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