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