#include <vector>
+#include <map>
+#include <set>
#ifdef __APPLE__
#include <GL/glew.h>
#else
#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);
};