Mesh.h

#pragma once
 
#include "krpch.h"
 
#include "Buffer.h"
#include <assimp/Importer.hpp>
#include <assimp/scene.h>
#include <assimp/postprocess.h>
 
namespace Karma
{   
    enum class MeshType
    {
        Mesh = 0,
        SkeletalMesh
    };
 
    enum class MeshAttribute
    {
        Vertices = 0,
        Normals,
        Tangents,
        Bitangents,
        Colors,
        TextureCoords,
        AnimMeshes
    };
    
    class KARMA_API Mesh
    {
    public:
        Mesh(std::shared_ptr<VertexBuffer> vertexBuffer, std::shared_ptr<IndexBuffer> indexBuffer, const std::string& meshName = "NoName",
            MeshType mType = MeshType::Mesh);
        Mesh(const std::string& filePath);
 
        virtual void ProcessMesh(aiMesh* meshToProcess);
        static std::shared_ptr<Mesh> ProcessTheRawMesh(aiMesh* meshToProcess, const std::string& mName = "NoName");
        static void DealVertexIndexBufferData(float*& vertexData, uint32_t& vertexDataSize, uint32_t*& indexData, uint32_t& indexDataLength,
            aiMesh* meshToProcess, BufferLayout& buffLayout);
 
        void ProcessNode(aiNode* nodeToProcess, const aiScene* theScene);
 
        static void GaugeVertexDataLayout(aiMesh* meshToProcess, BufferLayout& buffLayout);
 
        std::shared_ptr<VertexBuffer> GetVertexBuffer() const { return m_VertexBuffer; }
        std::shared_ptr<IndexBuffer> GetIndexBuffer() const { return m_IndexBuffer; }
 
        void SetVertexBuffer(std::shared_ptr<VertexBuffer> vBuffer) { m_VertexBuffer = vBuffer; }
        void SetIndexBuffer(std::shared_ptr<IndexBuffer> iBuffer) { m_IndexBuffer = iBuffer; }
 
        // Useful dictionary related functions
        static float LayoutElementToAttributeValue(unsigned int vertexNumber, uint32_t counter, aiMesh* meshToProcess, const BufferElement& layoutElem);
        static void InitializeAttributeDictionary();
 
    protected:
        std::shared_ptr<VertexBuffer> m_VertexBuffer;
        std::shared_ptr<IndexBuffer> m_IndexBuffer;
 
        std::string m_MeshName;
        MeshType m_MeshType;
 
        static std::shared_ptr<std::unordered_map<std::string, MeshAttribute>> m_NameToAttributeDictionary;
    };
}