Major upgrade of AliRoot code
[u/mrichter/AliRoot.git] / TPC / AliTPC.h
CommitLineData
73042f01 1#ifndef ALITPC_H
2#define ALITPC_H
3da30618 3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id$ */
7
fe4da5cc 8////////////////////////////////////////////////
9// Manager class for TPC //
10////////////////////////////////////////////////
fe4da5cc 11#include "AliDetector.h"
12#include "AliHit.h"
13#include "AliDigit.h"
fe4da5cc 14
73042f01 15
2b06d5c3 16class TMatrix;
17class TTree;
cc80f89e 18
2b06d5c3 19class TFile;
20class AliTPCParam;
cc80f89e 21class AliTPCDigitsArray;
22class AliTPCClustersArray;
39c8eb58 23class AliTPCTrackHits; // M.I.
fe4da5cc 24
25class AliTPC : public AliDetector {
26protected:
27 Int_t fSens; // ISENS
28 Int_t fSecAL; // Upper sector selector
29 Int_t fSecAU; // Lower sector selector
30 Int_t fSecLows[6]; // List of lower sectors selected
31 Int_t fSecUps[12]; // List of upper sectors selected
32 Int_t fNsectors; // Number of sectors in TPC
cc80f89e 33 //MI changes
34 AliTPCDigitsArray * fDigitsArray; //detector digit object
35 AliTPCClustersArray * fClustersArray; //detector cluster object
39c8eb58 36 AliTPCParam *fTPCParam; // pointer to TPC parameters
37 AliTPCTrackHits *fTrackHits; //hits for given track M.I.
38 Int_t fHitType; // if fNewHit = 1 old data structure if 2 new hits
39 // 3 both types
4b0fdcad 40
41 //MK changes
42
43 Float_t fSide; // selects left(-1), right(+1), or both(0) sides of the TPC
1283eee5 44 Int_t fNoComp; // number of a drift gas components
45 Int_t fMixtComp[3]; // drift gas components
46 Float_t fMixtProp[3]; // mixture proportions
47
fe4da5cc 48public:
49 AliTPC();
50 AliTPC(const char *name, const char *title);
51 virtual ~AliTPC();
73042f01 52 virtual void AddHit(Int_t a1, Int_t *a2, Float_t *a3);
fe4da5cc 53 Int_t DistancetoPrimitive(Int_t px, Int_t py);
54 virtual void BuildGeometry();
55 virtual void CreateGeometry() {}
56 virtual void CreateMaterials();
73042f01 57 virtual void Hits2Clusters(TFile *of);
cc80f89e 58 virtual void Hits2ExactClustersSector(Int_t isec); // MI change calculate "exact" cluster position
8c555625 59
60 virtual void Hits2Digits(); //MI change
61 virtual void Hits2DigitsSector(Int_t isec); //MI change
fe4da5cc 62 virtual void Init();
63 virtual Int_t IsVersion() const =0;
73042f01 64 virtual void Digits2Clusters(TFile *of);
65 virtual void Clusters2Tracks(TFile *of);
cc80f89e 66
fe4da5cc 67 Int_t GetNsectors() {return fNsectors;}
68 virtual void MakeBranch(Option_t *opt=" ");
69 virtual void ResetDigits();
70 virtual void SetSecAL(Int_t sec);
71 virtual void SetSecAU(Int_t sec);
72 virtual void SetSecLows(Int_t s1,Int_t s2,Int_t s3,Int_t s4,Int_t s5, Int_t s6);
73 virtual void SetSecUps (Int_t s1,Int_t s2,Int_t s3,Int_t s4,Int_t s5, Int_t s6,
74 Int_t s7,Int_t s8,Int_t s9,Int_t s10, Int_t s11, Int_t s12);
75 virtual void SetSens(Int_t sens);
cc80f89e 76
4b0fdcad 77
78 virtual void SetSide(Float_t side);
1283eee5 79 virtual void SetGasMixt(Int_t nc,Int_t c1,Int_t c2,Int_t c3,Float_t p1,
80 Float_t p2,Float_t p3);
81
fe4da5cc 82 virtual void StepManager()=0;
8c555625 83 virtual void DrawDetector() {}
cc80f89e 84 AliTPCDigitsArray* GetDigitsArray() {return fDigitsArray;} //MI change
85 AliTPCClustersArray* GetClustersArray(){return fClustersArray;} //MI change
86 AliTPCParam *GetParam(){return fTPCParam;} // M.K, M.I changes
87 void SetParam(AliTPCParam *param){fTPCParam=param;} // M.K, M.I changes
88 void SetDigitsArray(AliTPCDigitsArray* param) {fDigitsArray=param;} //MI change
89 void SetClustersArray(AliTPCClustersArray *clusters) {fClustersArray = clusters;} //MI change
39c8eb58 90
91// additional function neccesary for the new hits
92 virtual void MakeBranch2(Option_t *opt=" "); //
93 virtual void SetTreeAddress();
94 virtual void SetTreeAddress2();
95 virtual void AddHit2(Int_t a1, Int_t *a2, Float_t *a3); //
96 virtual void ResetHits();
97 virtual void ResetHits2();
98 virtual AliHit* FirstHit(Int_t track);
99 virtual AliHit* NextHit();
100 virtual AliHit* FirstHit2(Int_t track);
101 virtual AliHit* NextHit2();
102
103 virtual void LoadPoints(Int_t dummy);
104 virtual void LoadPoints2(Int_t dummy);
105 virtual void LoadPoints3(Int_t dumy);
106 virtual void FinishPrimary();
107 virtual void RemapTrackHitIDs(Int_t *map);
108 virtual void FindTrackHitsIntersection(TClonesArray * arr);
109 //fill clones array with intersection of current point with the
110 //middle of the row
111 void SetHitType(Int_t type){fHitType =type;} //set type of hit container
112
113
fe4da5cc 114private:
115 //
8c555625 116 void DigitizeRow(Int_t irow,Int_t isec,TObjArray **rowTriplet);
cc80f89e 117 Float_t GetSignal(TObjArray *p1, Int_t ntr, TMatrix *m1, TMatrix *m2,
8c555625 118 Int_t *IndexRange);
119 void GetList (Float_t label,Int_t np,TMatrix *m,Int_t *IndexRange,
120 Float_t **pList);
121 void MakeSector(Int_t isec,Int_t nrows,TTree *TH,Stat_t ntracks,TObjArray **row);
cc80f89e 122 void TransportElectron(Float_t *xyz, Int_t *index);
123 Int_t fCurrentIndex[4];// index[0] indicates coordinate system,
124 // index[1] sector number,
125 // index[2] pad row number
126 // index[3] pad row number for which signal is calculated
fe4da5cc 127
128 ClassDef(AliTPC,2) // Time Projection Chamber class
129};
130
fe4da5cc 131
132//_____________________________________________________________________________
133
134class AliTPCdigit : public AliDigit {
135public:
136 Int_t fSector; //array of volumes
137 Int_t fPadRow; //Row number
138 Int_t fPad ; //Pad number
139 Int_t fTime; //Time bucket
140 Int_t fSignal; //Signal amplitude
141
142public:
143 AliTPCdigit() {}
144 AliTPCdigit(Int_t *tracks, Int_t *digits);
145 virtual ~AliTPCdigit() {}
146
147 ClassDef(AliTPCdigit,1) // Time Projection Chamber digits
148};
149
150
151//_____________________________________________________________________________
152
153class AliTPChit : public AliHit {
154public:
155 Int_t fSector; //sector number
156 Int_t fPadRow; //Pad Row number
157 Float_t fQ ; //charge
158
159public:
160 AliTPChit() {}
161 AliTPChit(Int_t shunt, Int_t track, Int_t *vol, Float_t *hits);
162 virtual ~AliTPChit() {}
39c8eb58 163 void SetX(Float_t x){fX = x;}
164 void SetY(Float_t y){fY = y;}
165 void SetZ(Float_t z){fZ = z;}
fe4da5cc 166
167 ClassDef(AliTPChit,1) // Time Projection Chamber hits
168};
3c0f9266 169
73042f01 170#endif
fe4da5cc 171
fe4da5cc 172
3c0f9266 173
fe4da5cc 174
fe4da5cc 175
fe4da5cc 176
fe4da5cc 177
fe4da5cc 178
fe4da5cc 179