vertex color

hoho/vis.py

@@ -41,7 +41,7 @@ def line(p1, p2, c=(255,0,0), resolution=10, radius=0.05):
          
     return mesh
 
-def show_wf(row, radius=10, show_vertices=False):
+def show_wf(row, radius=10, show_vertices=False, vertex_color=(255,0,0, 255)):
     EDGE_CLASSES = ['eave',
                     'ridge',
                     'step_flashing',
@@ -53,7 +53,9 @@ def show_wf(row, radius=10, show_vertices=False):
                     'transition_line']
     out_meshes = []
     if show_vertices:
-        out_meshes.extend([trimesh.primitives.Sphere(radius=radius+5, center = center, color=(255,0,0)) for center in row['wf_vertices']])
+        out_meshes.extend([trimesh.primitives.Sphere(radius=radius+5, center = center, color=vertex_color) for center in row['wf_vertices']])
+        for m in out_meshes:
+            m.visual.vertex_colors = np.ones_like(m.visual.vertex_colors)*vertex_color
     if 'edge_semantics' not in row:
         print ("Warning: edge semantics is not here, skipping")
         out_meshes.extend([line(a,b, radius=radius, c=(214, 251, 248)) for a,b in np.stack([*row['wf_vertices']])[np.stack(row['wf_edges'])]])