From 4be052af48b9bb296b7ed04b84995491995e57ce Mon Sep 17 00:00:00 2001 From: jriegel Date: Sun, 5 Jan 2014 22:31:04 +0100 Subject: [PATCH] StepShape class and python binding for handling STep-Shape transfers. --- src/Mod/Import/App/AppImport.cpp | 13 +++- src/Mod/Import/App/CMakeLists.txt | 6 ++ src/Mod/Import/App/StepShape.cpp | 102 ++++++++++++++++++++++++++ src/Mod/Import/App/StepShape.h | 50 +++++++++++++ src/Mod/Import/App/StepShapePy.xml | 30 ++++++++ src/Mod/Import/App/StepShapePyImp.cpp | 58 +++++++++++++++ 6 files changed, 258 insertions(+), 1 deletion(-) create mode 100644 src/Mod/Import/App/StepShape.cpp create mode 100644 src/Mod/Import/App/StepShape.h create mode 100644 src/Mod/Import/App/StepShapePy.xml create mode 100644 src/Mod/Import/App/StepShapePyImp.cpp diff --git a/src/Mod/Import/App/AppImport.cpp b/src/Mod/Import/App/AppImport.cpp index c21b306b2..cd5dd6ab7 100644 --- a/src/Mod/Import/App/AppImport.cpp +++ b/src/Mod/Import/App/AppImport.cpp @@ -29,15 +29,20 @@ #include #include +#include "StepShapePy.h" +#include "StepShape.h" /* registration table */ extern struct PyMethodDef Import_Import_methods[]; +PyDoc_STRVAR(module_doc, +"This module is about import/export files formates.\n" +"\n"); extern "C" { void ImportExport initImport() { - (void) Py_InitModule("Import", Import_Import_methods); /* mod name, table ptr */ + PyObject* importModule = Py_InitModule3("Import", Import_Import_methods, module_doc); /* mod name, table ptr */ try { Base::Interpreter().loadModule("Part"); @@ -47,6 +52,12 @@ void ImportExport initImport() return; } + // add mesh elements + Base::Interpreter().addType(&Import::StepShapePy ::Type,importModule,"StepShape"); + + // init Type system + //Import::StepShape ::init(); + Base::Console().Log("Loading Import module... done\n"); } diff --git a/src/Mod/Import/App/CMakeLists.txt b/src/Mod/Import/App/CMakeLists.txt index 4aa081e91..e8e063aa8 100644 --- a/src/Mod/Import/App/CMakeLists.txt +++ b/src/Mod/Import/App/CMakeLists.txt @@ -27,6 +27,10 @@ SET(Import_SRCS AppImportPy.cpp ImportOCAF.cpp ImportOCAF.h + StepShapePy.xml + StepShape.h + StepShape.cpp + StepShapePyImp.cpp PreCompiled.cpp PreCompiled.h ) @@ -51,6 +55,8 @@ SET(SCL_Resources ) SOURCE_GROUP("SCL" FILES ${SCL_Resources}) +generate_from_xml(StepShapePy) + add_library(Import SHARED ${Import_SRCS}) target_link_libraries(Import ${Import_LIBS}) diff --git a/src/Mod/Import/App/StepShape.cpp b/src/Mod/Import/App/StepShape.cpp new file mode 100644 index 000000000..67092a597 --- /dev/null +++ b/src/Mod/Import/App/StepShape.cpp @@ -0,0 +1,102 @@ +/*************************************************************************** + * (c) Jürgen Riegel (juergen.riegel@web.de) 2014 * + * * + * 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 +#endif + +#include "StepShape.h" + +# include +# include +# include +# include +# include +# include + +# include +# include + +using namespace Import; + +StepShape::StepShape(const char* fileName) +{ +} + + +StepShape::~StepShape() +{ +} + +int StepShape::read(const char* fileName) +{ + STEPControl_Reader aReader; + + Base::FileInfo fi(fileName); + + if (!fi.exists()) { + std::stringstream str; + str << "File '" << fileName << "' does not exist!"; + throw Base::Exception(str.str().c_str()); + } + + if (aReader.ReadFile((Standard_CString)fileName) != IFSelect_RetDone) { + throw Base::Exception("Cannot open STEP file"); + } + + //Standard_Integer ic = Interface_Static::IVal("read.precision.mode"); + //Standard_Real rp = Interface_Static::RVal("read.maxprecision.val"); + //Standard_Integer ic = Interface_Static::IVal("read.maxprecision.mode"); + //Standard_Integer mv = Interface_Static::IVal("read.stdsameparameter.mode"); + //Standard_Integer rp = Interface_Static::IVal("read.surfacecurve.mode"); + //Standard_Real era = Interface_Static::RVal("read.encoderegularity.angle"); + //Standard_Integer ic = Interface_Static::IVal("read.step.product.mode"); + //Standard_Integer ic = Interface_Static::IVal("read.step.product.context"); + //Standard_Integer ic = Interface_Static::IVal("read.step.shape.repr"); + //Standard_Integer ic = Interface_Static::IVal("read.step.assembly.level"); + //Standard_Integer ic = Interface_Static::IVal("read.step.shape.relationship"); + //Standard_Integer ic = Interface_Static::IVal("read.step.shape.aspect"); + + Handle(TColStd_HSequenceOfTransient) list = aReader.GiveList(); + + //Use method StepData_StepModel::NextNumberForLabel to find its rank with the following: + Standard_CString label = "#..."; + Handle_StepData_StepModel model = aReader.StepModel(); + //rank = model->NextNumberForLabe(label, 0, Standard_False); + + Handle_Message_PrinterOStream mstr = new Message_PrinterOStream(); + Handle_Message_Messenger msg = new Message_Messenger(mstr); + + std::cout << "dump of step header:" << std::endl; + + model->DumpHeader(msg); + + for(int nent=1;nent<=model->NbEntities();nent++) { + Handle(Standard_Transient) entity=model->Entity(nent); + + std::cout << "label entity " << nent << ":" ; + model->PrintLabel(entity,msg); + std::cout << ";"<< entity->DynamicType()->Name() << std::endl; + } +} diff --git a/src/Mod/Import/App/StepShape.h b/src/Mod/Import/App/StepShape.h new file mode 100644 index 000000000..3e64cf306 --- /dev/null +++ b/src/Mod/Import/App/StepShape.h @@ -0,0 +1,50 @@ +/*************************************************************************** + * (c) Jürgen Riegel (juergen.riegel@web.de) 2014 * + * * + * 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 * + * * + ***************************************************************************/ + + +#ifndef IMPORT_STEPSHAPE_H +#define IMPORT_STEPSHAPE_H + + + +namespace Import +{ + +/** The StepShape helper class + * The MeshFacet class provides an interface for the MeshFacetPy class for + * convenient access to the Mesh data structure. This class should not be used + * for programming algorithms in C++. Use Mesh Core classes instead! + */ +class ImportExport StepShape +{ +public: + StepShape(const char* fileName=""); + ~StepShape(); + + int read(const char* fileName); + +}; + +} // namespace Import + + +#endif // IMPORT_STEPSHAPE_H diff --git a/src/Mod/Import/App/StepShapePy.xml b/src/Mod/Import/App/StepShapePy.xml new file mode 100644 index 000000000..ec17cd991 --- /dev/null +++ b/src/Mod/Import/App/StepShapePy.xml @@ -0,0 +1,30 @@ + + + + + + StepShape in a Import + StepShape in Import +This class gives a interface to retrive TopoShapes out of an loaded STEP file of any kind. + + + + + method read() +Read a STEP file into memory and makeit accessably + + + + + + \ No newline at end of file diff --git a/src/Mod/Import/App/StepShapePyImp.cpp b/src/Mod/Import/App/StepShapePyImp.cpp new file mode 100644 index 000000000..bc0587715 --- /dev/null +++ b/src/Mod/Import/App/StepShapePyImp.cpp @@ -0,0 +1,58 @@ + +#include "PreCompiled.h" + +#include "Mod/Import/App/StepShape.h" + +// inclusion of the generated files (generated out of StepShapePy.xml) +#include "StepShapePy.h" +#include "StepShapePy.cpp" + +using namespace Import; + +// returns a string which represents the object e.g. when printed in python +std::string StepShapePy::representation(void) const +{ + return std::string(""); +} + +PyObject *StepShapePy::PyMake(struct _typeobject *, PyObject *, PyObject *) // Python wrapper +{ + // create a new instance of StepShapePy and the Twin object + return new StepShapePy(new StepShape); +} + +// constructor method +int StepShapePy::PyInit(PyObject* args, PyObject* /*kwd*/) +{ + char* fileName; + if (PyArg_ParseTuple(args, "s", &fileName)) { + getStepShapePtr()->read(fileName); + return 0; + } + + PyErr_SetString(PyExc_TypeError, "StepShape needs a file name\n"); + return -1; +} + + +PyObject* StepShapePy::read(PyObject * /*args*/) +{ + PyErr_SetString(PyExc_NotImplementedError, "Not yet implemented"); + return 0; +} + + + + + +PyObject *StepShapePy::getCustomAttributes(const char* /*attr*/) const +{ + return 0; +} + +int StepShapePy::setCustomAttributes(const char* /*attr*/, PyObject* /*obj*/) +{ + return 0; +} + +