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