Skip to content

Commit

Permalink
fix: adding mpl syntax for 3d view points
Browse files Browse the repository at this point in the history
adding c argument for colouring points
  • Loading branch information
Lachlan Grose committed Dec 7, 2021
1 parent d94f79b commit cc7788f
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion LoopStructural/visualisation/lavavu.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,7 @@ def _add_surface(
vmax = kwargs.get("vmax", np.nanmax(surfaceval))
surf.colourmap(cmap, range=(vmin, vmax))

def _add_points(self, points, name, value=None, **kwargs):
def _add_points(self, points, name, value=None, c = None, **kwargs):
"""Virtual function to be overwritten by subclasses for adding points to the viewer
Parameters
Expand All @@ -126,6 +126,8 @@ def _add_points(self, points, name, value=None, **kwargs):
name = "Unnamed points"
p = self.lv.points(name, **kwargs)
p.vertices(points)
if value is None and c is not None:
value = c
if value is not None:
p.values(value, "v")
p["colourby"] = "v"
Expand Down

0 comments on commit cc7788f

Please sign in to comment.