475 lines
14 KiB
C++
475 lines
14 KiB
C++
/*
|
|
* ***** BEGIN GPL LICENSE BLOCK *****
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
* ***** END GPL LICENSE BLOCK *****
|
|
*/
|
|
|
|
/** \file source/blender/freestyle/intern/python/Interface0D/BPy_SVertex.cpp
|
|
* \ingroup freestyle
|
|
*/
|
|
|
|
#include "BPy_SVertex.h"
|
|
|
|
#include "../BPy_Convert.h"
|
|
#include "../BPy_Id.h"
|
|
#include "../Interface1D/BPy_FEdge.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
/*----------------------SVertex methods ----------------------------*/
|
|
|
|
PyDoc_STRVAR(SVertex_doc,
|
|
"Class hierarchy: :class:`Interface0D` > :class:`SVertex`\n"
|
|
"\n"
|
|
"Class to define a vertex of the embedding.\n"
|
|
"\n"
|
|
".. method:: __init__()\n"
|
|
"\n"
|
|
" Default constructor.\n"
|
|
"\n"
|
|
".. method:: __init__(brother)\n"
|
|
"\n"
|
|
" Copy constructor.\n"
|
|
"\n"
|
|
" :arg brother: A SVertex object.\n"
|
|
" :type brother: :class:`SVertex`\n"
|
|
"\n"
|
|
".. method:: __init__(point_3d, id)\n"
|
|
"\n"
|
|
" Builds a SVertex from 3D coordinates and an Id.\n"
|
|
"\n"
|
|
" :arg point_3d: A three-dimensional vector.\n"
|
|
" :type point_3d: :class:`mathutils.Vector`\n"
|
|
" :arg id: An Id object.\n"
|
|
" :type id: :class:`Id`");
|
|
|
|
static int convert_v3(PyObject *obj, void *v)
|
|
{
|
|
return float_array_from_PyObject(obj, (float *)v, 3);
|
|
}
|
|
|
|
static int SVertex_init(BPy_SVertex *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
static const char *kwlist_1[] = {"brother", NULL};
|
|
static const char *kwlist_2[] = {"point_3d", "id", NULL};
|
|
PyObject *obj = 0;
|
|
float v[3];
|
|
|
|
if (PyArg_ParseTupleAndKeywords(args, kwds, "|O!", (char **)kwlist_1, &SVertex_Type, &obj)) {
|
|
if (!obj)
|
|
self->sv = new SVertex();
|
|
else
|
|
self->sv = new SVertex(*(((BPy_SVertex *)obj)->sv));
|
|
}
|
|
else if (PyErr_Clear(),
|
|
PyArg_ParseTupleAndKeywords(args, kwds, "O&O!", (char **)kwlist_2, convert_v3, v, &Id_Type, &obj))
|
|
{
|
|
Vec3r point_3d(v[0], v[1], v[2]);
|
|
self->sv = new SVertex(point_3d, *(((BPy_Id *)obj)->id));
|
|
}
|
|
else {
|
|
PyErr_SetString(PyExc_TypeError, "invalid argument(s)");
|
|
return -1;
|
|
}
|
|
self->py_if0D.if0D = self->sv;
|
|
self->py_if0D.borrowed = false;
|
|
return 0;
|
|
}
|
|
|
|
PyDoc_STRVAR(SVertex_add_normal_doc,
|
|
".. method:: add_normal(normal)\n"
|
|
"\n"
|
|
" Adds a normal to the SVertex's set of normals. If the same normal\n"
|
|
" is already in the set, nothing changes.\n"
|
|
"\n"
|
|
" :arg normal: A three-dimensional vector.\n"
|
|
" :type normal: :class:`mathutils.Vector`, list or tuple of 3 real numbers");
|
|
|
|
static PyObject *SVertex_add_normal(BPy_SVertex *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
static const char *kwlist[] = {"normal", NULL};
|
|
PyObject *py_normal;
|
|
Vec3r n;
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O", (char **)kwlist, &py_normal))
|
|
return NULL;
|
|
if (!Vec3r_ptr_from_PyObject(py_normal, n)) {
|
|
PyErr_SetString(PyExc_TypeError, "argument 1 must be a 3D vector (either a list of 3 elements or Vector)");
|
|
return NULL;
|
|
}
|
|
self->sv->AddNormal(n);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
PyDoc_STRVAR(SVertex_add_fedge_doc,
|
|
".. method:: add_fedge(fedge)\n"
|
|
"\n"
|
|
" Add an FEdge to the list of edges emanating from this SVertex.\n"
|
|
"\n"
|
|
" :arg fedge: An FEdge.\n"
|
|
" :type fedge: :class:`FEdge`");
|
|
|
|
static PyObject *SVertex_add_fedge(BPy_SVertex *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
static const char *kwlist[] = {"fedge", NULL};
|
|
PyObject *py_fe;
|
|
|
|
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O!", (char **)kwlist, &FEdge_Type, &py_fe))
|
|
return NULL;
|
|
self->sv->AddFEdge(((BPy_FEdge *)py_fe)->fe);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
// virtual bool operator== (const SVertex &brother)
|
|
|
|
static PyMethodDef BPy_SVertex_methods[] = {
|
|
{"add_normal", (PyCFunction)SVertex_add_normal, METH_VARARGS | METH_KEYWORDS, SVertex_add_normal_doc},
|
|
{"add_fedge", (PyCFunction)SVertex_add_fedge, METH_VARARGS | METH_KEYWORDS, SVertex_add_fedge_doc},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
/*----------------------mathutils callbacks ----------------------------*/
|
|
|
|
/* subtype */
|
|
#define MATHUTILS_SUBTYPE_POINT3D 1
|
|
#define MATHUTILS_SUBTYPE_POINT2D 2
|
|
|
|
static int SVertex_mathutils_check(BaseMathObject *bmo)
|
|
{
|
|
if (!BPy_SVertex_Check(bmo->cb_user))
|
|
return -1;
|
|
return 0;
|
|
}
|
|
|
|
static int SVertex_mathutils_get(BaseMathObject *bmo, int subtype)
|
|
{
|
|
BPy_SVertex *self = (BPy_SVertex *)bmo->cb_user;
|
|
switch (subtype) {
|
|
case MATHUTILS_SUBTYPE_POINT3D:
|
|
bmo->data[0] = self->sv->getX();
|
|
bmo->data[1] = self->sv->getY();
|
|
bmo->data[2] = self->sv->getZ();
|
|
break;
|
|
case MATHUTILS_SUBTYPE_POINT2D:
|
|
bmo->data[0] = self->sv->getProjectedX();
|
|
bmo->data[1] = self->sv->getProjectedY();
|
|
bmo->data[2] = self->sv->getProjectedZ();
|
|
break;
|
|
default:
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int SVertex_mathutils_set(BaseMathObject *bmo, int subtype)
|
|
{
|
|
BPy_SVertex *self = (BPy_SVertex *)bmo->cb_user;
|
|
switch (subtype) {
|
|
case MATHUTILS_SUBTYPE_POINT3D:
|
|
{
|
|
Vec3r p(bmo->data[0], bmo->data[1], bmo->data[2]);
|
|
self->sv->setPoint3D(p);
|
|
}
|
|
break;
|
|
case MATHUTILS_SUBTYPE_POINT2D:
|
|
{
|
|
Vec3r p(bmo->data[0], bmo->data[1], bmo->data[2]);
|
|
self->sv->setPoint2D(p);
|
|
}
|
|
break;
|
|
default:
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int SVertex_mathutils_get_index(BaseMathObject *bmo, int subtype, int index)
|
|
{
|
|
BPy_SVertex *self = (BPy_SVertex *)bmo->cb_user;
|
|
switch (subtype) {
|
|
case MATHUTILS_SUBTYPE_POINT3D:
|
|
switch (index) {
|
|
case 0: bmo->data[0] = self->sv->getX(); break;
|
|
case 1: bmo->data[1] = self->sv->getY(); break;
|
|
case 2: bmo->data[2] = self->sv->getZ(); break;
|
|
default:
|
|
return -1;
|
|
}
|
|
break;
|
|
case MATHUTILS_SUBTYPE_POINT2D:
|
|
switch (index) {
|
|
case 0: bmo->data[0] = self->sv->getProjectedX(); break;
|
|
case 1: bmo->data[1] = self->sv->getProjectedY(); break;
|
|
case 2: bmo->data[2] = self->sv->getProjectedZ(); break;
|
|
default:
|
|
return -1;
|
|
}
|
|
break;
|
|
default:
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static int SVertex_mathutils_set_index(BaseMathObject *bmo, int subtype, int index)
|
|
{
|
|
BPy_SVertex *self = (BPy_SVertex *)bmo->cb_user;
|
|
switch (subtype) {
|
|
case MATHUTILS_SUBTYPE_POINT3D:
|
|
{
|
|
Vec3r p(self->sv->point3D());
|
|
p[index] = bmo->data[index];
|
|
self->sv->setPoint3D(p);
|
|
}
|
|
break;
|
|
case MATHUTILS_SUBTYPE_POINT2D:
|
|
{
|
|
Vec3r p(self->sv->point2D());
|
|
p[index] = bmo->data[index];
|
|
self->sv->setPoint2D(p);
|
|
}
|
|
break;
|
|
default:
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static Mathutils_Callback SVertex_mathutils_cb = {
|
|
SVertex_mathutils_check,
|
|
SVertex_mathutils_get,
|
|
SVertex_mathutils_set,
|
|
SVertex_mathutils_get_index,
|
|
SVertex_mathutils_set_index
|
|
};
|
|
|
|
static unsigned char SVertex_mathutils_cb_index = -1;
|
|
|
|
void SVertex_mathutils_register_callback()
|
|
{
|
|
SVertex_mathutils_cb_index = Mathutils_RegisterCallback(&SVertex_mathutils_cb);
|
|
}
|
|
|
|
/*----------------------SVertex get/setters ----------------------------*/
|
|
|
|
PyDoc_STRVAR(SVertex_point_3d_doc,
|
|
"The 3D coordinates of the SVertex.\n"
|
|
"\n"
|
|
":type: mathutils.Vector");
|
|
|
|
static PyObject *SVertex_point_3d_get(BPy_SVertex *self, void *UNUSED(closure))
|
|
{
|
|
return Vector_CreatePyObject_cb((PyObject *)self, 3, SVertex_mathutils_cb_index, MATHUTILS_SUBTYPE_POINT3D);
|
|
}
|
|
|
|
static int SVertex_point_3d_set(BPy_SVertex *self, PyObject *value, void *UNUSED(closure))
|
|
{
|
|
float v[3];
|
|
if (!float_array_from_PyObject(value, v, 3)) {
|
|
PyErr_SetString(PyExc_ValueError, "value must be a 3-dimensional vector");
|
|
return -1;
|
|
}
|
|
Vec3r p(v[0], v[1], v[2]);
|
|
self->sv->setPoint3D(p);
|
|
return 0;
|
|
}
|
|
|
|
PyDoc_STRVAR(SVertex_point_2d_doc,
|
|
"The projected 3D coordinates of the SVertex.\n"
|
|
"\n"
|
|
":type: mathutils.Vector");
|
|
|
|
static PyObject *SVertex_point_2d_get(BPy_SVertex *self, void *UNUSED(closure))
|
|
{
|
|
return Vector_CreatePyObject_cb((PyObject *)self, 3, SVertex_mathutils_cb_index, MATHUTILS_SUBTYPE_POINT2D);
|
|
}
|
|
|
|
static int SVertex_point_2d_set(BPy_SVertex *self, PyObject *value, void *UNUSED(closure))
|
|
{
|
|
float v[3];
|
|
if (!float_array_from_PyObject(value, v, 3)) {
|
|
PyErr_SetString(PyExc_ValueError, "value must be a 3-dimensional vector");
|
|
return -1;
|
|
}
|
|
Vec3r p(v[0], v[1], v[2]);
|
|
self->sv->setPoint2D(p);
|
|
return 0;
|
|
}
|
|
|
|
PyDoc_STRVAR(SVertex_id_doc,
|
|
"The Id of this SVertex.\n"
|
|
"\n"
|
|
":type: :class:`Id`");
|
|
|
|
static PyObject *SVertex_id_get(BPy_SVertex *self, void *UNUSED(closure))
|
|
{
|
|
Id id(self->sv->getId());
|
|
return BPy_Id_from_Id(id); // return a copy
|
|
}
|
|
|
|
static int SVertex_id_set(BPy_SVertex *self, PyObject *value, void *UNUSED(closure))
|
|
{
|
|
if (!BPy_Id_Check(value)) {
|
|
PyErr_SetString(PyExc_TypeError, "value must be an Id");
|
|
return -1;
|
|
}
|
|
self->sv->setId(*(((BPy_Id *)value)->id));
|
|
return 0;
|
|
}
|
|
|
|
PyDoc_STRVAR(SVertex_normals_doc,
|
|
"The normals for this Vertex as a list. In a sharp surface, an SVertex\n"
|
|
"has exactly one normal. In a smooth surface, an SVertex can have any\n"
|
|
"number of normals.\n"
|
|
"\n"
|
|
":type: list of :class:`mathutils.Vector` objects");
|
|
|
|
static PyObject *SVertex_normals_get(BPy_SVertex *self, void *UNUSED(closure))
|
|
{
|
|
PyObject *py_normals;
|
|
set< Vec3r > normals;
|
|
|
|
py_normals = PyList_New(0);
|
|
normals = self->sv->normals();
|
|
for (set< Vec3r >::iterator set_iterator = normals.begin(); set_iterator != normals.end(); set_iterator++) {
|
|
Vec3r v(*set_iterator);
|
|
PyList_Append(py_normals, Vector_from_Vec3r(v));
|
|
}
|
|
return py_normals;
|
|
}
|
|
|
|
PyDoc_STRVAR(SVertex_normals_size_doc,
|
|
"The number of different normals for this SVertex.\n"
|
|
"\n"
|
|
":type: int");
|
|
|
|
static PyObject *SVertex_normals_size_get(BPy_SVertex *self, void *UNUSED(closure))
|
|
{
|
|
return PyLong_FromLong(self->sv->normalsSize());
|
|
}
|
|
|
|
PyDoc_STRVAR(SVertex_viewvertex_doc,
|
|
"If this SVertex is also a ViewVertex, this property refers to the\n"
|
|
"ViewVertex, and None otherwise.\n"
|
|
"\n"
|
|
":type: :class:`ViewVertex`");
|
|
|
|
static PyObject *SVertex_viewvertex_get(BPy_SVertex *self, void *UNUSED(closure))
|
|
{
|
|
ViewVertex *vv = self->sv->viewvertex();
|
|
if (vv)
|
|
return Any_BPy_ViewVertex_from_ViewVertex(*vv);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
PyDoc_STRVAR(SVertex_curvatures_doc,
|
|
"Curvature information expressed in the form of a seven-element tuple\n"
|
|
"(K1, e1, K2, e2, Kr, er, dKr), where K1 and K2 are scalar values\n"
|
|
"representing the first (maximum) and second (minimum) principal\n"
|
|
"curvatures at this SVertex, respectively; e1 and e2 are\n"
|
|
"three-dimensional vectors representing the first and second principal\n"
|
|
"directions, i.e. the directions of the normal plane where the\n"
|
|
"curvature takes its maximum and minimum values, respectively; and Kr,\n"
|
|
"er and dKr are the radial curvature, radial direction, and the\n"
|
|
"derivative of the radial curvature at this SVertex, repectively.\n"
|
|
"\n"
|
|
":type: tuple");
|
|
|
|
static PyObject *SVertex_curvatures_get(BPy_SVertex *self, void *UNUSED(closure))
|
|
{
|
|
const CurvatureInfo *info = self->sv->getCurvatureInfo();
|
|
if (!info)
|
|
Py_RETURN_NONE;
|
|
Vec3r e1(info->e1.x(), info->e1.y(), info->e1.z());
|
|
Vec3r e2(info->e2.x(), info->e2.y(), info->e2.z());
|
|
Vec3r er(info->er.x(), info->er.y(), info->er.z());
|
|
PyObject *retval = PyTuple_New(7);
|
|
PyTuple_SET_ITEM(retval, 0, PyFloat_FromDouble(info->K1));
|
|
PyTuple_SET_ITEM(retval, 2, Vector_from_Vec3r(e1));
|
|
PyTuple_SET_ITEM(retval, 1, PyFloat_FromDouble(info->K2));
|
|
PyTuple_SET_ITEM(retval, 3, Vector_from_Vec3r(e2));
|
|
PyTuple_SET_ITEM(retval, 4, PyFloat_FromDouble(info->Kr));
|
|
PyTuple_SET_ITEM(retval, 5, Vector_from_Vec3r(er));
|
|
PyTuple_SET_ITEM(retval, 6, PyFloat_FromDouble(info->dKr));
|
|
return retval;
|
|
}
|
|
|
|
static PyGetSetDef BPy_SVertex_getseters[] = {
|
|
{(char *)"point_3d", (getter)SVertex_point_3d_get, (setter)SVertex_point_3d_set,
|
|
(char *)SVertex_point_3d_doc, NULL},
|
|
{(char *)"point_2d", (getter)SVertex_point_2d_get, (setter)SVertex_point_2d_set,
|
|
(char *)SVertex_point_2d_doc, NULL},
|
|
{(char *)"id", (getter)SVertex_id_get, (setter)SVertex_id_set, (char *)SVertex_id_doc, NULL},
|
|
{(char *)"normals", (getter)SVertex_normals_get, (setter)NULL, (char *)SVertex_normals_doc, NULL},
|
|
{(char *)"normals_size", (getter)SVertex_normals_size_get, (setter)NULL, (char *)SVertex_normals_size_doc, NULL},
|
|
{(char *)"viewvertex", (getter)SVertex_viewvertex_get, (setter)NULL, (char *)SVertex_viewvertex_doc, NULL},
|
|
{(char *)"curvatures", (getter)SVertex_curvatures_get, (setter)NULL, (char *)SVertex_curvatures_doc, NULL},
|
|
{NULL, NULL, NULL, NULL, NULL} /* Sentinel */
|
|
};
|
|
|
|
/*-----------------------BPy_SVertex type definition ------------------------------*/
|
|
PyTypeObject SVertex_Type = {
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
|
"SVertex", /* tp_name */
|
|
sizeof(BPy_SVertex), /* 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 */
|
|
SVertex_doc, /* tp_doc */
|
|
0, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
0, /* tp_iter */
|
|
0, /* tp_iternext */
|
|
BPy_SVertex_methods, /* tp_methods */
|
|
0, /* tp_members */
|
|
BPy_SVertex_getseters, /* tp_getset */
|
|
&Interface0D_Type, /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
(initproc)SVertex_init, /* tp_init */
|
|
0, /* tp_alloc */
|
|
0, /* tp_new */
|
|
};
|
|
|
|
///////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|