Auto stash before merge of "master" and "Gitea/master"

2024-09-25 17:27:21 +02:00
parent 6417ab940d
commit 81f29d1b3f
16 changed files with 518 additions and 147 deletions
@@ -0,0 +1,134 @@
+#include "TerrainRenderer.h"
+#include "Asset/AssetRegistry.h"
+#include "Component/TerrainTile.h"
+#include "Graphics/Graphics.h"
+#include "Graphics/Shader.h"
+
+using namespace Seele;
+
+TerrainRenderer::TerrainRenderer(Gfx::PGraphics graphics, PScene scene, Gfx::PDescriptorLayout viewParamsLayout)
+    : graphics(graphics), scene(scene) {
+    tilesBuffer = graphics->createShaderBuffer(ShaderBufferCreateInfo{
+        .dynamic = true,
+        .name = "TilesBuffer",
+    });
+    float displacementData = 0;
+    displacementMap = graphics->createTexture2D(TextureCreateInfo{
+        .sourceData =
+            {
+                .size = sizeof(float),
+                .data = (uint8*)&displacementData,
+            },
+        .format = Gfx::SE_FORMAT_R32_SFLOAT,
+        .width = 1,
+        .height = 1,
+        .depth = 1,
+        .name = "TerrainDisplacement",
+    });
+    sampler = graphics->createSampler(SamplerCreateInfo{});
+    layout = graphics->createDescriptorLayout("pTerrainData");
+    layout->addDescriptorBinding(Gfx::DescriptorBinding{
+        .binding = 0,
+        .descriptorType = Gfx::SE_DESCRIPTOR_TYPE_STORAGE_BUFFER,
+    });
+    layout->addDescriptorBinding(Gfx::DescriptorBinding{
+        .binding = 1,
+        .descriptorType = Gfx::SE_DESCRIPTOR_TYPE_SAMPLED_IMAGE,
+    });
+    layout->addDescriptorBinding(Gfx::DescriptorBinding{
+        .binding = 2,
+        .descriptorType = Gfx::SE_DESCRIPTOR_TYPE_SAMPLER,
+    });
+    layout->create();
+    pipelineLayout = graphics->createPipelineLayout("TerrainLayout");
+    pipelineLayout->addDescriptorLayout(viewParamsLayout);
+    pipelineLayout->addDescriptorLayout(layout);
+    ShaderCompilationInfo s{
+        .name = "TerrainShaders",
+        .modules = {"TerrainPass"},
+        .entryPoints =
+            {
+                {"taskMain", "TerrainPass"},
+                {"meshMain", "TerrainPass"},
+                {"fragmentMain", "TerrainPass"},
+            },
+        .rootSignature = pipelineLayout,
+        .dumpIntermediate = false,
+    };
+    graphics->beginShaderCompilation(s);
+    task = graphics->createTaskShader({0});
+    mesh = graphics->createMeshShader({1});
+    frag = graphics->createFragmentShader({2});
+    pipelineLayout->create();
+}
+
+TerrainRenderer::~TerrainRenderer() {}
+
+void TerrainRenderer::beginFrame() {
+    struct TerrainTile {
+        IVector2 offset;
+        float extent;
+        float height;
+    };
+    Array<TerrainTile> payloads;
+    scene->view<Component::TerrainTile>([&](Component::TerrainTile& tile) {
+        payloads.add(TerrainTile{
+            .offset = IVector2(tile.location),
+            .extent = Component::TerrainTile::DIMENSIONS,
+            .height = tile.height,
+        });
+    });
+    if (payloads.size() == 0)
+        return;
+    tilesBuffer->rotateBuffer(sizeof(TerrainTile) * payloads.size());
+    tilesBuffer->updateContents(0, sizeof(TerrainTile) * payloads.size(), payloads.data());
+    tilesBuffer->pipelineBarrier(Gfx::SE_ACCESS_TRANSFER_WRITE_BIT, Gfx::SE_PIPELINE_STAGE_TRANSFER_BIT, Gfx::SE_ACCESS_SHADER_READ_BIT,
+                                 Gfx::SE_PIPELINE_STAGE_TASK_SHADER_BIT_EXT);
+
+    set = layout->allocateDescriptorSet();
+    set->updateBuffer(0, tilesBuffer);
+    set->updateTexture(1, 0, Gfx::PTexture2D(displacementMap));
+    set->updateSampler(2, sampler);
+    set->writeChanges();
+}
+
+Gfx::ORenderCommand TerrainRenderer::render(Gfx::PDescriptorSet viewParamsSet) {
+    Gfx::ORenderCommand command = graphics->createRenderCommand("TerrainRender");
+    command->setViewport(viewport);
+    command->bindPipeline(pipeline);
+    command->bindDescriptor({viewParamsSet, set});
+    command->drawMesh(400, 4, 1);
+    return command;
+}
+
+void TerrainRenderer::setViewport(Gfx::PViewport _viewport, Gfx::PRenderPass renderPass) {
+    viewport = _viewport;
+
+    Gfx::MeshPipelineCreateInfo pipelineInfo = {
+        .taskShader = task,
+        .meshShader = mesh,
+        .fragmentShader = frag,
+        .renderPass = renderPass,
+        .pipelineLayout = pipelineLayout,
+        .multisampleState =
+            {
+                .samples = viewport->getSamples(),
+            },
+        .rasterizationState =
+            {
+                //.polygonMode = Gfx::SE_POLYGON_MODE_LINE,
+                .cullMode = Gfx::SE_CULL_MODE_BACK_BIT,
+            },
+        .colorBlend =
+            {
+                .attachmentCount = 1,
+                .blendAttachments =
+                    {
+                        Gfx::ColorBlendState::BlendAttachment{
+                            .blendEnable = false,
+                        },
+                    },
+            },
+    };
+    pipeline = graphics->createGraphicsPipeline(std::move(pipelineInfo));
+}