set magnetic field for tracks

[u/mrichter/AliRoot.git] / TRD / AliTRD.h

diff --git a/TRD/AliTRD.h b/TRD/AliTRD.h
index a7d926478c2818da674a266923b03bdb49857464..64abc4b32a2d15ea8bf409e1d733853c3a6de07e 100644 (file)
--- a/TRD/AliTRD.h
+++ b/TRD/AliTRD.h
@@ -8,14 +8,20 @@
 ////////////////////////////////////////////////
 //  Manager and hits classes for set: TRD     //
 ////////////////////////////////////////////////
- 
-#include "AliRun.h"
+
+#include <TLorentzVector.h>
 #include "AliDetector.h"
-#include "AliTRDhit.h" 
-#include "AliDigit.h"
+#include <TVirtualMC.h>
+
+class TFile;
+class TLorentzVector;
+
+class AliRun;
+class AliDigit;
 
-#include "AliTRDconst.h"
-#include "AliTRDgeometry.h"
+class AliTRDhit;
+class AliTRDsim;
+class AliTRDgeometry;
 
 //_____________________________________________________________________________
 class AliTRD : public AliDetector {
@@ -29,51 +35,68 @@ class AliTRD : public AliDetector {
 
           AliTRD    &operator=(const AliTRD &trd);
 
-  virtual void       AddHit(Int_t track, Int_t *det, Float_t *hits);
-  virtual void       AddDigit(Int_t *digits, Int_t *amp);    
-  virtual void       AddRecPoint(Float_t *pos, Int_t *digits
-                               , Int_t det, Float_t amp);
+  virtual void       AddHit(Int_t track, Int_t det, Float_t *hits, Int_t q, Bool_t inDrift); 
   virtual void       BuildGeometry();
   virtual void       Copy(TObject &trd);
   virtual void       CreateGeometry();
   virtual void       CreateMaterials();
-  virtual void       DrawModule();
-  Int_t              DistancetoPrimitive(Int_t px, Int_t py);
-  TObjArray         *RecPoints()           { return fRecPoints;   };
+  virtual void       DrawModule() const;
+  Int_t              DistancetoPrimitive(Int_t px, Int_t py) const;
+  virtual void       LoadPoints(Int_t track);    
   virtual void       Init();
   virtual Int_t      IsVersion() const = 0;
-  virtual void       MakeBranch(Option_t* option);     
-  virtual void       ResetRecPoints();
+  virtual void       MakeBranch(Option_t* option);
+  virtual void       ResetDigits();     
   virtual void       StepManager() = 0; 
   virtual void       SetTreeAddress();
 
+  virtual void       StepManagerErmilova()      = 0;
+  virtual void       StepManagerGeant()         = 0;
+  virtual void       StepManagerFixedStep()     = 0;
+  virtual void       SelectStepManager(Int_t t) = 0;
+  virtual void       SetStepSize(Double_t s)    = 0;
+
   virtual void       SetGasMix(Int_t imix = 0);
   virtual void       SetHits()             {};
-  virtual void       SetPHOShole()         { fGeometry->SetPHOShole(); };
-  virtual void       SetRICHhole()         { fGeometry->SetRICHhole(); };
+  virtual void       SetPHOShole();
+  virtual void       SetRICHhole();
+  virtual void       SetDrawTR(Int_t idraw = 1)     { fDrawTR      = idraw; };
+  virtual void       SetDisplayType(Int_t type = 0) { fDisplayType = type;  };
 
-  AliTRDgeometry    *GetGeometry()         { return fGeometry; };
+  AliTRDgeometry    *GetGeometry() const            { return fGeometry; };
 
   virtual void       SetSensChamber(Int_t ichamber)              = 0;
   virtual void       SetSensPlane(Int_t iplane)                  = 0;
   virtual void       SetSensSector(Int_t isector)                = 0;
   virtual void       SetSensSector(Int_t isector, Int_t nsector) = 0;
 
-  virtual Int_t      GetSensChamber()     = 0;
-  virtual Int_t      GetSensPlane()       = 0;
-  virtual Int_t      GetSensSector()      = 0;
-  virtual Int_t      GetSensSectorRange() = 0; 
+  virtual Int_t      GetSensChamber() const     = 0;
+  virtual Int_t      GetSensPlane() const       = 0;
+  virtual Int_t      GetSensSector() const      = 0;
+  virtual Int_t      GetSensSectorRange() const = 0; 
+ 
+  virtual void       Hits2Digits();
+  virtual void       Hits2SDigits();
+  virtual AliDigitizer* CreateDigitizer(AliRunDigitizer* manager) const; 
+  virtual void       SDigits2Digits();
+  virtual void       Digits2Raw();
+
+  virtual AliTRDsim *CreateTR()     = 0;
+  virtual AliTRDsim *GetTR() const  = 0;
 
  protected:
 
-  Int_t              fGasMix;            //  Gas mixture. 0: Xe/Isobutane 1: Xe/CO2
+  Int_t                fGasMix;             //  Gas mixture. 0: Xe/Isobutane 1: Xe/CO2
+
+  AliTRDgeometry      *fGeometry;           //  The TRD geometry
 
-  AliTRDgeometry    *fGeometry;          //  The TRD geometry
+  Float_t              fGasDensity;         //  The density of the drift gas
+  Float_t              fFoilDensity;        //  The density of the entrance window foil
 
-  TObjArray         *fRecPoints;         //  Array of reconstructed points
-  Int_t              fNRecPoints;        //! Number of reconstructed points
+  Int_t                fDrawTR;             //  Switches marking the TR photons in the display
+  Int_t                fDisplayType;        //  Display type (0: normal, 1: detailed) 
 
-  ClassDef(AliTRD,1)                     //  Transition Radiation Detector base class
+  ClassDef(AliTRD,7)                        //  Transition Radiation Detector base class
 
 };