descriptor set stuff from old engine

2026-08-23 21:19:01 -07:00
parent 518c664741
commit ce075403d5
6 changed files with 341 additions and 0 deletions
@@ -0,0 +1,52 @@
+#pragma once
+
+#include <memory>
+#include <vector>
+
+#include "device.hpp"
+
+namespace sophia {
+
+class DescriptorPool {
+ public:
+  DescriptorPool(const DescriptorPool&) = delete;
+  DescriptorPool& operator=(const DescriptorPool&) = delete;
+
+  class Builder {
+   public:
+    Builder(Device& device) : device{device} {}
+
+    Builder& addPoolSize(vk::DescriptorType descriptorType, u32 count);
+    Builder& setPoolFlags(vk::DescriptorPoolCreateFlags flags);
+    Builder& setMaxSets(u32 count);
+    std::unique_ptr<DescriptorPool> build() const;
+
+   private:
+    Device& device;
+    std::vector<vk::DescriptorPoolSize> poolSizes{};
+    u32 maxSets = 1000;
+    vk::DescriptorPoolCreateFlags poolFlags{};
+  };
+
+  DescriptorPool(Device& device, u32 maxSets,
+                 vk::DescriptorPoolCreateFlags poolFlags,
+                 const std::vector<vk::DescriptorPoolSize>& poolSizes);
+  ~DescriptorPool();
+
+  bool allocateDescriptorSet(const vk::DescriptorSetLayout descriptorSetLayout,
+                             vk::DescriptorSet& descriptor) const;
+
+  void freeDescriptors(std::vector<vk::DescriptorSet>& descriptors) const;
+
+  void resetPool();
+
+  const vk::DescriptorPool& get() { return this->descriptorPool; }
+
+ private:
+  Device& device;
+  vk::DescriptorPool descriptorPool;
+
+  friend class DescriptorWriter;
+};
+
+}  // namespace sophia