diff options
Diffstat (limited to 'source/blender/freestyle/intern/python/BPy_Interface1D.cpp')
| -rw-r--r-- | source/blender/freestyle/intern/python/BPy_Interface1D.cpp | 321 |
1 files changed, 321 insertions, 0 deletions
diff --git a/source/blender/freestyle/intern/python/BPy_Interface1D.cpp b/source/blender/freestyle/intern/python/BPy_Interface1D.cpp new file mode 100644 index 00000000000..f6046b4cad3 --- /dev/null +++ b/source/blender/freestyle/intern/python/BPy_Interface1D.cpp @@ -0,0 +1,321 @@ +#include "BPy_Interface1D.h" + +#include "BPy_Convert.h" +#include "Interface1D/BPy_FrsCurve.h" +#include "Interface1D/Curve/BPy_Chain.h" +#include "Interface1D/BPy_FEdge.h" +#include "Interface1D/FEdge/BPy_FEdgeSharp.h" +#include "Interface1D/FEdge/BPy_FEdgeSmooth.h" +#include "Interface1D/BPy_Stroke.h" +#include "Interface1D/BPy_ViewEdge.h" + +#include "BPy_MediumType.h" + +#ifdef __cplusplus +extern "C" { +#endif + +/////////////////////////////////////////////////////////////////////////////////////////// + +//-------------------MODULE INITIALIZATION-------------------------------- +int Interface1D_Init(PyObject *module) +{ + if (module == NULL) + return -1; + + if (PyType_Ready(&Interface1D_Type) < 0) + return -1; + Py_INCREF(&Interface1D_Type); + PyModule_AddObject(module, "Interface1D", (PyObject *)&Interface1D_Type); + + if (PyType_Ready(&FrsCurve_Type) < 0) + return -1; + Py_INCREF(&FrsCurve_Type); + PyModule_AddObject(module, "Curve", (PyObject *)&FrsCurve_Type); + + if (PyType_Ready(&Chain_Type) < 0) + return -1; + Py_INCREF(&Chain_Type); + PyModule_AddObject(module, "Chain", (PyObject *)&Chain_Type); + + if (PyType_Ready(&FEdge_Type) < 0) + return -1; + Py_INCREF(&FEdge_Type); + PyModule_AddObject(module, "FEdge", (PyObject *)&FEdge_Type); + + if (PyType_Ready(&FEdgeSharp_Type) < 0) + return -1; + Py_INCREF(&FEdgeSharp_Type); + PyModule_AddObject(module, "FEdgeSharp", (PyObject *)&FEdgeSharp_Type); + + if (PyType_Ready(&FEdgeSmooth_Type) < 0) + return -1; + Py_INCREF(&FEdgeSmooth_Type); + PyModule_AddObject(module, "FEdgeSmooth", (PyObject *)&FEdgeSmooth_Type); + + if (PyType_Ready(&Stroke_Type) < 0) + return -1; + Py_INCREF(&Stroke_Type); + PyModule_AddObject(module, "Stroke", (PyObject *)&Stroke_Type); + + PyDict_SetItemString(Stroke_Type.tp_dict, "DRY_MEDIUM", BPy_MediumType_DRY_MEDIUM); + PyDict_SetItemString(Stroke_Type.tp_dict, "HUMID_MEDIUM", BPy_MediumType_HUMID_MEDIUM); + PyDict_SetItemString(Stroke_Type.tp_dict, "OPAQUE_MEDIUM", BPy_MediumType_OPAQUE_MEDIUM); + + if (PyType_Ready(&ViewEdge_Type) < 0) + return -1; + Py_INCREF(&ViewEdge_Type); + PyModule_AddObject(module, "ViewEdge", (PyObject *)&ViewEdge_Type); + + FEdgeSharp_mathutils_register_callback(); + FEdgeSmooth_mathutils_register_callback(); + + return 0; +} + +/*----------------------Interface1D methods ----------------------------*/ + +PyDoc_STRVAR(Interface1D_doc, +"Base class for any 1D element.\n" +"\n" +".. method:: __init__()\n" +"\n" +" Default constructor."); + +static int Interface1D_init(BPy_Interface1D *self, PyObject *args, PyObject *kwds) +{ + if (!PyArg_ParseTuple(args, "")) + return -1; + self->if1D = new Interface1D(); + self->borrowed = 0; + return 0; +} + +static void Interface1D_dealloc(BPy_Interface1D* self) +{ + if (self->if1D && !self->borrowed) + delete self->if1D; + Py_TYPE(self)->tp_free((PyObject*)self); +} + +static PyObject * Interface1D_repr(BPy_Interface1D* self) +{ + return PyUnicode_FromFormat("type: %s - address: %p", self->if1D->getExactTypeName().c_str(), self->if1D); +} + +PyDoc_STRVAR(Interface1D_vertices_begin_doc, +".. method:: vertices_begin()\n" +"\n" +" Returns an iterator over the Interface1D vertices, pointing to the\n" +" first vertex.\n" +"\n" +" :return: An Interface0DIterator pointing to the first vertex.\n" +" :rtype: :class:`Interface0DIterator`"); + +static PyObject * Interface1D_vertices_begin(BPy_Interface1D *self) +{ + Interface0DIterator if0D_it(self->if1D->verticesBegin()); + return BPy_Interface0DIterator_from_Interface0DIterator(if0D_it, 0); +} + +PyDoc_STRVAR(Interface1D_vertices_end_doc, +".. method:: vertices_end()\n" +"\n" +" Returns an iterator over the Interface1D vertices, pointing after\n" +" the last vertex.\n" +"\n" +" :return: An Interface0DIterator pointing after the last vertex.\n" +" :rtype: :class:`Interface0DIterator`"); + +static PyObject * Interface1D_vertices_end(BPy_Interface1D *self) +{ + Interface0DIterator if0D_it(self->if1D->verticesEnd()); + return BPy_Interface0DIterator_from_Interface0DIterator(if0D_it, 1); +} + +PyDoc_STRVAR(Interface1D_points_begin_doc, +".. method:: points_begin(t=0.0)\n" +"\n" +" Returns an iterator over the Interface1D points, pointing to the\n" +" first point. The difference with vertices_begin() is that here we can\n" +" iterate over points of the 1D element at a any given sampling.\n" +" Indeed, for each iteration, a virtual point is created.\n" +"\n" +" :arg t: A sampling with which we want to iterate over points of\n" +" this 1D element.\n" +" :type t: float\n" +" :return: An Interface0DIterator pointing to the first point.\n" +" :rtype: :class:`Interface0DIterator`"); + +static PyObject * Interface1D_points_begin(BPy_Interface1D *self, PyObject *args) +{ + float f = 0; + + if(!(PyArg_ParseTuple(args, "|f", &f))) + return NULL; + Interface0DIterator if0D_it(self->if1D->pointsBegin(f)); + return BPy_Interface0DIterator_from_Interface0DIterator(if0D_it, 0); +} + +PyDoc_STRVAR(Interface1D_points_end_doc, +".. method:: points_end(t=0.0)\n" +"\n" +" Returns an iterator over the Interface1D points, pointing after the\n" +" last point. The difference with vertices_end() is that here we can\n" +" iterate over points of the 1D element at a given sampling. Indeed,\n" +" for each iteration, a virtual point is created.\n" +"\n" +" :arg t: A sampling with which we want to iterate over points of\n" +" this 1D element.\n" +" :type t: float\n" +" :return: An Interface0DIterator pointing after the last point.\n" +" :rtype: :class:`Interface0DIterator`"); + +static PyObject * Interface1D_points_end(BPy_Interface1D *self, PyObject *args) +{ + float f = 0; + + if (!PyArg_ParseTuple(args, "|f", &f)) + return NULL; + Interface0DIterator if0D_it(self->if1D->pointsEnd(f)); + return BPy_Interface0DIterator_from_Interface0DIterator(if0D_it, 1); +} + +static PyMethodDef BPy_Interface1D_methods[] = { + {"vertices_begin", (PyCFunction)Interface1D_vertices_begin, METH_NOARGS, Interface1D_vertices_begin_doc}, + {"vertices_end", (PyCFunction)Interface1D_vertices_end, METH_NOARGS, Interface1D_vertices_end_doc}, + {"points_begin", (PyCFunction)Interface1D_points_begin, METH_VARARGS, Interface1D_points_begin_doc}, + {"points_end", (PyCFunction)Interface1D_points_end, METH_VARARGS, Interface1D_points_end_doc}, + {NULL, NULL, 0, NULL} +}; + +/*----------------------Interface1D get/setters ----------------------------*/ + +PyDoc_STRVAR(Interface1D_exact_type_name_doc, +"The string of the name of the 1D element.\n" +"\n" +":type: str"); + +static PyObject *Interface1D_exact_type_name_get(BPy_Interface1D *self, void *UNUSED(closure)) +{ + return PyUnicode_FromString(self->if1D->getExactTypeName().c_str()); +} + +PyDoc_STRVAR(Interface1D_id_doc, +"The Id of this Interface1D.\n" +"\n" +":type: :class:`Id`"); + +static PyObject *Interface1D_id_get(BPy_Interface1D *self, void *UNUSED(closure)) +{ + Id id(self->if1D->getId()); + if (PyErr_Occurred()) + return NULL; + return BPy_Id_from_Id(id); // return a copy +} + +PyDoc_STRVAR(Interface1D_nature_doc, +"The nature of this Interface1D.\n" +"\n" +":type: :class:`Nature`"); + +static PyObject *Interface1D_nature_get(BPy_Interface1D *self, void *UNUSED(closure)) +{ + Nature::VertexNature nature = self->if1D->getNature(); + if (PyErr_Occurred()) + return NULL; + return BPy_Nature_from_Nature(nature); +} + +PyDoc_STRVAR(Interface1D_length_2d_doc, +"The 2D length of this Interface1D.\n" +"\n" +":type: float"); + +static PyObject *Interface1D_length_2d_get(BPy_Interface1D *self, void *UNUSED(closure)) +{ + real length = self->if1D->getLength2D(); + if (PyErr_Occurred()) + return NULL; + return PyFloat_FromDouble((double)length); +} + +PyDoc_STRVAR(Interface1D_time_stamp_doc, +"The time stamp of the 1D element, mainly used for selection.\n" +"\n" +":type: int"); + +static PyObject *Interface1D_time_stamp_get(BPy_Interface1D *self, void *UNUSED(closure)) +{ + return PyLong_FromLong(self->if1D->getTimeStamp()); +} + +static int Interface1D_time_stamp_set(BPy_Interface1D *self, PyObject *value, void *UNUSED(closure)) +{ + int timestamp; + + if ((timestamp = PyLong_AsLong(value)) == -1 && PyErr_Occurred()) { + PyErr_SetString(PyExc_TypeError, "value must be a number"); + return -1; + } + self->if1D->setTimeStamp(timestamp); + return 0; +} + +static PyGetSetDef BPy_Interface1D_getseters[] = { + {(char *)"exact_type_name", (getter)Interface1D_exact_type_name_get, (setter)NULL, (char *)Interface1D_exact_type_name_doc, NULL}, + {(char *)"id", (getter)Interface1D_id_get, (setter)NULL, (char *)Interface1D_id_doc, NULL}, + {(char *)"nature", (getter)Interface1D_nature_get, (setter)NULL, (char *)Interface1D_nature_doc, NULL}, + {(char *)"length_2d", (getter)Interface1D_length_2d_get, (setter)NULL, (char *)Interface1D_length_2d_doc, NULL}, + {(char *)"time_stamp", (getter)Interface1D_time_stamp_get, (setter)Interface1D_time_stamp_set, (char *)Interface1D_time_stamp_doc, NULL}, + {NULL, NULL, NULL, NULL, NULL} /* Sentinel */ +}; + +/*-----------------------BPy_Interface1D type definition ------------------------------*/ + +PyTypeObject Interface1D_Type = { + PyVarObject_HEAD_INIT(NULL, 0) + "Interface1D", /* tp_name */ + sizeof(BPy_Interface1D), /* tp_basicsize */ + 0, /* tp_itemsize */ + (destructor)Interface1D_dealloc, /* tp_dealloc */ + 0, /* tp_print */ + 0, /* tp_getattr */ + 0, /* tp_setattr */ + 0, /* tp_reserved */ + (reprfunc)Interface1D_repr, /* tp_repr */ + 0, /* tp_as_number */ + 0, /* tp_as_sequence */ + 0, /* tp_as_mapping */ + 0, /* tp_hash */ + 0, /* tp_call */ + 0, /* tp_str */ + 0, /* tp_getattro */ + 0, /* tp_setattro */ + 0, /* tp_as_buffer */ + Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */ + Interface1D_doc, /* tp_doc */ + 0, /* tp_traverse */ + 0, /* tp_clear */ + 0, /* tp_richcompare */ + 0, /* tp_weaklistoffset */ + 0, /* tp_iter */ + 0, /* tp_iternext */ + BPy_Interface1D_methods, /* tp_methods */ + 0, /* tp_members */ + BPy_Interface1D_getseters, /* tp_getset */ + 0, /* tp_base */ + 0, /* tp_dict */ + 0, /* tp_descr_get */ + 0, /* tp_descr_set */ + 0, /* tp_dictoffset */ + (initproc)Interface1D_init, /* tp_init */ + 0, /* tp_alloc */ + PyType_GenericNew, /* tp_new */ +}; + +/////////////////////////////////////////////////////////////////////////////////////////// + +#ifdef __cplusplus +} +#endif |