Skip to content

Commit

Permalink
Part: expose ChFi2d_FilletAPI to Python and move to ChFi2d module
Browse files Browse the repository at this point in the history
  • Loading branch information
wwmayer committed May 28, 2022
1 parent bd853e5 commit b07be87
Show file tree
Hide file tree
Showing 8 changed files with 263 additions and 10 deletions.
7 changes: 6 additions & 1 deletion src/Mod/Part/App/AppPart.cpp
Expand Up @@ -52,6 +52,7 @@
#include "Mod/Part/App/BSplineCurvePy.h"
#include "Mod/Part/App/BSplineSurfacePy.h"
#include <Mod/Part/App/ChFi2d_FilletAlgoPy.h>
#include <Mod/Part/App/ChFi2d_FilletAPIPy.h>
#include "Mod/Part/App/CirclePy.h"
#include "Mod/Part/App/ConePy.h"
#include "Mod/Part/App/ConicPy.h"
Expand Down Expand Up @@ -291,7 +292,6 @@ PyMOD_INIT_FUNC(Part)
Base::Interpreter().addType(&Part::GeometryBoolExtensionPy ::Type,partModule,"GeometryBoolExtension");
Base::Interpreter().addType(&Part::GeometryDoubleExtensionPy ::Type,partModule,"GeometryDoubleExtension");
Base::Interpreter().addType(&Part::PrecisionPy ::Type,partModule,"Precision");
Base::Interpreter().addType(&Part::ChFi2d_FilletAlgoPy::Type,partModule,"ChFi2d_FilletAlgo");

// BRepFeat package
PyObject* brepfeatModule(module.getAttr("BRepFeat").ptr());
Expand Down Expand Up @@ -339,6 +339,11 @@ PyMOD_INIT_FUNC(Part)
PyObject* shapeUpgrade(module.getAttr("ShapeUpgrade").ptr());
Base::Interpreter().addType(&Part::UnifySameDomainPy::Type, shapeUpgrade, "UnifySameDomain");

// ChFi2d sub-module
PyObject* chFi2d(module.getAttr("ChFi2d").ptr());
Base::Interpreter().addType(&Part::ChFi2d_FilletAlgoPy::Type, chFi2d, "FilletAlgo");
Base::Interpreter().addType(&Part::ChFi2d_FilletAPIPy::Type, chFi2d, "FilletAPI");

Part::TopoShape ::init();
Part::PropertyPartShape ::init();
Part::PropertyGeometryList ::init();
Expand Down
13 changes: 13 additions & 0 deletions src/Mod/Part/App/AppPartPy.cpp
Expand Up @@ -314,6 +314,17 @@ class ShapeUpgradeModule : public Py::ExtensionModule<ShapeUpgradeModule>
virtual ~ShapeUpgradeModule() {}
};

class ChFi2dModule : public Py::ExtensionModule<ChFi2dModule>
{
public:
ChFi2dModule() : Py::ExtensionModule<ChFi2dModule>("ChFi2d")
{
initialize("This is a module working with the ChFi2d framework."); // register with Python
}

virtual ~ChFi2dModule() {}
};

class Module : public Py::ExtensionModule<Module>
{
BRepFeatModule brepFeat;
Expand All @@ -322,6 +333,7 @@ class Module : public Py::ExtensionModule<Module>
GeomPlateModule geomPlate;
HLRBRepModule HLRBRep;
ShapeUpgradeModule shapeUpgrade;
ChFi2dModule chFi2d;
public:
Module() : Py::ExtensionModule<Module>("Part")
{
Expand Down Expand Up @@ -548,6 +560,7 @@ class Module : public Py::ExtensionModule<Module>
PyModule_AddObject(m_module, "GeomPlate", geomPlate.module().ptr());
PyModule_AddObject(m_module, "HLRBRep", HLRBRep.module().ptr());
PyModule_AddObject(m_module, "ShapeUpgrade", shapeUpgrade.module().ptr());
PyModule_AddObject(m_module, "ChFi2d", chFi2d.module().ptr());
}

virtual ~Module() {}
Expand Down
3 changes: 3 additions & 0 deletions src/Mod/Part/App/CMakeLists.txt
Expand Up @@ -92,6 +92,7 @@ generate_from_xml(TopoShapeWirePy)
generate_from_xml(BRepOffsetAPI_MakePipeShellPy)
generate_from_xml(BRepOffsetAPI_MakeFillingPy)
generate_from_xml(ChFi2d_FilletAlgoPy)
generate_from_xml(ChFi2d_FilletAPIPy)
generate_from_xml(PrecisionPy)

# make sure to create the directory at configure time
Expand Down Expand Up @@ -321,6 +322,8 @@ SET(Python_SRCS
BRepOffsetAPI_MakeFillingPyImp.cpp
ChFi2d_FilletAlgoPy.xml
ChFi2d_FilletAlgoPyImp.cpp
ChFi2d_FilletAPIPy.xml
ChFi2d_FilletAPIPyImp.cpp
PrecisionPy.xml
PrecisionPyImp.cpp
PartPyCXX.cpp
Expand Down
44 changes: 44 additions & 0 deletions src/Mod/Part/App/ChFi2d_FilletAPIPy.xml
@@ -0,0 +1,44 @@
<?xml version="1.0" encoding="UTF-8"?>
<GenerateModel xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="generateMetaModel_Module.xsd">
<PythonExport
Father="PyObjectBase"
Name="ChFi2d_FilletAPIPy"
PythonName="Part.ChFi2d.FilletAPI"
Twin="ChFi2d_FilletAPI"
TwinPointer="ChFi2d_FilletAPI"
Include="ChFi2d_FilletAPI.hxx"
Namespace="Part"
FatherInclude="Base/PyObjectBase.h"
FatherNamespace="Base"
Constructor="true"
Delete="true">
<Documentation>
<Author Licence="LGPL" Name="Werner Mayer" EMail="wmayer[at]users.sourceforge.net" />
<UserDocu>Algorithm that creates fillet edge</UserDocu>
</Documentation>
<Methode Name="init">
<Documentation>
<UserDocu>Initializes a fillet algorithm: accepts a wire consisting of two edges in a plane</UserDocu>
</Documentation>
</Methode>
<Methode Name="perform">
<Documentation>
<UserDocu>perform(radius) -> bool

Constructs a fillet edge</UserDocu>
</Documentation>
</Methode>
<Methode Name="numberOfResults">
<Documentation>
<UserDocu>Returns number of possible solutions</UserDocu>
</Documentation>
</Methode>
<Methode Name="result">
<Documentation>
<UserDocu>result(point, solution=-1)

Returns result (fillet edge, modified edge1, modified edge2)</UserDocu>
</Documentation>
</Methode>
</PythonExport>
</GenerateModel>
188 changes: 188 additions & 0 deletions src/Mod/Part/App/ChFi2d_FilletAPIPyImp.cpp
@@ -0,0 +1,188 @@
/***************************************************************************
* Copyright (c) 2022 Werner Mayer <wmayer[at]users.sourceforge.net> *
* *
* 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 <gp_Pln.hxx>
# include <Standard_Failure.hxx>
# include <TopoDS.hxx>
# include <TopoDS_Edge.hxx>
# include <TopoDS_Wire.hxx>
#endif

#include <Mod/Part/App/ChFi2d_FilletAPIPy.h>
#include <Mod/Part/App/ChFi2d_FilletAPIPy.cpp>
#include <Mod/Part/App/TopoShapeEdgePy.h>
#include <Mod/Part/App/TopoShapeWirePy.h>
#include <Mod/Part/App/Geometry.h>
#include <Mod/Part/App/PlanePy.h>
#include <Base/VectorPy.h>
#include "Tools.h"

using namespace Part;


PyObject *ChFi2d_FilletAPIPy::PyMake(struct _typeobject *, PyObject *, PyObject *) // Python wrapper
{
// create a new instance of ChFi2d_FilletAPIPy and the Twin object
return new ChFi2d_FilletAPIPy(new ChFi2d_FilletAPI);
}

// constructor method
int ChFi2d_FilletAPIPy::PyInit(PyObject* args, PyObject* /*kwd*/)
{
if (PyArg_ParseTuple(args, ""))
return 0;

PyErr_Clear();
PyObject* wire;
PyObject* plane;
if (PyArg_ParseTuple(args, "O!O!", &TopoShapeWirePy::Type, &wire, &PlanePy::Type, &plane)) {
TopoDS_Shape shape = static_cast<TopoShapeWirePy*>(wire)->getTopoShapePtr()->getShape();
Handle(Geom_Plane) hPlane = Handle(Geom_Plane)::DownCast(static_cast<PlanePy*>(plane)->getGeomPlanePtr()->handle());
getChFi2d_FilletAPIPtr()->Init(TopoDS::Wire(shape), hPlane->Pln());
return 0;
}

PyErr_Clear();
PyObject* edge1;
PyObject* edge2;
if (PyArg_ParseTuple(args, "O!O!O!", &TopoShapeEdgePy::Type, &edge1,
&TopoShapeEdgePy::Type, &edge2,
&PlanePy::Type, &plane)) {
TopoDS_Shape shape1 = static_cast<TopoShapeEdgePy*>(edge1)->getTopoShapePtr()->getShape();
TopoDS_Shape shape2 = static_cast<TopoShapeEdgePy*>(edge2)->getTopoShapePtr()->getShape();
Handle(Geom_Plane) hPlane = Handle(Geom_Plane)::DownCast(static_cast<PlanePy*>(plane)->getGeomPlanePtr()->handle());
getChFi2d_FilletAPIPtr()->Init(TopoDS::Edge(shape1), TopoDS::Edge(shape2), hPlane->Pln());
return 0;
}

PyErr_SetString(PyExc_TypeError, "Wrong arguments:\n"
"-- FilletAPI()\n"
"-- FilletAPI(wire, plane)"
"-- FilletAPI(edge, edge, plane)\n");
return -1;
}

// returns a string which represents the object e.g. when printed in python
std::string ChFi2d_FilletAPIPy::representation() const
{
return std::string("<FilletAPI object>");
}

PyObject* ChFi2d_FilletAPIPy::init(PyObject *args)
{
PyObject* wire;
PyObject* plane;
if (PyArg_ParseTuple(args, "O!O!", &TopoShapeWirePy::Type, &wire, &PlanePy::Type, &plane)) {
TopoDS_Shape shape = static_cast<TopoShapeWirePy*>(wire)->getTopoShapePtr()->getShape();
Handle(Geom_Plane) hPlane = Handle(Geom_Plane)::DownCast(static_cast<PlanePy*>(plane)->getGeomPlanePtr()->handle());
getChFi2d_FilletAPIPtr()->Init(TopoDS::Wire(shape), hPlane->Pln());
Py_Return;
}

PyErr_Clear();
PyObject* edge1;
PyObject* edge2;
if (PyArg_ParseTuple(args, "O!O!O!", &TopoShapeEdgePy::Type, &edge1,
&TopoShapeEdgePy::Type, &edge2,
&PlanePy::Type, &plane)) {
TopoDS_Shape shape1 = static_cast<TopoShapeEdgePy*>(edge1)->getTopoShapePtr()->getShape();
TopoDS_Shape shape2 = static_cast<TopoShapeEdgePy*>(edge2)->getTopoShapePtr()->getShape();
Handle(Geom_Plane) hPlane = Handle(Geom_Plane)::DownCast(static_cast<PlanePy*>(plane)->getGeomPlanePtr()->handle());
getChFi2d_FilletAPIPtr()->Init(TopoDS::Edge(shape1), TopoDS::Edge(shape2), hPlane->Pln());
Py_Return;
}

PyErr_SetString(PyExc_TypeError, "Wrong arguments:\n"
"-- init(wire, plane)"
"-- init(edge, edge, plane)\n");
return nullptr;
}

PyObject* ChFi2d_FilletAPIPy::perform(PyObject *args)
{
double radius;
if (!PyArg_ParseTuple(args, "d", &radius))
return nullptr;

try {
bool ok = getChFi2d_FilletAPIPtr()->Perform(radius);
return Py::new_reference_to(Py::Boolean(ok));
}
catch (Standard_Failure& e) {
PyErr_SetString(Base::PyExc_FC_CADKernelError, e.GetMessageString());
return nullptr;
}
}

PyObject* ChFi2d_FilletAPIPy::numberOfResults(PyObject *args)
{
PyObject* pnt;
if (!PyArg_ParseTuple(args, "O!", &Base::VectorPy::Type, &pnt))
return nullptr;

try {
Base::Vector3d* vec = static_cast<Base::VectorPy*>(pnt)->getVectorPtr();
Standard_Integer num = getChFi2d_FilletAPIPtr()->NbResults(gp_Pnt(vec->x, vec->y, vec->z));
return Py::new_reference_to(Py::Long(num));
}
catch (Standard_Failure& e) {
PyErr_SetString(Base::PyExc_FC_CADKernelError, e.GetMessageString());
return nullptr;
}
}

PyObject* ChFi2d_FilletAPIPy::result(PyObject *args)
{
PyObject* pnt;
int solution = -1;
if (!PyArg_ParseTuple(args, "O!|i", &Base::VectorPy::Type, &pnt, &solution))
return nullptr;

Base::Vector3d* vec = static_cast<Base::VectorPy*>(pnt)->getVectorPtr();

try {
TopoDS_Edge theEdge1, theEdge2;
TopoDS_Shape res_edge = getChFi2d_FilletAPIPtr()->Result(Base::convertTo<gp_Pnt>(*vec), theEdge1, theEdge2, solution);

Py::TupleN tuple(Py::asObject(TopoShape(res_edge).getPyObject()),
Py::asObject(TopoShape(theEdge1).getPyObject()),
Py::asObject(TopoShape(theEdge2).getPyObject()));
return Py::new_reference_to(tuple);
}
catch (Standard_Failure& e) {
PyErr_SetString(Base::PyExc_FC_CADKernelError, e.GetMessageString());
return nullptr;
}
}

PyObject *ChFi2d_FilletAPIPy::getCustomAttributes(const char* /*attr*/) const
{
return nullptr;
}

int ChFi2d_FilletAPIPy::setCustomAttributes(const char* /*attr*/, PyObject* /*obj*/)
{
return 0;
}
2 changes: 1 addition & 1 deletion src/Mod/Part/App/ChFi2d_FilletAlgoPy.xml
Expand Up @@ -3,7 +3,7 @@
<PythonExport
Father="PyObjectBase"
Name="ChFi2d_FilletAlgoPy"
PythonName="Part.ChFi2d_FilletAlgo"
PythonName="Part.ChFi2d.FilletAlgo"
Twin="ChFi2d_FilletAlgo"
TwinPointer="ChFi2d_FilletAlgo"
Include="ChFi2d_FilletAlgo.hxx"
Expand Down
8 changes: 4 additions & 4 deletions src/Mod/Part/App/ChFi2d_FilletAlgoPyImp.cpp
Expand Up @@ -78,16 +78,16 @@ int ChFi2d_FilletAlgoPy::PyInit(PyObject* args, PyObject* /*kwd*/)
}

PyErr_SetString(PyExc_TypeError, "Wrong arguments:\n"
"-- ChFi2d_FilletAlgo()\n"
"-- ChFi2d_FilletAlgo(wire, plane)"
"-- ChFi2d_FilletAlgo(edge, edge, plane)\n");
"-- FilletAlgo()\n"
"-- FilletAlgo(wire, plane)"
"-- FilletAlgo(edge, edge, plane)\n");
return -1;
}

// returns a string which represents the object e.g. when printed in python
std::string ChFi2d_FilletAlgoPy::representation() const
{
return std::string("<ChFi2d_FilletAlgo object>");
return std::string("<FilletAlgo object>");
}

PyObject* ChFi2d_FilletAlgoPy::init(PyObject *args)
Expand Down
8 changes: 4 additions & 4 deletions src/Mod/Part/TestPartApp.py
Expand Up @@ -240,9 +240,9 @@ def testChFi2d_FilletAlgo(self):
pln = Part.Plane()

with self.assertRaises(TypeError):
alg = Part.ChFi2d_FilletAlgo(pln)
alg = Part.ChFi2d.FilletAlgo(pln)

alg = Part.ChFi2d_FilletAlgo()
alg = Part.ChFi2d.FilletAlgo()
with self.assertRaises(TypeError):
alg.init()

Expand All @@ -254,11 +254,11 @@ def testChFi2d_FilletAlgo(self):
with self.assertRaises(TypeError):
alg.perform()

alg = Part.ChFi2d_FilletAlgo(wire, pln)
alg = Part.ChFi2d.FilletAlgo(wire, pln)
alg.init(edge1, edge2, pln)
alg.init(wire, pln)

alg = Part.ChFi2d_FilletAlgo(edge1, edge2, pln)
alg = Part.ChFi2d.FilletAlgo(edge1, edge2, pln)
alg.perform(1.0)

with self.assertRaises(TypeError):
Expand Down

0 comments on commit b07be87

Please sign in to comment.