Poincare options

src/simsopt/field/tracing.py

@@ -799,7 +799,8 @@
     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:
 
@@ -835,9 +836,9 @@
         if col == 1:
             axs[row, col].set_yticklabels([])
         if xlims is not None:
             axs[row, col].set_xlim(xlims)
         if ylims is not None:
             axs[row, col].set_ylim(ylims)
         for j in range(len(fieldlines_phi_hits)):
             lost = fieldlines_phi_hits[j][-1, 1] < 0
             if mark_lost:
@@ -846,17 +847,17 @@
             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)
 
         # if passed a surface, plot the plasma surface outline
         if surf is not None:
             cross_section = surf.cross_section(phi=phis[i])
             r_interp = np.sqrt(cross_section[:, 0] ** 2 + cross_section[:, 1] ** 2)
             z_interp = cross_section[:, 2]
             axs[row, col].plot(r_interp, z_interp, linewidth=1, c='k')
 
     plt.tight_layout()
     plt.savefig(filename, dpi=dpi)