Update AliMUONRecoCheck to find the corresponding trackRefs also for the trigger...
[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 class AliMUONLocalTrigger;
32
33 class AliMUONVTrackReconstructor : public TObject {
34
35  public:
36   AliMUONVTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer); // default Constructor
37   virtual ~AliMUONVTrackReconstructor(); // Destructor
38
39
40   void EventReconstruct(AliMUONVClusterStore& clusterStore,
41                         AliMUONVTrackStore& trackStore);
42   
43   void EventReconstructTrigger(const AliMUONTriggerCircuit& triggerCircuit,
44                                const AliMUONVTriggerStore& triggerStore,
45                                AliMUONVTriggerTrackStore& triggerTrackStore);
46   
47   void ValidateTracksWithTrigger(AliMUONVTrackStore& trackStore,
48                                  const AliMUONVTriggerTrackStore& triggerTrackStore,
49                                  const AliMUONVTriggerStore& triggerStore,
50                                  const AliMUONTrackHitPattern& trackHitPattern);
51
52   /// Return reco parameters
53   const AliMUONRecoParam* GetRecoParam() const { return fkRecoParam; }
54   
55   /// Re-fit the given track
56   virtual Bool_t RefitTrack(AliMUONTrack &track, Bool_t enableImprovement = kTRUE) = 0;
57   
58   void TriggerToTrack(const AliMUONTriggerCircuit& circuit,
59                       const AliMUONLocalTrigger& locTrg,
60                       AliMUONTriggerTrack& triggerTrack,
61                       UChar_t globalTriggerPattern = 0);
62   
63   
64   
65  protected:
66
67   TClonesArray *fRecTracksPtr; ///< pointer to array of reconstructed tracks
68   Int_t fNRecTracks; ///< number of reconstructed tracks
69
70   AliMUONVClusterServer* fClusterServer; ///< reference to our cluster server
71
72   const AliMUONRecoParam* fkRecoParam; ///< reference to reco parameters
73   
74   Double_t* fMaxMCSAngle2; ///< maximum angle dispersion due to MCS
75   
76   // Functions
77   AliMUONVTrackReconstructor (const AliMUONVTrackReconstructor& rhs); ///< copy constructor
78   AliMUONVTrackReconstructor& operator=(const AliMUONVTrackReconstructor& rhs); ///< assignment operator
79   
80   /// Make track candidates from clusters in stations(1..) 4 and 5
81   virtual Bool_t MakeTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
82   /// Make extra track candidates from clusters in stations(1..) 4 and 5
83   virtual Bool_t MakeMoreTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
84   /// Follow tracks in stations(1..) 3, 2 and 1
85   virtual Bool_t FollowTracks(AliMUONVClusterStore& clusterStore) = 0;
86   /// Complement the reconstructed tracks
87   virtual Bool_t ComplementTracks(const AliMUONVClusterStore& clusterStore) = 0;
88   void ImproveTracks();
89   /// Improve the given reconstructed track
90   virtual void ImproveTrack(AliMUONTrack &track) = 0;
91   void Finalize();
92   /// Finalize the given track
93   virtual Bool_t FinalizeTrack(AliMUONTrack &track) = 0;
94   
95   Bool_t IsAcceptable(AliMUONTrackParam &trackParam);
96   
97   TClonesArray* MakeSegmentsBetweenChambers(const AliMUONVClusterStore& clusterStore, Int_t ch1, Int_t ch2);
98
99   void RemoveUsedSegments(TClonesArray& segments);
100   void RemoveIdenticalTracks();
101   void RemoveDoubleTracks();
102   void RemoveConnectedTracks(Int_t stMin, Int_t stMax, Bool_t all);
103   void TagConnectedTracks(Int_t stMin, Int_t stMax, Bool_t all);
104
105   void AskForNewClustersInStation(const AliMUONTrackParam &trackParam,
106                                   AliMUONVClusterStore& clusterStore, Int_t station);
107   void AskForNewClustersInChamber(const AliMUONTrackParam &trackParam,
108                                   AliMUONVClusterStore& clusterStore, Int_t chamber);
109   
110   Double_t TryOneCluster(const AliMUONTrackParam &trackParam, AliMUONVCluster* cluster,
111                          AliMUONTrackParam &trackParamAtCluster, Bool_t updatePropagator = kFALSE);
112   Bool_t   TryOneClusterFast(const AliMUONTrackParam &trackParam, const AliMUONVCluster* cluster);
113   Double_t TryTwoClustersFast(const AliMUONTrackParam &trackParamAtCluster1, AliMUONVCluster* cluster2,
114                               AliMUONTrackParam &trackParamAtCluster2);
115
116   Bool_t FollowLinearTrackInChamber(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextChamber);
117   Bool_t FollowLinearTrackInStation(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextStation);
118
119  private:
120   
121   // Functions
122   void ResetTracks();
123   
124   
125   ClassDef(AliMUONVTrackReconstructor, 0) // MUON track reconstructor in ALICE
126 };
127         
128 #endif