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

#ifndef ALIMUONTRACKPARAM_H
#define ALIMUONTRACKPARAM_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 AliMUONTrackParam
/// \brief Track parameters in ALICE dimuon spectrometer
///
////////////////////////////////////////////////////
/// Track parameters in ALICE dimuon spectrometer
////////////////////////////////////////////////////

#include <TObject.h>
#include <TMatrixD.h>

class AliMUONVCluster;
class AliESDMuonTrack;

class AliMUONTrackParam : public TObject 
{
 public:
  AliMUONTrackParam(); // Constructor
  virtual ~AliMUONTrackParam(); // Destructor
  
  AliMUONTrackParam(const AliMUONTrackParam& theMUONTrackParam);
  AliMUONTrackParam& operator=(const  AliMUONTrackParam& theMUONTrackParam);

  void GetParamFrom(const AliESDMuonTrack& esdMuonTrack);
  void SetParamFor(AliESDMuonTrack& esdMuonTrack) const;
  void GetParamFromDCA(const AliESDMuonTrack& esdMuonTrack);
  void SetParamForDCA(AliESDMuonTrack& esdMuonTrack) const;
  void GetParamFromUncorrected(const AliESDMuonTrack& esdMuonTrack);
  void SetParamForUncorrected(AliESDMuonTrack& esdMuonTrack) const;
  
  void GetCovFrom(const AliESDMuonTrack& esdMuonTrack);
  void SetCovFor(AliESDMuonTrack& esdMuonTrack) const;

  // Get and Set methods for data
	/// return Z coordinate (cm)
  Double_t GetZ() const {return fZ;}
	/// set Z coordinate (cm)
  void     SetZ(Double_t z) {fZ = z;}
	/// return non bending coordinate (cm)
  Double_t GetNonBendingCoor() const {return fParameters(0,0);}
	/// set non bending coordinate (cm)
  void     SetNonBendingCoor(Double_t nonBendingCoor) {fParameters(0,0) = nonBendingCoor;}
	/// return non bending slope (cm ** -1)
  Double_t GetNonBendingSlope() const {return fParameters(1,0);}
	/// set non bending slope (cm ** -1)
  void     SetNonBendingSlope(Double_t nonBendingSlope) {fParameters(1,0) = nonBendingSlope;}
	/// return bending coordinate (cm)
  Double_t GetBendingCoor() const {return fParameters(2,0);}
	/// set bending coordinate (cm)
  void     SetBendingCoor(Double_t bendingCoor) {fParameters(2,0) = bendingCoor;}
	/// return bending slope (cm ** -1)
  Double_t GetBendingSlope() const {return fParameters(3,0);}
	/// set bending slope (cm ** -1)
  void     SetBendingSlope(Double_t bendingSlope) {fParameters(3,0) = bendingSlope;}
	/// return inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)
  Double_t GetInverseBendingMomentum() const {return fParameters(4,0);}
	/// set inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)
  void     SetInverseBendingMomentum(Double_t inverseBendingMomentum) {fParameters(4,0) = inverseBendingMomentum;}
	/// return the charge (assumed forward motion)
  Double_t GetCharge() const {return TMath::Sign(1.,fParameters(4,0));}
	/// set the charge (assumed forward motion)
  void     SetCharge(Double_t charge) {if (charge*fParameters(4,0) < 0.) fParameters(4,0) *= -1.;}
  
  	/// return track parameters
  const TMatrixD& GetParameters() const {return fParameters;}
  	/// set track parameters
  void            SetParameters(const TMatrixD& parameters) {fParameters = parameters;}
  	/// add track parameters
  void            AddParameters(const TMatrixD& parameters) {fParameters += parameters;}
  
  Double_t Px() const;  // return px
  Double_t Py() const;  // return py
  Double_t Pz() const;  // return pz
  Double_t P()  const;  // return total momentum

	/// return kTRUE if the covariance matrix exist, kFALSE if not
  Bool_t    CovariancesExist() const {return (fCovariances) ? kTRUE : kFALSE;}
  
  const TMatrixD& GetCovariances() const;
  void            SetCovariances(const TMatrixD& covariances);
  void            SetCovariances(const Double_t matrix[5][5]);
  void            SetVariances(const Double_t matrix[5][5]);
  void            DeleteCovariances();
  
  const TMatrixD& GetPropagator() const;
  void            ResetPropagator();
  void            UpdatePropagator(const TMatrixD& propagator);
  
  const TMatrixD& GetExtrapParameters() const;
  void            SetExtrapParameters(const TMatrixD& parameters);
  
  const TMatrixD& GetExtrapCovariances() const;
  void            SetExtrapCovariances(const TMatrixD& covariances);
  
  const TMatrixD& GetSmoothParameters() const;
  void            SetSmoothParameters(const TMatrixD& parameters);
  
  const TMatrixD& GetSmoothCovariances() const;
  void            SetSmoothCovariances(const TMatrixD& covariances);
  
        /// get pointeur to associated cluster
  AliMUONVCluster* GetClusterPtr() const {return fClusterPtr;}
	/// set pointeur to associated cluster
  void             SetClusterPtr(AliMUONVCluster* cluster, Bool_t owner = kFALSE) {fClusterPtr = cluster; fOwnCluster = owner;}
  
  	/// return kTRUE if the associated cluster can be removed from the track it belongs to
  Bool_t IsRemovable() const {return fRemovable;}
  	/// set the flag telling whether the associated cluster can be removed from the track it belongs to or not
  void   SetRemovable(Bool_t removable) {fRemovable = removable;}
  
	/// return kTRUE if the associated cluster alone in its chamber
  Bool_t IsAloneInChamber() const {return fAloneInChamber;}
	/// set the flag telling whether the associated hi alone in its chamber or not
  void   SetAloneInChamber(Bool_t aloneInChamber) {fAloneInChamber = aloneInChamber;}
  
  /// return the chi2 of the track when the associated cluster was attached
  Double_t GetTrackChi2() const {return fTrackChi2;}
  	/// set the chi2 of the track when the associated cluster was attached
  void     SetTrackChi2(Double_t chi2) {fTrackChi2 = chi2;}
  	/// return the local chi2 of the associated cluster with respect to the track
  Double_t GetLocalChi2() const {return fLocalChi2;}
  	/// set the local chi2 of the associated cluster with respect to the track
  void     SetLocalChi2(Double_t chi2) {fLocalChi2 = chi2;}
  
	/// necessary for sorting TClonesArray of AliMUONTrackParam
  Bool_t IsSortable () const {return kTRUE;}
  Int_t Compare(const TObject* trackParam) const;

  Bool_t CompatibleTrackParam(const AliMUONTrackParam &trackParam, Double_t sigma2Cut, Double_t &normChi2) const;

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

 private:
  
  Double_t fZ; ///< Z coordinate (cm)
  
  /// Track parameters ordered as follow:      <pre>
  /// X       = Non bending coordinate   (cm)
  /// SlopeX  = Non bending slope        (cm ** -1)
  /// Y       = Bending coordinate       (cm)
  /// SlopeY  = Bending slope            (cm ** -1)
  /// InvP_yz = Inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)  </pre>
  TMatrixD fParameters; ///< \brief Track parameters 
  
  /// Covariance matrix of track parameters, ordered as follow:      <pre>
  ///    <X,X>      <X,SlopeX>        <X,Y>      <X,SlopeY>       <X,InvP_yz>
  /// <X,SlopeX>  <SlopeX,SlopeX>  <Y,SlopeX>  <SlopeX,SlopeY>  <SlopeX,InvP_yz>
  ///    <X,Y>      <Y,SlopeX>        <Y,Y>      <Y,SlopeY>       <Y,InvP_yz>
  /// <X,SlopeY>  <SlopeX,SlopeY>  <Y,SlopeY>  <SlopeY,SlopeY>  <SlopeY,InvP_yz>
  /// <X,InvP_yz> <SlopeX,InvP_yz> <Y,InvP_yz> <SlopeY,InvP_yz> <InvP_yz,InvP_yz>  </pre>
  mutable TMatrixD *fCovariances; ///< \brief Covariance matrix of track parameters 
  
  mutable TMatrixD *fPropagator;        //!< Jacobian used to extrapolate the track parameters and covariances to the actual z position
  mutable TMatrixD *fExtrapParameters;  //!< Track parameters extrapolated to the actual z position (not filtered by Kalman)
  mutable TMatrixD *fExtrapCovariances; //!< Covariance matrix extrapolated to the actual z position (not filtered by Kalman)
  
  mutable TMatrixD *fSmoothParameters;  //!< Track parameters obtained using smoother
  mutable TMatrixD *fSmoothCovariances; //!< Covariance matrix obtained using smoother
  
  AliMUONVCluster *fClusterPtr; //!< Pointer to associated cluster if any
  Bool_t           fOwnCluster; //!< Ownership of the associated cluster
  
  Bool_t fRemovable; //!< kTRUE if the associated cluster can be removed from the track it belongs to
  
  Bool_t fAloneInChamber; //!< kTRUE if the associated cluster is alone in its chamber

  Double_t fTrackChi2; //!< Chi2 of the track when the associated cluster was attached
  Double_t fLocalChi2; //!< Local chi2 of the associated cluster with respect to the track
  
  ClassDef(AliMUONTrackParam, 4) // Track parameters in ALICE dimuon spectrometer
};
	
#endif
Commit	Line	Data
a9e2aefa	1	#ifndef ALIMUONTRACKPARAM_H
	2	#define ALIMUONTRACKPARAM_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
a9e2aefa	8
692de412	9	/// \ingroup rec
	10	/// \class AliMUONTrackParam
	11	/// \brief Track parameters in ALICE dimuon spectrometer
	12	///
	13	////////////////////////////////////////////////////
	14	/// Track parameters in ALICE dimuon spectrometer
	15	////////////////////////////////////////////////////
a9e2aefa	16
3831f268	17	#include <TObject.h>
ea94c18b	18	#include <TMatrixD.h>
a9e2aefa	19
96ebe67e	20	class AliMUONVCluster;
211c52eb	21	class AliESDMuonTrack;
211c52eb	22
30178c30	23	class AliMUONTrackParam : public TObject
30178c30	24	{
a9e2aefa	25	public:
30178c30	26	AliMUONTrackParam(); // Constructor
de2cd600	27	virtual ~AliMUONTrackParam(); // Destructor
61adb9bd	28
de2cd600	29	AliMUONTrackParam(const AliMUONTrackParam& theMUONTrackParam);
de2cd600	30	AliMUONTrackParam& operator=(const AliMUONTrackParam& theMUONTrackParam);
211c52eb	31
211c52eb	32	void GetParamFrom(const AliESDMuonTrack& esdMuonTrack);
ea94c18b	33	void SetParamFor(AliESDMuonTrack& esdMuonTrack) const;
61fed964	34	void GetParamFromDCA(const AliESDMuonTrack& esdMuonTrack);
61fed964	35	void SetParamForDCA(AliESDMuonTrack& esdMuonTrack) const;
22ccc301	36	void GetParamFromUncorrected(const AliESDMuonTrack& esdMuonTrack);
ea94c18b	37	void SetParamForUncorrected(AliESDMuonTrack& esdMuonTrack) const;
60765b06	38
	39	void GetCovFrom(const AliESDMuonTrack& esdMuonTrack);
	40	void SetCovFor(AliESDMuonTrack& esdMuonTrack) const;
211c52eb	41
a9e2aefa	42	// Get and Set methods for data
2457f726	43	/// return Z coordinate (cm)
ea94c18b	44	Double_t GetZ() const {return fZ;}
2457f726	45	/// set Z coordinate (cm)
208f139e	46	void SetZ(Double_t z) {fZ = z;}
2457f726	47	/// return non bending coordinate (cm)
ea94c18b	48	Double_t GetNonBendingCoor() const {return fParameters(0,0);}
2457f726	49	/// set non bending coordinate (cm)
ea94c18b	50	void SetNonBendingCoor(Double_t nonBendingCoor) {fParameters(0,0) = nonBendingCoor;}
	51	/// return non bending slope (cm ** -1)
	52	Double_t GetNonBendingSlope() const {return fParameters(1,0);}
	53	/// set non bending slope (cm ** -1)
	54	void SetNonBendingSlope(Double_t nonBendingSlope) {fParameters(1,0) = nonBendingSlope;}
	55	/// return bending coordinate (cm)
	56	Double_t GetBendingCoor() const {return fParameters(2,0);}
	57	/// set bending coordinate (cm)
	58	void SetBendingCoor(Double_t bendingCoor) {fParameters(2,0) = bendingCoor;}
	59	/// return bending slope (cm ** -1)
	60	Double_t GetBendingSlope() const {return fParameters(3,0);}
	61	/// set bending slope (cm ** -1)
	62	void SetBendingSlope(Double_t bendingSlope) {fParameters(3,0) = bendingSlope;}
	63	/// return inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)
	64	Double_t GetInverseBendingMomentum() const {return fParameters(4,0);}
	65	/// set inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion)
	66	void SetInverseBendingMomentum(Double_t inverseBendingMomentum) {fParameters(4,0) = inverseBendingMomentum;}
	67	/// return the charge (assumed forward motion)
	68	Double_t GetCharge() const {return TMath::Sign(1.,fParameters(4,0));}
	69	/// set the charge (assumed forward motion)
	70	void SetCharge(Double_t charge) {if (chargefParameters(4,0) < 0.) fParameters(4,0) = -1.;}
208f139e	71
ea94c18b	72	/// return track parameters
	73	const TMatrixD& GetParameters() const {return fParameters;}
	74	/// set track parameters
	75	void SetParameters(const TMatrixD& parameters) {fParameters = parameters;}
	76	/// add track parameters
	77	void AddParameters(const TMatrixD& parameters) {fParameters += parameters;}
de2cd600	78
6464217e	79	Double_t Px() const; // return px
	80	Double_t Py() const; // return py
	81	Double_t Pz() const; // return pz
	82	Double_t P() const; // return total momentum
a9e2aefa	83
208f139e	84	/// return kTRUE if the covariance matrix exist, kFALSE if not
ea94c18b	85	Bool_t CovariancesExist() const {return (fCovariances) ? kTRUE : kFALSE;}
	86
	87	const TMatrixD& GetCovariances() const;
	88	void SetCovariances(const TMatrixD& covariances);
	89	void SetCovariances(const Double_t matrix[5][5]);
	90	void SetVariances(const Double_t matrix[5][5]);
	91	void DeleteCovariances();
	92
	93	const TMatrixD& GetPropagator() const;
	94	void ResetPropagator();
	95	void UpdatePropagator(const TMatrixD& propagator);
	96
	97	const TMatrixD& GetExtrapParameters() const;
	98	void SetExtrapParameters(const TMatrixD& parameters);
	99
	100	const TMatrixD& GetExtrapCovariances() const;
	101	void SetExtrapCovariances(const TMatrixD& covariances);
	102
	103	const TMatrixD& GetSmoothParameters() const;
	104	void SetSmoothParameters(const TMatrixD& parameters);
	105
	106	const TMatrixD& GetSmoothCovariances() const;
	107	void SetSmoothCovariances(const TMatrixD& covariances);
	108
96ebe67e	109	/// get pointeur to associated cluster
	110	AliMUONVCluster* GetClusterPtr() const {return fClusterPtr;}
	111	/// set pointeur to associated cluster
	112	void SetClusterPtr(AliMUONVCluster* cluster, Bool_t owner = kFALSE) {fClusterPtr = cluster; fOwnCluster = owner;}
ea94c18b	113
96ebe67e	114	/// return kTRUE if the associated cluster can be removed from the track it belongs to
ea94c18b	115	Bool_t IsRemovable() const {return fRemovable;}
96ebe67e	116	/// set the flag telling whether the associated cluster can be removed from the track it belongs to or not
ea94c18b	117	void SetRemovable(Bool_t removable) {fRemovable = removable;}
ea94c18b	118
96ebe67e	119	/// return kTRUE if the associated cluster alone in its chamber
b709ac13	120	Bool_t IsAloneInChamber() const {return fAloneInChamber;}
	121	/// set the flag telling whether the associated hi alone in its chamber or not
	122	void SetAloneInChamber(Bool_t aloneInChamber) {fAloneInChamber = aloneInChamber;}
	123
96ebe67e	124	/// return the chi2 of the track when the associated cluster was attached
ea94c18b	125	Double_t GetTrackChi2() const {return fTrackChi2;}
96ebe67e	126	/// set the chi2 of the track when the associated cluster was attached
ea94c18b	127	void SetTrackChi2(Double_t chi2) {fTrackChi2 = chi2;}
96ebe67e	128	/// return the local chi2 of the associated cluster with respect to the track
ea94c18b	129	Double_t GetLocalChi2() const {return fLocalChi2;}
96ebe67e	130	/// set the local chi2 of the associated cluster with respect to the track
ea94c18b	131	void SetLocalChi2(Double_t chi2) {fLocalChi2 = chi2;}
ea94c18b	132
96ebe67e	133	/// necessary for sorting TClonesArray of AliMUONTrackParam
2457f726	134	Bool_t IsSortable () const {return kTRUE;}
208f139e	135	Int_t Compare(const TObject* trackParam) const;
de2cd600	136
61fed964	137	Bool_t CompatibleTrackParam(const AliMUONTrackParam &trackParam, Double_t sigma2Cut, Double_t &normChi2) const;
61fed964	138
1a38e749	139	virtual void Print(Option_t* opt="") const;
1a38e749	140
686772dc	141	virtual void Clear(Option_t* opt="");
1a38e749	142
a9e2aefa	143	private:
ea94c18b	144
829425a5	145	Double_t fZ; ///< Z coordinate (cm)
208f139e	146
ea94c18b	147	/// Track parameters ordered as follow: <pre>
	148	/// X = Non bending coordinate (cm)
	149	/// SlopeX = Non bending slope (cm ** -1)
	150	/// Y = Bending coordinate (cm)
	151	/// SlopeY = Bending slope (cm ** -1)
	152	/// InvP_yz = Inverse bending momentum (GeV/c ** -1) times the charge (assumed forward motion) </pre>
	153	TMatrixD fParameters; ///< \brief Track parameters
	154
b58638a9	155	/// Covariance matrix of track parameters, ordered as follow: <pre>
208f139e	156	/// <X,X> <X,SlopeX> <X,Y> <X,SlopeY> <X,InvP_yz>
	157	/// <X,SlopeX> <SlopeX,SlopeX> <Y,SlopeX> <SlopeX,SlopeY> <SlopeX,InvP_yz>
	158	/// <X,Y> <Y,SlopeX> <Y,Y> <Y,SlopeY> <Y,InvP_yz>
	159	/// <X,SlopeY> <SlopeX,SlopeY> <Y,SlopeY> <SlopeY,SlopeY> <SlopeY,InvP_yz>
b58638a9	160	/// <X,InvP_yz> <SlopeX,InvP_yz> <Y,InvP_yz> <SlopeY,InvP_yz> <InvP_yz,InvP_yz> </pre>
686772dc	161	mutable TMatrixD *fCovariances; ///< \brief Covariance matrix of track parameters
208f139e	162
96ebe67e	163	mutable TMatrixD *fPropagator; //!< Jacobian used to extrapolate the track parameters and covariances to the actual z position
96ebe67e	164	mutable TMatrixD *fExtrapParameters; //!< Track parameters extrapolated to the actual z position (not filtered by Kalman)
ea94c18b	165	mutable TMatrixD *fExtrapCovariances; //!< Covariance matrix extrapolated to the actual z position (not filtered by Kalman)
ea94c18b	166
96ebe67e	167	mutable TMatrixD *fSmoothParameters; //!< Track parameters obtained using smoother
ea94c18b	168	mutable TMatrixD *fSmoothCovariances; //!< Covariance matrix obtained using smoother
ea94c18b	169
96ebe67e	170	AliMUONVCluster *fClusterPtr; //!< Pointer to associated cluster if any
96ebe67e	171	Bool_t fOwnCluster; //!< Ownership of the associated cluster
de2cd600	172
96ebe67e	173	Bool_t fRemovable; //!< kTRUE if the associated cluster can be removed from the track it belongs to
ea94c18b	174
96ebe67e	175	Bool_t fAloneInChamber; //!< kTRUE if the associated cluster is alone in its chamber
b709ac13	176
96ebe67e	177	Double_t fTrackChi2; //!< Chi2 of the track when the associated cluster was attached
96ebe67e	178	Double_t fLocalChi2; //!< Local chi2 of the associated cluster with respect to the track
ea94c18b	179
ea94c18b	180	ClassDef(AliMUONTrackParam, 4) // Track parameters in ALICE dimuon spectrometer
37827b29	181	};
a9e2aefa	182
a9e2aefa	183	#endif