test
CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
TauDecay.cc
Go to the documentation of this file.
3 #include <iomanip>
4 #include <cstdlib>
5 #include <iostream>
6 
8  Reset();
9 }
10 
12 
13 }
14 
16  n_pi=0;
17  n_pi0=0;
18  n_K=0;
19  n_K0L=0;
20  n_K0S=0;
21  n_gamma=0;
22  n_nu=0;
23  n_e=0;
24  n_mu=0;
25  n_a1=0;
26  n_a10=0;
27  n_rho=0;
28  n_rho0=0;
29  n_eta=0;
30  n_omega=0;
31  n_Kstar0=0;
32  n_Kstar=0;
33  unknown=0;
34 }
35 
37  int id=abs(pdgid);
38  if(id==PdtPdgMini::e_minus) return true; // e+-
39  if(id==PdtPdgMini::nu_e) return true; // nu_e
40  if(id==PdtPdgMini::mu_minus) return true; // mu+-
41  if(id==PdtPdgMini::nu_mu) return true; // nu_mu
42  if(id==PdtPdgMini::nu_tau) return true; // nu_tau
43  if(id==PdtPdgMini::gamma) return true; // gamma happends in generator
44  if(id==PdtPdgMini::pi0) return true; // pi0
45  if(id==PdtPdgMini::pi_plus) return true; // pi+-
46  if(id==PdtPdgMini::K_L0) return true; // K0L
47  if(id==PdtPdgMini::K_S0) return true; // KS
48  if(id==PdtPdgMini::eta) return true;
49  if(id==PdtPdgMini::omega) return true;
50 
51  if(id==PdtPdgMini::K_plus) return true; // K+-
52  return false;
53 }
54 
56  int id=abs(pdgid);
57  //count particles
58  if(id==PdtPdgMini::pi_plus) { n_pi++; return true;}
59  if(id==PdtPdgMini::pi0) { n_pi0++; return true;}
60  if(id==PdtPdgMini::K_plus) { n_K++; return true;}
61  if(id==PdtPdgMini::K_L0) { n_K0L++; return true;}
62  if(id==PdtPdgMini::K_S0) { n_K0S++; return true;}
63  if(id==PdtPdgMini::eta) { n_eta++; return true;}
64  if(id==PdtPdgMini::omega) { n_omega++; return true;}
65  if(id==PdtPdgMini::gamma) { n_gamma++; return true;}
66  if(id==PdtPdgMini::nu_tau ||
67  id==PdtPdgMini::nu_e ||
68  id==PdtPdgMini::nu_mu) { n_nu++; return true;}
69  if(id==PdtPdgMini::e_minus) { n_e++; return true;}
70  if(id==PdtPdgMini::mu_minus) { n_mu++; return true;}
71  if(abs(id)==PdtPdgMini::K0) std::cout << "TauDecay::isTauParticleCounter: ERROR unidentified Particle: " << id << std::endl;
72  return false;
73 }
74 
76  int id=abs(pdgid);
77  //count resonances
78  if(id==PdtPdgMini::a_1_plus) { n_a1++; return true;}
79  if(id==PdtPdgMini::a_10) { n_a10++; return true;}
80  if(id==PdtPdgMini::rho_plus) { n_rho++; return true;}
81  if(id==PdtPdgMini::rho0) { n_rho0++; return true;}
82  if(id==PdtPdgMini::K_star0) { n_Kstar0++; return true;}
83  if(id==PdtPdgMini::K_star_plus){ n_Kstar++; return true;}
84  if(id==PdtPdgMini::W_plus) { return true;}
85  unknown++;
86  return false;
87 }
88 
89 void TauDecay::ClassifyDecayMode(unsigned int &MODE_ID,unsigned int &TauBitMask){
90  //Reset Bits
91  MODE_ID=0;
92  TauBitMask=0;
93  // Classify according to MODE and TauDecayStructure
94  if(n_pi+n_K+n_e+n_mu==1)TauBitMask=OneProng;
95  if(n_pi+n_K==3)TauBitMask=ThreeProng;
96  if(n_pi+n_K==5)TauBitMask=FiveProng;
97  if(n_pi0==1)TauBitMask+=OnePi0;
98  if(n_pi0==2)TauBitMask+=TwoPi0;
99  if(n_pi0==3)TauBitMask+=ThreePi0;
100  ClassifyDecayResonance(TauBitMask);
101 
102  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==0 && n_K==0 && n_K0S+n_K0L==2 && n_nu==1){
103  MODE_ID=MODE_K0BK0PI; return;
104  }
105  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==0 && n_K==0 && (n_K0L+n_K0S)==1 && n_nu==1){
106  MODE_ID=MODE_K0PI; return;
107  }
108  if(n_e==0 && n_mu==0 && n_pi==0 && n_pi0==1 && n_K==1 && n_K0L==0 && n_K0S==0 && n_nu==1){
109  MODE_ID=MODE_KPI0; return;
110  }
111  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==1 && n_K==0 && n_K0L+n_K0S==1 && n_nu==1){
112  MODE_ID=MODE_PIK0PI0; return;
113  }
114  if(n_e==0 && n_mu==0 && n_pi==0 && n_pi0==0 && n_K==1 && (n_K0L+n_K0S)==1 && n_nu==1){
115  MODE_ID=MODE_KK0B; return;
116  }
117  if(n_e==0 && n_mu==0 && n_pi==0 && n_pi0==1 && n_K==1 && n_K0L+n_K0S==1 && n_nu==1){
118  MODE_ID=MODE_KK0BPI0; return;
119  }
120  if(n_e==1 && n_mu==0 && n_pi==0 && n_pi0==0 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==2){
121  MODE_ID=MODE_ELECTRON; return;
122  }
123  if(n_e==0 && n_mu==1 && n_pi==0 && n_pi0==0 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==2){
124  MODE_ID=MODE_MUON; return;
125  }
126  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==0 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
127  MODE_ID=MODE_PION; return;
128  }
129  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==1 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
130  MODE_ID=MODE_PIPI0; return;
131  }
132  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==2 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
133  MODE_ID=MODE_PI2PI0; return;
134  }
135  if(n_e==0 && n_mu==0 && n_pi==3 && n_pi0==0 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
136  MODE_ID=MODE_3PI; return;
137  }
138  if(n_e==0 && n_mu==0 && n_pi==0 && n_pi0==0 && n_K==1 && n_K0L==0 && n_K0S==0 && n_nu==1){
139  MODE_ID=MODE_KAON; return;
140  }
141  if(n_e==0 && n_mu==0 && n_pi==3 && n_pi0==1 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
142  MODE_ID=MODE_3PIPI0; return;
143  }
144  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==3 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
145  MODE_ID=MODE_PI3PI0; return;
146  }
147  if(n_e==0 && n_mu==0 && n_pi==3 && n_pi0==2 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
148  MODE_ID=MODE_3PI2PI0; return;
149  }
150  if(n_e==0 && n_mu==0 && n_pi==5 && n_pi0==0 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
151  MODE_ID=MODE_5PI; return;
152  }
153  if(n_e==0 && n_mu==0 && n_pi==5 && n_pi0==1 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
154  MODE_ID=MODE_5PIPI0; return;
155  }
156  if(n_e==0 && n_mu==0 && n_pi==3 && n_pi0==3 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
157  MODE_ID=MODE_3PI3PI0; return;
158  }
159  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==0 && n_K==2 && n_K0L==0 && n_K0S==0 && n_nu==1){
160  MODE_ID=MODE_KPIK; return;
161  }
162  if(n_e==0 && n_mu==0 && n_pi==0 && n_pi0==2 && n_K==1 && n_K0L==0 && n_K0S==0 && n_nu==1){
163  MODE_ID=MODE_K2PI0; return;
164  }
165  if(n_e==0 && n_mu==0 && n_pi==2 && n_pi0==0 && n_K==1 && n_K0L==0 && n_K0S==0 && n_nu==1){
166  MODE_ID=MODE_KPIPI; return;
167  }
168  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==1 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1 && n_gamma>=1 && n_rho==0){
169  MODE_ID=MODE_PIPI0GAM; return ; // Obsolete should not be called
170  }
171  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==4 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1){
172  MODE_ID=MODE_PI4PI0; return;
173  }
174  if(n_e==0 && n_mu==0 && n_pi==3 && n_pi0==0 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1 && n_eta==1){
175  MODE_ID=MODE_3PIETA; return;
176  }
177  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==2 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1 && n_eta==1){
178  MODE_ID=MODE_PI2PI0ETA; return;
179  }
180  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==2 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1 && n_omega==1){
181  MODE_ID=MODE_PI2PI0OMEGA; return;
182  }
183  if(n_e==0 && n_mu==0 && n_pi==3 && n_pi0==0 && n_K==0 && n_K0L==0 && n_K0S==0 && n_nu==1 && n_omega==1){
184  MODE_ID=MODE_3PIOMEGA; return;
185  }
186  if(n_e==0 && n_mu==0 && n_pi==0 && n_pi0==0 && n_K==1 && n_K0L==0 && n_K0S==0 && n_nu==1 && n_omega==1){
187  MODE_ID=MODE_KOMEGA; return;
188  }
189  if(n_e==0 && n_mu==0 && n_pi==0 && n_pi0==3 && n_K==1 && n_K0L==0 && n_K0S==0 && n_nu==1){
190  MODE_ID=MODE_K3PI0; return;
191  }
192  if(n_e==0 && n_mu==0 && n_pi==2 && n_pi0==1 && n_K==1 && n_K0L==0 && n_K0S==0 && n_nu==1){
193  MODE_ID=MODE_K2PIPI0; return;
194  }
195  if(n_e==0 && n_mu==0 && n_pi==0 && n_pi0==0 && n_K==1 && n_K0L==0 && n_K0S==0 && n_nu==1 && n_eta==1){
196  MODE_ID=MODE_KETA; return;
197  }
198  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==2 && n_K==0 && (n_K0L+n_K0S)==0 && n_nu==1){
199  MODE_ID=MODE_K0PI2PI0; return;
200  }
201  if(n_e==0 && n_mu==0 && n_pi==3 && n_pi0==0 && n_K==0 && (n_K0L+n_K0S)==1 && n_nu==1){
202  MODE_ID=MODE_K03PI; return;
203  }
204  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==1 && n_K==0 && (n_K0L+n_K0S)==2 && n_nu==1){
205  MODE_ID=MODE_2K0PIPI0; return;
206  }
207  if(n_e==0 && n_mu==0 && n_pi==1 && n_pi0==1 && n_K==2 && n_K0L==0 && n_K0S==0 && n_nu==1){
208  MODE_ID=MODE_KPIKPI0; return;
209  }
210  if(n_pi==1 && n_pi0==1 && n_nu==1 && n_eta==1){ // eta modes
211  MODE_ID=MODE_ETAPIPI0; return;
212  }
213 
214  std::cout << "Tau Mode not found: n_e " << n_e << " n_mu " << n_mu << " n_pi " << n_pi << " n_pi0 " << n_pi0 << " n_K " << n_K << " n_K0L " << n_K0L << " n_K0S " << n_K0S << " n_nu " << n_nu << " n_gamma " << n_gamma << std::endl;
215  MODE_ID=MODE_UNKNOWN;
216 }
217 
218 unsigned int TauDecay::nProng(unsigned int &TauBitMask){
219  if(OneProng&TauBitMask) return 1;
220  if(ThreeProng&TauBitMask) return 3;
221  if(FiveProng&TauBitMask) return 5;
222  return 7;
223 }
224 unsigned int TauDecay::nPi0(unsigned int &TauBitMask){
225  if(OnePi0&TauBitMask) return 1;
226  if(TwoPi0&TauBitMask) return 2;
227  if(ThreePi0&TauBitMask) return 3;
228  return 0;
229 }
230 
231 bool TauDecay::hasResonance(unsigned int &TauBitMask, int pdgid){
232  int p=abs(pdgid);
233  if(p==PdtPdgMini::a_1_plus && Res_a1_pm&TauBitMask) return true;
234  if(p==PdtPdgMini::a_10 && Res_a1_0&TauBitMask) return true;
235  if(p==PdtPdgMini::rho_plus && Res_rho_pm&TauBitMask) return true;
236  if(p==PdtPdgMini::rho0 && Res_rho_0&TauBitMask) return true;
237  if(p==PdtPdgMini::eta && Res_eta&TauBitMask) return true;
238  if(p==PdtPdgMini::omega && Res_omega&TauBitMask) return true;
239  if(p==PdtPdgMini::K_star0 && Res_Kstar_pm&TauBitMask) return true;
240  if(p==PdtPdgMini::K_star_plus && Res_Kstar_0&TauBitMask) return true;
241  if(p==PdtPdgMini::K_S0 && KS0_to_pipi&TauBitMask) return true;
242  return false;
243 }
244 
245 
246 void TauDecay::ClassifyDecayResonance(unsigned int &TauBitMask){
247  // Add Resonance info to TauBitMask
248  if(n_a1>0) TauBitMask+=Res_a1_pm;
249  if(n_a10>0) TauBitMask+=Res_a1_0;
250  if(n_rho>0) TauBitMask+=Res_rho_pm;
251  if(n_rho0>0) TauBitMask+=Res_rho_0;
252  if(n_eta>0) TauBitMask+=Res_eta;
253  if(n_omega>0) TauBitMask+=Res_omega;
254  if(n_Kstar>0) TauBitMask+=Res_Kstar_pm;
255  if(n_Kstar0>0) TauBitMask+=Res_Kstar_0;
256 }
257 
258 std::string TauDecay::DecayMode(unsigned int &MODE_ID){
259  if(MODE_ID==MODE_ELECTRON) return "#tau^{#pm} #rightarrow e^{#pm}#nu#nu";
260  else if(MODE_ID==MODE_MUON) return "#tau^{#pm} #rightarrow #mu^{#pm}#nu#nu";
261  else if(MODE_ID==MODE_PION) return "#tau^{#pm} #rightarrow #pi^{#pm}#nu";
262  else if(MODE_ID==MODE_PIPI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{0}#nu";
263  else if(MODE_ID==MODE_3PI) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{#pm}#pi^{#mp}#nu";
264  else if(MODE_ID==MODE_PI2PI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{0}#pi^{0}#nu";
265  else if(MODE_ID==MODE_KAON) return "#tau^{#pm} #rightarrow K^{#pm}#nu";
266  else if(MODE_ID==MODE_KPI0) return "#tau^{#pm} #rightarrow K^{#pm}#pi^{0}#nu";
267  else if(MODE_ID==MODE_K0PI) return "#tau^{#pm} #rightarrow K^{0}#pi^{#pm}#nu";
268  else if(MODE_ID==MODE_3PIPI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{#pm}#pi^{#mp}#pi^{0}#nu";
269  else if(MODE_ID==MODE_PI3PI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{0}#pi^{0}#pi^{0}#nu";
270  else if(MODE_ID==MODE_3PI2PI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{#pm}#pi^{#mp}#pi^{0}#pi^{0}#nu";
271  else if(MODE_ID==MODE_5PI) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{#pm}#pi^{#pm}#pi^{#mp}#pi^{#mp}#nu";
272  else if(MODE_ID==MODE_5PIPI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{#pm}#pi^{#pm}#pi^{#mp}#pi^{#mp}#pi^{0}#nu";
273  else if(MODE_ID==MODE_3PI3PI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{#pm}#pi^{#mp}#pi^{0}#pi^{0}#pi^{0}#nu";
274  else if(MODE_ID==MODE_KPIK) return "#tau^{#pm} #rightarrow K^{#pm}#pi^{#pm}K^{#mp}#nu";
275  else if(MODE_ID==MODE_K0BK0PI) return "#tau^{#pm} #rightarrow #bar{K}^{0}K^{0}#pi^{#pm}#nu";
276  else if(MODE_ID==MODE_KK0BPI0) return "#tau^{#pm} #rightarrow #bar{K}^{0}K^{0}#pi^{#pm}#pi^{0}#nu";
277  else if(MODE_ID==MODE_K2PI0) return "#tau^{#pm} #rightarrow K^{#pm}#pi^{0}#pi^{0}#nu";
278  else if(MODE_ID==MODE_KPIPI) return "#tau^{#pm} #rightarrow K^{#pm}#pi^{#pm}#pi^{#mp}#nu";
279  else if(MODE_ID==MODE_PIK0PI0) return "#tau^{#pm} #rightarrow K^{0}#pi^{#pm}#pi^{0}#nu";
280  else if(MODE_ID==MODE_ETAPIPI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#eta#pi^{0}#nu";
281  else if(MODE_ID==MODE_PIPI0GAM) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{0}#nu#gamma (obsolete)";
282  else if(MODE_ID==MODE_KK0B) return "#tau^{#pm} #rightarrow K^{#pm}#bar{K}^{0}#nu";
283  else if(MODE_ID==MODE_PI4PI0) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{0}#pi^{0}#pi^{0}#pi^{0}#nu";
284  else if(MODE_ID==MODE_3PIETA) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{#pm}#pi^{#mp}#eta#nu";
285  else if(MODE_ID==MODE_PI2PI0ETA) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{0}#pi^{0}#eta#nu";
286  else if(MODE_ID==MODE_PI2PI0OMEGA) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{0}#pi^{0}#omega#nu";
287  else if(MODE_ID==MODE_3PIOMEGA) return "#tau^{#pm} #rightarrow #pi^{#pm}#pi^{#pm}#pi^{#mp}#omega#nu";
288  else if(MODE_ID==MODE_KOMEGA) return "#tau^{#pm} #rightarrow K^{#pm}#omega#nu";
289  else if(MODE_ID==MODE_K3PI0) return "#tau^{#pm} #rightarrow K#pi^{0}#pi^{0}#pi^{0}#nu";
290  else if(MODE_ID==MODE_K2PIPI0) return "#tau^{#pm} #rightarrow K^{#pm}#pi^{#pm}#pi^{#mp}#pi^{0}#nu";
291  else if(MODE_ID==MODE_KETA) return "#tau^{#pm} #rightarrow K^{#pm}#eta#nu";
292  else if(MODE_ID==MODE_K0PI2PI0) return "#tau^{#pm} #rightarrow K^{0}#pi^{#pm}#pi^{0}#pi^{0}#nu";
293  else if(MODE_ID==MODE_K03PI) return "#tau^{#pm} #rightarrow K^{0}#pi^{#pm}#pi^{#pm}#pi^{#mp}#nu";
294  else if(MODE_ID==MODE_2K0PIPI0) return "#tau^{#pm} #rightarrow K^{0}#bar{K}^{0}#pi^{#pm}#pi^{-}#nu";
295  else if(MODE_ID==MODE_KPIKPI0) return "#tau^{#pm} #rightarrow K^{#pm}#pi^{#pm}K^{#mp}#pi^{0}#nu";
296  return "UnKnown";
297 }
unsigned int n_pi
Definition: TauDecay.h:167
void Reset()
Definition: TauDecay.cc:15
bool hasResonance(unsigned int &TauBitMask, int pdgid)
Definition: TauDecay.cc:231
unsigned int unknown
Definition: TauDecay.h:168
unsigned int n_nu
Definition: TauDecay.h:167
~TauDecay()
Definition: TauDecay.cc:11
unsigned int n_omega
Definition: TauDecay.h:168
unsigned int n_K
Definition: TauDecay.h:167
bool isTauFinalStateParticle(int pdgid)
Definition: TauDecay.cc:36
unsigned int n_mu
Definition: TauDecay.h:167
unsigned int n_eta
Definition: TauDecay.h:168
unsigned int n_rho0
Definition: TauDecay.h:168
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
unsigned int n_K0S
Definition: TauDecay.h:167
unsigned int n_a10
Definition: TauDecay.h:168
TauDecay()
Definition: TauDecay.cc:7
bool isTauResonanceCounter(int pdgid)
Definition: TauDecay.cc:75
unsigned int n_a1
Definition: TauDecay.h:168
void ClassifyDecayMode(unsigned int &MODE_ID, unsigned int &TauBitMask)
Definition: TauDecay.cc:89
static std::string DecayMode(unsigned int &MODE_ID)
Definition: TauDecay.cc:258
unsigned int n_e
Definition: TauDecay.h:167
return(e1-e2)*(e1-e2)+dp *dp
unsigned int n_Kstar0
Definition: TauDecay.h:168
void ClassifyDecayResonance(unsigned int &TauBitMask)
Definition: TauDecay.cc:246
unsigned int n_rho
Definition: TauDecay.h:168
tuple cout
Definition: gather_cfg.py:145
unsigned int n_Kstar
Definition: TauDecay.h:168
unsigned int n_K0L
Definition: TauDecay.h:167
unsigned int n_pi0
Definition: TauDecay.h:167
unsigned int nPi0(unsigned int &TauBitMask)
Definition: TauDecay.cc:224
unsigned int n_gamma
Definition: TauDecay.h:167
bool isTauParticleCounter(int pdgid)
Definition: TauDecay.cc:55
unsigned int nProng(unsigned int &TauBitMask)
Definition: TauDecay.cc:218