kindred/create
1
1
Fork 0
Code Issues 30 Pull Requests Actions Packages Projects 9 Releases 2 Wiki Activity

Part: expose ChFi2d_AnaFilletAlgo to Python

Browse Source
This commit is contained in:
wmayer
2022-05-29 11:31:43 +02:00
parent b47f8d2605
commit f94ab3ec84
5 changed files with 248 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/Mod/Part/App/AppPart.cpp
View File

@@ -51,6 +51,7 @@
#include "Mod/Part/App/BRepOffsetAPI_MakePipeShellPy.h"
#include "Mod/Part/App/BSplineCurvePy.h"
#include "Mod/Part/App/BSplineSurfacePy.h"
#include <Mod/Part/App/ChFi2d/ChFi2d_AnaFilletAlgoPy.h>
#include <Mod/Part/App/ChFi2d/ChFi2d_FilletAlgoPy.h>
#include <Mod/Part/App/ChFi2d/ChFi2d_FilletAPIPy.h>
#include "Mod/Part/App/CirclePy.h"
@@ -341,6 +342,7 @@ PyMOD_INIT_FUNC(Part)
// ChFi2d sub-module
PyObject* chFi2d(module.getAttr("ChFi2d").ptr());
Base::Interpreter().addType(&Part::ChFi2d_AnaFilletAlgoPy::Type, chFi2d, "AnaFilletAlgo");
Base::Interpreter().addType(&Part::ChFi2d_FilletAlgoPy::Type, chFi2d, "FilletAlgo");
Base::Interpreter().addType(&Part::ChFi2d_FilletAPIPy::Type, chFi2d, "FilletAPI");

3
src/Mod/Part/App/CMakeLists.txt
View File

@@ -103,6 +103,7 @@ file(MAKE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/ShapeUpgrade)
generate_from_xml(BRepFeat/MakePrismPy)
generate_from_xml(ChFi2d/ChFi2d_AnaFilletAlgoPy)
generate_from_xml(ChFi2d/ChFi2d_FilletAlgoPy)
generate_from_xml(ChFi2d/ChFi2d_FilletAPIPy)
@@ -338,6 +339,8 @@ SOURCE_GROUP("BRepFeat" FILES ${BRepFeatPy_SRCS})
# ChFi2d wrappers
SET(ChFi2dPy_SRCS
ChFi2d/ChFi2d_AnaFilletAlgoPy.xml
ChFi2d/ChFi2d_AnaFilletAlgoPyImp.cpp
ChFi2d/ChFi2d_FilletAlgoPy.xml
ChFi2d/ChFi2d_FilletAlgoPyImp.cpp
ChFi2d/ChFi2d_FilletAPIPy.xml

40
src/Mod/Part/App/ChFi2d/ChFi2d_AnaFilletAlgoPy.xml Normal file
View File

@@ -0,0 +1,40 @@
<?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_AnaFilletAlgoPy"
PythonName="Part.ChFi2d.AnaFilletAlgo"
Twin="ChFi2d_AnaFilletAlgo"
TwinPointer="ChFi2d_AnaFilletAlgo"
Include="ChFi2d_AnaFilletAlgo.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>An analytical algorithm for calculation of the fillets.
It is implemented for segments and arcs of circle only.</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="result">
<Documentation>
<UserDocu>result()
Returns result (fillet edge, modified edge1, modified edge2)</UserDocu>
</Documentation>
</Methode>
</PythonExport>
</GenerateModel>

167
src/Mod/Part/App/ChFi2d/ChFi2d_AnaFilletAlgoPyImp.cpp Normal file
View File

@@ -0,0 +1,167 @@
/***************************************************************************
* 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/ChFi2d_AnaFilletAlgoPy.h>
#include <Mod/Part/App/ChFi2d/ChFi2d_AnaFilletAlgoPy.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_AnaFilletAlgoPy::PyMake(struct _typeobject *, PyObject *, PyObject *) // Python wrapper
{
// create a new instance of ChFi2d_AnaFilletAlgoPy and the Twin object
return new ChFi2d_AnaFilletAlgoPy(new ChFi2d_AnaFilletAlgo);
}
// constructor method
int ChFi2d_AnaFilletAlgoPy::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_AnaFilletAlgoPtr()->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_AnaFilletAlgoPtr()->Init(TopoDS::Edge(shape1), TopoDS::Edge(shape2), hPlane->Pln());
return 0;
}
PyErr_SetString(PyExc_TypeError, "Wrong arguments:\n"
"-- AnaFilletAlgo()\n"
"-- AnaFilletAlgo(wire, plane)"
"-- AnaFilletAlgo(edge, edge, plane)\n");
return -1;
}
// returns a string which represents the object e.g. when printed in python
std::string ChFi2d_AnaFilletAlgoPy::representation() const
{
return std::string("<AnaFilletAlgo object>");
}
PyObject* ChFi2d_AnaFilletAlgoPy::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_AnaFilletAlgoPtr()->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_AnaFilletAlgoPtr()->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_AnaFilletAlgoPy::perform(PyObject *args)
{
double radius;
if (!PyArg_ParseTuple(args, "d", &radius))
return nullptr;
try {
bool ok = getChFi2d_AnaFilletAlgoPtr()->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_AnaFilletAlgoPy::result(PyObject *args)
{
if (!PyArg_ParseTuple(args, ""))
return nullptr;
try {
TopoDS_Edge theEdge1, theEdge2;
TopoDS_Shape res_edge = getChFi2d_AnaFilletAlgoPtr()->Result(theEdge1, theEdge2);
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_AnaFilletAlgoPy::getCustomAttributes(const char* /*attr*/) const
{
return nullptr;
}
int ChFi2d_AnaFilletAlgoPy::setCustomAttributes(const char* /*attr*/, PyObject* /*obj*/)
{
return 0;
}

36
src/Mod/Part/TestPartApp.py
View File

@@ -273,3 +273,39 @@ class PartTestFilletAlgo(unittest.TestCase):
self.assertEqual(type(curve), Part.Circle)
self.assertEqual(curve.Axis, pln.Axis)
self.assertEqual(curve.Radius, 1.0)
def testChFi2d_AnaFilletAlgo(self):
v = FreeCAD.Vector
edge1 = Part.makeLine(v(0,0,0), v(0,10,0))
edge2 = Part.makeLine(v(0,10,0), v(10,10,0))
wire = Part.Wire([edge1, edge2])
pln = Part.Plane()
with self.assertRaises(TypeError):
alg = Part.ChFi2d.AnaFilletAlgo(pln)
alg = Part.ChFi2d.AnaFilletAlgo()
with self.assertRaises(TypeError):
alg.init()
print (alg)
# Test without shape
self.assertFalse(alg.perform(1))
with self.assertRaises(TypeError):
alg.perform()
alg = Part.ChFi2d.AnaFilletAlgo(wire, pln)
alg.init(edge1, edge2, pln)
alg.init(wire, pln)
alg = Part.ChFi2d.AnaFilletAlgo(edge1, edge2, pln)
alg.perform(1.0)
with self.assertRaises(TypeError):
alg.result(1)
result = alg.result()
curve = result[0].Curve
self.assertEqual(type(curve), Part.Circle)
self.assertEqual(curve.Radius, 1.0)