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