f7a64f6ffc5a03675204fa9c08fff155fdc2feff
[u/mrichter/AliRoot.git] / TPC / AliTPC.h
1 #ifndef ALITPC_H
2 #define ALITPC_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 ////////////////////////////////////////////////
9 //  Manager class for TPC                     //
10 ////////////////////////////////////////////////
11 #include "AliDetector.h"
12 #include "AliHit.h" 
13 #include "AliDigit.h" 
14 #include <TVector.h>
15
16
17 class TMatrix;
18 class AliTPCFastMatrix;  //MI change
19 class AliTPCFastVector;  //MI change
20
21 class TTree;
22
23 class TFile;
24 class AliTPCParam;
25 class AliTPCDigitsArray;
26 class AliTPCTrackHitsV2; // M.I.
27 class AliTPCTrackHits; // M.I.  -MI4 old hits - to be removed later
28 //class TLorentzVector;
29 class AliTPCLoader;
30
31
32 class AliTPC : public AliDetector {
33
34 public:
35   AliTPC(); 
36   AliTPC(const char *name, const char *title);
37   AliTPC(const AliTPC& t);
38   
39   virtual AliLoader* MakeLoader(const char* topfoldername);
40   
41   virtual      ~AliTPC();
42   virtual void  AddHit(Int_t a1, Int_t *a2, Float_t *a3);
43   Int_t         DistancetoPrimitive(Int_t px, Int_t py) const;
44   virtual void  BuildGeometry();
45   virtual void  CreateGeometry() {}
46   virtual void  CreateMaterials();
47   
48   virtual AliDigitizer* CreateDigitizer(AliRunDigitizer* manager) const;
49   virtual void  SDigits2Digits(){;} //MI change -cycling to production
50   virtual void  SDigits2Digits2(Int_t eventnumber=0);
51
52   virtual void  Hits2SDigits(); // MI - cycling around
53   virtual void  Hits2SDigits2(Int_t eventnumber=0);
54
55   virtual void  Hits2Digits();
56   virtual void  Hits2Digits(Int_t eventnumber);   //MI change
57   virtual void  Hits2DigitsSector(Int_t isec);  //MI change
58   virtual void  Init();
59   virtual Int_t IsVersion() const =0;
60
61   virtual void  Digits2Raw();
62
63   Int_t         GetNsectors() const  {return fNsectors;}
64   virtual void  MakeBranch(Option_t *opt=" ");
65   virtual void  ResetDigits();
66   virtual void  SetSecAL(Int_t sec);
67   virtual void  SetSecAU(Int_t sec);
68   virtual void  SetSecLows(Int_t s1,Int_t s2,Int_t s3,Int_t s4,Int_t s5, Int_t s6);
69   virtual void  SetSecUps (Int_t s1,Int_t s2,Int_t s3,Int_t s4,Int_t s5, Int_t s6,
70           Int_t s7,Int_t s8,Int_t s9,Int_t s10, Int_t s11, Int_t s12);
71   virtual void  SetSens(Int_t sens);
72
73
74   virtual void  SetSide(Float_t side);
75   virtual void  SetGasMixt(Int_t nc,Int_t c1,Int_t c2,Int_t c3,Float_t p1,
76                            Float_t p2,Float_t p3); 
77
78   virtual void  StepManager()=0;
79   virtual void  DrawDetector() {}
80   AliTPCDigitsArray*  GetDigitsArray() {return fDigitsArray;} //MI change
81   AliTPCParam *GetParam(){return fTPCParam;} // M.K, M.I changes
82   void SetParam(AliTPCParam *param){fTPCParam=param;} // M.K, M.I changes
83   void SetDigitsArray(AliTPCDigitsArray* param) {fDigitsArray=param;}  //MI change
84
85 // additional function neccesary for the new hits 
86    virtual void MakeBranch2(Option_t *opt=" ", const char *file=0);  //
87    virtual void SetTreeAddress();
88    virtual void SetTreeAddress2();
89    virtual void AddHit2(Int_t a1,  Int_t *a2, Float_t *a3);  //
90    virtual void ResetHits();
91    virtual void ResetHits2();     
92    virtual AliHit* FirstHit(Int_t track);
93    virtual AliHit* NextHit();
94    virtual AliHit* FirstHit2(Int_t track);
95    virtual AliHit* NextHit2();
96    virtual void LoadPoints(Int_t dummy);
97    virtual void LoadPoints2(Int_t dummy);
98    virtual void LoadPoints3(Int_t dumy);
99    virtual void FinishPrimary();
100    virtual void RemapTrackHitIDs(Int_t *map);
101    void SetHitType(Int_t type){fHitType =type;} //set type of hit container
102    void SetDigitsSwitch(Int_t sw){fDigitsSwitch = sw;}
103    void SetDefSwitch(Int_t def){fDefaults = def;}
104    Float_t GetNoise();  //get Current noise  
105    void    GenerNoise(Int_t tablasize);  // make noise table
106    Bool_t  IsSectorActive(Int_t sec) const;    // check if the sector is active
107    void    SetActiveSectors(Int_t * sectors, Int_t n);  //set active sectors
108    Int_t GetHitType() const {return fHitType;}
109    void    SetActiveSectors(Int_t flag=0); //loop over al hits and set active only hitted sectors
110    Bool_t  TrackInVolume(Int_t id,Int_t track);  //return true if current track is in volume
111
112 // static functions
113    static AliTPCParam* LoadTPCParam(TFile *file); 
114 protected:
115    Int_t          fDefaults; // defaults switch
116   Int_t          fSens;             // ISENS
117   Int_t          fSecAL;            // Upper sector selector
118   Int_t          fSecAU;            // Lower sector selector
119   Int_t          fSecLows[6];       // List of lower sectors selected
120   Int_t          fSecUps[12];       // List of upper sectors selected
121   Int_t          fNsectors;         // Number of sectors in TPC
122   //MI changes
123   AliTPCDigitsArray * fDigitsArray;              //!detector digit object  
124   AliTPCParam *fTPCParam;           // pointer to TPC parameters 
125   AliTPCTrackHitsV2 *fTrackHits;      //!hits for given track M.I.
126   AliTPCTrackHits *fTrackHitsOld;      //!hits for given track M.I. MIold -
127
128   Int_t  fHitType; // if fNewHit = 1 old data structure if 2 new hits  if 4  old MI stucture
129   //  3 both types 
130   Int_t fDigitsSwitch; // digits type, 0->normal, 1->summable
131
132   //MK changes
133
134   Float_t        fSide;  // selects left(-1), right(+1), or both(0) sides of the TPC
135   Int_t          fNoComp; // number of a drift gas components
136   Int_t          fMixtComp[3]; // drift gas components
137   Float_t        fMixtProp[3]; // mixture proportions 
138    
139  private:
140   //
141   void SetDefaults();
142   void DigitizeRow(Int_t irow,Int_t isec,TObjArray **rowTriplet);
143   Float_t GetSignal(TObjArray *p1, Int_t ntr, AliTPCFastMatrix *m1, 
144                    AliTPCFastMatrix *m2,Int_t *IndexRange);
145   void GetList (Float_t label,Int_t np,AliTPCFastMatrix *m,Int_t *IndexRange,
146                 Float_t **pList);
147   void MakeSector(Int_t isec,Int_t nrows,TTree *TH,Stat_t ntracks,TObjArray **row);
148   void TransportElectron(Float_t *xyz, Int_t *index);
149   Int_t fCurrentIndex[4];// index[0] indicates coordinate system, 
150                          // index[1] sector number, 
151                          // index[2] pad row number  
152                          // index[3] pad row number for which signal is calculated
153   Int_t      fNoiseDepth;  //!noise table
154   Float_t *  fNoiseTable;  //![fNoiseDepth] table with noise
155   Int_t      fCurrentNoise; //!index of the noise in  the noise table 
156   Bool_t*    fActiveSectors; //!bool indicating which sectors are active
157
158   ClassDef(AliTPC,10)  // Time Projection Chamber class
159 };
160
161
162 //_____________________________________________________________________________
163  
164 class AliTPChit : public AliHit {
165 public:
166    Int_t     fSector;     //sector number
167    Int_t     fPadRow;     //Pad Row number
168    Float_t   fQ ;         //charge
169  
170 public:
171    AliTPChit() {}
172    AliTPChit(Int_t shunt, Int_t track, Int_t *vol, Float_t *hits);
173    virtual ~AliTPChit() {}
174    void SetX(Float_t x){fX = x;}
175    void SetY(Float_t y){fY = y;}
176    void SetZ(Float_t z){fZ = z;}
177  
178    ClassDef(AliTPChit,1)  // Time Projection Chamber hits
179 };
180
181 //_____________________________________________________________________________
182 class AliTPCFastVector : public TVector {
183 public :
184   AliTPCFastVector(Int_t size):TVector(size){};
185   virtual ~AliTPCFastVector(){;}
186   Float_t & UncheckedAt(Int_t index) const  {return  fElements[index];} //fast acces  
187   
188 };
189
190 #endif
191
192
193
194
195
196
197
198
199