Adding upstream version 124.0.1.upstream/124.0.1

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

1 files changed, 265 insertions, 0 deletions

diff --git a/gfx/webrender_bindings/RenderExternalTextureHost.cpp b/gfx/webrender_bindings/RenderExternalTextureHost.cpp
new file mode 100644
index 0000000000..db8107ff0a
--- /dev/null
+++ b/gfx/webrender_bindings/RenderExternalTextureHost.cpp
@@ -0,0 +1,265 @@
+/* -*- 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 "RenderExternalTextureHost.h"
+
+#include "mozilla/gfx/Logging.h"
+#include "mozilla/layers/ImageDataSerializer.h"
+
+#include "GLContext.h"
+
+namespace mozilla {
+namespace wr {
+
+RenderExternalTextureHost::RenderExternalTextureHost(
+    uint8_t* aBuffer, const layers::BufferDescriptor& aDescriptor)
+    : mBuffer(aBuffer),
+      mDescriptor(aDescriptor),
+      mInitialized(false),
+      mTextureUpdateNeeded(true) {
+  MOZ_COUNT_CTOR_INHERITED(RenderExternalTextureHost, RenderTextureHost);
+
+  switch (mDescriptor.type()) {
+    case layers::BufferDescriptor::TYCbCrDescriptor: {
+      const layers::YCbCrDescriptor& ycbcr = mDescriptor.get_YCbCrDescriptor();
+      mSize = ycbcr.display().Size();
+      mFormat = gfx::SurfaceFormat::YUV;
+      break;
+    }
+    case layers::BufferDescriptor::TRGBDescriptor: {
+      const layers::RGBDescriptor& rgb = mDescriptor.get_RGBDescriptor();
+      mSize = rgb.size();
+      mFormat = rgb.format();
+      break;
+    }
+    default:
+      gfxCriticalError() << "Bad buffer host descriptor "
+                         << (int)mDescriptor.type();
+      MOZ_CRASH("GFX: Bad descriptor");
+  }
+}
+
+RenderExternalTextureHost::~RenderExternalTextureHost() {
+  MOZ_COUNT_DTOR_INHERITED(RenderExternalTextureHost, RenderTextureHost);
+
+  if (NS_WARN_IF(!IsReadyForDeletion())) {
+    gfxCriticalNote << "RenderExternalTextureHost sync failed";
+  }
+
+  DeleteTextures();
+}
+
+bool RenderExternalTextureHost::CreateSurfaces() {
+  if (!IsYUV()) {
+    mSurfaces[0] = gfx::Factory::CreateWrappingDataSourceSurface(
+        GetBuffer(),
+        layers::ImageDataSerializer::GetRGBStride(
+            mDescriptor.get_RGBDescriptor()),
+        mSize, mFormat);
+  } else {
+    const layers::YCbCrDescriptor& desc = mDescriptor.get_YCbCrDescriptor();
+    const gfx::SurfaceFormat surfaceFormat =
+        SurfaceFormatForColorDepth(desc.colorDepth());
+    auto cbcrSize = layers::ImageDataSerializer::GetCroppedCbCrSize(desc);
+
+    mSurfaces[0] = gfx::Factory::CreateWrappingDataSourceSurface(
+        layers::ImageDataSerializer::GetYChannel(GetBuffer(), desc),
+        desc.yStride(), desc.display().Size(), surfaceFormat);
+    mSurfaces[1] = gfx::Factory::CreateWrappingDataSourceSurface(
+        layers::ImageDataSerializer::GetCbChannel(GetBuffer(), desc),
+        desc.cbCrStride(), cbcrSize, surfaceFormat);
+    mSurfaces[2] = gfx::Factory::CreateWrappingDataSourceSurface(
+        layers::ImageDataSerializer::GetCrChannel(GetBuffer(), desc),
+        desc.cbCrStride(), cbcrSize, surfaceFormat);
+  }
+
+  for (size_t i = 0; i < PlaneCount(); ++i) {
+    if (NS_WARN_IF(!mSurfaces[i])) {
+      gfxCriticalNote << "Surface is null";
+      return false;
+    }
+  }
+
+  return true;
+}
+
+void RenderExternalTextureHost::DeleteSurfaces() {
+  for (size_t i = 0; i < PlaneCount(); ++i) {
+    mSurfaces[i] = nullptr;
+  }
+}
+
+void RenderExternalTextureHost::DeleteTextures() {
+  for (size_t i = 0; i < PlaneCount(); ++i) {
+    mTextureSources[i] = nullptr;
+    mImages[i] = InvalidToWrExternalImage();
+  }
+}
+
+bool RenderExternalTextureHost::InitializeIfNeeded() {
+  if (mInitialized) {
+    return true;
+  }
+
+  if (!GetBuffer()) {
+    // We hit some problems to get the shmem.
+    gfxCriticalNote << "GetBuffer Failed";
+    return false;
+  }
+
+  if (!CreateSurfaces()) {
+    DeleteSurfaces();
+    return false;
+  }
+
+  mInitialized = true;
+  return mInitialized;
+}
+
+bool RenderExternalTextureHost::IsReadyForDeletion() {
+  if (!mInitialized) {
+    return true;
+  }
+
+  auto& textureSource = mTextureSources[0];
+  if (textureSource) {
+    return textureSource->Sync(false);
+  }
+
+  return true;
+}
+
+wr::WrExternalImage RenderExternalTextureHost::Lock(uint8_t aChannelIndex,
+                                                    gl::GLContext* aGL) {
+  if (mGL.get() != aGL) {
+    mGL = aGL;
+    mGL->MakeCurrent();
+  }
+
+  if (!mGL || !mGL->MakeCurrent()) {
+    return InvalidToWrExternalImage();
+  }
+
+  if (!InitializeIfNeeded()) {
+    return InvalidToWrExternalImage();
+  }
+
+  UpdateTextures();
+  return mImages[aChannelIndex];
+}
+
+void RenderExternalTextureHost::PrepareForUse() { mTextureUpdateNeeded = true; }
+
+void RenderExternalTextureHost::Unlock() {}
+
+void RenderExternalTextureHost::UpdateTexture(size_t aIndex) {
+  MOZ_ASSERT(mSurfaces[aIndex]);
+
+  auto& texture = mTextureSources[aIndex];
+
+  if (texture) {
+    texture->Update(mSurfaces[aIndex]);
+  } else {
+    texture = new layers::DirectMapTextureSource(mGL, mSurfaces[aIndex]);
+
+    const GLuint handle = texture->GetTextureHandle();
+    const auto uvs = GetUvCoords(texture->GetSize());
+    mImages[aIndex] = NativeTextureToWrExternalImage(
+        handle, uvs.first.x, uvs.first.y, uvs.second.x, uvs.second.y);
+  }
+
+  MOZ_ASSERT(mGL->GetError() == LOCAL_GL_NO_ERROR);
+}
+
+void RenderExternalTextureHost::UpdateTextures() {
+  if (!mTextureUpdateNeeded) {
+    // Nothing to do here.
+    return;
+  }
+
+  for (size_t i = 0; i < PlaneCount(); ++i) {
+    UpdateTexture(i);
+  }
+
+  mTextureSources[0]->MaybeFenceTexture();
+  mTextureUpdateNeeded = false;
+}
+
+size_t RenderExternalTextureHost::GetPlaneCount() const { return PlaneCount(); }
+
+gfx::SurfaceFormat RenderExternalTextureHost::GetFormat() const {
+  return mFormat;
+}
+
+gfx::ColorDepth RenderExternalTextureHost::GetColorDepth() const {
+  switch (mDescriptor.type()) {
+    case layers::BufferDescriptor::TYCbCrDescriptor:
+      return mDescriptor.get_YCbCrDescriptor().colorDepth();
+    default:
+      return gfx::ColorDepth::COLOR_8;
+  }
+}
+
+gfx::YUVRangedColorSpace RenderExternalTextureHost::GetYUVColorSpace() const {
+  switch (mDescriptor.type()) {
+    case layers::BufferDescriptor::TYCbCrDescriptor:
+      return gfx::GetYUVRangedColorSpace(mDescriptor.get_YCbCrDescriptor());
+    default:
+      return gfx::YUVRangedColorSpace::Default;
+  }
+}
+
+bool RenderExternalTextureHost::MapPlane(RenderCompositor* aCompositor,
+                                         uint8_t aChannelIndex,
+                                         PlaneInfo& aPlaneInfo) {
+  if (!mBuffer) {
+    // We hit some problems to get the shmem.
+    gfxCriticalNote << "GetBuffer Failed";
+    return false;
+  }
+
+  switch (mDescriptor.type()) {
+    case layers::BufferDescriptor::TYCbCrDescriptor: {
+      const layers::YCbCrDescriptor& desc = mDescriptor.get_YCbCrDescriptor();
+      switch (aChannelIndex) {
+        case 0:
+          aPlaneInfo.mData =
+              layers::ImageDataSerializer::GetYChannel(mBuffer, desc);
+          aPlaneInfo.mStride = desc.yStride();
+          aPlaneInfo.mSize = desc.display().Size();
+          break;
+        case 1:
+          aPlaneInfo.mData =
+              layers::ImageDataSerializer::GetCbChannel(mBuffer, desc);
+          aPlaneInfo.mStride = desc.cbCrStride();
+          aPlaneInfo.mSize =
+              layers::ImageDataSerializer::GetCroppedCbCrSize(desc);
+          break;
+        case 2:
+          aPlaneInfo.mData =
+              layers::ImageDataSerializer::GetCrChannel(mBuffer, desc);
+          aPlaneInfo.mStride = desc.cbCrStride();
+          aPlaneInfo.mSize =
+              layers::ImageDataSerializer::GetCroppedCbCrSize(desc);
+          break;
+      }
+      break;
+    }
+    default: {
+      const layers::RGBDescriptor& desc = mDescriptor.get_RGBDescriptor();
+      aPlaneInfo.mData = mBuffer;
+      aPlaneInfo.mStride = layers::ImageDataSerializer::GetRGBStride(desc);
+      aPlaneInfo.mSize = desc.size();
+      break;
+    }
+  }
+  return true;
+}
+
+void RenderExternalTextureHost::UnmapPlanes() {}
+
+}  // namespace wr
+}  // namespace mozilla