GENIEGenerator
Loading...
Searching...
No Matches
Functions | Variables
NuSmear.h File Reference
#include <iostream>
#include <string>
#include <map>
#include <random>
#include "GHepParticle.h"
#include "PDGCodes.h"
#include "TVector3.h"
#include "PDGUtils.h"
Include dependency graph for NuSmear.h:

Go to the source code of this file.

Functions

double PiP_Res_duneCdr (double myE, double myKE, double myPmag)
double PiM_Res_duneCdr (double myE, double myKE, double myPmag)
double Pi0_Res_duneCdr (double myE, double myKE, double myPmag)
double KP_Res_duneCdr (double myE, double myKE, double myPmag)
double KM_Res_duneCdr (double myE, double myKE, double myPmag)
double Gamma_Res_duneCdr (double myE, double myKE, double myPmag)
double Proton_Res_duneCdr (double myE, double myKE, double myPmag)
double Electron_Res_duneCdr (double myE, double myKE, double myPmag)
double Positron_Res_duneCdr (double myE, double myKE, double myPmag)
double Muon_Res_duneCdr (double myE, double myKE, double myPmag)
double AntiMuon_Res_duneCdr (double myE, double myKE, double myPmag)
double Neutron_Res_duneCdr (double myE, double myKE, double myPmag)
double K0_Res_duneCdr (double myE, double myKE, double myPmag)
double AntiK0_Res_duneCdr (double myE, double myKE, double myPmag)
double Lambda_Res_duneCdr (double myE, double myKE, double myPmag)
double SigmaP_Res_duneCdr (double myE, double myKE, double myPmag)
double Sigma0_Res_duneCdr (double myE, double myKE, double myPmag)
double SigmaM_Res_duneCdr (double myE, double myKE, double myPmag)
std::mt19937 gen (rd())
double smearE (int myPdg, double myE, double myKE, double myPx, double myPy, double myPz, std::string model)
double smearA (int myPdg, double myPx, double myPy, double myPz, std::string model)

Variables

std::random_device rd

Function Documentation

◆ AntiK0_Res_duneCdr()

double AntiK0_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 196 of file NuSmear.h.

197{
198 if (myKE >= 0.05)
199 {
200 return pow((pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2)), 0.5);
201 }
202 else
203 {
204 return -1;
205 }
206}

Referenced by smearE().

◆ AntiMuon_Res_duneCdr()

double AntiMuon_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 160 of file NuSmear.h.

161{
162 if (myKE >= 0.03)
163 {
164 return 0.15;
165 }
166 else
167 {
168 return -1;
169 }
170}

Referenced by smearE().

◆ Electron_Res_duneCdr()

double Electron_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 124 of file NuSmear.h.

125{
126 if (myKE >= 0.03)
127 {
128 return pow(pow(0.02, 2) + pow(0.15 / (pow(myE, 0.5)), 2), 0.5);
129 }
130 else
131 {
132 return -1;
133 }
134}

Referenced by smearE().

◆ Gamma_Res_duneCdr()

double Gamma_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 93 of file NuSmear.h.

94{
95 if (myKE >= 0.03)
96 {
97 return pow(pow(0.02, 2) + pow(0.15 / (pow(myE, 0.5)), 2), 0.5);
98 }
99 else
100 {
101 return -1;
102 }
103}

Referenced by smearE().

◆ gen()

std::mt19937 gen ( rd() )

Referenced by genie::ZExpELFormFactorModel::FixEL0(), genie::BBA03ELFormFactorsModel::Gen(), genie::BBA05ELFormFactorsModel::Gen(), genie::BBA07ELFormFactorsModel::Gen(), genie::GalsterELFormFactorsModel::Gen(), genie::ZExpELFormFactorModel::Gen(), smearA(), and smearE().

◆ K0_Res_duneCdr()

double K0_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 184 of file NuSmear.h.

185{
186 if (myKE >= 0.05)
187 {
188 return pow((pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2)), 0.5);
189 }
190 else
191 {
192 return -1;
193 }
194}

Referenced by smearE().

◆ KM_Res_duneCdr()

double KM_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 81 of file NuSmear.h.

82{
83 if (myKE >= 0.05)
84 {
85 return pow(pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2), 0.5);
86 }
87 else
88 {
89 return -1;
90 }
91}

Referenced by smearE().

◆ KP_Res_duneCdr()

double KP_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 69 of file NuSmear.h.

70{
71 if (myKE >= 0.05)
72 {
73 return pow((pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2)), 0.5);
74 }
75 else
76 {
77 return -1;
78 }
79}

Referenced by smearE().

◆ Lambda_Res_duneCdr()

double Lambda_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 208 of file NuSmear.h.

209{
210 if (myKE >= 0.05)
211 {
212 return pow((pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2)), 0.5);
213 }
214 else
215 {
216 return -1;
217 }
218}

Referenced by smearE().

◆ Muon_Res_duneCdr()

double Muon_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 148 of file NuSmear.h.

149{
150 if (myKE >= 0.03)
151 {
152 return 0.15;
153 }
154 else
155 {
156 return -1;
157 }
158}

Referenced by smearE().

◆ Neutron_Res_duneCdr()

double Neutron_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 172 of file NuSmear.h.

173{
174 if (myKE >= 0.05)
175 {
176 return 0.4 / (pow(myKE, 0.5));
177 }
178 else
179 {
180 return -1;
181 }
182}

Referenced by smearE().

◆ Pi0_Res_duneCdr()

double Pi0_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 57 of file NuSmear.h.

58{
59 if (myKE >= 0.05)
60 {
61 return pow((pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2)), 0.5);
62 }
63 else
64 {
65 return -1;
66 }
67}

Referenced by smearE().

◆ PiM_Res_duneCdr()

double PiM_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 45 of file NuSmear.h.

46{
47 if (myKE >= 0.1)
48 {
49 return 0.15;
50 }
51 else
52 {
53 return -1;
54 }
55}

Referenced by smearE().

◆ PiP_Res_duneCdr()

double PiP_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 33 of file NuSmear.h.

34{
35 if (myKE >= 0.1)
36 {
37 return 0.15;
38 }
39 else
40 {
41 return -1;
42 }
43}

Referenced by smearE().

◆ Positron_Res_duneCdr()

double Positron_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 136 of file NuSmear.h.

137{
138 if (myKE >= 0.03)
139 {
140 return pow(pow(0.02, 2) + pow(0.15 / (pow(myE, 0.5)), 2), 0.5);
141 }
142 else
143 {
144 return -1;
145 }
146}

Referenced by smearE().

◆ Proton_Res_duneCdr()

double Proton_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 105 of file NuSmear.h.

106{
107 if (myKE >= 0.05)
108 {
109 if (myPmag < 0.4)
110 {
111 return 0.1;
112 }
113 else
114 {
115 return pow(pow(0.05, 2) + pow(0.3 / (pow(myKE, 0.5)), 2), 0.5);
116 }
117 }
118 else
119 {
120 return -1;
121 }
122}

Referenced by smearE().

◆ Sigma0_Res_duneCdr()

double Sigma0_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 232 of file NuSmear.h.

233{
234 if (myKE >= 0.05)
235 {
236 return pow((pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2)), 0.5);
237 }
238 else
239 {
240 return -1;
241 }
242}

Referenced by smearE().

◆ SigmaM_Res_duneCdr()

double SigmaM_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 244 of file NuSmear.h.

245{
246 if (myKE >= 0.05)
247 {
248 return pow((pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2)), 0.5);
249 }
250 else
251 {
252 return -1;
253 }
254}

Referenced by smearE().

◆ SigmaP_Res_duneCdr()

double SigmaP_Res_duneCdr ( double myE,
double myKE,
double myPmag )

Definition at line 220 of file NuSmear.h.

221{
222 if (myKE >= 0.05)
223 {
224 return pow((pow(0.05, 2) + pow((0.3 / (pow(myE, 0.5))), 2)), 0.5);
225 }
226 else
227 {
228 return -1;
229 }
230}

Referenced by smearE().

◆ smearA()

double smearA ( int myPdg,
double myPx,
double myPy,
double myPz,
std::string model )

Definition at line 440 of file NuSmear.h.

441{
442 std::map<int, int> myMap;
443
444 myMap[kPdgPiP] = 0;
445 myMap[kPdgPiM] = 1;
446 myMap[kPdgPi0] = 2;
447 myMap[kPdgKP] = 3;
448 myMap[kPdgKM] = 4;
449 myMap[kPdgGamma] = 5;
450 myMap[kPdgProton] = 6;
451 myMap[kPdgElectron] = 7;
452 myMap[kPdgPositron] = 8;
453 myMap[kPdgMuon] = 9;
454 myMap[kPdgAntiMuon] = 10;
455 myMap[kPdgNeutron] = 11;
456 myMap[kPdgK0] = 12;
457 myMap[kPdgAntiK0] = 13;
458 myMap[kPdgLambda] = 14;
459 myMap[kPdgSigmaP] = 15;
460 myMap[kPdgSigma0] = 16;
461 myMap[kPdgSigmaM] = 17;
462
463 TVector3 P(myPx, myPy, myPz);
464 TVector3 Z(0, 0, 1);
465 double theta = (P.Angle(Z)) / M_PI * 180;
466
467 int in = myMap.find(myPdg)->second;
468
469 // DUNE-CDR and Default angular smearing values
470
471 double angularResDeg_DuneCdr[18] = {
472 1,
473 1,
474 5,
475 5,
476 5,
477 1,
478 5,
479 1,
480 1,
481 1,
482 1,
483 5,
484 5,
485 5,
486 5,
487 5,
488 5,
489 5};
490
491 double angularResDeg_Default[18] = {
492 2,
493 2,
494 8,
495 2,
496 2,
497 3,
498 8,
499 2,
500 2,
501 2,
502 2,
503 10,
504 8,
505 8,
506 8,
507 8,
508 8,
509 8};
510
511 double resolution = 0;
512
513 if (model == "duneCdr")
514 {
515 resolution += (angularResDeg_DuneCdr[in]);
516 }
517 else if (model == "default")
518 {
519 resolution += (angularResDeg_Default[in]);
520 }
521
522 std::normal_distribution<double> distNorm(theta, resolution);
523 return distNorm(gen);
524}
gen
std::mt19937 gen(rd())
genie::kPdgPiM
const int kPdgPiM
Definition PDGCodes.h:159
genie::kPdgSigma0
const int kPdgSigma0
Definition PDGCodes.h:88
genie::kPdgAntiMuon
const int kPdgAntiMuon
Definition PDGCodes.h:38
genie::kPdgProton
const int kPdgProton
Definition PDGCodes.h:81
genie::kPdgSigmaP
const int kPdgSigmaP
Definition PDGCodes.h:87
genie::kPdgPi0
const int kPdgPi0
Definition PDGCodes.h:160
genie::kPdgKP
const int kPdgKP
Definition PDGCodes.h:172
genie::kPdgNeutron
const int kPdgNeutron
Definition PDGCodes.h:83
genie::kPdgKM
const int kPdgKM
Definition PDGCodes.h:173
genie::kPdgPiP
const int kPdgPiP
Definition PDGCodes.h:158
genie::kPdgLambda
const int kPdgLambda
Definition PDGCodes.h:85
genie::kPdgSigmaM
const int kPdgSigmaM
Definition PDGCodes.h:89
genie::kPdgMuon
const int kPdgMuon
Definition PDGCodes.h:37
genie::kPdgPositron
const int kPdgPositron
Definition PDGCodes.h:36
genie::kPdgGamma
const int kPdgGamma
Definition PDGCodes.h:189
genie::kPdgAntiK0
const int kPdgAntiK0
Definition PDGCodes.h:175
genie::kPdgElectron
const int kPdgElectron
Definition PDGCodes.h:35
genie::kPdgK0
const int kPdgK0
Definition PDGCodes.h:174

References gen(), genie::kPdgAntiK0, genie::kPdgAntiMuon, genie::kPdgElectron, genie::kPdgGamma, genie::kPdgK0, genie::kPdgKM, genie::kPdgKP, genie::kPdgLambda, genie::kPdgMuon, genie::kPdgNeutron, genie::kPdgPi0, genie::kPdgPiM, genie::kPdgPiP, genie::kPdgPositron, genie::kPdgProton, genie::kPdgSigma0, genie::kPdgSigmaM, and genie::kPdgSigmaP.

◆ smearE()

double smearE ( int myPdg,
double myE,
double myKE,
double myPx,
double myPy,
double myPz,
std::string model )

Definition at line 263 of file NuSmear.h.

264{
265 std::map<int, int> myMap;
266
267 myMap[kPdgPiP] = 0;
268 myMap[kPdgPiM] = 1;
269 myMap[kPdgPi0] = 2;
270 myMap[kPdgKP] = 3;
271 myMap[kPdgKM] = 4;
272 myMap[kPdgGamma] = 5;
273 myMap[kPdgProton] = 6;
274 myMap[kPdgElectron] = 7;
275 myMap[kPdgPositron] = 8;
276 myMap[kPdgMuon] = 9;
277 myMap[kPdgAntiMuon] = 10;
278 myMap[kPdgNeutron] = 11;
279 myMap[kPdgK0] = 12;
280 myMap[kPdgAntiK0] = 13;
281 myMap[kPdgLambda] = 14;
282 myMap[kPdgSigmaP] = 15;
283 myMap[kPdgSigma0] = 16;
284 myMap[kPdgSigmaM] = 17;
285
286 TVector3 P(myPx, myPy, myPz);
287 TVector3 Z(0, 0, 1);
288 double myPmag = P.Mag();
289
290 int in = myMap.find(myPdg)->second;
291
292 if (model == "duneCdr")
293 {
294
295 double (*resolution_ptr_duneCdr[18])(double, double, double) = {
296 PiP_Res_duneCdr,
297 PiM_Res_duneCdr,
298 Pi0_Res_duneCdr,
299 KP_Res_duneCdr,
300 KM_Res_duneCdr,
301 Gamma_Res_duneCdr,
302 Proton_Res_duneCdr,
303 Electron_Res_duneCdr,
304 Positron_Res_duneCdr,
305 Muon_Res_duneCdr,
306 AntiMuon_Res_duneCdr,
307 Neutron_Res_duneCdr,
308 K0_Res_duneCdr,
309 AntiK0_Res_duneCdr,
310 Lambda_Res_duneCdr,
311 SigmaP_Res_duneCdr,
312 Sigma0_Res_duneCdr,
313 SigmaM_Res_duneCdr};
314
315 double resolution = (*resolution_ptr_duneCdr[in])(myE, myKE, myPmag);
316
317 if (resolution == -1)
318 {
319 return 0;
320 }
321 else
322 {
323 double var = 0;
324 double eSq = 0;
325
326 if (pdg::IsNeutronOrProton(myPdg))
327 {
328 var += pow((resolution * myKE), 2);
329 eSq += pow(myKE, 2);
330 }
331 else
332 {
333 var += pow((resolution * myE), 2);
334 eSq += pow(myE, 2);
335 }
336
337 double m = log(eSq / (pow(var + eSq, 0.5)));
338 double s = pow(log(1 + (var / eSq)), 0.5);
339
340 std::lognormal_distribution<double> distLognorm(m, s);
341
342 if (myPdg == kPdgNeutron)
343 {
344 if (myPmag < 1)
345 {
346 std::uniform_real_distribution<double> distUni(0, 1);
347
348 if (distUni(gen) < 0.1)
349 {
350 return 0;
351 }
352 else
353 {
354 return 0.6 * (distLognorm(gen));
355 }
356 }
357 else
358 {
359 return 0.6 * (distLognorm(gen));
360 }
361 }
362 else
363 {
364 return distLognorm(gen);
365 }
366 }
367 }
368
369 // Default model resolutions and particle detection dependencies
370
371 else if (model == "default")
372 {
373 double info[18][2] = {
374
375 {0.15, 1},
376 {0.15, 1},
377 {0.15, 1},
378 {0.2, 1},
379 {0.2, 1},
380 {0.3, 0.5},
381 {0.4, 1},
382 {0.4, 1},
383 {0.4, 1},
384 {0.15, 1},
385 {0.15, 1},
386 {0.5, 0.5},
387 {0.2, 1},
388 {0.2, 1},
389 {0.3, 1},
390 {0.3, 1},
391 {0.3, 1},
392 {0.3, 1}};
393
394 double resolution = info[in][0];
395 double chanceToSee = info[in][1];
396 double var = 0;
397 double eSq = 0;
398
399 if (pdg::IsNeutronOrProton(myPdg))
400 {
401 var += pow((resolution * myKE), 2);
402 eSq += pow(myKE, 2);
403 }
404 else
405 {
406 var += pow((resolution * myE), 2);
407 eSq += pow(myE, 2);
408 }
409
410 double m = log(eSq / (pow(var + eSq, 0.5)));
411 double s = pow(log(1 + (var / eSq)), 0.5);
412
413 std::lognormal_distribution<double> distLognorm(m, s);
414 std::uniform_real_distribution<double> distUni(0, 1);
415
416 if (distUni(gen) < chanceToSee)
417 {
418 if (myKE > 0.05)
419 {
420 return distLognorm(gen);
421 }
422 else
423 {
424 return 0;
425 }
426 }
427 else
428 {
429 return 0;
430 }
431 }
432 else
433 {
434 std::cout << "Error: Resolution model not found./n";
435 }
436}
Gamma_Res_duneCdr
double Gamma_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:93
Electron_Res_duneCdr
double Electron_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:124
Lambda_Res_duneCdr
double Lambda_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:208
AntiK0_Res_duneCdr
double AntiK0_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:196
Positron_Res_duneCdr
double Positron_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:136
PiM_Res_duneCdr
double PiM_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:45
K0_Res_duneCdr
double K0_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:184
Neutron_Res_duneCdr
double Neutron_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:172
AntiMuon_Res_duneCdr
double AntiMuon_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:160
SigmaP_Res_duneCdr
double SigmaP_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:220
Pi0_Res_duneCdr
double Pi0_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:57
SigmaM_Res_duneCdr
double SigmaM_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:244
Proton_Res_duneCdr
double Proton_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:105
PiP_Res_duneCdr
double PiP_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:33
KP_Res_duneCdr
double KP_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:69
Muon_Res_duneCdr
double Muon_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:148
Sigma0_Res_duneCdr
double Sigma0_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:232
KM_Res_duneCdr
double KM_Res_duneCdr(double myE, double myKE, double myPmag)
Definition NuSmear.h:81
genie::pdg::IsNeutronOrProton
bool IsNeutronOrProton(int pdgc)
Definition PDGUtils.cxx:351

References AntiK0_Res_duneCdr(), AntiMuon_Res_duneCdr(), Electron_Res_duneCdr(), Gamma_Res_duneCdr(), gen(), genie::pdg::IsNeutronOrProton(), K0_Res_duneCdr(), KM_Res_duneCdr(), KP_Res_duneCdr(), genie::kPdgAntiK0, genie::kPdgAntiMuon, genie::kPdgElectron, genie::kPdgGamma, genie::kPdgK0, genie::kPdgKM, genie::kPdgKP, genie::kPdgLambda, genie::kPdgMuon, genie::kPdgNeutron, genie::kPdgPi0, genie::kPdgPiM, genie::kPdgPiP, genie::kPdgPositron, genie::kPdgProton, genie::kPdgSigma0, genie::kPdgSigmaM, genie::kPdgSigmaP, Lambda_Res_duneCdr(), Muon_Res_duneCdr(), Neutron_Res_duneCdr(), Pi0_Res_duneCdr(), PiM_Res_duneCdr(), PiP_Res_duneCdr(), Positron_Res_duneCdr(), Proton_Res_duneCdr(), Sigma0_Res_duneCdr(), SigmaM_Res_duneCdr(), and SigmaP_Res_duneCdr().

Variable Documentation

◆ rd

std::random_device rd

Definition at line 258 of file NuSmear.h.

Generated on for GENIEGenerator by doxygen 1.14.0