Vertex picking

[opengl.git] / model.hpp

diff --git a/model.hpp b/model.hpp
index 92aab0dd21cb80cb95791f475819522ee0b58056..ecc8842b7e3807d901d642dc721f572a8cd91b78 100644 (file)
--- a/model.hpp
+++ b/model.hpp
@@ -1,5 +1,9 @@
+#ifndef MODEL_HPP
+#define MODEL_HPP
+
 #include <vector>
 #include <map>
+#include <set>
 #ifdef __APPLE__
 #include <GL/glew.h>
 #else
@@ -11,13 +15,6 @@
 #include "program.hpp"
 #include "skybox.hpp"
 
-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;
-}
 
 class Model {
 
@@ -26,7 +23,7 @@ class Model {
                std::vector<const aiNodeAnim*> nodeAnims;
        };
 
-       typedef std::map<std::string, std::pair<unsigned int, glm::mat4>> BoneMap;
+       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;
 
@@ -40,6 +37,7 @@ class Model {
                GLuint progId, vao, numIndices;
                unsigned int materialIndex;
                BoneMap boneMap;
+               const aiMesh &ai;
        };
        
        public:
@@ -48,11 +46,22 @@ class Model {
 
                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, glm::mat4 parentModel, BoneTransforms boneTransforms) 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);
-                               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;
@@ -60,11 +69,14 @@ class Model {
                                const AnimMap *animMap;
                                std::vector<Node*> children;
                                std::vector<unsigned int> meshIndices;
+                               const bool isBone;
                };
                
-               Node* getRoot() { return root; }
-               Node* find(const aiString name);
-               Node* find(const std::string &name);
+               Node& getRoot() const { return *root; }
+               Node* find(const aiString name) const;
+               Node* find(const std::string &name) const;
+
+               std::pair<glm::vec3, float> closestVertex(Model::Node &node, glm::vec3 a, glm::vec3 b, glm::mat4 parentTrans = glm::mat4(1)) const;
        
        private:
                const Program program;
@@ -74,5 +86,10 @@ class Model {
 
                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);
 };
+
+#endif