Merge pull request #362 from hiddenSymmetries/poincare_options

Poincare options
hiddenSymmetries · Oct 6, 2023 · 490282b · 490282b
2 parents 085177e + 51b37b4
commit 490282b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/simsopt/field/tracing.py b/src/simsopt/field/tracing.py
@@ -799,7 +799,8 @@ class IterationStoppingCriterion(sopp.IterationStoppingCriterion):
     pass
 
 
-def plot_poincare_data(fieldlines_phi_hits, phis, filename, mark_lost=False, aspect='equal', dpi=300, xlims=None, ylims=None, surf=None):
+def plot_poincare_data(fieldlines_phi_hits, phis, filename, mark_lost=False, aspect='equal', dpi=300, xlims=None, 
+                       ylims=None, surf=None, s=2, marker='o'):
     """
     Create a poincare plot. Usage:
 
@@ -846,7 +847,7 @@ def plot_poincare_data(fieldlines_phi_hits, phis, filename, mark_lost=False, asp
             if data_this_phi.size == 0:
                 continue
             r = np.sqrt(data_this_phi[:, 2]**2+data_this_phi[:, 3]**2)
-            axs[row, col].scatter(r, data_this_phi[:, 4], marker='o', s=2, linewidths=0, c=color)
+            axs[row, col].scatter(r, data_this_phi[:, 4], marker=marker, s=s, linewidths=0, c=color)
 
         plt.rc('axes', axisbelow=True)
         axs[row, col].grid(True, linewidth=0.5)