griefith/test
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Python utils for Python <-> MT_Vector, MT_Point & MT_Matrix

Browse Source
This commit is contained in:
Kester Maddock
2004-05-16 12:53:54 +00:00
parent 0a27ae972e
commit 3e6b05f607
3 changed files with 321 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

238
source/gameengine/Ketsji/KX_PyMath.cpp Normal file
View File

@@ -0,0 +1,238 @@
/**
* $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 *****
* Initialize Python thingies.
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#ifdef WIN32
#pragma warning (disable : 4786)
#endif //WIN32
#include "MT_Vector3.h"
#include "MT_Vector4.h"
#include "MT_Matrix4x4.h"
#include "ListValue.h"
#include "KX_Python.h"
MT_Vector3 MT_Vector3FromPyList(PyObject* pylist)
{
MT_Vector3 vec(0., 0., 0.);
bool error=false;
if (pylist->ob_type == &CListValue::Type)
{
CListValue* listval = (CListValue*) pylist;
unsigned int numitems = listval->GetCount();
if (numitems <= 3)
{
for (unsigned int index=0;index<numitems;index++)
{
vec[index] = listval->GetValue(index)->GetNumber();
}
} else
{
error = true;
}
} else
{
// assert the list is long enough...
unsigned int numitems = PyList_Size(pylist);
if (numitems <= 3)
{
for (unsigned int index=0;index<numitems;index++)
{
vec[index] = PyFloat_AsDouble(PyList_GetItem(pylist,index));
}
}
else
{
error = true;
}
}
if (error)
PyErr_SetString(PyExc_TypeError, "Expected list of three items for vector argument.");
return vec;
}
MT_Point3 MT_Point3FromPyList(PyObject* pylist)
{
MT_Point3 point(0., 0., 0.);
bool error=false;
if (pylist->ob_type == &CListValue::Type)
{
CListValue* listval = (CListValue*) pylist;
unsigned int numitems = listval->GetCount();
if (numitems <= 3)
{
for (unsigned int index=0;index<numitems;index++)
{
point[index] = listval->GetValue(index)->GetNumber();
}
} else
{
error = true;
}
} else
{
// assert the list is long enough...
unsigned int numitems = PyList_Size(pylist);
if (numitems <= 3)
{
for (unsigned int index=0;index<numitems;index++)
{
point[index] = PyFloat_AsDouble(PyList_GetItem(pylist,index));
}
}
else
{
error = true;
}
}
if (error)
PyErr_SetString(PyExc_TypeError, "Expected list of three items for point argument.");
return point;
}
MT_Vector4 MT_Vector4FromPyList(PyObject* pylist)
{
MT_Vector4 vec(0., 0., 0., 1.);
bool error=false;
if (pylist->ob_type == &CListValue::Type)
{
CListValue* listval = (CListValue*) pylist;
unsigned int numitems = listval->GetCount();
if (numitems <= 4)
{
for (unsigned index=0;index<numitems;index++)
{
vec[index] = listval->GetValue(index)->GetNumber();
}
} else
{
error = true;
}
} else
{
// assert the list is long enough...
unsigned int numitems = PyList_Size(pylist);
if (numitems <= 4)
{
for (unsigned index=0;index<numitems;index++)
{
vec[index] = PyFloat_AsDouble(PyList_GetItem(pylist,index));
}
}
else
{
error = true;
}
}
if (error)
PyErr_SetString(PyExc_TypeError, "Expected list of four items for Vector argument.");
return vec;
}
MT_Matrix4x4 MT_Matrix4x4FromPyObject(PyObject *pymat)
{
MT_Matrix4x4 mat;
bool error = false;
mat.setIdentity();
if (PySequence_Check(pymat))
{
unsigned int rows = PySequence_Size(pymat);
for (unsigned int y = 0; y < rows && y < 4; y++)
{
PyObject *pyrow = PySequence_GetItem(pymat, y);
if (PySequence_Check(pyrow))
{
unsigned int cols = PySequence_Size(pyrow);
for( unsigned int x = 0; x < cols && x < 4; x++)
{
mat[y][x] = PyFloat_AsDouble(PySequence_GetItem(pyrow, x));
}
}
}
}
return mat;
}
MT_Matrix3x3 MT_Matrix3x3FromPyObject(PyObject *pymat)
{
MT_Matrix3x3 mat;
bool error = false;
mat.setIdentity();
if (PySequence_Check(pymat))
{
unsigned int rows = PySequence_Size(pymat);
for (unsigned int y = 0; y < rows && y < 3; y++)
{
PyObject *pyrow = PySequence_GetItem(pymat, y);
if (PySequence_Check(pyrow))
{
unsigned int cols = PySequence_Size(pyrow);
for( unsigned int x = 0; x < cols && x < 3; x++)
{
mat[y][x] = PyFloat_AsDouble(PySequence_GetItem(pyrow, x));
}
}
}
}
return mat;
}
PyObject* PyObjectFromMT_Matrix4x4(const MT_Matrix4x4 &mat)
{
PyObject *pymat = PyList_New(0);
for (unsigned int y = 0; y < 4; y++)
{
PyObject *row = PyList_New(0);
for( unsigned int x = 0; x < 4; x++ )
{
PyList_Append(row, PyFloat_FromDouble(mat[y][x]));
}
PyList_Append(pymat, row);
}
return pymat;
}