blender-archive/source/blender/freestyle/intern/python/Interface0D/BPy_ViewVertex.cpp

#include "BPy_ViewVertex.h"

#include "../BPy_Convert.h"
#include "../Interface1D/BPy_ViewEdge.h"
#include "../BPy_Nature.h"

#ifdef __cplusplus
extern "C" {
#endif

///////////////////////////////////////////////////////////////////////////////////////////

/*----------------------ViewVertex methods----------------------------*/

PyDoc_STRVAR(ViewVertex_doc,
"Class hierarchy: :class:`Interface0D` > :class:`ViewVertex`\n"
"\n"
"Class to define a view vertex.  A view vertex is a feature vertex\n"
"corresponding to a point of the image graph, where the characteristics\n"
"of an edge (e.g., nature and visibility) might change.  A\n"
":class:`ViewVertex` can be of two kinds: A :class:`TVertex` when it\n"
"corresponds to the intersection between two ViewEdges or a\n"
":class:`NonTVertex` when it corresponds to a vertex of the initial\n"
"input mesh (it is the case for vertices such as corners for example).\n"
"Thus, this class can be specialized into two classes, the\n"
":class:`TVertex` class and the :class:`NonTVertex` class.\n"
"\n"
".. method:: __init__()\n"
"\n"
"   Default constructor.\n"
"\n"
".. method:: __init__(iBrother)\n"
"\n"
"   Copy constructor.\n"
"\n"
"   :arg iBrother: A ViewVertex object.\n"
"   :type iBrother: :class:`ViewVertex`");

static int ViewVertex_init(BPy_ViewVertex *self, PyObject *args, PyObject *kwds)
{
	if (!PyArg_ParseTuple(args, ""))
		return -1;
	self->vv = 0; // ViewVertex is abstract
	self->py_if0D.if0D = self->vv;
	self->py_if0D.borrowed = 0;
	return 0;
}

PyDoc_STRVAR(ViewVertex_edges_begin_doc,
".. method:: edges_begin()\n"
"\n"
"   Returns an iterator over the ViewEdges that goes to or comes from\n"
"   this ViewVertex pointing to the first ViewEdge of the list. The\n"
"   orientedViewEdgeIterator allows to iterate in CCW order over these\n"
"   ViewEdges and to get the orientation for each ViewEdge\n"
"   (incoming/outgoing).\n"
"\n"
"   :return: An orientedViewEdgeIterator pointing to the first ViewEdge.\n"
"   :rtype: :class:`orientedViewEdgeIterator`");

static PyObject * ViewVertex_edges_begin(BPy_ViewVertex *self)
{
	if (!self->vv)
		Py_RETURN_NONE;
	ViewVertexInternal::orientedViewEdgeIterator ove_it(self->vv->edgesBegin());
	return BPy_orientedViewEdgeIterator_from_orientedViewEdgeIterator(ove_it, 0);
}

PyDoc_STRVAR(ViewVertex_edges_end_doc,
".. method:: edges_end()\n"
"\n"
"   Returns an orientedViewEdgeIterator over the ViewEdges around this\n"
"   ViewVertex, pointing after the last ViewEdge.\n"
"\n"
"   :return: An orientedViewEdgeIterator pointing after the last ViewEdge.\n"
"   :rtype: :class:`orientedViewEdgeIterator`");

static PyObject * ViewVertex_edges_end(BPy_ViewVertex *self)
{
#if 0
	if (!self->vv)
		Py_RETURN_NONE;
	ViewVertexInternal::orientedViewEdgeIterator ove_it(self->vv->edgesEnd());
	return BPy_orientedViewEdgeIterator_from_orientedViewEdgeIterator(ove_it, 1);
#else
	PyErr_SetString(PyExc_NotImplementedError, "edges_end method currently disabled");
	return NULL;
#endif
}

PyDoc_STRVAR(ViewVertex_edges_iterator_doc,
".. method:: edges_iterator(iEdge)\n"
"\n"
"   Returns an orientedViewEdgeIterator pointing to the ViewEdge given\n"
"   as argument.\n"
"\n"
"   :arg iEdge: A ViewEdge object.\n"
"   :type iEdge: :class:`ViewEdge`\n"
"   :return: An orientedViewEdgeIterator pointing to the given ViewEdge.\n"
"   :rtype: :class:`orientedViewEdgeIterator`");

static PyObject * ViewVertex_edges_iterator(BPy_ViewVertex *self, PyObject *args)
{
	PyObject *py_ve;

	if (!self->vv)
		Py_RETURN_NONE;
	if (!PyArg_ParseTuple(args, "O!", &ViewEdge_Type, &py_ve))
		return NULL;
	ViewEdge *ve = ((BPy_ViewEdge *)py_ve)->ve;
	ViewVertexInternal::orientedViewEdgeIterator ove_it(self->vv->edgesIterator(ve));
	return BPy_orientedViewEdgeIterator_from_orientedViewEdgeIterator(ove_it, 0);
}

static PyMethodDef BPy_ViewVertex_methods[] = {
	{"edges_begin", (PyCFunction)ViewVertex_edges_begin, METH_NOARGS, ViewVertex_edges_begin_doc},
	{"edges_end", (PyCFunction)ViewVertex_edges_end, METH_NOARGS, ViewVertex_edges_end_doc},
	{"edges_iterator", (PyCFunction)ViewVertex_edges_iterator, METH_VARARGS, ViewVertex_edges_iterator_doc},
	{NULL, NULL, 0, NULL}
};

/*----------------------ViewVertex get/setters ----------------------------*/

PyDoc_STRVAR(ViewVertex_nature_doc,
"The nature of this ViewVertex.\n"
"\n"
":type: :class:`Nature`");

static PyObject *ViewVertex_nature_get(BPy_ViewVertex *self, void *UNUSED(closure))
{
	Nature::VertexNature nature = self->vv->getNature();
	if (PyErr_Occurred())
		return NULL;
	return BPy_Nature_from_Nature(nature); // return a copy
}

static int ViewVertex_nature_set(BPy_ViewVertex *self, PyObject *value, void *UNUSED(closure))
{
	if (!BPy_Nature_Check(value)) {
		PyErr_SetString(PyExc_TypeError, "value must be a Nature");
		return -1;
	}
	self->vv->setNature(PyLong_AsLong((PyObject *)&((BPy_Nature *)value)->i));
	return 0;
}

static PyGetSetDef BPy_ViewVertex_getseters[] = {
	{(char *)"nature", (getter)ViewVertex_nature_get, (setter)ViewVertex_nature_set, (char *)ViewVertex_nature_doc, NULL},
	{NULL, NULL, NULL, NULL, NULL}  /* Sentinel */
};

/*-----------------------BPy_ViewVertex type definition ------------------------------*/
PyTypeObject ViewVertex_Type = {
	PyVarObject_HEAD_INIT(NULL, 0)
	"ViewVertex",                   /* tp_name */
	sizeof(BPy_ViewVertex),         /* tp_basicsize */
	0,                              /* tp_itemsize */
	0,                              /* tp_dealloc */
	0,                              /* tp_print */
	0,                              /* tp_getattr */
	0,                              /* tp_setattr */
	0,                              /* tp_reserved */
	0,                              /* 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 */
	ViewVertex_doc,                 /* tp_doc */
	0,                              /* tp_traverse */
	0,                              /* tp_clear */
	0,                              /* tp_richcompare */
	0,                              /* tp_weaklistoffset */
	0,                              /* tp_iter */
	0,                              /* tp_iternext */
	BPy_ViewVertex_methods,         /* tp_methods */
	0,                              /* tp_members */
	BPy_ViewVertex_getseters,       /* tp_getset */
	&Interface0D_Type,              /* tp_base */
	0,                              /* tp_dict */
	0,                              /* tp_descr_get */
	0,                              /* tp_descr_set */
	0,                              /* tp_dictoffset */
	(initproc)ViewVertex_init,      /* tp_init */
	0,                              /* tp_alloc */
	0,                              /* tp_new */
};

///////////////////////////////////////////////////////////////////////////////////////////

#ifdef __cplusplus
}
#endif