gecko-dev/gfx/webrender_bindings/RenderBufferTextureHost.cpp

156 строки
5.8 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 "RenderBufferTextureHost.h"
#include "mozilla/gfx/Logging.h"
#include "mozilla/layers/ImageDataSerializer.h"
namespace mozilla {
namespace wr {
RenderBufferTextureHost::RenderBufferTextureHost(uint8_t* aBuffer,
const layers::BufferDescriptor& aDescriptor)
: mBuffer(aBuffer)
, mDescriptor(aDescriptor)
, mMap()
, mYMap()
, mCbMap()
, mCrMap()
, mLocked(false)
{
MOZ_COUNT_CTOR_INHERITED(RenderBufferTextureHost, RenderTextureHost);
switch (mDescriptor.type()) {
case layers::BufferDescriptor::TYCbCrDescriptor: {
const layers::YCbCrDescriptor& ycbcr = mDescriptor.get_YCbCrDescriptor();
mSize = ycbcr.ySize();
mFormat = gfx::SurfaceFormat::YUV;
break;
}
case layers::BufferDescriptor::TRGBDescriptor: {
const layers::RGBDescriptor& rgb = mDescriptor.get_RGBDescriptor();
mSize = rgb.size();
mFormat = rgb.format();
break;
}
default:
gfxCriticalError() << "Bad buffer host descriptor " << (int)mDescriptor.type();
MOZ_CRASH("GFX: Bad descriptor");
}
}
RenderBufferTextureHost::~RenderBufferTextureHost()
{
MOZ_COUNT_DTOR_INHERITED(RenderBufferTextureHost, RenderTextureHost);
}
wr::WrExternalImage
RenderBufferTextureHost::Lock(uint8_t aChannelIndex,
gl::GLContext* aGL,
wr::ImageRendering aRendering)
{
if (!mLocked) {
if (!GetBuffer()) {
// We hit some problems to get the shmem.
return InvalidToWrExternalImage();
}
if (mFormat != gfx::SurfaceFormat::YUV) {
mSurface = gfx::Factory::CreateWrappingDataSourceSurface(GetBuffer(),
layers::ImageDataSerializer::GetRGBStride(mDescriptor.get_RGBDescriptor()),
mSize,
mFormat);
if (NS_WARN_IF(!mSurface)) {
return InvalidToWrExternalImage();
}
if (NS_WARN_IF(!mSurface->Map(gfx::DataSourceSurface::MapType::READ_WRITE, &mMap))) {
mSurface = nullptr;
return InvalidToWrExternalImage();
}
} else {
const layers::YCbCrDescriptor& desc = mDescriptor.get_YCbCrDescriptor();
mYSurface = gfx::Factory::CreateWrappingDataSourceSurface(layers::ImageDataSerializer::GetYChannel(GetBuffer(), desc),
desc.yStride(),
desc.ySize(),
gfx::SurfaceFormat::A8);
mCbSurface = gfx::Factory::CreateWrappingDataSourceSurface(layers::ImageDataSerializer::GetCbChannel(GetBuffer(), desc),
desc.cbCrStride(),
desc.cbCrSize(),
gfx::SurfaceFormat::A8);
mCrSurface = gfx::Factory::CreateWrappingDataSourceSurface(layers::ImageDataSerializer::GetCrChannel(GetBuffer(), desc),
desc.cbCrStride(),
desc.cbCrSize(),
gfx::SurfaceFormat::A8);
if (NS_WARN_IF(!mYSurface || !mCbSurface || !mCrSurface)) {
mYSurface = mCbSurface = mCrSurface = nullptr;
return InvalidToWrExternalImage();
}
if (NS_WARN_IF(!mYSurface->Map(gfx::DataSourceSurface::MapType::READ_WRITE, &mYMap) ||
!mCbSurface->Map(gfx::DataSourceSurface::MapType::READ_WRITE, &mCbMap) ||
!mCrSurface->Map(gfx::DataSourceSurface::MapType::READ_WRITE, &mCrMap))) {
mYSurface = mCbSurface = mCrSurface = nullptr;
return InvalidToWrExternalImage();
}
}
mLocked = true;
}
RenderBufferData data = GetBufferDataForRender(aChannelIndex);
return RawDataToWrExternalImage(data.mData, data.mBufferSize);
}
void
RenderBufferTextureHost::Unlock()
{
if (mLocked) {
if (mSurface) {
mSurface->Unmap();
mSurface = nullptr;
} else if (mYSurface) {
mYSurface->Unmap();
mCbSurface->Unmap();
mCrSurface->Unmap();
mYSurface = mCbSurface = mCrSurface = nullptr;
}
mLocked = false;
}
}
RenderBufferTextureHost::RenderBufferData
RenderBufferTextureHost::GetBufferDataForRender(uint8_t aChannelIndex)
{
MOZ_ASSERT(mFormat != gfx::SurfaceFormat::YUV || aChannelIndex < 3);
MOZ_ASSERT(mFormat == gfx::SurfaceFormat::YUV || aChannelIndex < 1);
MOZ_ASSERT(mLocked);
if (mFormat != gfx::SurfaceFormat::YUV) {
MOZ_ASSERT(mSurface);
return RenderBufferData(mMap.mData, mMap.mStride * mSurface->GetSize().height);
} else {
MOZ_ASSERT(mYSurface && mCbSurface && mCrSurface);
switch (aChannelIndex) {
case 0:
return RenderBufferData(mYMap.mData, mYMap.mStride * mYSurface->GetSize().height);
break;
case 1:
return RenderBufferData(mCbMap.mData, mCbMap.mStride * mCbSurface->GetSize().height);
break;
case 2:
return RenderBufferData(mCrMap.mData, mCrMap.mStride * mCrSurface->GetSize().height);
break;
default:
MOZ_ASSERT_UNREACHABLE("unexpected to be called");
return RenderBufferData(nullptr, 0);
}
}
}
} // namespace wr
} // namespace mozilla