Expose separatrix kwargs setting in animate_poloidal

boutproject · May 28, 2024 · ba35f46 · ba35f46
1 parent afbeaf0
commit ba35f46
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/xbout/plotting/animate.py b/xbout/plotting/animate.py
@@ -97,6 +97,7 @@ def animate_poloidal(
     cax=None,
     animate_over=None,
     separatrix=True,
+    separatrix_kwargs=dict(),
     targets=True,
     add_limiter_hatching=True,
     cmap=None,
@@ -130,6 +131,8 @@ def animate_poloidal(
         Dimension over which to animate, defaults to the time dimension
     separatrix : bool, optional
         Add dashed lines showing separatrices
+    separatrix_kwargs : dict, optional
+        Options to pass to the separatrix plotter (e.g. line color)
     targets : bool, optional
         Draw solid lines at the target surfaces
     add_limiter_hatching : bool, optional
@@ -277,7 +280,7 @@ def animate_poloidal(
         targets = False
 
     if separatrix:
-        plot_separatrices(da_regions, ax, x=x, y=y)
+        plot_separatrices(da_regions, ax, x=x, y=y, **separatrix_kwargs)
 
     if targets:
         plot_targets(da_regions, ax, x=x, y=y, hatching=add_limiter_hatching)