Cleanup: manually remove header text not handled by automation
[blender.git] / source / blender / collada / BCSampleData.cpp
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  * The Original Code is Copyright (C) 2008 Blender Foundation.
17  * All rights reserved.
18  */
19
20 #include "BCSampleData.h"
21 #include "collada_utils.h"
22
23 BCSample::BCSample(Object *ob):
24         obmat(ob)
25 {}
26
27 BCSample::~BCSample()
28 {
29         BCBoneMatrixMap::iterator it;
30         for (it = bonemats.begin(); it != bonemats.end(); ++it) {
31                 delete it->second;
32         }
33 }
34
35 void BCSample::add_bone_matrix(Bone *bone, Matrix &mat)
36 {
37         BCMatrix *matrix;
38         BCBoneMatrixMap::const_iterator it = bonemats.find(bone);
39         if (it != bonemats.end()) {
40                 throw std::invalid_argument("bone " + std::string(bone->name) + " already defined before");
41         }
42         matrix = new BCMatrix(mat);
43         bonemats[bone] = matrix;
44 }
45
46 BCMatrix::BCMatrix(Matrix &mat)
47 {
48         set_transform(mat);
49 }
50
51 BCMatrix::BCMatrix(Object *ob)
52 {
53         set_transform(ob);
54 }
55
56 void BCMatrix::set_transform(Matrix &mat)
57 {
58         copy_m4_m4(matrix, mat);
59         mat4_decompose(this->loc, this->q, this->size, mat);
60         quat_to_eul(this->rot, this->q);
61 }
62
63 void BCMatrix::set_transform(Object *ob)
64 {
65         Matrix lmat;
66
67         BKE_object_matrix_local_get(ob, lmat);
68         copy_m4_m4(matrix, lmat);
69
70         mat4_decompose(this->loc, this->q, this->size, lmat);
71         quat_to_compatible_eul(this->rot, ob->rot, this->q);
72 }
73
74 const BCMatrix *BCSample::get_matrix(Bone *bone) const
75 {
76         BCBoneMatrixMap::const_iterator it = bonemats.find(bone);
77         if (it == bonemats.end()) {
78                 return NULL;
79         }
80         return it->second;
81 }
82
83 const BCMatrix &BCSample::get_matrix() const
84 {
85         return obmat;
86 }
87
88
89 /* Get channel value */
90 const bool BCSample::get_value(std::string channel_target, const int array_index, float *val) const
91 {
92         if (channel_target == "location") {
93                 *val = obmat.location()[array_index];
94         }
95         else if (channel_target == "scale") {
96                 *val = obmat.scale()[array_index];
97         }
98         else if (
99                 channel_target == "rotation" ||
100                 channel_target == "rotation_euler") {
101                 *val = obmat.rotation()[array_index];
102         }
103         else if (channel_target == "rotation_quat") {
104                 *val = obmat.quat()[array_index];
105         }
106         else {
107                 *val = 0;
108                 return false;
109         }
110
111         return true;
112 }
113
114 void BCMatrix::copy(Matrix &out, Matrix &in)
115 {
116         /* destination comes first: */
117         memcpy(out, in, sizeof(Matrix));
118 }
119
120 void BCMatrix::transpose(Matrix &mat)
121 {
122         transpose_m4(mat);
123 }
124
125 void BCMatrix::sanitize(Matrix &mat, int precision)
126 {
127         bc_sanitize_mat(mat, precision);
128 }
129
130 void BCMatrix::unit()
131 {
132         unit_m4(matrix);
133 }
134
135 /*
136 We need double here because the OpenCollada API needs it.
137 precision = -1 indicates to not limit the precision
138 */
139 void BCMatrix::get_matrix(double(&mat)[4][4], const bool transposed, const int precision) const
140 {
141         for (int i = 0; i < 4; i++)
142                 for (int j = 0; j < 4; j++) {
143                         float val = (transposed) ? matrix[j][i] : matrix[i][j];
144                         if (precision >= 0)
145                                 val = floor((val * pow(10, precision) + 0.5)) / pow(10, precision);
146                         mat[i][j] = val;
147                 }
148 }
149
150 const bool BCMatrix::in_range(const BCMatrix &other, float distance) const
151 {
152         for (int i = 0; i < 4; i++) {
153                 for (int j = 0; j < 4; j++) {
154                         if (fabs(other.matrix[i][j] - matrix[i][j]) > distance) {
155                                 return false;
156                         }
157                 }
158         }
159         return true;
160 }
161
162 float(&BCMatrix::location() const)[3]
163 {
164         return loc;
165 }
166
167 float(&BCMatrix::rotation() const)[3]
168 {
169         return rot;
170 }
171
172 float(&BCMatrix::scale() const)[3]
173 {
174         return size;
175 }
176
177 float(&BCMatrix::quat() const)[4]
178 {
179         return q;
180 }