ForkMaster/netgen
1
0
Fork 0
mirror of https://github.com/NGSolve/netgen.git synced 2024-09-22 17:52:50 +05:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'stl_util_pyexport' into 'master'

Browse Source 
Export some STL utility functions

See merge request ngsolve/netgen!616
This commit is contained in:
Hochsteger, Matthias 2023-11-22 20:46:50 +01:00
commit 6442b47b99
2 changed files with 60 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
libsrc/stlgeom/python_stl.cpp
View File

@ -223,6 +223,65 @@ NGCORE_API_EXPORT void ExportSTL(py::module & m)
ng_geometry = self;
})
)
.def("GetVicinity", [] (shared_ptr<STLGeometry> self, int node, int size, string type) {
NgArray<int> vic;
int trig;
if(type == "trig")
trig = node;
if(type == "point")
trig = self->TrigPerPoint(node, 1);
self->GetVicinity(trig, size, vic);
auto geo = make_shared<STLGeometry>();
NgArray<STLReadTriangle> trigs;
for(auto i : Range(vic.Size())) {
int trigi = vic[i];
STLReadTriangle t;
Vec<3> normal = self->GetTriangle(trigi).Normal();
Point<3> pts[3];
auto trig = self->GetTriangle(trigi);
for(auto pi : Range(3))
pts[pi] = self->GetPoint(trig[pi]);
trigs.Append(STLReadTriangle(pts, normal));
}
geo->SetSurfaceSTL(true);
geo->InitSTLGeometry(trigs);
return geo;
}, py::arg("node"), py::arg("size"), py::arg("node_type") = "trig")
.def("SmoothDirtyTrigs", [] (shared_ptr<STLGeometry> self, py::kwargs kwargs) {
STLParameters stlparam;
CreateSTLParametersFromKwargs(stlparam, kwargs);
self->SmoothDirtyTrigs(stlparam);
})
.def("GetDirtyTrigs", [] (shared_ptr<STLGeometry> self, py::kwargs kwargs) {
STLParameters stlparam;
CreateSTLParametersFromKwargs(stlparam, kwargs);
self->MarkDirtyTrigs(stlparam);
py::list dirty;
for(auto i : Range(self->GetNT()))
if(self->IsMarkedTrig(i+1))
dirty.append(i);
})
.def("MovePointToMiddle", [] (shared_ptr<STLGeometry> self, int node, int count) {
auto trignr = self->TrigPerPoint(node, 1);
auto trig = self->GetTriangle(trignr);
int point_in_trig = -1;
for(auto i : Range(3))
if(trig[i] == node)
point_in_trig = i;
if(point_in_trig == -1)
throw Exception("Point not found in triangle");
self->SetSelectTrig(trignr);
self->SetNodeOfSelTrig(point_in_trig);
for(auto i : Range(count))
self->MoveSelectedPointToMiddle();
})
.def("Save", &STLGeometry::Save)
;
m.def("LoadSTLGeometry", [] (const string & filename)
{

1
libsrc/stlgeom/stltopology.hpp
View File

@ -327,6 +327,7 @@ public:
void SaveSTLE (const filesystem::path & filename) const; // stores trigs and edges
bool IsSurfaceSTL() const { return surface; }
void SetSurfaceSTL( bool surface_ ) { surface = surface_; }
virtual void DoArchive(Archive& ar)
{