summaryrefslogtreecommitdiffstats
path: root/gfx/layers/MacIOSurfaceHelpers.cpp
diff options
context:
space:
mode:
authorMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
committerMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
commit5f8de423f190bbb79a62f804151bc24824fa32d8 (patch)
tree10027f336435511475e392454359edea8e25895d /gfx/layers/MacIOSurfaceHelpers.cpp
parent49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff)
downloadUXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip
Add m-esr52 at 52.6.0
Diffstat (limited to 'gfx/layers/MacIOSurfaceHelpers.cpp')
-rw-r--r--gfx/layers/MacIOSurfaceHelpers.cpp176
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