58 lines
1.7 KiB
C++
58 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#ifndef NOMINMAX
|
|
#define NOMINMAX
|
|
#endif
|
|
|
|
#include "Host/ShaderResourceDescriptorAllocator.h"
|
|
|
|
#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 ShaderResourceDescriptorAllocator {
|
|
public:
|
|
bool Initialize(::XCEngine::RHI::RHIDevice& device, UINT descriptorCount = 64u) override;
|
|
void Shutdown() override;
|
|
|
|
bool IsInitialized() const override;
|
|
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) override;
|
|
void Free(
|
|
D3D12_CPU_DESCRIPTOR_HANDLE cpuHandle,
|
|
D3D12_GPU_DESCRIPTOR_HANDLE gpuHandle) override;
|
|
|
|
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
|