Add track length to track references
[u/mrichter/AliRoot.git] / 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 "AliDetector.h"
13
14 class TFile;
15 class TLorentzVector;
16
17 class AliRun;
18 class AliDigit;
19
20 class AliTRDhit;
21 class AliTRDsim;
22 class AliTRDgeometry;
23 class AliTRDtrackHits;
24
25 //_____________________________________________________________________________
26 class AliTRD : public AliDetector {
27
28  public:
29
30   AliTRD();
31   AliTRD(const char *name, const char *title);
32   AliTRD(const AliTRD &trd);
33   virtual           ~AliTRD();
34
35           AliTRD    &operator=(const AliTRD &trd);
36
37   virtual void       AddHit(Int_t track, Int_t det, Float_t *hits, Int_t q, Bool_t inDrift); 
38   virtual void       AddCluster(Float_t *pos, Int_t det, Float_t amp, Int_t *tracks
39                               , Float_t *sig, Int_t iType);
40   virtual void       AddTrackReference(Int_t label, TLorentzVector p
41                                      , TLorentzVector x, Float_t length);
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, const char *file=0);     
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,5)                        //  Transition Radiation Detector base class
121
122 };
123
124 #endif