update latest examples

Author: tomvanmele

docs/examples/example_booleans.py

@@ -26,7 +26,6 @@ def boolean_difference():
     V, F = boolean_difference_mesh_mesh(A, B)
     shape = Polyhedron(V.tolist(), F.tolist())
     shape = shape.to_mesh()
-
     return shape
 
 
@@ -36,7 +35,6 @@ def boolean_intersection():
     V, F = boolean_intersection_mesh_mesh(A, B)
     shape = Polyhedron(V.tolist(), F.tolist())
     shape = shape.to_mesh()
-
     return shape
 
 
@@ -46,7 +44,6 @@ def boolean_union():
     V, F = boolean_union_mesh_mesh(A, B)
     shape = Polyhedron(V.tolist(), F.tolist())
     shape = shape.to_mesh()
-
     return shape
 
 
@@ -58,33 +55,37 @@ def split():
     return mesh.exploded()
 
 
-def main():
-    """Compute the boolean difference, intersection, union, and split of two triangle meshes."""
-    difference = boolean_difference()
-    intersection = boolean_intersection()
-    union = boolean_union()
-    splits = split()
-    return difference, intersection, union, splits
+# =============================================================================
+# Booleans
+# =============================================================================
+
+difference = boolean_difference()
+intersection = boolean_intersection()
+union = boolean_union()
+splits = split()
 
+# =============================================================================
+# Spread
+# =============================================================================
 
-difference, intersection, union, splits = main()
+difference.transform(Translation.from_vector([-6, 0, 0]))
+intersection.transform(Translation.from_vector([-2, 0, 0]))
+union.transform(Translation.from_vector([2, 0, 0]))
 
+for m in splits:
+    m.transform(Translation.from_vector([6, 0, 0]))
 
 # ==============================================================================
 # Visualize
 # ==============================================================================
 
 viewer = Viewer()
 
-difference.transform(Translation.from_vector([-6, 0, 0]))
-intersection.transform(Translation.from_vector([-2, 0, 0]))
-union.transform(Translation.from_vector([2, 0, 0]))
-for m in splits:
-    m.transform(Translation.from_vector([6, 0, 0]))
-
 viewer.scene.add(difference, lineswidth=1, show_points=False)
 viewer.scene.add(intersection, lineswidth=1, show_points=False)
 viewer.scene.add(union, lineswidth=1, show_points=False)
-viewer.scene.add(splits, lineswidth=1, show_points=False)
+
+group = viewer.scene.add_group(name="Splits")
+group.add_from_list(splits, lineswidth=1, show_points=False)
 
 viewer.show()

docs/examples/example_geodesics.py

@@ -1,33 +1,20 @@
 from pathlib import Path
 
 import numpy as np
-from compas.colors import Color, ColorMap
+from compas.colors import Color
+from compas.colors import ColorMap
 from compas.datastructures import Mesh
-from compas.geometry import Box, Point, Polyline, Translation
+from compas.geometry import Box
+from compas.geometry import Point
+from compas.geometry import Polyline
+from compas.geometry import Translation
 from compas_viewer import Viewer
 from compas_viewer.config import Config
 
-from compas_cgal.geodesics import (
-    HeatGeodesicSolver,
-    geodesic_isolines,
-    geodesic_isolines_split,
-    heat_geodesic_distances,
-)
-
-# Load mesh
-FILE = Path(__file__).parent.parent.parent / "data" / "elephant.off"
-mesh = Mesh.from_off(FILE)
-mesh.quads_to_triangles()
-V, F = mesh.to_vertices_and_faces()
-V_np = np.array(V)
-
-# Config
-X_OFF, Y_OFF = 0.75, 1.0
-ISOVALUES = [i/50 for i in range(50)]
-SPLIT_ISOVALUES = [i/20 for i in range(20)]
-COLORS = [Color.red(), Color.orange(), Color.yellow(), Color.green(),
-          Color.cyan(), Color.blue(), Color.purple(), Color.magenta()]
-cmap = ColorMap.from_two_colors(Color.blue(), Color.red())
+from compas_cgal.geodesics import HeatGeodesicSolver
+from compas_cgal.geodesics import geodesic_isolines
+from compas_cgal.geodesics import geodesic_isolines_split
+from compas_cgal.geodesics import heat_geodesic_distances
 
 
 def make_mesh(V, F, offset):
@@ -61,41 +48,98 @@ def make_source_points(sources, offset):
     return [Point(V[s][0] + offset[0], V[s][1] + offset[1], V[s][2]) for s in sources]
 
 
-# Row 1: Single source
+# =============================================================================
+# Load mesh
+# =============================================================================
+
+FILE = Path(__file__).parent.parent.parent / "data" / "elephant.off"
+mesh = Mesh.from_off(FILE)
+mesh.quads_to_triangles()
+V, F = mesh.to_vertices_and_faces()
+V_np = np.array(V)
+
+# =============================================================================
+# Config
+# =============================================================================
+
+X_OFF, Y_OFF = 0.75, 1.0
+ISOVALUES = [i / 50 for i in range(50)]
+SPLIT_ISOVALUES = [i / 20 for i in range(20)]
+COLORS = [
+    Color.red(),
+    Color.orange(),
+    Color.yellow(),
+    Color.green(),
+    Color.cyan(),
+    Color.blue(),
+    Color.purple(),
+    Color.magenta(),
+]
+cmap = ColorMap.from_two_colors(Color.blue(), Color.red())
+
+# =============================================================================
+# Single sources
+# =============================================================================
+
 src1 = [0]
 dist1 = heat_geodesic_distances((V, F), src1)
 
-# Row 2: Multiple sources (bbox corners)
+# =============================================================================
+# Multiple sources
+# =============================================================================
+
 solver = HeatGeodesicSolver((V, F))
 bbox = Box.from_points(V_np)
 src2 = list(dict.fromkeys([int(np.argmin(np.linalg.norm(V_np - c, axis=1))) for c in bbox.vertices]))
 dist2 = solver.solve(src2)
 
-# Viewer
+# =============================================================================
+# Viz
+# =============================================================================
+
 config = Config()
 config.camera.target = [X_OFF, -Y_OFF / 2, 0]
 config.camera.position = [X_OFF, -2.0, 0.8]
+
 viewer = Viewer(config=config)
 
 # Row 1: Single Source
+
 g1 = viewer.scene.add_group("Single Source")
-g1.add(make_mesh(V, F, (0, 0)), use_vertexcolors=True, vertexcolor=make_vertex_colors(dist1), show_lines=False)
+
+g1.add(
+    make_mesh(V, F, (0, 0)),
+    use_vertexcolors=True,
+    vertexcolor=make_vertex_colors(dist1),
+    show_lines=False,
+)
+
 for pt in make_source_points(src1, (0, 0)):
     g1.add(pt, pointcolor=Color.black(), pointsize=20)
-# g1.add(make_mesh(V, F, (X_OFF, 0)), facecolor=Color.grey(), show_lines=True)
+
 for pl in make_isolines(src1, (X_OFF, 0)):
     g1.add(pl, linecolor=Color.red(), lineswidth=5)
+
 for m, c in make_split_meshes(src1, (2 * X_OFF, 0)):
     g1.add(m, facecolor=c, show_lines=False)
 
 # Row 2: Multiple Sources
+
 g2 = viewer.scene.add_group("Multiple Sources")
-g2.add(make_mesh(V, F, (0, -Y_OFF)), use_vertexcolors=True, vertexcolor=make_vertex_colors(dist2), show_lines=False)
+
+g2.add(
+    make_mesh(V, F, (0, -Y_OFF)),
+    use_vertexcolors=True,
+    vertexcolor=make_vertex_colors(dist2),
+    show_lines=False,
+)
+
 for pt in make_source_points(src2, (0, -Y_OFF)):
     g2.add(pt, pointcolor=Color.black(), pointsize=20)
-# g2.add(make_mesh(V, F, (X_OFF, -Y_OFF)), facecolor=Color.grey(), show_lines=True)
+
 for pl in make_isolines(src2, (X_OFF, -Y_OFF)):
     g2.add(pl, linecolor=Color.red(), lineswidth=5)
+
 for m, c in make_split_meshes(src2, (2 * X_OFF, -Y_OFF)):
     g2.add(m, facecolor=c, show_lines=False)