Depsgraph: remove EvaluationContext, pass Depsgraph instead.
[blender.git] / source / blender / collada / MeshImporter.h
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * Contributor(s): Chingiz Dyussenov, Arystanbek Dyussenov, Nathan Letwory.
19  *
20  * ***** END GPL LICENSE BLOCK *****
21  */
22
23 /** \file MeshImporter.h
24  *  \ingroup collada
25  */
26
27 #ifndef __MESHIMPORTER_H__
28 #define __MESHIMPORTER_H__
29
30 #include <map>
31 #include <vector>
32
33 #include "COLLADAFWIndexList.h"
34 #include "COLLADAFWPolygons.h"
35 #include "COLLADAFWInstanceGeometry.h"
36 #include "COLLADAFWMaterialBinding.h"
37 #include "COLLADAFWMesh.h"
38 #include "COLLADAFWMeshVertexData.h"
39 #include "COLLADAFWNode.h"
40 #include "COLLADAFWTextureCoordinateBinding.h"
41 #include "COLLADAFWTypes.h"
42 #include "COLLADAFWUniqueId.h"
43
44 #include "ArmatureImporter.h"
45 #include "collada_utils.h"
46
47 extern "C" {
48 #include "BLI_edgehash.h"
49 #include "DNA_material_types.h"
50 #include "DNA_mesh_types.h"
51 #include "DNA_meshdata_types.h"
52 #include "DNA_object_types.h"
53 #include "DNA_scene_types.h"
54 #include "DNA_texture_types.h"
55
56 }
57
58 // only for ArmatureImporter to "see" MeshImporter::get_object_by_geom_uid
59 class MeshImporterBase
60 {
61 public:
62         virtual Object *get_object_by_geom_uid(const COLLADAFW::UniqueId& geom_uid) = 0;
63         virtual Mesh *get_mesh_by_geom_uid(const COLLADAFW::UniqueId& mesh_uid) = 0;
64         virtual std::string *get_geometry_name(const std::string &mesh_name) = 0;
65 };
66
67 class UVDataWrapper
68 {
69         COLLADAFW::MeshVertexData *mVData;
70 public:
71         UVDataWrapper(COLLADAFW::MeshVertexData& vdata);
72
73 #ifdef COLLADA_DEBUG
74         void print();
75 #endif
76
77         void getUV(int uv_index, float *uv);
78 };
79
80 class VCOLDataWrapper
81 {
82         COLLADAFW::MeshVertexData *mVData;
83 public:
84         VCOLDataWrapper(COLLADAFW::MeshVertexData& vdata);
85         void get_vcol(int v_index, MLoopCol *mloopcol);
86 };
87
88 class MeshImporter : public MeshImporterBase
89 {
90 private:
91
92         UnitConverter *unitconverter;
93
94         Scene *scene;
95         ViewLayer *view_layer;
96
97         ArmatureImporter *armature_importer;
98
99         std::map<std::string, std::string> mesh_geom_map; // needed for correct shape key naming
100         std::map<COLLADAFW::UniqueId, Mesh*> uid_mesh_map; // geometry unique id-to-mesh map
101         std::map<COLLADAFW::UniqueId, Object*> uid_object_map; // geom uid-to-object
102         std::vector<Object*> imported_objects; // list of imported objects
103
104         // this structure is used to assign material indices to polygons
105         // it holds a portion of Mesh faces and corresponds to a DAE primitive list (<triangles>, <polylist>, etc.)
106         struct Primitive {
107                 MPoly *mpoly;
108                 unsigned int totpoly;
109         };
110         typedef std::map<COLLADAFW::MaterialId, std::vector<Primitive> > MaterialIdPrimitiveArrayMap;
111         std::map<COLLADAFW::UniqueId, MaterialIdPrimitiveArrayMap> geom_uid_mat_mapping_map; // crazy name!
112         std::multimap<COLLADAFW::UniqueId, COLLADAFW::UniqueId> materials_mapped_to_geom; //< materials that have already been mapped to a geometry. A pair of geom uid and mat uid, one geometry can have several materials
113         
114         bool set_poly_indices(MPoly *mpoly,
115                                                   MLoop *mloop,
116                                                   int loop_index,
117                                                   unsigned int *indices,
118                                                   int loop_count);
119
120         void set_face_uv(MLoopUV *mloopuv,
121                                          UVDataWrapper &uvs,
122                                          int loop_index,
123                                          COLLADAFW::IndexList& index_list,
124                                          int count);
125
126         void set_vcol(MLoopCol *mloopcol,
127                           VCOLDataWrapper &vob,
128                           int loop_index,
129                           COLLADAFW::IndexList& index_list,
130                           int count);
131
132 #ifdef COLLADA_DEBUG
133         void print_index_list(COLLADAFW::IndexList& index_list);
134 #endif
135
136         bool is_nice_mesh(COLLADAFW::Mesh *mesh);
137
138         void read_vertices(COLLADAFW::Mesh *mesh, Mesh *me);
139                         
140         bool primitive_has_useable_normals(COLLADAFW::MeshPrimitive *mp);
141         bool primitive_has_faces(COLLADAFW::MeshPrimitive *mp);
142
143         static void mesh_add_edges(Mesh *mesh, int len);
144
145         unsigned int get_loose_edge_count(COLLADAFW::Mesh *mesh);
146
147         CustomData create_edge_custom_data(EdgeHash *eh);
148
149         void allocate_poly_data(COLLADAFW::Mesh *collada_mesh, Mesh *me);
150
151         // TODO: import uv set names
152         void read_polys(COLLADAFW::Mesh *mesh, Mesh *me);
153         void read_lines(COLLADAFW::Mesh *mesh, Mesh *me);
154         unsigned int get_vertex_count(COLLADAFW::Polygons *mp, int index);
155
156         void get_vector(float v[3], COLLADAFW::MeshVertexData& arr, int i, int stride);
157
158         bool is_flat_face(unsigned int *nind, COLLADAFW::MeshVertexData& nor, int count);
159
160         std::vector<Object *> get_all_users_of(Mesh *reference_mesh);
161
162 public:
163
164         MeshImporter(UnitConverter *unitconv, ArmatureImporter *arm, Scene *sce, ViewLayer *view_layer);
165
166         virtual Object *get_object_by_geom_uid(const COLLADAFW::UniqueId& geom_uid);
167
168         virtual Mesh *get_mesh_by_geom_uid(const COLLADAFW::UniqueId& geom_uid);
169         
170         MTex *assign_textures_to_uvlayer(COLLADAFW::TextureCoordinateBinding &ctexture,
171                                          Mesh *me, TexIndexTextureArrayMap& texindex_texarray_map,
172                                          MTex *color_texture);
173
174         void optimize_material_assignements();
175
176         void assign_material_to_geom(
177                 COLLADAFW::MaterialBinding cmaterial,
178                 std::map<COLLADAFW::UniqueId, Material*>& uid_material_map,
179                 Object *ob, const COLLADAFW::UniqueId *geom_uid,
180                 std::map<Material*, TexIndexTextureArrayMap>& material_texture_mapping_map, short mat_index);
181         
182         
183         Object *create_mesh_object(COLLADAFW::Node *node, COLLADAFW::InstanceGeometry *geom,
184                                    bool isController,
185                                    std::map<COLLADAFW::UniqueId, Material*>& uid_material_map,
186                                    std::map<Material*, TexIndexTextureArrayMap>& material_texture_mapping_map);
187
188         // create a mesh storing a pointer in a map so it can be retrieved later by geometry UID
189         bool write_geometry(const COLLADAFW::Geometry* geom);
190         std::string *get_geometry_name(const std::string &mesh_name);
191 };
192
193 #endif