mirror of
https://github.com/Blinue/Magpie.git
synced 2026-06-24 02:04:10 +00:00
423 lines
12 KiB
C++
423 lines
12 KiB
C++
#include "pch.h"
|
||
#include "Renderer.h"
|
||
#include "App.h"
|
||
#include "Utils.h"
|
||
#include "shaders/FillVS.h"
|
||
#include "shaders/CopyPS.h"
|
||
#include "shaders/SimpleVS.h"
|
||
#include <VertexTypes.h>
|
||
|
||
extern std::shared_ptr<spdlog::logger> logger;
|
||
|
||
|
||
|
||
Renderer::~Renderer() {
|
||
CloseHandle(_frameLatencyWaitableObject);
|
||
}
|
||
|
||
bool Renderer::Initialize() {
|
||
if (!_InitD3D()) {
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
bool Renderer::InitializeEffectsAndCursor() {
|
||
Effect& effect = _effects.emplace_back();
|
||
if (!effect.InitializeFsr()) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, "初始化 Effect 失败");
|
||
return false;
|
||
}
|
||
|
||
_effectInput = App::GetInstance().GetFrameSource().GetOutput();
|
||
D3D11_TEXTURE2D_DESC inputDesc;
|
||
_effectInput->GetDesc(&inputDesc);
|
||
SIZE hostSize = App::GetInstance().GetHostWndSize();
|
||
|
||
// 等比缩放到最大
|
||
float fillScale = std::min(float(hostSize.cx) / inputDesc.Width, float(hostSize.cy) / inputDesc.Height);
|
||
SIZE outputSize = { lroundf(inputDesc.Width * fillScale), lroundf(inputDesc.Height * fillScale) };
|
||
|
||
RECT destRect{};
|
||
destRect.left = (hostSize.cx - outputSize.cx) / 2;
|
||
destRect.right = destRect.left + outputSize.cx;
|
||
destRect.top = (hostSize.cy - outputSize.cy) / 2;
|
||
destRect.bottom = destRect.top + outputSize.cy;
|
||
|
||
effect.SetOutputSize(outputSize);
|
||
|
||
if (!effect.Build(_effectInput, _backBuffer)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, "构建 Effect 失败");
|
||
return false;
|
||
}
|
||
|
||
if (App::GetInstance().IsShowFPS()) {
|
||
if (!_frameRateRenderer.Initialize(_backBuffer, destRect)) {
|
||
return false;
|
||
}
|
||
}
|
||
|
||
if (!_cursorRenderer.Initialize(_backBuffer, destRect)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, "构建 CursorRenderer 失败");
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
|
||
void Renderer::Render() {
|
||
if (!_waitingForNextFrame) {
|
||
WaitForSingleObject(_frameLatencyWaitableObject, 1000);
|
||
}
|
||
|
||
if (!_CheckSrcState()) {
|
||
SPDLOG_LOGGER_INFO(logger, "源窗口状态改变，退出全屏");
|
||
DestroyWindow(App::GetInstance().GetHwndHost());
|
||
return;
|
||
}
|
||
|
||
_waitingForNextFrame = !App::GetInstance().GetFrameSource().Update();
|
||
if (_waitingForNextFrame) {
|
||
return;
|
||
}
|
||
|
||
_d3dDC->ClearState();
|
||
// 所有渲染都使用三角形带拓扑
|
||
_d3dDC->IASetPrimitiveTopology(D3D11_PRIMITIVE_TOPOLOGY_TRIANGLESTRIP);
|
||
|
||
for (Effect& effect : _effects) {
|
||
effect.Draw();
|
||
}
|
||
|
||
if (App::GetInstance().IsShowFPS()) {
|
||
_frameRateRenderer.Draw();
|
||
}
|
||
|
||
_cursorRenderer.Draw();
|
||
|
||
_dxgiSwapChain->Present(0, 0);
|
||
}
|
||
|
||
bool Renderer::GetRenderTargetView(ID3D11Texture2D* texture, ID3D11RenderTargetView** result) {
|
||
auto it = _rtvMap.find(texture);
|
||
if (it != _rtvMap.end()) {
|
||
*result = it->second.Get();
|
||
return true;
|
||
}
|
||
|
||
ComPtr<ID3D11RenderTargetView>& r = _rtvMap[texture];
|
||
HRESULT hr = _d3dDevice->CreateRenderTargetView(texture, nullptr, &r);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("CreateRenderTargetView 失败", hr));
|
||
return false;
|
||
} else {
|
||
*result = r.Get();
|
||
return true;
|
||
}
|
||
}
|
||
|
||
bool Renderer::GetShaderResourceView(ID3D11Texture2D* texture, ID3D11ShaderResourceView** result) {
|
||
auto it = _srvMap.find(texture);
|
||
if (it != _srvMap.end()) {
|
||
*result = it->second.Get();
|
||
return true;
|
||
}
|
||
|
||
ComPtr<ID3D11ShaderResourceView>& r = _srvMap[texture];
|
||
HRESULT hr = _d3dDevice->CreateShaderResourceView(texture, nullptr, &r);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("CreateShaderResourceView 失败", hr));
|
||
return false;
|
||
} else {
|
||
*result = r.Get();
|
||
return true;
|
||
}
|
||
}
|
||
|
||
bool Renderer::SetFillVS() {
|
||
if (!_fillVS) {
|
||
HRESULT hr = _d3dDevice->CreateVertexShader(FillVSShaderByteCode, sizeof(FillVSShaderByteCode), nullptr, &_fillVS);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("创建 FillVS 失败", hr));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
_d3dDC->IASetInputLayout(nullptr);
|
||
_d3dDC->IASetVertexBuffers(0, 0, nullptr, nullptr, nullptr);
|
||
_d3dDC->VSSetShader(_fillVS.Get(), nullptr, 0);
|
||
|
||
return true;
|
||
}
|
||
|
||
|
||
bool Renderer::SetCopyPS(ID3D11SamplerState* sampler, ID3D11ShaderResourceView* input) {
|
||
if (!_copyPS) {
|
||
HRESULT hr = _d3dDevice->CreatePixelShader(CopyPSShaderByteCode, sizeof(CopyPSShaderByteCode), nullptr, &_copyPS);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("创建 CopyPS 失败", hr));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
_d3dDC->PSSetShader(_copyPS.Get(), nullptr, 0);
|
||
_d3dDC->PSSetConstantBuffers(0, 0, nullptr);
|
||
_d3dDC->PSSetShaderResources(0, 1, &input);
|
||
_d3dDC->PSSetSamplers(0, 1, &sampler);
|
||
|
||
return true;
|
||
}
|
||
|
||
bool Renderer::SetSimpleVS(ID3D11Buffer* simpleVB) {
|
||
if (!_simpleVS) {
|
||
HRESULT hr = _d3dDevice->CreateVertexShader(SimpleVSShaderByteCode, sizeof(SimpleVSShaderByteCode), nullptr, &_simpleVS);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("创建 SimpleVS 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
hr = _d3dDevice->CreateInputLayout(
|
||
VertexPositionTexture::InputElements,
|
||
VertexPositionTexture::InputElementCount,
|
||
SimpleVSShaderByteCode,
|
||
sizeof(SimpleVSShaderByteCode),
|
||
&_simpleIL
|
||
);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, fmt::sprintf("创建 SimpleVS 输入布局失败\n\tHRESULT：0x%X", hr));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
_d3dDC->IASetInputLayout(_simpleIL.Get());
|
||
|
||
UINT stride = sizeof(VertexPositionTexture);
|
||
UINT offset = 0;
|
||
_d3dDC->IASetVertexBuffers(0, 1, &simpleVB, &stride, &offset);
|
||
|
||
_d3dDC->VSSetShader(_simpleVS.Get(), nullptr, 0);
|
||
|
||
return true;
|
||
}
|
||
|
||
bool Renderer::_InitD3D() {
|
||
HRESULT hr;
|
||
const SIZE hostSize = App::GetInstance().GetHostWndSize();
|
||
|
||
{
|
||
UINT createDeviceFlags = 0;
|
||
#ifdef _DEBUG
|
||
createDeviceFlags |= D3D11_CREATE_DEVICE_DEBUG;
|
||
#endif
|
||
|
||
D3D_FEATURE_LEVEL featureLevels[] = {
|
||
D3D_FEATURE_LEVEL_11_0,
|
||
D3D_FEATURE_LEVEL_10_1,
|
||
D3D_FEATURE_LEVEL_10_0,
|
||
D3D_FEATURE_LEVEL_9_3,
|
||
D3D_FEATURE_LEVEL_9_2,
|
||
D3D_FEATURE_LEVEL_9_1,
|
||
};
|
||
|
||
ComPtr<ID3D11Device> d3dDevice;
|
||
ComPtr<ID3D11DeviceContext> d3dDC;
|
||
D3D_FEATURE_LEVEL featureLevel;
|
||
hr = D3D11CreateDevice(
|
||
nullptr,
|
||
D3D_DRIVER_TYPE_HARDWARE,
|
||
nullptr,
|
||
createDeviceFlags,
|
||
featureLevels,
|
||
1,
|
||
D3D11_SDK_VERSION,
|
||
&d3dDevice,
|
||
&featureLevel,
|
||
&d3dDC
|
||
);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("D3D11CreateDevice 失败", hr));
|
||
return false;
|
||
}
|
||
SPDLOG_LOGGER_INFO(logger, fmt::format("已创建 D3D Device\n\t功能级别：{}", featureLevel));
|
||
|
||
hr = d3dDevice.As<ID3D11Device1>(&_d3dDevice);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("获取 ID3D11Device5 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
hr = d3dDC.As<ID3D11DeviceContext1>(&_d3dDC);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("获取 ID3D11DeviceContext4 失败", hr));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
{
|
||
ComPtr<IDXGIFactory2> dxgiFactory;
|
||
|
||
hr = _d3dDevice.As<IDXGIDevice4>(&_dxgiDevice);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("获取 IDXGIDevice 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
ComPtr<IDXGIAdapter> dxgiAdapter;
|
||
hr = _dxgiDevice->GetAdapter(&dxgiAdapter);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("获取 IDXGIAdapter 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
hr = dxgiAdapter->GetParent(IID_PPV_ARGS(&dxgiFactory));
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("获取 IDXGIFactory2 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
DXGI_SWAP_CHAIN_DESC1 sd = {};
|
||
sd.Width = hostSize.cx;
|
||
sd.Height = hostSize.cy;
|
||
sd.Format = DXGI_FORMAT_B8G8R8A8_UNORM;
|
||
sd.SampleDesc.Count = 1;
|
||
sd.SampleDesc.Quality = 0;
|
||
sd.BufferUsage = DXGI_USAGE_RENDER_TARGET_OUTPUT | DXGI_USAGE_SHADER_INPUT;
|
||
sd.BufferCount = 2;
|
||
sd.SwapEffect = DXGI_SWAP_EFFECT_FLIP_DISCARD;
|
||
sd.Flags = DXGI_SWAP_CHAIN_FLAG_FRAME_LATENCY_WAITABLE_OBJECT;
|
||
|
||
ComPtr<IDXGISwapChain1> dxgiSwapChain = nullptr;
|
||
hr = dxgiFactory->CreateSwapChainForHwnd(
|
||
_d3dDevice.Get(),
|
||
App::GetInstance().GetHwndHost(),
|
||
&sd,
|
||
nullptr,
|
||
nullptr,
|
||
&dxgiSwapChain
|
||
);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("创建交换链失败", hr));
|
||
return false;
|
||
}
|
||
|
||
hr = dxgiSwapChain.As<IDXGISwapChain4>(&_dxgiSwapChain);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("获取 IDXGISwapChain4 失败", hr));
|
||
return false;
|
||
}
|
||
|
||
_frameLatencyWaitableObject = _dxgiSwapChain->GetFrameLatencyWaitableObject();
|
||
|
||
hr = dxgiFactory->MakeWindowAssociation(App::GetInstance().GetHwndHost(), DXGI_MWA_NO_ALT_ENTER);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("MakeWindowAssociation 失败", hr));
|
||
}
|
||
}
|
||
|
||
hr = _dxgiSwapChain->GetBuffer(0, IID_PPV_ARGS(&_backBuffer));
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("获取后缓冲区失败", hr));
|
||
return false;
|
||
}
|
||
|
||
SPDLOG_LOGGER_INFO(logger, "Renderer 初始化完成");
|
||
return true;
|
||
}
|
||
|
||
bool Renderer::_CheckSrcState() {
|
||
HWND hwndSrc = App::GetInstance().GetHwndSrc();
|
||
if (GetForegroundWindow() != hwndSrc) {
|
||
SPDLOG_LOGGER_INFO(logger, "前台窗口已改变");
|
||
return false;
|
||
}
|
||
|
||
RECT rect = Utils::GetClientScreenRect(hwndSrc);
|
||
if (App::GetInstance().GetSrcClientRect() != rect) {
|
||
SPDLOG_LOGGER_INFO(logger, "源窗口位置或大小改变");
|
||
return false;
|
||
}
|
||
if (Utils::GetWindowShowCmd(hwndSrc) != SW_NORMAL) {
|
||
SPDLOG_LOGGER_INFO(logger, "源窗口显示状态改变");
|
||
return false;
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
bool Renderer::SetAlphaBlend(bool enable) {
|
||
if (!enable) {
|
||
_d3dDC->OMSetBlendState(nullptr, nullptr, 0xffffffff);
|
||
return true;
|
||
}
|
||
|
||
if (!_alphaBlendState) {
|
||
D3D11_BLEND_DESC desc{};
|
||
desc.RenderTarget[0].BlendEnable = TRUE;
|
||
desc.RenderTarget[0].SrcBlend = D3D11_BLEND_SRC_ALPHA;
|
||
desc.RenderTarget[0].SrcBlendAlpha = D3D11_BLEND_ONE;
|
||
desc.RenderTarget[0].DestBlend = D3D11_BLEND_INV_SRC_ALPHA;
|
||
desc.RenderTarget[0].DestBlendAlpha = D3D11_BLEND_ZERO;
|
||
desc.RenderTarget[0].BlendOp = desc.RenderTarget[0].BlendOpAlpha = D3D11_BLEND_OP_ADD;
|
||
desc.RenderTarget[0].RenderTargetWriteMask = D3D11_COLOR_WRITE_ENABLE_ALL;
|
||
|
||
HRESULT hr = _d3dDevice->CreateBlendState(&desc, &_alphaBlendState);
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_CRITICAL(logger, MakeComErrorMsg("CreateBlendState 失败", hr));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
_d3dDC->OMSetBlendState(_alphaBlendState.Get(), nullptr, 0xffffffff);
|
||
return true;
|
||
}
|
||
|
||
bool Renderer::GetSampler(FilterType filterType, ID3D11SamplerState** result) {
|
||
if (filterType == FilterType::LINEAR) {
|
||
if (!_linearSampler) {
|
||
D3D11_SAMPLER_DESC desc{};
|
||
desc.Filter = D3D11_FILTER_MIN_MAG_MIP_LINEAR;
|
||
desc.AddressU = D3D11_TEXTURE_ADDRESS_CLAMP;
|
||
desc.AddressV = D3D11_TEXTURE_ADDRESS_CLAMP;
|
||
desc.AddressW = D3D11_TEXTURE_ADDRESS_CLAMP;
|
||
desc.ComparisonFunc = D3D11_COMPARISON_NEVER;
|
||
desc.MinLOD = 0;
|
||
desc.MaxLOD = 0;
|
||
HRESULT hr = _d3dDevice->CreateSamplerState(&desc, &_linearSampler);
|
||
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("创建 ID3D11SamplerState 出错", hr));
|
||
return false;
|
||
} else {
|
||
SPDLOG_LOGGER_INFO(logger, "已创建 ID3D11SamplerState\n\tFilter：D3D11_FILTER_MIN_MAG_MIP_LINEAR");
|
||
}
|
||
}
|
||
|
||
*result = _linearSampler.Get();
|
||
} else {
|
||
if (!_pointSampler) {
|
||
D3D11_SAMPLER_DESC desc{};
|
||
desc.Filter = D3D11_FILTER_MIN_MAG_MIP_POINT;
|
||
desc.AddressU = D3D11_TEXTURE_ADDRESS_CLAMP;
|
||
desc.AddressV = D3D11_TEXTURE_ADDRESS_CLAMP;
|
||
desc.AddressW = D3D11_TEXTURE_ADDRESS_CLAMP;
|
||
desc.ComparisonFunc = D3D11_COMPARISON_NEVER;
|
||
desc.MinLOD = 0;
|
||
desc.MaxLOD = 0;
|
||
HRESULT hr = _d3dDevice->CreateSamplerState(&desc, &_pointSampler);
|
||
|
||
if (FAILED(hr)) {
|
||
SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("创建 ID3D11SamplerState 出错", hr));
|
||
return false;
|
||
} else {
|
||
SPDLOG_LOGGER_INFO(logger, "已创建 ID3D11SamplerState\n\tFilter：D3D11_FILTER_MIN_MAG_MIP_POINT");
|
||
}
|
||
}
|
||
|
||
*result = _pointSampler.Get();
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|