blender/source/gameengine/Ketsji/KX_VertexProxy.cpp
Kester Maddock c9b4585618 Switch to using floats instead of shorts for normal data - they're supposed to be faster.
Also use shorts instead of ints for the index data, since index arrays are limited anyhow.
2004-06-30 13:41:19 +00:00

366 lines
7.8 KiB
C++

/**
* $Id$
* ***** BEGIN GPL/BL DUAL LICENSE BLOCK *****
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version. The Blender
* Foundation also sells licenses for use in proprietary software under
* the Blender License. See http://www.blender.org/BL/ for information
* about this.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*
* The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
* All rights reserved.
*
* The Original Code is: all of this file.
*
* Contributor(s): none yet.
*
* ***** END GPL/BL DUAL LICENSE BLOCK *****
*/
#include "KX_VertexProxy.h"
#include "RAS_TexVert.h"
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include "KX_PyMath.h"
PyTypeObject KX_VertexProxy::Type = {
PyObject_HEAD_INIT(&PyType_Type)
0,
"KX_VertexProxy",
sizeof(KX_VertexProxy),
0,
PyDestructor,
0,
__getattr,
__setattr,
0, //&MyPyCompare,
__repr,
0, //&cvalue_as_number,
0,
0,
0,
0
};
PyParentObject KX_VertexProxy::Parents[] = {
&KX_VertexProxy::Type,
&SCA_IObject::Type,
&CValue::Type,
NULL
};
PyMethodDef KX_VertexProxy::Methods[] = {
{"getXYZ", (PyCFunction)KX_VertexProxy::sPyGetXYZ,METH_VARARGS},
{"setXYZ", (PyCFunction)KX_VertexProxy::sPySetXYZ,METH_VARARGS},
{"getUV", (PyCFunction)KX_VertexProxy::sPyGetUV,METH_VARARGS},
{"setUV", (PyCFunction)KX_VertexProxy::sPySetUV,METH_VARARGS},
{"getRGBA", (PyCFunction)KX_VertexProxy::sPyGetRGBA,METH_VARARGS},
{"setRGBA", (PyCFunction)KX_VertexProxy::sPySetRGBA,METH_VARARGS},
{"getNormal", (PyCFunction)KX_VertexProxy::sPyGetNormal,METH_VARARGS},
{"setNormal", (PyCFunction)KX_VertexProxy::sPySetNormal,METH_VARARGS},
{NULL,NULL} //Sentinel
};
PyObject*
KX_VertexProxy::_getattr(const STR_String& attr)
{
if (attr == "XYZ")
return PyObjectFromMT_Vector3(m_vertex->getLocalXYZ());
if (attr == "UV")
return PyObjectFromMT_Point2(MT_Point2(m_vertex->getUV1()));
if (attr == "colour" || attr == "color")
{
unsigned int icol = m_vertex->getRGBA();
unsigned char *colp = (unsigned char *) &icol;
MT_Vector4 colour(colp[0], colp[1], colp[2], colp[3]);
colour /= 255.0;
return PyObjectFromMT_Vector4(colour);
}
if (attr == "normal")
{
return PyObjectFromMT_Vector3(m_vertex->getNormal());
}
// pos
if (attr == "x")
return PyFloat_FromDouble(m_vertex->getLocalXYZ()[0]);
if (attr == "y")
return PyFloat_FromDouble(m_vertex->getLocalXYZ()[1]);
if (attr == "z")
return PyFloat_FromDouble(m_vertex->getLocalXYZ()[2]);
// Col
if (attr == "r")
return PyFloat_FromDouble(((unsigned char*)m_vertex->getRGBA())[0]/255.0);
if (attr == "g")
return PyFloat_FromDouble(((unsigned char*)m_vertex->getRGBA())[1]/255.0);
if (attr == "b")
return PyFloat_FromDouble(((unsigned char*)m_vertex->getRGBA())[2]/255.0);
if (attr == "a")
return PyFloat_FromDouble(((unsigned char*)m_vertex->getRGBA())[3]/255.0);
// UV
if (attr == "u")
return PyFloat_FromDouble(m_vertex->getUV1()[0]);
if (attr == "v")
return PyFloat_FromDouble(m_vertex->getUV1()[1]);
_getattr_up(SCA_IObject);
}
int KX_VertexProxy::_setattr(const STR_String& attr, PyObject *pyvalue)
{
if (PySequence_Check(pyvalue))
{
if (attr == "XYZ")
{
m_vertex->SetXYZ(MT_Point3FromPyList(pyvalue));
return 0;
}
if (attr == "UV")
{
m_vertex->SetUV(MT_Point2FromPyList(pyvalue));
return 0;
}
if (attr == "colour" || attr == "color")
{
m_vertex->SetRGBA(MT_Vector4FromPyList(pyvalue));
return 0;
}
if (attr == "normal")
{
m_vertex->SetNormal(MT_Vector3FromPyList(pyvalue));
return 0;
}
}
if (PyFloat_Check(pyvalue))
{
float val = PyFloat_AsDouble(pyvalue);
// pos
MT_Point3 pos(m_vertex->getLocalXYZ());
if (attr == "x")
{
pos.x() = val;
m_vertex->SetXYZ(pos);
return 0;
}
if (attr == "y")
{
pos.y() = val;
m_vertex->SetXYZ(pos);
return 0;
}
if (attr == "z")
{
pos.z() = val;
m_vertex->SetXYZ(pos);
return 0;
}
// uv
MT_Point2 uv = m_vertex->getUV1();
if (attr == "u")
{
uv[0] = val;
m_vertex->SetUV(uv);
return 0;
}
if (attr == "v")
{
uv[1] = val;
m_vertex->SetUV(uv);
return 0;
}
// col
unsigned int icol = m_vertex->getRGBA();
unsigned char *cp = (unsigned char*) &icol;
val *= 255.0;
if (attr == "r")
{
cp[0] = (unsigned char) val;
m_vertex->SetRGBA(icol);
return 0;
}
if (attr == "g")
{
cp[1] = (unsigned char) val;
m_vertex->SetRGBA(icol);
return 0;
}
if (attr == "b")
{
cp[2] = (unsigned char) val;
m_vertex->SetRGBA(icol);
return 0;
}
if (attr == "a")
{
cp[3] = (unsigned char) val;
m_vertex->SetRGBA(icol);
return 0;
}
}
return SCA_IObject::_setattr(attr, pyvalue);
}
KX_VertexProxy::KX_VertexProxy(RAS_TexVert* vertex)
:m_vertex(vertex)
{
}
KX_VertexProxy::~KX_VertexProxy()
{
}
// stuff for cvalue related things
CValue* KX_VertexProxy::Calc(VALUE_OPERATOR op, CValue *val) { return NULL;}
CValue* KX_VertexProxy::CalcFinal(VALUE_DATA_TYPE dtype, VALUE_OPERATOR op, CValue *val) { return NULL;}
STR_String sVertexName="vertex";
const STR_String & KX_VertexProxy::GetText() {return sVertexName;};
float KX_VertexProxy::GetNumber() { return -1;}
STR_String KX_VertexProxy::GetName() { return sVertexName;}
void KX_VertexProxy::SetName(STR_String name) { };
CValue* KX_VertexProxy::GetReplica() { return NULL;}
void KX_VertexProxy::ReplicaSetName(STR_String name) {};
// stuff for python integration
PyObject* KX_VertexProxy::PyGetXYZ(PyObject* self,
PyObject* args,
PyObject* kwds)
{
MT_Point3 pos = m_vertex->getLocalXYZ();
PyObject* resultlist = PyList_New(3);
int index;
for (index=0;index<3;index++)
{
PyList_SetItem(resultlist,index,PyFloat_FromDouble(pos[index]));
}
return resultlist;
}
PyObject* KX_VertexProxy::PySetXYZ(PyObject* self,
PyObject* args,
PyObject* kwds)
{
MT_Point3 pos = ConvertPythonVectorArg(args);
m_vertex->SetXYZ(pos);
Py_Return;
}
PyObject* KX_VertexProxy::PyGetNormal(PyObject* self,
PyObject* args,
PyObject* kwds)
{
MT_Vector3 normal(m_vertex->getNormal());
PyObject* resultlist = PyList_New(3);
int index;
for (index=0;index<3;index++)
{
PyList_SetItem(resultlist,index,PyFloat_FromDouble(normal[index]));
}
return resultlist;
}
PyObject* KX_VertexProxy::PySetNormal(PyObject* self,
PyObject* args,
PyObject* kwds)
{
MT_Point3 normal = ConvertPythonVectorArg(args);
m_vertex->SetNormal(normal);
Py_Return;
}
PyObject* KX_VertexProxy::PyGetRGBA(PyObject* self,
PyObject* args,
PyObject* kwds)
{
int rgba = m_vertex->getRGBA();
return PyInt_FromLong(rgba);
}
PyObject* KX_VertexProxy::PySetRGBA(PyObject* self,
PyObject* args,
PyObject* kwds)
{
int rgba;
if (PyArg_ParseTuple(args,"i",&rgba))
{
m_vertex->SetRGBA(rgba);
}
Py_Return;
}
PyObject* KX_VertexProxy::PyGetUV(PyObject* self,
PyObject* args,
PyObject* kwds)
{
MT_Vector2 uv = m_vertex->getUV1();
PyObject* resultlist = PyList_New(2);
int index;
for (index=0;index<2;index++)
{
PyList_SetItem(resultlist,index,PyFloat_FromDouble(uv[index]));
}
return resultlist;
}
PyObject* KX_VertexProxy::PySetUV(PyObject* self,
PyObject* args,
PyObject* kwds)
{
MT_Point3 uv = ConvertPythonVectorArg(args);
m_vertex->SetUV(MT_Point2(uv[0],uv[1]));
Py_Return;
}