TRD/AliTRD.h

   1 #ifndef ALITRD_H
   2 #define ALITRD_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 and hits classes for set: TRD     //
  10 ////////////////////////////////////////////////
  11
  12 #include <TLorentzVector.h>
  13 #include "AliDetector.h"
  14 #include <TVirtualMC.h>
  15
  16 class TFile;
  17 class TLorentzVector;
  18
  19 class AliRun;
  20 class AliDigit;
  21
  22 class AliTRDhit;
  23 class AliTRDsim;
  24 class AliTRDgeometry;
  25 class AliTRDtrackHits;
  26
  27 //_____________________________________________________________________________
  28 class AliTRD : public AliDetector {
  29
  30  public:
  31
  32   AliTRD();
  33   AliTRD(const char *name, const char *title);
  34   AliTRD(const AliTRD &trd);
  35   virtual           ~AliTRD();
  36
  37           AliTRD    &operator=(const AliTRD &trd);
  38
  39   virtual void       AddHit(Int_t track, Int_t det, Float_t *hits, Int_t q, Bool_t inDrift);
  40   virtual void       AddCluster(Float_t *pos, Int_t det, Float_t amp, Int_t *tracks
  41                               , Float_t *sig, Int_t iType);
  42   virtual void       BuildGeometry();
  43   virtual void       Copy(TObject &trd);
  44   virtual void       CreateGeometry();
  45   virtual void       CreateMaterials();
  46   virtual void       DrawModule() const;
  47   Int_t              DistancetoPrimitive(Int_t px, Int_t py) const;
  48   virtual void       LoadPoints(Int_t track);
  49   TObjArray         *RecPoints() const              { return fRecPoints;   };
  50   virtual void       Init();
  51   virtual Int_t      IsVersion() const = 0;
  52   virtual void       MakeBranch(Option_t* option);
  53   virtual void       ResetDigits();
  54   virtual void       ResetRecPoints();
  55   virtual void       StepManager() = 0;
  56   virtual void       SetTreeAddress();
  57
  58   virtual void       SetGasMix(Int_t imix = 0);
  59   virtual void       SetHits()             {};
  60   virtual void       SetPHOShole();
  61   virtual void       SetRICHhole();
  62   virtual void       SetDrawTR(Int_t idraw = 1)     { fDrawTR      = idraw; };
  63   virtual void       SetDisplayType(Int_t type = 0) { fDisplayType = type;  };
  64
  65   AliTRDgeometry    *GetGeometry() const            { return fGeometry; };
  66
  67   virtual void       SetSensChamber(Int_t ichamber)              = 0;
  68   virtual void       SetSensPlane(Int_t iplane)                  = 0;
  69   virtual void       SetSensSector(Int_t isector)                = 0;
  70   virtual void       SetSensSector(Int_t isector, Int_t nsector) = 0;
  71
  72   virtual void       SetHitTypeStandard()           { fHitType = 1; };
  73   virtual void       SetHitTypeCompressed()         { fHitType = 2; };
  74   virtual void       SetHitTypeBoth()               { fHitType = 3; };
  75
  76   virtual Int_t      GetSensChamber() const     = 0;
  77   virtual Int_t      GetSensPlane() const       = 0;
  78   virtual Int_t      GetSensSector() const      = 0;
  79   virtual Int_t      GetSensSectorRange() const = 0;
  80
  81   virtual void       Hits2Digits();
  82   virtual void       Hits2SDigits();
  83   virtual void       SDigits2Digits();
  84
  85   virtual AliTRDsim *CreateTR()     = 0;
  86   virtual AliTRDsim *GetTR() const  = 0;
  87
  88   virtual void       MakeBranch2(Option_t *opt=" ", const char *file=0);
  89   virtual void       SetTreeAddress2();
  90   virtual void       AddHit2(Int_t track, Int_t det, Float_t *hits, Int_t q, Bool_t inDrift);
  91   virtual void       ResetHits();
  92
  93   virtual AliHit*    FirstHit(Int_t track);
  94   virtual AliHit*    NextHit();
  95   virtual AliHit*    FirstHit2(Int_t track);
  96   virtual AliHit*    NextHit2();
  97
  98   virtual void       FinishPrimary();
  99   virtual void       RemapTrackHitIDs(Int_t *map);
 100
 101  protected:
 102
 103   Int_t                fGasMix;             //  Gas mixture. 0: Xe/Isobutane 1: Xe/CO2
 104
 105   AliTRDgeometry      *fGeometry;           //  The TRD geometry
 106
 107   TObjArray           *fRecPoints;          //  Array of reconstructed points / cluster
 108   Int_t                fNRecPoints;         //! Number of reconstructed points / cluster
 109
 110   Float_t              fGasDensity;         //  The density of the drift gas
 111   Float_t              fFoilDensity;        //  The density of the entrance window foil
 112
 113   Int_t                fDrawTR;             //  Switches marking the TR photons in the display
 114   Int_t                fDisplayType;        //  Display type (0: normal, 1: detailed)
 115
 116
 117   AliTRDtrackHits     *fTrackHits;          //! Compressed hits for a given track
 118   Int_t                fHitType;            //  Switch for the different hit data structures
 119
 120   ClassDef(AliTRD,6)                        //  Transition Radiation Detector base class
 121
 122 };
 123
 124 #endif