/*************************************************************************** * Copyright (c) 2016 Werner Mayer * * * * This file is part of the FreeCAD CAx development system. * * * * This library is free software; you can redistribute it and/or * * modify it under the terms of the GNU Library General Public * * License as published by the Free Software Foundation; either * * version 2 of the License, or (at your option) any later version. * * * * This library 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 Library General Public License for more details. * * * * You should have received a copy of the GNU Library General Public * * License along with this library; see the file COPYING.LIB. If not, * * write to the Free Software Foundation, Inc., 59 Temple Place, * * Suite 330, Boston, MA 02111-1307, USA * * * ***************************************************************************/ #include "PreCompiled.h" #ifndef _PreComp_ # include # include # include # include #endif #include #include #include #include #include using namespace Part; // returns a string which represents the object e.g. when printed in python std::string BezierCurve2dPy::representation(void) const { return ""; } PyObject *BezierCurve2dPy::PyMake(struct _typeobject *, PyObject *, PyObject *) // Python wrapper { // create a new instance of BezierCurve2dPy and the Twin object return new BezierCurve2dPy(new Geom2dBezierCurve); } // constructor method int BezierCurve2dPy::PyInit(PyObject* /*args*/, PyObject* /*kwd*/) { return 0; } PyObject* BezierCurve2dPy::isRational(PyObject *args) { if (!PyArg_ParseTuple(args, "")) return 0; Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); Standard_Boolean val = curve->IsRational(); return PyBool_FromLong(val ? 1 : 0); } PyObject* BezierCurve2dPy::isPeriodic(PyObject *args) { if (!PyArg_ParseTuple(args, "")) return 0; Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); Standard_Boolean val = curve->IsPeriodic(); return PyBool_FromLong(val ? 1 : 0); } PyObject* BezierCurve2dPy::isClosed(PyObject *args) { if (!PyArg_ParseTuple(args, "")) return 0; Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); Standard_Boolean val = curve->IsClosed(); return PyBool_FromLong(val ? 1 : 0); } PyObject* BezierCurve2dPy::increase(PyObject * args) { int degree; if (!PyArg_ParseTuple(args, "i", °ree)) return 0; Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); curve->Increase(degree); Py_Return; } PyObject* BezierCurve2dPy::insertPoleAfter(PyObject * args) { int index; double weight=1.0; PyObject* p; if (!PyArg_ParseTuple(args, "iO!|d", &index, Base::Vector2dPy::type_object(), &p, &weight)) return 0; Base::Vector2d vec = Py::Vector2d(p).getCxxObject()->value(); gp_Pnt2d pnt(vec.x, vec.y); try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); curve->InsertPoleAfter(index,pnt,weight); Py_Return; } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::insertPoleBefore(PyObject * args) { int index; double weight=1.0; PyObject* p; if (!PyArg_ParseTuple(args, "iO!|d", &index, Base::Vector2dPy::type_object(), &p, &weight)) return 0; Base::Vector2d vec = Py::Vector2d(p).getCxxObject()->value(); gp_Pnt2d pnt(vec.x, vec.y); try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); curve->InsertPoleBefore(index,pnt,weight); Py_Return; } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::removePole(PyObject * args) { int index; if (!PyArg_ParseTuple(args, "i", &index)) return 0; try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); curve->RemovePole(index); Py_Return; } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::segment(PyObject * args) { double u1,u2; if (!PyArg_ParseTuple(args, "dd", &u1,&u2)) return 0; try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); curve->Segment(u1,u2); Py_Return; } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::setPole(PyObject * args) { int index; double weight=-1.0; PyObject* p; if (!PyArg_ParseTuple(args, "iO!|d", &index, Base::Vector2dPy::type_object(), &p, &weight)) return 0; Base::Vector2d vec = Py::Vector2d(p).getCxxObject()->value(); gp_Pnt2d pnt(vec.x, vec.y); try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); if (weight < 0.0) curve->SetPole(index,pnt); else curve->SetPole(index,pnt,weight); Py_Return; } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::getPole(PyObject * args) { int index; if (!PyArg_ParseTuple(args, "i", &index)) return 0; try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); Standard_OutOfRange_Raise_if (index < 1 || index > curve->NbPoles(), "Pole index out of range"); gp_Pnt2d pnt = curve->Pole(index); Py::Module module("__FreeCADBase__"); Py::Callable method(module.getAttr("Vector2d")); Py::Tuple arg(2); arg.setItem(0, Py::Float(pnt.X())); arg.setItem(1, Py::Float(pnt.Y())); return Py::new_reference_to(method.apply(arg)); } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::getPoles(PyObject * args) { if (!PyArg_ParseTuple(args, "")) return 0; try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); TColgp_Array1OfPnt2d p(1,curve->NbPoles()); curve->Poles(p); Py::List poles; Py::Module module("__FreeCADBase__"); Py::Callable method(module.getAttr("Vector2d")); Py::Tuple arg(2); for (Standard_Integer i=p.Lower(); i<=p.Upper(); i++) { gp_Pnt2d pnt = p(i); arg.setItem(0, Py::Float(pnt.X())); arg.setItem(1, Py::Float(pnt.Y())); poles.append(method.apply(arg)); } return Py::new_reference_to(poles); } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::setPoles(PyObject * args) { PyObject* plist; if (!PyArg_ParseTuple(args, "O", &plist)) return 0; try { Py::Sequence list(plist); TColgp_Array1OfPnt2d poles(1,list.size()); int index = poles.Lower(); for (Py::Sequence::iterator it = list.begin(); it != list.end(); ++it) { Base::Vector2d pole = Py::Vector2d(*it).getCxxObject()->value(); poles.SetValue(index++, gp_Pnt2d(pole.x,pole.y)); } Handle_Geom2d_BezierCurve bezier = new Geom2d_BezierCurve(poles); this->getGeom2dBezierCurvePtr()->setHandle(bezier); Py_Return; } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::setWeight(PyObject * args) { int index; double weight; if (!PyArg_ParseTuple(args, "id", &index,&weight)) return 0; try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); curve->SetWeight(index,weight); Py_Return; } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::getWeight(PyObject * args) { int index; if (!PyArg_ParseTuple(args, "i", &index)) return 0; try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); Standard_OutOfRange_Raise_if (index < 1 || index > curve->NbPoles() , "Weight index out of range"); double weight = curve->Weight(index); return Py_BuildValue("d", weight); } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::getWeights(PyObject * args) { if (!PyArg_ParseTuple(args, "")) return 0; try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); TColStd_Array1OfReal w(1,curve->NbPoles()); curve->Weights(w); Py::List weights; for (Standard_Integer i=w.Lower(); i<=w.Upper(); i++) { weights.append(Py::Float(w(i))); } return Py::new_reference_to(weights); } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } PyObject* BezierCurve2dPy::getResolution(PyObject* args) { double tol; if (!PyArg_ParseTuple(args, "d", &tol)) return 0; try { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); double utol; curve->Resolution(tol,utol); return Py_BuildValue("d",utol); } catch (Standard_Failure) { Handle_Standard_Failure e = Standard_Failure::Caught(); PyErr_SetString(PartExceptionOCCError, e->GetMessageString()); return 0; } } Py::Long BezierCurve2dPy::getDegree(void) const { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); return Py::Long(curve->Degree()); } Py::Long BezierCurve2dPy::getMaxDegree(void) const { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); return Py::Long(curve->MaxDegree()); } Py::Long BezierCurve2dPy::getNbPoles(void) const { Handle_Geom2d_BezierCurve curve = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); return Py::Long(curve->NbPoles()); } Py::Object BezierCurve2dPy::getStartPoint(void) const { Handle_Geom2d_BezierCurve c = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); gp_Pnt2d pnt = c->StartPoint(); Py::Module module("__FreeCADBase__"); Py::Callable method(module.getAttr("Vector2d")); Py::Tuple arg(2); arg.setItem(0, Py::Float(pnt.X())); arg.setItem(1, Py::Float(pnt.Y())); return method.apply(arg); } Py::Object BezierCurve2dPy::getEndPoint(void) const { Handle_Geom2d_BezierCurve c = Handle_Geom2d_BezierCurve::DownCast (getGeometry2dPtr()->handle()); gp_Pnt2d pnt = c->EndPoint(); Py::Module module("__FreeCADBase__"); Py::Callable method(module.getAttr("Vector2d")); Py::Tuple arg(2); arg.setItem(0, Py::Float(pnt.X())); arg.setItem(1, Py::Float(pnt.Y())); return method.apply(arg); } PyObject *BezierCurve2dPy::getCustomAttributes(const char* /*attr*/) const { return 0; } int BezierCurve2dPy::setCustomAttributes(const char* /*attr*/, PyObject* /*obj*/) { return 0; }