From 1aa7e55e923a300cf24cc1dd6e63f7bec44d2914 Mon Sep 17 00:00:00 2001 From: wmayer Date: Tue, 24 Jan 2012 15:23:58 +0000 Subject: [PATCH 1/2] + implement Python interface to discretize wires or edge with given deflection or number of points git-svn-id: https://free-cad.svn.sourceforge.net/svnroot/free-cad/trunk@5423 e8eeb9e2-ec13-0410-a4a9-efa5cf37419d --- src/Mod/Part/App/GeometryCurvePy.xml | 2 +- src/Mod/Part/App/GeometryCurvePyImp.cpp | 25 +++++++++---- src/Mod/Part/App/TopoShapeEdgePy.xml | 5 +++ src/Mod/Part/App/TopoShapeEdgePyImp.cpp | 47 +++++++++++++++++++++++++ src/Mod/Part/App/TopoShapeWirePy.xml | 17 +++++---- src/Mod/Part/App/TopoShapeWirePyImp.cpp | 47 +++++++++++++++++++++++++ 6 files changed, 129 insertions(+), 14 deletions(-) diff --git a/src/Mod/Part/App/GeometryCurvePy.xml b/src/Mod/Part/App/GeometryCurvePy.xml index eaae3fb19..78090909c 100644 --- a/src/Mod/Part/App/GeometryCurvePy.xml +++ b/src/Mod/Part/App/GeometryCurvePy.xml @@ -23,7 +23,7 @@ - Discretizes the curve using a given deflection and returns a list of points + Discretizes the curve using a given deflection or number of points and returns a list of points diff --git a/src/Mod/Part/App/GeometryCurvePyImp.cpp b/src/Mod/Part/App/GeometryCurvePyImp.cpp index 2c9deb4e6..01f91500a 100644 --- a/src/Mod/Part/App/GeometryCurvePyImp.cpp +++ b/src/Mod/Part/App/GeometryCurvePyImp.cpp @@ -108,22 +108,33 @@ PyObject* GeometryCurvePy::toShape(PyObject *args) PyObject* GeometryCurvePy::discretize(PyObject *args) { - double d; - if (!PyArg_ParseTuple(args, "d", &d)) + PyObject* defl_or_num; + if (!PyArg_ParseTuple(args, "O", &defl_or_num)) return 0; - Handle_Geom_Geometry g = getGeometryPtr()->handle(); - Handle_Geom_Curve c = Handle_Geom_Curve::DownCast(g); try { + Handle_Geom_Geometry g = getGeometryPtr()->handle(); + Handle_Geom_Curve c = Handle_Geom_Curve::DownCast(g); if (!c.IsNull()) { - GeomAdaptor_Curve curve_adaptator(c); + GeomAdaptor_Curve adapt(c); GCPnts_UniformAbscissa discretizer; - discretizer.Initialize (curve_adaptator, d); + if (PyInt_Check(defl_or_num)) { + int num = PyInt_AsLong(defl_or_num); + discretizer.Initialize (adapt, num); + } + else if (PyFloat_Check(defl_or_num)) { + double defl = PyFloat_AsDouble(defl_or_num); + discretizer.Initialize (adapt, defl); + } + else { + PyErr_SetString(PyExc_TypeError, "Either int or float expected"); + return 0; + } if (discretizer.IsDone () && discretizer.NbPoints () > 0) { Py::List points; int nbPoints = discretizer.NbPoints (); for (int i=1; i<=nbPoints; i++) { - gp_Pnt p = curve_adaptator.Value (discretizer.Parameter (i)); + gp_Pnt p = adapt.Value (discretizer.Parameter (i)); points.append(Py::Vector(Base::Vector3d(p.X(),p.Y(),p.Z()))); } diff --git a/src/Mod/Part/App/TopoShapeEdgePy.xml b/src/Mod/Part/App/TopoShapeEdgePy.xml index 7e01e5a42..873b3fb72 100644 --- a/src/Mod/Part/App/TopoShapeEdgePy.xml +++ b/src/Mod/Part/App/TopoShapeEdgePy.xml @@ -59,6 +59,11 @@ Set the tolerance for the edge. + + + Discretizes the edge using a given deflection or number of points and returns a list of points + + Set or get the tolerance of the vertex diff --git a/src/Mod/Part/App/TopoShapeEdgePyImp.cpp b/src/Mod/Part/App/TopoShapeEdgePyImp.cpp index d33159b6f..ec5bc0905 100644 --- a/src/Mod/Part/App/TopoShapeEdgePyImp.cpp +++ b/src/Mod/Part/App/TopoShapeEdgePyImp.cpp @@ -53,6 +53,7 @@ #include #include #include +#include #include #include @@ -396,6 +397,52 @@ PyObject* TopoShapeEdgePy::derivative3At(PyObject *args) } } +PyObject* TopoShapeEdgePy::discretize(PyObject *args) +{ + PyObject* defl_or_num; + if (!PyArg_ParseTuple(args, "O", &defl_or_num)) + return 0; + + try { + BRepAdaptor_Curve adapt(TopoDS::Edge(getTopoShapePtr()->_Shape)); + GCPnts_UniformAbscissa discretizer; + if (PyInt_Check(defl_or_num)) { + int num = PyInt_AsLong(defl_or_num); + discretizer.Initialize (adapt, num); + } + else if (PyFloat_Check(defl_or_num)) { + double defl = PyFloat_AsDouble(defl_or_num); + discretizer.Initialize (adapt, defl); + } + else { + PyErr_SetString(PyExc_TypeError, "Either int or float expected"); + return 0; + } + if (discretizer.IsDone () && discretizer.NbPoints () > 0) { + Py::List points; + int nbPoints = discretizer.NbPoints (); + for (int i=1; i<=nbPoints; i++) { + gp_Pnt p = adapt.Value (discretizer.Parameter (i)); + points.append(Py::Vector(Base::Vector3d(p.X(),p.Y(),p.Z()))); + } + + return Py::new_reference_to(points); + } + else { + PyErr_SetString(PyExc_Exception, "Descretization of curve failed"); + return 0; + } + } + catch (Standard_Failure) { + Handle_Standard_Failure e = Standard_Failure::Caught(); + PyErr_SetString(PyExc_Exception, e->GetMessageString()); + return 0; + } + + PyErr_SetString(PyExc_Exception, "Geometry is not a curve"); + return 0; +} + PyObject* TopoShapeEdgePy::setTolerance(PyObject *args) { double tol; diff --git a/src/Mod/Part/App/TopoShapeWirePy.xml b/src/Mod/Part/App/TopoShapeWirePy.xml index d1558e9a3..057bfc9c9 100644 --- a/src/Mod/Part/App/TopoShapeWirePy.xml +++ b/src/Mod/Part/App/TopoShapeWirePy.xml @@ -45,13 +45,18 @@ - - Approximate B-Spline-curve from this wire - - - - Returns the center of mass of the current system. + Approximate B-Spline-curve from this wire + + + + + Discretizes the wire using a given deflection or number of points and returns a list of points + + + + + Returns the center of mass of the current system. If the gravitational field is uniform, it is the center of gravity. The coordinates returned for the center of mass are expressed in the absolute Cartesian coordinate system. diff --git a/src/Mod/Part/App/TopoShapeWirePyImp.cpp b/src/Mod/Part/App/TopoShapeWirePyImp.cpp index 895b17dff..0011e2198 100644 --- a/src/Mod/Part/App/TopoShapeWirePyImp.cpp +++ b/src/Mod/Part/App/TopoShapeWirePyImp.cpp @@ -37,6 +37,7 @@ #include #include +#include #include #include @@ -320,6 +321,52 @@ PyObject* TopoShapeWirePy::approximate(PyObject *args) } } +PyObject* TopoShapeWirePy::discretize(PyObject *args) +{ + PyObject* defl_or_num; + if (!PyArg_ParseTuple(args, "O", &defl_or_num)) + return 0; + + try { + BRepAdaptor_CompCurve adapt(TopoDS::Wire(getTopoShapePtr()->_Shape)); + GCPnts_UniformAbscissa discretizer; + if (PyInt_Check(defl_or_num)) { + int num = PyInt_AsLong(defl_or_num); + discretizer.Initialize (adapt, num); + } + else if (PyFloat_Check(defl_or_num)) { + double defl = PyFloat_AsDouble(defl_or_num); + discretizer.Initialize (adapt, defl); + } + else { + PyErr_SetString(PyExc_TypeError, "Either int or float expected"); + return 0; + } + if (discretizer.IsDone () && discretizer.NbPoints () > 0) { + Py::List points; + int nbPoints = discretizer.NbPoints (); + for (int i=1; i<=nbPoints; i++) { + gp_Pnt p = adapt.Value (discretizer.Parameter (i)); + points.append(Py::Vector(Base::Vector3d(p.X(),p.Y(),p.Z()))); + } + + return Py::new_reference_to(points); + } + else { + PyErr_SetString(PyExc_Exception, "Descretization of wire failed"); + return 0; + } + } + catch (Standard_Failure) { + Handle_Standard_Failure e = Standard_Failure::Caught(); + PyErr_SetString(PyExc_Exception, e->GetMessageString()); + return 0; + } + + PyErr_SetString(PyExc_Exception, "Geometry is not a curve"); + return 0; +} + Py::Object TopoShapeWirePy::getCenterOfMass(void) const { GProp_GProps props; From 281c2620bc9113b42d3e8069c32b7bd7571c34d8 Mon Sep 17 00:00:00 2001 From: wmayer Date: Tue, 24 Jan 2012 16:38:06 +0000 Subject: [PATCH 2/2] + patch: improve face union (tanderson69) git-svn-id: https://free-cad.svn.sourceforge.net/svnroot/free-cad/trunk@5424 e8eeb9e2-ec13-0410-a4a9-efa5cf37419d --- src/Mod/Part/App/TopoShape.cpp | 46 +++++++++++++++++++++----------- src/Mod/Part/App/modelRefine.cpp | 19 ++----------- src/Mod/Part/App/modelRefine.h | 4 +-- 3 files changed, 34 insertions(+), 35 deletions(-) diff --git a/src/Mod/Part/App/TopoShape.cpp b/src/Mod/Part/App/TopoShape.cpp index 86e50b98e..490ba3c78 100644 --- a/src/Mod/Part/App/TopoShape.cpp +++ b/src/Mod/Part/App/TopoShape.cpp @@ -1771,17 +1771,24 @@ TopoDS_Shape TopoShape::removeSplitter() const Standard_Failure::Raise("Cannot remove splitter from empty shape"); if (_Shape.ShapeType() == TopAbs_SOLID) { - const TopoDS_Solid& solid = TopoDS::Solid(_Shape); - ModelRefine::FaceUniter uniter(solid); - if (uniter.process()) { - TopoDS_Solid solidMod; - if (!uniter.getSolid(solidMod)) - Standard_Failure::Raise("Getting solid failed"); - return solidMod; - } - else { - Standard_Failure::Raise("Removing splitter failed"); + const TopoDS_Solid &solid = TopoDS::Solid(_Shape); + BRepTools_ReShape reshape; + TopExp_Explorer it; + for (it.Init(solid, TopAbs_SHELL); it.More(); it.Next()) { + const TopoDS_Shell ¤tShell = TopoDS::Shell(it.Current()); + ModelRefine::FaceUniter uniter(currentShell); + if (uniter.process()) { + if (uniter.isModified()) { + const TopoDS_Shell &newShell = uniter.getShell(); + reshape.Replace(currentShell, newShell); + } + } + else { + Standard_Failure::Raise("Removing splitter failed"); + return _Shape; + } } + return reshape.Apply(solid); } else if (_Shape.ShapeType() == TopAbs_SHELL) { const TopoDS_Shell& shell = TopoDS::Shell(_Shape); @@ -1801,13 +1808,20 @@ TopoDS_Shape TopoShape::removeSplitter() const TopExp_Explorer xp; // solids for (xp.Init(_Shape, TopAbs_SOLID); xp.More(); xp.Next()) { - const TopoDS_Solid& solid = TopoDS::Solid(xp.Current()); - ModelRefine::FaceUniter uniter(solid); - if (uniter.process()) { - TopoDS_Solid solidMod; - if (uniter.getSolid(solidMod)) - builder.Add(comp, solidMod); + const TopoDS_Solid &solid = TopoDS::Solid(xp.Current()); + BRepTools_ReShape reshape; + TopExp_Explorer it; + for (it.Init(solid, TopAbs_SHELL); it.More(); it.Next()) { + const TopoDS_Shell ¤tShell = TopoDS::Shell(it.Current()); + ModelRefine::FaceUniter uniter(currentShell); + if (uniter.process()) { + if (uniter.isModified()) { + const TopoDS_Shell &newShell = uniter.getShell(); + reshape.Replace(currentShell, newShell); + } + } } + builder.Add(comp, reshape.Apply(solid)); } // free shells for (xp.Init(_Shape, TopAbs_SHELL, TopAbs_SOLID); xp.More(); xp.Next()) { diff --git a/src/Mod/Part/App/modelRefine.cpp b/src/Mod/Part/App/modelRefine.cpp index 72b6c6486..1b66050c2 100644 --- a/src/Mod/Part/App/modelRefine.cpp +++ b/src/Mod/Part/App/modelRefine.cpp @@ -436,19 +436,11 @@ FaceTypedCylinder& ModelRefine::getCylinderObject() ///////////////////////////////////////////////////////////////////////////////////////////////////////// -FaceUniter::FaceUniter(const TopoDS_Shell &shellIn) +FaceUniter::FaceUniter(const TopoDS_Shell &shellIn) : modifiedSignal(false) { workShell = shellIn; } -FaceUniter::FaceUniter(const TopoDS_Solid &solidIn) -{ - //get first shell - TopExp_Explorer it; - it.Init(solidIn, TopAbs_SHELL); - workShell = TopoDS::Shell(it.Current()); -} - bool FaceUniter::process() { if (workShell.IsNull()) @@ -495,6 +487,7 @@ bool FaceUniter::process() } if (facesToSew.size() > 0) { + modifiedSignal = true; workShell = ModelRefine::removeFaces(workShell, facesToRemove); TopExp_Explorer xp; bool emptyShell = true; @@ -537,11 +530,3 @@ bool FaceUniter::process() } return true; } - -bool FaceUniter::getSolid(TopoDS_Solid &outSolid) const -{ - BRepBuilderAPI_MakeSolid solidMaker; - solidMaker.Add(workShell); - outSolid = solidMaker.Solid(); - return solidMaker.IsDone() ? true : false; -} diff --git a/src/Mod/Part/App/modelRefine.h b/src/Mod/Part/App/modelRefine.h index 1f51b8ded..c008b3053 100644 --- a/src/Mod/Part/App/modelRefine.h +++ b/src/Mod/Part/App/modelRefine.h @@ -151,14 +151,14 @@ namespace ModelRefine FaceUniter(){} public: FaceUniter(const TopoDS_Shell &shellIn); - FaceUniter(const TopoDS_Solid &solidIn);//get first shell bool process(); const TopoDS_Shell& getShell() const {return workShell;} - bool getSolid(TopoDS_Solid &outSolid) const;//tries to make solid from shell. + bool isModified(){return modifiedSignal;} private: TopoDS_Shell workShell; std::vector typeObjects; + bool modifiedSignal; }; }