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