X-Git-Url: https://git.lukelau.me/?p=opengl.git;a=blobdiff_plain;f=model.hpp;h=bf6f08f346133920d792870d6a0f42249fe5e0cd;hp=0339538429cc94ae87292ff3f53e56cd460777a6;hb=dbd855720a9af7d6e599ddc50bbbb0dee85458a5;hpb=b64cd5a5ec09e6f051583371045ef7080c69b776

diff --git a/model.hpp b/model.hpp
index 0339538..bf6f08f 100644
--- a/model.hpp
+++ b/model.hpp
@@ -1,4 +1,6 @@
 #include <vector>
+#include <map>
+#include <set>
 #ifdef __APPLE__
 #include <GL/glew.h>
 #else
@@ -6,27 +8,80 @@
 #endif
 #include <glm/glm.hpp>
 #include <assimp/scene.h>
+#include "material.hpp"
+#include "program.hpp"
+#include "skybox.hpp"
 
-class Model {
-	public:
-		Model(const std::string &path, GLuint progId): progId(progId) {
-			loadModel(path);
+inline glm::mat4 aiMatrixToMat4(aiMatrix4x4 from) {
+	glm::mat4 to;
+	for (int i = 0; i < 4; i++)
+		for (int j = 0; j < 4; j++)
+			to[i][j] = from[j][i];
+	return to;
 }
-		void draw();
-	private:
-		const GLuint progId;
+
+class Model {
+
+	struct Animation {
+		double duration;
+		std::vector<const aiNodeAnim*> nodeAnims;
+	};
+
+	typedef std::map<std::string, std::pair<unsigned int, aiBone*>> BoneMap;
+	typedef std::map<std::string, std::vector<const Animation>> AnimMap;
+	typedef std::map<std::string, glm::mat4> BoneTransforms;
+
+	struct VertBones {
+		unsigned int ids[4] = {0, 0, 0 ,0};
+		float weights[4] = {1, 0, 0, 0};
+	};
 
 	struct Mesh {
 		Mesh(const aiMesh *aiMesh, GLuint progId);
+		GLuint progId, vao, numIndices;
+		unsigned int materialIndex;
+		BoneMap boneMap;
+	};
+	
+	public:
+		Model(const aiScene *scene, Program p);
+		void draw(Skybox skybox, const float tick) const;
+
+		class Node {
+			public:
+				Node(const aiNode &aiNode, GLuint progId, AnimMap *animMap);
+				void draw(const std::vector<Mesh> &meshes, const std::vector<Material> &materials, const Skybox s, const float tick, const BoneTransforms &boneTransforms, glm::mat4 parentModel) const;
+				const std::vector<Node*> &getChildren() const { return children; }
+				Node* findNode(const aiNode &aiNode);
+				const aiNode &ai;
+
+				// an extra transform
+				glm::mat4 transform = glm::mat4(1);
 				
-			GLuint vao, vertexVbo, normalVbo, indicesVbo;
+				glm::mat4 totalTrans(const glm::mat4 parentTrans, const float tick) const;
+
+			private:
+				const GLuint progId;
 
-			std::vector<glm::vec3> vertices;
-			std::vector<glm::vec3> normals;
-			std::vector<glm::vec2> texCoords;
-			std::vector<GLuint> indices;
+				const AnimMap *animMap;
+				std::vector<Node*> children;
+				std::vector<unsigned int> meshIndices;
 		};
 		
+		Node* getRoot() { return root; }
+		Node* find(const aiString name) const;
+		Node* find(const std::string &name) const;
+	
+	private:
+		const Program program;
+		
 		std::vector<Mesh> meshes;
+		Node *root;
+
+		std::vector<Material> materials;
+
+		AnimMap animMap;
+
+		BoneTransforms calcBoneTransforms(const Node &n, const float tick, const std::set<std::string> bones, const glm::mat4 parentTrans) const;
 		void loadModel(const std::string &path);
 };