Adding support for Matrix Transformation export
[blender.git] / source / blender / collada / collada_utils.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 collada_utils.h
24  *  \ingroup collada
25  */
26
27 #ifndef __COLLADA_UTILS_H__
28 #define __COLLADA_UTILS_H__
29
30 #include "COLLADAFWMeshPrimitive.h"
31 #include "COLLADAFWGeometry.h"
32 #include "COLLADAFWFloatOrDoubleArray.h"
33 #include "COLLADAFWTypes.h"
34
35 #include <vector>
36 #include <map>
37 #include <set>
38 #include <algorithm>
39
40 extern "C" {
41 #include "DNA_object_types.h"
42 #include "DNA_mesh_types.h"
43 #include "DNA_customdata_types.h"
44 #include "DNA_texture_types.h"
45 #include "DNA_scene_types.h"
46
47 #include "RNA_access.h"
48
49 #include "BLI_linklist.h"
50 #include "BLI_utildefines.h"
51 #include "BLI_string.h"
52
53 #include "BKE_context.h"
54 #include "BKE_object.h"
55 #include "BKE_DerivedMesh.h"
56 #include "BKE_scene.h"
57 #include "BKE_idprop.h"
58 }
59
60 #include "ImportSettings.h"
61 #include "ExportSettings.h"
62 #include "collada_internal.h"
63
64 typedef std::map<COLLADAFW::TextureMapId, std::vector<MTex *> > TexIndexTextureArrayMap;
65
66 extern Main *bc_get_main();
67 extern EvaluationContext *bc_get_evaluation_context();
68 extern void bc_update_scene(Scene *scene, float ctime);
69
70 extern float bc_get_float_value(const COLLADAFW::FloatOrDoubleArray& array, unsigned int index);
71 extern int bc_test_parent_loop(Object *par, Object *ob);
72 extern int bc_set_parent(Object *ob, Object *par, bContext *C, bool is_parent_space = true);
73 extern Object *bc_add_object(Scene *scene, int type, const char *name);
74 extern Mesh *bc_get_mesh_copy(Scene *scene, Object *ob, BC_export_mesh_type export_mesh_type, bool apply_modifiers, bool triangulate);
75
76 extern Object *bc_get_assigned_armature(Object *ob);
77 extern Object *bc_get_highest_selected_ancestor_or_self(LinkNode *export_set, Object *ob);
78 extern bool bc_is_base_node(LinkNode *export_set, Object *ob);
79 extern bool bc_is_in_Export_set(LinkNode *export_set, Object *ob);
80 extern bool bc_has_object_type(LinkNode *export_set, short obtype);
81
82 extern int bc_is_marked(Object *ob);
83 extern void bc_remove_mark(Object *ob);
84 extern void bc_set_mark(Object *ob);
85
86 extern char *bc_CustomData_get_layer_name(const CustomData *data, int type, int n);
87 extern char *bc_CustomData_get_active_layer_name(const CustomData *data, int type);
88
89 extern void bc_bubble_sort_by_Object_name(LinkNode *export_set);
90 extern bool bc_is_root_bone(Bone *aBone, bool deform_bones_only);
91 extern int  bc_get_active_UVLayer(Object *ob);
92
93 extern std::string bc_replace_string(std::string data, const std::string& pattern, const std::string& replacement); 
94 extern std::string bc_url_encode(std::string data); 
95 extern void bc_match_scale(Object *ob, UnitConverter &bc_unit, bool scale_to_scene);
96 extern void bc_match_scale(std::vector<Object *> *objects_done, UnitConverter &unit_converter, bool scale_to_scene);
97
98 extern void bc_decompose(float mat[4][4], float *loc, float eul[3], float quat[4], float *size);
99 extern void bc_rotate_from_reference_quat(float quat_to[4], float quat_from[4], float mat_to[4][4]);
100
101 extern void bc_triangulate_mesh(Mesh *me);
102 extern bool bc_is_leaf_bone(Bone *bone);
103 extern EditBone *bc_get_edit_bone(bArmature * armature, char *name);
104 extern int bc_set_layer(int bitfield, int layer, bool enable);
105 extern int bc_set_layer(int bitfield, int layer);
106
107 inline bool bc_in_range(float a, float b, float range) {
108         return abs(a - b) < range;
109 }
110 void bc_copy_m4_farray(float r[4][4], float *a);
111 void bc_copy_farray_m4(float *r, float a[4][4]);
112
113 extern void bc_sanitize_mat(float mat[4][4], int precision);
114 extern void bc_sanitize_mat(double mat[4][4], int precision);
115
116 extern IDProperty *bc_get_IDProperty(Bone *bone, std::string key);
117 extern void bc_set_IDProperty(EditBone *ebone, const char *key, float value);
118 extern void bc_set_IDPropertyMatrix(EditBone *ebone, const char *key, float mat[4][4]);
119
120 extern float bc_get_property(Bone *bone, std::string key, float def);
121 extern void bc_get_property_vector(Bone *bone, std::string key, float val[3], const float def[3]);
122 extern bool bc_get_property_matrix(Bone *bone, std::string key, float mat[4][4]);
123
124 extern void bc_create_restpose_mat(const ExportSettings *export_settings, Bone *bone, float to_mat[4][4], float world[4][4], bool use_local_space);
125
126 extern std::string bc_get_active_uvlayer_name(Object *ob);
127 extern std::string bc_get_active_uvlayer_name(Mesh *me);
128 extern std::string bc_get_uvlayer_name(Mesh *me, int layer);
129
130 extern std::set<Image *> bc_getUVImages(Scene *sce, bool all_uv_layers);
131 extern std::set<Image *> bc_getUVImages(Object *ob, bool all_uv_layers);
132 extern std::set<Object *> bc_getUVTexturedObjects(Scene *sce, bool all_uv_layers);
133
134 class BCPolygonNormalsIndices
135 {
136         std::vector<unsigned int> normal_indices;
137
138         public:
139
140         void add_index(unsigned int index) {
141                 normal_indices.push_back(index);
142         }
143
144         unsigned int operator[](unsigned int i) { 
145                 return normal_indices[i]; 
146         }
147
148 };
149
150 class BoneExtended {
151
152 private:
153         char  name[MAXBONENAME];
154         int   chain_length;
155         bool  is_leaf;
156         float tail[3];
157         float roll;
158
159         int   bone_layers;
160         int   use_connect;
161         bool  has_custom_tail;
162         bool  has_custom_roll;
163
164 public:
165
166         BoneExtended(EditBone *aBone);
167
168         void set_name(char *aName);
169         char *get_name();
170
171         void set_chain_length(const int aLength);
172         int  get_chain_length();
173
174         void set_leaf_bone(bool state);
175         bool is_leaf_bone();
176
177         void set_bone_layers(std::string layers, std::vector<std::string> &layer_labels);
178         int get_bone_layers();
179         static std::string get_bone_layers(int bitfield);
180
181         void set_roll(float roll);
182         bool has_roll();
183         float get_roll();
184
185         void set_tail(float *vec);
186         float *get_tail();
187         bool has_tail();
188
189         void set_use_connect(int use_connect);
190         int get_use_connect();
191 };
192
193 /* a map to store bone extension maps
194 | std:string     : an armature name
195 | BoneExtended * : a map that contains extra data for bones
196 */
197 typedef std::map<std::string, BoneExtended *> BoneExtensionMap;
198
199 /*
200 | A class to organise bone extendion data for multiple Armatures.
201 | this is needed for the case where a Collada file contains 2 or more
202 | separate armatures.
203 */
204 class BoneExtensionManager {
205 private:
206         std::map<std::string, BoneExtensionMap *> extended_bone_maps;
207
208 public:
209         BoneExtensionMap &getExtensionMap(bArmature *armature);
210         ~BoneExtensionManager();
211 };
212
213 #endif