add missing files after merging
[blender.git] / source / gameengine / Ketsji / KX_PyMath.cpp
1 /*
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): none yet.
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  * Initialize Python thingies.
29  */
30
31 /** \file gameengine/Ketsji/KX_PyMath.cpp
32  *  \ingroup ketsji
33  */
34
35
36 #if defined(WIN32) && !defined(FREE_WINDOWS)
37 #pragma warning (disable : 4786)
38 #endif //WIN32
39
40 #ifdef WITH_PYTHON
41
42 #include "MT_Vector3.h"
43 #include "MT_Vector4.h"
44 #include "MT_Matrix4x4.h"
45 #include "MT_Point2.h"
46
47 #include "ListValue.h"
48
49 #include "KX_Python.h"
50 #include "KX_PyMath.h"
51
52 bool PyOrientationTo(PyObject* pyval, MT_Matrix3x3 &rot, const char *error_prefix)
53 {
54         int size= PySequence_Size(pyval);
55         
56         if (size == 4)
57         {
58                 MT_Quaternion qrot;
59                 if (PyQuatTo(pyval, qrot))
60                 {
61                         rot.setRotation(qrot);
62                         return true;
63                 }
64         }
65         else if (size == 3) {
66                 /* 3x3 matrix or euler */
67                 MT_Vector3 erot;
68                 if (PyVecTo(pyval, erot))
69                 {
70                         rot.setEuler(erot);
71                         return true;
72                 }
73                 PyErr_Clear();
74                 
75                 if (PyMatTo(pyval, rot))
76                 {
77                         return true;
78                 }
79         }
80         
81         PyErr_Format(PyExc_TypeError, "%s, could not set the orientation from a 3x3 matrix, quaternion or euler sequence", error_prefix);
82         return false;
83 }
84
85 bool PyQuatTo(PyObject* pyval, MT_Quaternion &qrot)
86 {
87         if(!PyVecTo(pyval, qrot))
88                 return false;
89
90         /* annoying!, Blender/Mathutils have the W axis first! */
91         MT_Scalar w= qrot[0]; /* from python, this is actually the W */
92         qrot[0]= qrot[1];
93         qrot[1]= qrot[2];
94         qrot[2]= qrot[3];
95         qrot[3]= w;
96
97         return true;
98 }
99
100 PyObject* PyObjectFrom(const MT_Matrix4x4 &mat)
101 {
102 #ifdef USE_MATHUTILS
103         float fmat[16];
104         mat.getValue(fmat);
105         return newMatrixObject(fmat, 4, 4, Py_NEW, NULL);
106 #else
107         PyObject *collist = PyList_New(4);
108         PyObject *col;
109         int i;
110         
111         for(i=0; i < 4; i++) {
112                 col = PyList_New(4);
113                 PyList_SET_ITEM(col, 0, PyFloat_FromDouble(mat[0][i]));
114                 PyList_SET_ITEM(col, 1, PyFloat_FromDouble(mat[1][i]));
115                 PyList_SET_ITEM(col, 2, PyFloat_FromDouble(mat[2][i]));
116                 PyList_SET_ITEM(col, 3, PyFloat_FromDouble(mat[3][i]));
117                 PyList_SET_ITEM(collist, i, col);
118         }
119         
120         return collist;
121 #endif
122 }
123
124 PyObject* PyObjectFrom(const MT_Matrix3x3 &mat)
125 {
126 #ifdef USE_MATHUTILS
127         float fmat[9];
128         mat.getValue3x3(fmat);
129         return newMatrixObject(fmat, 3, 3, Py_NEW, NULL);
130 #else
131         PyObject *collist = PyList_New(3);
132         PyObject *col;
133         int i;
134         
135         for(i=0; i < 3; i++) {
136                 col = PyList_New(3);
137                 PyList_SET_ITEM(col, 0, PyFloat_FromDouble(mat[0][i]));
138                 PyList_SET_ITEM(col, 1, PyFloat_FromDouble(mat[1][i]));
139                 PyList_SET_ITEM(col, 2, PyFloat_FromDouble(mat[2][i]));
140                 PyList_SET_ITEM(collist, i, col);
141         }
142         
143         return collist;
144 #endif
145 }
146
147 #ifdef USE_MATHUTILS
148 PyObject* PyObjectFrom(const MT_Quaternion &qrot)
149 {
150         /* NOTE, were re-ordering here for Mathutils compat */
151         float fvec[4]= {qrot[3], qrot[0], qrot[1], qrot[2]};
152         return newQuaternionObject(fvec, Py_NEW, NULL);
153 }
154 #endif
155
156 PyObject* PyObjectFrom(const MT_Tuple4 &vec)
157 {
158 #ifdef USE_MATHUTILS
159         float fvec[4]= {vec[0], vec[1], vec[2], vec[3]};
160         return newVectorObject(fvec, 4, Py_NEW, NULL);
161 #else
162         PyObject *list = PyList_New(4);
163         PyList_SET_ITEM(list, 0, PyFloat_FromDouble(vec[0]));
164         PyList_SET_ITEM(list, 1, PyFloat_FromDouble(vec[1]));
165         PyList_SET_ITEM(list, 2, PyFloat_FromDouble(vec[2]));
166         PyList_SET_ITEM(list, 3, PyFloat_FromDouble(vec[3]));
167         return list;
168 #endif
169 }
170
171 PyObject* PyObjectFrom(const MT_Tuple3 &vec)
172 {
173 #ifdef USE_MATHUTILS
174         float fvec[3]= {vec[0], vec[1], vec[2]};
175         return newVectorObject(fvec, 3, Py_NEW, NULL);
176 #else
177         PyObject *list = PyList_New(3);
178         PyList_SET_ITEM(list, 0, PyFloat_FromDouble(vec[0]));
179         PyList_SET_ITEM(list, 1, PyFloat_FromDouble(vec[1]));
180         PyList_SET_ITEM(list, 2, PyFloat_FromDouble(vec[2]));
181         return list;
182 #endif  
183 }
184
185 PyObject* PyObjectFrom(const MT_Tuple2 &vec)
186 {
187 #ifdef USE_MATHUTILS
188         float fvec[2]= {vec[0], vec[1]};
189         return newVectorObject(fvec, 2, Py_NEW, NULL);
190 #else
191         PyObject *list = PyList_New(2);
192         PyList_SET_ITEM(list, 0, PyFloat_FromDouble(vec[0]));
193         PyList_SET_ITEM(list, 1, PyFloat_FromDouble(vec[1]));
194         return list;
195 #endif
196 }
197
198 #endif // WITH_PYTHON