Fix T67002: Crash redoing an action after ID rename
[blender.git] / source / blender / collada / BCMath.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 "BCMath.h"
21 #include "BlenderContext.h"
22
23 void BCQuat::rotate_to(Matrix &mat_to)
24 {
25   Quat qd;
26   Matrix matd;
27   Matrix mati;
28   Matrix mat_from;
29
30   quat_to_mat4(mat_from, q);
31
32   /* Calculate the difference matrix matd between mat_from and mat_to */
33   invert_m4_m4(mati, mat_from);
34   mul_m4_m4m4(matd, mati, mat_to);
35
36   mat4_to_quat(qd, matd);
37
38   mul_qt_qtqt(q, qd, q); /* rotate to the final rotation to mat_to */
39 }
40
41 BCMatrix::BCMatrix(const BCMatrix &mat)
42 {
43   set_transform(mat.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 BCMatrix::BCMatrix()
57 {
58   unit();
59 }
60
61 BCMatrix::BCMatrix(BC_global_forward_axis global_forward_axis, BC_global_up_axis global_up_axis)
62 {
63   float mrot[3][3];
64   float mat[4][4];
65   mat3_from_axis_conversion(
66       BC_DEFAULT_FORWARD, BC_DEFAULT_UP, global_forward_axis, global_up_axis, mrot);
67
68   transpose_m3(mrot);  // TODO: Verify that mat3_from_axis_conversion() returns a transposed matrix
69   copy_m4_m3(mat, mrot);
70   set_transform(mat);
71 }
72
73 void BCMatrix::add_transform(const Matrix &mat, bool inverse)
74 {
75   add_transform(this->matrix, mat, this->matrix, inverse);
76 }
77
78 void BCMatrix::add_transform(const BCMatrix &mat, bool inverse)
79 {
80   add_transform(this->matrix, mat.matrix, this->matrix, inverse);
81 }
82
83 void BCMatrix::apply_transform(const BCMatrix &mat, bool inverse)
84 {
85   apply_transform(this->matrix, mat.matrix, this->matrix, inverse);
86 }
87
88 void BCMatrix::add_transform(Matrix &to, const Matrix &transform, const Matrix &from, bool inverse)
89 {
90   if (inverse) {
91     Matrix globinv;
92     invert_m4_m4(globinv, transform);
93     add_transform(to, globinv, from, /*inverse=*/false);
94   }
95   else {
96     mul_m4_m4m4(to, transform, from);
97   }
98 }
99
100 void BCMatrix::apply_transform(Matrix &to,
101                                const Matrix &transform,
102                                const Matrix &from,
103                                bool inverse)
104 {
105   Matrix globinv;
106   invert_m4_m4(globinv, transform);
107   if (inverse) {
108     add_transform(to, globinv, from, /*inverse=*/false);
109   }
110   else {
111     mul_m4_m4m4(to, transform, from);
112     mul_m4_m4m4(to, to, globinv);
113   }
114 }
115
116 void BCMatrix::add_inverted_transform(Matrix &to, const Matrix &transform, const Matrix &from)
117 {
118   Matrix workmat;
119   invert_m4_m4(workmat, transform);
120   mul_m4_m4m4(to, workmat, from);
121 }
122
123 void BCMatrix::set_transform(Object *ob)
124 {
125   Matrix lmat;
126
127   BKE_object_matrix_local_get(ob, lmat);
128   copy_m4_m4(matrix, lmat);
129
130   mat4_decompose(this->loc, this->q, this->size, lmat);
131   quat_to_compatible_eul(this->rot, ob->rot, this->q);
132 }
133
134 void BCMatrix::set_transform(Matrix &mat)
135 {
136   copy_m4_m4(matrix, mat);
137   mat4_decompose(this->loc, this->q, this->size, mat);
138   quat_to_eul(this->rot, this->q);
139 }
140
141 void BCMatrix::copy(Matrix &out, Matrix &in)
142 {
143   /* destination comes first: */
144   memcpy(out, in, sizeof(Matrix));
145 }
146
147 void BCMatrix::transpose(Matrix &mat)
148 {
149   transpose_m4(mat);
150 }
151
152 void BCMatrix::sanitize(Matrix &mat, int precision)
153 {
154   for (int i = 0; i < 4; i++) {
155     for (int j = 0; j < 4; j++) {
156       double val = (double)mat[i][j];
157       val = double_round(val, precision);
158       mat[i][j] = (float)val;
159     }
160   }
161 }
162
163 void BCMatrix::sanitize(DMatrix &mat, int precision)
164 {
165   for (int i = 0; i < 4; i++) {
166     for (int j = 0; j < 4; j++) {
167       mat[i][j] = double_round(mat[i][j], precision);
168     }
169   }
170 }
171
172 void BCMatrix::unit()
173 {
174   unit_m4(this->matrix);
175   mat4_decompose(this->loc, this->q, this->size, this->matrix);
176   quat_to_eul(this->rot, this->q);
177 }
178
179 /* We need double here because the OpenCollada API needs it.
180  * precision = -1 indicates to not limit the precision. */
181 void BCMatrix::get_matrix(DMatrix &mat, const bool transposed, const int precision) const
182 {
183   for (int i = 0; i < 4; i++) {
184     for (int j = 0; j < 4; j++) {
185       float val = (transposed) ? matrix[j][i] : matrix[i][j];
186       if (precision >= 0) {
187         val = floor((val * pow(10, precision) + 0.5)) / pow(10, precision);
188       }
189       mat[i][j] = val;
190     }
191   }
192 }
193
194 void BCMatrix::get_matrix(Matrix &mat,
195                           const bool transposed,
196                           const int precision,
197                           const bool inverted) const
198 {
199   for (int i = 0; i < 4; i++) {
200     for (int j = 0; j < 4; j++) {
201       float val = (transposed) ? matrix[j][i] : matrix[i][j];
202       if (precision >= 0) {
203         val = floor((val * pow(10, precision) + 0.5)) / pow(10, precision);
204       }
205       mat[i][j] = val;
206     }
207   }
208
209   if (inverted) {
210     invert_m4(mat);
211   }
212 }
213
214 const bool BCMatrix::in_range(const BCMatrix &other, float distance) const
215 {
216   for (int i = 0; i < 4; i++) {
217     for (int j = 0; j < 4; j++) {
218       if (fabs(other.matrix[i][j] - matrix[i][j]) > distance) {
219         return false;
220       }
221     }
222   }
223   return true;
224 }
225
226 float (&BCMatrix::location() const)[3]
227 {
228   return loc;
229 }
230
231 float (&BCMatrix::rotation() const)[3]
232 {
233   return rot;
234 }
235
236 float (&BCMatrix::scale() const)[3]
237 {
238   return size;
239 }
240
241 float (&BCMatrix::quat() const)[4]
242 {
243   return q;
244 }