From 7fb4fc542c79df1072f201cd9b633f90b401ff92 Mon Sep 17 00:00:00 2001
From: James Szalkie <jamesszalkie122@gmail.com>
Date: Fri, 5 Jun 2026 16:00:31 -0400
Subject: [PATCH] screwy plot

---
 ELoss/PCEnergyAnalysis.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/ELoss/PCEnergyAnalysis.py b/ELoss/PCEnergyAnalysis.py
index fad25b3..29f85b9 100644
--- a/ELoss/PCEnergyAnalysis.py
+++ b/ELoss/PCEnergyAnalysis.py
@@ -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)