Add MCNSD flag to trigger mask

[u/mrichter/AliRoot.git] / MUON / AliMUONVTrackReconstructor.h

diff --git a/MUON/AliMUONVTrackReconstructor.h b/MUON/AliMUONVTrackReconstructor.h
index 9630de614a95f1cd2a838bd732461238450188a5..be29209708e038518dc2e9c5140516ea80cb786c 100644 (file)
--- a/MUON/AliMUONVTrackReconstructor.h
+++ b/MUON/AliMUONVTrackReconstructor.h
@@ -11,9 +11,6 @@
 ///
 //  Author: Philippe Pillot
 
-#include "AliMUONReconstructor.h"
-#include "AliMUONRecoParam.h"
-
 #include <TObject.h>
 
 class AliMUONTrack;
@@ -21,6 +18,7 @@ class AliMUONTrackParam;
 class AliMUONVCluster;
 class AliMUONTriggerTrack;
 class AliMUONTrackHitPattern;
+class AliMUONVClusterServer;
 class AliMUONVClusterStore;
 class AliMUONVTrackStore;
 class AliMUONVTriggerTrackStore;
@@ -29,23 +27,17 @@ class AliMUONGeometryTransformer;
 class AliMUONDigitMaker;
 class AliMUONTriggerCircuit;
 class TClonesArray;
+class AliMUONRecoParam;
+class AliMUONLocalTrigger;
 
 class AliMUONVTrackReconstructor : public TObject {
 
  public:
-  AliMUONVTrackReconstructor(); // default Constructor
+  AliMUONVTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer); // default Constructor
   virtual ~AliMUONVTrackReconstructor(); // Destructor
 
-  // Reconstructed tracks
-           /// Return number of reconstructed tracks
-  Int_t GetNRecTracks() const {return fNRecTracks;} // Number
-           /// Set number of reconstructed tracks
-  void SetNRecTracks(Int_t NRecTracks) {fNRecTracks = NRecTracks;}
-           /// Return array of reconstructed tracks
-  TClonesArray* GetRecTracksPtr() const {return fRecTracksPtr;} // Array
- 
-  // Functions
-  void EventReconstruct(const AliMUONVClusterStore& clusterStore,
+
+  void EventReconstruct(AliMUONVClusterStore& clusterStore,
                         AliMUONVTrackStore& trackStore);
   
   void EventReconstructTrigger(const AliMUONTriggerCircuit& triggerCircuit,
@@ -56,6 +48,18 @@ class AliMUONVTrackReconstructor : public TObject {
                                  const AliMUONVTriggerTrackStore& triggerTrackStore,
                                  const AliMUONVTriggerStore& triggerStore,
                                  const AliMUONTrackHitPattern& trackHitPattern);
+
+  /// Return reco parameters
+  const AliMUONRecoParam* GetRecoParam() const { return fkRecoParam; }
+  
+  /// Re-fit the given track
+  virtual Bool_t RefitTrack(AliMUONTrack &track, Bool_t enableImprovement = kTRUE) = 0;
+  
+  void TriggerToTrack(const AliMUONTriggerCircuit& circuit,
+                      const AliMUONLocalTrigger& locTrg,
+                      AliMUONTriggerTrack& triggerTrack,
+                      UChar_t globalTriggerPattern = 0);
+  
   
   
  protected:
@@ -63,36 +67,54 @@ class AliMUONVTrackReconstructor : public TObject {
   TClonesArray *fRecTracksPtr; ///< pointer to array of reconstructed tracks
   Int_t fNRecTracks; ///< number of reconstructed tracks
 
+  AliMUONVClusterServer* fClusterServer; ///< reference to our cluster server
 
+  const AliMUONRecoParam* fkRecoParam; ///< reference to reco parameters
+  
+  Double_t* fMaxMCSAngle2; ///< maximum angle dispersion due to MCS
+  
   // Functions
   AliMUONVTrackReconstructor (const AliMUONVTrackReconstructor& rhs); ///< copy constructor
   AliMUONVTrackReconstructor& operator=(const AliMUONVTrackReconstructor& rhs); ///< assignment operator
   
-  /// Make track candidats from clusters in stations(1..) 4 and 5
-  virtual void MakeTrackCandidates(const AliMUONVClusterStore& clusterStore) = 0;
+  /// Make track candidates from clusters in stations(1..) 4 and 5
+  virtual Bool_t MakeTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
+  /// Make extra track candidates from clusters in stations(1..) 4 and 5
+  virtual Bool_t MakeMoreTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
   /// Follow tracks in stations(1..) 3, 2 and 1
-  virtual void FollowTracks(const AliMUONVClusterStore& clusterStore) = 0;
+  virtual Bool_t FollowTracks(AliMUONVClusterStore& clusterStore) = 0;
   /// Complement the reconstructed tracks
-  virtual void ComplementTracks(const AliMUONVClusterStore& clusterStore) = 0;
-  /// Improve the reconstructed tracks
-  virtual void ImproveTracks() = 0;
-  /// Finalize the tracking results
-  virtual void Finalize() = 0;
+  virtual Bool_t ComplementTracks(const AliMUONVClusterStore& clusterStore) = 0;
+  void ImproveTracks();
+  /// Improve the given reconstructed track
+  virtual void ImproveTrack(AliMUONTrack &track) = 0;
+  void Finalize();
+  /// Finalize the given track
+  virtual Bool_t FinalizeTrack(AliMUONTrack &track) = 0;
   
-  TClonesArray* MakeSegmentsInStation(const AliMUONVClusterStore& clusterStore, Int_t station);
+  Bool_t IsAcceptable(AliMUONTrackParam &trackParam);
+  
+  TClonesArray* MakeSegmentsBetweenChambers(const AliMUONVClusterStore& clusterStore, Int_t ch1, Int_t ch2);
 
+  void RemoveUsedSegments(TClonesArray& segments);
   void RemoveIdenticalTracks();
   void RemoveDoubleTracks();
+  void RemoveConnectedTracks(Int_t stMin, Int_t stMax, Bool_t all);
+  void TagConnectedTracks(Int_t stMin, Int_t stMax, Bool_t all);
 
+  void AskForNewClustersInStation(const AliMUONTrackParam &trackParam,
+                                 AliMUONVClusterStore& clusterStore, Int_t station);
+  void AskForNewClustersInChamber(const AliMUONTrackParam &trackParam,
+                                 AliMUONVClusterStore& clusterStore, Int_t chamber);
+  
   Double_t TryOneCluster(const AliMUONTrackParam &trackParam, AliMUONVCluster* cluster,
                         AliMUONTrackParam &trackParamAtCluster, Bool_t updatePropagator = kFALSE);
-  Bool_t   TryOneClusterFast(const AliMUONTrackParam &trackParam, AliMUONVCluster* cluster);
+  Bool_t   TryOneClusterFast(const AliMUONTrackParam &trackParam, const AliMUONVCluster* cluster);
   Double_t TryTwoClustersFast(const AliMUONTrackParam &trackParamAtCluster1, AliMUONVCluster* cluster2,
                              AliMUONTrackParam &trackParamAtCluster2);
 
-  Bool_t FollowLinearTrackInStation(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore,
-                                   Int_t nextStation);
-  
+  Bool_t FollowLinearTrackInChamber(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextChamber);
+  Bool_t FollowLinearTrackInStation(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextStation);
 
  private: