]>
Commit | Line | Data |
---|---|---|
1 | #ifndef ALIMULTIPLICITY_H | |
2 | #define ALIMULTIPLICITY_H | |
3 | ||
4 | #include<TObject.h> | |
5 | #include <TBits.h> | |
6 | #include<TMath.h> | |
7 | ||
8 | //////////////////////////////////////////////////////// | |
9 | //// Class containing multiplicity information // | |
10 | //// to stored in the ESD // | |
11 | //////////////////////////////////////////////////////// | |
12 | ||
13 | class AliMultiplicity : public TObject { | |
14 | ||
15 | public: | |
16 | ||
17 | AliMultiplicity(); // default constructor | |
18 | // standard constructor | |
19 | AliMultiplicity(Int_t ntr,Float_t *th, Float_t *ph, Float_t *dth, Float_t *dph, Int_t *labels, | |
20 | Int_t* labelsL2, Int_t ns, Float_t *ts, Float_t *ps, Short_t nfcL1, Short_t nfcL2, TBits fFastOrFiredChips); | |
21 | AliMultiplicity(const AliMultiplicity& m); | |
22 | AliMultiplicity& operator=(const AliMultiplicity& m); | |
23 | virtual void Copy(TObject &obj) const; | |
24 | virtual ~AliMultiplicity(); | |
25 | // methods to access tracklet information | |
26 | Int_t GetNumberOfTracklets() const {return fNtracks;} | |
27 | Double_t GetTheta(Int_t i) const { | |
28 | if(i>=0 && i<fNtracks) return fTh[i]; | |
29 | Error("GetTheta","Invalid track number %d",i); return -9999.; | |
30 | } | |
31 | Double_t GetEta(Int_t i) const { | |
32 | if(i>=0 && i<fNtracks) return -TMath::Log(TMath::Tan(fTh[i]/2.)); | |
33 | Error("GetEta","Invalid track number %d",i); return -9999.; | |
34 | } | |
35 | Double_t GetPhi(Int_t i) const { | |
36 | if(i>=0 && i<fNtracks) return fPhi[i]; | |
37 | Error("GetPhi","Invalid track number %d",i); return -9999.; | |
38 | } | |
39 | Double_t GetDeltaTheta(Int_t i) const { | |
40 | if(fDeltTh && i>=0 && i<fNtracks) return fDeltTh[i]; | |
41 | Error("GetDeltaTheta","DeltaTheta not available in data or Invalid track number %d(max %d)",i, fNtracks); return -9999.; | |
42 | } | |
43 | Double_t GetDeltaPhi(Int_t i) const { | |
44 | if(i>=0 && i<fNtracks) return fDeltPhi[i]; | |
45 | Error("GetDeltaPhi","Invalid track number %d",i); return -9999.; | |
46 | } | |
47 | ||
48 | Int_t GetLabel(Int_t i, Int_t layer) const; | |
49 | void SetLabel(Int_t i, Int_t layer, Int_t label); | |
50 | ||
51 | // methods to access single cluster information | |
52 | Int_t GetNumberOfSingleClusters() const {return fNsingle;} | |
53 | Double_t GetThetaSingle(Int_t i) const { | |
54 | if(i>=0 && i<fNsingle) return fThsingle[i]; | |
55 | Error("GetThetaSingle","Invalid cluster number %d",i); return -9999.; | |
56 | } | |
57 | ||
58 | Double_t GetPhiSingle(Int_t i) const { | |
59 | if(i>=0 && i<fNsingle) return fPhisingle[i]; | |
60 | Error("GetPhisingle","Invalid cluster number %d",i); return -9999.; | |
61 | } | |
62 | ||
63 | Short_t GetNumberOfFiredChips(Int_t layer) const { return fFiredChips[layer]; } | |
64 | void SetFiredChips(Int_t layer, Short_t firedChips) { fFiredChips[layer] = firedChips; } | |
65 | ||
66 | UInt_t GetNumberOfITSClusters(Int_t layer) const { return layer<6 ? fITSClusters[layer] : 0; } | |
67 | UInt_t GetNumberOfITSClusters(Int_t layMin, Int_t layMax) const ; | |
68 | void SetITSClusters(Int_t layer, UInt_t clusters) { fITSClusters[layer] = clusters; } | |
69 | ||
70 | void SetFastOrFiredChips(UInt_t chipKey){fFastOrFiredChips.SetBitNumber(chipKey);} | |
71 | TBits GetFastOrFiredChips() const {return fFastOrFiredChips;} | |
72 | Bool_t TestFastOrFiredChips(UInt_t chipKey) const {return fFastOrFiredChips.TestBitNumber(chipKey);} | |
73 | ||
74 | protected: | |
75 | void Duplicate(const AliMultiplicity &m); // used by copy ctr. | |
76 | ||
77 | Int_t fNtracks; // Number of tracklets | |
78 | Int_t fNsingle; // Number of clusters on SPD layer 1, not associated | |
79 | // with a tracklet on SPD layer 2 | |
80 | Int_t *fLabels; //[fNtracks] array with labels of cluster in L1 used for tracklet | |
81 | Int_t *fLabelsL2; //[fNtracks] array with labels of cluster in L2 used for tracklet | |
82 | Double32_t *fTh; //[fNtracks] array with theta values | |
83 | Double32_t *fPhi; //[fNtracks] array with phi values | |
84 | Double32_t *fDeltTh; //[fNtracks] array with delta theta values | |
85 | Double32_t *fDeltPhi; //[fNtracks] array with delta phi values | |
86 | Double32_t *fThsingle; //[fNsingle] array with theta values of L1 clusters | |
87 | Double32_t *fPhisingle; //[fNsingle] array with phi values of L2 clusters | |
88 | Short_t fFiredChips[2]; // Number of fired chips in the two SPD layers | |
89 | UInt_t fITSClusters[6]; // Number of ITS cluster per layer | |
90 | TBits fFastOrFiredChips; // Map of FastOr fired chips | |
91 | ||
92 | ClassDef(AliMultiplicity,10); | |
93 | }; | |
94 | ||
95 | inline Int_t AliMultiplicity::GetLabel(Int_t i, Int_t layer) const | |
96 | { | |
97 | if(i>=0 && i<fNtracks) { | |
98 | if (layer == 0) { | |
99 | return fLabels[i]; | |
100 | } else if (layer == 1) { | |
101 | if (fLabelsL2) { | |
102 | return fLabelsL2[i]; | |
103 | } else { | |
104 | Warning("GetLabel", "No information for layer 2 available !"); | |
105 | return -9999; | |
106 | } | |
107 | } else { | |
108 | Error("GetLabel","Invalid layer number %d",layer); return -9999; | |
109 | } | |
110 | } else { | |
111 | Error("GetLabel","Invalid track number %d",i); return -9999; | |
112 | } | |
113 | return -9999; | |
114 | } | |
115 | #endif |