mirror of
https://gitlab.winehq.org/wine/wine-gecko.git
synced 2024-09-13 09:24:08 -07:00
355 lines
11 KiB
C++
355 lines
11 KiB
C++
/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
|
* 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 "SharedPlanarYCbCrImage.h"
|
|
#include <stddef.h> // for size_t
|
|
#include <stdio.h> // for printf
|
|
#include "ISurfaceAllocator.h" // for ISurfaceAllocator, etc
|
|
#include "gfxPoint.h" // for gfxIntSize
|
|
#include "mozilla/Assertions.h" // for MOZ_ASSERT, etc
|
|
#include "mozilla/gfx/Types.h" // for SurfaceFormat::FORMAT_YUV
|
|
#include "mozilla/ipc/SharedMemory.h" // for SharedMemory, etc
|
|
#include "mozilla/layers/ImageClient.h" // for ImageClient
|
|
#include "mozilla/layers/LayersSurfaces.h" // for SurfaceDescriptor, etc
|
|
#include "mozilla/layers/TextureClient.h" // for BufferTextureClient, etc
|
|
#include "mozilla/layers/YCbCrImageDataSerializer.h"
|
|
#include "mozilla/mozalloc.h" // for operator delete
|
|
#include "nsISupportsImpl.h" // for Image::AddRef
|
|
|
|
class gfxASurface;
|
|
|
|
namespace mozilla {
|
|
namespace layers {
|
|
|
|
using namespace mozilla::ipc;
|
|
|
|
SharedPlanarYCbCrImage::SharedPlanarYCbCrImage(ImageClient* aCompositable)
|
|
: PlanarYCbCrImage(nullptr)
|
|
{
|
|
mTextureClient = aCompositable->CreateBufferTextureClient(gfx::FORMAT_YUV);
|
|
MOZ_COUNT_CTOR(SharedPlanarYCbCrImage);
|
|
}
|
|
|
|
SharedPlanarYCbCrImage::~SharedPlanarYCbCrImage() {
|
|
MOZ_COUNT_DTOR(SharedPlanarYCbCrImage);
|
|
}
|
|
|
|
|
|
DeprecatedSharedPlanarYCbCrImage::~DeprecatedSharedPlanarYCbCrImage() {
|
|
MOZ_COUNT_DTOR(DeprecatedSharedPlanarYCbCrImage);
|
|
|
|
if (mAllocated) {
|
|
SurfaceDescriptor desc;
|
|
DropToSurfaceDescriptor(desc);
|
|
mSurfaceAllocator->DestroySharedSurface(&desc);
|
|
}
|
|
}
|
|
|
|
TextureClient*
|
|
SharedPlanarYCbCrImage::GetTextureClient()
|
|
{
|
|
return mTextureClient.get();
|
|
}
|
|
|
|
uint8_t*
|
|
SharedPlanarYCbCrImage::GetBuffer()
|
|
{
|
|
return mTextureClient->GetBuffer();
|
|
}
|
|
|
|
already_AddRefed<gfxASurface>
|
|
SharedPlanarYCbCrImage::GetAsSurface()
|
|
{
|
|
if (!mTextureClient->IsAllocated()) {
|
|
NS_WARNING("Can't get as surface");
|
|
return nullptr;
|
|
}
|
|
return PlanarYCbCrImage::GetAsSurface();
|
|
}
|
|
|
|
void
|
|
SharedPlanarYCbCrImage::SetData(const PlanarYCbCrImage::Data& aData)
|
|
{
|
|
// If mShmem has not been allocated (through Allocate(aData)), allocate it.
|
|
// This code path is slower than the one used when Allocate has been called
|
|
// since it will trigger a full copy.
|
|
if (!mTextureClient->IsAllocated()) {
|
|
Data data = aData;
|
|
if (!Allocate(data)) {
|
|
printf("SharedPlanarYCbCrImage::SetData failed to allocate :(\n");
|
|
return;
|
|
}
|
|
}
|
|
|
|
MOZ_ASSERT(mTextureClient->AsTextureClientYCbCr());
|
|
|
|
if (!mTextureClient->AsTextureClientYCbCr()->UpdateYCbCr(aData)) {
|
|
MOZ_ASSERT(false, "Failed to copy YCbCr data into the TextureClient");
|
|
return;
|
|
}
|
|
|
|
// do not set mBuffer like in PlanarYCbCrImage because the later
|
|
// will try to manage this memory without knowing it belongs to a
|
|
// shmem.
|
|
mBufferSize = YCbCrImageDataSerializer::ComputeMinBufferSize(mData.mYSize,
|
|
mData.mCbCrSize);
|
|
mSize = mData.mPicSize;
|
|
|
|
YCbCrImageDataSerializer serializer(mTextureClient->GetBuffer());
|
|
mData.mYChannel = serializer.GetYData();
|
|
mData.mCbChannel = serializer.GetCbData();
|
|
mData.mCrChannel = serializer.GetCrData();
|
|
mTextureClient->MarkImmutable();
|
|
}
|
|
|
|
// needs to be overriden because the parent class sets mBuffer which we
|
|
// do not want to happen.
|
|
uint8_t*
|
|
SharedPlanarYCbCrImage::AllocateAndGetNewBuffer(uint32_t aSize)
|
|
{
|
|
NS_ABORT_IF_FALSE(!mTextureClient->IsAllocated(), "This image already has allocated data");
|
|
size_t size = YCbCrImageDataSerializer::ComputeMinBufferSize(aSize);
|
|
// update buffer size
|
|
mBufferSize = size;
|
|
|
|
// get new buffer _without_ setting mBuffer.
|
|
bool status = mTextureClient->Allocate(mBufferSize);
|
|
MOZ_ASSERT(status);
|
|
YCbCrImageDataSerializer serializer(mTextureClient->GetBuffer());
|
|
|
|
return serializer.GetData();
|
|
}
|
|
|
|
void
|
|
SharedPlanarYCbCrImage::SetDataNoCopy(const Data &aData)
|
|
{
|
|
mData = aData;
|
|
mSize = aData.mPicSize;
|
|
YCbCrImageDataSerializer serializer(mTextureClient->GetBuffer());
|
|
serializer.InitializeBufferInfo(aData.mYSize,
|
|
aData.mCbCrSize,
|
|
aData.mStereoMode);
|
|
}
|
|
|
|
uint8_t*
|
|
SharedPlanarYCbCrImage::AllocateBuffer(uint32_t aSize)
|
|
{
|
|
NS_ABORT_IF_FALSE(!mTextureClient->IsAllocated(),
|
|
"This image already has allocated data");
|
|
if (!mTextureClient->Allocate(aSize)) {
|
|
return nullptr;
|
|
}
|
|
return mTextureClient->GetBuffer();
|
|
}
|
|
|
|
bool
|
|
SharedPlanarYCbCrImage::IsValid() {
|
|
return mTextureClient->IsAllocated();
|
|
}
|
|
|
|
bool
|
|
SharedPlanarYCbCrImage::Allocate(PlanarYCbCrImage::Data& aData)
|
|
{
|
|
NS_ABORT_IF_FALSE(!mTextureClient->IsAllocated(),
|
|
"This image already has allocated data");
|
|
|
|
size_t size = YCbCrImageDataSerializer::ComputeMinBufferSize(aData.mYSize,
|
|
aData.mCbCrSize);
|
|
|
|
if (AllocateBuffer(static_cast<uint32_t>(size)) == nullptr) {
|
|
return false;
|
|
}
|
|
|
|
YCbCrImageDataSerializer serializer(mTextureClient->GetBuffer());
|
|
serializer.InitializeBufferInfo(aData.mYSize,
|
|
aData.mCbCrSize,
|
|
aData.mStereoMode);
|
|
MOZ_ASSERT(serializer.IsValid());
|
|
|
|
aData.mYChannel = serializer.GetYData();
|
|
aData.mCbChannel = serializer.GetCbData();
|
|
aData.mCrChannel = serializer.GetCrData();
|
|
|
|
// copy some of aData's values in mData (most of them)
|
|
mData.mYChannel = aData.mYChannel;
|
|
mData.mCbChannel = aData.mCbChannel;
|
|
mData.mCrChannel = aData.mCrChannel;
|
|
mData.mYSize = aData.mYSize;
|
|
mData.mCbCrSize = aData.mCbCrSize;
|
|
mData.mPicX = aData.mPicX;
|
|
mData.mPicY = aData.mPicY;
|
|
mData.mPicSize = aData.mPicSize;
|
|
mData.mStereoMode = aData.mStereoMode;
|
|
// those members are not always equal to aData's, due to potentially different
|
|
// packing.
|
|
mData.mYSkip = 0;
|
|
mData.mCbSkip = 0;
|
|
mData.mCrSkip = 0;
|
|
mData.mYStride = mData.mYSize.width;
|
|
mData.mCbCrStride = mData.mCbCrSize.width;
|
|
|
|
return true;
|
|
}
|
|
|
|
void
|
|
DeprecatedSharedPlanarYCbCrImage::SetData(const PlanarYCbCrImage::Data& aData)
|
|
{
|
|
// If mShmem has not been allocated (through Allocate(aData)), allocate it.
|
|
// This code path is slower than the one used when Allocate has been called
|
|
// since it will trigger a full copy.
|
|
if (!mAllocated) {
|
|
Data data = aData;
|
|
if (!Allocate(data)) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
// do not set mBuffer like in PlanarYCbCrImage because the later
|
|
// will try to manage this memory without knowing it belongs to a
|
|
// shmem.
|
|
mBufferSize = YCbCrImageDataSerializer::ComputeMinBufferSize(mData.mYSize,
|
|
mData.mCbCrSize);
|
|
mSize = mData.mPicSize;
|
|
|
|
YCbCrImageDataSerializer serializer(mShmem.get<uint8_t>());
|
|
MOZ_ASSERT(aData.mCbSkip == aData.mCrSkip);
|
|
if (!serializer.CopyData(aData.mYChannel, aData.mCbChannel, aData.mCrChannel,
|
|
aData.mYSize, aData.mYStride,
|
|
aData.mCbCrSize, aData.mCbCrStride,
|
|
aData.mYSkip, aData.mCbSkip)) {
|
|
NS_WARNING("Failed to copy image data!");
|
|
}
|
|
mData.mYChannel = serializer.GetYData();
|
|
mData.mCbChannel = serializer.GetCbData();
|
|
mData.mCrChannel = serializer.GetCrData();
|
|
}
|
|
|
|
// needs to be overriden because the parent class sets mBuffer which we
|
|
// do not want to happen.
|
|
uint8_t*
|
|
DeprecatedSharedPlanarYCbCrImage::AllocateAndGetNewBuffer(uint32_t aSize)
|
|
{
|
|
NS_ABORT_IF_FALSE(!mAllocated, "This image already has allocated data");
|
|
size_t size = YCbCrImageDataSerializer::ComputeMinBufferSize(aSize);
|
|
// update buffer size
|
|
mBufferSize = size;
|
|
|
|
// get new buffer _without_ setting mBuffer.
|
|
AllocateBuffer(mBufferSize);
|
|
YCbCrImageDataSerializer serializer(mShmem.get<uint8_t>());
|
|
|
|
return serializer.GetData();
|
|
}
|
|
|
|
|
|
void
|
|
DeprecatedSharedPlanarYCbCrImage::SetDataNoCopy(const Data &aData)
|
|
{
|
|
mData = aData;
|
|
mSize = aData.mPicSize;
|
|
YCbCrImageDataSerializer serializer(mShmem.get<uint8_t>());
|
|
serializer.InitializeBufferInfo(aData.mYSize,
|
|
aData.mCbCrSize,
|
|
aData.mStereoMode);
|
|
}
|
|
|
|
uint8_t*
|
|
DeprecatedSharedPlanarYCbCrImage::AllocateBuffer(uint32_t aSize)
|
|
{
|
|
NS_ABORT_IF_FALSE(!mAllocated, "This image already has allocated data");
|
|
SharedMemory::SharedMemoryType shmType = OptimalShmemType();
|
|
if (!mSurfaceAllocator->AllocUnsafeShmem(aSize, shmType, &mShmem)) {
|
|
return nullptr;
|
|
}
|
|
mAllocated = true;
|
|
return mShmem.get<uint8_t>();
|
|
}
|
|
|
|
|
|
bool
|
|
DeprecatedSharedPlanarYCbCrImage::Allocate(PlanarYCbCrImage::Data& aData)
|
|
{
|
|
NS_ABORT_IF_FALSE(!mAllocated, "This image already has allocated data");
|
|
|
|
size_t size = YCbCrImageDataSerializer::ComputeMinBufferSize(aData.mYSize,
|
|
aData.mCbCrSize);
|
|
|
|
if (AllocateBuffer(static_cast<uint32_t>(size)) == nullptr) {
|
|
return false;
|
|
}
|
|
|
|
YCbCrImageDataSerializer serializer(mShmem.get<uint8_t>());
|
|
serializer.InitializeBufferInfo(aData.mYSize,
|
|
aData.mCbCrSize,
|
|
aData.mStereoMode);
|
|
if (!serializer.IsValid() || mShmem.Size<uint8_t>() < size) {
|
|
mSurfaceAllocator->DeallocShmem(mShmem);
|
|
return false;
|
|
}
|
|
|
|
aData.mYChannel = serializer.GetYData();
|
|
aData.mCbChannel = serializer.GetCbData();
|
|
aData.mCrChannel = serializer.GetCrData();
|
|
|
|
// copy some of aData's values in mData (most of them)
|
|
mData.mYChannel = aData.mYChannel;
|
|
mData.mCbChannel = aData.mCbChannel;
|
|
mData.mCrChannel = aData.mCrChannel;
|
|
mData.mYSize = aData.mYSize;
|
|
mData.mCbCrSize = aData.mCbCrSize;
|
|
mData.mPicX = aData.mPicX;
|
|
mData.mPicY = aData.mPicY;
|
|
mData.mPicSize = aData.mPicSize;
|
|
mData.mStereoMode = aData.mStereoMode;
|
|
// those members are not always equal to aData's, due to potentially different
|
|
// packing.
|
|
mData.mYSkip = 0;
|
|
mData.mCbSkip = 0;
|
|
mData.mCrSkip = 0;
|
|
mData.mYStride = mData.mYSize.width;
|
|
mData.mCbCrStride = mData.mCbCrSize.width;
|
|
|
|
mAllocated = true;
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
DeprecatedSharedPlanarYCbCrImage::ToSurfaceDescriptor(SurfaceDescriptor& aDesc) {
|
|
if (!mAllocated) {
|
|
return false;
|
|
}
|
|
aDesc = YCbCrImage(mShmem, reinterpret_cast<uint64_t>(this));
|
|
this->AddRef();
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
DeprecatedSharedPlanarYCbCrImage::DropToSurfaceDescriptor(SurfaceDescriptor& aDesc) {
|
|
if (!mAllocated) {
|
|
return false;
|
|
}
|
|
aDesc = YCbCrImage(mShmem, 0);
|
|
mShmem = Shmem();
|
|
mAllocated = false;
|
|
return true;
|
|
}
|
|
|
|
DeprecatedSharedPlanarYCbCrImage*
|
|
DeprecatedSharedPlanarYCbCrImage::FromSurfaceDescriptor(const SurfaceDescriptor& aDescriptor)
|
|
{
|
|
if (aDescriptor.type() != SurfaceDescriptor::TYCbCrImage) {
|
|
return nullptr;
|
|
}
|
|
const YCbCrImage& ycbcr = aDescriptor.get_YCbCrImage();
|
|
if (ycbcr.owner() == 0) {
|
|
return nullptr;
|
|
}
|
|
return reinterpret_cast<DeprecatedSharedPlanarYCbCrImage*>(ycbcr.owner());
|
|
}
|
|
|
|
|
|
} // namespace
|
|
} // namespace
|