]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/ESD/AliMultiplicity.h
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
test
[u/mrichter/AliRoot.git] / STEER / ESD / AliMultiplicity.h
CommitLineData
32e449be 1#ifndef ALIMULTIPLICITY_H
2#define ALIMULTIPLICITY_H
3
59495bf1 4#include <TObject.h>
ff44c37c 5#include <TBits.h>
59495bf1 6#include <TMath.h>
7class AliRefArray;
32e449be 8
9////////////////////////////////////////////////////////
10//// Class containing multiplicity information //
11//// to stored in the ESD //
12////////////////////////////////////////////////////////
13
14class AliMultiplicity : public TObject {
15
16 public:
59495bf1 17 //
b9be2602 18 enum {kMultTrackRefs =BIT(14), // in new format (old is default for bwd.comp.) multiple cluster->track references are allowed
19 kScaleDThtbySin2=BIT(15), // scale Dtheta by 1/sin^2(theta). Default is DON'T scale, for bwd.comp.
20 kSPD2Sng =BIT(16) // are SPD2 singles stored?
59495bf1 21 };
32e449be 22 AliMultiplicity(); // default constructor
23 // standard constructor
fa9ed8e9 24 AliMultiplicity(Int_t ntr,Float_t *th, Float_t *ph, Float_t *dth, Float_t *dph, Int_t *labels,
d7c5c1e4 25 Int_t* labelsL2, Int_t ns, Float_t *ts, Float_t *ps, Int_t *labelss, Short_t nfcL1, Short_t nfcL2, const TBits & fFastOrFiredChips);
1f9831ab 26 AliMultiplicity(Int_t ntr, Int_t ns, Short_t nfcL1, Short_t nfcL2, const TBits & fFastOr);
32e449be 27 AliMultiplicity(const AliMultiplicity& m);
28 AliMultiplicity& operator=(const AliMultiplicity& m);
732a24fe 29 virtual void Copy(TObject &obj) const;
1f9831ab 30 virtual void Clear(Option_t* opt="");
32e449be 31 virtual ~AliMultiplicity();
59495bf1 32 // methods to access tracklet information
33 Bool_t GetMultTrackRefs() const {return TestBit(kMultTrackRefs);}
34 void SetMultTrackRefs(Bool_t v) {SetBit(kMultTrackRefs,v);}
35 Bool_t GetScaleDThetaBySin2T() const {return TestBit(kScaleDThtbySin2);}
36 void SetScaleDThetaBySin2T(Bool_t v) {SetBit(kScaleDThtbySin2,v);}
37
38 //
32e449be 39 Int_t GetNumberOfTracklets() const {return fNtracks;}
4f499767 40 Double_t GetTheta(Int_t i) const {
41 if(i>=0 && i<fNtracks) return fTh[i];
42 Error("GetTheta","Invalid track number %d",i); return -9999.;
43 }
44 Double_t GetEta(Int_t i) const {
45 if(i>=0 && i<fNtracks) return -TMath::Log(TMath::Tan(fTh[i]/2.));
46 Error("GetEta","Invalid track number %d",i); return -9999.;
47 }
48 Double_t GetPhi(Int_t i) const {
49 if(i>=0 && i<fNtracks) return fPhi[i];
50 Error("GetPhi","Invalid track number %d",i); return -9999.;
51 }
52 Double_t GetDeltaTheta(Int_t i) const {
53 if(fDeltTh && i>=0 && i<fNtracks) return fDeltTh[i];
54 Error("GetDeltaTheta","DeltaTheta not available in data or Invalid track number %d(max %d)",i, fNtracks); return -9999.;
55 }
56 Double_t GetDeltaPhi(Int_t i) const {
57 if(i>=0 && i<fNtracks) return fDeltPhi[i];
58 Error("GetDeltaPhi","Invalid track number %d",i); return -9999.;
59 }
eda42f29 60
59495bf1 61 Double_t CalcDist(Int_t it) const;
b9be2602 62 Float_t GetThetaAll(int icl, int lr) const;
63 Float_t GetPhiAll(int icl, int lr) const;
64 Int_t GetLabelAll(int icl, int lr) const;
59495bf1 65
eda42f29 66 Int_t GetLabel(Int_t i, Int_t layer) const;
56258350 67 void SetLabel(Int_t i, Int_t layer, Int_t label);
d7c5c1e4 68 Int_t GetLabelSingle(Int_t i) const;
b9be2602 69 Int_t GetLabelSingleLr(Int_t i, Int_t layer) const;
d7c5c1e4 70 void SetLabelSingle(Int_t i, Int_t label);
71
34581d1e 72 Bool_t FreeClustersTracklet(Int_t i, Int_t mode) const;
73 Bool_t FreeSingleCluster(Int_t i, Int_t mode) const;
1f9831ab 74
eda42f29 75
968e8539 76// methods to access single cluster information
b9be2602 77 Int_t SetNumberOfSingleClustersSPD2(Int_t n) {return fNsingleSPD2 = n;}
968e8539 78 Int_t GetNumberOfSingleClusters() const {return fNsingle;}
b9be2602 79 Int_t GetNumberOfSingleClustersLr(Int_t lr) const;
80 Bool_t AreSPD2SinglesStored() const {return TestBit(kSPD2Sng);}
81 void SetSPD2SinglesStored(Bool_t v=kTRUE) {return SetBit(kSPD2Sng,v);}
82
4f499767 83 Double_t GetThetaSingle(Int_t i) const {
84 if(i>=0 && i<fNsingle) return fThsingle[i];
85 Error("GetThetaSingle","Invalid cluster number %d",i); return -9999.;
86 }
87
b9be2602 88 Double_t GetThetaSingleLr(Int_t i, Int_t lr) const {
89 if (lr==1) {
90 if (!AreSPD2SinglesStored()) {Error("GetThetaSingle","Invalid cluster number %d for lr %d",i,lr); return -9999.;}
91 else i += GetNumberOfSingleClustersLr(0);
92 }
93 if(i>=0 && i<fNsingle) return fThsingle[i];
94 Error("GetThetaSingle","Invalid cluster number %d",i); return -9999.;
95 }
96
4f499767 97 Double_t GetPhiSingle(Int_t i) const {
98 if(i>=0 && i<fNsingle) return fPhisingle[i];
99 Error("GetPhisingle","Invalid cluster number %d",i); return -9999.;
100 }
32e449be 101
b9be2602 102 Double_t GetPhiSingleLr(Int_t i, Int_t lr) const {
103 if (lr==1) {
104 if (!AreSPD2SinglesStored()) {Error("GetPhiSingle","Invalid cluster number %d for lr %d",i,lr); return -9999.;}
105 else i += GetNumberOfSingleClustersLr(0);
106 }
107 if(i>=0 && i<fNsingle) return fPhisingle[i];
108 Error("GetPhisingle","Invalid cluster number %d",i); return -9999.;
109 }
110
ab37e14c 111 Short_t GetNumberOfFiredChips(Int_t layer) const { return fFiredChips[layer]; }
112 void SetFiredChips(Int_t layer, Short_t firedChips) { fFiredChips[layer] = firedChips; }
466ffa28 113
d325d83f 114 UInt_t GetNumberOfITSClusters(Int_t layer) const { return layer<6 ? fITSClusters[layer] : 0; }
115 UInt_t GetNumberOfITSClusters(Int_t layMin, Int_t layMax) const ;
116 void SetITSClusters(Int_t layer, UInt_t clusters) { fITSClusters[layer] = clusters; }
117
ee87788b 118 void SetFastOrFiredChips(UInt_t chipKey){fFastOrFiredChips.SetBitNumber(chipKey);}
adc4b00f 119 const TBits & GetFastOrFiredChips() const {return fFastOrFiredChips;}
ee87788b 120 Bool_t TestFastOrFiredChips(UInt_t chipKey) const {return fFastOrFiredChips.TestBitNumber(chipKey);}
ff44c37c 121
ab37e14c 122 void SetFiredChipMap(TBits & firedChips){fClusterFiredChips = firedChips;}
123 void SetFiredChipMap(UInt_t chipKey){fClusterFiredChips.SetBitNumber(chipKey);}
124 const TBits & GetFiredChipMap() const {return fClusterFiredChips;}
125 Bool_t TestFiredChipMap(UInt_t chipKey) const {return fClusterFiredChips.TestBitNumber(chipKey);}
adc4b00f 126
34581d1e 127 Bool_t GetTrackletTrackIDs(Int_t i, Int_t mode, Int_t &spd1, Int_t &spd2) const;
59495bf1 128 Int_t GetTrackletTrackIDsLay(Int_t lr,Int_t i, Int_t mode, UInt_t* refs, UInt_t maxRef) const;
34581d1e 129 Bool_t GetSingleClusterTrackID(Int_t i, Int_t mode, Int_t &tr) const;
59495bf1 130 Int_t GetSingleClusterTrackIDs(Int_t i, Int_t mode, UInt_t* refs, UInt_t maxRef) const;
1f9831ab 131
132 // array getters
133 Double_t* GetTheta() const {return (Double_t*)fTh;}
134 Double_t* GetPhi() const {return (Double_t*)fPhi;}
135 Double_t* GetDeltTheta() const {return (Double_t*)fDeltTh;}
136 Double_t* GetDeltPhi() const {return (Double_t*)fDeltPhi;}
137 Double_t* GetThetaSingle() const {return (Double_t*)fThsingle;}
138 Double_t* GetPhiSingle() const {return (Double_t*)fPhisingle;}
d7c5c1e4 139 Int_t* GetLabels() const {return (Int_t*)fLabels;}
1f9831ab 140 Int_t* GetLabels2() const {return (Int_t*)fLabelsL2;}
d7c5c1e4 141 Int_t* GetLabelsSingle() const {return (Int_t*)fLabelssingle;}
1f9831ab 142
59495bf1 143 void AttachTracklet2TrackRefs(AliRefArray* l1t1,AliRefArray* l1t2,AliRefArray* l2t1,AliRefArray* l2t2) {
144 fTCl2Tracks[0][0] = l1t1; fTCl2Tracks[0][1] = l1t2; fTCl2Tracks[1][0] = l2t1; fTCl2Tracks[1][1] = l2t2;
145 }
146 void AttachCluster2TrackRefs(AliRefArray* l1t1,AliRefArray* l1t2) {
147 fSCl2Tracks[0] = l1t1; fSCl2Tracks[1] = l1t2;
148 }
149 void SetTrackletData(Int_t id, const Float_t* tlet, UInt_t trSPD1=0, UInt_t trSPD2=0);
150 void SetSingleClusterData(Int_t id, const Float_t* scl,UInt_t tr=0);
151 void CompactBits();
152 //
153 void SetDPhiWindow2(Float_t v=-1) {fDPhiWindow2 = v;}
154 void SetDThetaWindow2(Float_t v=-1) {fDThetaWindow2 = v;}
155 void SetDPhiShift(Float_t v=-1) {fDPhiShift = v;}
156 void SetNStdDev(Float_t v=1) {fNStdDev = v;}
157 //
158 Float_t GetDPhiWindow2() const {return fDPhiWindow2;}
159 Float_t GetDThetaWindow2() const {return fDThetaWindow2;}
160 Float_t GetDPhiShift() const {return fDPhiShift;}
161 Float_t GetNStdDev() const {return fNStdDev;}
162
163 //
1f9831ab 164 virtual void Print(Option_t *opt="") const;
165
32e449be 166 protected:
167 void Duplicate(const AliMultiplicity &m); // used by copy ctr.
32e449be 168
cd888a89 169 Int_t fNtracks; // Number of tracklets
b9be2602 170 Int_t fNsingle; // Number of clusters on SPD layer 1 and 2 (if storage of spd2 singles requested), not associated with a tracklet on otherSPD
171 Int_t fNsingleSPD2; // Number of clusters on SPD layer 2 not associated (if stored)
59495bf1 172 //
173 Float_t fDPhiWindow2; // sigma^2 in dphi used in reco
174 Float_t fDThetaWindow2; // sigma^2 in dtheta used in reco
175 Float_t fDPhiShift; // bending shift used
176 Float_t fNStdDev; // number of standard deviations kept
177 //
0939e22a 178 Int_t *fLabels; //[fNtracks] array with labels of cluster in L1 used for tracklet
179 Int_t *fLabelsL2; //[fNtracks] array with labels of cluster in L2 used for tracklet
59495bf1 180 UInt_t* fUsedClusS; //[fNsingle] id+1 of the tracks using cluster, coded as (TPC/ITS+ITS_SA)+(ITS_SA_PURE<<16) !!! Outphased for multiple refs
c8122432 181 ULong64_t* fUsedClusT; //[fNtracks] id+1 of the tracks using clusters, coded as (TPC/ITS+ITS_SA)+(ITS_SA_PURE<<16) for SPD1 and SPD2 in low and high parts
59495bf1 182 AliRefArray *fTCl2Tracks[2][2]; // container with multiple tracklet_cluster->track references
183 AliRefArray *fSCl2Tracks[2]; // container with multiple single_cluster->track references
cd888a89 184 Double32_t *fTh; //[fNtracks] array with theta values
185 Double32_t *fPhi; //[fNtracks] array with phi values
fa9ed8e9 186 Double32_t *fDeltTh; //[fNtracks] array with delta theta values
cd888a89 187 Double32_t *fDeltPhi; //[fNtracks] array with delta phi values
188 Double32_t *fThsingle; //[fNsingle] array with theta values of L1 clusters
1f9831ab 189 Double32_t *fPhisingle; //[fNsingle] array with phi values of L1 clusters
d7c5c1e4 190 Int_t *fLabelssingle; //[fNsingle] array with labels of clusters in L1 not used for tracklets
ab37e14c 191 Short_t fFiredChips[2]; // Number of fired chips in the two SPD layers
d7c5c1e4 192 UInt_t fITSClusters[6]; // Number of ITS cluster per layer
ff44c37c 193 TBits fFastOrFiredChips; // Map of FastOr fired chips
1f9831ab 194 TBits fClusterFiredChips; // Map of fired chips (= at least one cluster)
ff44c37c 195
b9be2602 196 ClassDef(AliMultiplicity,19);
32e449be 197};
198
eda42f29 199inline Int_t AliMultiplicity::GetLabel(Int_t i, Int_t layer) const
200{
201 if(i>=0 && i<fNtracks) {
202 if (layer == 0) {
203 return fLabels[i];
204 } else if (layer == 1) {
205 if (fLabelsL2) {
206 return fLabelsL2[i];
207 } else {
208 Warning("GetLabel", "No information for layer 2 available !");
209 return -9999;
210 }
211 } else {
212 Error("GetLabel","Invalid layer number %d",layer); return -9999;
213 }
214 } else {
215 Error("GetLabel","Invalid track number %d",i); return -9999;
216 }
4f499767 217 return -9999;
eda42f29 218}
d7c5c1e4 219
220inline Int_t AliMultiplicity::GetLabelSingle(Int_t i) const
221{
222 if(i>=0 && i<fNsingle) {
223 return fLabelssingle[i];
224 } else {
225 Error("GetLabelSingle","Invalid cluster number %d",i); return -9999;
226 }
227 return -9999;
228}
229
230
59495bf1 231inline Double_t AliMultiplicity::CalcDist(Int_t i) const
232{
233 // calculate eliptical distance. theta is the angle of cl1, dtheta = tht(cl1)-tht(cl2)
234 if (i<0 && i>=fNtracks) return -1;
235 if (fDPhiWindow2<1E-9 || fDThetaWindow2<1E-9) return -1; // not stored
236 double dphi = TMath::Abs(fDeltPhi[i]) - fDPhiShift;
237 double dtheta = fDeltTh[i];
238 if (GetScaleDThetaBySin2T()) {
239 double sinTI = TMath::Sin(fTh[i]-dtheta/2);
240 sinTI *= sinTI;
241 dtheta /= sinTI>1.e-6 ? sinTI : 1.e-6;
242 }
243 return dphi*dphi/fDPhiWindow2 + dtheta*dtheta/fDThetaWindow2;
244}
245
b9be2602 246inline Int_t AliMultiplicity::GetNumberOfSingleClustersLr(Int_t lr) const
247{
248 // return number of singles at given layer
249 if (lr==0) return fNsingle - fNsingleSPD2;
250 return AreSPD2SinglesStored() ? fNsingleSPD2 : -1;
251}
252
d7c5c1e4 253
254
32e449be 255#endif