- Update and add new ESD QA histograms
[u/mrichter/AliRoot.git] / MUON / AliMUONVTrackReconstructor.h
1 #ifndef ALIMUONVTRACKRECONSTRUCTOR_H
2 #define ALIMUONVTRACKRECONSTRUCTOR_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
8 /// \ingroup rec
9 /// \class AliMUONVTrackReconstructor
10 /// \brief Virtual class for the MUON track reconstruction
11 ///
12 //  Author: Philippe Pillot
13
14 #include <TObject.h>
15
16 class AliMUONTrack;
17 class AliMUONTrackParam;
18 class AliMUONVCluster;
19 class AliMUONTriggerTrack;
20 class AliMUONTrackHitPattern;
21 class AliMUONVClusterServer;
22 class AliMUONVClusterStore;
23 class AliMUONVTrackStore;
24 class AliMUONVTriggerTrackStore;
25 class AliMUONVTriggerStore;
26 class AliMUONGeometryTransformer;
27 class AliMUONDigitMaker;
28 class AliMUONTriggerCircuit;
29 class TClonesArray;
30 class AliMUONRecoParam;
31
32 class AliMUONVTrackReconstructor : public TObject {
33
34  public:
35   AliMUONVTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer); // default Constructor
36   virtual ~AliMUONVTrackReconstructor(); // Destructor
37
38
39   void EventReconstruct(AliMUONVClusterStore& clusterStore,
40                         AliMUONVTrackStore& trackStore);
41   
42   void EventReconstructTrigger(const AliMUONTriggerCircuit& triggerCircuit,
43                                const AliMUONVTriggerStore& triggerStore,
44                                AliMUONVTriggerTrackStore& triggerTrackStore);
45   
46   void ValidateTracksWithTrigger(AliMUONVTrackStore& trackStore,
47                                  const AliMUONVTriggerTrackStore& triggerTrackStore,
48                                  const AliMUONVTriggerStore& triggerStore,
49                                  const AliMUONTrackHitPattern& trackHitPattern);
50
51   const AliMUONRecoParam* GetRecoParam() const { return fkRecoParam; }
52   
53   /// re-fit the given track
54   virtual Bool_t RefitTrack(AliMUONTrack &track, Bool_t enableImprovement = kTRUE) = 0;
55   
56   
57  protected:
58
59   TClonesArray *fRecTracksPtr; ///< pointer to array of reconstructed tracks
60   Int_t fNRecTracks; ///< number of reconstructed tracks
61
62   AliMUONVClusterServer* fClusterServer; ///< reference to our cluster server
63
64   const AliMUONRecoParam* fkRecoParam; ///< reference to reco parameters
65   
66   // Functions
67   AliMUONVTrackReconstructor (const AliMUONVTrackReconstructor& rhs); ///< copy constructor
68   AliMUONVTrackReconstructor& operator=(const AliMUONVTrackReconstructor& rhs); ///< assignment operator
69   
70   /// Make track candidates from clusters in stations(1..) 4 and 5
71   virtual void MakeTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
72   /// Make extra track candidates from clusters in stations(1..) 4 and 5
73   virtual void MakeMoreTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
74   /// Follow tracks in stations(1..) 3, 2 and 1
75   virtual void FollowTracks(AliMUONVClusterStore& clusterStore) = 0;
76   /// Complement the reconstructed tracks
77   virtual void ComplementTracks(const AliMUONVClusterStore& clusterStore) = 0;
78   void ImproveTracks();
79   /// Improve the given reconstructed track
80   virtual void ImproveTrack(AliMUONTrack &track) = 0;
81   void Finalize();
82   /// Finalize the given track
83   virtual void FinalizeTrack(AliMUONTrack &track) = 0;
84   
85   TClonesArray* MakeSegmentsBetweenChambers(const AliMUONVClusterStore& clusterStore, Int_t ch1, Int_t ch2);
86
87   void RemoveIdenticalTracks();
88   void RemoveDoubleTracks();
89   void RemoveConnectedTracks(Bool_t inSt345 = kFALSE);
90
91   void AskForNewClustersInStation(const AliMUONTrackParam &trackParam,
92                                   AliMUONVClusterStore& clusterStore, Int_t station);
93   void AskForNewClustersInChamber(const AliMUONTrackParam &trackParam,
94                                   AliMUONVClusterStore& clusterStore, Int_t chamber);
95   
96   Double_t TryOneCluster(const AliMUONTrackParam &trackParam, AliMUONVCluster* cluster,
97                          AliMUONTrackParam &trackParamAtCluster, Bool_t updatePropagator = kFALSE);
98   Bool_t   TryOneClusterFast(const AliMUONTrackParam &trackParam, const AliMUONVCluster* cluster);
99   Double_t TryTwoClustersFast(const AliMUONTrackParam &trackParamAtCluster1, AliMUONVCluster* cluster2,
100                               AliMUONTrackParam &trackParamAtCluster2);
101
102   Bool_t FollowLinearTrackInChamber(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextChamber);
103   Bool_t FollowLinearTrackInStation(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextStation);
104   
105
106  private:
107   
108   // Functions
109   void ResetTracks();
110   
111   
112   ClassDef(AliMUONVTrackReconstructor, 0) // MUON track reconstructor in ALICE
113 };
114         
115 #endif