screwy plot

This commit is contained in:
James Szalkie 2026-06-05 16:00:31 -04:00
parent 19928ac432
commit 7fb4fc542c

View File

@ -378,6 +378,7 @@ def prepare_tree_data(tree, treename, particle, max_events=None, z_max=34.86):
Eqqq = energy_loss(particle, Eiqqq, dqqq)
Elost = Eisx3 - Esx3
Elostqqq = Eiqqq - Eqqq
Eprop = EA - EC
return {
@ -389,6 +390,7 @@ def prepare_tree_data(tree, treename, particle, max_events=None, z_max=34.86):
"EC": EC,
"Esx3": Esx3,
"Elost": Elost,
"Elostqqq": Elostqqq,
"Eprop": Eprop,
"dA": dA,
"dC": dC,
@ -999,6 +1001,7 @@ class MyInteractiveApp(cmd.Cmd):
qqqY = data["qqqY"]
qqqZ = data["qqqZ"]
qqqE = data["Eqqq"]
Elostqqq = data["Elostqqq"]
update_plot_data(f"{particle}_{treename}_Ei", Ei)
update_plot_data(f"{particle}_{treename}_sx3Z", sx3Z)
@ -1093,9 +1096,9 @@ class MyInteractiveApp(cmd.Cmd):
mask1 = ~np.isnan(qqqE) & ~np.isnan(thetab)
plt.figure(figsize=(7,6))
plt.hist2d(thetab[mask1], (np.sin(np.deg2rad(thetab)) * Elost)[mask1], bins=200)
plt.xlabel("Elost x sin(theta)")
plt.ylabel("Eqqq (MeV)")
plt.hist2d(qqqE[mask1], (np.sin(np.deg2rad(thetab)) * Elostqqq)[mask1], bins=200)
plt.ylabel("Elost x sin(theta)")
plt.xlabel("Eqqq (MeV)")
plt.title(f"{particle} ({treename}) Energy QQQ vs Elost * Theta")
plt.colorbar(label="Counts")
#plt.xlim(0,30)