diff options
Diffstat (limited to 'source/blender/python/generic/vector.c')
-rw-r--r-- | source/blender/python/generic/vector.c | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/source/blender/python/generic/vector.c b/source/blender/python/generic/vector.c index b3e54ac2250..b8f2ca6f1df 100644 --- a/source/blender/python/generic/vector.c +++ b/source/blender/python/generic/vector.c @@ -48,6 +48,7 @@ static PyObject *Vector_Negate( VectorObject * self ); static PyObject *Vector_Resize2D( VectorObject * self ); static PyObject *Vector_Resize3D( VectorObject * self ); static PyObject *Vector_Resize4D( VectorObject * self ); +static PyObject *Vector_ToTuple( VectorObject * self, PyObject *value ); static PyObject *Vector_ToTrackQuat( VectorObject * self, PyObject * args ); static PyObject *Vector_Reflect( VectorObject *self, VectorObject *value ); static PyObject *Vector_Cross( VectorObject * self, VectorObject * value ); @@ -61,6 +62,7 @@ static struct PyMethodDef Vector_methods[] = { {"resize2D", (PyCFunction) Vector_Resize2D, METH_NOARGS, NULL}, {"resize3D", (PyCFunction) Vector_Resize3D, METH_NOARGS, NULL}, {"resize4D", (PyCFunction) Vector_Resize4D, METH_NOARGS, NULL}, + {"toTuple", (PyCFunction) Vector_ToTuple, METH_O, NULL}, {"toTrackQuat", ( PyCFunction ) Vector_ToTrackQuat, METH_VARARGS, NULL}, {"reflect", ( PyCFunction ) Vector_Reflect, METH_O, NULL}, {"cross", ( PyCFunction ) Vector_Cross, METH_O, NULL}, @@ -236,6 +238,30 @@ static PyObject *Vector_Resize4D(VectorObject * self) Py_INCREF(self); return (PyObject*)self; } + +/*----------------------------Vector.resize4D() ------------------ + resize the vector to x,y,z,w */ +static PyObject *Vector_ToTuple(VectorObject * self, PyObject *value) +{ + int ndigits= PyLong_AsSsize_t(value); + int x; + + PyObject *ret; + + if(ndigits > 22 || ndigits < 0) { /* accounts for non ints */ + PyErr_SetString(PyExc_TypeError, "vector.key(ndigits): ndigits must be between 0 and 21"); + return NULL; + } + + ret= PyTuple_New(self->size); + + for(x = 0; x < self->size; x++) { + PyTuple_SET_ITEM(ret, x, PyFloat_FromDouble(double_round((double)self->vec[x], ndigits))); + } + + return ret; +} + /*----------------------------Vector.toTrackQuat(track, up) ---------------------- extract a quaternion from the vector and the track and up axis */ static PyObject *Vector_ToTrackQuat( VectorObject * self, PyObject * args ) |