self define descriptors

engine/xe_input.hpp

@@ -0,0 +1,15 @@
+#pragma once
+
+#include "xe_window.hpp"
+
+namespace xe {
+
+class XeInput {
+
+  private:
+
+    XeWindow &xeWindow;
+
+};
+  
+}

engine/xe_render_system.cpp

@@ -20,18 +20,19 @@ XeRenderSystem::XeRenderSystem(
   XeEngine &xeEngine,
   std::string vert,
   std::string frag,
-  uint32_t pushCunstantDataSize, 
+  std::map<uint32_t, uint32_t> uniformBindings,
-  uint32_t uniformBufferDataSize,
+  std::map<uint32_t, XeImage*> imageBindings,
-  XeImage *image
+  uint32_t pushCunstantDataSize
 ) : xeDevice{xeEngine.xeDevice}, 
     xeRenderer{xeEngine.xeRenderer},
     xeDescriptorPool{xeEngine.xeDescriptorPool},
     pushCunstantDataSize{pushCunstantDataSize},
-    uniformBufferDataSize{uniformBufferDataSize},
+    uniformBindings{uniformBindings},
-    textureSamplerBinding{image != nullptr} {
+    imageBindings{imageBindings} {
+  createTextureSampler();
   createDescriptorSetLayout();
   createUniformBuffers();
-  createDescriptorSets(image);
+  createDescriptorSets();
   createPipelineLayout();
   createPipeline(xeRenderer.getSwapChainRenderPass(), vert, frag);
 }
@@ -39,21 +40,10 @@ XeRenderSystem::XeRenderSystem(
 
 XeRenderSystem::~XeRenderSystem() {
   vkDestroyPipelineLayout(xeDevice.device(), pipelineLayout, nullptr);
-  if ( textureSamplerBinding ) {
   vkDestroySampler(xeDevice.device(), textureSampler, nullptr);
-  }
 };
 
-void XeRenderSystem::createDescriptorSetLayout() {
+void XeRenderSystem::createTextureSampler() {
-  XeDescriptorSetLayout::Builder builder{xeDevice};
-  int binding = 0;
-
-  if (uniformBufferDataSize > 0) {
-    builder.addBinding(binding, VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, VK_SHADER_STAGE_VERTEX_BIT, nullptr);
-    binding += 1;
-  }
-  if (textureSamplerBinding) {
-
   VkSamplerCreateInfo samplerInfo{};
   samplerInfo.sType = VK_STRUCTURE_TYPE_SAMPLER_CREATE_INFO;
   samplerInfo.magFilter = VK_FILTER_LINEAR;
@@ -74,59 +64,67 @@ void XeRenderSystem::createDescriptorSetLayout() {
   if (vkCreateSampler(xeDevice.device(), &samplerInfo, nullptr, &textureSampler) != VK_SUCCESS) {
     throw std::runtime_error("failed to create texture sampler!");
   }
+}
 
+void XeRenderSystem::createDescriptorSetLayout() {
+  XeDescriptorSetLayout::Builder builder{xeDevice};
+  
+  for ( const auto &[binding, size]: uniformBindings) {
+    builder.addBinding(binding, VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER, VK_SHADER_STAGE_VERTEX_BIT, nullptr);
+  }
+
+  for ( const auto &[binding, image]: imageBindings) {
     builder.addBinding(binding, VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER, VK_SHADER_STAGE_FRAGMENT_BIT, &textureSampler);
-    binding += 1;
   }
 
   xeDescriptorSetLayout = builder.build();
 }
 
 void XeRenderSystem::createUniformBuffers() {
-  if(uniformBufferDataSize == 0) return;
+  for ( const auto &[binding, bufferSize]: uniformBindings) {
-
+    uboBuffers[binding] = std::vector<std::unique_ptr<XeBuffer>>(XeSwapChain::MAX_FRAMES_IN_FLIGHT);
-  uboBuffers = std::vector<std::unique_ptr<XeBuffer>>(XeSwapChain::MAX_FRAMES_IN_FLIGHT);
+    for (int i = 0; i < uboBuffers[binding].size(); i++) {
-  for (int i = 0; i < uboBuffers.size(); i++) {
+      uboBuffers[binding][i] = std::make_unique<XeBuffer>(
-    uboBuffers[i] = std::make_unique<XeBuffer>(
         xeDevice,
-      uniformBufferDataSize,
+        bufferSize,
         XeSwapChain::MAX_FRAMES_IN_FLIGHT,
-
         VK_BUFFER_USAGE_UNIFORM_BUFFER_BIT,
         VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT | VK_MEMORY_PROPERTY_HOST_COHERENT_BIT);
-    uboBuffers[i]->map();
+      uboBuffers[binding][i]->map();
+    }
   }
-
 }
 
-void XeRenderSystem::createDescriptorSets(XeImage *image) {
+void XeRenderSystem::createDescriptorSets() {
 
   descriptorSets = std::vector<VkDescriptorSet>(XeSwapChain::MAX_FRAMES_IN_FLIGHT);
   for (int i = 0; i < descriptorSets.size(); i++) {
-    updateDescriptorSet(image, i, true);
+    updateDescriptorSet(i, true);
   }
 
 }
 
-void XeRenderSystem::updateDescriptorSet(XeImage *image, int frameIndex, bool allocate) {
+void XeRenderSystem::updateDescriptorSet(int frameIndex, bool allocate) {
-  auto bufferInfo = uboBuffers[frameIndex]->descriptorInfo();
+
   XeDescriptorWriter writer{*xeDescriptorSetLayout, *xeDescriptorPool};
 
-    int binding = 0;
+  std::vector<VkDescriptorBufferInfo> bufferInfos{};
 
-    if (uniformBufferDataSize > 0) {
+  int i = 0;
-      writer.writeBuffer(binding, &bufferInfo);
+  for ( const auto &[binding, size]: uniformBindings) {
-      binding += 1;
+    bufferInfos.push_back(uboBuffers[binding][frameIndex]->descriptorInfo());
+    writer.writeBuffer(binding, &bufferInfos[i]);
+    i++;
   }
 
-    if (textureSamplerBinding) {
+  for ( const auto &[binding, image]: imageBindings) {
     VkDescriptorImageInfo imageInfo{};
     imageInfo.imageLayout = VK_IMAGE_LAYOUT_SHADER_READ_ONLY_OPTIMAL;
     imageInfo.imageView = image->textureImageView;
     imageInfo.sampler = textureSampler;
     writer.writeImage(binding, &imageInfo);
-      binding += 1;
   }
+
   if (allocate) {
     writer.build(descriptorSets[frameIndex]);
   } else {
@@ -143,8 +141,12 @@ void XeRenderSystem::createPipelineLayout() {
   pushConstantRange.offset = 0;
   pushConstantRange.size = pushCunstantDataSize;
 
+  std::vector<VkDescriptorSetLayout> descriptorSetLayouts{xeDescriptorSetLayout->getDescriptorSetLayout()};
+
   VkPipelineLayoutCreateInfo pipelineLayoutInfo{};
   pipelineLayoutInfo.sType = VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO;
+  pipelineLayoutInfo.setLayoutCount = static_cast<uint32_t>(descriptorSetLayouts.size());
+  pipelineLayoutInfo.pSetLayouts = descriptorSetLayouts.data();
 
   if (pushCunstantDataSize > 0) {
     pipelineLayoutInfo.pushConstantRangeCount = 1;
@@ -154,17 +156,6 @@ void XeRenderSystem::createPipelineLayout() {
     pipelineLayoutInfo.pPushConstantRanges = nullptr;
   }
 
-  std::vector<VkDescriptorSetLayout> descriptorSetLayouts{xeDescriptorSetLayout->getDescriptorSetLayout()};
-
-  if (uniformBufferDataSize > 0) {
-    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!");
   }
@@ -215,12 +206,13 @@ void XeRenderSystem::loadPushConstant(void *pushConstantData) {
         pushConstantData);
 }
 
-void XeRenderSystem::loadUniformObject(void *uniformBufferData) {
+void XeRenderSystem::loadUniformObject(uint32_t binding, void *uniformBufferData) {
-  uboBuffers[xeRenderer.getFrameIndex()]->writeToBuffer(uniformBufferData);
+  uboBuffers[binding][xeRenderer.getFrameIndex()]->writeToBuffer(uniformBufferData);
 }
 
-void XeRenderSystem::loadTexture(XeImage *image) {
+void XeRenderSystem::loadTexture(uint32_t binding, XeImage *image) {
-  updateDescriptorSet(image, xeRenderer.getFrameIndex(), false);
+  imageBindings[binding] = image;
+  updateDescriptorSet(xeRenderer.getFrameIndex(), false);
 }
 
 void XeRenderSystem::render(XeGameObject &gameObject) {

engine/xe_render_system.hpp

@@ -10,19 +10,55 @@
 #include "xe_image.hpp"
 
 #include <memory>
+#include <map>
 
 namespace xe {
 
 class XeRenderSystem {
   public:
   
+    class Builder {
+      public:
+        Builder(XeEngine &xeEngine, std::string vert, std::string frag) : xeEngine{xeEngine}, vert{vert}, frag{frag} {}
+
+        Builder& addPushConstant(uint32_t size) {
+          pushCunstantDataSize = size;
+          return *this;
+        }
+
+        Builder& addUniformBinding(uint32_t binding, uint32_t size) {
+          uniformBindings[binding] = size;
+          return *this;
+        }
+
+        Builder& addTextureBinding(uint32_t binding, XeImage* image) {
+          imageBindings[binding] = image;
+          return *this;
+        }
+
+        std::unique_ptr<XeRenderSystem> build() {
+          return std::make_unique<XeRenderSystem>(xeEngine, std::move(vert), std::move(frag), std::move(uniformBindings), std::move(imageBindings), std::move(pushCunstantDataSize));
+        }
+
+      private:
+
+        std::map<uint32_t, uint32_t> uniformBindings{};
+        std::map<uint32_t, XeImage*> imageBindings{};
+        uint32_t pushCunstantDataSize{0};
+
+        std::string vert;
+        std::string frag;
+
+        XeEngine &xeEngine;
+    };
+
     XeRenderSystem(
       XeEngine &xeEngine,
       std::string vert,
       std::string frag,
-      uint32_t pushCunstantDataSize,
+      std::map<uint32_t, uint32_t> uniformBindings,
-      uint32_t uniformBufferDataSize,
+      std::map<uint32_t, XeImage*> imageBindings,
-      XeImage *image
+      uint32_t pushCunstantDataSize
     );
 
     ~XeRenderSystem();
@@ -32,37 +68,39 @@ class XeRenderSystem {
 
     void start();
     void loadPushConstant(void *pushConstantData);
-    void loadUniformObject(void *uniformBufferData);
+    void loadUniformObject(uint32_t binding, void *uniformBufferData);
-    void loadTexture(XeImage *image);
+    void loadTexture(uint32_t binding, XeImage *image);
     void render(XeGameObject &gameObject);
     void stop();
 
   private:
   
+    void createTextureSampler();
     void createDescriptorSetLayout();
     void createUniformBuffers();
-    void createDescriptorSets(XeImage *image);
+    void createDescriptorSets();
-    void updateDescriptorSet(XeImage *image, int frameIndex, bool allocate);
+    void updateDescriptorSet(int frameIndex, bool allocate);
     void createPipelineLayout();
     void createPipeline(VkRenderPass renderPass, std::string vert, std::string frag);
 
     bool boundPipeline{false};
     bool boundDescriptor{false};
 
-    uint32_t uniformBufferDataSize;
-    uint32_t pushCunstantDataSize;
-    bool textureSamplerBinding;
   
     XeDevice& xeDevice;
     XeRenderer& xeRenderer;
 
-    std::unique_ptr<XePipeline> xePipeline;
+    std::map<uint32_t, std::vector<std::unique_ptr<XeBuffer>>> uboBuffers{};
-    std::vector<std::unique_ptr<XeBuffer>> uboBuffers;
+    std::map<uint32_t, uint32_t> uniformBindings;
+    std::map<uint32_t, XeImage*> imageBindings;
     std::vector<VkDescriptorSet> descriptorSets;
 
+    uint32_t pushCunstantDataSize;
+
     VkSampler textureSampler;
     
     VkPipelineLayout pipelineLayout;
+    std::unique_ptr<XePipeline> xePipeline;
     std::unique_ptr<XeDescriptorPool> &xeDescriptorPool;
     std::unique_ptr<XeDescriptorSetLayout> xeDescriptorSetLayout;
 

src/simple_renderer.cpp

@@ -1,29 +1,37 @@
 #include "simple_renderer.hpp"
+#include "xe_render_system.hpp"
+
+#include <iostream>
 
 namespace app {
 
-SimpleRenderer::SimpleRenderer(xe::XeEngine &xeEngine, xe::XeImage *xeImage) 
+SimpleRenderer::SimpleRenderer(xe::XeEngine &xeEngine, xe::XeImage *xeImage) {
-  : xeRenderSystem{xeEngine, "res/shaders/simple_shader.vert.spv", "res/shaders/simple_shader.frag.spv", sizeof(PushConstant), sizeof(UniformBuffer), xeImage} {};
+  xeRenderSystem = xe::XeRenderSystem::Builder(xeEngine, "res/shaders/simple_shader.vert.spv", "res/shaders/simple_shader.frag.spv")
+    .addPushConstant(sizeof(PushConstant))
+    .addUniformBinding(0, sizeof(UniformBuffer))
+    .addTextureBinding(1, xeImage)
+    .build();
+}
 
 void SimpleRenderer::render(std::vector<xe::XeGameObject> &gameObjects, xe::XeCamera &xeCamera, xe::XeImage *xeImage) {
 
-  xeRenderSystem.loadTexture(xeImage);
+  xeRenderSystem->loadTexture(1, xeImage);
 
-  xeRenderSystem.start();
+  xeRenderSystem->start();
 
   UniformBuffer ubo{};
   ubo.projectionView = xeCamera.getProjection() * xeCamera.getView();
-  xeRenderSystem.loadUniformObject(&ubo);
+  xeRenderSystem->loadUniformObject(0, &ubo);
 
   for(auto &obj : gameObjects) {
     PushConstant pc{};
     pc.modelMatrix = obj.transform.mat4();
     pc.normalMatrix = obj.transform.normalMatrix();
-    xeRenderSystem.loadPushConstant(&pc);
+    xeRenderSystem->loadPushConstant(&pc);
-    xeRenderSystem.render(obj);
+    xeRenderSystem->render(obj);
   }
 
-  xeRenderSystem.stop();
+  xeRenderSystem->stop();
 
 }
 

src/simple_renderer.hpp

@@ -29,7 +29,7 @@ class SimpleRenderer {
     void render(std::vector<xe::XeGameObject> &gameObjects, xe::XeCamera &xeCamera, xe::XeImage *xeImage);
 
   private:
-    xe::XeRenderSystem xeRenderSystem;
+    std::unique_ptr<xe::XeRenderSystem> xeRenderSystem;
 
 };