Added true Nu PDG branch to PhaseIITreeMaker

ANNIEsoft · Feb 27, 2025 · 918a01c · 918a01c
1 parent 5b3a549
commit 918a01c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/UserTools/PhaseIITreeMaker/PhaseIITreeMaker.cpp b/UserTools/PhaseIITreeMaker/PhaseIITreeMaker.cpp
@@ -371,6 +371,7 @@ bool PhaseIITreeMaker::Initialise(std::string configfile, DataModel &data){
       fPhaseIITrigTree->Branch("trueNeutCapE",&fTrueNeutCapE);
       fPhaseIITrigTree->Branch("trueNeutCapGammaE",&fTrueNeutCapGammaE);
       fPhaseIITrigTree->Branch("trueNeutrinoEnergy",&fTrueNeutrinoEnergy,"trueNeutrinoEnergy/D");
+      fPhaseIITrigTree->Branch("trueNuPDG",&fTrueNuPDG,"trueNuPDG/I");
       fPhaseIITrigTree->Branch("trueNeutrinoMomentum_X",&fTrueNeutrinoMomentum_X,"trueNeutrinoMomentum_X/D");
       fPhaseIITrigTree->Branch("trueNeutrinoMomentum_Y",&fTrueNeutrinoMomentum_Y,"trueNeutrinoMomentum_Y/D");
       fPhaseIITrigTree->Branch("trueNeutrinoMomentum_Z",&fTrueNeutrinoMomentum_Z,"trueNeutrinoMomentum_Z/D");
@@ -1059,6 +1060,7 @@ void PhaseIITreeMaker::ResetVariables() {
     fTrueNeutCapE->clear();
     fTrueNeutCapGammaE->clear();
     fTrueNeutrinoEnergy = -9999;
+    fTrueNuPDG = -9999;
     fTrueNeutrinoMomentum_X = -9999;
     fTrueNeutrinoMomentum_Y = -9999;
     fTrueNeutrinoMomentum_Z = -9999;
@@ -2091,6 +2093,7 @@ bool PhaseIITreeMaker::FillMCTruthInfo() {
     std::cout <<"get_fsl_vtx: "<<get_fsl_vtx<<", get_fsl_momentum: "<<get_fsl_momentum<<", get_fsl_time: "<<get_fsl_time<<", get_fsl_mass: "<<get_fsl_mass<<", get_fsl_pdg: "<<get_fsl_pdg<<", get_fsl_energy: "<<get_fsl_energy<<std::endl;
     if (get_neutrino_energy && get_neutrino_mom && get_neutrino_vtxx && get_neutrino_vtxy && get_neutrino_vtxz && get_neutrino_vtxt && get_q2 && get_cc && get_nc && get_qel && get_res && get_dis && get_coh && get_mec && get_n && get_p && get_pi0 && get_piplus && get_pipluscher && get_piminus && get_piminuscher && get_kplus && get_kpluscher && get_kminus && get_kminuscher && get_fsl_vtx && get_fsl_momentum && get_fsl_time && get_fsl_mass && get_fsl_pdg && get_fsl_energy && get_bjx && get_y && get_targetZ && get_q0 && get_q3 && get_w ){
       fTrueNeutrinoEnergy = TrueNeutrinoEnergy;
+      fTrueNuPDG = TrueNuPDG;
       fTrueNeutrinoMomentum_X = TrueNeutrinoMomentum.X();
       fTrueNeutrinoMomentum_Y = TrueNeutrinoMomentum.Y();
       fTrueNeutrinoMomentum_Z = TrueNeutrinoMomentum.Z();

diff --git a/UserTools/PhaseIITreeMaker/PhaseIITreeMaker.h b/UserTools/PhaseIITreeMaker/PhaseIITreeMaker.h
@@ -301,6 +301,7 @@ class PhaseIITreeMaker: public Tool {
 
   //Genie information for event
   double fTrueNeutrinoEnergy;
+  int fTrueNuPDG;
   double fTrueNeutrinoMomentum_X;
   double fTrueNeutrinoMomentum_Y;
   double fTrueNeutrinoMomentum_Z;