netgen/libsrc/stlgeom/python_stl.cpp
Christopher Lackner 561cf4e75f add files
2016-12-09 16:34:22 +01:00

50 lines
1.0 KiB
C++

#ifdef NG_PYTHON
#include <../general/ngpython.hpp>
#include <stlgeom.hpp>
using namespace netgen;
namespace netgen
{
extern shared_ptr<NetgenGeometry> ng_geometry;
}
DLL_HEADER void ExportSTL(py::module & m)
{
py::class_<STLGeometry,shared_ptr<STLGeometry>> (m,"STLGeometry")
.def(py::init<>())
.def("Load", FunctionPointer([] (shared_ptr<STLGeometry> self, const string & filename)
{
ifstream ist(filename);
self->Load(ist);
}))
;
m.def("GenerateMesh", FunctionPointer([] (shared_ptr<STLGeometry> geo, MeshingParameters &param)
{
auto mesh = make_shared<Mesh>();
SetGlobalMesh(mesh);
mesh->SetGeometry(geo);
ng_geometry = geo;
try
{
geo->GenerateMesh(mesh,param);
}
catch (NgException ex)
{
cout << "Caught NgException: " << ex.What() << endl;
}
return mesh;
}))
;
}
PYBIND11_PLUGIN(libstl) {
py::module m("stl", "pybind stl");
ExportSTL(m);
return m.ptr();
}
#endif