forked from mirrors/gecko-dev
a5eed5ff52
Make VideoFramePool thread safe to avoid multiple access during software decode to DMABuf: - Create Mutex for VideoFramePool access - Mark surface as used when it's provided by VideoFramePool to avoid race conditions. Differential Revision: https://phabricator.services.mozilla.com/D135557
118 lines
4 KiB
C++
118 lines
4 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "DMABUFSurfaceImage.h"
|
|
#include "mozilla/widget/DMABufSurface.h"
|
|
#include "mozilla/layers/CompositableClient.h"
|
|
#include "mozilla/layers/CompositableForwarder.h"
|
|
#include "mozilla/layers/DMABUFTextureClientOGL.h"
|
|
#include "mozilla/layers/TextureForwarder.h"
|
|
#include "mozilla/ScopeExit.h"
|
|
#include "mozilla/StaticMutex.h"
|
|
#include "GLContext.h"
|
|
#include "GLContextProvider.h"
|
|
#include "GLBlitHelper.h"
|
|
#include "GLReadTexImageHelper.h"
|
|
#include "GLContextTypes.h" // for GLContext, etc
|
|
#include "GLContextEGL.h"
|
|
#include "GLContextProvider.h"
|
|
#include "ScopedGLHelpers.h"
|
|
|
|
using namespace mozilla;
|
|
using namespace mozilla::layers;
|
|
using namespace mozilla::gfx;
|
|
using namespace mozilla::gl;
|
|
|
|
DMABUFSurfaceImage::DMABUFSurfaceImage(DMABufSurface* aSurface)
|
|
: Image(nullptr, ImageFormat::DMABUF), mSurface(aSurface) {
|
|
MOZ_DIAGNOSTIC_ASSERT(mSurface->IsGlobalRefSet(),
|
|
"DMABufSurface must be marked as used!");
|
|
}
|
|
|
|
DMABUFSurfaceImage::~DMABUFSurfaceImage() {
|
|
// Unref as we're done with this surface.
|
|
mSurface->GlobalRefRelease();
|
|
}
|
|
|
|
StaticRefPtr<GLContext> sSnapshotContext;
|
|
static StaticMutex sSnapshotContextMutex;
|
|
|
|
already_AddRefed<gfx::SourceSurface> DMABUFSurfaceImage::GetAsSourceSurface() {
|
|
if (!sSnapshotContext) {
|
|
nsCString discardFailureId;
|
|
sSnapshotContext = GLContextProvider::CreateHeadless({}, &discardFailureId);
|
|
if (!sSnapshotContext) {
|
|
gfxCriticalError() << "Failed to create snapshot GLContext.";
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
StaticMutexAutoLock lock(sSnapshotContextMutex);
|
|
sSnapshotContext->MakeCurrent();
|
|
|
|
auto releaseTextures =
|
|
mozilla::MakeScopeExit([&] { mSurface->ReleaseTextures(); });
|
|
|
|
for (int i = 0; i < mSurface->GetTextureCount(); i++) {
|
|
if (!mSurface->GetTexture(i) &&
|
|
!mSurface->CreateTexture(sSnapshotContext, i)) {
|
|
gfxCriticalError()
|
|
<< "GetAsSourceSurface: Failed to create DMABuf textures.";
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
ScopedTexture scopedTex(sSnapshotContext);
|
|
ScopedBindTexture boundTex(sSnapshotContext, scopedTex.Texture());
|
|
|
|
gfx::IntSize size = GetSize();
|
|
sSnapshotContext->fTexImage2D(LOCAL_GL_TEXTURE_2D, 0, LOCAL_GL_RGBA,
|
|
size.width, size.height, 0, LOCAL_GL_RGBA,
|
|
LOCAL_GL_UNSIGNED_BYTE, nullptr);
|
|
|
|
ScopedFramebufferForTexture autoFBForTex(sSnapshotContext,
|
|
scopedTex.Texture());
|
|
if (!autoFBForTex.IsComplete()) {
|
|
gfxCriticalError()
|
|
<< "GetAsSourceSurface: ScopedFramebufferForTexture failed.";
|
|
return nullptr;
|
|
}
|
|
|
|
const gl::OriginPos destOrigin = gl::OriginPos::BottomLeft;
|
|
{
|
|
const ScopedBindFramebuffer bindFB(sSnapshotContext, autoFBForTex.FB());
|
|
if (!sSnapshotContext->BlitHelper()->BlitImageToFramebuffer(this, size,
|
|
destOrigin)) {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
RefPtr<gfx::DataSourceSurface> source =
|
|
gfx::Factory::CreateDataSourceSurface(size, gfx::SurfaceFormat::B8G8R8A8);
|
|
if (NS_WARN_IF(!source)) {
|
|
return nullptr;
|
|
}
|
|
|
|
ScopedBindFramebuffer bind(sSnapshotContext, autoFBForTex.FB());
|
|
ReadPixelsIntoDataSurface(sSnapshotContext, source);
|
|
|
|
return source.forget();
|
|
}
|
|
|
|
TextureClient* DMABUFSurfaceImage::GetTextureClient(
|
|
KnowsCompositor* aKnowsCompositor) {
|
|
if (!mTextureClient) {
|
|
BackendType backend = BackendType::NONE;
|
|
mTextureClient = TextureClient::CreateWithData(
|
|
DMABUFTextureData::Create(mSurface, backend), TextureFlags::DEFAULT,
|
|
aKnowsCompositor->GetTextureForwarder());
|
|
}
|
|
return mTextureClient;
|
|
}
|
|
|
|
gfx::IntSize DMABUFSurfaceImage::GetSize() const {
|
|
return gfx::IntSize::Truncate(mSurface->GetWidth(), mSurface->GetHeight());
|
|
}
|