Use kcp as QoS module

2025-10-26 20:25:34 +08:00 · 2023-08-30 17:44:22 +08:00
parent 1e90d99980
commit 72d4982f54
34 changed files with 157 additions and 3119 deletions
--- a/screen_capture/wgc_session_impl.h
+++ b/screen_capture/wgc_session_impl.h
@@ -0,0 +1,116 @@
+#ifndef _WGC_SESSION_IMPL_H_
+#define _WGC_SESSION_IMPL_H_
+
+#include <d3d11_4.h>
+#include <winrt/Windows.Foundation.h>
+#include <winrt/Windows.Graphics.Capture.h>
+
+#include <mutex>
+#include <thread>
+
+#include "wgc_session.h"
+
+class WgcSessionImpl : public WgcSession {
+  struct __declspec(uuid("A9B3D012-3DF2-4EE3-B8D1-8695F457D3C1"))
+      IDirect3DDxgiInterfaceAccess : ::IUnknown {
+    virtual HRESULT __stdcall GetInterface(GUID const &id, void **object) = 0;
+  };
+
+  template <typename T>
+  inline auto GetDXGIInterfaceFromObject(
+      winrt::Windows::Foundation::IInspectable const &object) {
+    auto access = object.as<IDirect3DDxgiInterfaceAccess>();
+    winrt::com_ptr<T> result;
+    winrt::check_hresult(
+        access->GetInterface(winrt::guid_of<T>(), result.put_void()));
+    return result;
+  }
+
+  struct {
+    union {
+      HWND hwnd;
+      HMONITOR hmonitor;
+    };
+    bool is_window;
+  } target_{0};
+
+ public:
+  WgcSessionImpl();
+  ~WgcSessionImpl() override;
+
+ public:
+  void Release() override;
+
+  int Initialize(HWND hwnd) override;
+  int Initialize(HMONITOR hmonitor) override;
+
+  void RegisterObserver(wgc_session_observer *observer) override;
+
+  int Start() override;
+  int Stop() override;
+
+  int Pause() override;
+  int Resume() override;
+
+ private:
+  auto CreateD3D11Device();
+  auto CreateCaptureItemForWindow(HWND hwnd);
+  auto CreateCaptureItemForMonitor(HMONITOR hmonitor);
+
+  HRESULT CreateMappedTexture(winrt::com_ptr<ID3D11Texture2D> src_texture,
+                              unsigned int width = 0, unsigned int height = 0);
+  void OnFrame(
+      winrt::Windows::Graphics::Capture::Direct3D11CaptureFramePool const
+          &sender,
+      winrt::Windows::Foundation::IInspectable const &args);
+  void OnClosed(winrt::Windows::Graphics::Capture::GraphicsCaptureItem const &,
+                winrt::Windows::Foundation::IInspectable const &);
+
+  int Initialize();
+  void CleanUp();
+
+  // void message_func();
+
+ private:
+  std::mutex lock_;
+  bool is_initialized_ = false;
+  bool is_running_ = false;
+  bool is_paused_ = false;
+
+  wgc_session_observer *observer_ = nullptr;
+
+  // wgc
+  winrt::Windows::Graphics::Capture::GraphicsCaptureItem capture_item_{nullptr};
+  winrt::Windows::Graphics::Capture::GraphicsCaptureSession capture_session_{
+      nullptr};
+  winrt::Windows::Graphics::SizeInt32 capture_frame_size_;
+
+  winrt::Windows::Graphics::DirectX::Direct3D11::IDirect3DDevice
+      d3d11_direct_device_{nullptr};
+  winrt::com_ptr<ID3D11DeviceContext> d3d11_device_context_{nullptr};
+  winrt::com_ptr<ID3D11Texture2D> d3d11_texture_mapped_{nullptr};
+
+  std::atomic<bool> cleaned_ = false;
+  winrt::Windows::Graphics::Capture::Direct3D11CaptureFramePool
+      capture_framepool_{nullptr};
+  winrt::Windows::Graphics::Capture::Direct3D11CaptureFramePool::
+      FrameArrived_revoker capture_framepool_trigger_;
+  winrt::Windows::Graphics::Capture::GraphicsCaptureItem::Closed_revoker
+      capture_close_trigger_;
+
+  // message loop
+  std::thread loop_;
+  HWND hwnd_ = nullptr;
+};
+
+// template <typename T>
+// inline auto WgcSessionImpl::GetDXGIInterfaceFromObject(
+//     winrt::Windows::Foundation::IInspectable const &object) {
+//   auto access = object.as<IDirect3DDxgiInterfaceAccess>();
+//   winrt::com_ptr<T> result;
+//   winrt::check_hresult(
+//       access->GetInterface(winrt::guid_of<T>(), result.put_void()));
+//   return result;
+// }
+
+#endif