reduce_prod5_nue.C
Go to the documentation of this file.
1 #include "StandardRecord/StandardRecord.h"
2 #include "StandardRecord/Proxy/SRProxy.h"
3 #include "CAFAna/Core/FileReducer.h"
4 #include "CAFAna/Cuts/Cuts.h"
5 #include "3FlavorAna/Cuts/NueCuts2017.h"
6 #include "3FlavorAna/Cuts/NueCuts2018.h"
7 #include "3FlavorAna/Cuts/NueCuts2020.h"
8 #include "3FlavorAna/Cuts/NumuCuts2017.h"
9 #include "3FlavorAna/Cuts/NumuCuts2018.h"
10 #include "3FlavorAna/Cuts/NumuCuts2020.h"
11 #include "3FlavorAna/Cuts/NumuCuts.h"
12 #include "3FlavorAna/Cuts/BeamNueCuts.h"
13 
14 using namespace ana;
15 void ReduceForSignalDecaf(caf::StandardRecord* sr,
16  const caf::SRProxy* srProxy)
17 {
18 
19  ClearMultiNuInfo(sr);
20  ClearSecondaryTrackInfo(sr);
21 
22  if(sr->hdr.det == caf::kFARDET) return;
23 
24  if(!ana::kNue2020NDDecafCut(srProxy) && (ana::kNumuUncontainNDDecafCut(srProxy) || ana::kNumuContainNDDecafCut(srProxy))){
25 
26  // remove these completely
27  sr->parent = caf::SRParentBranch();
28  sr->training = caf::SRTrainingBranch();
29  sr->me = caf::SRMichelE();
30 
31  // slc branch
32  // keeping some basic slice info for various cuts
33  unsigned int tempncontplanes = sr->slc.ncontplanes;
34  unsigned int tempnhit = sr->slc.nhit;
35  unsigned int tempfirstplane = sr->slc.firstplane;
36  unsigned int templastplane = sr->slc.lastplane;
37  unsigned int tempncellsfromedge = sr->slc.ncellsfromedge;
38  float tempcalE = sr->slc.calE;
39  caf::SRVector3D tempmeanpos = sr->slc.meanpos;
40  caf::SRVector3D tempboxmax = sr->slc.boxmax;
41  caf::SRVector3D tempboxmin = sr->slc.boxmin;
42 
43  sr->slc = caf::SRSlice();
44 
45  sr->slc.ncontplanes = tempncontplanes;
46  sr->slc.nhit = tempnhit;
47  sr->slc.firstplane = tempfirstplane;
48  sr->slc.lastplane = templastplane;
49  sr->slc.ncellsfromedge = tempncellsfromedge;
50  sr->slc.calE = tempcalE;
51  sr->slc.meanpos = tempmeanpos;
52  sr->slc.boxmin = tempboxmin;
53  sr->slc.boxmax = tempboxmax;
54 
55  // energy branch
56  // keeping only a few numu energy vars
57  float temprecotrkcchadE = sr->energy.numu.recotrkcchadE;
58  float temptrkccE = sr->energy.numu.trkccE;
59  float temptrknonqeE = sr->energy.numu.trknonqeE;
60  float temptrkqeE = sr->energy.numu.trkqeE;
61  float tempndhadcaltranE = sr->energy.numu.ndhadcaltranE;
62  float tempndhadcalcatE = sr->energy.numu.ndhadcalcatE;
63  float tempE = sr->energy.numu.E;
64  float temphadcalE = sr->energy.numu.hadcalE;
65  float temphadtrkE = sr->energy.numu.hadtrkE;
66  float tempndtrkcaltranE = sr->energy.numu.ndtrkcaltranE;
67  float tempndtrkcalcatE = sr->energy.numu.ndtrkcalcatE;
68  float tempndtrkcalactE = sr->energy.numu.ndtrkcalactE;
69  float tempndtrklencat = sr->energy.numu.ndtrklencat;
70  float tempndtrklenact = sr->energy.numu.ndtrklenact;
71 
72  sr->energy.numu = caf::SRNumuEnergy();
73 
74  sr->energy.numu.recotrkcchadE = temprecotrkcchadE;
75  sr->energy.numu.trkccE = temptrkccE;
76  sr->energy.numu.trknonqeE = temptrknonqeE;
77  sr->energy.numu.trkqeE = temptrkqeE;
78  sr->energy.numu.ndhadcaltranE = tempndhadcaltranE;
79  sr->energy.numu.ndhadcalcatE = tempndhadcalcatE;
80  sr->energy.numu.E = tempE;
81  sr->energy.numu.hadcalE = temphadcalE;
82  sr->energy.numu.hadtrkE = temphadtrkE;
83  sr->energy.numu.ndtrkcaltranE = tempndtrkcaltranE;
84  sr->energy.numu.ndtrkcalcatE = tempndtrkcalcatE;
85  sr->energy.numu.ndtrkcalactE = tempndtrkcalactE;
86  sr->energy.numu.ndtrklencat = tempndtrklencat;
87  sr->energy.numu.ndtrklenact = tempndtrklenact;
88 
89  ResetBPFEnergy(sr);
90 
91  // sel branch
92 
93  sr->sel.bpfid = caf::SRBpfId();
94  sr->sel.xnuepid = caf::SRXnue();
95 
96  ResetLEMInfo(sr);
97  ResetLIDInfo(sr);
98  ResetRVPInfo(sr);
99  ResetCosRejInfo(sr);
100 
101  //
102  // track branch
103  // need number of cosmic tracks, therefore we create a dummy vector
104  sr->trk.cosmic.tracks.clear();
105  sr->trk.cosmic.tracks.resize(sr->trk.cosmic.ntracks);
106 
107  // need start and stop coordinates
108  if(sr->trk.kalman.ntracks > 0){
109  std::vector<caf::SRVector3D> temptrackstart;
110  std::vector<caf::SRVector3D> temptrackstop;
111  std::vector<caf::SRVector3D> temptrackdir;
112  unsigned int tempidxremid = sr->trk.kalman.idxremid;
113  for(int i = 0; i < (int)sr->trk.kalman.tracks.size(); i++){
114  temptrackstart.push_back(sr->trk.kalman.tracks[i].start);
115  temptrackstop.push_back(sr->trk.kalman.tracks[i].stop);
116  temptrackdir.push_back(sr->trk.kalman.tracks[i].dir);
117  }
118  float temptrackslen = sr->trk.kalman.tracks[0].len;
119 
120  sr->trk.kalman.tracks.clear();
121  sr->trk.kalman.tracks.resize(sr->trk.kalman.ntracks);
122 
123  for(int i = 0; i < (int)sr->trk.kalman.tracks.size(); i++){
124  sr->trk.kalman.tracks[i].start = temptrackstart[i];
125  sr->trk.kalman.tracks[i].stop = temptrackstop[i];
126  sr->trk.kalman.tracks[i].dir = temptrackdir[i];
127  }
128  sr->trk.kalman.tracks[0].len = temptrackslen;
129  sr->trk.kalman.idxremid = tempidxremid;
130  }
131 
132  // vertex branch
133  // need vtx coordinates
134  if(sr->vtx.elastic.IsValid && sr->vtx.elastic.fuzzyk.npng > 0){
135  float tempvtxX = sr->vtx.elastic.vtx.X();
136  float tempvtxY = sr->vtx.elastic.vtx.Y();
137  float tempvtxZ = sr->vtx.elastic.vtx.Z();
138  float tempnshwlid = sr->vtx.elastic.fuzzyk.nshwlid;
139  float tempnpng = sr->vtx.elastic.fuzzyk.npng;
140  float tempdedx0 = sr->vtx.elastic.fuzzyk.png[0].shwlid.lid.dedx0;
141  int templongestidx = sr->vtx.elastic.fuzzyk.longestidx;
142  float templen = sr->vtx.elastic.fuzzyk.png[templongestidx].len;
143 
144  std::vector<caf::SRVector3D> tempshwlidstart;
145  std::vector<caf::SRVector3D> tempshwlidstop;
146  std::vector<caf::SRVector3D> temppngstart;
147  std::vector<caf::SRCVNParticleResult> tempcvnpart;
148  std::vector<float> tempshwlidcale;
149 
150  for(int pngidx = 0; pngidx < (int)sr->vtx.elastic.fuzzyk.nshwlid; pngidx++){
151  tempshwlidstart.push_back(sr->vtx.elastic.fuzzyk.png[pngidx].shwlid.start);
152  tempshwlidstop.push_back(sr->vtx.elastic.fuzzyk.png[pngidx].shwlid.stop);
153  }
154  for(int pngidx = 0; pngidx < (int)sr->vtx.elastic.fuzzyk.npng; pngidx++){
155  temppngstart.push_back(sr->vtx.elastic.fuzzyk.png[pngidx].start);
156  tempcvnpart.push_back(sr->vtx.elastic.fuzzyk.png[pngidx].cvnpart);
157  tempshwlidcale.push_back(sr->vtx.elastic.fuzzyk.png[pngidx].shwlid.calE);
158  }
159 
160 
161  sr->vtx.elastic.IsValid = true;
162  sr->vtx.elastic.vtx.x = tempvtxX;
163  sr->vtx.elastic.vtx.y = tempvtxY;
164  sr->vtx.elastic.vtx.z = tempvtxZ;
165 
166  sr->vtx.elastic.fuzzyk.npng = tempnpng;
167  sr->vtx.elastic.fuzzyk.nshwlid = tempnshwlid;
168  sr->vtx.elastic.fuzzyk.png.resize(tempnpng);
169  for(int pngidx = 0; pngidx < (int)sr->vtx.elastic.fuzzyk.nshwlid; pngidx++){
170  sr->vtx.elastic.fuzzyk.png[pngidx].shwlid.start = tempshwlidstart[pngidx];
171  sr->vtx.elastic.fuzzyk.png[pngidx].shwlid.stop = tempshwlidstop[pngidx];
172  }
173  for(int pngidx = 0; pngidx < (int)sr->vtx.elastic.fuzzyk.npng; pngidx++){
174  sr->vtx.elastic.fuzzyk.png[pngidx].start = temppngstart[pngidx];
175  sr->vtx.elastic.fuzzyk.png[pngidx].cvnpart = tempcvnpart[pngidx];
176  sr->vtx.elastic.fuzzyk.png[pngidx].shwlid.calE = tempshwlidcale[pngidx];
177  }
178  sr->vtx.elastic.fuzzyk.png[0].shwlid.lid.dedx0 = tempdedx0;
179  sr->vtx.elastic.fuzzyk.longestidx = templongestidx;
180  sr->vtx.elastic.fuzzyk.png[templongestidx].len = templen;
181 
182  ClearHoughVertexInfo(sr);
183  }
184 
185  // truth branch
186  if(sr->hdr.ismc == 1 && sr->mc.nnu > 0) {
187 
188  sr->mc.nu[0].rwgt.ppfx.vuniv.clear();
189  sr->mc.nu[0].rwgt.ppfx.nvuniv = 0;
190 
191  ClearMichelTruthInfo(sr);
192  }
193 
194  }
195 
196 }
197 
198 void reduce_prod5_nue(std::string fname, std::string outname)
199 {
200  FileReducer reduce(fname, outname);
201  reduce.SetMetadata("nova.decaf_skim", "nue2020_numundv2");
202 
203  const Cut kIsND(
204  [](const caf::SRProxy* sr)
205  {
206  return sr->hdr.det==caf::kNEARDET;
207  });
208 
209  const Cut kIsFD(
210  [](const caf::SRProxy* sr)
211  {
212  return sr->hdr.det==caf::kFARDET;
213  });
214 
215  const Cut fdCut = (kIsFD && kNue2020FDDecafCut);
216  const Cut ndCut = (kIsND && (kNue2020NDDecafCut || kNumuContainNDDecafCut || kNumuUncontainNDDecafCut));
217 
218  reduce.AddEventCut(fdCut || ndCut);
219  reduce.AddReductionStep(ReduceForSignalDecaf);
220  reduce.AddReductionStep(ClearMultiNuInfo);
221  reduce.AddReductionStep(ClearSecondaryTrackInfo);
222  reduce.Go();
223 }
caf::SRHeader::det
Det_t det
Detector, ND = 1, FD = 2, NDOS = 3.
Definition: SRHeader.h:28
caf::kNEARDET
Near Detector underground.
Definition: SREnums.h:10
PandAna.cut.analysis_cuts.kIsFD
kIsFD
Definition: analysis_cuts.py:8
plot_validation_datamc.fname
fname
Definition: plot_validation_datamc.py:180
caf::kFARDET
Far Detector at Ash River.
Definition: SREnums.h:11
caf::SRVector3D
A 3-vector with more efficient storage than TVector3.
Definition: SRVector3D.h:14
ana
Cuts and Vars for the 2020 FD DiF Study.
Definition: vars.h:6
caf::SRTrainingBranch
Event ID training variables.
Definition: SRTrainingBranch.h:13
caf::StandardRecord::hdr
SRHeader hdr
Header branch: run, subrun, etc.
Definition: StandardRecord.h:40
caf::SRNumuEnergy::ndtrklenact
float ndtrklenact
Near detector – muon track length in active region [cm].
Definition: SRNumuEnergy.h:37
caf::SRNumuEnergy::ndtrkcaltranE
float ndtrkcaltranE
Near detector – muon calorimetric energy in transition plane [GeV].
Definition: SRNumuEnergy.h:40
ana::ResetCosRejInfo
void ResetCosRejInfo(caf::StandardRecord *sr)
Definition: FileReducer.cxx:309
ana::ClearSecondaryTrackInfo
void ClearSecondaryTrackInfo(caf::StandardRecord *sr)
Definition: FileReducer.cxx:194
caf::SRSlice::lastplane
unsigned int lastplane
last plane
Definition: SRSlice.h:27
caf::SRSlice::firstplane
unsigned int firstplane
first plane
Definition: SRSlice.h:26
caf::SRKalman::ntracks
size_t ntracks
Definition: SRKalman.h:23
caf::SRNumuEnergy::trknonqeE
float trknonqeE
Track length non-quasielastic neutrino energy [GeV].
Definition: SRNumuEnergy.h:26
caf::SRKalman::idxremid
unsigned int idxremid
index number of the best ReMId track
Definition: SRKalman.h:32
caf::SRSlice::boxmax
SRVector3D boxmax
Maximum coordinates box containing all the hits [cm].
Definition: SRSlice.h:47
caf::Proxy< caf::StandardRecord >::hdr
caf::Proxy< caf::SRHeader > hdr
Definition: SRProxy.h:2137
caf::SRFuzzyK::png
std::vector< SRFuzzyKProng > png
Vector of 3D prong objects.
Definition: SRFuzzyK.h:19
caf::StandardRecord::me
SRMichelE me
Michel electron branch.
Definition: StandardRecord.h:45
caf::Proxy< caf::StandardRecord >
Proxy for caf::StandardRecord.
Definition: SRProxy.h:2126
caf::SRVector3D::Y
float Y() const
Definition: SRVector3D.h:33
caf::SRNumuEnergy::ndtrkcalcatE
float ndtrkcalcatE
Near detector – muon calorimetric energy in muon catcher [GeV].
Definition: SRNumuEnergy.h:41
caf::StandardRecord::training
SRTrainingBranch training
Extra training information for prototyping PIDs etc.
Definition: StandardRecord.h:50
reduce_prod5_nue
void reduce_prod5_nue(std::string fname, std::string outname)
Definition: reduce_prod5_nue.C:198
caf::SRFuzzyK::longestidx
unsigned int longestidx
index of longest prong
Definition: SRFuzzyK.h:23
ana::FileReducer::SetMetadata
void SetMetadata(const std::string &key, const std::string &val)
Cause the output file to be systematically shifted.
Definition: FileReducer.h:102
caf::SRHeader::ismc
bool ismc
data or MC? True if MC
Definition: SRHeader.h:27
caf::SRXnue
Store BDT variables for the short-baseline oscillation study.
Definition: SRXnue.h:12
caf::SRBpfId
Breakpoint ID (BpfId) output.
Definition: SRBpfId.h:18
ana::FileReducer::AddReductionStep
void AddReductionStep(const std::function< ReductionFunc > &f)
Run the specified reduction function over each event.
Definition: FileReducer.h:85
caf::SRVector3D::X
float X() const
Definition: SRVector3D.h:32
ana::ResetRVPInfo
void ResetRVPInfo(caf::StandardRecord *sr)
Definition: FileReducer.cxx:303
caf::SRNumuEnergy::trkccE
float trkccE
Track length cc neutrino energy [GeV].
Definition: SRNumuEnergy.h:27
caf::SRTrackBranch::kalman
SRKalman kalman
Tracks produced by KalmanTrack.
Definition: SRTrackBranch.h:24
caf::SRFuzzyK::nshwlid
unsigned int nshwlid
number of shwlid showers - either 0 or number of 3d prongs
Definition: SRFuzzyK.h:24
caf::SRSlice::boxmin
SRVector3D boxmin
Minimum coordinates box containing all the hits [cm].
Definition: SRSlice.h:46
caf::SRNumuEnergy::ndhadcalcatE
float ndhadcalcatE
Near detector – hadronic calorimetric energy NOT on the muon track in muon catcher [GeV]...
Definition: SRNumuEnergy.h:44
caf::SRNumuEnergy::trkqeE
float trkqeE
Track length quasielastic neutrino energy [GeV].
Definition: SRNumuEnergy.h:25
caf::SRNumuEnergy::hadtrkE
float hadtrkE
Hadronic calorimetric energy on the muon track[GeV].
Definition: SRNumuEnergy.h:36
ana::kNue2020NDDecafCut
const Cut kNue2020NDDecafCut
Definition: NueCuts2020.h:176
ana::ClearHoughVertexInfo
void ClearHoughVertexInfo(caf::StandardRecord *sr)
Definition: FileReducer.cxx:265
ana::ClearMichelTruthInfo
void ClearMichelTruthInfo(caf::StandardRecord *sr)
Definition: FileReducer.cxx:42
caf::SRSlice::calE
float calE
Calorimetric energy of the cluster [GeV].
Definition: SRSlice.h:38
ana::FileReducer::AddEventCut
void AddEventCut(const Cut &cut)
Only copy records to the output file if they pass this cut.
Definition: FileReducer.cxx:368
caf::SRNumuEnergy
Numu energy estimator output.
Definition: SRNumuEnergy.h:17
sr
caf::StandardRecord * sr
Definition: FillTrainingTrees.h:18
ana::kNue2020FDDecafCut
const Cut kNue2020FDDecafCut
Definition: NueCuts2020.h:31
caf::SRElastic::vtx
SRVector3D vtx
Vertex position in detector coordinates. [cm].
Definition: SRElastic.h:40
caf::SRTruthBranch::nnu
short nnu
Number of neutrinos in nu vector (0 or 1)
Definition: SRTruthBranch.h:37
caf::SRFuzzyK::npng
size_t npng
Definition: SRFuzzyK.h:26
caf::SRNumuEnergy::E
float E
Neutrino energy, set to match trkccE [GeV].
Definition: SRNumuEnergy.h:23
caf::SRSlice
An SRSlice contains overarching information for a slice.
Definition: SRSlice.h:15
caf::SRVector3D::Z
float Z() const
Definition: SRVector3D.h:34
caf::SRSlice::nhit
unsigned int nhit
number of hits
Definition: SRSlice.h:22
ana::FileReducer::Go
virtual void Go() override
Load all the registered spectra.
Definition: FileReducer.cxx:405
ana::FileReducer
Create smaller CAFs.
Definition: FileReducer.h:56
caf::SRIDBranch::bpfid
SRBpfId bpfid
Output from the BreakPointFitter PID (BPFPIdMaker) package.
Definition: SRIDBranch.h:41
ReduceForSignalDecaf
void ReduceForSignalDecaf(caf::StandardRecord *sr, const caf::SRProxy *srProxy)
Definition: reduce_prod5_nue.C:15
caf::SRSlice::ncellsfromedge
unsigned int ncellsfromedge
minimum number of cells to edge of detector
Definition: SRSlice.h:30
ana::kNumuUncontainNDDecafCut
const Cut kNumuUncontainNDDecafCut
Definition: BeamNueCuts.h:23
caf::StandardRecord
The StandardRecord is the primary top-level object in the Common Analysis File trees.
Definition: StandardRecord.h:33
ana::ResetBPFEnergy
void ResetBPFEnergy(caf::StandardRecord *sr)
Definition: FileReducer.cxx:285
caf::SRNumuEnergy::ndhadcaltranE
float ndhadcaltranE
Near detector – hadronic calorimetric energy NOT on the muon track in transition plane [GeV]...
Definition: SRNumuEnergy.h:43
caf::SREnergyBranch::numu
SRNumuEnergy numu
Numu energy estimator.
Definition: SREnergyBranch.h:27
caf::SRNumuEnergy::ndtrkcalactE
float ndtrkcalactE
Near detector – muon calorimetric energy in active region [GeV].
Definition: SRNumuEnergy.h:39
ana::ResetLEMInfo
void ResetLEMInfo(caf::StandardRecord *sr)
Definition: FileReducer.cxx:291
caf::SRNumuEnergy::ndtrklencat
float ndtrklencat
Near detector – muon track length in muon catcher [cm].
Definition: SRNumuEnergy.h:38
caf::StandardRecord::sel
SRIDBranch sel
Selector (PID) branch.
Definition: StandardRecord.h:47
caf::SRVertexBranch::elastic
SRElastic elastic
Single vertex found by Elastic Arms.
Definition: SRVertexBranch.h:26
caf::StandardRecord::slc
SRSlice slc
Slice branch: nhit, extents, time, etc.
Definition: StandardRecord.h:42
caf::StandardRecord::mc
SRTruthBranch mc
Truth branch for MC: energy, flavor, etc.
Definition: StandardRecord.h:48
caf::SRElastic::fuzzyk
SRFuzzyK fuzzyk
Primary 3D prong object.
Definition: SRElastic.h:44
caf::SRNumuEnergy::recotrkcchadE
float recotrkcchadE
Reconstructed hadronic energy for track cc neutrino energy estimator [GeV].
Definition: SRNumuEnergy.h:34
ana::ResetLIDInfo
void ResetLIDInfo(caf::StandardRecord *sr)
Definition: FileReducer.cxx:297
caf::SRSlice::meanpos
SRVector3D meanpos
Mean position of hits in slice, weighted by charge [cm].
Definition: SRSlice.h:48
caf::StandardRecord::parent
SRParentBranch parent
True parent branch for matching, e.g. MRCC.
Definition: StandardRecord.h:49
ana::_Cut< caf::SRProxy >
caf::SRSlice::ncontplanes
unsigned int ncontplanes
number of continuous planes
Definition: SRSlice.h:25
caf::StandardRecord::trk
SRTrackBranch trk
Track branch: nhit, len, etc.
Definition: StandardRecord.h:43
caf::StandardRecord::energy
SREnergyBranch energy
Energy estimator branch.
Definition: StandardRecord.h:46
caf::SRTrackBranch::cosmic
SRTrackBase cosmic
Tracks produced by CosmicTrack.
Definition: SRTrackBranch.h:26
caf::SRTruthBranch::nu
std::vector< SRNeutrino > nu
implemented as a vector to maintain mc.nu structure, i.e. not a pointer, but with 0 or 1 entries...
Definition: SRTruthBranch.h:25
caf::SRTrackBase::tracks
std::vector< SRTrack > tracks
Definition: SRTrackBase.h:15
caf::SRNumuEnergy::hadcalE
float hadcalE
Hadronic calorimetric energy NOT on the muon track[GeV].
Definition: SRNumuEnergy.h:35
ana::ClearMultiNuInfo
void ClearMultiNuInfo(caf::StandardRecord *sr)
Definition: FileReducer.cxx:25
caf::SRKalman::tracks
std::vector< SRKalmanTrack > tracks
3D Tracks produced by KalmanTrack
Definition: SRKalman.h:16
ana::kNumuContainNDDecafCut
const Cut kNumuContainNDDecafCut
Definition: BeamNueCuts.h:22
caf::Proxy< caf::SRHeader >::det
caf::Proxy< caf::Det_t > det
Definition: SRProxy.h:232
caf::StandardRecord::vtx
SRVertexBranch vtx
Vertex branch: location, time, etc.
Definition: StandardRecord.h:44
caf::SRIDBranch::xnuepid
SRXnue xnuepid
Output from BDT (XnuePID)
Definition: SRIDBranch.h:57
string
enum BeamMode string
Generated on Mon Mar 8 2021 14:11:25 for NOvA by   doxygen 1.8.11