Add matricies to Python module.

1 files changed, 12 insertions, 14 deletions

diff --git a/libdimension-python/Vector.c b/libdimension-python/Vector.c
index d4ed533..8cd7f05 100644
--- a/libdimension-python/Vector.c
+++ b/libdimension-python/Vector.c
@@ -19,18 +19,19 @@
  *************************************************************************/
 
 #include "Vector.h"
-#include "dimension.h"
 
-static int
-dmnsn_py_Vector_init(dmnsn_py_Vector *self, PyObject *args, PyObject *kwds)
+bool
+dmnsn_py_Vector_args(dmnsn_vector *v, PyObject *args, PyObject *kwds)
 {
   static char *kwlist[] = { "x", "y", "z", NULL };
+  return PyArg_ParseTupleAndKeywords(args, kwds, "ddd", kwlist,
+                                     &v->x, &v->y, &v->z);
+}
 
-  if (!PyArg_ParseTupleAndKeywords(args, kwds, "ddd", kwlist,
-                                   &self->v.x, &self->v.y, &self->v.z))
-    return -1;
-
-  return 0;
+static int
+dmnsn_py_Vector_init(dmnsn_py_Vector *self, PyObject *args, PyObject *kwds)
+{
+  return dmnsn_py_Vector_args(&self->v, args, kwds) ? 0 : -1;
 }
 
 static PyObject *
@@ -79,12 +80,9 @@ dmnsn_py_Vector_str(dmnsn_py_Vector *self)
 static PyObject *
 dmnsn_py_Vector_richcompare(PyObject *lhs, PyObject *rhs, int op)
 {
-  if (!PyObject_TypeCheck(lhs, &dmnsn_py_VectorType)) {
-    PyErr_SetString(PyExc_TypeError,
-                    "Vectors can only be compared with Vectors");
-    return NULL;
-  }
-  if (!PyObject_TypeCheck(lhs, &dmnsn_py_VectorType)) {
+  if (!PyObject_TypeCheck(lhs, &dmnsn_py_VectorType)
+      || !PyObject_TypeCheck(rhs, &dmnsn_py_VectorType))
+  {
     PyErr_SetString(PyExc_TypeError,
                     "Vectors can only be compared with Vectors");
     return NULL;