ea9bc3e0fbc92348d650be26c99b7c4fd0049d54
[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 #if (PY_VERSION_HEX >= 0x02060000)
50         PyVarObject_HEAD_INIT(NULL, 0)
51 #else
52         /* python 2.5 and below */
53         PyObject_HEAD_INIT( NULL )  /* required py macro */
54         0,                          /* ob_size */
55 #endif
56         "KX_MeshProxy",
57         sizeof(PyObjectPlus_Proxy),
58         0,
59         py_base_dealloc,
60         0,
61         0,
62         0,
63         0,
64         py_base_repr,
65         0,0,0,0,0,0,0,0,0,
66         Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
67         0,0,0,0,0,0,0,
68         Methods,
69         0,
70         0,
71         &CValue::Type,
72         0,0,0,0,0,0,
73         py_base_new
74 };
75
76 PyMethodDef KX_MeshProxy::Methods[] = {
77 // Deprecated ----->
78 {"getNumMaterials", (PyCFunction)KX_MeshProxy::sPyGetNumMaterials,METH_VARARGS},
79 {"getNumPolygons", (PyCFunction)KX_MeshProxy::sPyGetNumPolygons,METH_NOARGS},
80 // <-----
81
82 {"getMaterialName", (PyCFunction)KX_MeshProxy::sPyGetMaterialName,METH_VARARGS},
83 {"getTextureName", (PyCFunction)KX_MeshProxy::sPyGetTextureName,METH_VARARGS},
84 {"getVertexArrayLength", (PyCFunction)KX_MeshProxy::sPyGetVertexArrayLength,METH_VARARGS},
85 {"getVertex", (PyCFunction)KX_MeshProxy::sPyGetVertex,METH_VARARGS},
86 {"getPolygon", (PyCFunction)KX_MeshProxy::sPyGetPolygon,METH_VARARGS},
87 //{"getIndexArrayLength", (PyCFunction)KX_MeshProxy::sPyGetIndexArrayLength,METH_VARARGS},
88
89   {NULL,NULL} //Sentinel
90 };
91
92 PyAttributeDef KX_MeshProxy::Attributes[] = {
93         KX_PYATTRIBUTE_RO_FUNCTION("materials",         KX_MeshProxy, pyattr_get_materials),
94         KX_PYATTRIBUTE_RO_FUNCTION("numPolygons",       KX_MeshProxy, pyattr_get_numPolygons),
95         KX_PYATTRIBUTE_RO_FUNCTION("numMaterials",      KX_MeshProxy, pyattr_get_numMaterials),
96
97         { NULL }        //Sentinel
98 };
99
100 void KX_MeshProxy::SetMeshModified(bool v)
101 {
102         m_meshobj->SetMeshModified(v);
103 }
104
105 KX_MeshProxy::KX_MeshProxy(RAS_MeshObject* mesh)
106         : CValue(), m_meshobj(mesh)
107 {
108 }
109
110 KX_MeshProxy::~KX_MeshProxy()
111 {
112 }
113
114
115
116 // stuff for cvalue related things
117 CValue*         KX_MeshProxy::Calc(VALUE_OPERATOR op, CValue *val) { return NULL;}
118 CValue*         KX_MeshProxy::CalcFinal(VALUE_DATA_TYPE dtype, VALUE_OPERATOR op, CValue *val) { return NULL;}  
119
120 const STR_String &      KX_MeshProxy::GetText() {return m_meshobj->GetName();};
121 double          KX_MeshProxy::GetNumber() { return -1;}
122 STR_String&     KX_MeshProxy::GetName() { return m_meshobj->GetName();}
123 void            KX_MeshProxy::SetName(const char *name) { };
124 CValue*         KX_MeshProxy::GetReplica() { return NULL;}
125
126
127 // stuff for python integration
128         
129 PyObject* KX_MeshProxy::PyGetNumMaterials(PyObject* args, PyObject* kwds)
130 {
131         int num = m_meshobj->NumMaterials();
132         ShowDeprecationWarning("getNumMaterials()", "the numMaterials property");
133         return PyLong_FromSsize_t(num);
134 }
135
136 PyObject* KX_MeshProxy::PyGetNumPolygons()
137 {
138         int num = m_meshobj->NumPolygons();
139         ShowDeprecationWarning("getNumPolygons()", "the numPolygons property");
140         return PyLong_FromSsize_t(num);
141 }
142
143 PyObject* KX_MeshProxy::PyGetMaterialName(PyObject* args, PyObject* kwds)
144 {
145     int matid= 1;
146         STR_String matname;
147
148         if (PyArg_ParseTuple(args,"i:getMaterialName",&matid))
149         {
150                 matname = m_meshobj->GetMaterialName(matid);
151         }
152         else {
153                 return NULL;
154         }
155
156         return PyUnicode_FromString(matname.Ptr());
157                 
158 }
159         
160
161 PyObject* KX_MeshProxy::PyGetTextureName(PyObject* args, PyObject* kwds)
162 {
163     int matid= 1;
164         STR_String matname;
165
166         if (PyArg_ParseTuple(args,"i:getTextureName",&matid))
167         {
168                 matname = m_meshobj->GetTextureName(matid);
169         }
170         else {
171                 return NULL;
172         }
173
174         return PyUnicode_FromString(matname.Ptr());
175                 
176 }
177
178 PyObject* KX_MeshProxy::PyGetVertexArrayLength(PyObject* args, PyObject* kwds)
179 {
180     int matid= 0;
181         int length = 0;
182
183         
184         if (!PyArg_ParseTuple(args,"i:getVertexArrayLength",&matid))
185                 return NULL;
186         
187
188         RAS_MeshMaterial *mmat = m_meshobj->GetMeshMaterial(matid); /* can be NULL*/
189         
190         if (mmat)
191         {
192                 RAS_IPolyMaterial* mat = mmat->m_bucket->GetPolyMaterial();
193                 if (mat)
194                         length = m_meshobj->NumVertices(mat);
195         }
196         
197         return PyLong_FromSsize_t(length);
198 }
199
200
201 PyObject* KX_MeshProxy::PyGetVertex(PyObject* args, PyObject* kwds)
202 {
203     int vertexindex;
204         int matindex;
205
206         if (!PyArg_ParseTuple(args,"ii:getVertex",&matindex,&vertexindex))
207                 return NULL;
208         
209         RAS_TexVert* vertex = m_meshobj->GetVertex(matindex,vertexindex);
210         
211         if(vertex==NULL) {
212                 PyErr_SetString(PyExc_ValueError, "mesh.getVertex(mat_idx, vert_idx): KX_MeshProxy, could not get a vertex at the given indicies");
213                 return NULL;
214         }
215         
216         return (new KX_VertexProxy(this, vertex))->NewProxy(true);
217 }
218
219 PyObject* KX_MeshProxy::PyGetPolygon(PyObject* args, PyObject* kwds)
220 {
221     int polyindex= 1;
222         PyObject* polyob = NULL;
223
224         if (!PyArg_ParseTuple(args,"i:getPolygon",&polyindex))
225                 return NULL;
226         
227         if (polyindex<0 || polyindex >= m_meshobj->NumPolygons())
228         {
229                 PyErr_SetString(PyExc_AttributeError, "mesh.getPolygon(int): KX_MeshProxy, invalid polygon index");
230                 return NULL;
231         }
232                 
233
234         RAS_Polygon* polygon = m_meshobj->GetPolygon(polyindex);
235         if (polygon)
236         {
237                 polyob = (new KX_PolyProxy(m_meshobj, polygon))->NewProxy(true);
238         }
239         else {
240                 PyErr_SetString(PyExc_AttributeError, "mesh.getPolygon(int): KX_MeshProxy, polygon is NULL, unknown reason");
241         }
242         return polyob;
243 }
244
245 PyObject* KX_MeshProxy::pyattr_get_materials(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
246 {
247         KX_MeshProxy* self= static_cast<KX_MeshProxy*>(self_v);
248         
249         int tot= self->m_meshobj->NumMaterials();
250         int i;
251         
252         PyObject *materials = PyList_New( tot );
253         
254         list<RAS_MeshMaterial>::iterator mit= self->m_meshobj->GetFirstMaterial();
255         
256         
257         for(i=0; i<tot; mit++, i++) {
258                 RAS_IPolyMaterial *polymat = mit->m_bucket->GetPolyMaterial();   
259                 
260                 /* Why do we need to check for RAS_BLENDERMAT if both are cast to a (PyObject*)? - Campbell */
261                 if(polymat->GetFlag() & RAS_BLENDERMAT)          
262                 {        
263                         KX_BlenderMaterial *mat = static_cast<KX_BlenderMaterial*>(polymat);     
264                         PyList_SET_ITEM(materials, i, mat->GetProxy());
265                 }
266                 else {  
267                         KX_PolygonMaterial *mat = static_cast<KX_PolygonMaterial*>(polymat);
268                         PyList_SET_ITEM(materials, i, mat->GetProxy());
269                 }
270         }       
271         return materials;
272 }
273
274 PyObject * KX_MeshProxy::pyattr_get_numMaterials(void * selfv, const KX_PYATTRIBUTE_DEF * attrdef) {
275         KX_MeshProxy * self = static_cast<KX_MeshProxy *> (selfv);
276         return PyLong_FromSsize_t(self->m_meshobj->NumMaterials());
277 }
278
279 PyObject * KX_MeshProxy::pyattr_get_numPolygons(void * selfv, const KX_PYATTRIBUTE_DEF * attrdef) {
280         KX_MeshProxy * self = static_cast<KX_MeshProxy *> (selfv);
281         return PyLong_FromSsize_t(self->m_meshobj->NumPolygons());
282 }
283
284 /* a close copy of ConvertPythonToGameObject but for meshes */
285 bool ConvertPythonToMesh(PyObject * value, RAS_MeshObject **object, bool py_none_ok, const char *error_prefix)
286 {
287         if (value==NULL) {
288                 PyErr_Format(PyExc_TypeError, "%s, python pointer NULL, should never happen", error_prefix);
289                 *object = NULL;
290                 return false;
291         }
292                 
293         if (value==Py_None) {
294                 *object = NULL;
295                 
296                 if (py_none_ok) {
297                         return true;
298                 } else {
299                         PyErr_Format(PyExc_TypeError, "%s, expected KX_MeshProxy or a KX_MeshProxy name, None is invalid", error_prefix);
300                         return false;
301                 }
302         }
303         
304         if (PyUnicode_Check(value)) {
305                 *object = (RAS_MeshObject*)SCA_ILogicBrick::m_sCurrentLogicManager->GetMeshByName(STR_String( _PyUnicode_AsString(value) ));
306                 
307                 if (*object) {
308                         return true;
309                 } else {
310                         PyErr_Format(PyExc_ValueError, "%s, requested name \"%s\" did not match any KX_MeshProxy in this scene", error_prefix, _PyUnicode_AsString(value));
311                         return false;
312                 }
313         }
314         
315         if (PyObject_TypeCheck(value, &KX_MeshProxy::Type)) {
316                 KX_MeshProxy *kx_mesh = static_cast<KX_MeshProxy*>BGE_PROXY_REF(value);
317                 
318                 /* sets the error */
319                 if (kx_mesh==NULL) {
320                         PyErr_Format(PyExc_SystemError, "%s, " BGE_PROXY_ERROR_MSG, error_prefix);
321                         return false;
322                 }               
323                 
324                 *object = kx_mesh->GetMesh();
325                 return true;
326         }
327         
328         *object = NULL;
329         
330         if (py_none_ok) {
331                 PyErr_Format(PyExc_TypeError, "%s, expect a KX_MeshProxy, a string or None", error_prefix);
332         } else {
333                 PyErr_Format(PyExc_TypeError, "%s, expect a KX_MeshProxy or a string", error_prefix);
334         }
335         
336         return false;
337 }