зеркало из https://github.com/mozilla/gecko-dev.git
393 строки
12 KiB
C++
393 строки
12 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 "D3D11YCbCrImage.h"
|
|
#include "YCbCrUtils.h"
|
|
#include "mozilla/gfx/gfxVars.h"
|
|
#include "mozilla/layers/CompositableClient.h"
|
|
#include "mozilla/layers/CompositableForwarder.h"
|
|
#include "mozilla/layers/TextureD3D11.h"
|
|
|
|
using namespace mozilla::gfx;
|
|
|
|
namespace mozilla {
|
|
namespace layers {
|
|
|
|
class DXGIYCbCrTextureAllocationHelper : public ITextureClientAllocationHelper
|
|
{
|
|
public:
|
|
DXGIYCbCrTextureAllocationHelper(const PlanarYCbCrData& aData,
|
|
TextureFlags aTextureFlags,
|
|
ID3D11Device* aDevice);
|
|
|
|
bool IsCompatible(TextureClient* aTextureClient) override;
|
|
|
|
already_AddRefed<TextureClient> Allocate(KnowsCompositor* aAllocator) override;
|
|
|
|
protected:
|
|
const PlanarYCbCrData& mData;
|
|
RefPtr<ID3D11Device> mDevice;
|
|
};
|
|
|
|
D3D11YCbCrImage::D3D11YCbCrImage()
|
|
: Image(NULL, ImageFormat::D3D11_YCBCR_IMAGE)
|
|
{
|
|
}
|
|
|
|
D3D11YCbCrImage::~D3D11YCbCrImage() { }
|
|
|
|
bool
|
|
D3D11YCbCrImage::SetData(KnowsCompositor* aAllocator,
|
|
ImageContainer* aContainer,
|
|
const PlanarYCbCrData& aData)
|
|
{
|
|
mPictureRect = IntRect(
|
|
aData.mPicX, aData.mPicY, aData.mPicSize.width, aData.mPicSize.height);
|
|
mYSize = aData.mYSize;
|
|
mCbCrSize = aData.mCbCrSize;
|
|
mColorSpace = aData.mYUVColorSpace;
|
|
|
|
D3D11YCbCrRecycleAllocator* allocator =
|
|
aContainer->GetD3D11YCbCrRecycleAllocator(aAllocator);
|
|
if (!allocator) {
|
|
return false;
|
|
}
|
|
|
|
{
|
|
DXGIYCbCrTextureAllocationHelper helper(aData, TextureFlags::DEFAULT, allocator->GetDevice());
|
|
mTextureClient = allocator->CreateOrRecycle(helper);
|
|
}
|
|
|
|
if (!mTextureClient) {
|
|
return false;
|
|
}
|
|
|
|
DXGIYCbCrTextureData* data =
|
|
mTextureClient->GetInternalData()->AsDXGIYCbCrTextureData();
|
|
|
|
ID3D11Texture2D* textureY = data->GetD3D11Texture(0);
|
|
ID3D11Texture2D* textureCb = data->GetD3D11Texture(1);
|
|
ID3D11Texture2D* textureCr = data->GetD3D11Texture(2);
|
|
|
|
RefPtr<ID3D10Multithread> mt;
|
|
HRESULT hr = allocator->GetDevice()->QueryInterface(
|
|
(ID3D10Multithread**)getter_AddRefs(mt));
|
|
|
|
if (FAILED(hr) || !mt) {
|
|
gfxCriticalError() << "Multithread safety interface not supported. " << hr;
|
|
return false;
|
|
}
|
|
|
|
if (!mt->GetMultithreadProtected()) {
|
|
gfxCriticalError() << "Device used not marked as multithread-safe.";
|
|
return false;
|
|
}
|
|
|
|
D3D11MTAutoEnter mtAutoEnter(mt.forget());
|
|
|
|
RefPtr<ID3D11DeviceContext> ctx;
|
|
allocator->GetDevice()->GetImmediateContext(getter_AddRefs(ctx));
|
|
if (!ctx) {
|
|
gfxCriticalError() << "Failed to get immediate context.";
|
|
return false;
|
|
}
|
|
|
|
AutoLockD3D11Texture lockY(textureY);
|
|
AutoLockD3D11Texture lockCb(textureCb);
|
|
AutoLockD3D11Texture lockCr(textureCr);
|
|
|
|
ctx->UpdateSubresource(textureY,
|
|
0,
|
|
nullptr,
|
|
aData.mYChannel,
|
|
aData.mYStride,
|
|
aData.mYStride * aData.mYSize.height);
|
|
ctx->UpdateSubresource(textureCb,
|
|
0,
|
|
nullptr,
|
|
aData.mCbChannel,
|
|
aData.mCbCrStride,
|
|
aData.mCbCrStride * aData.mCbCrSize.height);
|
|
ctx->UpdateSubresource(textureCr,
|
|
0,
|
|
nullptr,
|
|
aData.mCrChannel,
|
|
aData.mCbCrStride,
|
|
aData.mCbCrStride * aData.mCbCrSize.height);
|
|
|
|
|
|
return true;
|
|
}
|
|
|
|
IntSize
|
|
D3D11YCbCrImage::GetSize() const
|
|
{
|
|
return mPictureRect.Size();
|
|
}
|
|
|
|
TextureClient*
|
|
D3D11YCbCrImage::GetTextureClient(KnowsCompositor* aForwarder)
|
|
{
|
|
return mTextureClient;
|
|
}
|
|
|
|
const DXGIYCbCrTextureData*
|
|
D3D11YCbCrImage::GetData() const
|
|
{
|
|
if (!mTextureClient)
|
|
return nullptr;
|
|
|
|
return mTextureClient->GetInternalData()->AsDXGIYCbCrTextureData();
|
|
}
|
|
|
|
already_AddRefed<SourceSurface>
|
|
D3D11YCbCrImage::GetAsSourceSurface()
|
|
{
|
|
if (!mTextureClient) {
|
|
gfxWarning()
|
|
<< "GetAsSourceSurface() called on uninitialized D3D11YCbCrImage.";
|
|
return nullptr;
|
|
}
|
|
|
|
gfx::IntSize size(mPictureRect.Size());
|
|
gfx::SurfaceFormat format =
|
|
gfx::ImageFormatToSurfaceFormat(gfxVars::OffscreenFormat());
|
|
HRESULT hr;
|
|
|
|
PlanarYCbCrData data;
|
|
|
|
DXGIYCbCrTextureData* dxgiData =
|
|
mTextureClient->GetInternalData()->AsDXGIYCbCrTextureData();
|
|
|
|
if (!dxgiData) {
|
|
gfxCriticalError() << "Failed to get texture client internal data.";
|
|
return nullptr;
|
|
}
|
|
|
|
RefPtr<ID3D11Texture2D> texY = dxgiData->GetD3D11Texture(0);
|
|
RefPtr<ID3D11Texture2D> texCb = dxgiData->GetD3D11Texture(1);
|
|
RefPtr<ID3D11Texture2D> texCr = dxgiData->GetD3D11Texture(2);
|
|
RefPtr<ID3D11Texture2D> softTexY, softTexCb, softTexCr;
|
|
D3D11_TEXTURE2D_DESC desc;
|
|
|
|
RefPtr<ID3D11Device> dev;
|
|
texY->GetDevice(getter_AddRefs(dev));
|
|
|
|
RefPtr<ID3D10Multithread> mt;
|
|
hr = dev->QueryInterface((ID3D10Multithread**)getter_AddRefs(mt));
|
|
|
|
if (FAILED(hr) || !mt) {
|
|
gfxCriticalError() << "Multithread safety interface not supported.";
|
|
return nullptr;
|
|
}
|
|
|
|
if (!mt->GetMultithreadProtected()) {
|
|
gfxCriticalError() << "Device used not marked as multithread-safe.";
|
|
return nullptr;
|
|
}
|
|
|
|
D3D11MTAutoEnter mtAutoEnter(mt.forget());
|
|
|
|
texY->GetDesc(&desc);
|
|
desc.BindFlags = 0;
|
|
desc.MiscFlags = 0;
|
|
desc.CPUAccessFlags = D3D11_CPU_ACCESS_READ;
|
|
desc.Usage = D3D11_USAGE_STAGING;
|
|
|
|
dev->CreateTexture2D(&desc, nullptr, getter_AddRefs(softTexY));
|
|
|
|
texCb->GetDesc(&desc);
|
|
desc.BindFlags = 0;
|
|
desc.MiscFlags = 0;
|
|
desc.CPUAccessFlags = D3D11_CPU_ACCESS_READ;
|
|
desc.Usage = D3D11_USAGE_STAGING;
|
|
|
|
dev->CreateTexture2D(&desc, nullptr, getter_AddRefs(softTexCb));
|
|
|
|
texCr->GetDesc(&desc);
|
|
desc.BindFlags = 0;
|
|
desc.MiscFlags = 0;
|
|
desc.CPUAccessFlags = D3D11_CPU_ACCESS_READ;
|
|
desc.Usage = D3D11_USAGE_STAGING;
|
|
|
|
dev->CreateTexture2D(&desc, nullptr, getter_AddRefs(softTexCr));
|
|
|
|
RefPtr<ID3D11DeviceContext> ctx;
|
|
dev->GetImmediateContext(getter_AddRefs(ctx));
|
|
if (!ctx) {
|
|
gfxCriticalError() << "Failed to get immediate context.";
|
|
return nullptr;
|
|
}
|
|
|
|
{
|
|
AutoLockD3D11Texture lockY(texY);
|
|
AutoLockD3D11Texture lockCb(texCb);
|
|
AutoLockD3D11Texture lockCr(texCr);
|
|
ctx->CopyResource(softTexY, texY);
|
|
ctx->CopyResource(softTexCb, texCb);
|
|
ctx->CopyResource(softTexCr, texCr);
|
|
}
|
|
|
|
D3D11_MAPPED_SUBRESOURCE mapY, mapCb, mapCr;
|
|
RefPtr<gfx::DataSourceSurface> surface;
|
|
mapY.pData = mapCb.pData = mapCr.pData = nullptr;
|
|
|
|
hr = ctx->Map(softTexY, 0, D3D11_MAP_READ, 0, &mapY);
|
|
if (FAILED(hr)) {
|
|
gfxCriticalError() << "Failed to map Y plane (" << hr << ")";
|
|
return nullptr;
|
|
}
|
|
hr = ctx->Map(softTexCb, 0, D3D11_MAP_READ, 0, &mapCb);
|
|
if (FAILED(hr)) {
|
|
gfxCriticalError() << "Failed to map Y plane (" << hr << ")";
|
|
return nullptr;
|
|
}
|
|
hr = ctx->Map(softTexCr, 0, D3D11_MAP_READ, 0, &mapCr);
|
|
if (FAILED(hr)) {
|
|
gfxCriticalError() << "Failed to map Y plane (" << hr << ")";
|
|
return nullptr;
|
|
}
|
|
|
|
MOZ_ASSERT(mapCb.RowPitch == mapCr.RowPitch);
|
|
|
|
data.mPicX = mPictureRect.X();
|
|
data.mPicY = mPictureRect.Y();
|
|
data.mPicSize = mPictureRect.Size();
|
|
data.mStereoMode = StereoMode::MONO;
|
|
data.mYUVColorSpace = mColorSpace;
|
|
data.mYSkip = data.mCbSkip = data.mCrSkip = 0;
|
|
data.mYSize = mYSize;
|
|
data.mCbCrSize = mCbCrSize;
|
|
data.mYChannel = static_cast<uint8_t*>(mapY.pData);
|
|
data.mYStride = mapY.RowPitch;
|
|
data.mCbChannel = static_cast<uint8_t*>(mapCb.pData);
|
|
data.mCrChannel = static_cast<uint8_t*>(mapCr.pData);
|
|
data.mCbCrStride = mapCb.RowPitch;
|
|
|
|
gfx::GetYCbCrToRGBDestFormatAndSize(data, format, size);
|
|
if (size.width > PlanarYCbCrImage::MAX_DIMENSION ||
|
|
size.height > PlanarYCbCrImage::MAX_DIMENSION) {
|
|
gfxCriticalError() << "Illegal image dest width or height";
|
|
return nullptr;
|
|
}
|
|
|
|
surface = gfx::Factory::CreateDataSourceSurface(size, format);
|
|
if (!surface) {
|
|
gfxCriticalError() << "Failed to create DataSourceSurface for image: "
|
|
<< size << " " << format;
|
|
return nullptr;
|
|
}
|
|
|
|
DataSourceSurface::ScopedMap mapping(surface, DataSourceSurface::WRITE);
|
|
if (!mapping.IsMapped()) {
|
|
gfxCriticalError() << "Failed to map DataSourceSurface for D3D11YCbCrImage";
|
|
return nullptr;
|
|
}
|
|
|
|
gfx::ConvertYCbCrToRGB(
|
|
data, format, size, mapping.GetData(), mapping.GetStride());
|
|
|
|
ctx->Unmap(softTexY, 0);
|
|
ctx->Unmap(softTexCb, 0);
|
|
ctx->Unmap(softTexCr, 0);
|
|
|
|
return surface.forget();
|
|
}
|
|
|
|
DXGIYCbCrTextureAllocationHelper::DXGIYCbCrTextureAllocationHelper(const PlanarYCbCrData& aData,
|
|
TextureFlags aTextureFlags,
|
|
ID3D11Device* aDevice)
|
|
: ITextureClientAllocationHelper(gfx::SurfaceFormat::YUV,
|
|
aData.mYSize,
|
|
BackendSelector::Content,
|
|
aTextureFlags,
|
|
ALLOC_DEFAULT)
|
|
, mData(aData)
|
|
, mDevice(aDevice)
|
|
{
|
|
}
|
|
|
|
bool
|
|
DXGIYCbCrTextureAllocationHelper::IsCompatible(TextureClient* aTextureClient)
|
|
{
|
|
MOZ_ASSERT(aTextureClient->GetFormat() == gfx::SurfaceFormat::YUV);
|
|
|
|
DXGIYCbCrTextureData* dxgiData = aTextureClient->GetInternalData()->AsDXGIYCbCrTextureData();
|
|
if (!dxgiData ||
|
|
aTextureClient->GetSize() != mData.mYSize ||
|
|
dxgiData->GetYSize() != mData.mYSize ||
|
|
dxgiData->GetCbCrSize() != mData.mCbCrSize ||
|
|
dxgiData->GetYUVColorSpace() != mData.mYUVColorSpace) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
already_AddRefed<TextureClient>
|
|
DXGIYCbCrTextureAllocationHelper::Allocate(KnowsCompositor* aAllocator)
|
|
{
|
|
CD3D11_TEXTURE2D_DESC newDesc(DXGI_FORMAT_R8_UNORM, mData.mYSize.width, mData.mYSize.height,
|
|
1, 1);
|
|
newDesc.MiscFlags = D3D11_RESOURCE_MISC_SHARED_KEYEDMUTEX;
|
|
|
|
RefPtr<ID3D10Multithread> mt;
|
|
HRESULT hr = mDevice->QueryInterface(
|
|
(ID3D10Multithread**)getter_AddRefs(mt));
|
|
|
|
if (FAILED(hr) || !mt) {
|
|
gfxCriticalError() << "Multithread safety interface not supported. " << hr;
|
|
return nullptr;
|
|
}
|
|
|
|
if (!mt->GetMultithreadProtected()) {
|
|
gfxCriticalError() << "Device used not marked as multithread-safe.";
|
|
return nullptr;
|
|
}
|
|
|
|
D3D11MTAutoEnter mtAutoEnter(mt.forget());
|
|
|
|
RefPtr<ID3D11Texture2D> textureY;
|
|
hr = mDevice->CreateTexture2D(&newDesc, nullptr, getter_AddRefs(textureY));
|
|
NS_ENSURE_TRUE(SUCCEEDED(hr), nullptr);
|
|
|
|
newDesc.Width = mData.mCbCrSize.width;
|
|
newDesc.Height = mData.mCbCrSize.height;
|
|
|
|
RefPtr<ID3D11Texture2D> textureCb;
|
|
hr = mDevice->CreateTexture2D(&newDesc, nullptr, getter_AddRefs(textureCb));
|
|
NS_ENSURE_TRUE(SUCCEEDED(hr), nullptr);
|
|
|
|
RefPtr<ID3D11Texture2D> textureCr;
|
|
hr = mDevice->CreateTexture2D(&newDesc, nullptr, getter_AddRefs(textureCr));
|
|
NS_ENSURE_TRUE(SUCCEEDED(hr), nullptr);
|
|
|
|
return TextureClient::CreateWithData(
|
|
DXGIYCbCrTextureData::Create(
|
|
textureY,
|
|
textureCb,
|
|
textureCr,
|
|
mData.mYSize,
|
|
mData.mYSize,
|
|
mData.mCbCrSize,
|
|
mData.mYUVColorSpace),
|
|
mTextureFlags,
|
|
aAllocator->GetTextureForwarder());
|
|
}
|
|
|
|
already_AddRefed<TextureClient>
|
|
D3D11YCbCrRecycleAllocator::Allocate(SurfaceFormat aFormat,
|
|
IntSize aSize,
|
|
BackendSelector aSelector,
|
|
TextureFlags aTextureFlags,
|
|
TextureAllocationFlags aAllocFlags)
|
|
{
|
|
MOZ_ASSERT_UNREACHABLE("unexpected to be called");
|
|
return nullptr;
|
|
}
|
|
|
|
} // namespace layers
|
|
} // namespace mozilla
|