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