2 * ***** BEGIN GPL LICENSE BLOCK *****
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.
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.
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.
18 * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19 * All rights reserved.
21 * The Original Code is: all of this file.
23 * Contributor(s): none yet.
25 * ***** END GPL LICENSE BLOCK *****
28 /** \file gameengine/Ketsji/KX_MeshProxy.cpp
35 #include "KX_MeshProxy.h"
36 #include "RAS_IPolygonMaterial.h"
37 #include "RAS_MeshObject.h"
39 #include "KX_VertexProxy.h"
40 #include "KX_PolyProxy.h"
42 #include "KX_PolygonMaterial.h"
43 #include "KX_BlenderMaterial.h"
45 #include "KX_PyMath.h"
46 #include "KX_ConvertPhysicsObject.h"
48 #include "PyObjectPlus.h"
50 PyTypeObject KX_MeshProxy::Type = {
51 PyVarObject_HEAD_INIT(NULL, 0)
53 sizeof(PyObjectPlus_Proxy),
62 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
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 {"transform", (PyCFunction)KX_MeshProxy::sPyTransform,METH_VARARGS},
79 {"transform_uv", (PyCFunction)KX_MeshProxy::sPyTransformUV,METH_VARARGS},
80 //{"getIndexArrayLength", (PyCFunction)KX_MeshProxy::sPyGetIndexArrayLength,METH_VARARGS},
81 {NULL,NULL} //Sentinel
84 PyAttributeDef KX_MeshProxy::Attributes[] = {
85 KX_PYATTRIBUTE_RO_FUNCTION("materials", KX_MeshProxy, pyattr_get_materials),
86 KX_PYATTRIBUTE_RO_FUNCTION("numPolygons", KX_MeshProxy, pyattr_get_numPolygons),
87 KX_PYATTRIBUTE_RO_FUNCTION("numMaterials", KX_MeshProxy, pyattr_get_numMaterials),
92 void KX_MeshProxy::SetMeshModified(bool v)
94 m_meshobj->SetMeshModified(v);
97 KX_MeshProxy::KX_MeshProxy(RAS_MeshObject* mesh)
98 : CValue(), m_meshobj(mesh)
102 KX_MeshProxy::~KX_MeshProxy()
108 // stuff for cvalue related things
109 CValue* KX_MeshProxy::Calc(VALUE_OPERATOR op, CValue *val) { return NULL;}
110 CValue* KX_MeshProxy::CalcFinal(VALUE_DATA_TYPE dtype, VALUE_OPERATOR op, CValue *val) { return NULL;}
112 const STR_String & KX_MeshProxy::GetText() {return m_meshobj->GetName();};
113 double KX_MeshProxy::GetNumber() { return -1;}
114 STR_String& KX_MeshProxy::GetName() { return m_meshobj->GetName();}
115 void KX_MeshProxy::SetName(const char *name) { };
116 CValue* KX_MeshProxy::GetReplica() { return NULL;}
119 // stuff for python integration
121 PyObject *KX_MeshProxy::PyGetMaterialName(PyObject *args, PyObject *kwds)
126 if (PyArg_ParseTuple(args,"i:getMaterialName",&matid))
128 matname = m_meshobj->GetMaterialName(matid);
134 return PyUnicode_From_STR_String(matname);
139 PyObject *KX_MeshProxy::PyGetTextureName(PyObject *args, PyObject *kwds)
144 if (PyArg_ParseTuple(args,"i:getTextureName",&matid))
146 matname = m_meshobj->GetTextureName(matid);
152 return PyUnicode_From_STR_String(matname);
156 PyObject *KX_MeshProxy::PyGetVertexArrayLength(PyObject *args, PyObject *kwds)
162 if (!PyArg_ParseTuple(args,"i:getVertexArrayLength",&matid))
166 RAS_MeshMaterial *mmat = m_meshobj->GetMeshMaterial(matid); /* can be NULL*/
170 RAS_IPolyMaterial* mat = mmat->m_bucket->GetPolyMaterial();
172 length = m_meshobj->NumVertices(mat);
175 return PyLong_FromSsize_t(length);
179 PyObject *KX_MeshProxy::PyGetVertex(PyObject *args, PyObject *kwds)
184 if (!PyArg_ParseTuple(args,"ii:getVertex",&matindex,&vertexindex))
187 RAS_TexVert* vertex = m_meshobj->GetVertex(matindex,vertexindex);
190 PyErr_SetString(PyExc_ValueError, "mesh.getVertex(mat_idx, vert_idx): KX_MeshProxy, could not get a vertex at the given indices");
194 return (new KX_VertexProxy(this, vertex))->NewProxy(true);
197 PyObject *KX_MeshProxy::PyGetPolygon(PyObject *args, PyObject *kwds)
200 PyObject *polyob = NULL;
202 if (!PyArg_ParseTuple(args,"i:getPolygon",&polyindex))
205 if (polyindex<0 || polyindex >= m_meshobj->NumPolygons())
207 PyErr_SetString(PyExc_AttributeError, "mesh.getPolygon(int): KX_MeshProxy, invalid polygon index");
212 RAS_Polygon* polygon = m_meshobj->GetPolygon(polyindex);
215 polyob = (new KX_PolyProxy(m_meshobj, polygon))->NewProxy(true);
218 PyErr_SetString(PyExc_AttributeError, "mesh.getPolygon(int): KX_MeshProxy, polygon is NULL, unknown reason");
223 PyObject *KX_MeshProxy::PyTransform(PyObject *args, PyObject *kwds)
229 MT_Matrix4x4 transform;
231 if (!PyArg_ParseTuple(args,"iO:transform", &matindex, &pymat) ||
232 !PyMatTo(pymat, transform))
237 MT_Matrix4x4 ntransform = transform.inverse().transposed();
238 ntransform[0][3] = ntransform[1][3] = ntransform[2][3] = 0.0f;
240 /* transform mesh verts */
241 unsigned int mit_index = 0;
242 for (list<RAS_MeshMaterial>::iterator mit = m_meshobj->GetFirstMaterial();
243 (mit != m_meshobj->GetLastMaterial());
246 if (matindex == -1) {
247 /* always transform */
249 else if (matindex == mit_index) {
250 /* we found the right index! */
256 RAS_MeshSlot *slot = mit->m_baseslot;
257 RAS_MeshSlot::iterator it;
260 for (slot->begin(it); !slot->end(it); slot->next(it)) {
262 for (i = it.startvertex; i < it.endvertex; i++) {
263 it.vertex[i].Transform(transform, ntransform);
267 /* if we set a material index, quit when done */
268 if (matindex == mit_index) {
274 PyErr_Format(PyExc_ValueError,
275 "mesh.transform(...): invalid material index %d", matindex);
279 m_meshobj->SetMeshModified(true);
284 PyObject *KX_MeshProxy::PyTransformUV(PyObject *args, PyObject *kwds)
291 MT_Matrix4x4 transform;
293 if (!PyArg_ParseTuple(args,"iO|ii:transform_uv", &matindex, &pymat, &uvindex) ||
294 !PyMatTo(pymat, transform))
299 if (uvindex < -1 || uvindex > 1) {
300 PyErr_Format(PyExc_ValueError,
301 "mesh.transform_uv(...): invalid uv index %d", uvindex);
305 /* transform mesh verts */
306 unsigned int mit_index = 0;
307 for (list<RAS_MeshMaterial>::iterator mit = m_meshobj->GetFirstMaterial();
308 (mit != m_meshobj->GetLastMaterial());
311 if (matindex == -1) {
312 /* always transform */
314 else if (matindex == mit_index) {
315 /* we found the right index! */
321 RAS_MeshSlot *slot = mit->m_baseslot;
322 RAS_MeshSlot::iterator it;
325 for (slot->begin(it); !slot->end(it); slot->next(it)) {
328 for (i = it.startvertex; i < it.endvertex; i++) {
331 it.vertex[i].TransformUV(transform);
334 it.vertex[i].TransformUV2(transform);
337 it.vertex[i].TransformUV(transform);
338 it.vertex[i].TransformUV2(transform);
344 /* if we set a material index, quit when done */
345 if (matindex == mit_index) {
351 PyErr_Format(PyExc_ValueError,
352 "mesh.transform_uv(...): invalid material index %d", matindex);
356 m_meshobj->SetMeshModified(true);
361 PyObject *KX_MeshProxy::pyattr_get_materials(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
363 KX_MeshProxy* self = static_cast<KX_MeshProxy*>(self_v);
365 int tot= self->m_meshobj->NumMaterials();
368 PyObject *materials = PyList_New( tot );
370 list<RAS_MeshMaterial>::iterator mit= self->m_meshobj->GetFirstMaterial();
373 for (i=0; i<tot; mit++, i++) {
374 RAS_IPolyMaterial *polymat = mit->m_bucket->GetPolyMaterial();
376 /* Why do we need to check for RAS_BLENDERMAT if both are cast to a (PyObject *)? - Campbell */
377 if (polymat->GetFlag() & RAS_BLENDERMAT)
379 KX_BlenderMaterial *mat = static_cast<KX_BlenderMaterial*>(polymat);
380 PyList_SET_ITEM(materials, i, mat->GetProxy());
383 KX_PolygonMaterial *mat = static_cast<KX_PolygonMaterial*>(polymat);
384 PyList_SET_ITEM(materials, i, mat->GetProxy());
390 PyObject * KX_MeshProxy::pyattr_get_numMaterials(void * selfv, const KX_PYATTRIBUTE_DEF * attrdef)
392 KX_MeshProxy * self = static_cast<KX_MeshProxy *> (selfv);
393 return PyLong_FromSsize_t(self->m_meshobj->NumMaterials());
396 PyObject * KX_MeshProxy::pyattr_get_numPolygons(void * selfv, const KX_PYATTRIBUTE_DEF * attrdef)
398 KX_MeshProxy * self = static_cast<KX_MeshProxy *> (selfv);
399 return PyLong_FromSsize_t(self->m_meshobj->NumPolygons());
402 /* a close copy of ConvertPythonToGameObject but for meshes */
403 bool ConvertPythonToMesh(PyObject *value, RAS_MeshObject **object, bool py_none_ok, const char *error_prefix)
406 PyErr_Format(PyExc_TypeError, "%s, python pointer NULL, should never happen", error_prefix);
411 if (value==Py_None) {
417 PyErr_Format(PyExc_TypeError, "%s, expected KX_MeshProxy or a KX_MeshProxy name, None is invalid", error_prefix);
422 if (PyUnicode_Check(value)) {
423 *object = (RAS_MeshObject*)SCA_ILogicBrick::m_sCurrentLogicManager->GetMeshByName(STR_String( _PyUnicode_AsString(value) ));
428 PyErr_Format(PyExc_ValueError, "%s, requested name \"%s\" did not match any KX_MeshProxy in this scene", error_prefix, _PyUnicode_AsString(value));
433 if (PyObject_TypeCheck(value, &KX_MeshProxy::Type)) {
434 KX_MeshProxy *kx_mesh = static_cast<KX_MeshProxy*>BGE_PROXY_REF(value);
438 PyErr_Format(PyExc_SystemError, "%s, " BGE_PROXY_ERROR_MSG, error_prefix);
442 *object = kx_mesh->GetMesh();
449 PyErr_Format(PyExc_TypeError, "%s, expect a KX_MeshProxy, a string or None", error_prefix);
451 PyErr_Format(PyExc_TypeError, "%s, expect a KX_MeshProxy or a string", error_prefix);
457 #endif // WITH_PYTHON