init

libs/hwcodec/cpp/common/platform/win/win.h

@@ -0,0 +1,135 @@
+#ifndef WIN_H
+#define WIN_H
+
+#include <DirectXMath.h>
+#include <d3d11.h>
+#include <d3d11_1.h>
+#include <directxcolors.h>
+#include <iostream>
+#include <vector>
+#include <wrl/client.h>
+
+#include "../../common.h"
+
+using Microsoft::WRL::ComPtr;
+using namespace std;
+using namespace DirectX;
+
+#define SAFE_RELEASE(p)                                                        \
+  {                                                                            \
+    if ((p)) {                                                                 \
+      (p)->Release();                                                          \
+      (p) = nullptr;                                                           \
+    }                                                                          \
+  }
+#define LUID(desc)                                                             \
+  (((int64_t)desc.AdapterLuid.HighPart << 32) | desc.AdapterLuid.LowPart)
+
+class NativeDevice {
+public:
+  bool Init(int64_t luid, ID3D11Device *device, int pool_size = 1);
+  bool EnsureTexture(int width, int height);
+  bool SetTexture(ID3D11Texture2D *texture);
+  HANDLE GetSharedHandle();
+  ID3D11Texture2D *GetCurrentTexture();
+  int next();
+  void BeginQuery();
+  void EndQuery();
+  bool Query();
+  bool Process(ID3D11Texture2D *in, ID3D11Texture2D *out, int width, int height,
+               D3D11_VIDEO_PROCESSOR_CONTENT_DESC content_desc,
+               DXGI_COLOR_SPACE_TYPE colorSpace_in,
+               DXGI_COLOR_SPACE_TYPE colorSpace_out, int arraySlice);
+  bool BgraToNv12(ID3D11Texture2D *bgraTexture, ID3D11Texture2D *nv12Texture,
+                  int width, int height, DXGI_COLOR_SPACE_TYPE colorSpace_in,
+                  DXGI_COLOR_SPACE_TYPE colorSpace_outt);
+  bool Nv12ToBgra(int width, int height, ID3D11Texture2D *nv12Texture,
+                  ID3D11Texture2D *bgraTexture, int nv12ArrayIndex);
+  AdapterVendor GetVendor();
+  bool support_decode(DataFormat format);
+
+private:
+  bool InitFromLuid(int64_t luid);
+  bool InitFromDevice(ID3D11Device *device);
+  bool SetMultithreadProtected();
+  bool InitQuery();
+  bool InitVideoDevice();
+  bool isFormatHybridDecodedByHardware(DataFormat format, unsigned int vendorId,
+                                       unsigned int deviceId);
+
+  // nv12 to bgra
+  bool nv12_to_bgra_set_srv(ID3D11Texture2D *nv12Texture, int width,
+                            int height);
+  bool nv12_to_bgra_set_rtv(ID3D11Texture2D *bgraTexture, int width,
+                            int height);
+  bool nv12_to_bgra_set_view_port(int width, int height);
+  bool nv12_to_bgra_set_sample();
+  bool nv12_to_bgra_set_shader();
+  bool nv12_to_bgra_set_vertex_buffer();
+  bool nv12_to_bgra_draw();
+
+public:
+  // Direct3D 11
+  ComPtr<IDXGIFactory1> factory1_ = nullptr;
+  ComPtr<IDXGIAdapter> adapter_ = nullptr;
+  ComPtr<IDXGIAdapter1> adapter1_ = nullptr;
+  ComPtr<ID3D11Device> device_ = nullptr;
+  ComPtr<ID3D11DeviceContext> context_ = nullptr;
+  ComPtr<ID3D11Query> query_ = nullptr;
+  ComPtr<ID3D11VideoDevice> video_device_ = nullptr;
+  ComPtr<ID3D11VideoContext> video_context_ = nullptr;
+  ComPtr<ID3D11VideoContext1> video_context1_ = nullptr;
+  ComPtr<ID3D11VideoProcessorEnumerator> video_processor_enumerator_ = nullptr;
+  ComPtr<ID3D11VideoProcessor> video_processor_ = nullptr;
+  D3D11_VIDEO_PROCESSOR_CONTENT_DESC last_content_desc_ = {};
+
+  ComPtr<ID3D11RenderTargetView> RTV_ = NULL;
+  ComPtr<ID3D11ShaderResourceView> SRV_[2] = {NULL, NULL};
+  ComPtr<ID3D11VertexShader> vertexShader_ = NULL;
+  ComPtr<ID3D11PixelShader> pixelShader_ = NULL;
+  ComPtr<ID3D11SamplerState> samplerLinear_ = NULL;
+  ComPtr<ID3D11Texture2D> nv12SrvTexture_ = nullptr;
+
+  int count_;
+  int index_ = 0;
+
+  int last_nv12_to_bgra_width_ = 0;
+  int last_nv12_to_bgra_height_ = 0;
+
+private:
+  std::vector<ComPtr<ID3D11Texture2D>> texture_;
+};
+
+class Adapter {
+public:
+  bool Init(IDXGIAdapter1 *adapter1);
+
+private:
+  bool SetMultithreadProtected();
+
+public:
+  ComPtr<IDXGIAdapter> adapter_ = nullptr;
+  ComPtr<IDXGIAdapter1> adapter1_ = nullptr;
+  ComPtr<ID3D11Device> device_ = nullptr;
+  ComPtr<ID3D11DeviceContext> context_ = nullptr;
+  DXGI_ADAPTER_DESC1 desc1_;
+};
+
+class Adapters {
+public:
+  bool Init(AdapterVendor vendor);
+  static int GetFirstAdapterIndex(AdapterVendor vendor);
+
+public:
+  ComPtr<IDXGIFactory1> factory1_ = nullptr;
+  std::vector<std::unique_ptr<Adapter>> adapters_;
+};
+
+extern "C" uint64_t GetHwcodecGpuSignature();
+
+extern "C" void hwcodec_get_d3d11_texture_width_height(ID3D11Texture2D *texture, int *w,
+                                             int *h);
+
+extern "C" int32_t add_process_to_new_job(DWORD process_id);
+
+#endif