feat: add plotting options to show_prism_layers

mdtanker · Feb 22, 2024 · 8c1874e · 8c1874e
1 parent 16bf313
commit 8c1874e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/invert4geom/plotting.py b/src/invert4geom/plotting.py
@@ -894,6 +894,7 @@ def show_prism_layers(
                 style=kwargs.get("style", "surface"),
                 show_edges=kwargs.get("show_edges", False),
                 opacity=trans,
+                scalar_bar_args=kwargs.get("scalar_bar_args", None),
             )
         else:
             plotter.add_mesh(
@@ -906,6 +907,7 @@ def show_prism_layers(
                 show_edges=kwargs.get("show_edges", False),
                 log_scale=kwargs.get("log_scale", True),
                 opacity=trans,
+                scalar_bar_args=kwargs.get("scalar_bar_args", None),
             )
         plotter.set_scale(
             zscale=kwargs.get("zscale", 75)
@@ -918,6 +920,7 @@ def show_prism_layers(
     # Add a ceiling light
     add_light(plotter, prisms[i])  # pylint: disable=undefined-loop-variable
 
-    plotter.show_axes()
+    if kwargs.get("show_axes", True):
+        plotter.show_axes()
 
     plotter.show(jupyter_backend=kwargs.get("backend", "client"))