rename and negate DISABLE_PYTHON --> WITH_PYTHON
[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 #if defined(WIN32) && !defined(FREE_WINDOWS)
32 #pragma warning (disable : 4786)
33 #endif //WIN32
34
35 #ifdef WITH_PYTHON
36
37 #include "MT_Vector3.h"
38 #include "MT_Vector4.h"
39 #include "MT_Matrix4x4.h"
40 #include "MT_Point2.h"
41
42 #include "ListValue.h"
43
44 #include "KX_Python.h"
45 #include "KX_PyMath.h"
46
47 bool PyOrientationTo(PyObject* pyval, MT_Matrix3x3 &rot, const char *error_prefix)
48 {
49         int size= PySequence_Size(pyval);
50         
51         if (size == 4)
52         {
53                 MT_Quaternion qrot;
54                 if (PyQuatTo(pyval, qrot))
55                 {
56                         rot.setRotation(qrot);
57                         return true;
58                 }
59         }
60         else if (size == 3) {
61                 /* 3x3 matrix or euler */
62                 MT_Vector3 erot;
63                 if (PyVecTo(pyval, erot))
64                 {
65                         rot.setEuler(erot);
66                         return true;
67                 }
68                 PyErr_Clear();
69                 
70                 if (PyMatTo(pyval, rot))
71                 {
72                         return true;
73                 }
74         }
75         
76         PyErr_Format(PyExc_TypeError, "%s, could not set the orientation from a 3x3 matrix, quaternion or euler sequence", error_prefix);
77         return false;
78 }
79
80 bool PyQuatTo(PyObject* pyval, MT_Quaternion &qrot)
81 {
82         if(!PyVecTo(pyval, qrot))
83                 return false;
84
85         /* annoying!, Blender/Mathutils have the W axis first! */
86         MT_Scalar w= qrot[0]; /* from python, this is actually the W */
87         qrot[0]= qrot[1];
88         qrot[1]= qrot[2];
89         qrot[2]= qrot[3];
90         qrot[3]= w;
91
92         return true;
93 }
94
95 PyObject* PyObjectFrom(const MT_Matrix4x4 &mat)
96 {
97 #ifdef USE_MATHUTILS
98         float fmat[16];
99         mat.getValue(fmat);
100         return newMatrixObject(fmat, 4, 4, Py_NEW, NULL);
101 #else
102         PyObject *collist = PyList_New(4);
103         PyObject *col;
104         int i;
105         
106         for(i=0; i < 4; i++) {
107                 col = PyList_New(4);
108                 PyList_SET_ITEM(col, 0, PyFloat_FromDouble(mat[0][i]));
109                 PyList_SET_ITEM(col, 1, PyFloat_FromDouble(mat[1][i]));
110                 PyList_SET_ITEM(col, 2, PyFloat_FromDouble(mat[2][i]));
111                 PyList_SET_ITEM(col, 3, PyFloat_FromDouble(mat[3][i]));
112                 PyList_SET_ITEM(collist, i, col);
113         }
114         
115         return collist;
116 #endif
117 }
118
119 PyObject* PyObjectFrom(const MT_Matrix3x3 &mat)
120 {
121 #ifdef USE_MATHUTILS
122         float fmat[9];
123         mat.getValue3x3(fmat);
124         return newMatrixObject(fmat, 3, 3, Py_NEW, NULL);
125 #else
126         PyObject *collist = PyList_New(3);
127         PyObject *col;
128         int i;
129         
130         for(i=0; i < 3; i++) {
131                 col = PyList_New(3);
132                 PyList_SET_ITEM(col, 0, PyFloat_FromDouble(mat[0][i]));
133                 PyList_SET_ITEM(col, 1, PyFloat_FromDouble(mat[1][i]));
134                 PyList_SET_ITEM(col, 2, PyFloat_FromDouble(mat[2][i]));
135                 PyList_SET_ITEM(collist, i, col);
136         }
137         
138         return collist;
139 #endif
140 }
141
142 #ifdef USE_MATHUTILS
143 PyObject* PyObjectFrom(const MT_Quaternion &qrot)
144 {
145         /* NOTE, were re-ordering here for Mathutils compat */
146         float fvec[4]= {qrot[3], qrot[0], qrot[1], qrot[2]};
147         return newQuaternionObject(fvec, Py_NEW, NULL);
148 }
149 #endif
150
151 PyObject* PyObjectFrom(const MT_Tuple4 &vec)
152 {
153 #ifdef USE_MATHUTILS
154         float fvec[4]= {vec[0], vec[1], vec[2], vec[3]};
155         return newVectorObject(fvec, 4, Py_NEW, NULL);
156 #else
157         PyObject *list = PyList_New(4);
158         PyList_SET_ITEM(list, 0, PyFloat_FromDouble(vec[0]));
159         PyList_SET_ITEM(list, 1, PyFloat_FromDouble(vec[1]));
160         PyList_SET_ITEM(list, 2, PyFloat_FromDouble(vec[2]));
161         PyList_SET_ITEM(list, 3, PyFloat_FromDouble(vec[3]));
162         return list;
163 #endif
164 }
165
166 PyObject* PyObjectFrom(const MT_Tuple3 &vec)
167 {
168 #ifdef USE_MATHUTILS
169         float fvec[3]= {vec[0], vec[1], vec[2]};
170         return newVectorObject(fvec, 3, Py_NEW, NULL);
171 #else
172         PyObject *list = PyList_New(3);
173         PyList_SET_ITEM(list, 0, PyFloat_FromDouble(vec[0]));
174         PyList_SET_ITEM(list, 1, PyFloat_FromDouble(vec[1]));
175         PyList_SET_ITEM(list, 2, PyFloat_FromDouble(vec[2]));
176         return list;
177 #endif  
178 }
179
180 PyObject* PyObjectFrom(const MT_Tuple2 &vec)
181 {
182 #ifdef USE_MATHUTILS
183         float fvec[2]= {vec[0], vec[1]};
184         return newVectorObject(fvec, 2, Py_NEW, NULL);
185 #else
186         PyObject *list = PyList_New(2);
187         PyList_SET_ITEM(list, 0, PyFloat_FromDouble(vec[0]));
188         PyList_SET_ITEM(list, 1, PyFloat_FromDouble(vec[1]));
189         return list;
190 #endif
191 }
192
193 #endif // WITH_PYTHON