mirror of
https://github.com/Blinue/Magpie.git
synced 2026-06-24 02:04:10 +00:00
b704fd8e30
在单独的线程中用单独的 D3D Device 接受屏幕帧，就像 https://github.com/microsoft/Windows-classic-samples/tree/main/Samples/DXGIDesktopDuplication 中所做的。涉及 D3D Device 间的纹理同步因此较为复杂，但可以有效提高流畅度
403 lines
11 KiB
C++
403 lines
11 KiB
C++
#include "pch.h"
|
||
#include "DesktopDuplicationFrameSource.h"
|
||
#include "App.h"
|
||
|
||
|
||
static ComPtr<IDXGIOutput1> FindMonitor(ComPtr<IDXGIAdapter1> adapter, HMONITOR hMonitor) {
|
||
ComPtr<IDXGIOutput> output;
|
||
|
||
for (UINT adapterIndex = 0;
|
||
SUCCEEDED(adapter->EnumOutputs(adapterIndex,
|
||
output.ReleaseAndGetAddressOf()));
|
||
adapterIndex++
|
||
) {
|
||
DXGI_OUTPUT_DESC desc;
|
||
HRESULT hr = output->GetDesc(&desc);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("GetDesc 失败", hr));
|
||
continue;
|
||
}
|
||
|
||
if (desc.Monitor == hMonitor) {
|
||
ComPtr<IDXGIOutput1> output1;
|
||
hr = output.As<IDXGIOutput1>(&output1);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("从 IDXGIOutput 获取 IDXGIOutput1 失败", hr));
|
||
return nullptr;
|
||
}
|
||
|
||
return output1;
|
||
}
|
||
}
|
||
|
||
return nullptr;
|
||
}
|
||
|
||
// 根据显示器句柄查找 IDXGIOutput1
|
||
static ComPtr<IDXGIOutput1> GetDXGIOutput(HMONITOR hMonitor) {
|
||
const Renderer& renderer = App::GetInstance().GetRenderer();
|
||
ComPtr<IDXGIAdapter1> curAdapter = App::GetInstance().GetRenderer().GetGraphicsAdapter();
|
||
|
||
// 首先在当前使用的图形适配器上查询显示器
|
||
ComPtr<IDXGIOutput1> output = FindMonitor(curAdapter, hMonitor);
|
||
if (output) {
|
||
return output;
|
||
}
|
||
|
||
// 未找到则在所有图形适配器上查找
|
||
ComPtr<IDXGIAdapter1> adapter;
|
||
ComPtr<IDXGIFactory2> dxgiFactory = renderer.GetDXGIFactory();
|
||
for (UINT adapterIndex = 0;
|
||
SUCCEEDED(dxgiFactory->EnumAdapters1(adapterIndex,
|
||
adapter.ReleaseAndGetAddressOf()));
|
||
adapterIndex++
|
||
) {
|
||
if (adapter == curAdapter) {
|
||
continue;
|
||
}
|
||
|
||
output = FindMonitor(adapter, hMonitor);
|
||
if (output) {
|
||
return output;
|
||
}
|
||
}
|
||
|
||
return nullptr;
|
||
}
|
||
|
||
ComPtr<ID3D11Device> d3dDevice;
|
||
ComPtr<ID3D11DeviceContext> d3dDC;
|
||
HANDLE hDDPThread = NULL;
|
||
HANDLE hTerminateThreadEvent = NULL;
|
||
ComPtr<IDXGIKeyedMutex> sharedTexMutex;
|
||
ComPtr<ID3D11Texture2D> sharedTex;
|
||
ComPtr<ID3D11Texture2D> sharedTex1;
|
||
ComPtr<IDXGIKeyedMutex> sharedTexMutex1;
|
||
HANDLE hSharedTex = NULL;
|
||
|
||
DesktopDuplicationFrameSource::~DesktopDuplicationFrameSource() {
|
||
SetEvent(hTerminateThreadEvent);
|
||
WaitForSingleObject(hDDPThread, INFINITE);
|
||
ResetEvent(hTerminateThreadEvent);
|
||
}
|
||
|
||
bool DesktopDuplicationFrameSource::Initialize() {
|
||
// WDA_EXCLUDEFROMCAPTURE 只在 Win10 v2004 及更新版本中可用
|
||
const RTL_OSVERSIONINFOW& version = Utils::GetOSVersion();
|
||
if (Utils::CompareVersion(version.dwMajorVersion, version.dwMinorVersion, version.dwBuildNumber, 10, 0, 19041) < 0) {
|
||
SPDLOG_LOGGER_ERROR(logger, "当前操作系统无法使用 Desktop Duplication");
|
||
return false;
|
||
}
|
||
|
||
hTerminateThreadEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr);
|
||
if (!hTerminateThreadEvent) {
|
||
return false;
|
||
}
|
||
|
||
RECT srcClientRect;
|
||
if (!Utils::GetClientScreenRect(App::GetInstance().GetHwndSrcClient(), srcClientRect)) {
|
||
SPDLOG_LOGGER_ERROR(logger, "GetClientScreenRect 失败");
|
||
return false;
|
||
}
|
||
|
||
D3D11_TEXTURE2D_DESC desc{};
|
||
desc.Format = DXGI_FORMAT_B8G8R8A8_UNORM;
|
||
desc.Width = srcClientRect.right - srcClientRect.left;
|
||
desc.Height = srcClientRect.bottom - srcClientRect.top;
|
||
desc.Usage = D3D11_USAGE_DEFAULT;
|
||
desc.MipLevels = 1;
|
||
desc.ArraySize = 1;
|
||
desc.SampleDesc.Count = 1;
|
||
desc.BindFlags = D3D11_BIND_SHADER_RESOURCE | D3D11_BIND_RENDER_TARGET;
|
||
HRESULT hr = App::GetInstance().GetRenderer().GetD3DDevice()->CreateTexture2D(&desc, nullptr, &_output);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("创建 Texture2D 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
desc.MiscFlags = D3D11_RESOURCE_MISC_SHARED_KEYEDMUTEX;
|
||
hr = App::GetInstance().GetRenderer().GetD3DDevice()->CreateTexture2D(&desc, nullptr, &sharedTex);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("创建 Texture2D 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
hr = sharedTex.As<IDXGIKeyedMutex>(&sharedTexMutex);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("检索 IDXGIKeyedMutex 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
ComPtr<IDXGIResource> sharedDxgiRes;
|
||
hr = sharedTex.As<IDXGIResource>(&sharedDxgiRes);
|
||
if (FAILED(hr)) {
|
||
return false;
|
||
}
|
||
|
||
hr = sharedDxgiRes->GetSharedHandle(&hSharedTex);
|
||
if (FAILED(hr)) {
|
||
return false;
|
||
}
|
||
|
||
// 使全屏窗口无法被捕获到
|
||
if (!SetWindowDisplayAffinity(App::GetInstance().GetHwndHost(), WDA_EXCLUDEFROMCAPTURE)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeWin32ErrorMsg("SetWindowDisplayAffinity 失败"));
|
||
return false;
|
||
}
|
||
|
||
|
||
|
||
hDDPThread = CreateThread(nullptr, 0, _DDPThreadProc, nullptr, 0, nullptr);
|
||
if (!hDDPThread) {
|
||
return false;
|
||
}
|
||
|
||
SPDLOG_LOGGER_INFO(logger, "DesktopDuplicationFrameSource 初始化完成");
|
||
return true;
|
||
}
|
||
|
||
std::atomic<UINT> frameCount = 0;
|
||
|
||
FrameSourceBase::UpdateState DesktopDuplicationFrameSource::Update() {
|
||
static UINT f = 0;
|
||
UINT newF = frameCount.load();
|
||
if (f == newF) {
|
||
SPDLOG_LOGGER_INFO(logger, "NoUpdate");
|
||
return UpdateState::NoUpdate;
|
||
}
|
||
|
||
f = newF;
|
||
|
||
HRESULT hr = sharedTexMutex->AcquireSync(1, 1);
|
||
if (hr == static_cast<HRESULT>(WAIT_TIMEOUT)) {
|
||
return UpdateState::Waiting;
|
||
}
|
||
|
||
SPDLOG_LOGGER_INFO(logger, "NewFrame");
|
||
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, "AcquireSync 失败");
|
||
return UpdateState::Error;
|
||
}
|
||
|
||
const auto& dc = App::GetInstance().GetRenderer().GetD3DDC();
|
||
dc->CopyResource(_output.Get(), sharedTex.Get());
|
||
|
||
sharedTexMutex->ReleaseSync(0);
|
||
|
||
_firstFrame = false;
|
||
|
||
return UpdateState::NewFrame;
|
||
}
|
||
|
||
DWORD WINAPI DesktopDuplicationFrameSource::_DDPThreadProc(LPVOID lpThreadParameter) {
|
||
DesktopDuplicationFrameSource& that = (DesktopDuplicationFrameSource&)App::GetInstance().GetFrameSource();
|
||
|
||
HWND hwndSrc = App::GetInstance().GetHwndSrc();
|
||
HMONITOR hMonitor = MonitorFromWindow(hwndSrc, MONITOR_DEFAULTTONEAREST);
|
||
if (!hMonitor) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeWin32ErrorMsg("MonitorFromWindow 失败"));
|
||
return false;
|
||
}
|
||
|
||
MONITORINFO mi{};
|
||
mi.cbSize = sizeof(mi);
|
||
if (!GetMonitorInfo(hMonitor, &mi)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeWin32ErrorMsg("GetMonitorInfo 失败"));
|
||
return false;
|
||
}
|
||
|
||
if (!_CenterWindowIfNecessary(hwndSrc, mi.rcWork)) {
|
||
SPDLOG_LOGGER_ERROR(logger, "居中源窗口失败");
|
||
return false;
|
||
}
|
||
|
||
ComPtr<IDXGIOutput1> output = GetDXGIOutput(hMonitor);
|
||
if (!output) {
|
||
SPDLOG_LOGGER_ERROR(logger, "无法找到 IDXGIOutput");
|
||
return false;
|
||
}
|
||
|
||
UINT createDeviceFlags = D3D11_CREATE_DEVICE_BGRA_SUPPORT;
|
||
D3D_FEATURE_LEVEL featureLevels[] = {
|
||
D3D_FEATURE_LEVEL_11_1,
|
||
D3D_FEATURE_LEVEL_11_0,
|
||
D3D_FEATURE_LEVEL_10_1,
|
||
D3D_FEATURE_LEVEL_10_0,
|
||
// 不支持功能级别 9.x，但这里加上没坏处
|
||
D3D_FEATURE_LEVEL_9_3,
|
||
D3D_FEATURE_LEVEL_9_2,
|
||
D3D_FEATURE_LEVEL_9_1,
|
||
};
|
||
UINT nFeatureLevels = ARRAYSIZE(featureLevels);
|
||
|
||
D3D_FEATURE_LEVEL fl;
|
||
|
||
HRESULT hr = D3D11CreateDevice(
|
||
App::GetInstance().GetRenderer().GetGraphicsAdapter().Get(),
|
||
D3D_DRIVER_TYPE_UNKNOWN,
|
||
nullptr,
|
||
createDeviceFlags,
|
||
featureLevels,
|
||
nFeatureLevels,
|
||
D3D11_SDK_VERSION,
|
||
&d3dDevice,
|
||
&fl,
|
||
&d3dDC
|
||
);
|
||
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_WARN(logger, "D3D11CreateDevice 失败");
|
||
return false;
|
||
}
|
||
|
||
hr = d3dDevice->OpenSharedResource(hSharedTex, IID_PPV_ARGS(&sharedTex1));
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("OpenSharedResource 失败", hr));
|
||
return 0;
|
||
}
|
||
|
||
hr = sharedTex1.As<IDXGIKeyedMutex>(&sharedTexMutex1);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("检索 IDXGIKeyedMutex 失败", hr));
|
||
return 0;
|
||
}
|
||
|
||
ComPtr<IDXGIOutputDuplication> outputDup;
|
||
hr = output->DuplicateOutput(
|
||
d3dDevice.Get(),
|
||
outputDup.ReleaseAndGetAddressOf()
|
||
);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("DuplicateOutput 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
RECT srcClientRect;
|
||
if (!Utils::GetClientScreenRect(App::GetInstance().GetHwndSrcClient(), srcClientRect)) {
|
||
SPDLOG_LOGGER_ERROR(logger, "GetClientScreenRect 失败");
|
||
return false;
|
||
}
|
||
|
||
// 计算源窗口客户区在该屏幕上的位置，用于计算新帧是否有更新
|
||
RECT _srcClientInMonitor = {
|
||
srcClientRect.left - mi.rcMonitor.left,
|
||
srcClientRect.top - mi.rcMonitor.top,
|
||
srcClientRect.right - mi.rcMonitor.left,
|
||
srcClientRect.bottom - mi.rcMonitor.top
|
||
};
|
||
|
||
D3D11_BOX _frameInMonitor = {
|
||
(UINT)_srcClientInMonitor.left,
|
||
(UINT)_srcClientInMonitor.top,
|
||
0,
|
||
(UINT)_srcClientInMonitor.right,
|
||
(UINT)_srcClientInMonitor.bottom,
|
||
1
|
||
};
|
||
|
||
DXGI_OUTDUPL_FRAME_INFO info{};
|
||
ComPtr<IDXGIResource> dxgiRes;
|
||
std::vector<BYTE> dupMetaData;
|
||
|
||
bool waitToProcessCurrentFrame = false;
|
||
|
||
while ((WaitForSingleObjectEx(hTerminateThreadEvent, 0, FALSE) == WAIT_TIMEOUT)) {
|
||
if (!waitToProcessCurrentFrame) {
|
||
if (dxgiRes) {
|
||
outputDup->ReleaseFrame();
|
||
}
|
||
HRESULT hr = outputDup->AcquireNextFrame(500, &info, &dxgiRes);
|
||
if (hr == DXGI_ERROR_WAIT_TIMEOUT) {
|
||
continue;
|
||
}
|
||
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("AcquireNextFrame 失败", hr));
|
||
continue;
|
||
}
|
||
|
||
bool noUpdate = true;
|
||
|
||
// 检索 move rects 和 dirty rects
|
||
// 这些区域如果和窗口客户区有重叠则表明画面有变化
|
||
if (info.TotalMetadataBufferSize) {
|
||
if (info.TotalMetadataBufferSize > dupMetaData.size()) {
|
||
dupMetaData.resize(info.TotalMetadataBufferSize);
|
||
}
|
||
|
||
UINT bufSize = info.TotalMetadataBufferSize;
|
||
|
||
// move rects
|
||
hr = outputDup->GetFrameMoveRects(bufSize, (DXGI_OUTDUPL_MOVE_RECT*)dupMetaData.data(), &bufSize);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("GetFrameMoveRects 失败", hr));
|
||
continue;
|
||
}
|
||
|
||
UINT nRect = bufSize / sizeof(DXGI_OUTDUPL_MOVE_RECT);
|
||
for (UINT i = 0; i < nRect; ++i) {
|
||
const DXGI_OUTDUPL_MOVE_RECT& rect = ((DXGI_OUTDUPL_MOVE_RECT*)dupMetaData.data())[i];
|
||
if (Utils::CheckOverlap(_srcClientInMonitor, rect.DestinationRect)) {
|
||
noUpdate = false;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (noUpdate) {
|
||
bufSize = info.TotalMetadataBufferSize;
|
||
|
||
// dirty rects
|
||
hr = outputDup->GetFrameDirtyRects(bufSize, (RECT*)dupMetaData.data(), &bufSize);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("GetFrameDirtyRects 失败", hr));
|
||
continue;
|
||
}
|
||
|
||
nRect = bufSize / sizeof(RECT);
|
||
for (UINT i = 0; i < nRect; ++i) {
|
||
const RECT& rect = ((RECT*)dupMetaData.data())[i];
|
||
OutputDebugString(fmt::format(L"{},{}\n", rect.right - rect.left, rect.bottom - rect.top).c_str());
|
||
if (Utils::CheckOverlap(_srcClientInMonitor, rect)) {
|
||
noUpdate = false;
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
if (noUpdate) {
|
||
continue;
|
||
}
|
||
}
|
||
|
||
HRESULT hr = sharedTexMutex1->AcquireSync(0, 100);
|
||
if (hr == static_cast<HRESULT>(WAIT_TIMEOUT)) {
|
||
waitToProcessCurrentFrame = true;
|
||
continue;
|
||
}
|
||
|
||
waitToProcessCurrentFrame = false;
|
||
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, "AcquireSync 失败");
|
||
continue;
|
||
}
|
||
|
||
Utils::ScopeExit se([&]() {
|
||
sharedTexMutex1->ReleaseSync(1);
|
||
});
|
||
|
||
ComPtr<ID3D11Resource> d3dRes;
|
||
hr = dxgiRes.As<ID3D11Resource>(&d3dRes);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("从 IDXGIResource 检索 ID3D11Resource 失败", hr));
|
||
continue;
|
||
}
|
||
|
||
d3dDC->CopySubresourceRegion(sharedTex1.Get(), 0, 0, 0, 0, d3dRes.Get(), 0, &_frameInMonitor);
|
||
++frameCount;
|
||
}
|
||
|
||
return 0;
|
||
}
|