]>
Commit | Line | Data |
---|---|---|
f1d945a1 | 1 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * |
2 | * See cxx source for full Copyright notice */ | |
3 | /* $Id$ */ | |
4 | ||
e1911c19 | 5 | #ifndef ALIFLOWTRACKSIMPLE_H |
6 | #define ALIFLOWTRACKSIMPLE_H | |
f1d945a1 | 7 | |
bc231a12 | 8 | #include "TObject.h" |
f1d945a1 | 9 | #include "TBits.h" |
929098e4 | 10 | class TParticle; |
f1d945a1 | 11 | |
12 | // AliFlowTrackSimple: | |
13 | // A simple track class to the the AliFlowEventSimple for flow analysis | |
14 | // author: N. van der Kolk (kolk@nikhef.nl) | |
929098e4 | 15 | // mods: Mikolaj Krzewicki (mikolaj.krzewicki@cern.ch) |
f1d945a1 | 16 | |
bc231a12 | 17 | class AliFlowTrackSimple: public TObject { |
f1d945a1 | 18 | |
244c607a | 19 | public: |
8eca5d19 | 20 | enum tagType { kInvalid=-1, |
21 | kRP=0, | |
22 | kPOI=1, | |
23 | kPOI1=2, | |
24 | kPOI2=3, | |
25 | }; | |
26 | ||
f1d945a1 | 27 | AliFlowTrackSimple(); |
701f71c1 | 28 | AliFlowTrackSimple(TParticle* p); |
bc6b015e | 29 | AliFlowTrackSimple(const AliFlowTrackSimple& aTrack); |
59327657 | 30 | AliFlowTrackSimple& operator=(const AliFlowTrackSimple& aTrack); |
f1d945a1 | 31 | virtual ~AliFlowTrackSimple(); |
bc231a12 | 32 | virtual AliFlowTrackSimple* Clone(const char* option="") const; |
f1d945a1 | 33 | |
c076fda8 | 34 | Bool_t IsFolder() const {return kTRUE;}; |
35 | // void Browse(TBrowser *b); | |
bc231a12 | 36 | virtual void Print(Option_t* option = "") const; |
c076fda8 | 37 | |
7d27a354 | 38 | void Set(TParticle* p); |
39 | ||
f1d945a1 | 40 | Double_t Eta() const; |
41 | Double_t Pt() const; | |
44e060e0 | 42 | Double_t Phi() const; |
43 | Double_t Weight() const; | |
701f71c1 | 44 | Int_t Charge() const; |
60875c3c | 45 | Double_t Mass() const; |
b4dba88d | 46 | Int_t PID() const {return 0;} |
701f71c1 | 47 | |
34b15925 | 48 | Bool_t InRPSelection() const; |
8fa6a5fa MK |
49 | Bool_t InPOISelection(Int_t poiType=1) const; |
50 | Bool_t IsPOItype(Int_t poiType) const; | |
51 | void SetPOItype(Int_t poiType, Bool_t b=kTRUE); | |
34b15925 | 52 | Bool_t InSubevent(Int_t i) const; |
8fa6a5fa MK |
53 | void TagRP(Bool_t b=kTRUE) {SetForRPSelection(b);} |
54 | void TagPOI(Bool_t b=kTRUE) {SetForPOISelection(b);} | |
55 | void Tag(Int_t n, Bool_t b=kTRUE) {fFlowBits.SetBitNumber(n,b);} | |
56 | Bool_t CheckTag(Int_t n) {return fFlowBits.TestBitNumber(n);} | |
57 | void SetForSubevent(Int_t i); | |
58 | void ResetFlowTags() {fFlowBits.ResetAllBits();} | |
59 | void ResetSubEventTags() {fSubEventBits.ResetAllBits();} | |
34b15925 | 60 | Bool_t IsDead() const {return (fFlowBits.CountBits()==0);} |
b125a454 | 61 | |
f1d945a1 | 62 | void SetEta(Double_t eta); |
63 | void SetPt(Double_t pt); | |
64 | void SetPhi(Double_t phi); | |
44e060e0 | 65 | void SetWeight(Double_t weight); |
701f71c1 | 66 | void SetCharge(Int_t charge); |
60875c3c | 67 | void SetMass(Double_t mass); |
7382279b | 68 | void SetForRPSelection(Bool_t b=kTRUE); |
69 | void SetForPOISelection(Bool_t b=kTRUE); | |
8fa6a5fa | 70 | virtual void Clear(Option_t* o=""); |
7382279b | 71 | |
72 | void ResolutionPt(Double_t resolution); | |
73 | ||
bc231a12 | 74 | void AddV1( Double_t v1, |
75 | Double_t reactionPlaneAngle, | |
76 | Double_t precision, | |
77 | Int_t maxNumberOfIterations=100 ); | |
78 | void AddV2( Double_t v2, | |
79 | Double_t reactionPlaneAngle, | |
80 | Double_t precision, | |
81 | Int_t maxNumberOfIterations=100 ); | |
bff213ad | 82 | void AddV3( Double_t v3, |
83 | Double_t reactionPlaneAngle, | |
84 | Double_t precision, | |
85 | Int_t maxNumberOfIterations=100 ); | |
bc231a12 | 86 | void AddV4( Double_t v4, |
87 | Double_t reactionPlaneAngle, | |
88 | Double_t precision, | |
89 | Int_t maxNumberOfIterations=100 ); | |
1a80f9f6 | 90 | void AddV5( Double_t v5, |
91 | Double_t reactionPlaneAngle, | |
92 | Double_t precision, | |
93 | Int_t maxNumberOfIterations=100 ); | |
bc231a12 | 94 | void AddFlow( Double_t v1, |
95 | Double_t v2, | |
bff213ad | 96 | Double_t v3, |
bc231a12 | 97 | Double_t v4, |
98 | Double_t reactionPlaneAngle, | |
99 | Double_t precision, | |
100 | Int_t maxNumberOfIterations=100 ); | |
1a80f9f6 | 101 | void AddFlow( Double_t v1, |
102 | Double_t v2, | |
103 | Double_t v3, | |
104 | Double_t v4, | |
105 | Double_t v5, | |
106 | Double_t reactionPlaneAngle, | |
107 | Double_t precision, | |
108 | Int_t maxNumberOfIterations=100 ); | |
109 | void AddFlow( Double_t v1, | |
110 | Double_t v2, | |
111 | Double_t v3, | |
112 | Double_t v4, | |
113 | Double_t v5, | |
114 | Double_t rp1, | |
115 | Double_t rp2, | |
116 | Double_t rp3, | |
117 | Double_t rp4, | |
118 | Double_t rp5, | |
119 | Double_t precision, | |
120 | Int_t maxNumberOfIterations=100 ); | |
121 | ||
34b15925 | 122 | const TBits* GetFlowBits() const {return &fFlowBits;} |
44e060e0 | 123 | |
41dc4195 | 124 | void SetID(Int_t i) {fID=i;} |
cc0afcfc | 125 | Int_t GetID() const {return fID;} |
6e214c87 | 126 | |
8fa6a5fa MK |
127 | virtual Int_t GetNDaughters() const {return 0;} |
128 | virtual void AddDaughter(Int_t /*value*/) {} | |
129 | virtual Int_t GetIDDaughter(Int_t /*value*/) const {return 0;} | |
130 | virtual void SetDaughter(Int_t /*value*/, AliFlowTrackSimple* /*track*/) {} | |
131 | virtual AliFlowTrackSimple *GetDaughter(Int_t /*value*/) const {return NULL;} | |
132 | ||
f1d945a1 | 133 | private: |
60875c3c | 134 | AliFlowTrackSimple(Double_t phi, Double_t eta, Double_t pt, Double_t weight, Int_t charge, Double_t mass=-1); |
44e060e0 | 135 | Double_t fEta; // eta |
136 | Double_t fPt; // pt | |
137 | Double_t fPhi; // phi | |
138 | Double_t fTrackWeight; // weight | |
701f71c1 | 139 | Int_t fCharge; //charge |
60875c3c | 140 | Double_t fMass; // mass |
44e060e0 | 141 | TBits fFlowBits; // bits to set if track is selected |
142 | TBits fSubEventBits;// bits to set if track is selected for a subevent | |
41dc4195 | 143 | Int_t fID; // Unique track ID, point back to the ESD track |
f1d945a1 | 144 | |
60875c3c | 145 | ClassDef(AliFlowTrackSimple,2) // macro for rootcint |
f1d945a1 | 146 | |
147 | }; | |
148 | ||
701f71c1 | 149 | //Getters |
f1d945a1 | 150 | inline Double_t AliFlowTrackSimple::Eta() const { |
88e00a8a | 151 | return this->fEta; } |
f1d945a1 | 152 | inline Double_t AliFlowTrackSimple::Pt() const { |
88e00a8a | 153 | return this->fPt;} |
f1d945a1 | 154 | inline Double_t AliFlowTrackSimple::Phi() const { |
88e00a8a | 155 | return this->fPhi; } |
44e060e0 | 156 | inline Double_t AliFlowTrackSimple::Weight() const { |
157 | return this->fTrackWeight; } | |
701f71c1 | 158 | inline Int_t AliFlowTrackSimple::Charge() const { |
159 | return this->fCharge; } | |
60875c3c | 160 | inline Double_t AliFlowTrackSimple::Mass() const { |
161 | return this->fMass; } | |
b125a454 | 162 | //TBits |
1918addd | 163 | inline Bool_t AliFlowTrackSimple::InRPSelection() const { |
8fa6a5fa MK |
164 | return fFlowBits.TestBitNumber(kRP); } |
165 | inline Bool_t AliFlowTrackSimple::InPOISelection(Int_t poiType) const { | |
166 | return fFlowBits.TestBitNumber(poiType); } | |
167 | inline Bool_t AliFlowTrackSimple::IsPOItype(Int_t poiType) const { | |
168 | return fFlowBits.TestBitNumber(poiType); } | |
b125a454 | 169 | inline Bool_t AliFlowTrackSimple::InSubevent(Int_t i) const { |
170 | return this->fSubEventBits.TestBitNumber(i); } | |
f1d945a1 | 171 | |
701f71c1 | 172 | //Setters |
f1d945a1 | 173 | inline void AliFlowTrackSimple::SetEta(Double_t val) { |
174 | fEta = val; } | |
175 | inline void AliFlowTrackSimple::SetPt(Double_t val) { | |
176 | fPt = val; } | |
f1d945a1 | 177 | inline void AliFlowTrackSimple::SetPhi(Double_t val) { |
178 | fPhi = val; } | |
44e060e0 | 179 | inline void AliFlowTrackSimple::SetWeight(Double_t val) { |
180 | fTrackWeight = val; } | |
701f71c1 | 181 | inline void AliFlowTrackSimple::SetCharge(Int_t val) { |
182 | fCharge = val; } | |
60875c3c | 183 | inline void AliFlowTrackSimple::SetMass(Double_t val) { |
184 | fMass = val; } | |
8fa6a5fa MK |
185 | |
186 | //TBits | |
1918addd | 187 | inline void AliFlowTrackSimple::SetForRPSelection(Bool_t val) { |
8eca5d19 | 188 | fFlowBits.SetBitNumber(kRP,val); } |
1918addd | 189 | inline void AliFlowTrackSimple::SetForPOISelection(Bool_t val) { |
8eca5d19 | 190 | fFlowBits.SetBitNumber(kPOI,val); } |
b125a454 | 191 | inline void AliFlowTrackSimple::SetForSubevent(Int_t i) { |
192 | fSubEventBits.SetBitNumber(i,kTRUE); } | |
193 | ||
8fa6a5fa MK |
194 | inline void AliFlowTrackSimple::SetPOItype(Int_t poiType, Bool_t b) { |
195 | fFlowBits.SetBitNumber(poiType,b); } | |
196 | ||
f1d945a1 | 197 | #endif |
198 |