Create the magnetic field map in the reconstruction macros (Yuri)
[u/mrichter/AliRoot.git] / MUON / AliMUONRawCluster.h
1 #ifndef ALIMUONRAWCLUSTER_H
2 #define ALIMUONRAWCLUSTER_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 /* $Id$ */
8 // Revision of includes 07/05/2004
9
10 /// \ingroup base
11 /// \class AliMUONRawCluster
12 /// \brief MUON raw cluster
13 ///
14 /// Class for the MUON RecPoint
15 /// It contains the properties of the physics cluters found in the tracking chambers
16 /// RawCluster contains also the information from the both cathode of the chambers.
17
18
19 #include "AliMUONVCluster.h"
20 #include <TMath.h> // because of inline funtion GetRadius
21 #include <TArrayF.h>
22
23 class AliMUONRawCluster : public AliMUONVCluster {
24
25 public:
26    AliMUONRawCluster();
27    virtual ~AliMUONRawCluster();
28    
29            /// Clear method (used by TClonesArray)
30    virtual void Clear(Option_t* = "") {}
31    
32            /// Create a copy of the current cluster
33    virtual AliMUONRawCluster* Clone(const char* = "") const {return new AliMUONRawCluster(*this);}
34    
35            /// Set coordinates (cm)
36   virtual void     SetXYZ(Double_t x, Double_t y, Double_t z) {fX[0] = x; fY[0] = y; fZ[0] = z;}
37            /// Return coordinate X (cm)
38   virtual Double_t GetX() const {return fX[0];}
39            /// Return coordinate Y (cm)
40   virtual Double_t GetY() const {return fY[0];}
41            /// Return coordinate Z (cm)
42   virtual Double_t GetZ() const {return fZ[0];}
43   
44            /// Set resolution (cm) on coordinates (X,Y)
45   virtual void     SetErrXY(Double_t errX, Double_t errY) {fErrXY[0] = errX; fErrXY[1] = errY;}
46            /// Return resolution (cm) on coordinate X
47   virtual Double_t GetErrX() const {return fErrXY[0];}
48            /// Return resolution**2 (cm**2) on coordinate X
49   virtual Double_t GetErrX2() const {return fErrXY[0] * fErrXY[0];}
50            /// Return resolution (cm) on coordinate Y
51   virtual Double_t GetErrY() const {return fErrXY[1];}
52            /// Return resolution**2 (cm**2) on coordinate Y
53   virtual Double_t GetErrY2() const {return fErrXY[1] * fErrXY[1];}
54   
55            /// Set the cluster charge
56   virtual void     SetCharge(Double_t q) {fQ[0] = q;}
57            /// Set the cluster charge
58   virtual Double_t GetCharge() const {return fQ[0];}
59   
60            /// Return chamber Id
61   virtual Int_t    GetChamberId() const {return fDetElemId/100 - 1;}
62            /// Set detection element Id
63           void     SetDetElemId(Int_t id) {fDetElemId = id;}
64            /// Return detection element Id
65   virtual Int_t    GetDetElemId() const {return fDetElemId;}
66   
67   virtual void     SetDigitsId(Int_t nDigits, const UInt_t *digitsId);
68            /// Add a digit Id to the array of associated digits
69   virtual void     AddDigitId(UInt_t id) {fIndexMap[fMultiplicity[0]++][0] = id;}
70
71            /// Return number of associated digits
72   virtual Int_t    GetNDigits() const {return fMultiplicity[0];}
73            /// Return Id of digits i
74   virtual UInt_t   GetDigitId(Int_t i) const {return (i < fMultiplicity[0] && i < 50) ? (UInt_t)fIndexMap[i][0] : 0;}
75   
76            /// Set chi2 of cluster
77   virtual void     SetChi2( Double_t chi2) {fChi2[0] = chi2;}
78            /// Return chi2 of cluster
79   virtual Double_t GetChi2() const {return fChi2[0];}
80    
81    /// Return radius
82    Float_t      GetRadius(Int_t i) {return TMath::Sqrt(fX[i]*fX[i]+fY[i]*fY[i]);}
83    /// Return true as the function Compare() is implemented
84    Bool_t       IsSortable() const {return kTRUE;}
85    Int_t        Compare(const TObject *obj) const;
86    Int_t        PhysicsContribution() const;
87    virtual void Print(Option_t* opt="") const;
88    static Int_t BinarySearch(Float_t r, TArrayF ccord, Int_t from, Int_t upto);
89    static void  SortMin(Int_t *idx,Float_t *xdarray, Float_t *xarray, Float_t *yarray, Float_t *qarray,Int_t ntr);
90    void         DumpIndex();
91
92    Int_t        AddCharge(Int_t i, Float_t Q);
93    Int_t        AddX(Int_t i, Float_t X);
94    Int_t        AddY(Int_t i, Float_t Y);
95    Int_t        AddZ(Int_t i, Float_t Z);
96
97    Float_t      GetCharge(Int_t i) const;
98    Float_t      GetX(Int_t i) const;
99    Float_t      GetY(Int_t i) const;
100    Float_t      GetZ(Int_t i) const;
101    Int_t        GetTrack(Int_t i=0) const;
102    Float_t      GetPeakSignal(Int_t i=0) const;
103    Int_t        GetMultiplicity(Int_t i=0) const;
104    Int_t        GetClusterType() const;
105    Int_t        GetGhost() const;
106    Int_t        GetNcluster(Int_t i=0) const;
107    Float_t      GetChi2(Int_t i) const;
108    Int_t        GetIndex(Int_t i, Int_t j) const;
109    Int_t        GetOffset(Int_t i, Int_t j) const;
110    Float_t      GetContrib(Int_t i, Int_t j) const;
111    Int_t        GetPhysics(Int_t i) const;
112
113    Int_t        SetCharge(Int_t i, Float_t Q);
114    Int_t        SetX(Int_t i, Float_t X);
115    Int_t        SetY(Int_t i, Float_t Y);
116    Int_t        SetZ(Int_t i, Float_t Z);
117    Int_t        SetTrack(Int_t i, Int_t track);
118    Int_t        SetPeakSignal(Int_t i, Float_t peaksignal);
119    Int_t        SetMultiplicity(Int_t i, Int_t mul);
120    Int_t        SetClusterType(Int_t type);
121    Int_t        SetGhost(Int_t ghost);
122    Int_t        SetNcluster(Int_t i, Int_t ncluster);
123    Int_t        SetChi2(Int_t i, Float_t chi2);
124    void         SetIndex(Int_t i, Int_t j, Int_t index);
125    void         SetOffset(Int_t i, Int_t j, Int_t offset);
126    void         SetContrib(Int_t i, Int_t j, Float_t contrib);
127    void         SetPhysics(Int_t i, Int_t physics);
128
129 private:
130    Int_t       fIndexMap[50][2];  ///< Indices of digits
131    Int_t       fOffsetMap[50][2]; ///< Emmanuel special
132    Float_t     fContMap[50][2];   ///< Contribution from digit
133    Int_t       fPhysicsMap[50];   ///< Distinguish signal and background contr.
134   
135    Float_t     fQ[2]  ;           ///< Q of cluster (in ADC counts)     
136    Float_t     fX[2]  ;           ///< X of cluster
137    Float_t     fY[2]  ;           ///< Y of cluster
138    Float_t     fZ[2]  ;           ///< Z of cluster
139    Int_t       fTracks[3];        ///< Labels of overlapped tracks
140    Float_t     fPeakSignal[2];    ///< Peak signal 
141    Int_t       fMultiplicity[2];  ///< Cluster multiplicity
142    Int_t       fClusterType;      ///< Cluster type
143    Int_t       fGhost;            ///< Ghost info
144                                   // 0 if not a ghost or ghost problem solved
145                                   // >0 if ghost problem remains because
146                                   // 1 both (true and ghost) satify 
147                                   //   charge chi2 compatibility
148                                   // 2 none give satisfactory chi2
149    Int_t       fNcluster[2];      ///< Number of clusters
150    Float_t     fChi2[2];          ///< Chi**2 of fit
151    Int_t       fDetElemId;        ///< ID number of the detection element (slat) on which the cluster is found. 
152    Float_t     fErrXY[2];         ///< coordinate errors
153    
154    ClassDef(AliMUONRawCluster,3)  //Cluster class for MUON
155 };
156
157 // inline functions
158
159 /// Return Indices of digits
160 inline  Int_t  AliMUONRawCluster::GetIndex(Int_t i, Int_t j) const
161 { return fIndexMap[i][j]; }
162
163 /// Return Emmanuel special offset map
164 inline  Int_t  AliMUONRawCluster::GetOffset(Int_t i, Int_t j) const
165 { return fOffsetMap[i][j]; }
166
167 /// Return Contribution from digit
168 inline  Float_t  AliMUONRawCluster::GetContrib(Int_t i, Int_t j) const
169 { return fContMap[i][j]; }
170
171 /// Return Distinguish signal and background contr.
172 inline  Int_t  AliMUONRawCluster::GetPhysics(Int_t i) const
173 { return fPhysicsMap[i]; }
174
175 /// Set Indices of digits
176 inline  void  AliMUONRawCluster::SetIndex(Int_t i, Int_t j, Int_t index)
177 { fIndexMap[i][j] = index; }
178
179 /// Set Emmanuel special offset map
180 inline  void  AliMUONRawCluster::SetOffset(Int_t i, Int_t j, Int_t offset)
181 { fOffsetMap[i][j] = offset; }
182
183 /// Set Contribution from digit
184 inline  void  AliMUONRawCluster::SetContrib(Int_t i, Int_t j, Float_t contrib)
185 { fContMap[i][j] = contrib; }
186
187 /// Set Distinguish signal and background contr.
188 inline  void  AliMUONRawCluster::SetPhysics(Int_t i, Int_t physics)
189 { fPhysicsMap[i] = physics; }
190
191
192 #endif
193