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