Emilio Cobos Álvarez
02d691fdef
Autogenerated with:
cp dom/.clang-format gfx/ && mach format gfx/**.{cpp,h,mm}
Manual changes:
* OSVRSession headers are not system headers and they are not
standalone / rely on stdint, so tweaked to preserve previous
ordering.
* Missing include in GLDefs.h
* Missing include in gfxOTSUtils.
* Need to keep the windows header order in DCLayerTree.
* missing hb_font include in MockScaledFont.h
Differential Revision: https://phabricator.services.mozilla.com/D311695
295 lines
11 KiB
C++
295 lines
11 KiB
C++
/* 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 "MacIOSurfaceHelpers.h"
|
|
|
|
#include "YCbCrUtils.h"
|
|
#include "libyuv.h"
|
|
#include "mozilla/ScopeExit.h"
|
|
#include "mozilla/gfx/MacIOSurface.h"
|
|
|
|
namespace mozilla {
|
|
|
|
using namespace gfx;
|
|
|
|
namespace layers {
|
|
|
|
#define ALIGNED_32(x) ((x + 31) & ~31)
|
|
#define ALIGNEDPTR_32(x) \
|
|
reinterpret_cast<uint8_t*>((reinterpret_cast<uintptr_t>(x) + 31) & ~31)
|
|
|
|
static nsresult CopyFromLockedMacIOSurface(MacIOSurface* aSurface,
|
|
uint8_t* aData, int32_t aStride,
|
|
const IntSize& aSize,
|
|
SurfaceFormat aFormat) {
|
|
size_t bytesPerRow = aSurface->GetBytesPerRow();
|
|
SurfaceFormat ioFormat = aSurface->GetFormat();
|
|
|
|
if ((ioFormat == SurfaceFormat::NV12 || ioFormat == SurfaceFormat::NV16 ||
|
|
ioFormat == SurfaceFormat::YUY2 || ioFormat == SurfaceFormat::P010 ||
|
|
ioFormat == SurfaceFormat::P210) &&
|
|
(aSize.width > PlanarYCbCrImage::MAX_DIMENSION ||
|
|
aSize.height > PlanarYCbCrImage::MAX_DIMENSION)) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
if (ioFormat == SurfaceFormat::NV12 || ioFormat == SurfaceFormat::NV16) {
|
|
/* Extract and separate the CbCr planes */
|
|
size_t cbCrStride = aSurface->GetBytesPerRow(1);
|
|
size_t cbCrWidth = aSurface->GetDevicePixelWidth(1);
|
|
size_t cbCrHeight = aSurface->GetDevicePixelHeight(1);
|
|
|
|
auto cbPlane = MakeUnique<uint8_t[]>(cbCrWidth * cbCrHeight);
|
|
auto crPlane = MakeUnique<uint8_t[]>(cbCrWidth * cbCrHeight);
|
|
|
|
uint8_t* src = (uint8_t*)aSurface->GetBaseAddressOfPlane(1);
|
|
uint8_t* cbDest = cbPlane.get();
|
|
uint8_t* crDest = crPlane.get();
|
|
|
|
for (size_t i = 0; i < cbCrHeight; i++) {
|
|
uint8_t* rowSrc = src + cbCrStride * i;
|
|
for (size_t j = 0; j < cbCrWidth; j++) {
|
|
*cbDest = *rowSrc;
|
|
cbDest++;
|
|
rowSrc++;
|
|
*crDest = *rowSrc;
|
|
crDest++;
|
|
rowSrc++;
|
|
}
|
|
}
|
|
|
|
/* Convert to RGB */
|
|
PlanarYCbCrData data;
|
|
data.mYChannel = (uint8_t*)aSurface->GetBaseAddressOfPlane(0);
|
|
data.mYStride = aSurface->GetBytesPerRow(0);
|
|
data.mCbChannel = cbPlane.get();
|
|
data.mCrChannel = crPlane.get();
|
|
data.mCbCrStride = cbCrWidth;
|
|
data.mPictureRect = IntRect(IntPoint(0, 0), aSize);
|
|
data.mYUVColorSpace = aSurface->GetYUVColorSpace();
|
|
data.mColorPrimaries = aSurface->mColorPrimaries;
|
|
data.mColorRange = aSurface->IsFullRange() ? gfx::ColorRange::FULL
|
|
: gfx::ColorRange::LIMITED;
|
|
data.mChromaSubsampling = ioFormat == SurfaceFormat::NV12
|
|
? ChromaSubsampling::HALF_WIDTH_AND_HEIGHT
|
|
: ChromaSubsampling::HALF_WIDTH;
|
|
|
|
nsresult result =
|
|
ConvertYCbCrToRGB(data, SurfaceFormat::B8G8R8X8, aSize, aData, aStride);
|
|
MOZ_ASSERT(NS_SUCCEEDED(result), "Failed to convert YUV into RGB data");
|
|
return result;
|
|
}
|
|
|
|
if (ioFormat == SurfaceFormat::P010 || ioFormat == SurfaceFormat::P210) {
|
|
// P010 (4:2:0) and P210 (4:2:2) store 10-bit values in the high bits of
|
|
// each uint16_t (shifted left by 6 via safeShift10BitBy6 on write).
|
|
// We de-interleave the CbCr plane and pass ColorDepth::COLOR_16 so that
|
|
// ConvertYCbCrToRGB uses scale=256, which correctly maps the high-bit
|
|
// storage to 8-bit output.
|
|
size_t cbCrStride = aSurface->GetBytesPerRow(1);
|
|
size_t cbCrWidth = aSurface->GetDevicePixelWidth(1);
|
|
size_t cbCrHeight = aSurface->GetDevicePixelHeight(1);
|
|
|
|
auto cbPlane = MakeUnique<uint16_t[]>(cbCrWidth * cbCrHeight);
|
|
auto crPlane = MakeUnique<uint16_t[]>(cbCrWidth * cbCrHeight);
|
|
|
|
uint8_t* src = (uint8_t*)aSurface->GetBaseAddressOfPlane(1);
|
|
uint16_t* cbDest = cbPlane.get();
|
|
uint16_t* crDest = crPlane.get();
|
|
|
|
for (size_t i = 0; i < cbCrHeight; i++) {
|
|
uint16_t* rowSrc = reinterpret_cast<uint16_t*>(src + cbCrStride * i);
|
|
for (size_t j = 0; j < cbCrWidth; j++) {
|
|
*cbDest++ = *rowSrc++;
|
|
*crDest++ = *rowSrc++;
|
|
}
|
|
}
|
|
|
|
PlanarYCbCrData data;
|
|
data.mYChannel = (uint8_t*)aSurface->GetBaseAddressOfPlane(0);
|
|
data.mYStride = aSurface->GetBytesPerRow(0);
|
|
data.mCbChannel = (uint8_t*)cbPlane.get();
|
|
data.mCrChannel = (uint8_t*)crPlane.get();
|
|
data.mCbCrStride = cbCrWidth * sizeof(uint16_t);
|
|
data.mPictureRect = IntRect(IntPoint(0, 0), aSize);
|
|
data.mYUVColorSpace = aSurface->GetYUVColorSpace();
|
|
data.mColorPrimaries = aSurface->mColorPrimaries;
|
|
data.mColorRange = aSurface->IsFullRange() ? gfx::ColorRange::FULL
|
|
: gfx::ColorRange::LIMITED;
|
|
data.mChromaSubsampling = (ioFormat == SurfaceFormat::P010)
|
|
? ChromaSubsampling::HALF_WIDTH_AND_HEIGHT
|
|
: ChromaSubsampling::HALF_WIDTH;
|
|
data.mColorDepth = ColorDepth::COLOR_16;
|
|
|
|
nsresult result =
|
|
ConvertYCbCrToRGB(data, SurfaceFormat::B8G8R8X8, aSize, aData, aStride);
|
|
MOZ_ASSERT(NS_SUCCEEDED(result), "Failed to convert YUV into RGB data");
|
|
return result;
|
|
}
|
|
|
|
if (ioFormat == SurfaceFormat::YUY2) {
|
|
if (aSize.width == ALIGNED_32(aSize.width)) {
|
|
// Optimization when width is aligned to 32.
|
|
libyuv::ConvertToARGB((uint8_t*)aSurface->GetBaseAddress(),
|
|
0 /* not used */, aData, aStride, 0, 0, aSize.width,
|
|
aSize.height, aSize.width, aSize.height,
|
|
libyuv::kRotate0, libyuv::FOURCC_YUYV);
|
|
} else {
|
|
/* Convert to YV16 */
|
|
size_t cbCrWidth = (aSize.width + 1) >> 1;
|
|
size_t cbCrHeight = aSize.height;
|
|
// Ensure our stride is a multiple of 32 to allow for memory aligned rows.
|
|
size_t cbCrStride = ALIGNED_32(cbCrWidth);
|
|
size_t strideDelta = cbCrStride - cbCrWidth;
|
|
MOZ_ASSERT(strideDelta <= 31);
|
|
|
|
auto yPlane = MakeUnique<uint8_t[]>(cbCrStride * 2 * aSize.height + 31);
|
|
auto cbPlane = MakeUnique<uint8_t[]>(cbCrStride * cbCrHeight + 31);
|
|
auto crPlane = MakeUnique<uint8_t[]>(cbCrStride * cbCrHeight + 31);
|
|
|
|
uint8_t* src = (uint8_t*)aSurface->GetBaseAddress();
|
|
uint8_t* yDest = ALIGNEDPTR_32(yPlane.get());
|
|
uint8_t* cbDest = ALIGNEDPTR_32(cbPlane.get());
|
|
uint8_t* crDest = ALIGNEDPTR_32(crPlane.get());
|
|
|
|
for (int32_t i = 0; i < aSize.height; i++) {
|
|
uint8_t* rowSrc = src + bytesPerRow * i;
|
|
for (size_t j = 0; j < cbCrWidth; j++) {
|
|
*yDest = *rowSrc;
|
|
yDest++;
|
|
rowSrc++;
|
|
*cbDest = *rowSrc;
|
|
cbDest++;
|
|
rowSrc++;
|
|
*yDest = *rowSrc;
|
|
yDest++;
|
|
rowSrc++;
|
|
*crDest = *rowSrc;
|
|
crDest++;
|
|
rowSrc++;
|
|
}
|
|
if (strideDelta) {
|
|
cbDest += strideDelta;
|
|
crDest += strideDelta;
|
|
yDest += strideDelta << 1;
|
|
}
|
|
}
|
|
|
|
/* Convert to RGB */
|
|
PlanarYCbCrData data;
|
|
data.mYChannel = ALIGNEDPTR_32(yPlane.get());
|
|
data.mYStride = cbCrStride * 2;
|
|
data.mCbChannel = ALIGNEDPTR_32(cbPlane.get());
|
|
data.mCrChannel = ALIGNEDPTR_32(crPlane.get());
|
|
data.mCbCrStride = cbCrStride;
|
|
data.mPictureRect = IntRect(IntPoint(0, 0), aSize);
|
|
data.mYUVColorSpace = aSurface->GetYUVColorSpace();
|
|
data.mColorPrimaries = aSurface->mColorPrimaries;
|
|
data.mColorRange = aSurface->IsFullRange() ? gfx::ColorRange::FULL
|
|
: gfx::ColorRange::LIMITED;
|
|
data.mChromaSubsampling = ChromaSubsampling::HALF_WIDTH;
|
|
|
|
nsresult result = ConvertYCbCrToRGB(data, SurfaceFormat::B8G8R8X8, aSize,
|
|
aData, aStride);
|
|
MOZ_ASSERT(NS_SUCCEEDED(result), "Failed to convert YUV into RGB data");
|
|
return result;
|
|
}
|
|
} else {
|
|
unsigned char* ioData = (unsigned char*)aSurface->GetBaseAddress();
|
|
|
|
for (int32_t i = 0; i < aSize.height; ++i) {
|
|
memcpy(aData + i * aStride, ioData + i * bytesPerRow, aSize.width * 4);
|
|
}
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
already_AddRefed<SourceSurface> CreateSourceSurfaceFromMacIOSurface(
|
|
MacIOSurface* aSurface, gfx::DataSourceSurface* aDataSurface) {
|
|
if (NS_WARN_IF(!aSurface->Lock())) {
|
|
return nullptr;
|
|
}
|
|
|
|
auto scopeExit = MakeScopeExit([&]() { aSurface->Unlock(); });
|
|
|
|
size_t ioWidth = aSurface->GetDevicePixelWidth();
|
|
size_t ioHeight = aSurface->GetDevicePixelHeight();
|
|
IntSize size((int32_t)ioWidth, (int32_t)ioHeight);
|
|
SurfaceFormat ioFormat = aSurface->GetFormat();
|
|
|
|
SurfaceFormat format =
|
|
(ioFormat == SurfaceFormat::NV12 || ioFormat == SurfaceFormat::NV16 ||
|
|
ioFormat == SurfaceFormat::YUY2 || ioFormat == SurfaceFormat::P010 ||
|
|
ioFormat == SurfaceFormat::P210)
|
|
? SurfaceFormat::B8G8R8X8
|
|
: SurfaceFormat::B8G8R8A8;
|
|
|
|
RefPtr<DataSourceSurface> dataSurface;
|
|
if (aDataSurface) {
|
|
MOZ_ASSERT(aDataSurface->GetSize() == size);
|
|
MOZ_ASSERT(aDataSurface->GetFormat() == format);
|
|
if (aDataSurface->GetSize() == size &&
|
|
aDataSurface->GetFormat() == format) {
|
|
dataSurface = aDataSurface;
|
|
}
|
|
}
|
|
|
|
if (!dataSurface) {
|
|
dataSurface = Factory::CreateDataSourceSurface(size, format);
|
|
if (NS_WARN_IF(!dataSurface)) {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
DataSourceSurface::ScopedMap map(dataSurface, DataSourceSurface::WRITE);
|
|
if (NS_WARN_IF(!map.IsMapped())) {
|
|
return nullptr;
|
|
}
|
|
|
|
nsresult rv = CopyFromLockedMacIOSurface(aSurface, map.GetData(),
|
|
map.GetStride(), size, format);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return nullptr;
|
|
}
|
|
|
|
return dataSurface.forget();
|
|
}
|
|
|
|
nsresult CreateSurfaceDescriptorBufferFromMacIOSurface(
|
|
MacIOSurface* aSurface, SurfaceDescriptorBuffer& aSdBuffer,
|
|
Image::BuildSdbFlags aFlags,
|
|
const std::function<MemoryOrShmem(uint32_t)>& aAllocate) {
|
|
if (NS_WARN_IF(!aSurface->Lock())) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
auto scopeExit = MakeScopeExit([&]() { aSurface->Unlock(); });
|
|
|
|
size_t ioWidth = aSurface->GetDevicePixelWidth();
|
|
size_t ioHeight = aSurface->GetDevicePixelHeight();
|
|
IntSize size((int32_t)ioWidth, (int32_t)ioHeight);
|
|
SurfaceFormat ioFormat = aSurface->GetFormat();
|
|
|
|
SurfaceFormat format =
|
|
(ioFormat == SurfaceFormat::NV12 || ioFormat == SurfaceFormat::NV16 ||
|
|
ioFormat == SurfaceFormat::YUY2 || ioFormat == SurfaceFormat::P010 ||
|
|
ioFormat == SurfaceFormat::P210)
|
|
? SurfaceFormat::B8G8R8X8
|
|
: SurfaceFormat::B8G8R8A8;
|
|
|
|
uint8_t* buffer = nullptr;
|
|
int32_t stride = 0;
|
|
nsresult rv = Image::AllocateSurfaceDescriptorBufferRgb(
|
|
size, format, buffer, aSdBuffer, stride, aAllocate);
|
|
if (NS_WARN_IF(NS_FAILED(rv))) {
|
|
return rv;
|
|
}
|
|
|
|
return CopyFromLockedMacIOSurface(aSurface, buffer, stride, size, format);
|
|
}
|
|
|
|
} // namespace layers
|
|
} // namespace mozilla
|