m2cce/internship
m2cce/internship
1
Fork 0
Code Issues Activity

Update plot outputs

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-06-07 12:09:22 +02:00
parent 0b5f8be42a
commit 339c27b99b
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
3 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
data/processing/orbitals.py
View file

@ -81,18 +81,21 @@ ax3d.quiver3D(
ax3d.set(xlabel="x (cm)", ylabel="y (cm)", zlabel="z (cm)") ax3d.set(xlabel="x (cm)", ylabel="y (cm)", zlabel="z (cm)")
theta = np.angle(raw_ts["x"] + 1j * raw_ts["y"]).mean() 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) 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( ax2dv.quiver(
x0[:-1], x0[:-1],
z0[:-1], z0[:-1],
np.diff(x0)[:], np.diff(x0)[:],
np.diff(z0)[:], np.diff(z0)[:],
color="#0066ff", color="k",
scale_units="xy",
scale=1,
) )
ax2dv.grid() ax2dv.grid(c="k", alpha=.2)
ax2dv.set(aspect="equal") ax2dv.set(aspect="equal", xlabel="x (cm)", ylabel="z (cm)")
fig2dv.savefig("out_orbitals.pdf") fig2dv.savefig("out_orbitals.pdf")
fig2dv.savefig("out_orbitals.jpg")
plt.show() plt.show()

2
olaflow/config.ini
View file

@ -17,6 +17,6 @@ tf = 14300
x0 = -150 x0 = -150
[post] [post]
out = out_post out = out_post_refined
x = -50 x = -50
z = 5 z = 5

2
olaflow/processing/post.py
View file

@ -87,4 +87,4 @@ with plt.rc_context(
loc="center left", loc="center left",
) )
plt.show() fig.savefig(out.joinpath("fig.pdf"))