Reworking VertexData

src/Engine/Graphics/VertexData.h

@@ -1,13 +1,80 @@
 #pragma once
 #include "GraphicsResources.h"
+#include "Material/MaterialInstance.h"
+#include "Component/Mesh.h"
+#include "Component/Transform.h"
 
 namespace Seele
 {
+struct MeshId
+{
+    uint64 id;
+};
 class VertexData
 {
 public:
-private:
-    Gfx::PShaderBuffer positions;
+    struct InstanceData
+    {
+        Matrix4 transformMatrix;
+    };
+    struct MeshInstanceData
+    {
+        MeshId id;
+        InstanceData instance;
+    };
+    struct MaterialInstanceData
+    {
+        PMaterialInstance materialInstance;
+        Gfx::PDescriptorSet descriptorSet;
+        uint32_t numMeshes; // not necessarily equal to meshes.size() if a MeshId has multiple meshes
+        Array<MeshInstanceData> meshes;
+    };
+    struct MaterialData
+    {
+        PMaterial material;
+        Map<uint64, MaterialInstanceData> instances;
+    };
+    struct MeshData
+    {
+        uint32 numMeshlets;
+        uint32 meshletOffset;
+    };
+    void resetMeshData();
+    void updateMesh(const Component::Transform& transform, const Component::Mesh& mesh);
+    void createDescriptors();
+    virtual void bindBuffers(Gfx::PRenderCommand command) = 0;
+    virtual Gfx::PDescriptorLayout getVertexDataLayout() = 0;
+    virtual Gfx::PDescriptorSet getVertexDataSet() = 0;
+    virtual Gfx::PDescriptorLayout getInstanceDataLayout() = 0;
+    virtual std::string getTypeName() const = 0;
+    const Map<std::string, MaterialData>& getMaterialData() const { return materialData; }
+    const Array<MeshData>& getMeshData(MeshId id) { return meshData[id]; }
+    static List<VertexData*> getList();
+protected:
+    struct MeshletAABB
+    {
+        Vector min;
+        float pad0;
+        Vector max;
+        float pad1;
+    };
+    struct MeshletData
+    {
+        MeshletAABB boundingBox;
+        uint32_t vertexCount;
+        uint32_t primiticeCount;
+        uint32_t vertexOffset;
+        uint32_t primitiveOffset;
+    };
+    Map<std::string, MaterialData> materialData;
+    Map<MeshId, Array<MeshData>> meshData;
+    Array<MeshletData> meshlets;
+    Gfx::PDescriptorLayout instanceDataLayout;
+    Gfx::PGraphics graphics;
+    // for mesh shading
+    Gfx::PShaderBuffer meshletBuffer;
+    // for legacy pipeline
+    Gfx::PIndexBuffer indexBuffer;
+    VertexData(Gfx::PGraphics graphics);
 };
-DEFINE_REF(VertexData)
 }