more recatoring, doesnt compile

1 files changed, 145 insertions, 0 deletions

diff --git a/engine/xe_render_system.cpp b/engine/xe_render_system.cpp
new file mode 100644
index 0000000..ec56972
--- /dev/null
+++ b/engine/xe_render_system.cpp
@@ -0,0 +1,145 @@
+#include "xe_render_system.hpp"
+
+#include <vulkan/vulkan.h>
+#include "xe_device.hpp"
+#include "xe_pipeline.hpp"
+#include "xe_frame_info.hpp"
+#include "xe_game_object.hpp"
+#include "xe_swap_chain.hpp"
+#include "xe_renderer.hpp"
+#include "xe_descriptors.hpp"
+
+#include <memory>
+#include <stdexcept>
+
+namespace xe {
+
+template <typename P, typename U>
+XeRenderSystem::XeRenderSystem(
+  XeDevice &device, 
+  XeRenderer &renderer, 
+  std::unique_ptr<XeDescriptorPool> &xeDescriptorPool, 
+  std::unique_ptr<XeDescriptorSetLayout> &xeDescriptorSetLayout, 
+  std::string vert,
+  std::string frag, 
+  uint32_t pushCunstantDataSize, 
+  uint32_t uniformBufferDataSize) 
+  : xeDevice{device} {
+  createUniformBuffers(xeDescriptorPool, xeDescriptorSetLayout, uniformBufferDataSize);
+  createPipelineLayout(xeDescriptorSetLayout, pushCunstantDataSize, uniformBufferDataSize);
+  createPipeline(renderer.getSwapChainRenderPass(), vert, frag);
+}
+
+
+template <typename P, typename U>
+XeRenderSystem::~XeRenderSystem() {};
+
+
+template <typename P, typename U>
+void XeRenderSystem::createUniformBuffers(std::unique_ptr<XeDescriptorPool> &xeDescriptorPool, std::unique_ptr<XeDescriptorSetLayout> &xeDescriptorSetLayout, uint32_t uniformBufferDataSize) {
+  if(uniformBufferDataSize == 0) return;
+
+  uboBuffers = std::vector<std::unique_ptr<XeBuffer>>(XeSwapChain::MAX_FRAMES_IN_FLIGHT);
+  for (int i = 0; i < uboBuffers.size(); i++) {
+    uboBuffers[i] = std::make_unique<XeBuffer>(
+      xeDevice,
+      uniformBufferDataSize,
+      XeSwapChain::MAX_FRAMES_IN_FLIGHT,
+      VK_BUFFER_USAGE_UNIFORM_BUFFER_BIT,
+      VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT);
+    uboBuffers[i]->map();
+  }
+
+  descriptorSets = std::vector<VkDescriptorSet>(XeSwapChain::MAX_FRAMES_IN_FLIGHT);
+  for (int i = 0; i < descriptorSets.size(); i++) {
+    auto bufferInfo = uboBuffers[i]->descriptorInfo();
+    XeDescriptorWriter(*xeDescriptorSetLayout, *xeDescriptorPool)
+      .writeBuffer(0, &bufferInfo)
+      .build(descriptorSets[i]);
+  }
+}
+
+
+template <typename P, typename U>
+void XeRenderSystem::createPipelineLayout(std::unique_ptr<XeDescriptorSetLayout> &xeDescriptorSetLayout, uint32_t pushCunstantDataSize, uint32_t uniformBufferDataSize) {
+
+  VkPipelineLayoutCreateInfo pipelineLayoutInfo{};
+  pipelineLayoutInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO;
+
+  if (pushCunstantDataSize > 0) {
+    VkPushConstantRange pushConstantRange;
+    pushConstantRange.stageFlags = VK_SHADER_STAGE_VERTEX_BIT | VK_SHADER_STAGE_FRAGMENT_BIT;
+    pushConstantRange.offset = 0;
+    pushConstantRange.size = pushCunstantDataSize;
+    pipelineLayoutInfo.pushConstantRangeCount = 1;
+    pipelineLayoutInfo.pPushConstantRanges = &pushConstantRange;
+  } else {
+    pipelineLayoutInfo.pushConstantRangeCount = 0;
+    pipelineLayoutInfo.pPushConstantRanges = nullptr;
+  }
+
+  if (uniformBufferDataSize > 0) {
+    std::vector<VkDescriptorSetLayout> descriptorSetLayouts{xeDescriptorSetLayout->getDescriptorSetLayout()};
+    pipelineLayoutInfo.setLayoutCount = static_cast<uint32_t>(descriptorSetLayouts.size());
+    pipelineLayoutInfo.pSetLayouts = descriptorSetLayouts.data();
+  } else {
+    pipelineLayoutInfo.setLayoutCount = 0;
+    pipelineLayoutInfo.pSetLayouts = nullptr;
+  }
+
+  if(vkCreatePipelineLayout(xeDevice.device(), &pipelineLayoutInfo, nullptr, &pipelineLayout) != VK_SUCCESS) {
+    std::runtime_error("failed to create pipeline layout!");
+  }
+}
+
+
+template <typename P, typename U>
+void XeRenderSystem::createPipeline(VkRenderPass renderPass, std::string vert, std::string frag) {
+  assert(pipelineLayout != nullptr && "Cannot create pipeline before pipeline layout");
+
+  PipelineConfigInfo pipelineConfig{};
+  XePipeline::defaultPipelineConfigInfo(pipelineConfig);
+  pipelineConfig.renderPass = renderPass;
+  pipelineConfig.pipelineLayout = pipelineLayout;
+  xePipeline = std::make_unique<XePipeline>(
+    xeDevice,
+    vert,
+    frag,
+    pipelineConfig
+  );
+}
+
+
+template <typename P, typename U>
+void XeRenderSystem::renderGameObjects(XeFrameInfo &frameInfo, std::vector<XeGameObject> &gameObjects, XeRenderSystem::XeData pushConstantData) {
+  xePipeline->bind(frameInfo.commandBuffer);
+
+  if(uboBuffers.size() > 0) {
+    vkCmdBindDescriptorSets(
+        frameInfo.commandBuffer,
+        VK_PIPELINE_BIND_POINT_GRAPHICS,
+        pipelineLayout,
+        0,
+        1,
+        &descriptorSets[frameInfo.frameIndex],
+        0,
+        nullptr);
+  }
+  
+  for (auto& obj: gameObjects) {
+
+    vkCmdPushConstants(
+      frameInfo.commandBuffer, 
+      pipelineLayout, 
+      VK_SHADER_STAGE_VERTEX_BIT | VK_SHADER_STAGE_FRAGMENT_BIT, 
+      0, 
+      sizeof(pushConstantData),
+      &pushConstantData);
+
+    obj.model->bind(frameInfo.commandBuffer);
+    obj.model->draw(frameInfo.commandBuffer);
+  }
+
+}
+
+}
\ No newline at end of file