diff --git a/src/simsopt/field/tracing.py b/src/simsopt/field/tracing.py index d8fe8c3ba..f5ab3d164 100644 --- 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)