Merge pull request #50 from roussel-ryan/dset-plot-aspect-ratio

dset plot aspect ratio

Author: jp-ga

docs/examples/example_reconstruction_4d.ipynb

@@ -176,6 +176,7 @@ def plot_data(
 
             ax[i].set_title(f"{parameters[i]:.1f}")
             ax[i].set_xlabel("x (mm)")
+            ax[i].set_aspect("equal")
 
         ax[0].set_ylabel("y (mm)")
         ax[0].text(
@@ -377,6 +378,8 @@ def plot_data(
                             ha="right",
                             transform=ax[row_number, 0].transAxes,
                         )
+
+                    ax[row_number, i].set_aspect("equal")
         # fig.tight_layout()
         for ele in ax[-1]:
             ele.set_xlabel("x (mm)")

docs/examples/example_reconstruction_6d.ipynb

@@ -99,13 +99,7 @@ def __init__(
         # Drift from Bend to YAG 2 (corrected for dipole on/off)
         l_d3 = l3 - l_bend / 2 / np.cos(theta_on)
 
-        q = Quadrupole(
-            torch.tensor(l_quad),
-            torch.tensor(0.0),
-            name="SCAN_QUAD",
-            num_steps=5,
-            tracking_method="bmadx",
-        )
+        q = Quadrupole(torch.tensor(l_quad), torch.tensor(0.0), name="SCAN_QUAD")
         d1 = Drift(torch.tensor(l_d1))
 
         tdc = TransverseDeflectingCavity(
@@ -125,9 +119,10 @@ def __init__(
         bend = Dipole(
             name="SCAN_DIPOLE",
             length=torch.tensor(l_arc).float(),
-            angle=torch.tensor(0.0).float(),
+            angle=torch.tensor(theta_on).float(),
             dipole_e1=torch.tensor(0.0).float(),
             dipole_e2=torch.tensor(theta_on).float(),
+            tracking_method="bmadx",
         )
 
         d3 = Drift(name="DIPOLE_TO_SCREEN", length=torch.tensor(l_d3).float())