Removing extra semicolons (FedoraCore3, gcc 3.4.2)
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALFastRecParticle.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 /* $Id$ */
17
18 //_________________________________________________________________________
19 //  A  Particle modified by EMCAL response and produced by AliEMCALvFast
20 //*--
21 //  To become a general class of AliRoot ?    
22 //*--               
23 //*-- Author: Yves Schutz (SUBATECH)
24 //*--
25 /////////////////////////////////////////////////////////////////////////////
26
27 // --- ROOT system ---
28
29 // --- Standard library ---
30
31 // --- AliRoot header files ---
32
33 #include "AliEMCALFastRecParticle.h"
34 #include "TPad.h"
35 #include "TPaveText.h"
36
37 ClassImp(AliEMCALFastRecParticle) 
38
39 //____________________________________________________________________________
40 AliEMCALFastRecParticle::AliEMCALFastRecParticle() : TParticle()
41 {
42   // ctor
43   fType = 0 ; 
44 }
45
46 //____________________________________________________________________________
47 AliEMCALFastRecParticle::AliEMCALFastRecParticle(const AliEMCALFastRecParticle & rp)
48   : TParticle(rp)
49 {
50   // copy ctor
51
52   fType        = rp.fType ;
53   fPdgCode     = rp.fPdgCode;
54   fStatusCode  = rp.fStatusCode;
55   fMother[0]   = rp.fMother[0];
56   fMother[1]   = rp.fMother[1];
57   fDaughter[0] = rp.fDaughter[0];
58   fDaughter[1] = rp.fDaughter[1];
59   fWeight      = rp.fWeight;
60   fCalcMass    = rp.fCalcMass;
61   fPx          = rp.fPx;
62   fPy          = rp.fPy;
63   fPz          = rp.fPz;
64   fE           = rp.fE;
65   fVx          = rp.fVx;
66   fVy          = rp.fVy;
67   fVz          = rp.fVz;
68   fVt          = rp.fVt;
69   fPolarTheta  = rp.fPolarTheta;
70   fPolarPhi    = rp.fPolarPhi;
71   fParticlePDG = rp.fParticlePDG; 
72 }
73
74 //____________________________________________________________________________
75  AliEMCALFastRecParticle::AliEMCALFastRecParticle(const TParticle & pp)
76 {
77   // ctor from a TParticle (crummy?!)
78  
79   TParticle & pnoconst = (TParticle &)(pp) ;
80   AliEMCALFastRecParticle & p = (AliEMCALFastRecParticle &)(pnoconst) ;
81   fType        = 0  ;
82   fPdgCode     = p.fPdgCode;
83   fStatusCode  = p.fStatusCode;
84   fMother[0]   = p.fMother[0];
85   fMother[1]   = p.fMother[1];
86   fDaughter[0] = p.fDaughter[0];
87   fDaughter[1] = p.fDaughter[1];
88   fWeight      = p.fWeight;
89   fCalcMass    = p.fCalcMass;
90   fPx          = p.fPx;
91   fPy          = p.fPy;
92   fPz          = p.fPz;
93   fE           = p.fE;
94   fVx          = p.fVx;
95   fVy          = p.fVy;
96   fVz          = p.fVz;
97   fVt          = p.fVt;
98   fPolarTheta  = p.fPolarTheta;
99   fPolarPhi    = p.fPolarPhi;
100   fParticlePDG = p.fParticlePDG; 
101
102 }
103
104 //____________________________________________________________________________
105 Int_t AliEMCALFastRecParticle::DistancetoPrimitive(Int_t px, Int_t py)
106 {
107   //  Compute distance from point px,py to a AliEMCALFastRecParticle considered as a Tmarker
108   //  Compute the closest distance of approach from point px,py to this marker.
109   //  The distance is computed in pixels units.
110
111   Double_t kRADDEG = 180. / TMath::Pi() ; 
112   Coord_t x = Phi() * kRADDEG     ;
113   Coord_t y = Theta() * kRADDEG     ;
114   const Int_t kMaxDiff = 10;
115   Int_t pxm  = gPad->XtoAbsPixel(x);
116   Int_t pym  = gPad->YtoAbsPixel(y);
117   Int_t dist = (px-pxm)*(px-pxm) + (py-pym)*(py-pym);
118   
119   if (dist > kMaxDiff) return 9999;
120   return dist;
121 }
122
123 //___________________________________________________________________________
124  void AliEMCALFastRecParticle::Draw(Option_t *option)
125  {
126    // Draw this AliEMCALFastRecParticle with its current attributes
127     
128    AppendPad(option);
129  }
130
131 //______________________________________________________________________________
132 void AliEMCALFastRecParticle::ExecuteEvent(Int_t event, Int_t /*px*/, Int_t /*py*/)
133 {
134   //  Execute action corresponding to one event
135   //  This member function is called when a AliEMCALFastRecParticle is clicked with the locator
136   
137   if (!gPad->IsEditable()) 
138     return ;
139   
140   static TPaveText * clustertext = 0 ; 
141   
142   switch (event) {
143     
144   case kButton1Down: {
145     Double_t kRADDEG = 180. / TMath::Pi() ; 
146     Coord_t x = Phi() * kRADDEG     ;
147     Coord_t y = Theta() * kRADDEG     ;
148     clustertext = new TPaveText(x-1, y+1, x+5, y+3, "") ;
149     Text_t  line1[40] ;
150     Text_t  line2[40] ;
151     sprintf( line1, "PID: %s ", (const char*)Name() ) ;
152     sprintf( line2, "ENERGY: %f ", Energy() ) ;
153     clustertext ->AddText(line1) ;
154     clustertext ->AddText(line2) ;
155     clustertext ->Draw("");   
156     gPad->Update() ; 
157     break ;
158   }
159   
160   case kButton1Up: {
161     delete clustertext ; 
162     clustertext = 0 ; 
163     gPad->Update() ; 
164     break ;
165   }
166   }
167   
168 }
169
170 //____________________________________________________________________________
171 Bool_t AliEMCALFastRecParticle::IsPhoton(TString purity) const
172 {
173   //FIXME : just a blind copy from PHOS
174   // Rec.Particle is a photon if it has a photon-like shape, fast and neutral
175   // photon-like shape is defined with a purity "low", "medium" or "high"
176
177   purity.ToLower();
178   Bool_t photonLike = kFALSE;
179   if      (purity == "low"   ) photonLike = TestPIDBit(6);
180   else if (purity == "medium") photonLike = TestPIDBit(7);
181   else if (purity == "high"  ) photonLike = TestPIDBit(8);
182   if (photonLike                                   && //  photon by PCA
183       (TestPIDBit(5)||TestPIDBit(4)||TestPIDBit(3))&& //  fast by TOF
184       (TestPIDBit(2)||TestPIDBit(1)||TestPIDBit(0))&& //  neutral by CPV
185       !TestPIDBit(14))                              //  no charged track
186     return kTRUE ;
187   else
188     return kFALSE;
189 }
190
191 //____________________________________________________________________________
192 Bool_t AliEMCALFastRecParticle::IsPi0(TString purity) const
193 {
194   //FIXME : just a blind copy from PHOS
195   // Rec.Particle is a pi0 if it has a pi0-like shape, fast and neutral
196   // pi0-like shape is defined with a purity "low", "medium" or "high"
197
198   purity.ToLower();
199   Bool_t pi0Like = kFALSE;
200   if      (purity == "low"   ) pi0Like = TestPIDBit(9);
201   else if (purity == "medium") pi0Like = TestPIDBit(10);
202   else if (purity == "high"  ) pi0Like = TestPIDBit(11);
203   else Error("IsPi0","Wrong purity type: %s",purity.Data());
204   if (pi0Like                                      && //  pi0 by PCA
205       (TestPIDBit(5)||TestPIDBit(4)||TestPIDBit(3))&& //  fast by TOF
206       (TestPIDBit(2)||TestPIDBit(1)||TestPIDBit(0))&& //  neutral by CPV
207       !TestPIDBit(14))                              //  no charged track
208     return kTRUE ;
209   else
210     return kFALSE;
211 }
212
213 //____________________________________________________________________________
214 Bool_t AliEMCALFastRecParticle::IsElectron(TString purity) const
215 {
216   //FIXME : just a blind copy from PHOS
217   // Rec.Particle is an electron if it has a photon-like shape, fast and charged
218   // photon-like shape is defined with a purity "low", "medium" or "high"
219
220   purity.ToLower();
221   Bool_t photonLike = kFALSE;
222   if      (purity == "low"   ) photonLike = TestPIDBit(6);
223   else if (purity == "medium") photonLike = TestPIDBit(7);
224   else if (purity == "high"  ) photonLike = TestPIDBit(8);
225   else Error("IsElectron","Wrong purity type: %s",purity.Data());
226   
227   if (photonLike                                   && //  photon by PCA
228       (TestPIDBit(5)|| TestPIDBit(4)|| TestPIDBit(3))&& //  fast by TOF
229       (!TestPIDBit(2)||!TestPIDBit(1)||!TestPIDBit(0))&& //  charged by CPV
230       TestPIDBit(14))                                  //  no charged track
231     return kTRUE ;
232   else
233     return kFALSE;
234 }
235
236 //____________________________________________________________________________
237 Bool_t AliEMCALFastRecParticle::IsEleCon(TString purity) const
238 {
239   //FIXME : just a blind copy from PHOS
240   // Rec.Particle is an electron if it has a photon-like shape, fast and charged
241   // photon-like shape is defined with a purity "low", "medium" or "high"
242
243   purity.ToLower();
244   Bool_t photonLike = kFALSE;
245   if      (purity == "low"   ) photonLike = TestPIDBit(6);
246   else if (purity == "medium") photonLike = TestPIDBit(7);
247   else if (purity == "high"  ) photonLike = TestPIDBit(8);
248   else Error("IsElectron","Wrong purity type: %s",purity.Data());
249   
250   if (photonLike                                   && //  photon by PCA
251       (TestPIDBit(5)|| TestPIDBit(4)|| TestPIDBit(3))&& //  fast by TOF
252       (!TestPIDBit(2)||!TestPIDBit(1)||!TestPIDBit(0))&& //  charged by CPV
253       !TestPIDBit(14))                                  //  no charged track
254     return kTRUE ;
255   else
256     return kFALSE;
257 }
258
259 //____________________________________________________________________________
260 Bool_t AliEMCALFastRecParticle::IsHardPhoton() const
261 {
262   //FIXME : just a blind copy from PHOS
263   // Rec.Particle is a hard photon (E > 30 GeV) if its second moment M2x
264   // corresponds to photons
265   if (TestPIDBit(12) && !TestPIDBit(14))
266     return kTRUE;
267   else
268     return kFALSE;
269 }
270
271 //____________________________________________________________________________
272 Bool_t AliEMCALFastRecParticle::IsHardPi0() const
273 {
274   //FIXME : just a blind copy from EMCAL
275   // Rec.Particle is a hard pi0 (E > 30 GeV) if its second moment M2x
276   // corresponds to pi0
277   if (TestPIDBit(13)&& !TestPIDBit(14))
278     return kTRUE;
279   else
280     return kFALSE;
281 }
282
283 //____________________________________________________________________________
284 Bool_t AliEMCALFastRecParticle::IsHadron() const
285 {
286   //FIXME : just a blind copy from EMCAL
287   // Rec.Particle is an hadron if it does not look like
288   // a low-purity photon nor low-purity pi0
289
290   if ( !TestPIDBit(6) && !TestPIDBit(9) )             // not photon nor pi0
291     return kTRUE ;
292   else
293     return kFALSE;
294 }
295
296 //____________________________________________________________________________
297 Bool_t AliEMCALFastRecParticle::IsChargedHadron() const
298 {
299   //FIXME : just a blind copy from EMCAL
300   // Rec.Particle is a charged hadron if it does not look like
301   // a low-purity photon nor low-purity pi0 and is low-purity charged
302
303   if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
304        !TestPIDBit(2))                                // charged by CPV
305     return kTRUE ;
306   else
307     return kFALSE;
308 }
309
310 //____________________________________________________________________________
311 Bool_t AliEMCALFastRecParticle::IsNeutralHadron() const
312 {
313   //FIXME : just a blind copy from EMCAL
314   // Rec.Particle is a neutral hadron if it does not look like
315   // a low-purity photon nor low-purity pi0 and is high-purity neutral
316
317   if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
318         TestPIDBit(2))                                // neutral by CPV
319     return kTRUE ;
320   else
321     return kFALSE;
322 }
323
324 //____________________________________________________________________________
325 Bool_t AliEMCALFastRecParticle::IsFastChargedHadron() const
326 {
327   //FIXME : just a blind copy from EMCAL
328   // Rec.Particle is a fast charged hadron if it does not look like
329   // a low-purity photon nor low-purity pi0, is low-purity charged
330   // and is high-purity fast
331
332   if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
333        !TestPIDBit(2) &&                              // charged by CPV
334         TestPIDBit(5))                                // fast by TOF
335     return kTRUE ;
336   else
337     return kFALSE;
338 }
339
340 //____________________________________________________________________________
341 Bool_t AliEMCALFastRecParticle::IsSlowChargedHadron() const
342 {
343   //FIXME : just a blind copy from EMCAL
344   // Rec.Particle is a slow neutral hadron if it does not look like
345   // a low-purity photon nor low-purity pi0, is high-purity neutral
346   // and is not high-purity fast
347
348   if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
349        !TestPIDBit(2) &&                              // charged by CPV
350        !TestPIDBit(5))                                // slow by TOF
351     return kTRUE ;
352   else
353     return kFALSE;
354
355 }
356
357 //____________________________________________________________________________
358 Bool_t AliEMCALFastRecParticle::IsFastNeutralHadron() const
359 {
360   //FIXME : just a blind copy from EMCAL
361   // Rec.Particle is a fast neutral hadron if it does not look like
362   // a low-purity photon nor low-purity pi0, is high-purity neutral
363   // and is high-purity fast
364
365   if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
366         TestPIDBit(2) &&                              // neutral by CPV
367         TestPIDBit(5))                                // fast by TOF
368     return kTRUE ;
369   else
370     return kFALSE;
371 }
372
373 //____________________________________________________________________________
374 Bool_t AliEMCALFastRecParticle::IsSlowNeutralHadron() const
375 {
376   //FIXME : just a blind copy from EMCAL
377   // Rec.Particle is a slow neutral hadron if it does not look like
378   // a low-purity photon nor low-purity pi0, is high-purity neutral
379   // and is not high-purity fast
380
381   if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
382         TestPIDBit(2) &&                              // neutral by CPV
383        !TestPIDBit(5))                                // slow by TOF
384     return kTRUE ;
385   else
386     return kFALSE;
387 }
388
389
390 //____________________________________________________________________________
391 TString AliEMCALFastRecParticle::Name()const
392 {
393    // Returns the name of the particle type (only valid if PIDv1 is employed)
394
395   TString  name ; 
396
397   name = "Undefined particle" ;
398   
399   if      (IsPhoton("low"))
400     name = "Photon low purity, ";
401   else if (IsPhoton("medium"))
402     name = "Photon medium purity, ";
403   else if (IsPhoton("high"))
404     name = "Photon high purity, ";
405
406   if      (IsPi0("low"))
407     name += "Pi0 low purity, ";
408   else if (IsPi0("medium"))
409     name += "Pi0 medium purity, ";
410   else if (IsPi0("high"))
411     name += "Pi0 high purity, ";
412
413   if      (IsElectron("low"))
414     name += "Electron low purity, ";
415   else if (IsElectron("medium"))
416     name += "Electron medium purity, ";
417   else if (IsElectron("high"))
418     name += "Electron high purity, ";
419
420   if     (IsHadron()) {
421     name = "hadron";
422     if      (IsChargedHadron()) {
423       name.Prepend("charged, ");
424       if      (IsFastChargedHadron())
425         name.Prepend("fast, ");
426       else if (IsSlowChargedHadron())
427         name.Prepend("slow, ");
428     }
429     else if (IsNeutralHadron()) {
430       name.Prepend("neutral, ");
431       if      (IsFastNeutralHadron())
432         name.Prepend("fast, ");
433       else if (IsSlowNeutralHadron())
434         name.Prepend("slow, ");
435     }
436   }
437
438   return name ; 
439 }
440
441 //______________________________________________________________________________
442 void AliEMCALFastRecParticle::SetType(Int_t type) { 
443   // sets the particle type 
444   // bit-mask of the particle type means the following:
445   // bits 0,1,2   - neutral particle with low, medium and high purity
446   // bits 3.4,5   - fast particle with low, medium and high purity
447   // bits 6.7,8   - photon shower with low, medium and high purity
448   // bits 9,10,11 - hard-pi0 shower with low, medium and high purity
449
450   fType = type ; 
451   
452   if((type == 127) || (fType == 511) || (fType == 255) ||(fType == 383)||(fType == 447)){
453     fPdgCode = 22 ; 
454     return ;
455   }
456   
457   if ((fType == 63)|| ((fType < 8)&&(fType > 0)) ){
458     fPdgCode = 2112 ; 
459     return ;
460   }
461   if ( ((fType == 504) || (fType == 505) ||(fType == 248)||(fType == 249)||(fType == 120)||(fType == 121)) ){
462     fPdgCode = 11 ; 
463     return ;
464   }
465   if ((fType == 448) || (fType == 449) ||(fType == 192)||(fType == 193)||(fType == 64)||(fType == 64)){
466     fPdgCode = 13 ; 
467     return ;
468   }
469   if((fType == 56)||(fType == 57)){
470     fPdgCode = 211 ; 
471     return ;
472   }
473   if (fType == 0){
474     fPdgCode = 2212 ; 
475     return ;
476   }
477
478 }           
479
480 //______________________________________________________________________________
481 void AliEMCALFastRecParticle::Paint(Option_t *)
482 {
483   // Paint this ALiRecParticle in theta,phi coordinate as a TMarker  with its current attributes
484
485   Double_t kRADDEG = 180. / TMath::Pi() ; 
486   Coord_t x = Phi() * kRADDEG     ;
487   Coord_t y = Theta() * kRADDEG     ;
488   Color_t markercolor = 1 ;
489   Size_t  markersize  = 1. ;
490   Style_t markerstyle = 5 ;
491   
492   if (!gPad->IsBatch()) {
493     gVirtualX->SetMarkerColor(markercolor) ;
494     gVirtualX->SetMarkerSize (markersize)  ;
495     gVirtualX->SetMarkerStyle(markerstyle) ;
496   }
497   gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
498   gPad->PaintPolyMarker(1,&x,&y,"") ;
499 }
500
501 //____________________________________________________________________________
502 void AliEMCALFastRecParticle::Print(Option_t * /*opt*/)const
503 {
504   // Print the type, energy and momentum of the reconstructed particle
505   
506   printf("Print Summary:") ; 
507   printf("AliEMCALFastRecParticle > type is  %s\n", Name().Data()) ; 
508   printf("                      Energy = %f\n", fE) ; 
509   printf("                         Px     = %f\n", fPx) ; 
510   printf("                         Py     = %f\n", fPy) ;
511   printf("                         Pz     = %f\n", fPz) ; 
512 }