Remove several warnings
[u/mrichter/AliRoot.git] / MUON / AliMUON.h
1 #ifndef MUON_H
2 #define MUON_H
3 ////////////////////////////////////////////////
4 //  Manager and hits classes for set:MUON     //
5 ////////////////////////////////////////////////
6 #include "AliDetector.h"
7 #include "AliHit.h"
8 #include "AliMUONConst.h"
9 #include "AliDigit.h"
10 #include "AliMUONchamber.h"
11 #include "AliMUONSegRes.h"
12 #include <TVector.h>
13 #include <TObjArray.h>
14 #include <TArrayF.h>
15 #include <TFile.h>
16 #include <TTree.h>
17 typedef enum {simple, medium, big} Cluster_t;
18
19 static const int NCH=14;
20
21 class AliMUONcluster;
22 class AliMUONRawCluster;
23 class AliMUONClusterFinder;
24 class AliMUONcorrelation;
25
26
27 //----------------------------------------------
28
29  
30 class AliMUONcluster : public TObject {
31 public:
32
33    Int_t     fHitNumber;    // Hit number
34    Int_t     fCathode;      // Cathode number
35    Int_t     fQ  ;          // Total charge      
36    Int_t     fPadX  ;       // Pad number along X
37    Int_t     fPadY  ;       // Pad number along Y
38    Int_t     fQpad  ;       // Charge per pad
39    Int_t     fRSec  ;       // R -sector of pad
40  
41 public:
42    AliMUONcluster() {
43       fHitNumber=fQ=fPadX=fPadY=fQpad=fRSec=0;   
44 }
45    AliMUONcluster(Int_t *clhits);
46    virtual ~AliMUONcluster() {;}
47  
48    ClassDef(AliMUONcluster,1)  //Cluster object for set:MUON
49 };
50
51  
52 class AliMUONreccluster : public TObject {
53 public:
54
55    Int_t     fTracks[3];      //labels of overlapped tracks
56
57    Int_t       fQ  ;          // Q of cluster (in ADC counts)     
58    Float_t     fX  ;          // X of cluster
59    Float_t     fY  ;          // Y of cluster
60
61 public:
62    AliMUONreccluster() {
63        fTracks[0]=fTracks[1]=fTracks[2]=-1; 
64        fQ=0; fX=fY=0; 
65    }
66    virtual ~AliMUONreccluster() {;}
67  
68    ClassDef(AliMUONreccluster,1)  //Cluster object for set:MUON
69 };
70
71 //_____________________________________________________________________________
72
73 class AliMUONdigit : public TObject {
74  public:
75     Int_t     fPadX;        // Pad number along x
76     Int_t     fPadY ;       // Pad number along y
77     Int_t     fSignal;      // Signal amplitude
78     Int_t     fTcharges[10];   // charge per track making this digit (up to 10)
79     Int_t     fTracks[10];     // primary tracks making this digit (up to 10)
80     Int_t     fPhysics;        // physics contribution to signal 
81     Int_t     fHit;            // hit number - temporary solution
82
83
84  
85  public:
86     AliMUONdigit() {}
87     AliMUONdigit(Int_t *digits);
88     AliMUONdigit(Int_t *tracks, Int_t *charges, Int_t *digits);
89     virtual ~AliMUONdigit();
90     
91     ClassDef(AliMUONdigit,1)  //Digits for set:MUON
92 };
93 //_____________________________________________________________________________
94
95 class AliMUONlist : public AliMUONdigit {
96  public:
97     Int_t          fChamber;       // chamber number of pad
98     TObjArray     *fTrackList; 
99  public:
100     AliMUONlist() {fTrackList=0;}
101     AliMUONlist(Int_t rpad, Int_t *digits);
102     virtual ~AliMUONlist() {delete fTrackList;}
103     TObjArray  *TrackList()   {return fTrackList;}
104     ClassDef(AliMUONlist,1)  //Digits for set:MUON
105 };
106 //___________________________________________
107
108
109 //___________________________________________
110  
111 class AliMUONhit : public AliHit {
112  public:
113     Int_t     fChamber;       // Chamber number
114     Float_t   fParticle;      // Geant3 particle type
115     Float_t   fTheta ;        // Incident theta angle in degrees      
116     Float_t   fPhi   ;        // Incident phi angle in degrees
117     Float_t   fTlength;       // Track length inside the chamber
118     Float_t   fEloss;         // ionisation energy loss in gas   
119     Int_t     fPHfirst;       // first padhit
120     Int_t     fPHlast;        // last padhit
121
122 // modifs perso
123     Float_t   fPTot;          // hit momentum P
124     Float_t   fCxHit;            // Px/P
125     Float_t   fCyHit;            // Py/P
126     Float_t   fCzHit;            // Pz/P
127
128  public:
129     AliMUONhit() {}
130     AliMUONhit(Int_t fIshunt, Int_t track, Int_t *vol, Float_t *hits);
131     virtual ~AliMUONhit() {}
132     
133     ClassDef(AliMUONhit,1)  //Hits object for set:MUON
134 };
135
136 class AliMUON : public  AliDetector {
137  public:
138     AliMUON();
139     AliMUON(const char *name, const char *title);
140     virtual       ~AliMUON();
141     virtual void   AddHit(Int_t, Int_t*, Float_t*);
142     virtual void   AddCluster(Int_t*);
143     virtual void   AddDigits(Int_t, Int_t*, Int_t*, Int_t*);
144     virtual void   AddRawCluster(Int_t, const AliMUONRawCluster&);
145     virtual void   AddCathCorrel(Int_t, Int_t*, Float_t*, Float_t*);
146     virtual void   BuildGeometry();
147     virtual void   CreateGeometry() {}
148     virtual void   CreateMaterials() {}
149     virtual void   StepManager();
150     Int_t          DistancetoPrimitive(Int_t px, Int_t py);
151     virtual Int_t  IsVersion() const =0;
152 //
153     TClonesArray  *Clusters() {return fClusters;}
154     virtual  void  MakeTreeC(Option_t *option="C");
155     void           GetTreeC(Int_t);
156     virtual void   MakeBranch(Option_t *opt=" ");
157     void           SetTreeAddress();
158     virtual void   ResetHits();
159     virtual void   ResetDigits();
160     virtual void   ResetRawClusters();
161     virtual void   ResetCorrelation();
162     virtual void   FindClusters(Int_t,Int_t);
163     virtual void   Digitise(Int_t,Int_t,Option_t *opt1=" ",Option_t *opt2=" ",Text_t *name=" ");
164     virtual void   CathodeCorrelation(Int_t);
165     virtual void   SortTracks(Int_t *,Int_t *,Int_t);
166 //
167 // modifs perso
168
169     void     InitTracking(Double_t &, Double_t &, Double_t &);
170     void     Reconst(Int_t &,Int_t &,Int_t,Int_t &,Int_t&,Int_t&, Option_t *option,Text_t *filename);
171     void     FinishEvent();
172     void     CloseTracking();
173     void     SetCutPxz(Double_t p) {fSPxzCut=p;}
174     void     SetSigmaCut(Double_t p) {fSSigmaCut=p;}
175     void     SetXPrec(Double_t p) {fSXPrec=p;}
176     void     SetYPrec(Double_t p) {fSYPrec=p;}
177     Double_t GetCutPxz() {return fSPxzCut;}
178     Double_t GetSigmaCut() {return fSSigmaCut;}
179     Double_t GetXPrec() {return fSXPrec;}
180     Double_t GetYPrec() {return fSYPrec;}
181 // fin modifs perso 
182     
183 // Configuration Methods (per station id)
184 //
185 // Set Chamber Segmentation Parameters
186 // id refers to the station and isec to the cathode plane   
187     virtual void   SetPADSIZ(Int_t id, Int_t isec, Float_t p1, Float_t p2);
188
189 // Set Signal Generation Parameters
190     virtual void   SetSigmaIntegration(Int_t id, Float_t p1);
191     virtual void   SetChargeSlope(Int_t id, Float_t p1);
192     virtual void   SetChargeSpread(Int_t id, Float_t p1, Float_t p2);
193     virtual void   SetMaxAdc(Int_t id, Float_t p1);
194 // Set Segmentation and Response Model
195     virtual void   SetSegmentationModel(Int_t id, Int_t isec, AliMUONsegmentation *segmentation);
196     virtual void   SetResponseModel(Int_t id, AliMUONresponse *response);
197     virtual void   SetNsec(Int_t id, Int_t nsec);
198 // Set Reconstruction Model
199     virtual void   SetReconstructionModel(Int_t id, AliMUONClusterFinder *reconstruction);
200 // Set Stepping Parameters
201     virtual void   SetMaxStepGas(Float_t p1);
202     virtual void   SetMaxStepAlu(Float_t p1);
203     virtual void   SetMaxDestepGas(Float_t p1);
204     virtual void   SetMaxDestepAlu(Float_t p1);
205     virtual void   SetMuonAcc(Bool_t acc=0, Float_t angmin=2, Float_t angmax=9);
206 // Response Simulation
207     virtual void   MakePadHits(Float_t xhit,Float_t yhit,Float_t eloss,Int_t id);
208 // Return reference to Chamber #id
209     virtual AliMUONchamber& Chamber(Int_t id) {return *((AliMUONchamber *) (*fChambers)[id]);}
210 // Retrieve pad hits for a given Hit
211     virtual AliMUONcluster* FirstPad(AliMUONhit *, TClonesArray *);
212     virtual AliMUONcluster* NextPad(TClonesArray *);
213 // Return pointers to digits 
214     TObjArray            *Dchambers() {return fDchambers;}
215     Int_t                *Ndch() {return fNdch;}
216     virtual TClonesArray *DigitsAddress(Int_t id) {return ((TClonesArray *) (*fDchambers)[id]);}
217 // Return pointers to reconstructed clusters
218     TObjArray            *RawClusters() {return fRawClusters;}
219     Int_t                *Nrawch() {return fNrawch;}
220     virtual TClonesArray *RawClustAddress(Int_t id) {return ((TClonesArray *) (*fRawClusters)[id]);}
221
222 // modifs perso
223     AliMUONRawCluster *RawCluster(Int_t ichamber, Int_t icathod, Int_t icluster);
224     
225     
226     // Return pointers to list of correlated clusters
227     TObjArray            *CathCorrel() {return fCathCorrel;}
228     Int_t                *Ncorch() {return fNcorch;}
229     virtual TClonesArray *CathCorrelAddress(Int_t id)
230         {return ((TClonesArray *) (*fCathCorrel)[id]);}
231
232 // Return pointer to TreeC
233     TTree      *TreeC() {return fTreeC;} 
234  protected:
235     TObjArray            *fChambers;           // List of Tracking Chambers
236     Int_t                fNclusters;           // Number of clusters
237     TClonesArray         *fClusters;           // List of clusters
238     TObjArray            *fDchambers;          // List of digits
239     Int_t                *fNdch;               // Number of digits
240
241
242     TObjArray            *fRawClusters;            // List of raw clusters
243     Int_t                *fNrawch;                 // Number of raw clusters
244     TObjArray            *fCathCorrel;             // List of correlated clusters
245     Int_t                *fNcorch;                 // Number of correl clusters
246     TTree                *fTreeC;                  // Cathode correl index tree
247
248 //
249     Bool_t   fAccCut;          //Transport acceptance cut
250     Float_t  fAccMin;          //Minimum acceptance cut used during transport
251     Float_t  fAccMax;          //Minimum acceptance cut used during transport
252 //  
253
254 //  Stepping Parameters
255    Float_t fMaxStepGas;      // Maximum step size inside the chamber gas
256    Float_t fMaxStepAlu;      // Maximum step size inside the chamber aluminum
257    Float_t fMaxDestepGas;    // Maximum relative energy loss in gas
258    Float_t fMaxDestepAlu;    // Maximum relative energy loss in aluminum
259 //
260 // modifs perso
261 //  Parameters for reconstruction program
262    Double_t fSPxzCut;        // Pxz cut  (GeV/c) to begin the track finding
263    Double_t fSSigmaCut;      // Number of sig. delimiting the searching areas
264    Double_t fSXPrec;         // Chamber precision in X (cm) 
265    Double_t fSYPrec;         // Chamber precision in Y (cm)
266
267    Text_t *fFileName;
268    
269  protected:
270
271    ClassDef(AliMUON,1)  //Hits manager for set:MUON
272 };
273 //___________________________________________
274
275 class AliMUONRawCluster : public TObject {
276 public:
277
278    Int_t     fTracks[3];      //labels of overlapped tracks
279    Int_t       fQ  ;          // Q of cluster (in ADC counts)     
280    Float_t     fX  ;          // X of cluster
281    Float_t     fY  ;          // Y of cluster
282    Int_t       fPeakSignal;
283    Int_t       fIndexMap[50];  //indeces of digits
284    Int_t       fOffsetMap[50]; //Emmanuel special
285    Float_t     fContMap[50];   //Contribution from digit
286    Int_t       fPhysicsMap[50];
287    Int_t       fMultiplicity;  //cluster multiplicity
288    Int_t       fNcluster[2];
289    Int_t       fClusterType;   
290  public:
291    AliMUONRawCluster() {
292        fTracks[0]=fTracks[1]=fTracks[2]=-1; 
293        fQ=0; fX=fY=0; fMultiplicity=0;
294        for (int k=0;k<50;k++) {
295            fIndexMap[k]=-1;
296            fOffsetMap[k]=0;
297            fContMap[k]=0;
298            fPhysicsMap[k]=-1;
299        }
300        fNcluster[0]=fNcluster[1]=-1;
301    }
302    virtual ~AliMUONRawCluster() {}
303
304    Float_t GetRadius() {return TMath::Sqrt(fX*fX+fY*fY);}
305
306    Bool_t IsSortable() const {return kTRUE;}
307    Int_t  Compare(TObject *obj);
308    Int_t PhysicsContribution();
309    static Int_t BinarySearch(Float_t r, TArrayF, Int_t from, Int_t upto);
310    static void  SortMin(Int_t *,Float_t *,Float_t *,Float_t *,Float_t *,Int_t);
311  
312    ClassDef(AliMUONRawCluster,1)  //Cluster object for set:MUON
313 };
314
315 //___________________________________________
316 class AliMUONcorrelation : public TObject {
317 public:
318
319   // correlation starts from the 1-st cathode  
320   // last number in arrays corresponds to cluster on 1-st cathode
321
322    Int_t       fCorrelIndex[4];  // entry number in TreeR for the associated 
323                                  // cluster candidates on the 2-nd cathode
324    Float_t     fX[4]  ;          // X of clusters on the 2-nd cathode  
325    Float_t     fY[4]  ;          // Y of clusters
326
327 public:
328    AliMUONcorrelation() {
329        fCorrelIndex[0]=fCorrelIndex[1]=fCorrelIndex[2]=fCorrelIndex[3]=0;
330        fX[0]=fX[1]=fX[2]=fX[3]=0; fY[0]=fY[1]=fY[2]=fY[3]=0; 
331    }
332    AliMUONcorrelation(Int_t *idx, Float_t *x, Float_t *y);
333    virtual ~AliMUONcorrelation() {}
334    ClassDef(AliMUONcorrelation,1)  //Cathode correlation object for set:MUON
335 };
336
337 #endif
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352