forked from bartvdbraak/blender
162 lines
3.9 KiB
C++
162 lines
3.9 KiB
C++
/**
|
|
* $Id$
|
|
*
|
|
* ***** 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., 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 LICENSE BLOCK *****
|
|
* Initialize Python thingies.
|
|
*/
|
|
|
|
#ifndef __KX_PYMATH_H__
|
|
#define __KX_PYMATH_H__
|
|
|
|
#include "MT_Point2.h"
|
|
#include "MT_Point3.h"
|
|
#include "MT_Vector2.h"
|
|
#include "MT_Vector3.h"
|
|
#include "MT_Vector4.h"
|
|
#include "MT_Matrix3x3.h"
|
|
#include "MT_Matrix4x4.h"
|
|
|
|
#include "KX_Python.h"
|
|
|
|
inline unsigned int Size(const MT_Matrix4x4&) { return 4; }
|
|
inline unsigned int Size(const MT_Matrix3x3&) { return 3; }
|
|
inline unsigned int Size(const MT_Tuple2&) { return 2; }
|
|
inline unsigned int Size(const MT_Tuple3&) { return 3; }
|
|
inline unsigned int Size(const MT_Tuple4&) { return 4; }
|
|
|
|
/**
|
|
* Converts the given python matrix to an MT class.
|
|
*/
|
|
template<class T>
|
|
bool PyMatTo(PyObject* pymat, T& mat)
|
|
{
|
|
bool noerror = true;
|
|
mat.setIdentity();
|
|
if (PySequence_Check(pymat))
|
|
{
|
|
unsigned int rows = PySequence_Size(pymat);
|
|
if (rows != Size(mat))
|
|
return false;
|
|
|
|
for (unsigned int y = 0; noerror && y < Size(mat); y++)
|
|
{
|
|
PyObject *pyrow = PySequence_GetItem(pymat, y); /* new ref */
|
|
if (!PyErr_Occurred() && PySequence_Check(pyrow))
|
|
{
|
|
unsigned int cols = PySequence_Size(pyrow);
|
|
if (cols != Size(mat))
|
|
noerror = false;
|
|
else
|
|
{
|
|
for( unsigned int x = 0; x < Size(mat); x++)
|
|
{
|
|
PyObject *item = PySequence_GetItem(pyrow, x); /* new ref */
|
|
mat[y][x] = PyFloat_AsDouble(item);
|
|
Py_DECREF(item);
|
|
}
|
|
}
|
|
} else
|
|
noerror = false;
|
|
Py_DECREF(pyrow);
|
|
}
|
|
} else
|
|
noerror = false;
|
|
|
|
return noerror;
|
|
}
|
|
|
|
/**
|
|
* Converts a python list to a MT class.
|
|
*/
|
|
template<class T>
|
|
bool PyVecTo(PyObject* pyval, T& vec)
|
|
{
|
|
if (PySequence_Check(pyval))
|
|
{
|
|
unsigned int numitems = PySequence_Size(pyval);
|
|
if (numitems != Size(vec))
|
|
return false;
|
|
|
|
for (unsigned int x = 0; x < numitems; x++)
|
|
{
|
|
PyObject *item = PySequence_GetItem(pyval, x); /* new ref */
|
|
vec[x] = PyFloat_AsDouble(item);
|
|
Py_DECREF(item);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Converts a python argument to an MT class.
|
|
* This paramater expects arguments as passed to a python method.
|
|
*/
|
|
template<class T>
|
|
bool PyVecArgTo(PyObject* args, T& vec)
|
|
{
|
|
PyObject* pylist;
|
|
if (PyArg_ParseTuple(args,"O",&pylist))
|
|
return PyVecTo(pylist, vec);
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Converts an MT_Matrix4x4 to a python object.
|
|
*/
|
|
PyObject* PyObjectFrom(const MT_Matrix4x4 &mat);
|
|
|
|
/**
|
|
* Converts an MT_Matrix3x3 to a python object.
|
|
*/
|
|
PyObject* PyObjectFrom(const MT_Matrix3x3 &mat);
|
|
|
|
/**
|
|
* Converts an MT_Tuple2 to a python object.
|
|
*/
|
|
PyObject* PyObjectFrom(const MT_Tuple2 &vec);
|
|
|
|
/**
|
|
* Converts an MT_Tuple3 to a python object
|
|
*/
|
|
PyObject* PyObjectFrom(const MT_Tuple3 &vec);
|
|
|
|
/**
|
|
* Converts an MT_Tuple4 to a python object.
|
|
*/
|
|
PyObject* PyObjectFrom(const MT_Tuple4 &pos);
|
|
|
|
/**
|
|
* True if the given PyObject can be converted to an MT_Matrix
|
|
* @param rank = 3 (for MT_Matrix3x3) or 4 (for MT_Matrix4x4)
|
|
*/
|
|
bool PyObject_IsMT_Matrix(PyObject *pymat, unsigned int rank);
|
|
|
|
#endif
|