1 files changed, 96 insertions, 0 deletions
diff --git a/gfx/layers/mlgpu/TextureSourceProviderMLGPU.cpp b/gfx/layers/mlgpu/TextureSourceProviderMLGPU.cpp
new file mode 100644
index 0000000000..7ddf475588
--- /dev/null
+++ b/gfx/layers/mlgpu/TextureSourceProviderMLGPU.cpp
@@ -0,0 +1,96 @@
+/* -*- 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 "TextureSourceProviderMLGPU.h"
+#include "LayerManagerMLGPU.h"
+#include "MLGDevice.h"
+#ifdef XP_WIN
+#  include "mozilla/layers/MLGDeviceD3D11.h"
+#endif
+
+namespace mozilla {
+namespace layers {
+
+TextureSourceProviderMLGPU::TextureSourceProviderMLGPU(
+    LayerManagerMLGPU* aLayerManager, MLGDevice* aDevice)
+    : mLayerManager(aLayerManager), mDevice(aDevice) {}
+
+TextureSourceProviderMLGPU::~TextureSourceProviderMLGPU() = default;
+
+int32_t TextureSourceProviderMLGPU::GetMaxTextureSize() const {
+  if (!mDevice) {
+    return 0;
+  }
+  return mDevice->GetMaxTextureSize();
+}
+
+bool TextureSourceProviderMLGPU::SupportsEffect(EffectTypes aEffect) {
+  switch (aEffect) {
+    case EffectTypes::YCBCR:
+      return true;
+    default:
+      MOZ_ASSERT_UNREACHABLE("NYI");
+  }
+  return false;
+}
+
+bool TextureSourceProviderMLGPU::IsValid() const { return !!mLayerManager; }
+
+void TextureSourceProviderMLGPU::Destroy() {
+  mLayerManager = nullptr;
+  mDevice = nullptr;
+  TextureSourceProvider::Destroy();
+}
+
+#ifdef XP_WIN
+ID3D11Device* TextureSourceProviderMLGPU::GetD3D11Device() const {
+  if (!mDevice) {
+    return nullptr;
+  }
+  return mDevice->AsD3D11()->GetD3D11Device();
+}
+#endif
+
+TimeStamp TextureSourceProviderMLGPU::GetLastCompositionEndTime() const {
+  if (!mLayerManager) {
+    return TimeStamp();
+  }
+  return mLayerManager->GetLastCompositionEndTime();
+}
+
+already_AddRefed<DataTextureSource>
+TextureSourceProviderMLGPU::CreateDataTextureSource(TextureFlags aFlags) {
+  RefPtr<DataTextureSource> texture = mDevice->CreateDataTextureSource(aFlags);
+  return texture.forget();
+}
+
+already_AddRefed<DataTextureSource>
+TextureSourceProviderMLGPU::CreateDataTextureSourceAround(
+    gfx::DataSourceSurface* aSurface) {
+  MOZ_ASSERT_UNREACHABLE("NYI");
+  return nullptr;
+}
+
+void TextureSourceProviderMLGPU::UnlockAfterComposition(TextureHost* aTexture) {
+  TextureSourceProvider::UnlockAfterComposition(aTexture);
+
+  // If this is being called after we shutdown the compositor, we must finish
+  // read unlocking now to prevent a cycle.
+  if (!IsValid()) {
+    ReadUnlockTextures();
+  }
+}
+
+bool TextureSourceProviderMLGPU::NotifyNotUsedAfterComposition(
+    TextureHost* aTextureHost) {
+  if (!IsValid()) {
+    return false;
+  }
+  return TextureSourceProvider::NotifyNotUsedAfterComposition(aTextureHost);
+}
+
+}  // namespace layers
+}  // namespace mozilla