diff options
Diffstat (limited to 'source/blender/freestyle/intern/python/BPy_Convert.cpp')
-rw-r--r-- | source/blender/freestyle/intern/python/BPy_Convert.cpp | 723 |
1 files changed, 723 insertions, 0 deletions
diff --git a/source/blender/freestyle/intern/python/BPy_Convert.cpp b/source/blender/freestyle/intern/python/BPy_Convert.cpp new file mode 100644 index 00000000000..daf643c20c0 --- /dev/null +++ b/source/blender/freestyle/intern/python/BPy_Convert.cpp @@ -0,0 +1,723 @@ +/* + * ***** 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. + * + * The Original Code is Copyright (C) 2010 Blender Foundation. + * All rights reserved. + * + * The Original Code is: all of this file. + * + * Contributor(s): none yet. + * + * ***** END GPL LICENSE BLOCK ***** + */ + +/** \file source/blender/freestyle/intern/python/BPy_Convert.cpp + * \ingroup freestyle + */ + +#include "BPy_Convert.h" + +#include "BPy_BBox.h" +#include "BPy_FrsMaterial.h" +#include "BPy_Id.h" +#include "BPy_IntegrationType.h" +#include "BPy_Interface0D.h" +#include "Interface0D/BPy_CurvePoint.h" +#include "Interface0D/CurvePoint/BPy_StrokeVertex.h" +#include "Interface0D/BPy_SVertex.h" +#include "Interface0D/BPy_ViewVertex.h" +#include "Interface0D/ViewVertex/BPy_NonTVertex.h" +#include "Interface0D/ViewVertex/BPy_TVertex.h" +#include "BPy_Interface1D.h" +#include "Interface1D/BPy_FEdge.h" +#include "Interface1D/BPy_Stroke.h" +#include "Interface1D/BPy_ViewEdge.h" +#include "Interface1D/Curve/BPy_Chain.h" +#include "Interface1D/FEdge/BPy_FEdgeSharp.h" +#include "Interface1D/FEdge/BPy_FEdgeSmooth.h" +#include "BPy_Nature.h" +#include "BPy_MediumType.h" +#include "BPy_SShape.h" +#include "BPy_StrokeAttribute.h" +#include "BPy_ViewShape.h" + +#include "Iterator/BPy_AdjacencyIterator.h" +#include "Iterator/BPy_ChainPredicateIterator.h" +#include "Iterator/BPy_ChainSilhouetteIterator.h" +#include "Iterator/BPy_ChainingIterator.h" +#include "Iterator/BPy_CurvePointIterator.h" +#include "Iterator/BPy_Interface0DIterator.h" +#include "Iterator/BPy_SVertexIterator.h" +#include "Iterator/BPy_StrokeVertexIterator.h" +#include "Iterator/BPy_ViewEdgeIterator.h" +#include "Iterator/BPy_orientedViewEdgeIterator.h" + +#include "../stroke/StrokeRep.h" + +#ifdef __cplusplus +extern "C" { +#endif + +/////////////////////////////////////////////////////////////////////////////////////////// + +//============================== +// C++ => Python +//============================== + +PyObject *PyBool_from_bool(bool b) +{ + return PyBool_FromLong(b ? 1 : 0); +} + +PyObject *Vector_from_Vec2f(Vec2f& vec) +{ + float vec_data[2]; // because vec->_coord is protected + vec_data[0] = vec.x(); + vec_data[1] = vec.y(); + return Vector_CreatePyObject(vec_data, 2, Py_NEW, NULL); +} + +PyObject *Vector_from_Vec3f(Vec3f& vec) +{ + float vec_data[3]; // because vec->_coord is protected + vec_data[0] = vec.x(); + vec_data[1] = vec.y(); + vec_data[2] = vec.z(); + return Vector_CreatePyObject(vec_data, 3, Py_NEW, NULL); +} + +PyObject *Vector_from_Vec3r(Vec3r& vec) +{ + float vec_data[3]; // because vec->_coord is protected + vec_data[0] = vec.x(); + vec_data[1] = vec.y(); + vec_data[2] = vec.z(); + return Vector_CreatePyObject(vec_data, 3, Py_NEW, NULL); +} + +PyObject *BPy_Id_from_Id(Id& id) +{ + PyObject *py_id = Id_Type.tp_new(&Id_Type, 0, 0); + ((BPy_Id *)py_id)->id = new Id(id.getFirst(), id.getSecond()); + return py_id; +} + +PyObject *Any_BPy_Interface0D_from_Interface0D(Interface0D& if0D) +{ + if (typeid(if0D) == typeid(CurvePoint)) { + return BPy_CurvePoint_from_CurvePoint(dynamic_cast<CurvePoint&>(if0D)); + } + else if (typeid(if0D) == typeid(StrokeVertex)) { + return BPy_StrokeVertex_from_StrokeVertex(dynamic_cast<StrokeVertex&>(if0D)); + } + else if (typeid(if0D) == typeid(SVertex)) { + return BPy_SVertex_from_SVertex(dynamic_cast<SVertex&>(if0D)); + } + else if (typeid(if0D) == typeid(ViewVertex)) { + return BPy_ViewVertex_from_ViewVertex(dynamic_cast<ViewVertex&>(if0D)); + } + else if (typeid(if0D) == typeid(NonTVertex)) { + return BPy_NonTVertex_from_NonTVertex(dynamic_cast<NonTVertex&>(if0D)); + } + else if (typeid(if0D) == typeid(TVertex)) { + return BPy_TVertex_from_TVertex(dynamic_cast<TVertex&>(if0D)); + } + else if (typeid(if0D) == typeid(Interface0D)) { + return BPy_Interface0D_from_Interface0D(if0D); + } + string msg("unexpected type: " + if0D.getExactTypeName()); + PyErr_SetString(PyExc_TypeError, msg.c_str()); + return NULL; +} + +PyObject *Any_BPy_Interface1D_from_Interface1D(Interface1D& if1D) +{ + if (typeid(if1D) == typeid(ViewEdge)) { + return BPy_ViewEdge_from_ViewEdge(dynamic_cast<ViewEdge&>(if1D)); + } + else if (typeid(if1D) == typeid(Chain)) { + return BPy_Chain_from_Chain(dynamic_cast<Chain&>(if1D)); + } + else if (typeid(if1D) == typeid(Stroke)) { + return BPy_Stroke_from_Stroke(dynamic_cast<Stroke&>(if1D)); + } + else if (typeid(if1D) == typeid(FEdgeSharp)) { + return BPy_FEdgeSharp_from_FEdgeSharp(dynamic_cast<FEdgeSharp&>(if1D)); + } + else if (typeid(if1D) == typeid(FEdgeSmooth)) { + return BPy_FEdgeSmooth_from_FEdgeSmooth(dynamic_cast<FEdgeSmooth&>(if1D)); + } + else if (typeid(if1D) == typeid(FEdge)) { + return BPy_FEdge_from_FEdge(dynamic_cast<FEdge&>(if1D)); + } + else if (typeid(if1D) == typeid(Interface1D)) { + return BPy_Interface1D_from_Interface1D(if1D); + } + string msg("unexpected type: " + if1D.getExactTypeName()); + PyErr_SetString(PyExc_TypeError, msg.c_str()); + return NULL; +} + +PyObject *Any_BPy_FEdge_from_FEdge(FEdge& fe) +{ + if (typeid(fe) == typeid(FEdgeSharp)) { + return BPy_FEdgeSharp_from_FEdgeSharp(dynamic_cast<FEdgeSharp&>(fe)); + } + else if (typeid(fe) == typeid(FEdgeSmooth)) { + return BPy_FEdgeSmooth_from_FEdgeSmooth(dynamic_cast<FEdgeSmooth&>(fe)); + } + else if (typeid(fe) == typeid(FEdge)) { + return BPy_FEdge_from_FEdge(fe); + } + string msg("unexpected type: " + fe.getExactTypeName()); + PyErr_SetString(PyExc_TypeError, msg.c_str()); + return NULL; +} + +PyObject *Any_BPy_ViewVertex_from_ViewVertex(ViewVertex& vv) +{ + if (typeid(vv) == typeid(NonTVertex)) { + return BPy_NonTVertex_from_NonTVertex(dynamic_cast<NonTVertex&>(vv)); + } + else if (typeid(vv) == typeid(TVertex)) { + return BPy_TVertex_from_TVertex(dynamic_cast<TVertex&>(vv)); + } + else if (typeid(vv) == typeid(ViewVertex)) { + return BPy_ViewVertex_from_ViewVertex(vv); + } + string msg("unexpected type: " + vv.getExactTypeName()); + PyErr_SetString(PyExc_TypeError, msg.c_str()); + return NULL; +} + +PyObject *BPy_Interface0D_from_Interface0D(Interface0D& if0D) +{ + PyObject *py_if0D = Interface0D_Type.tp_new(&Interface0D_Type, 0, 0); + ((BPy_Interface0D *)py_if0D)->if0D = &if0D; + ((BPy_Interface0D *)py_if0D)->borrowed = 1; + return py_if0D; +} + +PyObject *BPy_Interface1D_from_Interface1D(Interface1D& if1D) +{ + PyObject *py_if1D = Interface1D_Type.tp_new(&Interface1D_Type, 0, 0); + ((BPy_Interface1D *)py_if1D)->if1D = &if1D; + ((BPy_Interface1D *)py_if1D)->borrowed = 1; + return py_if1D; +} + +PyObject *BPy_SVertex_from_SVertex(SVertex& sv) +{ + PyObject *py_sv = SVertex_Type.tp_new(&SVertex_Type, 0, 0); + ((BPy_SVertex *)py_sv)->sv = &sv; + ((BPy_SVertex *)py_sv)->py_if0D.if0D = ((BPy_SVertex *)py_sv)->sv; + ((BPy_SVertex *)py_sv)->py_if0D.borrowed = 1; + return py_sv; +} + +PyObject *BPy_FEdgeSharp_from_FEdgeSharp(FEdgeSharp& fes) +{ + PyObject *py_fe = FEdgeSharp_Type.tp_new(&FEdgeSharp_Type, 0, 0); + ((BPy_FEdgeSharp *)py_fe)->fes = &fes; + ((BPy_FEdgeSharp *)py_fe)->py_fe.fe = ((BPy_FEdgeSharp *)py_fe)->fes; + ((BPy_FEdgeSharp *)py_fe)->py_fe.py_if1D.if1D = ((BPy_FEdgeSharp *)py_fe)->fes; + ((BPy_FEdgeSharp *)py_fe)->py_fe.py_if1D.borrowed = 1; + return py_fe; +} + +PyObject *BPy_FEdgeSmooth_from_FEdgeSmooth(FEdgeSmooth& fes) +{ + PyObject *py_fe = FEdgeSmooth_Type.tp_new(&FEdgeSmooth_Type, 0, 0); + ((BPy_FEdgeSmooth *)py_fe)->fes = &fes; + ((BPy_FEdgeSmooth *)py_fe)->py_fe.fe = ((BPy_FEdgeSmooth *)py_fe)->fes; + ((BPy_FEdgeSmooth *)py_fe)->py_fe.py_if1D.if1D = ((BPy_FEdgeSmooth *)py_fe)->fes; + ((BPy_FEdgeSmooth *)py_fe)->py_fe.py_if1D.borrowed = 1; + return py_fe; +} + +PyObject *BPy_FEdge_from_FEdge(FEdge& fe) +{ + PyObject *py_fe = FEdge_Type.tp_new(&FEdge_Type, 0, 0); + ((BPy_FEdge *)py_fe)->fe = &fe; + ((BPy_FEdge *)py_fe)->py_if1D.if1D = ((BPy_FEdge *)py_fe)->fe; + ((BPy_FEdge *)py_fe)->py_if1D.borrowed = 1; + return py_fe; +} + +PyObject *BPy_Nature_from_Nature(unsigned short n) +{ + PyObject *args = PyTuple_New(1); + PyTuple_SET_ITEM(args, 0, PyLong_FromLong(n)); + PyObject *py_n = Nature_Type.tp_new(&Nature_Type, args, NULL); + Py_DECREF(args); + return py_n; +} + +PyObject *BPy_Stroke_from_Stroke(Stroke& s) +{ + PyObject *py_s = Stroke_Type.tp_new(&Stroke_Type, 0, 0); + ((BPy_Stroke *)py_s)->s = &s; + ((BPy_Stroke *)py_s)->py_if1D.if1D = ((BPy_Stroke *)py_s)->s; + ((BPy_Stroke *)py_s)->py_if1D.borrowed = 1; + return py_s; +} + +PyObject *BPy_StrokeAttribute_from_StrokeAttribute(StrokeAttribute& sa) +{ + PyObject *py_sa = StrokeAttribute_Type.tp_new(&StrokeAttribute_Type, 0, 0); + ((BPy_StrokeAttribute *)py_sa)->sa = &sa; + ((BPy_StrokeAttribute *)py_sa)->borrowed = 1; + return py_sa; +} + +PyObject *BPy_MediumType_from_MediumType(Stroke::MediumType n) +{ + PyObject *args = PyTuple_New(1); + PyTuple_SET_ITEM(args, 0, PyLong_FromLong(n)); + PyObject *py_mt = MediumType_Type.tp_new(&MediumType_Type, args, NULL); + Py_DECREF(args); + return py_mt; +} + +PyObject *BPy_StrokeVertex_from_StrokeVertex(StrokeVertex& sv) +{ + PyObject *py_sv = StrokeVertex_Type.tp_new(&StrokeVertex_Type, 0, 0); + ((BPy_StrokeVertex *)py_sv)->sv = &sv; + ((BPy_StrokeVertex *)py_sv)->py_cp.cp = ((BPy_StrokeVertex *)py_sv)->sv; + ((BPy_StrokeVertex *)py_sv)->py_cp.py_if0D.if0D = ((BPy_StrokeVertex *)py_sv)->sv; + ((BPy_StrokeVertex *)py_sv)->py_cp.py_if0D.borrowed = 1; + return py_sv; +} + +PyObject *BPy_ViewVertex_from_ViewVertex(ViewVertex& vv) +{ + PyObject *py_vv = ViewVertex_Type.tp_new(&ViewVertex_Type, 0, 0); + ((BPy_ViewVertex *)py_vv)->vv = &vv; + ((BPy_ViewVertex *)py_vv)->py_if0D.if0D = ((BPy_ViewVertex *)py_vv)->vv; + ((BPy_ViewVertex *)py_vv)->py_if0D.borrowed = 1; + return py_vv; +} + +PyObject *BPy_NonTVertex_from_NonTVertex(NonTVertex& ntv) +{ + PyObject *py_ntv = NonTVertex_Type.tp_new(&NonTVertex_Type, 0, 0); + ((BPy_NonTVertex *)py_ntv)->ntv = &ntv; + ((BPy_NonTVertex *)py_ntv)->py_vv.vv = ((BPy_NonTVertex *)py_ntv)->ntv; + ((BPy_NonTVertex *)py_ntv)->py_vv.py_if0D.if0D = ((BPy_NonTVertex *)py_ntv)->ntv; + ((BPy_NonTVertex *)py_ntv)->py_vv.py_if0D.borrowed = 1; + return py_ntv; +} + +PyObject *BPy_TVertex_from_TVertex(TVertex& tv) +{ + PyObject *py_tv = TVertex_Type.tp_new(&TVertex_Type, 0, 0); + ((BPy_TVertex *)py_tv)->tv = &tv; + ((BPy_TVertex *)py_tv)->py_vv.vv = ((BPy_TVertex *)py_tv)->tv; + ((BPy_TVertex *)py_tv)->py_vv.py_if0D.if0D = ((BPy_TVertex *)py_tv)->tv; + ((BPy_TVertex *)py_tv)->py_vv.py_if0D.borrowed = 1; + return py_tv; +} + +PyObject *BPy_BBox_from_BBox(const BBox< Vec3r > &bb) +{ + PyObject *py_bb = BBox_Type.tp_new(&BBox_Type, 0, 0); + ((BPy_BBox *)py_bb)->bb = new BBox< Vec3r >(bb); + return py_bb; +} + +PyObject *BPy_ViewEdge_from_ViewEdge(ViewEdge& ve) +{ + PyObject *py_ve = ViewEdge_Type.tp_new(&ViewEdge_Type, 0, 0); + ((BPy_ViewEdge *)py_ve)->ve = &ve; + ((BPy_ViewEdge *)py_ve)->py_if1D.if1D = ((BPy_ViewEdge *)py_ve)->ve; + ((BPy_ViewEdge *)py_ve)->py_if1D.borrowed = 1; + return py_ve; +} + +PyObject *BPy_Chain_from_Chain(Chain& c) +{ + PyObject *py_c = Chain_Type.tp_new(&Chain_Type, 0, 0); + ((BPy_Chain *)py_c)->c = &c; + ((BPy_Chain *)py_c)->py_c.c = ((BPy_Chain *)py_c)->c; + ((BPy_Chain *)py_c)->py_c.py_if1D.if1D = ((BPy_Chain *)py_c)->c; + ((BPy_Chain *)py_c)->py_c.py_if1D.borrowed = 1; + return py_c; +} + +PyObject *BPy_SShape_from_SShape(SShape& ss) +{ + PyObject *py_ss = SShape_Type.tp_new(&SShape_Type, 0, 0); + ((BPy_SShape *)py_ss)->ss = &ss; + ((BPy_SShape *)py_ss)->borrowed = 1; + return py_ss; +} + +PyObject *BPy_ViewShape_from_ViewShape(ViewShape& vs) +{ + PyObject *py_vs = ViewShape_Type.tp_new(&ViewShape_Type, 0, 0); + ((BPy_ViewShape *)py_vs)->vs = &vs; + ((BPy_ViewShape *)py_vs)->borrowed = 1; + ((BPy_ViewShape *)py_vs)->py_ss = NULL; + return py_vs; +} + +PyObject *BPy_FrsMaterial_from_FrsMaterial(const FrsMaterial& m) +{ + PyObject *py_m = FrsMaterial_Type.tp_new(&FrsMaterial_Type, 0, 0); + ((BPy_FrsMaterial *) py_m)->m = new FrsMaterial(m); + return py_m; +} + +PyObject *BPy_IntegrationType_from_IntegrationType(IntegrationType i) +{ + PyObject *args = PyTuple_New(1); + PyTuple_SET_ITEM(args, 0, PyLong_FromLong(i)); + PyObject *py_it = IntegrationType_Type.tp_new(&IntegrationType_Type, args, NULL); + Py_DECREF(args); + return py_it; +} + +PyObject *BPy_CurvePoint_from_CurvePoint(CurvePoint& cp) +{ + PyObject *py_cp = CurvePoint_Type.tp_new(&CurvePoint_Type, 0, 0); + ((BPy_CurvePoint *) py_cp)->cp = &cp; + ((BPy_CurvePoint *) py_cp)->py_if0D.if0D = ((BPy_CurvePoint *)py_cp)->cp; + ((BPy_CurvePoint *) py_cp)->py_if0D.borrowed = 1; + return py_cp; +} + +PyObject *BPy_directedViewEdge_from_directedViewEdge(ViewVertex::directedViewEdge& dve) +{ + PyObject *py_dve = PyTuple_New(2); + PyTuple_SET_ITEM(py_dve, 0, BPy_ViewEdge_from_ViewEdge(*(dve.first))); + PyTuple_SET_ITEM(py_dve, 1, PyBool_from_bool(dve.second)); + return py_dve; +} + +//============================== +// Iterators +//============================== + +PyObject *BPy_AdjacencyIterator_from_AdjacencyIterator(AdjacencyIterator& a_it) +{ + PyObject *py_a_it = AdjacencyIterator_Type.tp_new(&AdjacencyIterator_Type, 0, 0); + ((BPy_AdjacencyIterator *)py_a_it)->a_it = new AdjacencyIterator(a_it); + ((BPy_AdjacencyIterator *)py_a_it)->py_it.it = ((BPy_AdjacencyIterator *)py_a_it)->a_it; + return py_a_it; +} + +PyObject *BPy_Interface0DIterator_from_Interface0DIterator(Interface0DIterator& if0D_it, int reversed) +{ + PyObject *py_if0D_it = Interface0DIterator_Type.tp_new(&Interface0DIterator_Type, 0, 0); + ((BPy_Interface0DIterator *)py_if0D_it)->if0D_it = new Interface0DIterator(if0D_it); + ((BPy_Interface0DIterator *)py_if0D_it)->py_it.it = ((BPy_Interface0DIterator *)py_if0D_it)->if0D_it; + ((BPy_Interface0DIterator *)py_if0D_it)->reversed = reversed; + return py_if0D_it; +} + +PyObject *BPy_CurvePointIterator_from_CurvePointIterator(CurveInternal::CurvePointIterator& cp_it) +{ + PyObject *py_cp_it = CurvePointIterator_Type.tp_new(&CurvePointIterator_Type, 0, 0); + ((BPy_CurvePointIterator *)py_cp_it)->cp_it = new CurveInternal::CurvePointIterator(cp_it); + ((BPy_CurvePointIterator *)py_cp_it)->py_it.it = ((BPy_CurvePointIterator *)py_cp_it)->cp_it; + return py_cp_it; +} + +PyObject *BPy_StrokeVertexIterator_from_StrokeVertexIterator(StrokeInternal::StrokeVertexIterator& sv_it, int reversed) +{ + PyObject *py_sv_it = StrokeVertexIterator_Type.tp_new(&StrokeVertexIterator_Type, 0, 0); + ((BPy_StrokeVertexIterator *)py_sv_it)->sv_it = new StrokeInternal::StrokeVertexIterator(sv_it); + ((BPy_StrokeVertexIterator *)py_sv_it)->py_it.it = ((BPy_StrokeVertexIterator *)py_sv_it)->sv_it; + ((BPy_StrokeVertexIterator *)py_sv_it)->reversed = reversed; + return py_sv_it; +} + +PyObject *BPy_SVertexIterator_from_SVertexIterator(ViewEdgeInternal::SVertexIterator& sv_it) +{ + PyObject *py_sv_it = SVertexIterator_Type.tp_new(&SVertexIterator_Type, 0, 0); + ((BPy_SVertexIterator *)py_sv_it)->sv_it = new ViewEdgeInternal::SVertexIterator(sv_it); + ((BPy_SVertexIterator *)py_sv_it)->py_it.it = ((BPy_SVertexIterator *)py_sv_it)->sv_it; + return py_sv_it; +} + +PyObject *BPy_orientedViewEdgeIterator_from_orientedViewEdgeIterator(ViewVertexInternal::orientedViewEdgeIterator& ove_it, int reversed) +{ + PyObject *py_ove_it = orientedViewEdgeIterator_Type.tp_new(&orientedViewEdgeIterator_Type, 0, 0); + ((BPy_orientedViewEdgeIterator *)py_ove_it)->ove_it = new ViewVertexInternal::orientedViewEdgeIterator(ove_it); + ((BPy_orientedViewEdgeIterator *)py_ove_it)->py_it.it = ((BPy_orientedViewEdgeIterator *)py_ove_it)->ove_it; + ((BPy_orientedViewEdgeIterator *)py_ove_it)->reversed = reversed; + return py_ove_it; +} + +PyObject *BPy_ViewEdgeIterator_from_ViewEdgeIterator(ViewEdgeInternal::ViewEdgeIterator& ve_it) +{ + PyObject *py_ve_it = ViewEdgeIterator_Type.tp_new(&ViewEdgeIterator_Type, 0, 0); + ((BPy_ViewEdgeIterator *)py_ve_it)->ve_it = new ViewEdgeInternal::ViewEdgeIterator(ve_it); + ((BPy_ViewEdgeIterator *)py_ve_it)->py_it.it = ((BPy_ViewEdgeIterator *)py_ve_it)->ve_it; + return py_ve_it; +} + +PyObject *BPy_ChainingIterator_from_ChainingIterator(ChainingIterator& c_it) +{ + PyObject *py_c_it = ChainingIterator_Type.tp_new(&ChainingIterator_Type, 0, 0); + ((BPy_ChainingIterator *)py_c_it)->c_it = new ChainingIterator(c_it); + ((BPy_ChainingIterator *)py_c_it)->py_ve_it.py_it.it = ((BPy_ChainingIterator *)py_c_it)->c_it; + return py_c_it; +} + +PyObject *BPy_ChainPredicateIterator_from_ChainPredicateIterator(ChainPredicateIterator& cp_it) +{ + PyObject *py_cp_it = ChainPredicateIterator_Type.tp_new(&ChainPredicateIterator_Type, 0, 0); + ((BPy_ChainPredicateIterator *)py_cp_it)->cp_it = new ChainPredicateIterator(cp_it); + ((BPy_ChainPredicateIterator *)py_cp_it)->py_c_it.py_ve_it.py_it.it = ((BPy_ChainPredicateIterator *)py_cp_it)->cp_it; + return py_cp_it; +} + +PyObject *BPy_ChainSilhouetteIterator_from_ChainSilhouetteIterator(ChainSilhouetteIterator& cs_it) +{ + PyObject *py_cs_it = ChainSilhouetteIterator_Type.tp_new(&ChainSilhouetteIterator_Type, 0, 0); + ((BPy_ChainSilhouetteIterator *)py_cs_it)->cs_it = new ChainSilhouetteIterator(cs_it); + ((BPy_ChainSilhouetteIterator *)py_cs_it)->py_c_it.py_ve_it.py_it.it = ((BPy_ChainSilhouetteIterator *)py_cs_it)->cs_it; + return py_cs_it; +} + +//============================== +// Python => C++ +//============================== + +bool bool_from_PyBool(PyObject *b) +{ + return PyObject_IsTrue(b) != 0; +} + +IntegrationType IntegrationType_from_BPy_IntegrationType(PyObject *obj) +{ + return static_cast<IntegrationType>(PyLong_AsLong(obj)); +} + +Stroke::MediumType MediumType_from_BPy_MediumType(PyObject *obj) +{ + return static_cast<Stroke::MediumType>(PyLong_AsLong(obj)); +} + +Nature::EdgeNature EdgeNature_from_BPy_Nature(PyObject *obj) +{ + return static_cast<Nature::EdgeNature>(PyLong_AsLong(obj)); +} + +Vec2f *Vec2f_ptr_from_PyObject(PyObject *obj) +{ + Vec2f *v; + if ((v = Vec2f_ptr_from_Vector(obj))) + return v; + if ((v = Vec2f_ptr_from_PyList(obj))) + return v; + if ((v = Vec2f_ptr_from_PyTuple(obj))) + return v; + return NULL; +} + +Vec3f *Vec3f_ptr_from_PyObject(PyObject *obj) +{ + Vec3f *v; + if ((v = Vec3f_ptr_from_Vector(obj))) + return v; + if ((v = Vec3f_ptr_from_Color(obj))) + return v; + if ((v = Vec3f_ptr_from_PyList(obj))) + return v; + if ((v = Vec3f_ptr_from_PyTuple(obj))) + return v; + return NULL; +} + +Vec3r *Vec3r_ptr_from_PyObject(PyObject *obj) +{ + Vec3r *v; + if ((v = Vec3r_ptr_from_Vector(obj))) + return v; + if ((v = Vec3r_ptr_from_Color(obj))) + return v; + if ((v = Vec3r_ptr_from_PyList(obj))) + return v; + if ((v = Vec3r_ptr_from_PyTuple(obj))) + return v; + return NULL; +} + +Vec2f *Vec2f_ptr_from_Vector(PyObject *obj) +{ + if (!VectorObject_Check(obj) || ((VectorObject *)obj)->size != 2) + return NULL; + float x = ((VectorObject *)obj)->vec[0]; + float y = ((VectorObject *)obj)->vec[1]; + return new Vec2f(x, y); +} + +Vec3f *Vec3f_ptr_from_Vector(PyObject *obj) +{ + if (!VectorObject_Check(obj) || ((VectorObject *)obj)->size != 3) + return NULL; + float x = ((VectorObject *)obj)->vec[0]; + float y = ((VectorObject *)obj)->vec[1]; + float z = ((VectorObject *)obj)->vec[2]; + return new Vec3f(x, y, z); +} + +Vec3r *Vec3r_ptr_from_Vector(PyObject *obj) +{ + if (!VectorObject_Check(obj) || ((VectorObject *)obj)->size != 3) + return NULL; + real x = ((VectorObject *)obj)->vec[0]; + real y = ((VectorObject *)obj)->vec[1]; + real z = ((VectorObject *)obj)->vec[2]; + return new Vec3r(x, y, z); +} + +Vec3f *Vec3f_ptr_from_Color(PyObject *obj) +{ + if (!ColorObject_Check(obj)) + return NULL; + float r = ((ColorObject *)obj)->col[0]; + float g = ((ColorObject *)obj)->col[1]; + float b = ((ColorObject *)obj)->col[2]; + return new Vec3f(r, g, b); +} + +Vec3r *Vec3r_ptr_from_Color(PyObject *obj) +{ + if (!ColorObject_Check(obj)) + return NULL; + real r = ((ColorObject *)obj)->col[0]; + real g = ((ColorObject *)obj)->col[1]; + real b = ((ColorObject *)obj)->col[2]; + return new Vec3r(r, g, b); +} + +static int float_array_from_PyList(PyObject *obj, float *v, int n) +{ + for (int i = 0; i < n; i++) { + v[i] = PyFloat_AsDouble(PyList_GetItem(obj, i)); + if (v[i] == -1.0f && PyErr_Occurred()) { + PyErr_SetString(PyExc_TypeError, "list elements must be a number"); + return 0; + } + } + return 1; +} + +Vec2f *Vec2f_ptr_from_PyList(PyObject *obj) +{ + float v[2]; + + if (!PyList_Check(obj) || PyList_Size(obj) != 2) + return NULL; + if (!float_array_from_PyList(obj, v, 2)) + return NULL; + return new Vec2f(v[0], v[1]); +} + +Vec3f *Vec3f_ptr_from_PyList(PyObject *obj) +{ + float v[3]; + + if (!PyList_Check(obj) || PyList_Size(obj) != 3) + return NULL; + if (!float_array_from_PyList(obj, v, 3)) + return NULL; + return new Vec3f(v[0], v[1], v[2]); +} + +Vec3r *Vec3r_ptr_from_PyList(PyObject *obj) +{ + float v[3]; + + if (!PyList_Check(obj) || PyList_Size(obj) != 3) + return NULL; + if (!float_array_from_PyList(obj, v, 3)) + return NULL; + return new Vec3r(v[0], v[1], v[2]); +} + +static int float_array_from_PyTuple(PyObject *obj, float *v, int n) +{ + for (int i = 0; i < n; i++) { + v[i] = PyFloat_AsDouble(PyTuple_GetItem(obj, i)); + if (v[i] == -1.0f && PyErr_Occurred()) { + PyErr_SetString(PyExc_TypeError, "tuple elements must be a number"); + return 0; + } + } + return 1; +} + +Vec2f *Vec2f_ptr_from_PyTuple(PyObject *obj) +{ + float v[2]; + + if (!PyTuple_Check(obj) || PyTuple_Size(obj) != 2) + return NULL; + if (!float_array_from_PyTuple(obj, v, 2)) + return NULL; + return new Vec2f(v[0], v[1]); +} + +Vec3f *Vec3f_ptr_from_PyTuple(PyObject *obj) +{ + float v[3]; + + if (!PyTuple_Check(obj) || PyTuple_Size(obj) != 3) + return NULL; + if (!float_array_from_PyTuple(obj, v, 3)) + return NULL; + return new Vec3f(v[0], v[1], v[2]); +} + +Vec3r *Vec3r_ptr_from_PyTuple(PyObject *obj) +{ + float v[3]; + + if (!PyTuple_Check(obj) || PyTuple_Size(obj) != 3) + return NULL; + if (!float_array_from_PyTuple(obj, v, 3)) + return NULL; + return new Vec3r(v[0], v[1], v[2]); +} + +// helper for argument parsing + +int float_array_from_PyObject(PyObject *obj, float *v, int n) +{ + if (VectorObject_Check(obj) && ((VectorObject *)obj)->size == n) { + for (int i = 0; i < n; i++) + v[i] = ((VectorObject *)obj)->vec[i]; + return 1; + } + else if (PyList_Check(obj) && PyList_Size(obj) == n) { + return float_array_from_PyList(obj, v, n); + } + else if (PyTuple_Check(obj) && PyTuple_Size(obj) == n) { + return float_array_from_PyList(obj, v, n); + } + return 0; +} + +/////////////////////////////////////////////////////////////////////////////////////////// + +#ifdef __cplusplus +} +#endif |