diff --git a/data/processing/orbitals.py b/data/processing/orbitals.py index 884ecfa..116231f 100644 --- a/data/processing/orbitals.py +++ b/data/processing/orbitals.py @@ -81,18 +81,21 @@ ax3d.quiver3D( ax3d.set(xlabel="x (cm)", ylabel="y (cm)", zlabel="z (cm)") theta = np.angle(raw_ts["x"] + 1j * raw_ts["y"]).mean() -fig2dv, ax2dv = plt.subplots() +fig2dv, ax2dv = plt.subplots(figsize=(5/2.54, 2/3*10/2.54), dpi=200, constrained_layout=True) x0 = ts_flt["x"] * np.cos(theta) + ts_flt["y"] * np.sin(theta) -ax2dv.plot(x0, z0, c="#0066ff", lw=1) +#ax2dv.plot(x0, z0, c="#0066ff", lw=1) ax2dv.quiver( x0[:-1], z0[:-1], np.diff(x0)[:], np.diff(z0)[:], - color="#0066ff", + color="k", + scale_units="xy", + scale=1, ) -ax2dv.grid() -ax2dv.set(aspect="equal") +ax2dv.grid(c="k", alpha=.2) +ax2dv.set(aspect="equal", xlabel="x (cm)", ylabel="z (cm)") fig2dv.savefig("out_orbitals.pdf") +fig2dv.savefig("out_orbitals.jpg") plt.show() diff --git a/olaflow/config.ini b/olaflow/config.ini index 3026449..1b97257 100644 --- a/olaflow/config.ini +++ b/olaflow/config.ini @@ -17,6 +17,6 @@ tf = 14300 x0 = -150 [post] -out = out_post +out = out_post_refined x = -50 z = 5 diff --git a/olaflow/processing/post.py b/olaflow/processing/post.py index 99bf295..36e4078 100644 --- a/olaflow/processing/post.py +++ b/olaflow/processing/post.py @@ -87,4 +87,4 @@ with plt.rc_context( loc="center left", ) - plt.show() + fig.savefig(out.joinpath("fig.pdf"))