From 92058a52ea78473326dbcebbbebfe61b3723342b Mon Sep 17 00:00:00 2001
From: Vardan Tadevosyan <tadevosn@jlab.org>
Date: Mon, 18 Jan 2016 11:53:47 +0400
Subject: [PATCH] Add track energy in THcShower class.

---
 src/THcShower.cxx | 10 ++++++----
 src/THcShower.h   |  1 +
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/THcShower.cxx b/src/THcShower.cxx
index e30987a..6e8e311 100644
--- a/src/THcShower.cxx
+++ b/src/THcShower.cxx
@@ -514,7 +514,8 @@ Int_t THcShower::DefineVariables( EMode mode )
     { "nhits", "Number of hits",                                 "fNhits" },
     { "nclust", "Number of clusters",                            "fNclust" },
     { "etot", "Total energy",                                    "fEtot" },
-    { "etotnorm", "Total energy divided by Central Momentum",   "fEtotNorm" },
+    { "etotnorm", "Total energy divided by Central Momentum",    "fEtotNorm" },
+    { "etrack", "Track energy",                                  "fEtrack" },
     { "ntracks", "Number of shower tracks",                      "fNtracks" },
     { 0 }
   };
@@ -568,6 +569,7 @@ void THcShower::Clear(Option_t* opt)
   fNtracks = 0;
   fEtot = 0.;
   fEtotNorm = 0.;
+  fEtrack = 0.;
 
   // Purge cluster list
 
@@ -1069,9 +1071,9 @@ Int_t THcShower::FineProcess( TClonesArray& tracks )
 
     THaTrack* theTrack = static_cast<THaTrack*>( tracks[itrk] );
 
-    Float_t energy = GetShEnergy(theTrack);
-    if (fHasArray) energy += fArray->GetShEnergy(theTrack);
-    theTrack->SetEnergy(energy);
+    fEtrack = GetShEnergy(theTrack);
+    if (fHasArray) fEtrack += fArray->GetShEnergy(theTrack);
+    theTrack->SetEnergy(fEtrack);
 
   }       //over tracks
 
diff --git a/src/THcShower.h b/src/THcShower.h
index a4a7454..b1573e3 100644
--- a/src/THcShower.h
+++ b/src/THcShower.h
@@ -128,6 +128,7 @@ protected:
                              // cluster-to-track association
   Double_t fEtot;            // Total energy 
   Double_t fEtotNorm;        // Total energy divided by spec central momentum 
+  Double_t fEtrack;          // Cluster energy associated to the last track
 
   THcShowerClusterList* fClusterList;   // List of hit clusters
 
-- 
GitLab