forked from bartvdbraak/blender
d38329b5aa
Added attributes to the vertex class.
430 lines
9.4 KiB
C++
430 lines
9.4 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 *****
|
|
* 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 "MT_Point2.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 = PySequence_Size(pylist);
|
|
if (numitems <= 3)
|
|
{
|
|
for (unsigned int index=0;index<numitems;index++)
|
|
{
|
|
PyObject *item = PySequence_GetItem(pylist,index); /* new ref */
|
|
vec[index] = PyFloat_AsDouble(item);
|
|
Py_DECREF(item);
|
|
}
|
|
}
|
|
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 = PySequence_Size(pylist);
|
|
if (numitems <= 3)
|
|
{
|
|
for (unsigned int index=0;index<numitems;index++)
|
|
{
|
|
PyObject *item = PySequence_GetItem(pylist,index); /* new ref */
|
|
point[index] = PyFloat_AsDouble(item);
|
|
Py_DECREF(item);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
error = true;
|
|
}
|
|
|
|
}
|
|
if (error)
|
|
PyErr_SetString(PyExc_TypeError, "Expected list of three items for point argument.");
|
|
|
|
return point;
|
|
}
|
|
|
|
MT_Point2 MT_Point2FromPyList(PyObject* pylist)
|
|
{
|
|
MT_Point2 point(0., 0.);
|
|
bool error=false;
|
|
if (pylist->ob_type == &CListValue::Type)
|
|
{
|
|
CListValue* listval = (CListValue*) pylist;
|
|
unsigned int numitems = listval->GetCount();
|
|
if (numitems <= 2)
|
|
{
|
|
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 = PySequence_Size(pylist);
|
|
if (numitems <= 2)
|
|
{
|
|
for (unsigned int index=0;index<numitems;index++)
|
|
{
|
|
PyObject *item = PySequence_GetItem(pylist,index); /* new ref */
|
|
point[index] = PyFloat_AsDouble(item);
|
|
Py_DECREF(item);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
error = true;
|
|
}
|
|
|
|
}
|
|
if (error)
|
|
PyErr_SetString(PyExc_TypeError, "Expected list of twos 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 = PySequence_Size(pylist);
|
|
if (numitems <= 4)
|
|
{
|
|
for (unsigned index=0;index<numitems;index++)
|
|
{
|
|
PyObject *item = PySequence_GetItem(pylist,index); /* new ref */
|
|
vec[index] = PyFloat_AsDouble(item);
|
|
Py_DECREF(item);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
error = true;
|
|
}
|
|
}
|
|
if (error)
|
|
PyErr_SetString(PyExc_TypeError, "Expected list of four items for Vector argument.");
|
|
return vec;
|
|
}
|
|
|
|
MT_Quaternion MT_QuaternionFromPyList(PyObject* pylist)
|
|
{
|
|
MT_Quaternion 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 = PySequence_Size(pylist);
|
|
if (numitems <= 4)
|
|
{
|
|
for (unsigned index=0;index<numitems;index++)
|
|
{
|
|
PyObject *item = PySequence_GetItem(pylist,index); /* new ref */
|
|
vec[index] = PyFloat_AsDouble(item);
|
|
Py_DECREF(item);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
error = true;
|
|
}
|
|
}
|
|
if (error)
|
|
PyErr_SetString(PyExc_TypeError, "Expected list of four items for Quaternion 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); /* new ref */
|
|
if (PySequence_Check(pyrow))
|
|
{
|
|
unsigned int cols = PySequence_Size(pyrow);
|
|
for( unsigned int x = 0; x < cols && x < 4; x++)
|
|
{
|
|
PyObject *item = PySequence_GetItem(pyrow, x); /* new ref */
|
|
mat[y][x] = PyFloat_AsDouble(item);
|
|
Py_DECREF(item);
|
|
}
|
|
}
|
|
Py_DECREF(pyrow);
|
|
}
|
|
}
|
|
|
|
return mat;
|
|
}
|
|
|
|
bool PyObject_IsMT_Matrix(PyObject *pymat, unsigned int rank)
|
|
{
|
|
if (!pymat)
|
|
return false;
|
|
|
|
unsigned int x, y;
|
|
if (pymat->ob_type == &CListValue::Type)
|
|
{
|
|
CListValue* listval = (CListValue*) pymat;
|
|
if (listval->GetCount() == rank)
|
|
{
|
|
for (y = 0; y < rank; y++)
|
|
{
|
|
CListValue* vecval = (CListValue*)listval->GetValue(y);
|
|
if (vecval->GetCount() != rank)
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
} else if (PySequence_Check(pymat))
|
|
{
|
|
unsigned int rows = PySequence_Size(pymat);
|
|
if (rows != rank)
|
|
return false;
|
|
|
|
for (y = 0; y < rank; y++)
|
|
{
|
|
PyObject *pyrow = PySequence_GetItem(pymat, y); /* new ref */
|
|
if (PySequence_Check(pyrow))
|
|
{
|
|
if (PySequence_Size(pyrow) != rank)
|
|
{
|
|
Py_DECREF(pyrow);
|
|
return false;
|
|
}
|
|
}
|
|
Py_DECREF(pyrow);
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
MT_Matrix3x3 MT_Matrix3x3FromPyObject(PyObject *pymat)
|
|
{
|
|
MT_Matrix3x3 mat;
|
|
bool error = false;
|
|
mat.setIdentity();
|
|
if (pymat->ob_type == &CListValue::Type)
|
|
{
|
|
unsigned int row, col;
|
|
CListValue* listval = (CListValue*) pymat;
|
|
if (listval->GetCount() == 3)
|
|
{
|
|
for (row=0;row<3;row++) // each row has a 3-vector [x,y,z]
|
|
{
|
|
CListValue* vecval = (CListValue*)listval->GetValue(row);
|
|
for (col=0;col<3 && col < vecval->GetCount();col++)
|
|
{
|
|
mat[row][col] = vecval->GetValue(col)->GetNumber();
|
|
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
error = true;
|
|
if (error)
|
|
PyErr_SetString(PyExc_TypeError, "Expected list of three items for 3x3 matrix argument.");
|
|
|
|
}
|
|
} else 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); /* new ref */
|
|
if (PySequence_Check(pyrow))
|
|
{
|
|
unsigned int cols = PySequence_Size(pyrow);
|
|
for( unsigned int x = 0; x < cols && x < 3; x++)
|
|
{
|
|
PyObject *pyitem = PySequence_GetItem(pyrow, x); /* new ref */
|
|
mat[y][x] = PyFloat_AsDouble(pyitem);
|
|
Py_DECREF(pyitem);
|
|
}
|
|
}
|
|
Py_DECREF(pyrow);
|
|
}
|
|
}
|
|
|
|
return mat;
|
|
}
|
|
|
|
PyObject* PyObjectFromMT_Matrix4x4(const MT_Matrix4x4 &mat)
|
|
{
|
|
return Py_BuildValue("[[ffff][ffff][ffff][ffff]]",
|
|
mat[0][0], mat[0][1], mat[0][2], mat[0][3],
|
|
mat[1][0], mat[1][1], mat[1][2], mat[1][3],
|
|
mat[2][0], mat[2][1], mat[2][2], mat[2][3],
|
|
mat[3][0], mat[3][1], mat[3][2], mat[3][3]);
|
|
}
|
|
|
|
PyObject* PyObjectFromMT_Matrix3x3(const MT_Matrix3x3 &mat)
|
|
{
|
|
return Py_BuildValue("[[fff][fff][fff]]",
|
|
mat[0][0], mat[0][1], mat[0][2],
|
|
mat[1][0], mat[1][1], mat[1][2],
|
|
mat[2][0], mat[2][1], mat[2][2]);
|
|
}
|
|
|
|
PyObject* PyObjectFromMT_Vector4(const MT_Vector4 &vec)
|
|
{
|
|
return Py_BuildValue("[ffff]",
|
|
vec[0], vec[1], vec[2], vec[3]);
|
|
}
|
|
|
|
PyObject* PyObjectFromMT_Vector3(const MT_Vector3 &vec)
|
|
{
|
|
return Py_BuildValue("[fff]",
|
|
vec[0], vec[1], vec[2]);
|
|
}
|
|
|
|
PyObject* PyObjectFromMT_Point3(const MT_Point3 &pos)
|
|
{
|
|
return Py_BuildValue("[fff]",
|
|
pos[0], pos[1], pos[2]);
|
|
}
|
|
|
|
PyObject* PyObjectFromMT_Point2(const MT_Point2 &pos)
|
|
{
|
|
return Py_BuildValue("[ff]", pos[0], pos[1]);
|
|
}
|