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
  15
  16 class TMatrix;
  17 class TTree;
  18
  19 class TFile;
  20 class AliTPCParam;
  21 class AliTPCDigitsArray;
  22 class AliTPCClustersArray;
  23 class AliTPCTrackHits; // M.I.
  24
  25 class AliTPC : public AliDetector {
  26 protected:
  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
  33   //MI changes
  34   AliTPCDigitsArray * fDigitsArray;              //detector digit object
  35   AliTPCClustersArray * fClustersArray; //detector cluster object
  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
  40
  41   //MK changes
  42
  43   Float_t        fSide;  // selects left(-1), right(+1), or both(0) sides of the TPC
  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
  48 public:
  49   AliTPC();
  50   AliTPC(const char *name, const char *title);
  51   virtual      ~AliTPC();
  52   virtual void  AddHit(Int_t a1, Int_t *a2, Float_t *a3);
  53   Int_t         DistancetoPrimitive(Int_t px, Int_t py);
  54   virtual void  BuildGeometry();
  55   virtual void  CreateGeometry() {}
  56   virtual void  CreateMaterials();
  57   virtual void  Hits2Clusters(TFile *of);
  58   virtual void  Hits2ExactClustersSector(Int_t isec); // MI change calculate "exact" cluster position
  59
  60   virtual void  Hits2Digits();   //MI change
  61   virtual void Hits2DigitsSector(Int_t isec);  //MI change
  62   virtual void  Init();
  63   virtual Int_t IsVersion() const =0;
  64   virtual void  Digits2Clusters(TFile *of);
  65   virtual void  Clusters2Tracks(TFile *of);
  66
  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);
  76
  77
  78   virtual void  SetSide(Float_t side);
  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
  82   virtual void  StepManager()=0;
  83   virtual void  DrawDetector() {}
  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
  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
 114 private:
 115   //
 116   void DigitizeRow(Int_t irow,Int_t isec,TObjArray **rowTriplet);
 117   Float_t GetSignal(TObjArray *p1, Int_t ntr, TMatrix *m1, TMatrix *m2,
 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);
 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
 127
 128   ClassDef(AliTPC,2)  // Time Projection Chamber class
 129 };
 130
 131
 132 //_____________________________________________________________________________
 133
 134 class AliTPCdigit : public AliDigit {
 135 public:
 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
 142 public:
 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
 153 class AliTPChit : public AliHit {
 154 public:
 155    Int_t     fSector;     //sector number
 156    Int_t     fPadRow;     //Pad Row number
 157    Float_t   fQ ;         //charge
 158
 159 public:
 160    AliTPChit() {}
 161    AliTPChit(Int_t shunt, Int_t track, Int_t *vol, Float_t *hits);
 162    virtual ~AliTPChit() {}
 163    void SetX(Float_t x){fX = x;}
 164    void SetY(Float_t y){fY = y;}
 165    void SetZ(Float_t z){fZ = z;}
 166
 167    ClassDef(AliTPChit,1)  // Time Projection Chamber hits
 168 };
 169
 170 #endif
 171
 172
 173
 174
 175
 176
 177
 178
 179