diff options
Diffstat (limited to 'gfx/layers/MacIOSurfaceHelpers.cpp')
| -rw-r--r-- | gfx/layers/MacIOSurfaceHelpers.cpp | 176 |
1 files changed, 176 insertions, 0 deletions
diff --git a/gfx/layers/MacIOSurfaceHelpers.cpp b/gfx/layers/MacIOSurfaceHelpers.cpp new file mode 100644 index 000000000..4ca32427a --- /dev/null +++ b/gfx/layers/MacIOSurfaceHelpers.cpp @@ -0,0 +1,176 @@ +/* -*- 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 "libyuv.h" +#include "MacIOSurfaceHelpers.h" +#include "mozilla/gfx/MacIOSurface.h" +#include "YCbCrUtils.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 already_AddRefed<SourceSurface> +CreateSourceSurfaceFromLockedMacIOSurface(MacIOSurface* aSurface) +{ + size_t bytesPerRow = aSurface->GetBytesPerRow(); + size_t ioWidth = aSurface->GetDevicePixelWidth(); + size_t ioHeight = aSurface->GetDevicePixelHeight(); + SurfaceFormat ioFormat = aSurface->GetFormat(); + + if ((ioFormat == SurfaceFormat::NV12 || ioFormat == SurfaceFormat::YUV422) && + (ioWidth > PlanarYCbCrImage::MAX_DIMENSION || + ioHeight > PlanarYCbCrImage::MAX_DIMENSION)) { + return nullptr; + } + + SurfaceFormat format = + (ioFormat == SurfaceFormat::NV12 || ioFormat == SurfaceFormat::YUV422) + ? SurfaceFormat::B8G8R8X8 + : SurfaceFormat::B8G8R8A8; + + RefPtr<DataSourceSurface> dataSurface = + Factory::CreateDataSourceSurface(IntSize::Truncate(ioWidth, ioHeight), format); + if (NS_WARN_IF(!dataSurface)) { + return nullptr; + } + + DataSourceSurface::MappedSurface mappedSurface; + if (!dataSurface->Map(DataSourceSurface::WRITE, &mappedSurface)) { + return nullptr; + } + + if (ioFormat == SurfaceFormat::NV12) { + /* 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.mYSize = IntSize::Truncate(ioWidth, ioHeight); + data.mCbChannel = cbPlane.get(); + data.mCrChannel = crPlane.get(); + data.mCbCrStride = cbCrWidth; + data.mCbCrSize = IntSize::Truncate(cbCrWidth, cbCrHeight); + data.mPicSize = data.mYSize; + + ConvertYCbCrToRGB(data, SurfaceFormat::B8G8R8X8, IntSize::Truncate(ioWidth, ioHeight), mappedSurface.mData, mappedSurface.mStride); + } else if (ioFormat == SurfaceFormat::YUV422) { + if (ioWidth == ALIGNED_32(ioWidth)) { + // Optimization when width is aligned to 32. + IntSize size = IntSize::Truncate(ioWidth, ioHeight); + libyuv::ConvertToARGB((uint8_t*)aSurface->GetBaseAddress(), 0 /* not used */, + mappedSurface.mData, mappedSurface.mStride, + 0, 0, + size.width, size.height, + size.width, size.height, + libyuv::kRotate0, libyuv::FOURCC_UYVY); + } else { + /* Convert to YV16 */ + size_t cbCrWidth = (ioWidth+1)>>1; + size_t cbCrHeight = ioHeight; + // 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 * ioHeight + 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 (size_t i = 0; i < ioHeight; i++) { + uint8_t* rowSrc = src + bytesPerRow * i; + for (size_t j = 0; j < cbCrWidth; j++) { + *cbDest = *rowSrc; + cbDest++; + rowSrc++; + *yDest = *rowSrc; + yDest++; + rowSrc++; + *crDest = *rowSrc; + crDest++; + rowSrc++; + *yDest = *rowSrc; + yDest++; + 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.mYSize = IntSize::Truncate(ioWidth, ioHeight); + data.mCbChannel = ALIGNEDPTR_32(cbPlane.get()); + data.mCrChannel = ALIGNEDPTR_32(crPlane.get()); + data.mCbCrStride = cbCrStride; + data.mCbCrSize = IntSize::Truncate(cbCrWidth, cbCrHeight); + data.mPicSize = data.mYSize; + + ConvertYCbCrToRGB(data, SurfaceFormat::B8G8R8X8, IntSize::Truncate(ioWidth, ioHeight), mappedSurface.mData, mappedSurface.mStride); + } + } else { + unsigned char* ioData = (unsigned char*)aSurface->GetBaseAddress(); + + for (size_t i = 0; i < ioHeight; ++i) { + memcpy(mappedSurface.mData + i * mappedSurface.mStride, + ioData + i * bytesPerRow, + ioWidth * 4); + } + } + + dataSurface->Unmap(); + + return dataSurface.forget(); +} + +already_AddRefed<SourceSurface> +CreateSourceSurfaceFromMacIOSurface(MacIOSurface* aSurface) +{ + aSurface->Lock(); + RefPtr<SourceSurface> result = CreateSourceSurfaceFromLockedMacIOSurface(aSurface); + aSurface->Unlock(); + return result.forget(); +} + +} // namespace gfx +} // namespace mozilla |