- assigning vectors in the BGE wasn't updating the vector from the callback
[blender.git] / source / gameengine / Ketsji / KX_MeshProxy.cpp
1 /**
2  * $Id$
3  * ***** BEGIN GPL LICENSE BLOCK *****
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU General Public License
7  * as published by the Free Software Foundation; either version 2
8  * of the License, or (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software Foundation,
17  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
18  *
19  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
20  * All rights reserved.
21  *
22  * The Original Code is: all of this file.
23  *
24  * Contributor(s): none yet.
25  *
26  * ***** END GPL LICENSE BLOCK *****
27  */
28
29 #ifdef HAVE_CONFIG_H
30 #include <config.h>
31 #endif
32
33 #include "KX_MeshProxy.h"
34 #include "RAS_IPolygonMaterial.h"
35 #include "RAS_MeshObject.h"
36
37 #include "KX_VertexProxy.h"
38 #include "KX_PolyProxy.h"
39
40 #include "KX_PolygonMaterial.h"
41 #include "KX_BlenderMaterial.h"
42
43 #include "KX_PyMath.h"
44 #include "KX_ConvertPhysicsObject.h"
45
46 #include "PyObjectPlus.h" 
47
48 PyTypeObject KX_MeshProxy::Type = {
49         PyVarObject_HEAD_INIT(NULL, 0)
50         "KX_MeshProxy",
51         sizeof(PyObjectPlus_Proxy),
52         0,
53         py_base_dealloc,
54         0,
55         0,
56         0,
57         0,
58         py_base_repr,
59         0,0,0,0,0,0,0,0,0,
60         Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
61         0,0,0,0,0,0,0,
62         Methods,
63         0,
64         0,
65         &CValue::Type,
66         0,0,0,0,0,0,
67         py_base_new
68 };
69
70 PyMethodDef KX_MeshProxy::Methods[] = {
71 {"getMaterialName", (PyCFunction)KX_MeshProxy::sPyGetMaterialName,METH_VARARGS},
72 {"getTextureName", (PyCFunction)KX_MeshProxy::sPyGetTextureName,METH_VARARGS},
73 {"getVertexArrayLength", (PyCFunction)KX_MeshProxy::sPyGetVertexArrayLength,METH_VARARGS},
74 {"getVertex", (PyCFunction)KX_MeshProxy::sPyGetVertex,METH_VARARGS},
75 {"getPolygon", (PyCFunction)KX_MeshProxy::sPyGetPolygon,METH_VARARGS},
76 //{"getIndexArrayLength", (PyCFunction)KX_MeshProxy::sPyGetIndexArrayLength,METH_VARARGS},
77   {NULL,NULL} //Sentinel
78 };
79
80 PyAttributeDef KX_MeshProxy::Attributes[] = {
81         KX_PYATTRIBUTE_RO_FUNCTION("materials",         KX_MeshProxy, pyattr_get_materials),
82         KX_PYATTRIBUTE_RO_FUNCTION("numPolygons",       KX_MeshProxy, pyattr_get_numPolygons),
83         KX_PYATTRIBUTE_RO_FUNCTION("numMaterials",      KX_MeshProxy, pyattr_get_numMaterials),
84
85         { NULL }        //Sentinel
86 };
87
88 void KX_MeshProxy::SetMeshModified(bool v)
89 {
90         m_meshobj->SetMeshModified(v);
91 }
92
93 KX_MeshProxy::KX_MeshProxy(RAS_MeshObject* mesh)
94         : CValue(), m_meshobj(mesh)
95 {
96 }
97
98 KX_MeshProxy::~KX_MeshProxy()
99 {
100 }
101
102
103
104 // stuff for cvalue related things
105 CValue*         KX_MeshProxy::Calc(VALUE_OPERATOR op, CValue *val) { return NULL;}
106 CValue*         KX_MeshProxy::CalcFinal(VALUE_DATA_TYPE dtype, VALUE_OPERATOR op, CValue *val) { return NULL;}  
107
108 const STR_String &      KX_MeshProxy::GetText() {return m_meshobj->GetName();};
109 double          KX_MeshProxy::GetNumber() { return -1;}
110 STR_String&     KX_MeshProxy::GetName() { return m_meshobj->GetName();}
111 void            KX_MeshProxy::SetName(const char *name) { };
112 CValue*         KX_MeshProxy::GetReplica() { return NULL;}
113
114
115 // stuff for python integration
116
117 PyObject* KX_MeshProxy::PyGetMaterialName(PyObject* args, PyObject* kwds)
118 {
119     int matid= 1;
120         STR_String matname;
121
122         if (PyArg_ParseTuple(args,"i:getMaterialName",&matid))
123         {
124                 matname = m_meshobj->GetMaterialName(matid);
125         }
126         else {
127                 return NULL;
128         }
129
130         return PyUnicode_FromString(matname.Ptr());
131                 
132 }
133         
134
135 PyObject* KX_MeshProxy::PyGetTextureName(PyObject* args, PyObject* kwds)
136 {
137     int matid= 1;
138         STR_String matname;
139
140         if (PyArg_ParseTuple(args,"i:getTextureName",&matid))
141         {
142                 matname = m_meshobj->GetTextureName(matid);
143         }
144         else {
145                 return NULL;
146         }
147
148         return PyUnicode_FromString(matname.Ptr());
149                 
150 }
151
152 PyObject* KX_MeshProxy::PyGetVertexArrayLength(PyObject* args, PyObject* kwds)
153 {
154     int matid= 0;
155         int length = 0;
156
157         
158         if (!PyArg_ParseTuple(args,"i:getVertexArrayLength",&matid))
159                 return NULL;
160         
161
162         RAS_MeshMaterial *mmat = m_meshobj->GetMeshMaterial(matid); /* can be NULL*/
163         
164         if (mmat)
165         {
166                 RAS_IPolyMaterial* mat = mmat->m_bucket->GetPolyMaterial();
167                 if (mat)
168                         length = m_meshobj->NumVertices(mat);
169         }
170         
171         return PyLong_FromSsize_t(length);
172 }
173
174
175 PyObject* KX_MeshProxy::PyGetVertex(PyObject* args, PyObject* kwds)
176 {
177     int vertexindex;
178         int matindex;
179
180         if (!PyArg_ParseTuple(args,"ii:getVertex",&matindex,&vertexindex))
181                 return NULL;
182         
183         RAS_TexVert* vertex = m_meshobj->GetVertex(matindex,vertexindex);
184         
185         if(vertex==NULL) {
186                 PyErr_SetString(PyExc_ValueError, "mesh.getVertex(mat_idx, vert_idx): KX_MeshProxy, could not get a vertex at the given indicies");
187                 return NULL;
188         }
189         
190         return (new KX_VertexProxy(this, vertex))->NewProxy(true);
191 }
192
193 PyObject* KX_MeshProxy::PyGetPolygon(PyObject* args, PyObject* kwds)
194 {
195     int polyindex= 1;
196         PyObject* polyob = NULL;
197
198         if (!PyArg_ParseTuple(args,"i:getPolygon",&polyindex))
199                 return NULL;
200         
201         if (polyindex<0 || polyindex >= m_meshobj->NumPolygons())
202         {
203                 PyErr_SetString(PyExc_AttributeError, "mesh.getPolygon(int): KX_MeshProxy, invalid polygon index");
204                 return NULL;
205         }
206                 
207
208         RAS_Polygon* polygon = m_meshobj->GetPolygon(polyindex);
209         if (polygon)
210         {
211                 polyob = (new KX_PolyProxy(m_meshobj, polygon))->NewProxy(true);
212         }
213         else {
214                 PyErr_SetString(PyExc_AttributeError, "mesh.getPolygon(int): KX_MeshProxy, polygon is NULL, unknown reason");
215         }
216         return polyob;
217 }
218
219 PyObject* KX_MeshProxy::pyattr_get_materials(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
220 {
221         KX_MeshProxy* self= static_cast<KX_MeshProxy*>(self_v);
222         
223         int tot= self->m_meshobj->NumMaterials();
224         int i;
225         
226         PyObject *materials = PyList_New( tot );
227         
228         list<RAS_MeshMaterial>::iterator mit= self->m_meshobj->GetFirstMaterial();
229         
230         
231         for(i=0; i<tot; mit++, i++) {
232                 RAS_IPolyMaterial *polymat = mit->m_bucket->GetPolyMaterial();   
233                 
234                 /* Why do we need to check for RAS_BLENDERMAT if both are cast to a (PyObject*)? - Campbell */
235                 if(polymat->GetFlag() & RAS_BLENDERMAT)          
236                 {        
237                         KX_BlenderMaterial *mat = static_cast<KX_BlenderMaterial*>(polymat);     
238                         PyList_SET_ITEM(materials, i, mat->GetProxy());
239                 }
240                 else {  
241                         KX_PolygonMaterial *mat = static_cast<KX_PolygonMaterial*>(polymat);
242                         PyList_SET_ITEM(materials, i, mat->GetProxy());
243                 }
244         }       
245         return materials;
246 }
247
248 PyObject * KX_MeshProxy::pyattr_get_numMaterials(void * selfv, const KX_PYATTRIBUTE_DEF * attrdef) {
249         KX_MeshProxy * self = static_cast<KX_MeshProxy *> (selfv);
250         return PyLong_FromSsize_t(self->m_meshobj->NumMaterials());
251 }
252
253 PyObject * KX_MeshProxy::pyattr_get_numPolygons(void * selfv, const KX_PYATTRIBUTE_DEF * attrdef) {
254         KX_MeshProxy * self = static_cast<KX_MeshProxy *> (selfv);
255         return PyLong_FromSsize_t(self->m_meshobj->NumPolygons());
256 }
257
258 /* a close copy of ConvertPythonToGameObject but for meshes */
259 bool ConvertPythonToMesh(PyObject * value, RAS_MeshObject **object, bool py_none_ok, const char *error_prefix)
260 {
261         if (value==NULL) {
262                 PyErr_Format(PyExc_TypeError, "%s, python pointer NULL, should never happen", error_prefix);
263                 *object = NULL;
264                 return false;
265         }
266                 
267         if (value==Py_None) {
268                 *object = NULL;
269                 
270                 if (py_none_ok) {
271                         return true;
272                 } else {
273                         PyErr_Format(PyExc_TypeError, "%s, expected KX_MeshProxy or a KX_MeshProxy name, None is invalid", error_prefix);
274                         return false;
275                 }
276         }
277         
278         if (PyUnicode_Check(value)) {
279                 *object = (RAS_MeshObject*)SCA_ILogicBrick::m_sCurrentLogicManager->GetMeshByName(STR_String( _PyUnicode_AsString(value) ));
280                 
281                 if (*object) {
282                         return true;
283                 } else {
284                         PyErr_Format(PyExc_ValueError, "%s, requested name \"%s\" did not match any KX_MeshProxy in this scene", error_prefix, _PyUnicode_AsString(value));
285                         return false;
286                 }
287         }
288         
289         if (PyObject_TypeCheck(value, &KX_MeshProxy::Type)) {
290                 KX_MeshProxy *kx_mesh = static_cast<KX_MeshProxy*>BGE_PROXY_REF(value);
291                 
292                 /* sets the error */
293                 if (kx_mesh==NULL) {
294                         PyErr_Format(PyExc_SystemError, "%s, " BGE_PROXY_ERROR_MSG, error_prefix);
295                         return false;
296                 }               
297                 
298                 *object = kx_mesh->GetMesh();
299                 return true;
300         }
301         
302         *object = NULL;
303         
304         if (py_none_ok) {
305                 PyErr_Format(PyExc_TypeError, "%s, expect a KX_MeshProxy, a string or None", error_prefix);
306         } else {
307                 PyErr_Format(PyExc_TypeError, "%s, expect a KX_MeshProxy or a string", error_prefix);
308         }
309         
310         return false;
311 }