Adding upstream version 124.0.1.upstream/124.0.1

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 215 insertions, 0 deletions

diff --git a/gfx/layers/MacIOSurfaceHelpers.cpp b/gfx/layers/MacIOSurfaceHelpers.cpp
new file mode 100644
index 0000000000..ff5b59b626
--- /dev/null
+++ b/gfx/layers/MacIOSurfaceHelpers.cpp
@@ -0,0 +1,215 @@
+/* -*- 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 "libyuv.h"
+#include "MacIOSurfaceHelpers.h"
+#include "mozilla/gfx/MacIOSurface.h"
+#include "mozilla/ScopeExit.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 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::YUV422) &&
+      (aSize.width > PlanarYCbCrImage::MAX_DIMENSION ||
+       aSize.height > PlanarYCbCrImage::MAX_DIMENSION)) {
+    return NS_ERROR_FAILURE;
+  }
+
+  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.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 = ChromaSubsampling::HALF_WIDTH_AND_HEIGHT;
+
+    ConvertYCbCrToRGB(data, SurfaceFormat::B8G8R8X8, aSize, aData, aStride);
+  } else if (ioFormat == SurfaceFormat::YUV422) {
+    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;
+
+      ConvertYCbCrToRGB(data, SurfaceFormat::B8G8R8X8, aSize, aData, aStride);
+    }
+  } 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) {
+  aSurface->Lock();
+  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::YUV422)
+          ? SurfaceFormat::B8G8R8X8
+          : SurfaceFormat::B8G8R8A8;
+
+  RefPtr<DataSourceSurface> 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) {
+  aSurface->Lock();
+  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::YUV422)
+          ? 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