New option to discard mono-cathod clusters (by changing their resolution) and refit...
[u/mrichter/AliRoot.git] / MUON / AliMUONVTrackReconstructor.h
index 259072c..f33b2b9 100644 (file)
@@ -11,9 +11,6 @@
 ///
 //  Author: Philippe Pillot
 
-#include "AliMUONReconstructor.h"
-#include "AliMUONRecoParam.h"
-
 #include <TObject.h>
 
 class AliMUONTrack;
@@ -30,13 +27,17 @@ class AliMUONGeometryTransformer;
 class AliMUONDigitMaker;
 class AliMUONTriggerCircuit;
 class TClonesArray;
+class AliMUONRecoParam;
+class AliMUONLocalTrigger;
 
 class AliMUONVTrackReconstructor : public TObject {
 
  public:
-  AliMUONVTrackReconstructor(AliMUONVClusterServer& clusterServer); // default Constructor
+  AliMUONVTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer,
+                            const AliMUONGeometryTransformer* transformer); // default Constructor
   virtual ~AliMUONVTrackReconstructor(); // Destructor
 
+
   void EventReconstruct(AliMUONVClusterStore& clusterStore,
                         AliMUONVTrackStore& trackStore);
   
@@ -48,9 +49,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;
+  
+  Bool_t TriggerToTrack(const AliMUONTriggerCircuit& circuit,
+                       const AliMUONLocalTrigger& locTrg,
+                       AliMUONTriggerTrack& triggerTrack,
+                       UChar_t globalTriggerPattern = 0);
   
-  /// re-fit the given track
-  virtual Bool_t RefitTrack(AliMUONTrack &track) = 0;
   
   
  protected:
@@ -58,31 +68,45 @@ 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
+  AliMUONVClusterServer* fClusterServer; ///< reference to our cluster server
 
+  const AliMUONRecoParam* fkRecoParam; ///< reference to reco parameters
+  const AliMUONGeometryTransformer* fkTransformer; //!< geometry transformer (not owner)
+  
+  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 candidates from clusters in stations(1..) 4 and 5
-  virtual void MakeTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
+  virtual Bool_t MakeTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
   /// Make extra track candidates from clusters in stations(1..) 4 and 5
-  virtual void MakeMoreTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
+  virtual Bool_t MakeMoreTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
   /// Follow tracks in stations(1..) 3, 2 and 1
-  virtual void FollowTracks(AliMUONVClusterStore& clusterStore) = 0;
+  virtual Bool_t FollowTracks(AliMUONVClusterStore& clusterStore) = 0;
   /// Complement the reconstructed tracks
-  virtual void ComplementTracks(const AliMUONVClusterStore& clusterStore) = 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 void FinalizeTrack(AliMUONTrack &track) = 0;
+  virtual Bool_t FinalizeTrack(AliMUONTrack &track) = 0;
+  
+  void DiscardMonoCathodClusters();
+  void ChangeMonoCathodClusterRes(AliMUONTrack &track);
+  
+  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 RemoveBadTracks();
+  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);
@@ -91,13 +115,12 @@ class AliMUONVTrackReconstructor : public TObject {
   
   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 FollowLinearTrackInChamber(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextChamber);
   Bool_t FollowLinearTrackInStation(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextStation);
-  
 
  private: