From 339c27b99b2d6f66b6cd43d2c16dcc59745b1d0c Mon Sep 17 00:00:00 2001
From: "Edgar P. Burkhart" <git@edgarpierre.fr>
Date: Tue, 7 Jun 2022 12:09:22 +0200
Subject: [PATCH] Update plot outputs

---
 data/processing/orbitals.py | 13 ++++++++-----
 olaflow/config.ini          |  2 +-
 olaflow/processing/post.py  |  2 +-
 3 files changed, 10 insertions(+), 7 deletions(-)

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"))