8eea8e2252b1a83dbc46075054add0410950fba3
[u/mrichter/AliRoot.git] / MUON / AliMUONTrackK.h
1 #ifndef ALIMUONTRACKK_H
2 #define ALIMUONTRACKK_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 // Revision of includes 07/05/2004
8
9 #include <TObject.h>
10
11 class TMatrixD;
12 class AliMUONEventReconstructor;
13 class TClonesArray;
14 class TObjArray;
15 class AliMUONSegment;
16 class AliMUON;
17 class AliMUONHitForRec;
18
19 class AliMUONTrackK : public TObject {
20
21  public:
22
23   AliMUONTrackK(); // Default constructor
24   virtual ~AliMUONTrackK(); // Destructor
25
26   //AliMUONTrackK(const AliMUONEventReconstructor *EventReconstructor, const AliMUONHitForRec *hitForRec); // Constructor
27   AliMUONTrackK(AliMUONEventReconstructor *EventReconstructor, TClonesArray *hitForRec); // Constructor
28   AliMUONTrackK(AliMUONSegment *segment); // Constructor from a segment
29
30   // Pointer to hits on track
31   TObjArray* GetHitOnTrack(void) const {return fTrackHitsPtr;} // ptr. to hits on track
32   Int_t GetNTrackHits(void) const {return fNTrackHits;} // hits on track
33   Double_t GetTrackQuality(void) const {return fChi2;} // track quality
34   TMatrixD* GetTrackParameters(void) const {return fTrackPar;} // track parameters
35   Double_t GetZ(void) const {return fPosition;} // Z-coordinate of track 
36   TMatrixD* GetCovariance(void) const {return fCovariance;} // covariance matrix
37   Int_t GetTrackDir(void) const {return fTrackDir;} // get track propagation direction
38   void SetTrackDir(Int_t iDir) {fTrackDir = iDir;} // set track propagation direction  
39   Bool_t GetBPFlag(void) const {return fBPFlag;} // get backpropagation flag
40   void SetBPFlag(Bool_t BPFlag) {fBPFlag = BPFlag;} // set backpropagation flag
41   Int_t GetRecover(void) const {return fRecover;} // return recover flag 
42   void SetRecover(Int_t iRecover) {fRecover = iRecover;} // set recover flag
43   AliMUONSegment* GetStartSegment(void) const {return fStartSegment;} // return seed segment
44   Bool_t KalmanFilter(Int_t ichamBeg, Int_t ichamEnd, Bool_t Back, Double_t zDipole1, Double_t zDipole2); // Kalman filter
45   void StartBack(void); // start backpropagator
46   void SetTrackQuality(Int_t iChi2); // compute track quality or Chi2
47   Bool_t KeepTrack(AliMUONTrackK* track0) const; // keep or discard track 
48   void Kill(void); // kill track candidate
49   void Branson(void); // Branson correction
50   void GoToZ(Double_t zEnd); // propagate track to given Z
51   void GoToVertex(void); // propagate track to the vertex
52
53   // What is necessary for sorting TClonesArray's
54   Bool_t IsSortable() const { return kTRUE; }
55   Int_t Compare(const TObject* trackK) const; // "Compare" function for sorting
56
57
58
59  protected:
60
61   AliMUONTrackK (const AliMUONTrackK& source); // copy constructor
62   AliMUONTrackK& operator=(const AliMUONTrackK& source); // assignment operator
63
64  private:
65  
66   static Int_t fgNOfPoints; // number of points in event
67   static AliMUON *fgMUON; // pointer to MUON module  
68   static AliMUONEventReconstructor *fgEventReconstructor; // pointer to event reconstructor
69   static TClonesArray *fgHitForRec; // pointer to hits
70
71   AliMUONSegment *fStartSegment; // seed segment  
72   Double_t fPosition; // Z-coordinate of track
73   Double_t fPositionNew; //! Z-coordinate of track
74   Double_t fChi2; // Chi2 of track
75   TObjArray *fTrackHitsPtr; // pointer to hits on track
76   Int_t fNTrackHits; // number of points on track
77   Int_t fTrackDir; // track direction (+(-) towards high (low) z)
78   Bool_t fBPFlag; // backpropagation flag (TRUE if backpropagation)
79   Int_t fRecover; // recover flag (!=0 if recovery procedure was applied)
80   AliMUONHitForRec *fSkipHit; // hit to skip during recovery procedure
81
82   TMatrixD *fTrackPar; // track parameters
83   TMatrixD *fTrackParNew; //! track parameters
84   TMatrixD *fCovariance; // covariance matrix
85   TMatrixD *fWeight; //! weight matrix (inverse of covariance)
86
87   // Functions
88
89   void EvalCovariance(Double_t dZ);
90   void ParPropagation(Double_t zEnd);
91   void WeightPropagation(Double_t zEnd);
92   void MSThin(Int_t sign);
93   void MSLine(Double_t dZ, Double_t X0);
94   Bool_t FindPoint(Int_t ichamb, Double_t zEnd, Int_t currIndx, Int_t iFB, AliMUONHitForRec *&hitAdd);
95   void TryPoint(TMatrixD &point, const TMatrixD &pointWeight, TMatrixD &trackParTmp, Double_t &dChi2);
96   void SetGeantParam(Double_t *VGeant3, Int_t iFB);
97   void GetFromGeantParam(Double_t *VGeant3, Int_t iFB);
98   void Recover(void);
99
100   private:
101    // Some constants
102    static const Int_t fgkSize; // number of track parameters
103    static const Int_t fgkNSigma; //4; // acceptance window width in sigmas
104    static const Int_t fgkTriesMax; // max number of attempts to find exact position during tracking
105    static const Double_t fgkEpsilon; // tracking precision (cm)
106
107   ClassDef(AliMUONTrackK,0) // Kalman track in MUON arm of ALICE
108     };
109 #endif