4974566bba15c558ccf9347c323ada3052541410
[blender.git] / source / blender / collada / TransformWriter.cpp
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, Jan Diederich, Tod Liverseed,
19  *                 Nathan Letwory
20  *
21  * ***** END GPL LICENSE BLOCK *****
22  */
23
24 /** \file blender/collada/TransformWriter.cpp
25  *  \ingroup collada
26  */
27
28
29 #include "BKE_object.h"
30
31 #include "TransformWriter.h"
32
33 #include "BLI_math.h"
34
35 void TransformWriter::add_node_transform(COLLADASW::Node& node, float mat[][4], float parent_mat[][4])
36 {
37         float loc[3], rot[3], scale[3];
38         float local[4][4];
39
40         if (parent_mat) {
41                 float invpar[4][4];
42                 invert_m4_m4(invpar, parent_mat);
43                 mult_m4_m4m4(local, invpar, mat);
44         }
45         else {
46                 copy_m4_m4(local, mat);
47         }
48
49         double dmat[4][4];
50         UnitConverter* converter = new UnitConverter();
51         converter->mat4_to_dae_double(dmat, local);
52
53         TransformBase::decompose(local, loc, rot, NULL, scale);
54         if ( node.getType() == COLLADASW::Node::JOINT)
55                 node.addMatrix("transform", dmat);
56         else
57                 add_transform(node, loc, rot, scale);
58 }
59
60 void TransformWriter::add_node_transform_ob(COLLADASW::Node& node, Object *ob)
61 {
62         /*
63         float rot[3], loc[3], scale[3];
64
65         if (ob->parent) {
66                 float C[4][4], tmat[4][4], imat[4][4], mat[4][4];
67
68                 // factor out scale from obmat
69
70                 copy_v3_v3(scale, ob->size);
71
72                 ob->size[0] = ob->size[1] = ob->size[2] = 1.0f;
73                 object_to_mat4(ob, C);
74                 copy_v3_v3(ob->size, scale);
75
76                 mul_serie_m4(tmat, ob->parent->obmat, ob->parentinv, C, NULL, NULL, NULL, NULL, NULL);
77
78                 // calculate local mat
79
80                 invert_m4_m4(imat, ob->parent->obmat);
81                 mult_m4_m4m4(mat, imat, tmat);
82
83                 // done
84
85                 mat4_to_eul(rot, mat);
86                 copy_v3_v3(loc, mat[3]);
87         }
88         else {
89                 copy_v3_v3(loc, ob->loc);
90                 copy_v3_v3(rot, ob->rot);
91                 copy_v3_v3(scale, ob->size);
92         }
93
94         add_transform(node, loc, rot, scale);
95         */
96
97         /* Using parentinv should allow use of existing curves */
98         if (ob->parent) {
99                 // If parentinv is identity don't add it.
100                 bool add_parinv = false;
101
102                 for (int i = 0; i < 16; ++i) {
103                         float f = (i % 4 == i / 4) ? 1.0f : 0.0f;
104                         add_parinv |= (ob->parentinv[i % 4][i / 4] != f);
105                 }
106
107                 if (add_parinv) {
108                         double dmat[4][4];
109                         UnitConverter converter;
110                         converter.mat4_to_dae_double(dmat, ob->parentinv);
111                         node.addMatrix("parentinverse", dmat);
112                 }
113         }
114
115         add_transform(node, ob->loc, ob->rot, ob->size);
116 }
117
118 void TransformWriter::add_node_transform_identity(COLLADASW::Node& node)
119 {
120         float loc[] = {0.0f, 0.0f, 0.0f}, scale[] = {1.0f, 1.0f, 1.0f}, rot[] = {0.0f, 0.0f, 0.0f};
121         add_transform(node, loc, rot, scale);
122 }
123
124 void TransformWriter::add_transform(COLLADASW::Node& node, float loc[3], float rot[3], float scale[3])
125 {
126         node.addTranslate("location", loc[0], loc[1], loc[2]);
127         /*node.addRotateZ("rotationZ", COLLADABU::Math::Utils::radToDegF(rot[2]));
128         node.addRotateY("rotationY", COLLADABU::Math::Utils::radToDegF(rot[1]));
129         node.addRotateX("rotationX", COLLADABU::Math::Utils::radToDegF(rot[0]));*/
130         node.addRotateZ("rotationZ", RAD2DEGF(rot[2]));
131         node.addRotateY("rotationY", RAD2DEGF(rot[1]));
132         node.addRotateX("rotationX", RAD2DEGF(rot[0]));
133
134         node.addScale("scale", scale[0], scale[1], scale[2]);
135 }