- Adding handling of track info in digits.

[u/mrichter/AliRoot.git] / MUON / AliMUONTrackK.h

diff --git a/MUON/AliMUONTrackK.h b/MUON/AliMUONTrackK.h
index 188fa4206d60e9b450e6110683f4a69debff38bf..5be2a5d91903a23ca24497ea4d8885f2fc7f709c 100644 (file)
--- a/MUON/AliMUONTrackK.h
+++ b/MUON/AliMUONTrackK.h
@@ -3,26 +3,33 @@
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
-#include <TROOT.h>
+/*$Id$*/
+// Revision of includes 07/05/2004
+
+/// \ingroup rec
+/// \class AliMUONTrack
+/// \brief Kalman track in MUON arm of ALICE
+
 class TArrayD;
-class TMatrixD;
-class AliMUONEventReconstructor;
 class TClonesArray;
-class AliMUONSegment;
-class AliMUON;
+class TObjArray;
+#include <TMatrixDfwd.h>
+#include <TObject.h>
+
+class AliMUONEventRecoCombi;
 class AliMUONHitForRec;
+class AliMUONSegment;
+class AliMUONTrackReconstructor;
+#include "AliMUONTrack.h" 
 
-class AliMUONTrackK : public TObject {
+class AliMUONTrackK : public AliMUONTrack {
 
  public:
 
   AliMUONTrackK(); // Default constructor
   virtual ~AliMUONTrackK(); // Destructor
-  AliMUONTrackK (const AliMUONTrackK& source); // copy constructor
-  AliMUONTrackK& operator=(const AliMUONTrackK& source); // assignment operator
 
-  //AliMUONTrackK(const AliMUONEventReconstructor *EventReconstructor, const AliMUONHitForRec *hitForRec); // Constructor
-  AliMUONTrackK(AliMUONEventReconstructor *EventReconstructor, TClonesArray *hitForRec); // Constructor
+  AliMUONTrackK(AliMUONTrackReconstructor *TrackReconstructor, TClonesArray *hitForRec); // Constructor
   AliMUONTrackK(AliMUONSegment *segment); // Constructor from a segment
 
   // Pointer to hits on track
@@ -42,11 +49,21 @@ class AliMUONTrackK : public TObject {
   Bool_t KalmanFilter(Int_t ichamBeg, Int_t ichamEnd, Bool_t Back, Double_t zDipole1, Double_t zDipole2); // Kalman filter
   void StartBack(void); // start backpropagator
   void SetTrackQuality(Int_t iChi2); // compute track quality or Chi2
-  Bool_t KeepTrack(AliMUONTrackK* track0); // keep or discard track 
+  Bool_t KeepTrack(AliMUONTrackK* track0) const; // keep or discard track 
   void Kill(void); // kill track candidate
   void Branson(void); // Branson correction
   void GoToZ(Double_t zEnd); // propagate track to given Z
-  void GoToVertex(void); // propagate track to the vertex
+  void GoToVertex(Int_t iflag); // propagate track to the vertex
+  Bool_t Smooth(void); // apply smoother
+  Double_t GetChi2PerPoint(Int_t iPoint) const; // return Chi2 at point
+  void Print(FILE *lun) const; // print track information
+  void Print(const char* /*opt*/) const {return;} // print track information
+  AliMUONHitForRec* GetHitLastOk(void); // get hit before the skipped one
+  Int_t GetStation0(void); // return seed station number
+  Int_t DebugLevel(void) const {return fgDebug;} // return debug level
+  void SetDebugLevel(Int_t iDebug) {fgDebug = iDebug;} // set debug level
+  void FillMUONTrack(void); // set track parameters as for AliMUONTrack
+  void SetTrackParam(AliMUONTrackParam *trackParam, TMatrixD *par, Double_t z); // fill AliMUONTrackParam object
 
   // What is necessary for sorting TClonesArray's
   Bool_t IsSortable() const { return kTRUE; }
@@ -56,12 +73,17 @@ class AliMUONTrackK : public TObject {
 
  protected:
 
+  AliMUONTrackK (const AliMUONTrackK& source); // copy constructor
+  AliMUONTrackK& operator=(const AliMUONTrackK& source); // assignment operator
+
  private:
  
+  static Int_t fgDebug; // debug level
   static Int_t fgNOfPoints; // number of points in event
-  static AliMUON *fgMUON; // pointer to MUON module  
-  static AliMUONEventReconstructor *fgEventReconstructor; // pointer to event reconstructor
+  //static AliMUON *fgMUON; // pointer to MUON module  
+  static AliMUONTrackReconstructor *fgTrackReconstructor; // pointer to event reconstructor
   static TClonesArray *fgHitForRec; // pointer to hits
+  static AliMUONEventRecoCombi *fgCombi; // pointer to combined cluster/track finder
 
   AliMUONSegment *fStartSegment; // seed segment  
   Double_t fPosition; // Z-coordinate of track
@@ -79,23 +101,48 @@ class AliMUONTrackK : public TObject {
   TMatrixD *fCovariance; // covariance matrix
   TMatrixD *fWeight; //! weight matrix (inverse of covariance)
 
+  // For smoother
+  TObjArray *fParExtrap; //! extrapolated track parameters
+  TObjArray *fParFilter; //! filtered track parameters
+  TObjArray *fParSmooth; //! smoothed track parameters
+
+  TObjArray *fCovExtrap; //! extrapolated covariance matrices
+  TObjArray *fCovFilter; //! filtered covariance matrices
+
+  TObjArray *fJacob; //! Jacobian matrices
+  Int_t fNSteps; //!
+  TArrayD *fSteps; //!
+  TArrayD *fChi2Array; //!
+  TArrayD *fChi2Smooth; //!
+
   // Functions
 
   void EvalCovariance(Double_t dZ);
   void ParPropagation(Double_t zEnd);
-  void WeightPropagation(Double_t zEnd);
+  void WeightPropagation(Double_t zEnd, Bool_t smooth);
   void MSThin(Int_t sign);
   void MSLine(Double_t dZ, Double_t X0);
-  Bool_t FindPoint(Int_t ichamb, Double_t zEnd, Int_t currIndx, Int_t iFB, AliMUONHitForRec *&hitAdd);
+  Bool_t FindPoint(Int_t ichamb, Double_t zEnd, Int_t currIndx, Int_t iFB, AliMUONHitForRec *&hitAdd, Int_t iz);
   void TryPoint(TMatrixD &point, const TMatrixD &pointWeight, TMatrixD &trackParTmp, Double_t &dChi2);
   void SetGeantParam(Double_t *VGeant3, Int_t iFB);
   void GetFromGeantParam(Double_t *VGeant3, Int_t iFB);
-  void Recover(void);
+  Bool_t Recover(void);
+  void AddMatrices(AliMUONTrackK *trackK, Double_t dChi2, AliMUONHitForRec *hitAdd);
+  void CreateMatrix(TObjArray *objArray);
+  void RemoveMatrices(Double_t zEnd);
+  void RemoveMatrices(AliMUONTrackK* trackK);
+  void Outlier();
+  void SortHits(Int_t iflag, TObjArray *array);
+  void DropBranches(Int_t imax, TObjArray *hits);
+  void DropBranches(AliMUONSegment *segment);
+  Bool_t ExistDouble(AliMUONHitForRec *hit);
+  Bool_t ExistDouble(void);
 
   private:
    // Some constants
    static const Int_t fgkSize; // number of track parameters
    static const Int_t fgkNSigma; //4; // acceptance window width in sigmas
+   static const Double_t fgkChi2max; //25; // chi2 cut in smoother for outlier detection
    static const Int_t fgkTriesMax; // max number of attempts to find exact position during tracking
    static const Double_t fgkEpsilon; // tracking precision (cm)