diff options
author | Tavian Barnes <tavianator@gmail.com> | 2011-06-13 00:16:27 -0600 |
---|---|---|
committer | Tavian Barnes <tavianator@gmail.com> | 2011-06-13 00:17:16 -0600 |
commit | 9b758508df283a533a4cfc605545a35f77bc9d5f (patch) | |
tree | dd7e471d07a32bff8702aed3624a084c9c209ccf /libdimension-python/Matrix.c | |
parent | 7acd8ea6673b7a90ed4041408ccf1b024b8a007a (diff) | |
download | dimension-9b758508df283a533a4cfc605545a35f77bc9d5f.tar.xz |
Use Cython for the Python module.
Diffstat (limited to 'libdimension-python/Matrix.c')
-rw-r--r-- | libdimension-python/Matrix.c | 292 |
1 files changed, 0 insertions, 292 deletions
diff --git a/libdimension-python/Matrix.c b/libdimension-python/Matrix.c deleted file mode 100644 index 0219965..0000000 --- a/libdimension-python/Matrix.c +++ /dev/null @@ -1,292 +0,0 @@ -/************************************************************************* - * Copyright (C) 2009-2011 Tavian Barnes <tavianator@tavianator.com> * - * * - * This file is part of The Dimension Python Module. * - * * - * The Dimension Python Module is free software; you can redistribute it * - * and/ or modify it under the terms of the GNU Lesser General Public * - * License as published by the Free Software Foundation; either version * - * 3 of the License, or (at your option) any later version. * - * * - * The Dimension Python Module 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 Lesser General Public License for more details. * - * * - * You should have received a copy of the GNU Lesser General Public * - * License along with this program. If not, see * - * <http://www.gnu.org/licenses/>. * - *************************************************************************/ - -#include "dimension-python.h" - -static int -dmnsn_py_Matrix_init(dmnsn_py_Matrix *self, PyObject *args, PyObject *kwds) -{ - if (kwds) { - PyErr_SetString(PyExc_TypeError, "Keyword arguments not supported"); - return -1; - } - - if (!PyArg_ParseTuple(args, "dddddddddddd", - &self->m.n[0][0], &self->m.n[0][1], &self->m.n[0][2], - &self->m.n[0][3], - &self->m.n[1][0], &self->m.n[1][1], &self->m.n[1][2], - &self->m.n[1][3], - &self->m.n[2][0], &self->m.n[2][1], &self->m.n[2][2], - &self->m.n[2][3])) - return -1; - - return 0; -} - -PyObject * -dmnsn_py_Matrix_scale(PyObject *self, PyObject *args) -{ - dmnsn_vector scale; - if (!PyArg_ParseTuple(args, "O&", dmnsn_py_VectorParse, &scale)) - return NULL; - - dmnsn_py_Matrix *ret = PyObject_New(dmnsn_py_Matrix, &dmnsn_py_MatrixType); - if (ret) { - ret->m = dmnsn_scale_matrix(scale); - } - return (PyObject *)ret; -} - -PyObject * -dmnsn_py_Matrix_translate(PyObject *self, PyObject *args) -{ - dmnsn_vector translate; - if (!PyArg_ParseTuple(args, "O&", dmnsn_py_VectorParse, &translate)) - return NULL; - - dmnsn_py_Matrix *ret = PyObject_New(dmnsn_py_Matrix, &dmnsn_py_MatrixType); - if (ret) { - ret->m = dmnsn_translation_matrix(translate); - } - return (PyObject *)ret; -} - -PyObject * -dmnsn_py_Matrix_rotate(PyObject *self, PyObject *args) -{ - dmnsn_vector rotate; - if (!PyArg_ParseTuple(args, "O&", dmnsn_py_VectorParse, &rotate)) - return NULL; - - dmnsn_py_Matrix *ret = PyObject_New(dmnsn_py_Matrix, &dmnsn_py_MatrixType); - if (ret) { - ret->m = dmnsn_rotation_matrix(dmnsn_vector_mul(dmnsn_radians(1.0), - rotate)); - } - return (PyObject *)ret; -} - -static PyObject * -dmnsn_py_Matrix_repr(dmnsn_py_Matrix *self) -{ - PyObject *floats[3][4] = { { NULL } }; - bool initialized = true; - for (size_t i = 0; i < 3; ++i) { - for (size_t j = 0; j < 4; ++j) { - floats[i][j] = PyFloat_FromDouble(self->m.n[i][j]); - if (!floats[i][j]) { - initialized = false; - break; - } - } - if (!initialized) - break; - } - - if (!initialized) { - for (size_t i = 0; i < 3; ++i) { - for (size_t j = 0; j < 4; ++j) { - Py_XDECREF(floats[i][j]); - } - } - return NULL; - } - - PyObject *repr = PyUnicode_FromFormat("dimension.Matrix(%R, %R, %R, %R, " - "%R, %R, %R, %R, " - "%R, %R, %R, %R)", - floats[0][0], floats[0][1], - floats[0][2], floats[0][3], - floats[1][0], floats[1][1], - floats[1][2], floats[1][3], - floats[2][0], floats[2][1], - floats[2][2], floats[2][3]); - for (size_t i = 0; i < 3; ++i) { - for (size_t j = 0; j < 4; ++j) { - Py_DECREF(floats[i][j]); - } - } - return repr; -} - -static PyObject * -dmnsn_py_Matrix_str(dmnsn_py_Matrix *self) -{ - PyObject *floats[3][4] = { { NULL } }; - bool initialized = true; - for (size_t i = 0; i < 3; ++i) { - for (size_t j = 0; j < 4; ++j) { - floats[i][j] = PyFloat_FromDouble(self->m.n[i][j]); - if (!floats[i][j]) { - initialized = false; - break; - } - } - if (!initialized) - break; - } - - if (!initialized) { - for (size_t i = 0; i < 3; ++i) { - for (size_t j = 0; j < 4; ++j) { - Py_XDECREF(floats[i][j]); - } - } - return NULL; - } - - PyObject *str = PyUnicode_FromFormat("\n[%S\t%S\t%S\t%S]\n" - "[%S\t%S\t%S\t%S]\n" - "[%S\t%S\t%S\t%S]\n" - "[0.0\t0.0\t0.0\t1.0]", - floats[0][0], floats[0][1], - floats[0][2], floats[0][3], - floats[1][0], floats[1][1], - floats[1][2], floats[1][3], - floats[2][0], floats[2][1], - floats[2][2], floats[2][3]); - for (size_t i = 0; i < 3; ++i) { - for (size_t j = 0; j < 4; ++j) { - Py_DECREF(floats[i][j]); - } - } - return str; -} - -static PyObject * -dmnsn_py_Matrix_richcompare(PyObject *lhs, PyObject *rhs, int op) -{ - if (!PyObject_TypeCheck(lhs, &dmnsn_py_MatrixType) - || !PyObject_TypeCheck(rhs, &dmnsn_py_MatrixType)) - { - PyErr_SetString(PyExc_TypeError, - "Matricies can only be compared with Matricies"); - return NULL; - } - - dmnsn_py_Matrix *mlhs = (dmnsn_py_Matrix *)lhs; - dmnsn_py_Matrix *mrhs = (dmnsn_py_Matrix *)rhs; - - double sqdiff = 0.0; - for (size_t i = 0; i < 3; ++i) { - for (size_t j = 0; j < 4; ++j) { - double diff = mlhs->m.n[i][j] - mrhs->m.n[i][j]; - sqdiff += diff*diff; - } - } - - bool equal = sqrt(sqdiff) < dmnsn_epsilon; - - PyObject *result; - switch (op) { - case Py_EQ: - result = equal ? Py_True : Py_False; - break; - case Py_NE: - result = !equal ? Py_True : Py_False; - break; - default: - result = Py_NotImplemented; - break; - } - - Py_INCREF(result); - return result; -} - -static PyObject * -dmnsn_py_Matrix_mul(PyObject *lhs, PyObject *rhs) -{ - if (!PyObject_TypeCheck(lhs, &dmnsn_py_MatrixType)) { - PyErr_SetString(PyExc_TypeError, - "Left-hand side of matrix multiplication must be a matrix"); - return NULL; - } - - dmnsn_py_Matrix *mlhs = (dmnsn_py_Matrix *)lhs; - - if (PyObject_TypeCheck(rhs, &dmnsn_py_MatrixType)) { - dmnsn_py_Matrix *mrhs = (dmnsn_py_Matrix *)rhs; - dmnsn_py_Matrix *ret = PyObject_New(dmnsn_py_Matrix, &dmnsn_py_MatrixType); - if (ret) { - ret->m = dmnsn_matrix_mul(mlhs->m, mrhs->m); - } - return (PyObject *)ret; - } else if (PyObject_TypeCheck(rhs, &dmnsn_py_VectorType)) { - dmnsn_py_Vector *vrhs = (dmnsn_py_Vector *)rhs; - dmnsn_py_Vector *ret = PyObject_New(dmnsn_py_Vector, &dmnsn_py_VectorType); - if (ret) { - ret->v = dmnsn_transform_vector(mlhs->m, vrhs->v); - } - return (PyObject *)ret; - } else { - PyErr_SetString(PyExc_TypeError, - "Right-hand side of matrix multiplication must be a matrix" - " or vector"); - return NULL; - } -} - -static PyNumberMethods dmnsn_py_Matrix_as_number = { - .nb_multiply = dmnsn_py_Matrix_mul, -}; - -static PyObject * -dmnsn_py_Matrix_inverse(dmnsn_py_Matrix *self) -{ - dmnsn_py_Matrix *ret = PyObject_New(dmnsn_py_Matrix, &dmnsn_py_MatrixType); - if (ret) { - ret->m = dmnsn_matrix_inverse(self->m); - } - return (PyObject *)ret; -} - -static PyMethodDef dmnsn_py_Matrix_methods[] = { - { "inverse", (PyCFunction)dmnsn_py_Matrix_inverse, METH_NOARGS, - "Return the inverse of the matrix" }, - { NULL } -}; - -static PyGetSetDef dmnsn_py_Matrix_getsetters[] = { - { NULL } -}; - -PyTypeObject dmnsn_py_MatrixType = { - PyVarObject_HEAD_INIT(NULL, 0) - .tp_name = "dimension.Matrix", - .tp_basicsize = sizeof(dmnsn_py_Matrix), - .tp_repr = (reprfunc)dmnsn_py_Matrix_repr, - .tp_str = (reprfunc)dmnsn_py_Matrix_str, - .tp_as_number = &dmnsn_py_Matrix_as_number, - .tp_flags = Py_TPFLAGS_DEFAULT, - .tp_doc = "Dimension matrix", - .tp_richcompare = dmnsn_py_Matrix_richcompare, - .tp_methods = dmnsn_py_Matrix_methods, - .tp_getset = dmnsn_py_Matrix_getsetters, - .tp_init = (initproc)dmnsn_py_Matrix_init, -}; - -bool -dmnsn_py_init_MatrixType(void) -{ - dmnsn_py_MatrixType.tp_new = PyType_GenericNew; - Py_INCREF(&dmnsn_py_MatrixType); - return PyType_Ready(&dmnsn_py_MatrixType) >= 0; -} |