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

Part: expose ChFi2d_FilletAlgo to Python

Browse Source
This commit is contained in:
wmayer
2022-05-28 10:37:04 +02:00
parent 4c2972cb23
commit bd853e5ef2
5 changed files with 280 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_FilletAlgoPy.h>
#include "Mod/Part/App/CirclePy.h"
#include "Mod/Part/App/ConePy.h"
#include "Mod/Part/App/ConicPy.h"
@@ -290,6 +291,7 @@ 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());

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

@@ -91,6 +91,7 @@ generate_from_xml(TopoShapeVertexPy)
generate_from_xml(TopoShapeWirePy)
generate_from_xml(BRepOffsetAPI_MakePipeShellPy)
generate_from_xml(BRepOffsetAPI_MakeFillingPy)
generate_from_xml(ChFi2d_FilletAlgoPy)
generate_from_xml(PrecisionPy)
# make sure to create the directory at configure time
@@ -318,6 +319,8 @@ SET(Python_SRCS
BRepOffsetAPI_MakePipeShellPyImp.cpp
BRepOffsetAPI_MakeFillingPy.xml
BRepOffsetAPI_MakeFillingPyImp.cpp
ChFi2d_FilletAlgoPy.xml
ChFi2d_FilletAlgoPyImp.cpp
PrecisionPy.xml
PrecisionPyImp.cpp
PartPyCXX.cpp

44
src/Mod/Part/App/ChFi2d_FilletAlgoPy.xml Normal file
View File

@@ -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_FilletAlgoPy"
PythonName="Part.ChFi2d_FilletAlgo"
Twin="ChFi2d_FilletAlgo"
TwinPointer="ChFi2d_FilletAlgo"
Include="ChFi2d_FilletAlgo.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
src/Mod/Part/App/ChFi2d_FilletAlgoPyImp.cpp Normal file
View File

@@ -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_FilletAlgoPy.h>
#include <Mod/Part/App/ChFi2d_FilletAlgoPy.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_FilletAlgoPy::PyMake(struct _typeobject *, PyObject *, PyObject *) // Python wrapper
{
// create a new instance of ChFi2d_FilletAlgoPy and the Twin object
return new ChFi2d_FilletAlgoPy(new ChFi2d_FilletAlgo);
}
// constructor method
int ChFi2d_FilletAlgoPy::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_FilletAlgoPtr()->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_FilletAlgoPtr()->Init(TopoDS::Edge(shape1), TopoDS::Edge(shape2), hPlane->Pln());
return 0;
}
PyErr_SetString(PyExc_TypeError, "Wrong arguments:\n"
"-- ChFi2d_FilletAlgo()\n"
"-- ChFi2d_FilletAlgo(wire, plane)"
"-- ChFi2d_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>");
}
PyObject* ChFi2d_FilletAlgoPy::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_FilletAlgoPtr()->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_FilletAlgoPtr()->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_FilletAlgoPy::perform(PyObject *args)
{
double radius;
if (!PyArg_ParseTuple(args, "d", &radius))
return nullptr;
try {
bool ok = getChFi2d_FilletAlgoPtr()->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_FilletAlgoPy::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_FilletAlgoPtr()->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_FilletAlgoPy::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_FilletAlgoPtr()->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_FilletAlgoPy::getCustomAttributes(const char* /*attr*/) const
{
return nullptr;
}
int ChFi2d_FilletAlgoPy::setCustomAttributes(const char* /*attr*/, PyObject* /*obj*/)
{
return 0;
}

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

@@ -230,3 +230,46 @@ class PartTestCone(unittest.TestCase):
self.assertAlmostEqual(v1.getAngle(w1), 0)
self.assertAlmostEqual(v2.getAngle(w2), 0)
self.assertAlmostEqual(v3.getAngle(w3), 0)
class PartTestFilletAlgo(unittest.TestCase):
def testChFi2d_FilletAlgo(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_FilletAlgo(pln)
alg = Part.ChFi2d_FilletAlgo()
with self.assertRaises(TypeError):
alg.init()
print (alg)
# Test without shape
with self.assertRaises(Base.CADKernelError):
alg.perform(1)
with self.assertRaises(TypeError):
alg.perform()
alg = Part.ChFi2d_FilletAlgo(wire, pln)
alg.init(edge1, edge2, pln)
alg.init(wire, pln)
alg = Part.ChFi2d_FilletAlgo(edge1, edge2, pln)
alg.perform(1.0)
with self.assertRaises(TypeError):
alg.numberOfResults()
with self.assertRaises(TypeError):
alg.result(1)
self.assertEqual(alg.numberOfResults(Base.Vector(0,10,0)), 1)
result = alg.result(Base.Vector(0,10,0))
curve = result[0].Curve
self.assertEqual(type(curve), Part.Circle)
self.assertEqual(curve.Axis, pln.Axis)
self.assertEqual(curve.Radius, 1.0)