GENIEGenerator
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
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

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

Definition at line 196 of file NuSmear.h.

Referenced by smearE().

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 }
double AntiMuon_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 160 of file NuSmear.h.

Referenced by smearE().

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

Definition at line 124 of file NuSmear.h.

Referenced by smearE().

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 }
double Gamma_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 93 of file NuSmear.h.

Referenced by smearE().

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 }
std::mt19937 gen ( rd()  )

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

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

Definition at line 184 of file NuSmear.h.

Referenced by smearE().

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 }
double KM_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 81 of file NuSmear.h.

Referenced by smearE().

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 }
double KP_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 69 of file NuSmear.h.

Referenced by smearE().

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 }
double Lambda_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 208 of file NuSmear.h.

Referenced by smearE().

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 }
double Muon_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 148 of file NuSmear.h.

Referenced by smearE().

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

Definition at line 172 of file NuSmear.h.

Referenced by smearE().

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

Definition at line 57 of file NuSmear.h.

Referenced by smearE().

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 }
double PiM_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 45 of file NuSmear.h.

Referenced by smearE().

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

Definition at line 33 of file NuSmear.h.

Referenced by smearE().

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

Definition at line 136 of file NuSmear.h.

Referenced by smearE().

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 }
double Proton_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 105 of file NuSmear.h.

Referenced by smearE().

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 }
double Sigma0_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 232 of file NuSmear.h.

Referenced by smearE().

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 }
double SigmaM_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 244 of file NuSmear.h.

Referenced by smearE().

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 }
double SigmaP_Res_duneCdr ( double  myE,
double  myKE,
double  myPmag 
)

Definition at line 220 of file NuSmear.h.

Referenced by smearE().

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 }
double smearA ( int  myPdg,
double  myPx,
double  myPy,
double  myPz,
std::string  model 
)

Definition at line 440 of file NuSmear.h.

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.

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 }
genie::kPdgLambda
const int kPdgLambda
Definition: PDGCodes.h:85
genie::kPdgAntiMuon
const int kPdgAntiMuon
Definition: PDGCodes.h:38
genie::kPdgSigma0
const int kPdgSigma0
Definition: PDGCodes.h:88
gen
std::mt19937 gen(rd())
genie::kPdgElectron
const int kPdgElectron
Definition: PDGCodes.h:35
genie::kPdgK0
const int kPdgK0
Definition: PDGCodes.h:174
genie::kPdgKM
const int kPdgKM
Definition: PDGCodes.h:173
genie::kPdgGamma
const int kPdgGamma
Definition: PDGCodes.h:189
genie::kPdgKP
const int kPdgKP
Definition: PDGCodes.h:172
genie::kPdgPiP
const int kPdgPiP
Definition: PDGCodes.h:158
genie::kPdgPi0
const int kPdgPi0
Definition: PDGCodes.h:160
genie::kPdgAntiK0
const int kPdgAntiK0
Definition: PDGCodes.h:175
genie::kPdgSigmaM
const int kPdgSigmaM
Definition: PDGCodes.h:89
genie::kPdgPiM
const int kPdgPiM
Definition: PDGCodes.h:159
genie::kPdgSigmaP
const int kPdgSigmaP
Definition: PDGCodes.h:87
genie::kPdgProton
const int kPdgProton
Definition: PDGCodes.h:81
genie::kPdgMuon
const int kPdgMuon
Definition: PDGCodes.h:37
genie::kPdgPositron
const int kPdgPositron
Definition: PDGCodes.h:36
genie::kPdgNeutron
const int kPdgNeutron
Definition: PDGCodes.h:83
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.

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(), genie::units::m, Muon_Res_duneCdr(), Neutron_Res_duneCdr(), Pi0_Res_duneCdr(), PiM_Res_duneCdr(), PiP_Res_duneCdr(), Positron_Res_duneCdr(), Proton_Res_duneCdr(), genie::units::s, Sigma0_Res_duneCdr(), SigmaM_Res_duneCdr(), and SigmaP_Res_duneCdr().

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
Proton_Res_duneCdr
double Proton_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:105
genie::kPdgLambda
const int kPdgLambda
Definition: PDGCodes.h:85
AntiMuon_Res_duneCdr
double AntiMuon_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:160
Electron_Res_duneCdr
double Electron_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:124
genie::units::s
static constexpr double s
Definition: Units.h:95
genie::kPdgAntiMuon
const int kPdgAntiMuon
Definition: PDGCodes.h:38
K0_Res_duneCdr
double K0_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:184
genie::kPdgSigma0
const int kPdgSigma0
Definition: PDGCodes.h:88
gen
std::mt19937 gen(rd())
Sigma0_Res_duneCdr
double Sigma0_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:232
genie::kPdgElectron
const int kPdgElectron
Definition: PDGCodes.h:35
KP_Res_duneCdr
double KP_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:69
genie::kPdgK0
const int kPdgK0
Definition: PDGCodes.h:174
AntiK0_Res_duneCdr
double AntiK0_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:196
SigmaM_Res_duneCdr
double SigmaM_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:244
Positron_Res_duneCdr
double Positron_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:136
genie::kPdgKM
const int kPdgKM
Definition: PDGCodes.h:173
genie::kPdgGamma
const int kPdgGamma
Definition: PDGCodes.h:189
genie::kPdgKP
const int kPdgKP
Definition: PDGCodes.h:172
genie::kPdgPiP
const int kPdgPiP
Definition: PDGCodes.h:158
genie::kPdgPi0
const int kPdgPi0
Definition: PDGCodes.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
genie::kPdgAntiK0
const int kPdgAntiK0
Definition: PDGCodes.h:175
PiP_Res_duneCdr
double PiP_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:33
Lambda_Res_duneCdr
double Lambda_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:208
genie::kPdgSigmaM
const int kPdgSigmaM
Definition: PDGCodes.h:89
Neutron_Res_duneCdr
double Neutron_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:172
genie::pdg::IsNeutronOrProton
bool IsNeutronOrProton(int pdgc)
Definition: PDGUtils.cxx:351
genie::kPdgPiM
const int kPdgPiM
Definition: PDGCodes.h:159
genie::kPdgSigmaP
const int kPdgSigmaP
Definition: PDGCodes.h:87
genie::kPdgProton
const int kPdgProton
Definition: PDGCodes.h:81
PiM_Res_duneCdr
double PiM_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:45
Muon_Res_duneCdr
double Muon_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:148
genie::kPdgMuon
const int kPdgMuon
Definition: PDGCodes.h:37
genie::kPdgPositron
const int kPdgPositron
Definition: PDGCodes.h:36
genie::kPdgNeutron
const int kPdgNeutron
Definition: PDGCodes.h:83
genie::units::m
static constexpr double m
Definition: Units.h:71
KM_Res_duneCdr
double KM_Res_duneCdr(double myE, double myKE, double myPmag)
Definition: NuSmear.h:81

Variable Documentation

std::random_device rd

Definition at line 258 of file NuSmear.h.

Generated on Wed Jul 17 2024 14:46:50 for GENIEGenerator by   doxygen 1.8.5