]>
Commit | Line | Data |
---|---|---|
1 | #ifndef AliAODCluster_H | |
2 | #define AliAODCluster_H | |
3 | /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. * | |
4 | * See cxx source for full Copyright notice */ | |
5 | ||
6 | /* $Id$ */ | |
7 | ||
8 | //------------------------------------------------------------------------- | |
9 | // AOD cluster base class | |
10 | // Author: Markus Oldenburg, CERN | |
11 | //------------------------------------------------------------------------- | |
12 | ||
13 | #include <TObject.h> | |
14 | ||
15 | class AliAODCluster : public TObject { | |
16 | ||
17 | public: | |
18 | ||
19 | enum AODClu_t {kUndef = -1, | |
20 | kPHOSNeutral, | |
21 | kPHOSCharged, | |
22 | kEMCALClusterv1, | |
23 | kPMDNeutral, | |
24 | kPMDCharged}; | |
25 | ||
26 | enum AODCluPID_t { | |
27 | kElectron = 0, | |
28 | kMuon = 1, | |
29 | kPion = 2, | |
30 | kKaon = 3, | |
31 | kProton = 4, | |
32 | kPhoton = 5, | |
33 | kPi0 = 6, | |
34 | kNeutron = 7, | |
35 | kKaon0 = 8, | |
36 | kEleCon = 9, | |
37 | kUnknown = 10, | |
38 | kCharged = 11, //For PMD? | |
39 | kNeutral =12 //For PMD? | |
40 | }; | |
41 | ||
42 | AliAODCluster(); | |
43 | AliAODCluster(Int_t id, | |
44 | UInt_t nLabel, | |
45 | Int_t *label, | |
46 | Double_t energy, | |
47 | Double_t x[3], | |
48 | Double_t pid[13], | |
49 | Char_t ttype=kUndef, | |
50 | UInt_t selectInfo=0); | |
51 | ||
52 | AliAODCluster(Int_t id, | |
53 | UInt_t nLabel, | |
54 | Int_t *label, | |
55 | Float_t energy, | |
56 | Float_t x[3], | |
57 | Float_t pid[13], | |
58 | Char_t ttype=kUndef, | |
59 | UInt_t selectInfo=0); | |
60 | ||
61 | virtual ~AliAODCluster(); | |
62 | AliAODCluster(const AliAODCluster& clus); | |
63 | AliAODCluster& operator=(const AliAODCluster& clus); | |
64 | ||
65 | Double_t Chi2() const { return fChi2; } | |
66 | ||
67 | virtual Double_t E() const { return fEnergy; } | |
68 | ||
69 | // PID | |
70 | virtual const Double_t *PID() const { return fPID; } | |
71 | AODCluPID_t GetMostProbablePID() const; | |
72 | ||
73 | template <class T> void GetPID(T *pid) const { | |
74 | for(Int_t i=0; i<13; ++i) pid[i]=fPID[i];} | |
75 | ||
76 | template <class T> void SetPID(const T *pid) { | |
77 | if(pid) for(Int_t i=0; i<13; ++i) fPID[i]=pid[i]; | |
78 | else {for(Int_t i=0; i<13; fPID[i++]=0) ;} fPID[AliAODCluster::kUnknown]=1.;} | |
79 | ||
80 | template <class T> void SetPIDFromESD(const T *pid) { | |
81 | if(pid) {for(Int_t i=0; i<11; ++i) fPID[i]=pid[i]; fPID[11]=0; fPID[12]=0;} | |
82 | else {for(Int_t i=0; i<13; fPID[i++]=0) ;} fPID[AliAODCluster::kUnknown]=1.;} | |
83 | ||
84 | Int_t GetID() const { return fID; } | |
85 | Int_t GetLabel(UInt_t i) const; | |
86 | Int_t * GetLabels() const {return fLabel ; } | |
87 | UInt_t GetNLabel() const { return (UInt_t)fNLabel; } | |
88 | Bool_t TestFilterBit(UInt_t filterBit) const { return (Bool_t) ((filterBit & fFilterMap) != 0); } | |
89 | Char_t GetType() const { return fType; } | |
90 | ||
91 | template <class T> Bool_t GetPosition(T *x) const { | |
92 | x[0]=fPosition[0]; x[1]=fPosition[1]; x[2]=fPosition[2]; | |
93 | return kTRUE;} | |
94 | ||
95 | Bool_t IsEMCALCluster() {if(fType == kEMCALClusterv1) return kTRUE; | |
96 | else return kFALSE;} | |
97 | Bool_t IsPHOSCluster() {if(fType == kPHOSCharged || fType == kPHOSNeutral) return kTRUE; | |
98 | else return kFALSE;} | |
99 | ||
100 | ||
101 | void Print(const Option_t *opt = "") const; | |
102 | ||
103 | // setters | |
104 | void SetE(Double32_t energy) {fEnergy = energy ; } | |
105 | void SetPosition(Int_t ipos, Double32_t pos) {fPosition[ipos] = pos ;} | |
106 | void SetID(Int_t id) { fID = id; } | |
107 | void SetType(AODClu_t ttype) { fType=ttype; } | |
108 | void SetLabel(Int_t *label, UInt_t size); | |
109 | void RemoveLabel(); | |
110 | ||
111 | template <class T> void SetPosition(const T *x); | |
112 | ||
113 | void SetChi2(Double_t chi2) { fChi2 = chi2; } | |
114 | ||
115 | private : | |
116 | ||
117 | // Energy & position | |
118 | Double32_t fEnergy; // energy | |
119 | Double32_t fPosition[3]; // position of the cluster | |
120 | ||
121 | Double32_t fChi2; // chi2 (probably not necessary for PMD) | |
122 | Double32_t fPID[13]; // [0.,1.,8] pointer to PID object | |
123 | ||
124 | Int_t fID; // unique cluster ID, points back to the ESD cluster | |
125 | Int_t fNLabel; // number of original track for this cluster | |
126 | Int_t *fLabel; // [fNLabel] particle label, points back to MC tracks | |
127 | UInt_t fFilterMap; // filter information, one bit per set of cuts | |
128 | ||
129 | Char_t fType; // cluster type | |
130 | ||
131 | ClassDef(AliAODCluster,5); | |
132 | }; | |
133 | ||
134 | #endif |