Merged everything into one heap for the UI because copy descriptors isn't implemented yet

This commit is contained in:
Mr-Wiseguy 2023-11-02 01:57:50 -04:00
parent 6fdf66b296
commit 458ccd81fc
2 changed files with 12 additions and 21 deletions

View File

@ -1,5 +1,5 @@
SamplerState gSampler : register(s1, space0); SamplerState gSampler : register(s1, space0);
Texture2D<float4> gTexture : register(t0, space1); Texture2D<float4> gTexture : register(t2, space0);
void PSMain( void PSMain(
in float4 iColor : COLOR, in float4 iColor : COLOR,

View File

@ -48,15 +48,17 @@ void CalculateTextureRowWidthPadding(uint32_t rowPitch, uint32_t &rowWidth, uint
rowWidth += rowPadding; rowWidth += rowPadding;
} }
struct RmlRenderInterfaceHeap : public RT64::RenderDescriptorHeapBase { struct RmlRenderInterfaceHeapBase : public RT64::RenderDescriptorHeapBase {
uint32_t gSampler; uint32_t gSampler;
uint32_t gTexture;
RmlRenderInterfaceHeap(const RT64::RenderSampler* linear_sampler) { RmlRenderInterfaceHeapBase(const RT64::RenderSampler* linear_sampler) {
assert(linear_sampler != nullptr); assert(linear_sampler != nullptr);
builder.begin(); builder.begin();
builder.beginSet(); builder.beginSet();
gSampler = builder.addImmutableSampler(1, linear_sampler); gSampler = builder.addImmutableSampler(1, linear_sampler);
gTexture = builder.addTexture(2);
builder.endSet(); builder.endSet();
builder.end(); builder.end();
} }
@ -123,8 +125,7 @@ class RmlRenderInterface_RT64 : public Rml::RenderInterface {
std::unique_ptr<RT64::RenderSampler> linearSampler_{}; std::unique_ptr<RT64::RenderSampler> linearSampler_{};
std::unique_ptr<RT64::RenderShader> vertex_shader_{}; std::unique_ptr<RT64::RenderShader> vertex_shader_{};
std::unique_ptr<RT64::RenderShader> pixel_shader_{}; std::unique_ptr<RT64::RenderShader> pixel_shader_{};
std::unique_ptr<RmlRenderInterfaceHeap> heap_{}; std::unique_ptr<RmlRenderInterfaceHeapBase> heap_base_{};
std::unique_ptr<RT64::RenderDescriptorHeapBuilder> texture_heap_builder_{};
std::unique_ptr<RT64::RenderPipelineLayout> layout_{}; std::unique_ptr<RT64::RenderPipelineLayout> layout_{};
std::unique_ptr<RT64::RenderPipeline> pipeline_{}; std::unique_ptr<RT64::RenderPipeline> pipeline_{};
uint32_t upload_buffer_size_ = 0; uint32_t upload_buffer_size_ = 0;
@ -172,24 +173,14 @@ public:
// Create the descriptor heap // Create the descriptor heap
heap_ = std::make_unique<RmlRenderInterfaceHeap>(linearSampler_.get()); heap_base_ = std::make_unique<RmlRenderInterfaceHeapBase>(linearSampler_.get());
heap_->create(render_context->device);
texture_heap_builder_ = std::make_unique<RT64::RenderDescriptorHeapBuilder>();
texture_heap_builder_->begin();
texture_heap_builder_->beginSet();
texture_heap_builder_->addTexture(0);
texture_heap_builder_->endSet();
texture_heap_builder_->end();
// Create the pipeline layout // Create the pipeline layout
RT64::RenderPipelineLayoutBuilder layout_builder{}; RT64::RenderPipelineLayoutBuilder layout_builder{};
layout_builder.begin(false, true); layout_builder.begin(false, true);
layout_builder.addPushConstant(0, 0, sizeof(RmlPushConstants), RT64::RenderShaderStageFlag::VERTEX); layout_builder.addPushConstant(0, 0, sizeof(RmlPushConstants), RT64::RenderShaderStageFlag::VERTEX);
// Add the descriptor set for descriptors changed once per frame. // Add the descriptor set for descriptors changed once per frame.
layout_builder.addDescriptorSetsFromHeap(heap_->builder); layout_builder.addDescriptorSetsFromHeap(heap_base_->builder);
// Add the descriptor set for descriptors changed once per draw.
layout_builder.addDescriptorSetsFromHeap(*texture_heap_builder_);
layout_builder.end(); layout_builder.end();
layout_ = layout_builder.create(render_context->device); layout_ = layout_builder.create(render_context->device);
@ -341,7 +332,7 @@ public:
list_->setIndexBuffer(&index_view); list_->setIndexBuffer(&index_view);
RT64::RenderVertexBufferView vertex_view{vertex_buffer_->at(0), vert_size_bytes}; RT64::RenderVertexBufferView vertex_view{vertex_buffer_->at(0), vert_size_bytes};
list_->setVertexBuffers(0, &vertex_view, 1, &vertex_slot_); list_->setVertexBuffers(0, &vertex_view, 1, &vertex_slot_);
list_->setGraphicsDescriptorHeap(textures_[texture].heap.get(), 0, per_draw_descriptor_set); list_->setGraphicsDescriptorHeap(textures_[texture].heap.get());
RmlPushConstants constants{ RmlPushConstants constants{
.transform = mvp_, .transform = mvp_,
@ -500,8 +491,9 @@ public:
list_->barriers(RT64::RenderTextureBarrier::Transition(texture.get(), RT64::RenderTextureState::PIXEL_SHADER_ACCESS)); list_->barriers(RT64::RenderTextureBarrier::Transition(texture.get(), RT64::RenderTextureState::PIXEL_SHADER_ACCESS));
// Create a descriptor heap with this texture in it. // Create a descriptor heap with this texture in it.
std::unique_ptr<RT64::RenderDescriptorHeap> heap = texture_heap_builder_->create(render_context_->device); std::unique_ptr<RT64::RenderDescriptorHeap> heap = heap_base_->builder.create(render_context_->device);
heap->setTexture(0, 0, texture.get(), RT64::RenderTextureState::PIXEL_SHADER_ACCESS);
heap->setTexture(heap_base_->gTexture, 0, texture.get(), RT64::RenderTextureState::PIXEL_SHADER_ACCESS);
textures_.emplace(texture_handle, TextureHandle{ std::move(texture), std::move(heap) }); textures_.emplace(texture_handle, TextureHandle{ std::move(texture), std::move(heap) });
@ -528,7 +520,6 @@ public:
list_ = list; list_ = list;
list_->setPipeline(pipeline_.get()); list_->setPipeline(pipeline_.get());
list_->setGraphicsPipelineLayout(layout_.get()); list_->setGraphicsPipelineLayout(layout_.get());
list_->setGraphicsDescriptorHeap(heap_->get(), 0, per_frame_descriptor_set);
projection_mtx_ = Rml::Matrix4f::ProjectOrtho(0.0f, static_cast<float>(image_width), static_cast<float>(image_height), 0.0f, -10000, 10000); projection_mtx_ = Rml::Matrix4f::ProjectOrtho(0.0f, static_cast<float>(image_width), static_cast<float>(image_height), 0.0f, -10000, 10000);
recalculate_mvp(); recalculate_mvp();