Major revision of the Kalman tracking: works with new data structure,
[u/mrichter/AliRoot.git] / MUON / AliMUONTrack.h
1 #ifndef ALIMUONTRACK_H
2 #define ALIMUONTRACK_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
10 ///////////////////////////////////////////////////
11 // Reconstructed track in ALICE dimuon spectrometer
12 ///////////////////////////////////////////////////
13
14 #include <TClonesArray.h>
15
16 #include "AliMUONTrackParam.h" // object belongs to the class
17 #include "AliMUONHitForRec.h"  // object belongs to the class
18
19 //const Int_t kMaxTrackingChamber=10;
20         // not used
21
22 class TObjArray;
23 class TVirtualFitter;
24 class AliMUONEventReconstructor;
25 class AliMUONHitForRec;
26 class AliMUONSegment;
27
28 class AliMUONTrack : public TObject 
29 {
30  public:
31   AliMUONTrack(); // Default constructor
32   virtual ~AliMUONTrack(); // Destructor
33   AliMUONTrack (const AliMUONTrack& AliMUONTrack); // copy constructor
34   AliMUONTrack& operator=(const AliMUONTrack& AliMUONTrack); // assignment operator
35
36   AliMUONTrack(AliMUONSegment* BegSegment, AliMUONSegment* EndSegment, AliMUONEventReconstructor* EventReconstructor); // Constructor from two Segment's
37   AliMUONTrack(AliMUONSegment* Segment, AliMUONHitForRec* HitForRec, AliMUONEventReconstructor* EventReconstructor); // Constructor from one Segment and one HitForRec
38   void Remove(void);
39
40   AliMUONEventReconstructor* GetEventReconstructor(void) const {return fEventReconstructor;}
41   AliMUONTrackParam* GetTrackParamAtVertex(void) {return &fTrackParamAtVertex;}
42   void SetTrackParamAtVertex(void); // Set track parameters at vertex from last stations 4 & 5
43   void SetTrackParamAtVertex(AliMUONTrackParam* TrackParam) {fTrackParamAtVertex = *TrackParam;}
44   TClonesArray *GetTrackParamAtHit(void) const {return fTrackParamAtHit;}
45   TClonesArray *GetHitForRecAtHit(void) const {return fHitForRecAtHit;}
46   void ResetTrackParamAtHit(void) { fTrackParamAtHit->Delete(); }
47   void ResetHitForRecAtHit(void) { fHitForRecAtHit->Delete(); }
48   void AddTrackParamAtHit(const AliMUONTrackParam *trackParam) 
49     {new ((*fTrackParamAtHit)[fTrackParamAtHit->GetEntriesFast()]) AliMUONTrackParam(*trackParam);}
50   void AddHitForRecAtHit(const AliMUONHitForRec *hitForRec) 
51     {new ((*fHitForRecAtHit)[fHitForRecAtHit->GetEntriesFast()]) AliMUONHitForRec(*hitForRec);}
52
53   TObjArray* GetTrackHitsPtr(void) const {return fTrackHitsPtr;}
54   Int_t GetNTrackHits(void) const {return fNTrackHits;}
55   void SetNTrackHits(Int_t nTrackHits) {fNTrackHits = nTrackHits;}
56   Int_t GetFitMCS(void) const {return fFitMCS;}
57   Int_t GetFitNParam(void) const {return fFitNParam;}
58   Int_t GetFitStart(void) const {return fFitStart;}
59   Double_t GetFitFMin(void) const {return fFitFMin;}
60   Bool_t GetMatchTrigger(void) const {return fMatchTrigger;}
61   Double_t GetChi2MatchTrigger(void) const {return fChi2MatchTrigger;}
62   void SetFitMCS(Int_t FitMCS);
63   void SetFitNParam(Int_t FitNParam);
64   void SetFitStart(Int_t FitStart);
65
66   AliMUONTrackParam* GetTrackParamAtFirstHit(void) const;
67
68   void RecursiveDump(void) const; // Recursive dump (with track hits)
69   void Fit(); // Fit
70   void AddSegment(AliMUONSegment* Segment); // Add Segment
71   void AddHitForRec(AliMUONHitForRec* HitForRec); // Add HitForRec
72   void SetTrackParamAtHit(Int_t indexHit, AliMUONTrackParam *TrackParam) const;
73   Int_t HitsInCommon(AliMUONTrack* Track) const;
74   void MatchTriggerTrack(TClonesArray* TriggerTrackArray);
75   Bool_t* CompatibleTrack(AliMUONTrack* Track, Double_t Sigma2Cut) const; // return array of compatible chamber
76   
77   Int_t GetTrackID() const {return fTrackID;}
78   void SetTrackID(Int_t trackID) {fTrackID = trackID;}
79
80   static TVirtualFitter* Fitter(void) {return fgFitter;}
81
82  protected:
83  private:
84   static TVirtualFitter* fgFitter; //!                  Pointer to track fitter
85   AliMUONEventReconstructor* fEventReconstructor; //!   Pointer to EventReconstructor
86   AliMUONTrackParam fTrackParamAtVertex; // Track parameters at vertex
87   TClonesArray *fTrackParamAtHit; // Track parameters at hit
88   TClonesArray *fHitForRecAtHit; // Cluster parameters at hit
89   TObjArray *fTrackHitsPtr; //!  Pointer to array of pointers to TrackHit's
90   Int_t fNTrackHits; // Number of TrackHit's
91   Int_t fFitMCS; // 0(1) for fit without(with) multiple Coulomb scattering
92   Int_t fFitNParam; // 3(5) for fit with 3(5) parameters
93   Int_t fFitStart; // 0 or 1 for fit starting from parameters at vertex (0) or at first TrackHit(1)
94   Double_t fFitFMin; // minimum value of the function minimized by the fit
95   Bool_t fMatchTrigger; // 1 if track matches with trigger track, 0 if not
96   Double_t fChi2MatchTrigger; // chi2 of trigger/track matching 
97  
98   Int_t fTrackID; // track ID = track number in TrackRefs
99
100   ClassDef(AliMUONTrack, 2) // Reconstructed track in ALICE dimuon spectrometer
101     };
102         
103 #endif