mirror of
https://github.com/gnif/LookingGlass.git
synced 2026-08-09 08:41:31 +00:00
[idd] project: organize driver sources by responsibility
Group the IDD sources and Visual Studio filters by subsystem. Split the device and swap-chain implementations into focused units, rename the context classes, and reduce header coupling.
This commit is contained in:
Geoffrey McRae
195
idd/LGIdd/postprocess/effect/CComputeEffect.cpp
Normal file
195
idd/LGIdd/postprocess/effect/CComputeEffect.cpp
Normal file
@@ -0,0 +1,195 @@
|
||||
/**
|
||||
* Looking Glass
|
||||
* Copyright © 2017-2026 The Looking Glass Authors
|
||||
* https://looking-glass.io
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License as published by the Free
|
||||
* Software Foundation; either version 2 of the License, or (at your option)
|
||||
* any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but WITHOUT
|
||||
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||||
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
||||
* more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License along
|
||||
* with this program; if not, write to the Free Software Foundation, Inc., 59
|
||||
* Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
*/
|
||||
|
||||
#include "CComputeEffect.h"
|
||||
|
||||
#include "CDebug.h"
|
||||
|
||||
#include <d3dcompiler.h>
|
||||
#include <cstring>
|
||||
|
||||
namespace PostProcessUtil
|
||||
{
|
||||
static bool CreateDefaultResource(const ComPtr<ID3D12Device3>& device,
|
||||
const D3D12_RESOURCE_DESC& desc, ComPtr<ID3D12Resource>& resource)
|
||||
{
|
||||
D3D12_HEAP_PROPERTIES heapProps = {};
|
||||
heapProps.Type = D3D12_HEAP_TYPE_DEFAULT;
|
||||
heapProps.CPUPageProperty = D3D12_CPU_PAGE_PROPERTY_UNKNOWN;
|
||||
heapProps.MemoryPoolPreference = D3D12_MEMORY_POOL_UNKNOWN;
|
||||
heapProps.CreationNodeMask = 1;
|
||||
heapProps.VisibleNodeMask = 1;
|
||||
|
||||
HRESULT hr = device->CreateCommittedResource(
|
||||
&heapProps,
|
||||
D3D12_HEAP_FLAG_CREATE_NOT_ZEROED,
|
||||
&desc,
|
||||
D3D12_RESOURCE_STATE_COMMON,
|
||||
nullptr,
|
||||
IID_PPV_ARGS(&resource));
|
||||
if (FAILED(hr))
|
||||
{
|
||||
DEBUG_ERROR_HR(hr, "Failed to create post-processing destination resource");
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CreateDefaultTexture(const ComPtr<ID3D12Device3>& device,
|
||||
const D3D12_RESOURCE_DESC& desc, ComPtr<ID3D12Resource>& resource)
|
||||
{
|
||||
return CreateDefaultResource(device, desc, resource);
|
||||
}
|
||||
|
||||
bool CreateDefaultBuffer(const ComPtr<ID3D12Device3>& device,
|
||||
UINT64 size, ComPtr<ID3D12Resource>& resource)
|
||||
{
|
||||
D3D12_RESOURCE_DESC desc = {};
|
||||
desc.Dimension = D3D12_RESOURCE_DIMENSION_BUFFER;
|
||||
desc.Width = size;
|
||||
desc.Height = 1;
|
||||
desc.DepthOrArraySize = 1;
|
||||
desc.MipLevels = 1;
|
||||
desc.SampleDesc.Count = 1;
|
||||
desc.Layout = D3D12_TEXTURE_LAYOUT_ROW_MAJOR;
|
||||
desc.Flags = D3D12_RESOURCE_FLAG_ALLOW_UNORDERED_ACCESS;
|
||||
return CreateDefaultResource(device, desc, resource);
|
||||
}
|
||||
}
|
||||
|
||||
bool CComputeEffect::InitCompute(const ComPtr<ID3D12Device3>& device,
|
||||
const D3D12_DESCRIPTOR_RANGE * ranges, UINT rangeCount,
|
||||
const D3D12_STATIC_SAMPLER_DESC * samplers, UINT samplerCount,
|
||||
const char * shader)
|
||||
{
|
||||
D3D12_ROOT_PARAMETER rootParam = {};
|
||||
rootParam.ParameterType = D3D12_ROOT_PARAMETER_TYPE_DESCRIPTOR_TABLE;
|
||||
rootParam.ShaderVisibility = D3D12_SHADER_VISIBILITY_ALL;
|
||||
rootParam.DescriptorTable.NumDescriptorRanges = rangeCount;
|
||||
rootParam.DescriptorTable.pDescriptorRanges = ranges;
|
||||
|
||||
D3D12_VERSIONED_ROOT_SIGNATURE_DESC rootSignatureDesc = {};
|
||||
rootSignatureDesc.Version = D3D_ROOT_SIGNATURE_VERSION_1;
|
||||
rootSignatureDesc.Desc_1_0.NumParameters = 1;
|
||||
rootSignatureDesc.Desc_1_0.pParameters = &rootParam;
|
||||
rootSignatureDesc.Desc_1_0.NumStaticSamplers = samplerCount;
|
||||
rootSignatureDesc.Desc_1_0.pStaticSamplers = samplers;
|
||||
rootSignatureDesc.Desc_1_0.Flags = D3D12_ROOT_SIGNATURE_FLAG_NONE;
|
||||
|
||||
ComPtr<ID3DBlob> blob;
|
||||
ComPtr<ID3DBlob> error;
|
||||
HRESULT hr = D3D12SerializeVersionedRootSignature(
|
||||
&rootSignatureDesc, &blob, &error);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
DEBUG_ERROR_HR(hr, "Failed to serialize post-processing root signature");
|
||||
if (error)
|
||||
DEBUG_ERROR("%s", (const char *)error->GetBufferPointer());
|
||||
return false;
|
||||
}
|
||||
|
||||
hr = device->CreateRootSignature(
|
||||
0,
|
||||
blob->GetBufferPointer(),
|
||||
blob->GetBufferSize(),
|
||||
IID_PPV_ARGS(&m_rootSignature));
|
||||
if (FAILED(hr))
|
||||
{
|
||||
DEBUG_ERROR_HR(hr, "Failed to create post-processing root signature");
|
||||
return false;
|
||||
}
|
||||
|
||||
blob.Reset();
|
||||
error.Reset();
|
||||
hr = D3DCompile(
|
||||
shader,
|
||||
std::strlen(shader),
|
||||
nullptr,
|
||||
nullptr,
|
||||
nullptr,
|
||||
"main",
|
||||
"cs_5_0",
|
||||
0,
|
||||
0,
|
||||
&blob,
|
||||
&error);
|
||||
if (FAILED(hr))
|
||||
{
|
||||
DEBUG_ERROR_HR(hr, "Failed to compile post-processing shader");
|
||||
if (error)
|
||||
DEBUG_ERROR("%s", (const char *)error->GetBufferPointer());
|
||||
return false;
|
||||
}
|
||||
|
||||
D3D12_COMPUTE_PIPELINE_STATE_DESC psoDesc = {};
|
||||
psoDesc.pRootSignature = m_rootSignature.Get();
|
||||
psoDesc.CS.pShaderBytecode = blob->GetBufferPointer();
|
||||
psoDesc.CS.BytecodeLength = blob->GetBufferSize();
|
||||
|
||||
hr = device->CreateComputePipelineState(&psoDesc, IID_PPV_ARGS(&m_pso));
|
||||
if (FAILED(hr))
|
||||
{
|
||||
DEBUG_ERROR_HR(hr, "Failed to create post-processing PSO");
|
||||
return false;
|
||||
}
|
||||
|
||||
UINT descriptorCount = 0;
|
||||
for (UINT i = 0; i < rangeCount; ++i)
|
||||
descriptorCount += ranges[i].NumDescriptors;
|
||||
|
||||
D3D12_DESCRIPTOR_HEAP_DESC heapDesc = {};
|
||||
heapDesc.Type = D3D12_DESCRIPTOR_HEAP_TYPE_CBV_SRV_UAV;
|
||||
heapDesc.NumDescriptors = descriptorCount;
|
||||
heapDesc.Flags = D3D12_DESCRIPTOR_HEAP_FLAG_SHADER_VISIBLE;
|
||||
|
||||
hr = device->CreateDescriptorHeap(&heapDesc, IID_PPV_ARGS(&m_descHeap));
|
||||
if (FAILED(hr))
|
||||
{
|
||||
DEBUG_ERROR_HR(hr, "Failed to create post-processing descriptor heap");
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void CComputeEffect::Bind(const ComPtr<ID3D12GraphicsCommandList>& commandList)
|
||||
{
|
||||
ID3D12DescriptorHeap * heaps[] = { m_descHeap.Get() };
|
||||
commandList->SetDescriptorHeaps(1, heaps);
|
||||
commandList->SetPipelineState(m_pso.Get());
|
||||
commandList->SetComputeRootSignature(m_rootSignature.Get());
|
||||
commandList->SetComputeRootDescriptorTable(
|
||||
0, m_descHeap->GetGPUDescriptorHandleForHeapStart());
|
||||
}
|
||||
|
||||
void CComputeEffect::TransitionDst(
|
||||
const ComPtr<ID3D12GraphicsCommandList>& commandList,
|
||||
D3D12_RESOURCE_STATES before, D3D12_RESOURCE_STATES after)
|
||||
{
|
||||
D3D12_RESOURCE_BARRIER barrier = {};
|
||||
barrier.Type = D3D12_RESOURCE_BARRIER_TYPE_TRANSITION;
|
||||
barrier.Flags = D3D12_RESOURCE_BARRIER_FLAG_NONE;
|
||||
barrier.Transition.pResource = m_dst.Get();
|
||||
barrier.Transition.StateBefore = before;
|
||||
barrier.Transition.StateAfter = after;
|
||||
barrier.Transition.Subresource = D3D12_RESOURCE_BARRIER_ALL_SUBRESOURCES;
|
||||
commandList->ResourceBarrier(1, &barrier);
|
||||
}
|
||||
Reference in New Issue
Block a user