mirror of
https://github.com/kunkundi/crossdesk.git
synced 2025-10-26 20:25:34 +08:00
116 lines
3.3 KiB
C++
116 lines
3.3 KiB
C++
#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 |