mirror of
https://github.com/Blinue/Magpie.git
synced 2026-06-24 02:04:10 +00:00
339 lines
10 KiB
C++
339 lines
10 KiB
C++
#include "pch.h"
|
||
#include "EffectDrawer.h"
|
||
#include "Logger.h"
|
||
#include "Utils.h"
|
||
#include "App.h"
|
||
#include "DeviceResources.h"
|
||
#include "TextureLoader.h"
|
||
#include "StrUtils.h"
|
||
|
||
#pragma push_macro("_UNICODE")
|
||
#undef _UNICODE
|
||
// Conan 的 muparser 不含 UNICODE 支持
|
||
#include <muParser.h>
|
||
#pragma pop_macro("_UNICODE")
|
||
|
||
|
||
bool EffectDrawer::Initialize(
|
||
const EffectDesc& desc,
|
||
const EffectParams& params,
|
||
ID3D11Texture2D* inputTex,
|
||
ID3D11Texture2D** outputTex,
|
||
RECT* outputRect
|
||
) {
|
||
_desc = desc;
|
||
|
||
SIZE inputSize{};
|
||
{
|
||
D3D11_TEXTURE2D_DESC inputDesc;
|
||
inputTex->GetDesc(&inputDesc);
|
||
inputSize = { (LONG)inputDesc.Width, (LONG)inputDesc.Height };
|
||
}
|
||
|
||
const SIZE hostSize = Utils::GetSizeOfRect(App::Get().GetHostWndRect());;
|
||
bool _isLastEffect = desc.Flags & EFFECT_FLAG_LAST_EFFECT;
|
||
|
||
DeviceResources& dr = App::Get().GetDeviceResources();
|
||
auto d3dDevice = dr.GetD3DDevice();
|
||
|
||
static mu::Parser exprParser;
|
||
exprParser.DefineConst("INPUT_WIDTH", inputSize.cx);
|
||
exprParser.DefineConst("INPUT_HEIGHT", inputSize.cy);
|
||
|
||
SIZE outputSize{};
|
||
|
||
if (desc.outSizeExpr.first.empty()) {
|
||
outputSize = hostSize;
|
||
|
||
// scale 属性
|
||
// [+, +]：缩放比例
|
||
// [0, 0]：非等比例缩放到屏幕大小
|
||
// [-, -]：相对于屏幕能容纳的最大等比缩放的比例
|
||
|
||
static float DELTA = 1e-5f;
|
||
|
||
float scaleX = params.scale.first;
|
||
float scaleY = params.scale.second;
|
||
|
||
float fillScale = std::min(float(outputSize.cx) / inputSize.cx, float(outputSize.cy) / inputSize.cy);
|
||
|
||
if (scaleX >= DELTA) {
|
||
outputSize.cx = std::lroundf(inputSize.cx * scaleX);
|
||
} else if (scaleX < -DELTA) {
|
||
outputSize.cx = std::lroundf(inputSize.cx * fillScale * -scaleX);
|
||
}
|
||
|
||
if (scaleY >= DELTA) {
|
||
outputSize.cy = std::lroundf(inputSize.cy * scaleY);
|
||
} else if (scaleY < -DELTA) {
|
||
outputSize.cy = std::lroundf(inputSize.cy * fillScale * -scaleY);
|
||
}
|
||
} else {
|
||
assert(!desc.outSizeExpr.second.empty());
|
||
|
||
try {
|
||
exprParser.SetExpr(desc.outSizeExpr.first);
|
||
outputSize.cx = std::lround(exprParser.Eval());
|
||
|
||
exprParser.SetExpr(desc.outSizeExpr.second);
|
||
outputSize.cy = std::lround(exprParser.Eval());
|
||
} catch (const mu::ParserError& e) {
|
||
Logger::Get().Error(fmt::format("计算输出尺寸 {} 失败：{}", e.GetExpr(), e.GetMsg()));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
if (outputSize.cx <= 0 || outputSize.cy <= 0) {
|
||
Logger::Get().Error("非法的输出尺寸");
|
||
return false;
|
||
}
|
||
|
||
exprParser.DefineConst("OUTPUT_WIDTH", outputSize.cx);
|
||
exprParser.DefineConst("OUTPUT_HEIGHT", outputSize.cy);
|
||
|
||
// 大小必须为 4 的倍数
|
||
size_t builtinConstantCount = _isLastEffect ? 16 : 10 ;
|
||
_constants.resize((builtinConstantCount + desc.params.size() + 3) / 4 * 4);
|
||
// cbuffer __CB2 : register(b1) {
|
||
// uint2 __inputSize;
|
||
// uint2 __outputSize;
|
||
// float2 __inputPt;
|
||
// float2 __outputPt;
|
||
// float2 __scale;
|
||
// [uint2 __viewport;]
|
||
// [uint4 __offset;]
|
||
// [PARAMETERS]
|
||
// );
|
||
_constants[0].uintVal = inputSize.cx;
|
||
_constants[1].uintVal = inputSize.cy;
|
||
_constants[2].uintVal = outputSize.cx;
|
||
_constants[3].uintVal = outputSize.cy;
|
||
_constants[4].floatVal = 1.0f / inputSize.cx;
|
||
_constants[5].floatVal = 1.0f / inputSize.cy;
|
||
_constants[6].floatVal = 1.0f / outputSize.cx;
|
||
_constants[7].floatVal = 1.0f / outputSize.cy;
|
||
_constants[8].floatVal = outputSize.cx / (FLOAT)inputSize.cx;
|
||
_constants[9].floatVal = outputSize.cy / (FLOAT)inputSize.cy;
|
||
|
||
if (_isLastEffect) {
|
||
// 输出尺寸可能比主窗口更大
|
||
RECT virtualOutputRect{};
|
||
virtualOutputRect.left = (hostSize.cx - outputSize.cx) / 2;
|
||
virtualOutputRect.top = (hostSize.cy - hostSize.cy) / 2;
|
||
virtualOutputRect.right = virtualOutputRect.left + outputSize.cx;
|
||
virtualOutputRect.bottom = virtualOutputRect.top + outputSize.cy;
|
||
|
||
RECT realOutputRect = {
|
||
std::max(0L, virtualOutputRect.left),
|
||
std::max(0L, virtualOutputRect.top),
|
||
std::min(hostSize.cx, virtualOutputRect.right),
|
||
std::min(hostSize.cy, virtualOutputRect.bottom)
|
||
};
|
||
|
||
if (outputRect) {
|
||
*outputRect = realOutputRect;
|
||
}
|
||
|
||
_constants[10].uintVal = realOutputRect.right - realOutputRect.left;
|
||
_constants[11].uintVal = realOutputRect.bottom - realOutputRect.top;
|
||
_constants[12].uintVal = -std::min(0L, virtualOutputRect.left);
|
||
_constants[13].uintVal = -std::min(0L, virtualOutputRect.top);
|
||
_constants[14].uintVal = realOutputRect.left;
|
||
_constants[15].uintVal = realOutputRect.top;
|
||
} else {
|
||
if (outputRect) {
|
||
*outputRect = RECT{ 0,0,outputSize.cx, outputSize.cy };
|
||
}
|
||
}
|
||
|
||
// 填入参数
|
||
for (UINT i = 0; i < desc.params.size(); ++i) {
|
||
const auto& paramDesc = desc.params[i];
|
||
auto it = params.params.find(i);
|
||
|
||
if (paramDesc.type == EffectConstantType::Float) {
|
||
float value;
|
||
|
||
if (it == params.params.end()) {
|
||
value = std::get<float>(paramDesc.defaultValue);
|
||
} else {
|
||
value= it->second.floatVal;
|
||
|
||
if ((paramDesc.minValue.index() == 1 && value < std::get<float>(paramDesc.minValue))
|
||
|| (paramDesc.maxValue.index() == 1 && value > std::get<float>(paramDesc.maxValue))
|
||
) {
|
||
Logger::Get().Error(fmt::format("参数 {} 的值非法", paramDesc.name));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
_constants[builtinConstantCount + i].floatVal = value;
|
||
} else {
|
||
int value;
|
||
|
||
if (it == params.params.end()) {
|
||
value = std::get<int>(paramDesc.defaultValue);
|
||
} else {
|
||
value = it->second.intVal;
|
||
|
||
if ((paramDesc.minValue.index() == 2 && value < std::get<int>(paramDesc.minValue))
|
||
|| (paramDesc.maxValue.index() == 2 && value > std::get<int>(paramDesc.maxValue))
|
||
) {
|
||
Logger::Get().Error(fmt::format("参数 {} 的值非法", paramDesc.name));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
_constants[builtinConstantCount + i].intVal = value;
|
||
}
|
||
}
|
||
|
||
D3D11_BUFFER_DESC bd{};
|
||
bd.Usage = D3D11_USAGE_DEFAULT;
|
||
bd.ByteWidth = 4 * (UINT)_constants.size();
|
||
bd.BindFlags = D3D11_BIND_CONSTANT_BUFFER;
|
||
|
||
D3D11_SUBRESOURCE_DATA initData{};
|
||
initData.pSysMem = _constants.data();
|
||
|
||
HRESULT hr = d3dDevice->CreateBuffer(&bd, &initData, _constantBuffer.put());
|
||
if (FAILED(hr)) {
|
||
Logger::Get().ComError("CreateBuffer 失败", hr);
|
||
return false;
|
||
}
|
||
|
||
_samplers.resize(desc.samplers.size());
|
||
for (UINT i = 0; i < _samplers.size(); ++i) {
|
||
const EffectSamplerDesc& samDesc = desc.samplers[i];
|
||
if (!dr.GetSampler(
|
||
samDesc.filterType == EffectSamplerFilterType::Linear ? D3D11_FILTER_MIN_MAG_MIP_LINEAR : D3D11_FILTER_MIN_MAG_MIP_POINT,
|
||
samDesc.addressType == EffectSamplerAddressType::Clamp ? D3D11_TEXTURE_ADDRESS_CLAMP : D3D11_TEXTURE_ADDRESS_WRAP,
|
||
&_samplers[i])
|
||
) {
|
||
Logger::Get().Error(fmt::format("创建采样器 {} 失败", samDesc.name));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
// 创建中间纹理
|
||
// 第一个为 INPUT，最后一个为 OUTPUT
|
||
_textures.resize(desc.textures.size() + 1);
|
||
_textures[0].copy_from(inputTex);
|
||
for (size_t i = 1; i < desc.textures.size(); ++i) {
|
||
const EffectIntermediateTextureDesc& texDesc = desc.textures[i];
|
||
|
||
if (!texDesc.source.empty()) {
|
||
// 从文件加载纹理
|
||
_textures[i] = TextureLoader::Load((L"effects\\" + StrUtils::UTF8ToUTF16(texDesc.source)).c_str());
|
||
if (!_textures[i]) {
|
||
Logger::Get().Error(fmt::format("加载纹理 {} 失败", texDesc.source));
|
||
return false;
|
||
}
|
||
} else {
|
||
SIZE texSize{};
|
||
try {
|
||
exprParser.SetExpr(texDesc.sizeExpr.first);
|
||
texSize.cx = std::lround(exprParser.Eval());
|
||
exprParser.SetExpr(texDesc.sizeExpr.second);
|
||
texSize.cy = std::lround(exprParser.Eval());
|
||
} catch (const mu::ParserError& e) {
|
||
Logger::Get().Error(fmt::format("计算中间纹理尺寸 {} 失败：{}", e.GetExpr(), e.GetMsg()));
|
||
return false;
|
||
}
|
||
|
||
if (texSize.cx <= 0 || texSize.cy <= 0) {
|
||
Logger::Get().Error("非法的中间纹理尺寸");
|
||
return false;
|
||
}
|
||
|
||
D3D11_TEXTURE2D_DESC desc{};
|
||
desc.Format = EffectIntermediateTextureDesc::DXGI_FORMAT_MAP[(UINT)texDesc.format];
|
||
desc.Width = texSize.cx;
|
||
desc.Height = texSize.cy;
|
||
desc.Usage = D3D11_USAGE_DEFAULT;
|
||
desc.MipLevels = 1;
|
||
desc.ArraySize = 1;
|
||
desc.SampleDesc.Count = 1;
|
||
desc.SampleDesc.Quality = 0;
|
||
desc.BindFlags = D3D11_BIND_SHADER_RESOURCE | D3D11_BIND_UNORDERED_ACCESS;
|
||
HRESULT hr = d3dDevice->CreateTexture2D(&desc, nullptr, _textures[i].put());
|
||
if (FAILED(hr)) {
|
||
Logger::Get().ComError("创建 Texture2D 失败", hr);
|
||
return false;
|
||
}
|
||
}
|
||
}
|
||
|
||
if (!_isLastEffect) {
|
||
// 创建输出纹理
|
||
D3D11_TEXTURE2D_DESC desc{};
|
||
desc.Format = DXGI_FORMAT_B8G8R8A8_UNORM;
|
||
desc.Width = outputSize.cx;
|
||
desc.Height = outputSize.cy;
|
||
desc.Usage = D3D11_USAGE_DEFAULT;
|
||
desc.MipLevels = 1;
|
||
desc.ArraySize = 1;
|
||
desc.SampleDesc.Count = 1;
|
||
desc.SampleDesc.Quality = 0;
|
||
desc.BindFlags = D3D11_BIND_SHADER_RESOURCE | D3D11_BIND_UNORDERED_ACCESS;
|
||
HRESULT hr = d3dDevice->CreateTexture2D(&desc, nullptr, _textures.back().put());
|
||
if (FAILED(hr)) {
|
||
Logger::Get().ComError("创建 Texture2D 失败", hr);
|
||
return false;
|
||
}
|
||
} else {
|
||
_textures.back().copy_from(dr.GetBackBuffer());
|
||
}
|
||
|
||
*outputTex = _textures.back().get();
|
||
|
||
_shaders.resize(desc.passes.size());
|
||
_srvs.resize(desc.passes.size());
|
||
_uavs.resize(desc.passes.size());
|
||
for (UINT i = 0; i < _shaders.size(); ++i) {
|
||
const EffectPassDesc& passDesc = desc.passes[i];
|
||
|
||
HRESULT hr = d3dDevice->CreateComputeShader(
|
||
passDesc.cso->GetBufferPointer(), passDesc.cso->GetBufferSize(), nullptr, _shaders[i].put());
|
||
if (FAILED(hr)) {
|
||
Logger::Get().ComError("创建计算着色器失败", hr);
|
||
return false;
|
||
}
|
||
|
||
_srvs[i].resize(passDesc.inputs.size() * 2);
|
||
for (UINT j = 0; j < passDesc.inputs.size(); ++j) {
|
||
if (!dr.GetShaderResourceView(_textures[passDesc.inputs[j]].get(), &_srvs[i][j])) {
|
||
Logger::Get().Error("GetShaderResourceView 失败");
|
||
return false;
|
||
}
|
||
}
|
||
|
||
if (!passDesc.outputs.empty()) {
|
||
_uavs[i].resize(passDesc.outputs.size());
|
||
for (UINT j = 0; j < passDesc.outputs.size(); ++j) {
|
||
if (!dr.GetUnorderedAccessView(_textures[passDesc.outputs[j]].get(), &_uavs[i][0])) {
|
||
Logger::Get().Error("GetUnorderedAccessView 失败");
|
||
return false;
|
||
}
|
||
}
|
||
} else {
|
||
// 最后一个 pass 输出到 OUTPUT
|
||
_uavs[i].resize(1);
|
||
if (!dr.GetUnorderedAccessView(_textures.back().get(), &_uavs[i][0])) {
|
||
Logger::Get().Error("GetUnorderedAccessView 失败");
|
||
return false;
|
||
}
|
||
}
|
||
}
|
||
|
||
if (_isLastEffect) {
|
||
// 为光标纹理预留空间
|
||
_srvs.back().resize(_srvs.back().size() + 2);
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|
||
void EffectDrawer::Draw()
|
||
{
|
||
}
|