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