X-Git-Url: https://git.lukelau.me/?p=opengl.git;a=blobdiff_plain;f=model.hpp;h=e7ad29cd2c9f97b0cf3b414c9e1c89aaf7d367b6;hp=358b92d07701211a99eff67692e1ec56045f6468;hb=76e0d790f76df0c32b09118870145ec5f9eaf153;hpb=511a2c92fcb9dda82dd5d38b91ea03790d0cb7b2

diff --git a/model.hpp b/model.hpp
index 358b92d..e7ad29c 100644
--- a/model.hpp
+++ b/model.hpp
@@ -1,4 +1,9 @@
+#ifndef MODEL_HPP
+#define MODEL_HPP
+
 #include <vector>
+#include <map>
+#include <set>
 #ifdef __APPLE__
 #include <GL/glew.h>
 #else
@@ -6,50 +11,93 @@
 #endif
 #include <glm/glm.hpp>
 #include <assimp/scene.h>
-#include <assimp/Importer.hpp>
 #include "material.hpp"
 #include "program.hpp"
 #include "skybox.hpp"
 
+
 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};
+	};
+	
+	public:
+		Model(std::vector<std::string> blendshapes, std::string neutral, Program p);
+		Model(const aiScene *scene, Program p);
+		void draw(Skybox skybox, const float tick) const;
+
 		struct Mesh {
 			Mesh(const aiMesh *aiMesh, GLuint progId);
 			GLuint progId, vao, numIndices;
+			GLuint vbos[6];
 			unsigned int materialIndex;
+			BoneMap boneMap;
+			const aiMesh &ai;
+			void updatePosBuffer() const;
 		};
 
-	public:
-		Model(const std::string &path, Program p);
-		void draw(Skybox skybox) const;
-
 		class Node {
 			public:
-				Node(const aiNode &aiNode, GLuint progId);
-				void draw(const std::vector<Mesh> &meshes, const std::vector<Material> &materials, const Skybox s, glm::mat4 parentModel) const;
+				Node(aiNode &aiNode, GLuint progId, AnimMap *animMap, std::set<std::string> allBones, Node *parent);
+
+				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);
-				glm::mat4 model = glm::mat4(1);
-				const aiNode &ai;
+				aiNode &ai;
+
+				// an extra transform
+				glm::mat4 transform = glm::mat4(1);
+				
+				glm::mat4 totalTrans(const glm::mat4 parentTrans, const float tick) const;
+
+				const Node *parent;
+				const Node &getRoot() const;
+
+				bool operator==(const Node &rhs) const;
+
 			private:
 				const GLuint progId;
 
+				const AnimMap *animMap;
 				std::vector<Node*> children;
 				std::vector<unsigned int> meshIndices;
+				const bool isBone;
 		};
 		
-		Node* getRoot() { return root; }
-		Node* find(const std::string &name);
+		Node& getRoot() const { return *root; }
+		Node* find(const aiString name) const;
+		Node* find(const std::string &name) const;
 
-	private:
-		const Program program;
+		struct VertexLookup {
+			glm::vec3 pos;
+			int meshIdx, vertIdx;
+			float distance;
+		};
+		VertexLookup closestVertex(Model::Node &node, glm::vec3 a, glm::vec3 b, glm::mat4 parentTrans = glm::mat4(1)) const;
 		
 		std::vector<Mesh> meshes;
+	
+	private:
+		const Program program;
 		Node *root;
 
 		std::vector<Material> materials;
 
-		void loadModel(const std::string &path);
+		AnimMap animMap;
 
-		Assimp::Importer importer;
+		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);
 };
+
+#endif