KX_PolyProxy.cpp

Go to the documentation of this file.

/*
 * $Id: KX_PolyProxy.cpp 35171 2011-02-25 13:35:59Z jesterking $
 * ***** BEGIN GPL 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.
 *
 * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, 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 LICENSE BLOCK *****
 */

#ifdef WITH_PYTHON

#include "KX_PolyProxy.h"
#include "KX_MeshProxy.h"
#include "RAS_MeshObject.h"
#include "KX_BlenderMaterial.h"
#include "KX_PolygonMaterial.h"

#include "KX_PyMath.h"

PyTypeObject KX_PolyProxy::Type = {
        PyVarObject_HEAD_INIT(NULL, 0)
        "KX_PolyProxy",
        sizeof(PyObjectPlus_Proxy),
        0,
        py_base_dealloc,
        0,
        0,
        0,
        0,
        py_base_repr,
        0,0,0,0,0,0,0,0,0,
        Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE,
        0,0,0,0,0,0,0,
        Methods,
        0,
        0,
        &CValue::Type,
        0,0,0,0,0,0,
        py_base_new
};

PyMethodDef KX_PolyProxy::Methods[] = {
        KX_PYMETHODTABLE_NOARGS(KX_PolyProxy,getMaterialIndex),
        KX_PYMETHODTABLE_NOARGS(KX_PolyProxy,getNumVertex),
        KX_PYMETHODTABLE_NOARGS(KX_PolyProxy,isVisible),
        KX_PYMETHODTABLE_NOARGS(KX_PolyProxy,isCollider),
        KX_PYMETHODTABLE_NOARGS(KX_PolyProxy,getMaterialName),
        KX_PYMETHODTABLE_NOARGS(KX_PolyProxy,getTextureName),
        KX_PYMETHODTABLE(KX_PolyProxy,getVertexIndex),
        KX_PYMETHODTABLE_NOARGS(KX_PolyProxy,getMesh),
        KX_PYMETHODTABLE_NOARGS(KX_PolyProxy,getMaterial),
        {NULL,NULL} //Sentinel
};

PyAttributeDef KX_PolyProxy::Attributes[] = {
        KX_PYATTRIBUTE_RO_FUNCTION("material_name", KX_PolyProxy, pyattr_get_material_name),
        KX_PYATTRIBUTE_RO_FUNCTION("texture_name", KX_PolyProxy, pyattr_get_texture_name),
        KX_PYATTRIBUTE_RO_FUNCTION("material", KX_PolyProxy, pyattr_get_material),
        KX_PYATTRIBUTE_RO_FUNCTION("material_id", KX_PolyProxy, pyattr_get_material_id),
        KX_PYATTRIBUTE_RO_FUNCTION("v1", KX_PolyProxy, pyattr_get_v1),
        KX_PYATTRIBUTE_RO_FUNCTION("v2", KX_PolyProxy, pyattr_get_v2),
        KX_PYATTRIBUTE_RO_FUNCTION("v3", KX_PolyProxy, pyattr_get_v3),
        KX_PYATTRIBUTE_RO_FUNCTION("v4", KX_PolyProxy, pyattr_get_v4),
        KX_PYATTRIBUTE_RO_FUNCTION("visible", KX_PolyProxy, pyattr_get_visible),
        KX_PYATTRIBUTE_RO_FUNCTION("collide", KX_PolyProxy, pyattr_get_collide),
        { NULL }        //Sentinel
};

KX_PolyProxy::KX_PolyProxy(const RAS_MeshObject*mesh, RAS_Polygon* polygon)
:       m_polygon(polygon),
        m_mesh((RAS_MeshObject*)mesh)
{
}

KX_PolyProxy::~KX_PolyProxy()
{
}


// stuff for cvalue related things
CValue*         KX_PolyProxy::Calc(VALUE_OPERATOR, CValue *) { return NULL;}
CValue*         KX_PolyProxy::CalcFinal(VALUE_DATA_TYPE, VALUE_OPERATOR, CValue *) { return NULL;}      
STR_String      sPolyName="polygone";
const STR_String &      KX_PolyProxy::GetText() {return sPolyName;};
double          KX_PolyProxy::GetNumber() { return -1;}
STR_String&     KX_PolyProxy::GetName() { return sPolyName;}
void            KX_PolyProxy::SetName(const char *) { };
CValue*         KX_PolyProxy::GetReplica() { return NULL;}

// stuff for python integration

PyObject* KX_PolyProxy::pyattr_get_material_name(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);
        return self->PygetMaterialName();
}

PyObject* KX_PolyProxy::pyattr_get_texture_name(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);
        return self->PygetTextureName();
}

PyObject* KX_PolyProxy::pyattr_get_material(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);
        return self->PygetMaterial();
}

PyObject* KX_PolyProxy::pyattr_get_material_id(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);
        return self->PygetMaterialIndex();
}

PyObject* KX_PolyProxy::pyattr_get_v1(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);

        return PyLong_FromSsize_t(self->m_polygon->GetVertexOffsetAbs(self->m_mesh, 0));
}

PyObject* KX_PolyProxy::pyattr_get_v2(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);

        return PyLong_FromSsize_t(self->m_polygon->GetVertexOffsetAbs(self->m_mesh, 1));
}

PyObject* KX_PolyProxy::pyattr_get_v3(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);

        return PyLong_FromSsize_t(self->m_polygon->GetVertexOffsetAbs(self->m_mesh, 2));
}

PyObject* KX_PolyProxy::pyattr_get_v4(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);

        if (3 < self->m_polygon->VertexCount())
        {
                return PyLong_FromSsize_t(self->m_polygon->GetVertexOffsetAbs(self->m_mesh, 3));
        }
        return PyLong_FromSsize_t(0);
}

PyObject* KX_PolyProxy::pyattr_get_visible(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);
        return self->PyisVisible();
}

PyObject* KX_PolyProxy::pyattr_get_collide(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef)
{
        KX_PolyProxy* self= static_cast<KX_PolyProxy*>(self_v);
        return self->PyisCollider();
}

KX_PYMETHODDEF_DOC_NOARGS(KX_PolyProxy, getMaterialIndex,
"getMaterialIndex() : return the material index of the polygon in the mesh\n")
{
        RAS_MaterialBucket* polyBucket = m_polygon->GetMaterial();
        unsigned int matid;
        for (matid=0; matid<(unsigned int)m_mesh->NumMaterials(); matid++)
        {
                RAS_MeshMaterial* meshMat = m_mesh->GetMeshMaterial(matid);
                if (meshMat->m_bucket == polyBucket)
                        // found it
                        break;
        }
        return PyLong_FromSsize_t(matid);
}

KX_PYMETHODDEF_DOC_NOARGS(KX_PolyProxy, getNumVertex,
"getNumVertex() : returns the number of vertex of the polygon, 3 or 4\n")
{
        return PyLong_FromSsize_t(m_polygon->VertexCount());
}

KX_PYMETHODDEF_DOC_NOARGS(KX_PolyProxy, isVisible,
"isVisible() : returns whether the polygon is visible or not\n")
{
        return PyLong_FromSsize_t(m_polygon->IsVisible());
}

KX_PYMETHODDEF_DOC_NOARGS(KX_PolyProxy, isCollider,
"isCollider() : returns whether the polygon is receives collision or not\n")
{
        return PyLong_FromSsize_t(m_polygon->IsCollider());
}

KX_PYMETHODDEF_DOC_NOARGS(KX_PolyProxy, getMaterialName,
"getMaterialName() : returns the polygon material name, \"NoMaterial\" if no material\n")
{
        return PyUnicode_FromString(m_polygon->GetMaterial()->GetPolyMaterial()->GetMaterialName());
}

KX_PYMETHODDEF_DOC_NOARGS(KX_PolyProxy, getTextureName,
"getTexturelName() : returns the polygon texture name, \"NULL\" if no texture\n")
{
        return PyUnicode_FromString(m_polygon->GetMaterial()->GetPolyMaterial()->GetTextureName());
}

KX_PYMETHODDEF_DOC(KX_PolyProxy, getVertexIndex,
"getVertexIndex(vertex) : returns the mesh vertex index of a polygon vertex\n"
"vertex: index of the vertex in the polygon: 0->3\n"
"return value can be used to retrieve the vertex details through mesh proxy\n"
"Note: getVertexIndex(3) on a triangle polygon returns 0\n")
{
        int index;
        if (!PyArg_ParseTuple(args,"i:getVertexIndex",&index))
        {
                return NULL;
        }
        if (index < 0 || index > 3)
        {
                PyErr_SetString(PyExc_AttributeError, "poly.getVertexIndex(int): KX_PolyProxy, expected an index between 0-3");
                return NULL;
        }
        if (index < m_polygon->VertexCount())
        {
                return PyLong_FromSsize_t(m_polygon->GetVertexOffsetAbs(m_mesh, index));
        }
        return PyLong_FromSsize_t(0);
}

KX_PYMETHODDEF_DOC_NOARGS(KX_PolyProxy, getMesh,
"getMesh() : returns a mesh proxy\n")
{
        KX_MeshProxy* meshproxy = new KX_MeshProxy((RAS_MeshObject*)m_mesh);
        return meshproxy->NewProxy(true);
}

KX_PYMETHODDEF_DOC_NOARGS(KX_PolyProxy, getMaterial,
"getMaterial() : returns a material\n")
{
        RAS_IPolyMaterial *polymat = m_polygon->GetMaterial()->GetPolyMaterial();
        if(polymat->GetFlag() & RAS_BLENDERMAT)
        {
                KX_BlenderMaterial* mat = static_cast<KX_BlenderMaterial*>(polymat);
                return mat->GetProxy();
        }
        else
        {
                KX_PolygonMaterial* mat = static_cast<KX_PolygonMaterial*>(polymat);
                return mat->GetProxy();
        }
}

#endif // WITH_PYTHON