Adding the charge to TMS_TreeWriter for them to be saved to the output

DUNE · Aug 2, 2024 · 8d8359f · 8d8359f
1 parent af38ede
commit 8d8359f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/TMS_TreeWriter.cpp b/src/TMS_TreeWriter.cpp
@@ -236,6 +236,7 @@ void TMS_TreeWriter::MakeBranches() {
   Reco_Tree->Branch("EnergyRange",  RecoTrackEnergyRange,   "EnergyRange[nTracks]/F");
   Reco_Tree->Branch("EnergyDeposit",RecoTrackEnergyDeposit, "EnergyDeposit[nTracks]/F");
   Reco_Tree->Branch("Length",       RecoTrackLength,        "Length[nTracks]/F");
+  Reco_Tree->Branch("Charge",       RecoTrackCharge,        "Charge[nTracks]/I");
 
 
   MakeTruthBranches(Truth_Info);
@@ -1204,6 +1205,7 @@ void TMS_TreeWriter::Fill(TMS_Event &event) {
     RecoTrackEnergyRange[itTrack]   =       RecoTrack->EnergyRange;
     RecoTrackLength[itTrack]        =       0.5 * (TrackLengthU[itTrack] + TrackLengthV[itTrack]); // RecoTrack->Length;, 2d is better estimate than 3d because of y jumps
     RecoTrackEnergyDeposit[itTrack] =       RecoTrack->EnergyDeposit;
+    RecoTrackCharge[itTrack]        =       RecoTrack->Charge;
     for (int j = 0; j < 3; j++) {
       RecoTrackStartPos[itTrack][j]  = RecoTrack->Start[j];
       RecoTrackEndPos[itTrack][j]    = RecoTrack->End[j];
@@ -1651,6 +1653,7 @@ void TMS_TreeWriter::Clear() {
     RecoTrackEnergyRange[i] = DEFAULT_CLEARING_FLOAT;
     RecoTrackEnergyDeposit[i] = DEFAULT_CLEARING_FLOAT;
     RecoTrackLength[i] = DEFAULT_CLEARING_FLOAT;
+    RecoTrackCharge[i] = DEFAULT_CLEARING_FLOAT;
   }
 
   RecoTrackN = 0;

diff --git a/src/TMS_TreeWriter.h b/src/TMS_TreeWriter.h
@@ -52,6 +52,7 @@ class TMS_TreeWriter {
     float RecoTrackEnergyRange[__TMS_MAX_TRACKS__];
     float RecoTrackEnergyDeposit[__TMS_MAX_TRACKS__];
     float RecoTrackLength[__TMS_MAX_TRACKS__];
+    int RecoTrackCharge[__TMS_MAX_TRACKS__];
 
   private:
     TMS_TreeWriter();