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

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

/*$Id$*/
// Revision of includes 07/05/2004

/// \ingroup rec
/// \class AliMUONTrack
/// \brief Reconstructed track in ALICE dimuon spectrometer
///
////////////////////////////////////////////////////
/// Reconstructed track in ALICE dimuon spectrometer
////////////////////////////////////////////////////

#include <TClonesArray.h>
#include <TMatrixD.h>

class AliMUONVCluster;
class AliMUONObjectPair;
class AliMUONTrackParam;

class AliMUONTrack : public TObject 
{
 public:
  AliMUONTrack(); // Default constructor
  AliMUONTrack(AliMUONObjectPair *segment, Double_t bendingVertexDispersion); // Constructor from a segment
  virtual ~AliMUONTrack(); // Destructor
  AliMUONTrack (const AliMUONTrack& track); // copy constructor
  AliMUONTrack& operator=(const AliMUONTrack& track); // assignment operator

  void Reset();
  
  TClonesArray* GetTrackParamAtCluster() const;
  void          AddTrackParamAtCluster(const AliMUONTrackParam &trackParam, AliMUONVCluster &cluster, Bool_t copy = kFALSE); 
  void          RemoveTrackParamAtCluster(AliMUONTrackParam *trackParam);
  Bool_t        UpdateTrackParamAtCluster();
  Bool_t        UpdateCovTrackParamAtCluster();
  
  Bool_t IsValid(UInt_t requestedStationMask, Bool_t request2ChInSameSt45 = kFALSE);
  
  void TagRemovableClusters(UInt_t requestedStationMask);
  
  /// return the number of clusters attached to the track
  Int_t GetNClusters() const {return fTrackParamAtCluster ? fTrackParamAtCluster->GetEntriesFast() : 0;}

  /// return kTrue if the vertex must be used to constrain the fit, kFalse if not
  Bool_t FitWithVertex() const {return fFitWithVertex;}
  /// set the flag telling whether the vertex must be used to constrain the fit or not
  void   FitWithVertex(Bool_t fitWithVertex) { fFitWithVertex = fitWithVertex; }
  /// return the vertex resolution square used during the tracking procedure
  void   GetVertexErrXY2(Double_t &nonBendingErr2, Double_t &bendingErr2) const
         { nonBendingErr2 = fVertexErrXY2[0]; bendingErr2 = fVertexErrXY2[1]; }
  /// set the vertex resolution square used during the tracking procedure
  void   SetVertexErrXY2(Double_t nonBendingErr2, Double_t bendingErr2)
	 { fVertexErrXY2[0] = nonBendingErr2; fVertexErrXY2[1] = bendingErr2; }

  /// return kTrue if the multiple scattering must be accounted for in the fit, kFalse if not
  Bool_t FitWithMCS() const {return fFitWithMCS;}
  /// set the flag telling whether the multiple scattering must be accounted for in the fit or not
  void   FitWithMCS(Bool_t fitWithMCS) {fFitWithMCS = fitWithMCS;}
  
  Bool_t   ComputeClusterWeights(TMatrixD* mcsCovariances = 0);
  Bool_t   ComputeLocalChi2(Bool_t accountForMCS);
  Double_t ComputeGlobalChi2(Bool_t accountForMCS);

  /// return the minimum value of the function minimized by the fit
  Double_t GetGlobalChi2() const {return fGlobalChi2;}
  /// set the minimum value of the function minimized by the fit
  void     SetGlobalChi2(Double_t chi2) { fGlobalChi2 = chi2;}
  
  /// return kTRUE if the track has been improved
  Bool_t IsImproved() const {return fImproved;}
  /// set the flag telling whether the track has been improved or not
  void   SetImproved(Bool_t improved) { fImproved = improved;}
  
  /// return 1,2,3 if track matches with trigger track, 0 if not
  Int_t    GetMatchTrigger(void) const {return fMatchTrigger;}
  /// returns the local trigger number corresponding to the trigger track (obsolete)
  Int_t    GetLoTrgNum(void) const {return LoCircuit();}
  /// set the flag telling whether track matches with trigger track or not
  void	   SetMatchTrigger(Int_t matchTrigger) {fMatchTrigger = matchTrigger;}
  /// return the chi2 of trigger/track matching 
  Double_t GetChi2MatchTrigger(void) const {return fChi2MatchTrigger;}
  /// set the chi2 of trigger/track matching 
  void     SetChi2MatchTrigger(Double_t chi2MatchTrigger) {fChi2MatchTrigger = chi2MatchTrigger;}

  Int_t ClustersInCommon(AliMUONTrack* track, Int_t stMin = 0, Int_t stMax = 4) const;

  Int_t    GetNDF() const;
  Double_t GetNormalizedChi2() const;

  Int_t  FindCompatibleClusters(AliMUONTrack &track, Double_t sigma2Cut, Bool_t compatibleCluster[10]) const;
  Bool_t Match(AliMUONTrack &track, Double_t sigma2Cut, Int_t &nMatchClusters) const;
  
  /// return pointer to track parameters at vertex (can be 0x0)
  AliMUONTrackParam* GetTrackParamAtVertex() const {return fTrackParamAtVertex;}
  void               SetTrackParamAtVertex(const AliMUONTrackParam* trackParam);

  /// set word telling which trigger chambers where hit by track
  UShort_t GetHitsPatternInTrigCh() const {return fHitsPatternInTrigCh;}
  /// set word telling which trigger chambers where hit by track
  void     SetHitsPatternInTrigCh(UShort_t hitsPatternInTrigCh) {fHitsPatternInTrigCh = hitsPatternInTrigCh;}

  /// set local trigger information for the matched trigger track
  void  SetLocalTrigger(Int_t loCirc, Int_t loStripX, Int_t loStripY, Int_t loDev, Int_t loLpt, Int_t loHpt, UChar_t respWithoutChamber=0);
  /// return local trigger information for the matched trigger track
  Int_t GetLocalTrigger(void) const { return fLocalTrigger;        }
  /// number of triggering circuit
  Int_t LoCircuit(void) const { return fLocalTrigger & 0xFF;       }
  /// x-strip local trigger 
  Int_t LoStripX(void) const  { return fLocalTrigger >>  8 & 0x1F; }
  /// y-strip local trigger 
  Int_t LoStripY(void) const  { return fLocalTrigger >> 13 & 0x0F; }
  /// deviation local trigger 
  Int_t LoDev(void)    const  { return fLocalTrigger >> 17 & 0x1F; }
  /// low pt decision local trigger 
  Int_t LoLpt(void)    const  { return fLocalTrigger >> 22 & 0x03; }
  /// high pt decision local trigger 
  Int_t LoHpt(void)    const  { return fLocalTrigger >> 24 & 0x03; }
  /// Word stating if trigger would be fired without one chamber
  Int_t GetTriggerWithoutChamber(void) const { return fLocalTrigger >> 26 & 0xF; }
  /// Check if trigger would be fired without chamber (ich [0,3])
  Bool_t TriggerFiredWithoutChamber(Int_t ich) const { return GetTriggerWithoutChamber() >> (3 - ich) & 0x1; }

  void  FindMCLabel();
  /// set the corresponding MC track number
  void  SetMCLabel(Int_t label) {fTrackID = label;}
  /// return the corresponding MC track number
  Int_t GetMCLabel() const {return fTrackID;}
  
  void RecursiveDump(void) const; // Recursive dump (with associated clusters)

  virtual void Print(Option_t* opt="") const;

  virtual void Clear(Option_t* opt="");
  
  /// return the maximum chi2 above which the track can be considered as abnormal (due to extrapolation failure, ...)
  static Double_t MaxChi2() {return fgkMaxChi2;}
  
  void   Connected(Bool_t flag = kTRUE) {fConnected = flag;}
  Bool_t IsConnected() const {return fConnected;}


 private:
  
  static const Double_t fgkMaxChi2; ///< maximum chi2 above which the track can be considered as abnormal
  
  mutable TClonesArray* fTrackParamAtCluster; ///< Track parameters at cluster
  
  Bool_t   fFitWithVertex;   //!< kTRUE if using the vertex to constrain the fit, kFALSE if not
  Double_t fVertexErrXY2[2]; //!< Vertex resolution square used during the tracking procedure if required
  
  Bool_t fFitWithMCS; //!< kTRUE if accounting for multiple scattering in the fit, kFALSE if not
  
  TMatrixD* fClusterWeightsNonBending; //!< weights matrix, in non bending direction, of clusters attached to the track
				       //!< (accounting for multiple scattering and cluster resolution)
  TMatrixD* fClusterWeightsBending;    //!< weights matrix, in bending direction, of clusters attached to the track
				       //!< (accounting for multiple scattering and cluster resolution)
  
  Double_t fGlobalChi2; ///< Global chi2 of the track
  
  Bool_t fImproved; //!< kTRUE if the track has been improved
  
  Int_t fMatchTrigger;  ///<  0 track does not match trigger
                        ///<  1 track match but does not pass pt cut
                        ///<  2 track match Low pt cut
                        ///<  3 track match High pt cut
  Double_t fChi2MatchTrigger; ///< chi2 of trigger/track matching 
  
  Int_t fTrackID; ///< Point to the corresponding MC track
  
  AliMUONTrackParam* fTrackParamAtVertex; //!< Track parameters at vertex
  
  UShort_t fHitsPatternInTrigCh; ///< Word containing info on the hits left in trigger chambers

  Int_t fLocalTrigger;    ///< packed local trigger information
  
  Bool_t fConnected; ///< kTRUE if that track shares cluster(s) with another
  
  // methods
  Bool_t ComputeClusterWeights(TMatrixD& clusterWeightsNB, TMatrixD& clusterWeightsB,
			       TMatrixD* mcsCovariances = 0, AliMUONVCluster* discardedCluster = 0) const;
  void   ComputeMCSCovariances(TMatrixD& mcsCovariances) const;
  
  
  ClassDef(AliMUONTrack, 9) // Reconstructed track in ALICE dimuon spectrometer
};
	
#endif
Commit	Line	Data
a9e2aefa	1	#ifndef ALIMUONTRACK_H
	2	#define ALIMUONTRACK_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/$Id$/
30178c30	7	// Revision of includes 07/05/2004
30178c30	8
692de412	9	/// \ingroup rec
	10	/// \class AliMUONTrack
	11	/// \brief Reconstructed track in ALICE dimuon spectrometer
	12	///
	13	////////////////////////////////////////////////////
	14	/// Reconstructed track in ALICE dimuon spectrometer
	15	////////////////////////////////////////////////////
3831f268	16
ea94c18b	17	#include <TClonesArray.h>
933daf4c	18	#include <TMatrixD.h>
a9e2aefa	19
96ebe67e	20	class AliMUONVCluster;
ea94c18b	21	class AliMUONObjectPair;
933daf4c	22	class AliMUONTrackParam;
a9e2aefa	23
30178c30	24	class AliMUONTrack : public TObject
30178c30	25	{
a9e2aefa	26	public:
d837040f	27	AliMUONTrack(); // Default constructor
a0dc65b4	28	AliMUONTrack(AliMUONObjectPair *segment, Double_t bendingVertexDispersion); // Constructor from a segment
8429a5e4	29	virtual ~AliMUONTrack(); // Destructor
ea94c18b	30	AliMUONTrack (const AliMUONTrack& track); // copy constructor
ea94c18b	31	AliMUONTrack& operator=(const AliMUONTrack& track); // assignment operator
a9e2aefa	32
b1fea02e	33	void Reset();
b1fea02e	34
7332f213	35	TClonesArray* GetTrackParamAtCluster() const;
96ebe67e	36	void AddTrackParamAtCluster(const AliMUONTrackParam &trackParam, AliMUONVCluster &cluster, Bool_t copy = kFALSE);
96ebe67e	37	void RemoveTrackParamAtCluster(AliMUONTrackParam *trackParam);
4ea3f013	38	Bool_t UpdateTrackParamAtCluster();
4ea3f013	39	Bool_t UpdateCovTrackParamAtCluster();
96ebe67e	40
f202486b	41	Bool_t IsValid(UInt_t requestedStationMask, Bool_t request2ChInSameSt45 = kFALSE);
7332f213	42
a0dc65b4	43	void TagRemovableClusters(UInt_t requestedStationMask);
7332f213	44
96ebe67e	45	/// return the number of clusters attached to the track
7332f213	46	Int_t GetNClusters() const {return fTrackParamAtCluster ? fTrackParamAtCluster->GetEntriesFast() : 0;}
96ebe67e	47
	48	/// return kTrue if the vertex must be used to constrain the fit, kFalse if not
	49	Bool_t FitWithVertex() const {return fFitWithVertex;}
	50	/// set the flag telling whether the vertex must be used to constrain the fit or not
	51	void FitWithVertex(Bool_t fitWithVertex) { fFitWithVertex = fitWithVertex; }
	52	/// return the vertex resolution square used during the tracking procedure
	53	void GetVertexErrXY2(Double_t &nonBendingErr2, Double_t &bendingErr2) const
	54	{ nonBendingErr2 = fVertexErrXY2[0]; bendingErr2 = fVertexErrXY2[1]; }
	55	/// set the vertex resolution square used during the tracking procedure
	56	void SetVertexErrXY2(Double_t nonBendingErr2, Double_t bendingErr2)
	57	{ fVertexErrXY2[0] = nonBendingErr2; fVertexErrXY2[1] = bendingErr2; }
	58
	59	/// return kTrue if the multiple scattering must be accounted for in the fit, kFalse if not
	60	Bool_t FitWithMCS() const {return fFitWithMCS;}
	61	/// set the flag telling whether the multiple scattering must be accounted for in the fit or not
	62	void FitWithMCS(Bool_t fitWithMCS) {fFitWithMCS = fitWithMCS;}
	63
	64	Bool_t ComputeClusterWeights(TMatrixD* mcsCovariances = 0);
	65	Bool_t ComputeLocalChi2(Bool_t accountForMCS);
	66	Double_t ComputeGlobalChi2(Bool_t accountForMCS);
	67
	68	/// return the minimum value of the function minimized by the fit
	69	Double_t GetGlobalChi2() const {return fGlobalChi2;}
	70	/// set the minimum value of the function minimized by the fit
	71	void SetGlobalChi2(Double_t chi2) { fGlobalChi2 = chi2;}
	72
	73	/// return kTRUE if the track has been improved
	74	Bool_t IsImproved() const {return fImproved;}
	75	/// set the flag telling whether the track has been improved or not
	76	void SetImproved(Bool_t improved) { fImproved = improved;}
	77
	78	/// return 1,2,3 if track matches with trigger track, 0 if not
	79	Int_t GetMatchTrigger(void) const {return fMatchTrigger;}
bdfb6eef	80	/// returns the local trigger number corresponding to the trigger track (obsolete)
bdfb6eef	81	Int_t GetLoTrgNum(void) const {return LoCircuit();}
96ebe67e	82	/// set the flag telling whether track matches with trigger track or not
96ebe67e	83	void SetMatchTrigger(Int_t matchTrigger) {fMatchTrigger = matchTrigger;}
96ebe67e	84	/// return the chi2 of trigger/track matching
	85	Double_t GetChi2MatchTrigger(void) const {return fChi2MatchTrigger;}
	86	/// set the chi2 of trigger/track matching
	87	void SetChi2MatchTrigger(Double_t chi2MatchTrigger) {fChi2MatchTrigger = chi2MatchTrigger;}
ea94c18b	88
5c15a68b	89	Int_t ClustersInCommon(AliMUONTrack* track, Int_t stMin = 0, Int_t stMax = 4) const;
ea94c18b	90
5a240757	91	Int_t GetNDF() const;
96ebe67e	92	Double_t GetNormalizedChi2() const;
ea94c18b	93
f202486b	94	Int_t FindCompatibleClusters(AliMUONTrack &track, Double_t sigma2Cut, Bool_t compatibleCluster[10]) const;
f202486b	95	Bool_t Match(AliMUONTrack &track, Double_t sigma2Cut, Int_t &nMatchClusters) const;
b8dc484b	96
61fed964	97	/// return pointer to track parameters at vertex (can be 0x0)
40a42243	98	AliMUONTrackParam* GetTrackParamAtVertex() const {return fTrackParamAtVertex;}
96ebe67e	99	void SetTrackParamAtVertex(const AliMUONTrackParam* trackParam);
	100
	101	/// set word telling which trigger chambers where hit by track
	102	UShort_t GetHitsPatternInTrigCh() const {return fHitsPatternInTrigCh;}
	103	/// set word telling which trigger chambers where hit by track
	104	void SetHitsPatternInTrigCh(UShort_t hitsPatternInTrigCh) {fHitsPatternInTrigCh = hitsPatternInTrigCh;}
7771752e	105
423b32ca	106	/// set local trigger information for the matched trigger track
00d46f24	107	void SetLocalTrigger(Int_t loCirc, Int_t loStripX, Int_t loStripY, Int_t loDev, Int_t loLpt, Int_t loHpt, UChar_t respWithoutChamber=0);
423b32ca	108	/// return local trigger information for the matched trigger track
61fed964	109	Int_t GetLocalTrigger(void) const { return fLocalTrigger; }
4a335550	110	/// number of triggering circuit
01413742	111	Int_t LoCircuit(void) const { return fLocalTrigger & 0xFF; }
4a335550	112	/// x-strip local trigger
423b32ca	113	Int_t LoStripX(void) const { return fLocalTrigger >> 8 & 0x1F; }
4a335550	114	/// y-strip local trigger
423b32ca	115	Int_t LoStripY(void) const { return fLocalTrigger >> 13 & 0x0F; }
4a335550	116	/// deviation local trigger
423b32ca	117	Int_t LoDev(void) const { return fLocalTrigger >> 17 & 0x1F; }
4a335550	118	/// low pt decision local trigger
423b32ca	119	Int_t LoLpt(void) const { return fLocalTrigger >> 22 & 0x03; }
4a335550	120	/// high pt decision local trigger
423b32ca	121	Int_t LoHpt(void) const { return fLocalTrigger >> 24 & 0x03; }
00d46f24	122	/// Word stating if trigger would be fired without one chamber
	123	Int_t GetTriggerWithoutChamber(void) const { return fLocalTrigger >> 26 & 0xF; }
	124	/// Check if trigger would be fired without chamber (ich [0,3])
	125	Bool_t TriggerFiredWithoutChamber(Int_t ich) const { return GetTriggerWithoutChamber() >> (3 - ich) & 0x1; }
423b32ca	126
2e2d0c44	127	void FindMCLabel();
	128	/// set the corresponding MC track number
	129	void SetMCLabel(Int_t label) {fTrackID = label;}
	130	/// return the corresponding MC track number
	131	Int_t GetMCLabel() const {return fTrackID;}
	132
96ebe67e	133	void RecursiveDump(void) const; // Recursive dump (with associated clusters)
6464217e	134
96ebe67e	135	virtual void Print(Option_t* opt="") const;
a9e2aefa	136
d2b1e7bb	137	virtual void Clear(Option_t* opt="");
4ea3f013	138
3e1866ad	139	/// return the maximum chi2 above which the track can be considered as abnormal (due to extrapolation failure, ...)
4ea3f013	140	static Double_t MaxChi2() {return fgkMaxChi2;}
5c15a68b	141
	142	void Connected(Bool_t flag = kTRUE) {fConnected = flag;}
	143	Bool_t IsConnected() const {return fConnected;}
956019b6	144
ea94c18b	145
a9e2aefa	146	private:
4ea3f013	147
	148	static const Double_t fgkMaxChi2; ///< maximum chi2 above which the track can be considered as abnormal
	149
7332f213	150	mutable TClonesArray* fTrackParamAtCluster; ///< Track parameters at cluster
de2cd600	151
96ebe67e	152	Bool_t fFitWithVertex; //!< kTRUE if using the vertex to constrain the fit, kFALSE if not
96ebe67e	153	Double_t fVertexErrXY2[2]; //!< Vertex resolution square used during the tracking procedure if required
208f139e	154
ea94c18b	155	Bool_t fFitWithMCS; //!< kTRUE if accounting for multiple scattering in the fit, kFALSE if not
ea94c18b	156
96ebe67e	157	TMatrixD* fClusterWeightsNonBending; //!< weights matrix, in non bending direction, of clusters attached to the track
	158	//!< (accounting for multiple scattering and cluster resolution)
	159	TMatrixD* fClusterWeightsBending; //!< weights matrix, in bending direction, of clusters attached to the track
	160	//!< (accounting for multiple scattering and cluster resolution)
ea94c18b	161
	162	Double_t fGlobalChi2; ///< Global chi2 of the track
	163
	164	Bool_t fImproved; //!< kTRUE if the track has been improved
	165
423b32ca	166	Int_t fMatchTrigger; ///< 0 track does not match trigger
	167	///< 1 track match but does not pass pt cut
	168	///< 2 track match Low pt cut
	169	///< 3 track match High pt cut
829425a5	170	Double_t fChi2MatchTrigger; ///< chi2 of trigger/track matching
de2cd600	171
2e2d0c44	172	Int_t fTrackID; ///< Point to the corresponding MC track
96ebe67e	173
	174	AliMUONTrackParam* fTrackParamAtVertex; //!< Track parameters at vertex
	175
7771752e	176	UShort_t fHitsPatternInTrigCh; ///< Word containing info on the hits left in trigger chambers
423b32ca	177
423b32ca	178	Int_t fLocalTrigger; ///< packed local trigger information
de2cd600	179
5c15a68b	180	Bool_t fConnected; ///< kTRUE if that track shares cluster(s) with another
ea94c18b	181
ea94c18b	182	// methods
96ebe67e	183	Bool_t ComputeClusterWeights(TMatrixD& clusterWeightsNB, TMatrixD& clusterWeightsB,
96ebe67e	184	TMatrixD* mcsCovariances = 0, AliMUONVCluster* discardedCluster = 0) const;
ea94c18b	185	void ComputeMCSCovariances(TMatrixD& mcsCovariances) const;
	186
	187
bdfb6eef	188	ClassDef(AliMUONTrack, 9) // Reconstructed track in ALICE dimuon spectrometer
de2cd600	189	};
a9e2aefa	190
a9e2aefa	191	#endif