1 files changed, 94 insertions, 0 deletions
diff --git a/dom/media/platforms/wmf/DXVA2Manager.h b/dom/media/platforms/wmf/DXVA2Manager.h
new file mode 100644
index 0000000000..1fb501b406
--- /dev/null
+++ b/dom/media/platforms/wmf/DXVA2Manager.h
@@ -0,0 +1,94 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim:set ts=2 sw=2 sts=2 et cindent: */
+/* 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/. */
+#if !defined(DXVA2Manager_h_)
+#  define DXVA2Manager_h_
+
+#  include "MediaInfo.h"
+#  include "WMF.h"
+#  include "mozilla/Mutex.h"
+#  include "mozilla/gfx/Rect.h"
+#  include "d3d11.h"
+
+namespace mozilla {
+
+namespace layers {
+class Image;
+class ImageContainer;
+class KnowsCompositor;
+}  // namespace layers
+
+class DXVA2Manager {
+ public:
+  // Creates and initializes a DXVA2Manager. We can use DXVA2 via either
+  // D3D9Ex or D3D11.
+  static DXVA2Manager* CreateD3D9DXVA(layers::KnowsCompositor* aKnowsCompositor,
+                                      nsACString& aFailureReason);
+  static DXVA2Manager* CreateD3D11DXVA(
+      layers::KnowsCompositor* aKnowsCompositor, nsACString& aFailureReason,
+      ID3D11Device* aDevice = nullptr);
+
+  // Returns a pointer to the D3D device manager responsible for managing the
+  // device we're using for hardware accelerated video decoding. If we're using
+  // D3D9Ex, this is an IDirect3DDeviceManager9. For D3D11 this is an
+  // IMFDXGIDeviceManager. It is safe to call this on any thread.
+  virtual IUnknown* GetDXVADeviceManager() = 0;
+
+  // Creates an Image for the video frame stored in aVideoSample.
+  virtual HRESULT CopyToImage(IMFSample* aVideoSample,
+                              const gfx::IntRect& aRegion,
+                              layers::Image** aOutImage) = 0;
+
+  virtual HRESULT WrapTextureWithImage(IMFSample* aVideoSample,
+                                       const gfx::IntRect& aRegion,
+                                       layers::Image** aOutImage) {
+    // Not implemented!
+    MOZ_CRASH("WrapTextureWithImage not implemented on this manager.");
+    return E_FAIL;
+  }
+
+  virtual HRESULT CopyToBGRATexture(ID3D11Texture2D* aInTexture,
+                                    uint32_t aArrayIndex,
+                                    ID3D11Texture2D** aOutTexture) {
+    // Not implemented!
+    MOZ_CRASH("CopyToBGRATexture not implemented on this manager.");
+    return E_FAIL;
+  }
+
+  virtual HRESULT ConfigureForSize(IMFMediaType* aInputType,
+                                   gfx::YUVColorSpace aColorSpace,
+                                   gfx::ColorRange aColorRange, uint32_t aWidth,
+                                   uint32_t aHeight) {
+    return S_OK;
+  }
+
+  virtual bool IsD3D11() { return false; }
+
+  virtual ~DXVA2Manager();
+
+  virtual bool SupportsConfig(const VideoInfo& aInfo, IMFMediaType* aInputType,
+                              IMFMediaType* aOutputType) = 0;
+
+  // Called before shutdown video MFTDecoder.
+  virtual void BeforeShutdownVideoMFTDecoder() {}
+
+  virtual bool SupportsZeroCopyNV12Texture() { return false; }
+
+  static bool IsNV12Supported(uint32_t aVendorID, uint32_t aDeviceID,
+                              const nsAString& aDriverVersionString);
+
+ protected:
+  Mutex mLock MOZ_UNANNOTATED;
+  DXVA2Manager();
+
+  bool IsUnsupportedResolution(const uint32_t& aWidth, const uint32_t& aHeight,
+                               const float& aFramerate) const;
+
+  bool mIsAMDPreUVD4 = false;
+};
+
+}  // namespace mozilla
+
+#endif  // DXVA2Manager_h_