New option to discard mono-cathod clusters (by changing their resolution) and refit...
[u/mrichter/AliRoot.git] / MUON / AliMUONVTrackReconstructor.h
index 509d2c7..f33b2b9 100644 (file)
 
 #include <TObject.h>
 
-class TClonesArray;
+class AliMUONTrack;
+class AliMUONTrackParam;
+class AliMUONVCluster;
 class AliMUONTriggerTrack;
 class AliMUONTrackHitPattern;
+class AliMUONVClusterServer;
 class AliMUONVClusterStore;
 class AliMUONVTrackStore;
 class AliMUONVTriggerTrackStore;
@@ -23,32 +26,19 @@ class AliMUONVTriggerStore;
 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,
+                            const AliMUONGeometryTransformer* transformer); // default Constructor
   virtual ~AliMUONVTrackReconstructor(); // Destructor
 
-  // Parameters for track reconstruction: public methods
-  // Get and Set, Set to defaults
-           /// Return minimum value (GeV/c) of momentum in bending plane
-  Double_t GetMinBendingMomentum(void) const {return fMinBendingMomentum;}
-           /// Return chamber resolution (cm) in bending plane
-  Double_t GetBendingResolution(void) const {return fBendingResolution;}
-           /// Return chamber resolution (cm) in non-bending plane
-  Double_t GetNonBendingResolution(void) const {return fNonBendingResolution;}
-
-  // 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(void) const {return fRecTracksPtr;} // Array
-  // Functions
-  void EventReconstruct(const AliMUONVClusterStore& clusterStore,
+
+  void EventReconstruct(AliMUONVClusterStore& clusterStore,
                         AliMUONVTrackStore& trackStore);
   
   void EventReconstructTrigger(const AliMUONTriggerCircuit& triggerCircuit,
@@ -59,61 +49,84 @@ 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);
+  
+  
+  
  protected:
 
-  // Defaults parameters for reconstruction
-  static const Double_t fgkDefaultMinBendingMomentum; ///< default min. bending momentum for reconstruction
-  static const Double_t fgkDefaultMaxBendingMomentum; ///< default max. bending momentum for reconstruction
-  static const Double_t fgkDefaultBendingResolution; ///< default bending coordinate resolution for reconstruction 
-  static const Double_t fgkDefaultNonBendingResolution; ///< default non bending coordinate resolution for reconstruction
-  static const Double_t fgkDefaultBendingVertexDispersion; ///< default vertex dispersion in bending plane for reconstruction
-  static const Double_t fgkDefaultNonBendingVertexDispersion; ///< default vertex dispersion in non bending plane for reconstruction
-  static const Double_t fgkDefaultMaxNormChi2MatchTrigger; ///< default maximum normalized chi2 of tracking/trigger track matching
-  
-  // Parameters for track reconstruction
-  Double_t fMinBendingMomentum; ///< minimum value (GeV/c) of momentum in bending plane
-  Double_t fMaxBendingMomentum; ///< maximum value (GeV/c) of momentum in bending plane
-  Double_t fBendingResolution; ///< chamber resolution (cm) in bending plane
-  Double_t fNonBendingResolution; ///< chamber resolution (cm) in non bending plane
-  Double_t fBendingVertexDispersion; ///< vextex dispersion (cm) in bending plane
-  Double_t fNonBendingVertexDispersion; ///< vextex dispersion (cm) in non bending plane
-  Double_t fMaxNormChi2MatchTrigger; ///< maximum normalized chi2 of tracking/trigger track matching
-  
-  TClonesArray* fHitsForRecPtr; ///< pointer to the array of hits for reconstruction
-  Int_t fNHitsForRec; ///< number of hits for reconstruction
-  Int_t* fNHitsForRecPerChamber; ///< number of HitsForRec
-  Int_t* fIndexOfFirstHitForRecPerChamber; ///< index (0...) of first HitForRec
-
-  // Reconstructed tracks
   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
+  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
   
-  void SortHitsForRecWithIncreasingChamber();
-  TClonesArray *MakeSegmentsInStation(Int_t station);
-
-               /// \todo add comment
-  virtual void AddHitsForRecFromRawClusters(const AliMUONVClusterStore& clusterStore);
-               /// \todo add comment
-  virtual void MakeTracks(void) = 0;
-               /// \todo add comment
-  virtual void MakeTrackCandidates(void) = 0;
-               /// \todo add comment
-  virtual void FollowTracks(void) = 0;
-               /// \todo add comment
-  virtual void RemoveDoubleTracks(void) = 0;
-               /// \todo add comment
-  virtual void FillMUONTrack(void) = 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 Bool_t FollowTracks(AliMUONVClusterStore& clusterStore) = 0;
+  /// Complement the reconstructed tracks
+  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;
+  
+  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);
+  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, 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:
   
   // Functions
-  void ResetTracks(void);
-  void ResetHitsForRec(void);
+  void ResetTracks();
+  
   
   ClassDef(AliMUONVTrackReconstructor, 0) // MUON track reconstructor in ALICE
 };