Jpp 20.0.0-rc.2
the software that should make you happy
Loading...
Searching...
No Matches
JMuonPostfit.cc File Reference

Example program to histogram fit results for the muon reconstruction chain. More...

#include <string>
#include <iostream>
#include <iomanip>
#include <vector>
#include <set>
#include "TROOT.h"
#include "TFile.h"
#include "TH1D.h"
#include "TH2D.h"
#include "TProfile.h"
#include "TMath.h"
#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/Evt.hh"
#include "km3net-dataformat/online/JDAQModuleIdentifier.hh"
#include "km3net-dataformat/definitions/reconstruction.hh"
#include "km3net-dataformat/definitions/fitparameters.hh"
#include "km3net-dataformat/tools/time_converter.hh"
#include "JAAnet/JHead.hh"
#include "JAAnet/JHeadToolkit.hh"
#include "JAAnet/JAAnetToolkit.hh"
#include "JAAnet/JVolume.hh"
#include "JDAQ/JDAQEventIO.hh"
#include "JPhysics/JConstants.hh"
#include "JTools/JQuantile.hh"
#include "JSupport/JTriggeredFileScanner.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"
#include "JSupport/JSupport.hh"
#include "JReconstruction/JEvt.hh"
#include "JReconstruction/JEvtToolkit.hh"
#include "JLang/JPredicate.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

Go to the source code of this file.

Functions

int main (int argc, char **argv)
 

Detailed Description

Example program to histogram fit results for the muon reconstruction chain.

Author
mdejong, bofearraigh.

Definition in file JMuonPostfit.cc.

Function Documentation

◆ main()

int main ( int argc,
char ** argv )

Definition at line 79 of file JMuonPostfit.cc.

80{
81 using namespace std;
82 using namespace JPP;
83 using namespace KM3NETDAQ;
84
85 typedef JTriggeredFileScanner<JEvt> JTriggeredFileScanner_t;
86 typedef JTriggeredFileScanner_t::multi_pointer_type multi_pointer_type;
87
88 JTriggeredFileScanner_t inputFile;
89 JLimit_t& numberOfEvents = inputFile.getLimit();
90 string outputFile;
91 size_t numberOfPrefits;
93 JAtmosphericMuon atmosphere;
94 double Emin_GeV;
95 int application;
96 string option;
97 string primary;
98 int debug;
99
100 try {
101
102 JParser<> zap("Example program to histogram fit results.");
103
104 zap['f'] = make_field(inputFile);
105 zap['o'] = make_field(outputFile) = "postfit.root";
106 zap['n'] = make_field(numberOfEvents) = JLimit::max();
107 zap['N'] = make_field(numberOfPrefits) = 1;
109 zap['E'] = make_field(Emin_GeV) = 0.0;
111 zap['a'] = make_field(atmosphere) = JAtmosphericMuon(90.0, 90.0);
112 zap['O'] = make_field(option) = "LINE", "LOGE", "LINN", "LOGN";
113 zap['p'] = make_field(primary) = muon_t, neutrino_t;
114 zap['d'] = make_field(debug) = 2;
115
116 zap(argc, argv);
117 }
118 catch(const exception& error) {
119 FATAL(error.what() << endl);
120 }
121
122 JHead head;
123
124 try {
125 head = getCommonHeader(inputFile);
126 }
127 catch(const exception& error) {
128 FATAL(error.what() << endl);
129 }
130
131 JVolume volume(head, option != "LINE");
132 JPosition3D offset = getPosition(getOffset(head));
134 JCylinder3D cylinder = getCylinder(head);
135
136 cylinder.add(offset);
137
138 NOTICE("Offset: " << offset << endl);
139 NOTICE("Cylinder: " << cylinder << endl);
140
141 TFile out(outputFile.c_str(), "recreate");
142
143 TH1D job("job", NULL, 100, 0.5, 100.5);
144
145 TH1D hn("hn", NULL, 250, -0.5, 249.5); // NDF
146 TH1D hq("hq", NULL, 300, 0.0, 600.0); // quality parameter
147 TH1D hi("hi", NULL, 101, -0.5, 100.5); // index of the best event in order of descending angular resolution
148 TH1D hv("hv", NULL, 200, -6.0, 0.0); // log of Poisson distribution of (#hits,number photo-electrons)
149 TH1D h1("h1", NULL, 200, -2.0, +2.0); // number of photo-electrons along the whole track
150 TH1D hc("hc", NULL, 200, -1.0, +1.0); // dz (z-slope)
151 TH1D hu("hu", NULL, 400, -1.0e3, 1.0e3); // quality difference between best up-going and down-going track (check of atmospheric muon)
152
153 TH1D hx("hx", NULL, 70, -3.0, +2.3); // angular deviation [log(deg)]
154 TH1D hd("hd", NULL, 100, 0.0, 10.0); // distance between true and best event at intersection point [m]
155 TH1D ht("ht", NULL, 100, -100.0, 100.0); // time between true and best track [ns]
156
157 TH1D hz0("hz0[start]", NULL, 400, -200.0, 200.0); // start point [m]
158 TH1D hz1("hz1[end]", NULL, 400, -200.0, 200.0); // end point [m]
159
160 TH1D E_0("E_0", NULL, 100, volume.getXmin(), volume.getXmax()); // true muon energy
161 TH1D E_1("E_1", NULL, 100, volume.getXmin(), volume.getXmax()); // uncorrected energy of best track
162 TH1D E_2("E_2", NULL, 100, volume.getXmin(), volume.getXmax()); // corrected energy of best track
163 TH1D E_E("E_E", NULL, 100, -5.0, +5.0); // ratio of reconstructed energy and true energy [log(E/E)]
164 TH2D ExE("ExE", NULL,
165 40, volume.getXmin(), volume.getXmax(),
166 40, volume.getXmin(), volume.getXmax()); // (Etrue,Ereco)
167
168 const Int_t ny = 28800;
169 const Double_t ymin = 0.0; // [deg]
170 const Double_t ymax = 180.0; // [deg]
171
173
174 if (option == "LINN") {
175
176 const double xmin = log((double) 3);
177 const double xmax = log((double) 300);
178
179 for (double x = xmin, dx = (xmax - xmin) / 20; x <= xmax; x += dx) {
180 X.push_back((int) exp(x));
181 }
182
183 } else if (option == "LOGN") {
184
185 const double xmin = log10((double) 3);
186 const double xmax = log10((double) 400);
187
188 for (double x = xmin, dx = (xmax - xmin) / 20; x <= xmax; x += dx) {
189 X.push_back(x);
190 }
191
192 } else {
193
194 for (double x = volume.getXmin(), dx = (volume.getXmax() - volume.getXmin()) / 20.0; x < volume.getXmax() + 0.5*dx; x += dx) {
195 X.push_back(x);
196 }
197 }
198
199 TH2D h2("h2", NULL, X.size() - 1, X.data(), ny, ymin, ymax);
200
201 TH2D Va("Va", NULL,
202 100, 0.0, cylinder.getRadius()*cylinder.getRadius(),
203 100, cylinder.getZmin() - 50.0, cylinder.getZmax() + 50.0);
204 TH2D Vb("Vb", NULL,
205 100, 0.0, cylinder.getRadius()*cylinder.getRadius(),
206 100, cylinder.getZmin() - 50.0, cylinder.getZmax() + 50.0);
207
208 JQuantile Q("Angle");
209 JQuantile O("Omega");
210
211
212 while (inputFile.hasNext()) {
213
214 STATUS("event: " << setw(10) << inputFile.getCounter() << '\r'); DEBUG(endl);
215
216 multi_pointer_type ps = inputFile.next();
217
218 JDAQEvent* tev = ps;
219 JEvt* evt = ps;
220 Evt* event = ps;
221
222 const time_converter converter(*event, *tev);
223
224 job.Fill(1.0);
225
226
227 double Enu = 0.0;
228 double Emu = 0.0;
229 double Emax = 0.0;
230
231 if (has_neutrino(*event)) {
232 Enu = get_neutrino(*event).E;
233 }
234
235 vector<Trk>::const_iterator muon = event->mc_trks.end();
236
237 for (vector<Trk>::const_iterator track = event->mc_trks.begin(); track != event->mc_trks.end(); ++track) {
238
239 if (is_muon(*track)) {
240
241 Emu += track->E;
242
243 if (track->E > Emax) {
244 muon = track;
245 Emax = track->E;
246 }
247 }
248 }
249
250 if (muon == event->mc_trks.end()) {
251 continue;
252 }
253
254 job.Fill(3.0);
255
256
257 // abscissa
258
259 Double_t x = 0.0;
260
261 {
262 const double E = event->mc_trks[0].E;
263 const int N = getCount(tev->begin<JDAQTriggeredHit>(), tev->end<JDAQTriggeredHit>());
264
265 if (option == "LINN")
266 x = N;
267 else if (option == "LOGN")
268 x = log10(N);
269 else
270 x = volume.getX(E);
271 }
272
273
274 if (!evt->empty()) {
275
276 JEvt::iterator __end = partition(evt->begin(), evt->end(), JHistory::is_application(application));
277
278 if (evt->begin() == __end) {
279 continue;
280 }
281
282 job.Fill(4.0);
283
284 if (numberOfPrefits > 0) {
285
286 JEvt::iterator __q = __end;
287
288 advance(__end = evt->begin(), min(numberOfPrefits, (size_t) distance(evt->begin(), __q)));
289
290 partial_sort(evt->begin(), __end, __q, quality_sorter);
291
292 } else {
293
294 sort(evt->begin(), __end, quality_sorter);
295 }
296
297 double E = 0.0;
298 JPointing pointing;
299
300 if (primary == muon_t) {
301 E = Emu;
302 pointing = JPointing(getDirection(*muon));
303 } else if (primary == neutrino_t) {
304 E = Enu;
305 pointing = JPointing(getDirection(get_neutrino(*event)));
306 }
307
308 JEvt::iterator best = pointing(evt->begin(), __end);
309 const Double_t beta = pointing.getAngle(*best);
310 const double Efit = best->getE();
311 const double Eraw = best->getW(JENERGY_ENERGY, numeric_limits<double>::min());
312
313 // selection of fit result
314
315 bool ok = (Efit >= Emin_GeV);
316
317 if (ok) {
318
319 job.Fill(5.0);
320
321 hn.Fill((Double_t) best->getNDF());
322 hq.Fill(best->getQ());
323 hi.Fill((Double_t) distance(evt->begin(), best));
324 hc.Fill(best->getDZ());
325
326 // atmospheric muon hypothesis test.
327
328 if (( has_neutrino(*event) && get_neutrino(*event).dir.z >= atmosphere.dot2) || // difference in quality between best upwards and best downgoing track.
329 (!has_neutrino(*event) && best->getDZ() >= atmosphere.dot2)) { // negative (positive) values imply a down(up) going track.
330 hu.Fill(atmosphere(evt->begin(), __end));
331 }
332
333 hx.Fill(max(log10(beta), hx.GetXaxis()->GetXmin()));
334
335 Q.put(beta);
336
337 JTrack3E ta = getTrack(*muon);
338 JTrack3E tb = getTrack(*best);
339
340 ta.add(offset);
341 tb.sub(converter.putTime());
342
343 static_cast<JTrack3D&>(ta).move(ta.getIntersection(tb), getSpeedOfLight()); // move to intersection point of both tracks
344 static_cast<JTrack3D&>(tb).move(tb.getIntersection(ta), getSpeedOfLight());
345
346 hd.Fill((tb.getPosition() - ta.getPosition()).getLength());
347
348 if (has_neutrino(*event)) {
349
350 job.Fill(6.0);
351
352 const Trk neutrino = get_neutrino(*event);
353
354 JPosition3D vertex = getPosition(neutrino);
355
356 vertex.add(offset);
357
358 if (cylinder.is_inside(vertex)) {
359
360 job.Fill(7.0);
361
362 Va.Fill(JVector2D(best->getX() - origin.getX(), best->getY() - origin.getY()).getLengthSquared(), best->getZ()); //R^{2} [m^2], z [m]
363
364 JTrack3E tc = getTrack(*best);
365
366 hz0.Fill(tc.getIntersection(vertex));
367 hz1.Fill(best->getW(JSTART_LENGTH_METRES, 0.0) - gWater(muon->E));
368 }
369 }
370
371 Vb.Fill(JVector2D(best->getX() - origin.getX(), best->getY() - origin.getY()).getLengthSquared(), best->getZ()); //R^{2} [m^2], z [m]
372 ht.Fill(tb.getT() - ta.getT());
373
374 E_0.Fill(volume.getX(E, true));
375 E_1.Fill(volume.getX(Eraw, true));
376 E_2.Fill(volume.getX(Efit, true));
377 E_E.Fill(volume.getX(Efit) - volume.getX(E));
378 ExE.Fill(volume.getX(E), volume.getX(Efit));
379
380 h2.Fill(x, beta);
381
382 if (best->hasW(JSTART_NPE_MIP_TOTAL)) {
383 h1.Fill(log10(best->getW(JSTART_NPE_MIP_TOTAL)));
384 }
385
386 if (best->hasW(JVETO_NPE) && best->hasW(JVETO_NUMBER_OF_HITS)) {
387
388 const double npe = best->getW(JVETO_NPE); // number of photo-electrons
389 const int count = best->getW(JVETO_NUMBER_OF_HITS); // number of hits
390 const double pv = TMath::PoissonI(count, npe); // Poisson distribution function for (#hits,npe)
391
392 hv.Fill(max(log10(pv), hv.GetXaxis()->GetXmin()));
393 }
394 }
395 }
396 }
397 STATUS(endl);
398
399 NOTICE("Number of events input " << setw(8) << right << job.GetBinContent(1) << endl);
400 NOTICE("Number of events with muon " << setw(8) << right << job.GetBinContent(3) << endl);
401 NOTICE("Number of events with fit " << setw(8) << right << job.GetBinContent(4) << endl);
402 NOTICE("Number of events selected " << setw(8) << right << job.GetBinContent(5) << endl);
403 NOTICE("Number of events with neutrino " << setw(8) << right << job.GetBinContent(6) << endl);
404 NOTICE("Number of events contained " << setw(8) << right << job.GetBinContent(7) << endl);
405
406 if (Q.getCount() != 0) {
407
408 NOTICE("Median space angle [deg] " << FIXED (6,3) << Q.getQuantile(0.5) << endl);
409
410 for (double q : {0.50, 0.90, 0.99}) {
411 NOTICE("Space angle " << FIXED(5,1) << (100.0 * q) << "% quantile [deg] " << FIXED(6,3) << Q.getQuantile(q) << endl);
412 }
413 }
414
415 out.Write();
416 out.Close();
417}
string outputFile
#define DEBUG(A)
Message macros.
Definition JMessage.hh:62
#define STATUS(A)
Definition JMessage.hh:63
#define NOTICE(A)
Definition JMessage.hh:64
#define FATAL(A)
Definition JMessage.hh:67
int debug
debug level
Definition JSirene.cc:72
#define make_field(A,...)
macro to convert parameter to JParserTemplateElement object
Definition JParser.hh:2142
std::vector< T >::difference_type distance(typename std::vector< T >::const_iterator first, typename PhysicsEvent::const_iterator< T > second)
Specialisation of STL distance.
Monte Carlo run header.
Definition JHead.hh:1236
double getRadius() const
Get radius.
Definition JCircle2D.hh:144
Data structure for vector in two dimensions.
Definition JVector2D.hh:34
double getLengthSquared() const
Get length squared.
Definition JVector2D.hh:188
double getIntersection(const JVector3D &pos) const
Get longitudinal position along axis of position of closest approach with given position.
Definition JAxis3D.hh:146
double getZmin() const
Get minimal z position.
bool is_inside(const JVector3D &pos) const
Check whether given point is inside cylinder.
double getZmax() const
Get maximal z position.
JCylinder3D & add(const JVector3D &pos)
Add position.
Data structure for position in three dimensions.
const JPosition3D & getPosition() const
Get position.
double getT(const JVector3D &pos) const
Get arrival time of Cherenkov light at given position.
Definition JTrack3D.hh:87
JTime & add(const JTime &value)
Addition operator.
JTime & sub(const JTime &value)
Subtraction operator.
3D track with energy.
Definition JTrack3E.hh:34
JVector3D & add(const JVector3D &vector)
Add vector.
Definition JVector3D.hh:142
Utility class to parse command line options.
Definition JParser.hh:1698
Auxiliary class to evaluate atmospheric muon hypothesis.
Auxiliary class to compare fit results with respect to a reference direction (e.g....
double getAngle(const JFit &fit) const
Get angle between reference direction and fit result.
const_iterator< T > end() const
Get end of data.
const_iterator< T > begin() const
Get begin of data.
Auxiliary class to convert DAQ hit time to/from Monte Carlo hit time.
static const int JENERGY_ENERGY
uncorrected energy [GeV] from JMuonEnergy
static const int JSTART_NPE_MIP_TOTAL
number of photo-electrons along the whole track from JMuonStart
static const int JVETO_NPE
number of photo-electrons from JVeto.cc
static const int JSTART_LENGTH_METRES
distance between projected positions on the track of optical modules for which the response does not ...
static const int JVETO_NUMBER_OF_HITS
number of hits from JVeto.cc
Vec getOrigin(const JHead &header)
Get origin.
JDirection3D getDirection(const Vec &dir)
Get direction.
JTrack3E getTrack(const Trk &track)
Get track.
JCylinder3D getCylinder(const JHead &header)
Get cylinder corresponding to the positions of generated tracks (i.e.
bool has_neutrino(const Evt &evt)
Test whether given event has an incoming neutrino.
JPosition3D getPosition(const Vec &pos)
Get position.
bool is_muon(const Trk &track)
Test whether given track is a (anti-)muon.
Vec getOffset(const JHead &header)
Get offset.
const Trk & get_neutrino(const Evt &evt)
Get incoming neutrino.
size_t getCount(const array_type< T > &buffer, const JCompare_t &compare)
Count number of unique values.
This name space includes all other name spaces (except KM3NETDAQ, KM3NET and ANTARES).
counter_type advance(counter_type &counter, const counter_type value, const counter_type limit=std::numeric_limits< counter_type >::max())
Advance counter.
Head getCommonHeader(const JMultipleFileScanner_t &file_list)
Get common Monte Carlo header.
KM3NeT DAQ data structures and auxiliaries.
Definition DataQueue.cc:39
const char *const neutrino_t
Definition io_ascii.hh:29
The Evt class respresent a Monte Carlo (MC) event as well as an offline event.
Definition Evt.hh:21
Auxiliary data structure for floating point format specification.
Definition JManip.hh:448
Auxiliary class for histogramming of effective volume.
Definition JVolume.hh:29
Primary particle.
Definition JHead.hh:1174
Acoustic event fit.
Auxiliary class to test history.
Definition JHistory.hh:188
Empty structure for specification of parser element that is initialised (i.e. does not require input)...
Definition JParser.hh:68
General purpose sorter of fit results.
Auxiliary class for defining the range of iterations of objects.
Definition JLimit.hh:45
const JLimit & getLimit() const
Get limit.
Definition JLimit.hh:84
static counter_type max()
Get maximum counter value.
Definition JLimit.hh:128
Auxiliary class to synchronously read DAQ events and Monte Carlo events (and optionally other events)...
Auxiliary data structure for running average, standard deviation and quantiles.
Definition JQuantile.hh:41
The Trk class represents a Monte Carlo (MC) particle as well as a reconstructed track/shower.
Definition Trk.hh:15
Vec dir
track direction
Definition Trk.hh:18
double E
Energy [GeV] (either MC truth or reconstructed)
Definition Trk.hh:20
double z
Definition Vec.hh:14
Reconstruction type dependent comparison of track quality.