SVN maintenance.
[blender-staging.git] / source / blender / collada / MeshImporter.h
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * Contributor(s): Chingiz Dyussenov, Arystanbek Dyussenov, Nathan Letwory.
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 #ifndef __BC__MESHIMPORTER_H__
26 #define __BC__MESHIMPORTER_H__
27
28 #include <map>
29 #include <vector>
30
31 #include "COLLADAFWIndexList.h"
32 #include "COLLADAFWInstanceGeometry.h"
33 #include "COLLADAFWMaterialBinding.h"
34 #include "COLLADAFWMesh.h"
35 #include "COLLADAFWMeshVertexData.h"
36 #include "COLLADAFWNode.h"
37 #include "COLLADAFWTextureCoordinateBinding.h"
38 #include "COLLADAFWTypes.h"
39 #include "COLLADAFWUniqueId.h"
40
41 #include "DNA_material_types.h"
42 #include "DNA_mesh_types.h"
43 #include "DNA_meshdata_types.h"
44 #include "DNA_object_types.h"
45 #include "DNA_scene_types.h"
46 #include "DNA_texture_types.h"
47
48 #include "ArmatureImporter.h"
49 #include "collada_utils.h"
50
51 // only for ArmatureImporter to "see" MeshImporter::get_object_by_geom_uid
52 class MeshImporterBase
53 {
54 public:
55         virtual Object *get_object_by_geom_uid(const COLLADAFW::UniqueId& geom_uid) = 0;
56 };
57
58 class UVDataWrapper
59 {
60         COLLADAFW::MeshVertexData *mVData;
61 public:
62         UVDataWrapper(COLLADAFW::MeshVertexData& vdata);
63
64 #ifdef COLLADA_DEBUG
65         void print();
66 #endif
67
68         void getUV(int uv_index[2], float *uv);
69 };
70
71 class MeshImporter : public MeshImporterBase
72 {
73 private:
74
75         Scene *scene;
76         ArmatureImporter *armature_importer;
77
78         std::map<COLLADAFW::UniqueId, Mesh*> uid_mesh_map; // geometry unique id-to-mesh map
79         std::map<COLLADAFW::UniqueId, Object*> uid_object_map; // geom uid-to-object
80         // this structure is used to assign material indices to faces
81         // it holds a portion of Mesh faces and corresponds to a DAE primitive list (<triangles>, <polylist>, etc.)
82         struct Primitive {
83                 MFace *mface;
84                 unsigned int totface;
85         };
86         typedef std::map<COLLADAFW::MaterialId, std::vector<Primitive> > MaterialIdPrimitiveArrayMap;
87         std::map<COLLADAFW::UniqueId, MaterialIdPrimitiveArrayMap> geom_uid_mat_mapping_map; // crazy name!
88         
89
90         void set_face_indices(MFace *mface, unsigned int *indices, bool quad);
91
92         // not used anymore, test_index_face from blenkernel is better
93 #if 0
94         // change face indices order so that v4 is not 0
95         void rotate_face_indices(MFace *mface);
96 #endif
97         
98         void set_face_uv(MTFace *mtface, UVDataWrapper &uvs,
99                                          COLLADAFW::IndexList& index_list, unsigned int *tris_indices);
100
101         void set_face_uv(MTFace *mtface, UVDataWrapper &uvs,
102                                         COLLADAFW::IndexList& index_list, int index, bool quad);
103
104 #ifdef COLLADA_DEBUG
105         void print_index_list(COLLADAFW::IndexList& index_list);
106 #endif
107
108         bool is_nice_mesh(COLLADAFW::Mesh *mesh);
109
110         void read_vertices(COLLADAFW::Mesh *mesh, Mesh *me);
111         
112         int triangulate_poly(unsigned int *indices, int totvert, MVert *verts, std::vector<unsigned int>& tri);
113         
114         int count_new_tris(COLLADAFW::Mesh *mesh, Mesh *me);
115         
116         // TODO: import uv set names
117         void read_faces(COLLADAFW::Mesh *mesh, Mesh *me, int new_tris);
118
119         void get_vector(float v[3], COLLADAFW::MeshVertexData& arr, int i);
120
121         bool flat_face(unsigned int *nind, COLLADAFW::MeshVertexData& nor, int count);
122
123 public:
124
125         MeshImporter(ArmatureImporter *arm, Scene *sce);
126
127         virtual Object *get_object_by_geom_uid(const COLLADAFW::UniqueId& geom_uid);
128         
129         MTex *assign_textures_to_uvlayer(COLLADAFW::TextureCoordinateBinding &ctexture,
130                                                                          Mesh *me, TexIndexTextureArrayMap& texindex_texarray_map,
131                                                                          MTex *color_texture);
132         
133         MTFace *assign_material_to_geom(COLLADAFW::MaterialBinding cmaterial,
134                                                                         std::map<COLLADAFW::UniqueId, Material*>& uid_material_map,
135                                                                         Object *ob, const COLLADAFW::UniqueId *geom_uid, 
136                                                                         MTex **color_texture, char *layername, MTFace *texture_face,
137                                                                         std::map<Material*, TexIndexTextureArrayMap>& material_texture_mapping_map, int mat_index);
138         
139         
140         Object *create_mesh_object(COLLADAFW::Node *node, COLLADAFW::InstanceGeometry *geom,
141                                                            bool isController,
142                                                            std::map<COLLADAFW::UniqueId, Material*>& uid_material_map,
143                                                            std::map<Material*, TexIndexTextureArrayMap>& material_texture_mapping_map);
144
145         // create a mesh storing a pointer in a map so it can be retrieved later by geometry UID
146         bool write_geometry(const COLLADAFW::Geometry* geom);
147
148 };
149
150 #endif