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 /// \ingroup rec
  10 /// \class AliMUONTrack
  11 /// \brief Kalman track in MUON arm of ALICE
  12
  13 class TArrayD;
  14 class TClonesArray;
  15 class TObjArray;
  16 #include <TMatrixDfwd.h>
  17 #include <TObject.h>
  18
  19 class AliMUONEventRecoCombi;
  20 class AliMUONHitForRec;
  21 class AliMUONSegment;
  22 class AliMUONTrackReconstructor;
  23 #include "AliMUONTrack.h"
  24
  25 class AliMUONTrackK : public AliMUONTrack {
  26
  27  public:
  28
  29   AliMUONTrackK(); // Default constructor
  30   virtual ~AliMUONTrackK(); // Destructor
  31
  32   AliMUONTrackK(AliMUONTrackReconstructor *TrackReconstructor, TClonesArray *hitForRec); // Constructor
  33   AliMUONTrackK(AliMUONSegment *segment); // Constructor from a segment
  34
  35   // Pointer to hits on track
  36   TObjArray* GetHitOnTrack(void) const {return fTrackHitsPtr;} // ptr. to hits on track
  37   Int_t GetNTrackHits(void) const {return fNTrackHits;} // hits on track
  38   Double_t GetTrackQuality(void) const {return fChi2;} // track quality
  39   TMatrixD* GetTrackParameters(void) const {return fTrackPar;} // track parameters
  40   Double_t GetZ(void) const {return fPosition;} // Z-coordinate of track
  41   TMatrixD* GetCovariance(void) const {return fCovariance;} // covariance matrix
  42   Int_t GetTrackDir(void) const {return fTrackDir;} // get track propagation direction
  43   void SetTrackDir(Int_t iDir) {fTrackDir = iDir;} // set track propagation direction
  44   Bool_t GetBPFlag(void) const {return fBPFlag;} // get backpropagation flag
  45   void SetBPFlag(Bool_t BPFlag) {fBPFlag = BPFlag;} // set backpropagation flag
  46   Int_t GetRecover(void) const {return fRecover;} // return recover flag
  47   void SetRecover(Int_t iRecover) {fRecover = iRecover;} // set recover flag
  48   AliMUONSegment* GetStartSegment(void) const {return fStartSegment;} // return seed segment
  49   Bool_t KalmanFilter(Int_t ichamBeg, Int_t ichamEnd, Bool_t Back, Double_t zDipole1, Double_t zDipole2); // Kalman filter
  50   void StartBack(void); // start backpropagator
  51   void SetTrackQuality(Int_t iChi2); // compute track quality or Chi2
  52   Bool_t KeepTrack(AliMUONTrackK* track0) const; // keep or discard track
  53   void Kill(void); // kill track candidate
  54   void Branson(void); // Branson correction
  55   void GoToZ(Double_t zEnd); // propagate track to given Z
  56   void GoToVertex(Int_t iflag); // propagate track to the vertex
  57   Bool_t Smooth(void); // apply smoother
  58   Double_t GetChi2PerPoint(Int_t iPoint) const; // return Chi2 at point
  59   void Print(FILE *lun) const; // print track information
  60   void Print(const char* /*opt*/) const {return;} // print track information
  61   AliMUONHitForRec* GetHitLastOk(void); // get hit before the skipped one
  62   Int_t GetStation0(void); // return seed station number
  63   Int_t DebugLevel(void) const {return fgDebug;} // return debug level
  64   void SetDebugLevel(Int_t iDebug) {fgDebug = iDebug;} // set debug level
  65   void FillMUONTrack(void); // set track parameters as for AliMUONTrack
  66   void SetTrackParam(AliMUONTrackParam *trackParam, TMatrixD *par, Double_t z); // fill AliMUONTrackParam object
  67
  68   // What is necessary for sorting TClonesArray's
  69   Bool_t IsSortable() const { return kTRUE; }
  70   Int_t Compare(const TObject* trackK) const; // "Compare" function for sorting
  71
  72
  73
  74  protected:
  75
  76   AliMUONTrackK (const AliMUONTrackK& source); // copy constructor
  77   AliMUONTrackK& operator=(const AliMUONTrackK& source); // assignment operator
  78
  79  private:
  80
  81   static Int_t fgDebug; // debug level
  82   static Int_t fgNOfPoints; // number of points in event
  83   //static AliMUON *fgMUON; // pointer to MUON module
  84   static AliMUONTrackReconstructor *fgTrackReconstructor; // pointer to event reconstructor
  85   static TClonesArray *fgHitForRec; // pointer to hits
  86   static AliMUONEventRecoCombi *fgCombi; // pointer to combined cluster/track finder
  87
  88   AliMUONSegment *fStartSegment; // seed segment
  89   Double_t fPosition; // Z-coordinate of track
  90   Double_t fPositionNew; //! Z-coordinate of track
  91   Double_t fChi2; // Chi2 of track
  92   TObjArray *fTrackHitsPtr; // pointer to hits on track
  93   Int_t fNTrackHits; // number of points on track
  94   Int_t fTrackDir; // track direction (+(-) towards high (low) z)
  95   Bool_t fBPFlag; // backpropagation flag (TRUE if backpropagation)
  96   Int_t fRecover; // recover flag (!=0 if recovery procedure was applied)
  97   AliMUONHitForRec *fSkipHit; // hit to skip during recovery procedure
  98
  99   TMatrixD *fTrackPar; // track parameters
 100   TMatrixD *fTrackParNew; //! track parameters
 101   TMatrixD *fCovariance; // covariance matrix
 102   TMatrixD *fWeight; //! weight matrix (inverse of covariance)
 103
 104   // For smoother
 105   TObjArray *fParExtrap; //! extrapolated track parameters
 106   TObjArray *fParFilter; //! filtered track parameters
 107   TObjArray *fParSmooth; //! smoothed track parameters
 108
 109   TObjArray *fCovExtrap; //! extrapolated covariance matrices
 110   TObjArray *fCovFilter; //! filtered covariance matrices
 111
 112   TObjArray *fJacob; //! Jacobian matrices
 113   Int_t fNSteps; //!
 114   TArrayD *fSteps; //!
 115   TArrayD *fChi2Array; //!
 116   TArrayD *fChi2Smooth; //!
 117
 118   // Functions
 119
 120   void EvalCovariance(Double_t dZ);
 121   void ParPropagation(Double_t zEnd);
 122   void WeightPropagation(Double_t zEnd, Bool_t smooth);
 123   void MSThin(Int_t sign);
 124   void MSLine(Double_t dZ, Double_t X0);
 125   Bool_t FindPoint(Int_t ichamb, Double_t zEnd, Int_t currIndx, Int_t iFB, AliMUONHitForRec *&hitAdd, Int_t iz);
 126   void TryPoint(TMatrixD &point, const TMatrixD &pointWeight, TMatrixD &trackParTmp, Double_t &dChi2);
 127   void SetGeantParam(Double_t *VGeant3, Int_t iFB);
 128   void GetFromGeantParam(Double_t *VGeant3, Int_t iFB);
 129   Bool_t Recover(void);
 130   void AddMatrices(AliMUONTrackK *trackK, Double_t dChi2, AliMUONHitForRec *hitAdd);
 131   void CreateMatrix(TObjArray *objArray);
 132   void RemoveMatrices(Double_t zEnd);
 133   void RemoveMatrices(AliMUONTrackK* trackK);
 134   void Outlier();
 135   void SortHits(Int_t iflag, TObjArray *array);
 136   void DropBranches(Int_t imax, TObjArray *hits);
 137   void DropBranches(AliMUONSegment *segment);
 138   Bool_t ExistDouble(AliMUONHitForRec *hit);
 139   Bool_t ExistDouble(void);
 140
 141   private:
 142    // Some constants
 143    static const Int_t fgkSize; // number of track parameters
 144    static const Int_t fgkNSigma; //4; // acceptance window width in sigmas
 145    static const Double_t fgkChi2max; //25; // chi2 cut in smoother for outlier detection
 146    static const Int_t fgkTriesMax; // max number of attempts to find exact position during tracking
 147    static const Double_t fgkEpsilon; // tracking precision (cm)
 148
 149   ClassDef(AliMUONTrackK,0) // Kalman track in MUON arm of ALICE
 150     };
 151 #endif