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

#ifndef ALIMUONVTRACKRECONSTRUCTOR_H
#define ALIMUONVTRACKRECONSTRUCTOR_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

/// \ingroup rec
/// \class AliMUONVTrackReconstructor
/// \brief Virtual class for the MUON track reconstruction
///
//  Author: Philippe Pillot

#include <TObject.h>

class AliMUONTrack;
class AliMUONTrackParam;
class AliMUONVCluster;
class AliMUONTriggerTrack;
class AliMUONTrackHitPattern;
class AliMUONVClusterServer;
class AliMUONVClusterStore;
class AliMUONVTrackStore;
class AliMUONVTriggerTrackStore;
class AliMUONVTriggerStore;
class AliMUONGeometryTransformer;
class AliMUONDigitMaker;
class AliMUONTriggerCircuit;
class TClonesArray;
class AliMUONRecoParam;
class AliMUONLocalTrigger;

class AliMUONVTrackReconstructor : public TObject {

 public:
  AliMUONVTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer,
			     const AliMUONGeometryTransformer* transformer); // default Constructor
  virtual ~AliMUONVTrackReconstructor(); // Destructor


  void EventReconstruct(AliMUONVClusterStore& clusterStore,
                        AliMUONVTrackStore& trackStore);
  
  void EventReconstructTrigger(const AliMUONTriggerCircuit& triggerCircuit,
                               const AliMUONVTriggerStore& triggerStore,
                               AliMUONVTriggerTrackStore& triggerTrackStore);
  
  void ValidateTracksWithTrigger(AliMUONVTrackStore& trackStore,
                                 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:

  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
  
  /// 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();
  
  
  ClassDef(AliMUONVTrackReconstructor, 0) // MUON track reconstructor in ALICE
};
	
#endif
Commit	Line	Data
	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 <TObject.h>
	15
	16	class AliMUONTrack;
	17	class AliMUONTrackParam;
	18	class AliMUONVCluster;
	19	class AliMUONTriggerTrack;
	20	class AliMUONTrackHitPattern;
	21	class AliMUONVClusterServer;
	22	class AliMUONVClusterStore;
	23	class AliMUONVTrackStore;
	24	class AliMUONVTriggerTrackStore;
	25	class AliMUONVTriggerStore;
	26	class AliMUONGeometryTransformer;
	27	class AliMUONDigitMaker;
	28	class AliMUONTriggerCircuit;
	29	class TClonesArray;
	30	class AliMUONRecoParam;
	31	class AliMUONLocalTrigger;
	32
	33	class AliMUONVTrackReconstructor : public TObject {
	34
	35	public:
	36	AliMUONVTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer,
	37	const AliMUONGeometryTransformer* transformer); // 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	/// Return reco parameters
	54	const AliMUONRecoParam* GetRecoParam() const { return fkRecoParam; }
	55
	56	/// Re-fit the given track
	57	virtual Bool_t RefitTrack(AliMUONTrack &track, Bool_t enableImprovement = kTRUE) = 0;
	58
	59	Bool_t TriggerToTrack(const AliMUONTriggerCircuit& circuit,
	60	const AliMUONLocalTrigger& locTrg,
	61	AliMUONTriggerTrack& triggerTrack,
	62	UChar_t globalTriggerPattern = 0);
	63
	64
	65
	66	protected:
	67
	68	TClonesArray *fRecTracksPtr; ///< pointer to array of reconstructed tracks
	69	Int_t fNRecTracks; ///< number of reconstructed tracks
	70
	71	AliMUONVClusterServer* fClusterServer; ///< reference to our cluster server
	72
	73	const AliMUONRecoParam* fkRecoParam; ///< reference to reco parameters
	74	const AliMUONGeometryTransformer* fkTransformer; //!< geometry transformer (not owner)
	75
	76	Double_t* fMaxMCSAngle2; ///< maximum angle dispersion due to MCS
	77
	78	// Functions
	79	AliMUONVTrackReconstructor (const AliMUONVTrackReconstructor& rhs); ///< copy constructor
	80	AliMUONVTrackReconstructor& operator=(const AliMUONVTrackReconstructor& rhs); ///< assignment operator
	81
	82	/// Make track candidates from clusters in stations(1..) 4 and 5
	83	virtual Bool_t MakeTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
	84	/// Make extra track candidates from clusters in stations(1..) 4 and 5
	85	virtual Bool_t MakeMoreTrackCandidates(AliMUONVClusterStore& clusterStore) = 0;
	86	/// Follow tracks in stations(1..) 3, 2 and 1
	87	virtual Bool_t FollowTracks(AliMUONVClusterStore& clusterStore) = 0;
	88	/// Complement the reconstructed tracks
	89	virtual Bool_t ComplementTracks(const AliMUONVClusterStore& clusterStore) = 0;
	90	void ImproveTracks();
	91	/// Improve the given reconstructed track
	92	virtual void ImproveTrack(AliMUONTrack &track) = 0;
	93	void Finalize();
	94	/// Finalize the given track
	95	virtual Bool_t FinalizeTrack(AliMUONTrack &track) = 0;
	96
	97	void DiscardMonoCathodClusters();
	98	void ChangeMonoCathodClusterRes(AliMUONTrack &track);
	99
	100	Bool_t IsAcceptable(AliMUONTrackParam &trackParam);
	101
	102	TClonesArray* MakeSegmentsBetweenChambers(const AliMUONVClusterStore& clusterStore, Int_t ch1, Int_t ch2);
	103
	104	void RemoveUsedSegments(TClonesArray& segments);
	105	void RemoveIdenticalTracks();
	106	void RemoveDoubleTracks();
	107	void RemoveBadTracks();
	108	void RemoveConnectedTracks(Int_t stMin, Int_t stMax, Bool_t all);
	109	void TagConnectedTracks(Int_t stMin, Int_t stMax, Bool_t all);
	110
	111	void AskForNewClustersInStation(const AliMUONTrackParam &trackParam,
	112	AliMUONVClusterStore& clusterStore, Int_t station);
	113	void AskForNewClustersInChamber(const AliMUONTrackParam &trackParam,
	114	AliMUONVClusterStore& clusterStore, Int_t chamber);
	115
	116	Double_t TryOneCluster(const AliMUONTrackParam &trackParam, AliMUONVCluster* cluster,
	117	AliMUONTrackParam &trackParamAtCluster, Bool_t updatePropagator = kFALSE);
	118	Bool_t TryOneClusterFast(const AliMUONTrackParam &trackParam, const AliMUONVCluster* cluster);
	119	Double_t TryTwoClustersFast(const AliMUONTrackParam &trackParamAtCluster1, AliMUONVCluster* cluster2,
	120	AliMUONTrackParam &trackParamAtCluster2);
	121
	122	Bool_t FollowLinearTrackInChamber(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextChamber);
	123	Bool_t FollowLinearTrackInStation(AliMUONTrack &trackCandidate, const AliMUONVClusterStore& clusterStore, Int_t nextStation);
	124
	125	private:
	126
	127	// Functions
	128	void ResetTracks();
	129
	130
	131	ClassDef(AliMUONVTrackReconstructor, 0) // MUON track reconstructor in ALICE
	132	};
	133
	134	#endif