ForkMaster/netgen
1
0
Fork 0
mirror of https://github.com/NGSolve/netgen.git synced 2024-11-11 16:49:16 +05:00
Code Issues Packages Projects Releases Wiki Activity

geometry edges in webgui

Browse Source
This commit is contained in:
Matthias Hochsteger 2021-08-13 16:46:49 +02:00
parent ae6b23fffc
commit fb64f0d873
2 changed files with 62 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
libsrc/occ/python_occ_shapes.cpp
View File

@ -56,10 +56,34 @@
using namespace netgen;
void ExtractEdgeData( const TopoDS_Edge & edge, int index, std::vector<double> * p, Box<3> & box )
{
if (BRep_Tool::Degenerated(edge)) return;
Handle(Poly_PolygonOnTriangulation) poly;
Handle(Poly_Triangulation) T;
TopLoc_Location loc;
BRep_Tool::PolygonOnTriangulation(edge, poly, T, loc);
int nbnodes = poly -> NbNodes();
for (int j = 1; j < nbnodes; j++)
{
auto p0 = occ2ng((T -> Nodes())(poly->Nodes()(j)).Transformed(loc));
auto p1 = occ2ng((T -> Nodes())(poly->Nodes()(j+1)).Transformed(loc));
for(auto k : Range(3))
{
p[0].push_back(p0[k]);
p[1].push_back(p1[k]);
}
p[0].push_back(index);
p[1].push_back(index);
box.Add(p0);
box.Add(p1);
}
}
void ExtractFaceData( const TopoDS_Face & face, int index, std::vector<double> * p, Box<3> & box )
{
Handle(Geom_Surface) surf = BRep_Tool::Surface (face);
TopLoc_Location loc;
Handle(Poly_Triangulation) triangulation = BRep_Tool::Triangulation (face, loc);
@ -606,6 +630,31 @@ DLL_HEADER void ExportNgOCCShapes(py::module &m)
names.append("");
index++;
}
std::vector<double> edge_p[2];
py::list edge_names, edge_colors;
index = 0;
for (TopExp_Explorer e(shape, TopAbs_EDGE); e.More(); e.Next())
{
TopoDS_Edge edge = TopoDS::Edge(e.Current());
ExtractEdgeData(edge, index, edge_p, box);
auto & props = OCCGeometry::global_shape_properties[edge.TShape()];
if(props.col)
{
auto & c = *props.col;
edge_colors.append(py::make_tuple(c[0], c[1], c[2]));
}
else
edge_colors.append(py::make_tuple(0.0, 0.0, 0.0));
if(props.name)
{
edge_names.append(*props.name);
}
else
edge_names.append("");
index++;
}
auto center = box.Center();
@ -623,9 +672,8 @@ DLL_HEADER void ExportNgOCCShapes(py::module &m)
data["draw_vol"] = false;
data["draw_surf"] = true;
data["funcdim"] = 0;
data["show_wireframe"] = false;
data["show_wireframe"] = true;
data["show_mesh"] = true;
data["edges"] = py::list{};
data["Bezier_points"] = py::list{};
py::list points;
points.append(p[0]);
@ -638,6 +686,13 @@ DLL_HEADER void ExportNgOCCShapes(py::module &m)
data["autoscale"] = false;
data["colors"] = colors;
data["names"] = names;
py::list edges;
edges.append(edge_p[0]);
edges.append(edge_p[1]);
data["edges"] = edges;
data["edge_names"] = edge_names;
data["edge_colors"] = edge_colors;
return data;
})
;

3
python/webgui.py
View File

@ -22,6 +22,9 @@ class WebGLScene(BaseWebGuiScene):
for i in range(len(bp)):
bp[i] = encodeData(np.array(bp[i]))
ep = d['edges']
for i in range(len(ep)):
ep[i] = encodeData(np.array(ep[i]))
if self.clipping is not None:
d['clipping'] = True