Example plot of triangles, edges and vertices with mayavi

1 files changed, 25 insertions, 0 deletions

diff --git a/src/python/example/plot_simplex_tree_dim012.py b/src/python/example/plot_simplex_tree_dim012.py
new file mode 100755
index 00000000..e6a1a4e0
--- /dev/null
+++ b/src/python/example/plot_simplex_tree_dim012.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+import numpy as np
+import gudhi
+
+# Coordinates of the points
+points=np.array([[0,0,0],[1,0,0],[0,1,0],[0,0,1],[1,1,1],[1,1,0],[0,1,1]])
+# Build the simplicial complex with a tetrahedon, an edge and an isolated vertex
+cplx=gudhi.SimplexTree()
+cplx.insert([1,2,3,5])
+cplx.insert([4,6])
+cplx.insert([0])
+
+from mayavi import mlab
+# Plot triangles
+triangles = np.array([s[0] for s in cplx.get_skeleton(2) if len(s[0])==3])
+mlab.triangular_mesh(points[:,0], points[:,1], points[:,2], triangles);
+# Plot edges
+for s in cplx.get_skeleton(1):
+    e = s[0]
+    if len(e) == 2:
+        pts = points[[e[0],e[1]]]
+        mlab.plot3d(pts[:,0],pts[:,1],pts[:,2],tube_radius=None)
+# Plot points
+mlab.points3d(points[:,0], points[:,1], points[:,2])
+mlab.show()