56 lines
1.6 KiB
C
56 lines
1.6 KiB
C
|
#pragma once
|
||
|
|
|
||
|
|
#ifndef NOMINMAX
|
||
|
|
#define NOMINMAX
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include <XCEngine/RHI/D3D12/D3D12DescriptorHeap.h>
|
||
|
|
#include <XCEngine/RHI/D3D12/D3D12Device.h>
|
||
|
|
#include <XCEngine/RHI/D3D12/D3D12Texture.h>
|
||
|
|
#include <XCEngine/RHI/RHIDescriptorPool.h>
|
||
|
|
#include <XCEngine/RHI/RHIDevice.h>
|
||
|
|
#include <XCEngine/RHI/RHITexture.h>
|
||
|
|
|
||
|
|
#include <d3d12.h>
|
||
|
|
#include <windows.h>
|
||
|
|
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
namespace XCEngine::UI::Editor::Host {
|
||
|
|
|
||
|
|
class D3D12ShaderResourceDescriptorAllocator {
|
||
|
|
public:
|
||
|
|
bool Initialize(::XCEngine::RHI::RHIDevice& device, UINT descriptorCount = 64u);
|
||
|
|
void Shutdown();
|
||
|
|
|
||
|
|
bool IsInitialized() const;
|
||
|
|
ID3D12DescriptorHeap* GetDescriptorHeap() const;
|
||
|
|
UINT GetDescriptorSize() const;
|
||
|
|
UINT GetDescriptorCount() const;
|
||
|
|
|
||
|
|
void Allocate(
|
||
|
|
D3D12_CPU_DESCRIPTOR_HANDLE* outCpuHandle,
|
||
|
|
D3D12_GPU_DESCRIPTOR_HANDLE* outGpuHandle);
|
||
|
|
bool CreateTextureDescriptor(
|
||
|
|
::XCEngine::RHI::RHITexture* texture,
|
||
|
|
D3D12_CPU_DESCRIPTOR_HANDLE* outCpuHandle,
|
||
|
|
D3D12_GPU_DESCRIPTOR_HANDLE* outGpuHandle);
|
||
|
|
void Free(
|
||
|
|
D3D12_CPU_DESCRIPTOR_HANDLE cpuHandle,
|
||
|
|
D3D12_GPU_DESCRIPTOR_HANDLE gpuHandle);
|
||
|
|
|
||
|
|
private:
|
||
|
|
void AllocateInternal(
|
||
|
|
D3D12_CPU_DESCRIPTOR_HANDLE* outCpuHandle,
|
||
|
|
D3D12_GPU_DESCRIPTOR_HANDLE* outGpuHandle);
|
||
|
|
|
||
|
|
::XCEngine::RHI::RHIDevice* m_device = nullptr;
|
||
|
|
::XCEngine::RHI::RHIDescriptorPool* m_descriptorPool = nullptr;
|
||
|
|
::XCEngine::RHI::D3D12DescriptorHeap* m_descriptorHeap = nullptr;
|
||
|
|
std::vector<bool> m_descriptorUsage = {};
|
||
|
|
UINT m_descriptorSize = 0u;
|
||
|
|
UINT m_descriptorCount = 0u;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace XCEngine::UI::Editor::Host
|