[u/mrichter/AliRoot.git] / ITS / AliITStrackV2.h

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

//-------------------------------------------------------------------------
//                       ITS Track Class
//
//        Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
//     dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
//-------------------------------------------------------------------------


/*****************************************************************************
 *                          December 18, 2000                                *
 *  Internal view of the ITS track parametrisation as well as the order of   *
 *           track parameters are subject for possible changes !             *
 *  Use GetExternalParameters() and GetExternalCovariance() to access ITS    *
 *      track information regardless of its internal representation.         *
 * This formation is now fixed in the following way:                         *
 *      external param0:   local Y-coordinate of a track (cm)                *
 *      external param1:   local Z-coordinate of a track (cm)                *
 *      external param2:   local sine of the track momentum azimuthal angle  *
 *      external param3:   tangent of the track momentum dip angle           *
 *      external param4:   1/pt (1/(GeV/c))                                  *
 *****************************************************************************/

#include <AliKalmanTrack.h>

#include "AliITSrecoV2.h"

class AliESDtrack;
class AliTPCtrack;

//_____________________________________________________________________________
class AliITStrackV2 : public AliKalmanTrack {
  friend class AliITStrackerV2;
  friend class AliITStrackerMI;
public:
  AliITStrackV2();
  AliITStrackV2(const AliTPCtrack& t) throw (const Char_t *);
  AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
  AliITStrackV2(const AliITStrackV2& t);
  Int_t PropagateToVertex(Double_t d=0., Double_t x0=0.);
  Int_t Propagate(Double_t alpha, Double_t xr);
  Int_t GetProlongationFast(Double_t alpha, Double_t xr,Double_t &y, Double_t &z);
  Int_t CorrectForMaterial(Double_t d, Double_t x0=21.82);
  Int_t PropagateTo(Double_t xr, Double_t d, Double_t x0=21.82);
  Int_t Update(const AliCluster* cl,Double_t chi2,UInt_t i);
  Int_t UpdateMI(Double_t cy, Double_t cz, Double_t cerry, Double_t cerrz, Double_t chi2,UInt_t i);  
  Int_t Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]);
  void SetdEdx(Double_t dedx) {fdEdx=dedx;}
  void SetSampledEdx(Float_t q, Int_t i);
  void CookdEdx(Double_t low=0., Double_t up=0.51);
  void SetDetectorIndex(Int_t i) {SetLabel(i);}
  void ResetCovariance();
  void ResetClusters() { SetChi2(0.); SetNumberOfClusters(0); }
  void UpdateESDtrack(ULong_t flags);
  void SetConstrainedESDtrack(Double_t chi2); 
  void SetReconstructed(Bool_t sr=kTRUE){fReconstructed = sr;}
  
  Bool_t GetReconstructed() const {return fReconstructed;}
  void SetChi2MIP(Int_t i,Float_t val){fChi2MIP[i]=val;}
  Float_t GetChi2MIP(Int_t i) const {return fChi2MIP[i];}  
  void IncrementNSkipped(){fNSkipped++;} // increment by 1 the # of skipped cls
  Float_t GetNSkipped() const {return fNSkipped;}
  void IncrementNUsed(){fNUsed++;} // increment by 1 the # of shared clusters
  Float_t GetNUsed() const {return fNUsed;}
  AliESDtrack *GetESDtrack() const {return fESDtrack;}
  Double_t GetCov33() const {return fC33;} // cov. matrix el. 3,3
  Double_t GetCov44() const {return fC44;} // cov. matrix el. 4,4
  Float_t GetDy(Int_t i) const {return fDy[i];}
  Float_t GetDz(Int_t i) const {return fDz[i];}
  Float_t GetSigmaY(Int_t i) const {return fSigmaY[i];}
  Float_t GetSigmaZ(Int_t i) const {return fSigmaZ[i];}


  Int_t GetDetectorIndex() const {return GetLabel();}
  Double_t GetX()    const {return fX;}
  Double_t GetAlpha()const {return fAlpha;}
  Double_t GetdEdx() const {return fdEdx;}
  Double_t GetPIDsignal() const {return GetdEdx();}
  Double_t GetY()    const {return fP0;}
  Double_t GetZ()    const {return fP1;}
  Double_t GetSnp()  const {return fP2;}
  Double_t GetTgl()  const {return fP3;}
  Double_t GetC()    const {return fP4;}
  Double_t
    Get1Pt() const { return (1e-9*TMath::Abs(fP4)/fP4 + fP4)*GetConvConst(); }
  Double_t GetD(Double_t x=0, Double_t y=0) const;
  Double_t GetZat(Double_t x=0) const;

  Double_t GetSigmaY2() const {return fC00;}
  Double_t GetSigmaZ2() const {return fC11;}
  Int_t Compare(const TObject *o) const;
  void GetExternalParameters(Double_t& xr, Double_t x[5]) const ;
  void GetExternalCovariance(Double_t cov[15]) const ;
  Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
  Int_t GetGlobalXYZat(Double_t r,Double_t &x,Double_t &y,Double_t &z) const;
  Double_t GetPredictedChi2(const AliCluster *cluster) const;
  Double_t GetPredictedChi2MI(Double_t cy, Double_t cz, Double_t cerry, Double_t cerrz) const;
  Int_t Invariant() const;
 
protected:
  Double_t fX;              // X-coordinate of this track (reference plane)
  Double_t fAlpha;          // rotation angle

  Double_t fdEdx;           // dE/dx

  Double_t fP0;             // Y-coordinate of a track 
  Double_t fP1;             // Z-coordinate of a track
  Double_t fP2;             // sine of the track momentum azimuthal angle
  Double_t fP3;             // tangent of the track momentum dip angle
  Double_t fP4;             // track curvature

  Double_t fC00;                         // covariance
  Double_t fC10, fC11;                   // matrix
  Double_t fC20, fC21, fC22;             // of the
  Double_t fC30, fC31, fC32, fC33;       // track
  Double_t fC40, fC41, fC42, fC43, fC44; // parameters 
  Float_t fNUsed;                          // number of shared clusters
  Float_t fNSkipped;                       // number of skipped clusters
  Float_t fNDeadZone;                     // number of clusters in dead zone
  Float_t fDeadZoneProbability;          // probability to cross dead zone
  Bool_t fReconstructed;                 // reconstructed - accepted flag
  Float_t fChi2MIP[12];                   // MIP chi squres 
  UInt_t fIndex[kMaxLayer]; // indices of associated clusters 
  Float_t fdEdxSample[4];   // array of dE/dx samples b.b.
  Float_t fDy[12];           //dy in layer
  Float_t fDz[12];           //dz in layer
  Float_t fSigmaY[12];       //sigma y 
  Float_t fSigmaZ[12];       //sigma z
  Float_t fNy[6];              //expected size of cluster
  Float_t fNz[6];              //expected size of cluster
  Float_t fD[2];            //distance to the vertex
  Float_t fNormQ[6];        // normalized Q
  Float_t fExpQ;            // expected Q
  Float_t fNormChi2[6];     // normalized chi2 
  Float_t fChi22;           // chi22
  Float_t fdEdxMismatch;    
  Bool_t fConstrain;        //indication of the vertex constrain
  Int_t  fClIndex[6];       //cluster Index
  AliESDtrack *fESDtrack;   //! pointer to the connected ESD track
  ClassDef(AliITStrackV2,3)   //ITS reconstructed track
};

inline 
void AliITStrackV2::GetExternalParameters(Double_t& xr, Double_t x[5]) const {
  //---------------------------------------------------------------------
  // This function return external ITS track representation
  //---------------------------------------------------------------------
     xr=fX;          
     x[0]=GetY(); x[1]=GetZ(); x[2]=GetSnp(); x[3]=GetTgl(); x[4]=Get1Pt();
}

inline
void AliITStrackV2::SetSampledEdx(Float_t q, Int_t i) {
  //----------------------------------------------------------------------
  // This function stores dEdx sample corrected for the track segment length 
  // Origin: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
  //----------------------------------------------------------------------
  if (i<0) return;
  if (i>3) return;
  Double_t s=GetSnp(), t=GetTgl();
  q *= TMath::Sqrt((1-s*s)/(1+t*t));
  fdEdxSample[i]=q;
}
#endif
Commit	Line	Data
006b5f7f	1	#ifndef ALIITSTRACKV2_H
	2	#define ALIITSTRACKV2_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	//-------------------------------------------------------------------------
	7	// ITS Track Class
	8	//
	9	// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
a9a2d814	10	// dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
006b5f7f	11	//-------------------------------------------------------------------------
	12
	13
	14	/*****************************************************************************
	15	* December 18, 2000 *
	16	* Internal view of the ITS track parametrisation as well as the order of *
	17	* track parameters are subject for possible changes ! *
	18	* Use GetExternalParameters() and GetExternalCovariance() to access ITS *
	19	* track information regardless of its internal representation. *
	20	* This formation is now fixed in the following way: *
	21	* external param0: local Y-coordinate of a track (cm) *
	22	* external param1: local Z-coordinate of a track (cm) *
	23	* external param2: local sine of the track momentum azimuthal angle *
	24	* external param3: tangent of the track momentum dip angle *
	25	* external param4: 1/pt (1/(GeV/c)) *
	26	*****************************************************************************/
	27
	28	#include <AliKalmanTrack.h>
	29
	30	#include "AliITSrecoV2.h"
	31
83d73500	32	class AliESDtrack;
e43c066c	33	class AliTPCtrack;
006b5f7f	34
006b5f7f	35	//_____________________________________________________________________________
8676d691	36	class AliITStrackV2 : public AliKalmanTrack {
e43c066c	37	friend class AliITStrackerV2;
e43c066c	38	friend class AliITStrackerMI;
006b5f7f	39	public:
22b13da0	40	AliITStrackV2();
e43c066c	41	AliITStrackV2(const AliTPCtrack& t) throw (const Char_t *);
67c3dcbe	42	AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
006b5f7f	43	AliITStrackV2(const AliITStrackV2& t);
880f41b9	44	Int_t PropagateToVertex(Double_t d=0., Double_t x0=0.);
a9a2d814	45	Int_t Propagate(Double_t alpha, Double_t xr);
e43c066c	46	Int_t GetProlongationFast(Double_t alpha, Double_t xr,Double_t &y, Double_t &z);
a9a2d814	47	Int_t CorrectForMaterial(Double_t d, Double_t x0=21.82);
a9a2d814	48	Int_t PropagateTo(Double_t xr, Double_t d, Double_t x0=21.82);
7f6ddf58	49	Int_t Update(const AliCluster* cl,Double_t chi2,UInt_t i);
e43c066c	50	Int_t UpdateMI(Double_t cy, Double_t cz, Double_t cerry, Double_t cerrz, Double_t chi2,UInt_t i);
8676d691	51	Int_t Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]);
7f6ddf58	52	void SetdEdx(Double_t dedx) {fdEdx=dedx;}
a9a2d814	53	void SetSampledEdx(Float_t q, Int_t i);
a9a2d814	54	void CookdEdx(Double_t low=0., Double_t up=0.51);
006b5f7f	55	void SetDetectorIndex(Int_t i) {SetLabel(i);}
7f6ddf58	56	void ResetCovariance();
7f6ddf58	57	void ResetClusters() { SetChi2(0.); SetNumberOfClusters(0); }
83d73500	58	void UpdateESDtrack(ULong_t flags);
babd135a	59	void SetConstrainedESDtrack(Double_t chi2);
1aedd96e	60	void SetReconstructed(Bool_t sr=kTRUE){fReconstructed = sr;}
e43c066c	61
1aedd96e	62	Bool_t GetReconstructed() const {return fReconstructed;}
	63	void SetChi2MIP(Int_t i,Float_t val){fChi2MIP[i]=val;}
	64	Float_t GetChi2MIP(Int_t i) const {return fChi2MIP[i];}
	65	void IncrementNSkipped(){fNSkipped++;} // increment by 1 the # of skipped cls
e43c066c	66	Float_t GetNSkipped() const {return fNSkipped;}
1aedd96e	67	void IncrementNUsed(){fNUsed++;} // increment by 1 the # of shared clusters
e43c066c	68	Float_t GetNUsed() const {return fNUsed;}
1aedd96e	69	AliESDtrack *GetESDtrack() const {return fESDtrack;}
	70	Double_t GetCov33() const {return fC33;} // cov. matrix el. 3,3
	71	Double_t GetCov44() const {return fC44;} // cov. matrix el. 4,4
	72	Float_t GetDy(Int_t i) const {return fDy[i];}
	73	Float_t GetDz(Int_t i) const {return fDz[i];}
	74	Float_t GetSigmaY(Int_t i) const {return fSigmaY[i];}
	75	Float_t GetSigmaZ(Int_t i) const {return fSigmaZ[i];}
	76
e43c066c	77
e43c066c	78
006b5f7f	79	Int_t GetDetectorIndex() const {return GetLabel();}
	80	Double_t GetX() const {return fX;}
	81	Double_t GetAlpha()const {return fAlpha;}
7f6ddf58	82	Double_t GetdEdx() const {return fdEdx;}
83d73500	83	Double_t GetPIDsignal() const {return GetdEdx();}
006b5f7f	84	Double_t GetY() const {return fP0;}
	85	Double_t GetZ() const {return fP1;}
	86	Double_t GetSnp() const {return fP2;}
	87	Double_t GetTgl() const {return fP3;}
67c3dcbe	88	Double_t GetC() const {return fP4;}
77f353c4	89	Double_t
77f353c4	90	Get1Pt() const { return (1e-9TMath::Abs(fP4)/fP4 + fP4)GetConvConst(); }
a6a6f799	91	Double_t GetD(Double_t x=0, Double_t y=0) const;
babd135a	92	Double_t GetZat(Double_t x=0) const;
babd135a	93
006b5f7f	94	Double_t GetSigmaY2() const {return fC00;}
006b5f7f	95	Double_t GetSigmaZ2() const {return fC11;}
006b5f7f	96	Int_t Compare(const TObject *o) const;
006b5f7f	97	void GetExternalParameters(Double_t& xr, Double_t x[5]) const ;
006b5f7f	98	void GetExternalCovariance(Double_t cov[15]) const ;
006b5f7f	99	Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
006b5f7f	100	Int_t GetGlobalXYZat(Double_t r,Double_t &x,Double_t &y,Double_t &z) const;
006b5f7f	101	Double_t GetPredictedChi2(const AliCluster *cluster) const;
e43c066c	102	Double_t GetPredictedChi2MI(Double_t cy, Double_t cz, Double_t cerry, Double_t cerrz) const;
006b5f7f	103	Int_t Invariant() const;
006b5f7f	104
235b705f	105	protected:
006b5f7f	106	Double_t fX; // X-coordinate of this track (reference plane)
	107	Double_t fAlpha; // rotation angle
	108
	109	Double_t fdEdx; // dE/dx
	110
	111	Double_t fP0; // Y-coordinate of a track
	112	Double_t fP1; // Z-coordinate of a track
	113	Double_t fP2; // sine of the track momentum azimuthal angle
	114	Double_t fP3; // tangent of the track momentum dip angle
	115	Double_t fP4; // track curvature
	116
	117	Double_t fC00; // covariance
	118	Double_t fC10, fC11; // matrix
	119	Double_t fC20, fC21, fC22; // of the
	120	Double_t fC30, fC31, fC32, fC33; // track
	121	Double_t fC40, fC41, fC42, fC43, fC44; // parameters
e43c066c	122	Float_t fNUsed; // number of shared clusters
	123	Float_t fNSkipped; // number of skipped clusters
	124	Float_t fNDeadZone; // number of clusters in dead zone
	125	Float_t fDeadZoneProbability; // probability to cross dead zone
babd135a	126	Bool_t fReconstructed; // reconstructed - accepted flag
e43c066c	127	Float_t fChi2MIP[12]; // MIP chi squres
006b5f7f	128	UInt_t fIndex[kMaxLayer]; // indices of associated clusters
a9a2d814	129	Float_t fdEdxSample[4]; // array of dE/dx samples b.b.
e43c066c	130	Float_t fDy[12]; //dy in layer
	131	Float_t fDz[12]; //dz in layer
	132	Float_t fSigmaY[12]; //sigma y
	133	Float_t fSigmaZ[12]; //sigma z
	134	Float_t fNy[6]; //expected size of cluster
	135	Float_t fNz[6]; //expected size of cluster
	136	Float_t fD[2]; //distance to the vertex
	137	Float_t fNormQ[6]; // normalized Q
	138	Float_t fExpQ; // expected Q
	139	Float_t fNormChi2[6]; // normalized chi2
	140	Float_t fChi22; // chi22
	141	Float_t fdEdxMismatch;
	142	Bool_t fConstrain; //indication of the vertex constrain
	143	Int_t fClIndex[6]; //cluster Index
83d73500	144	AliESDtrack *fESDtrack; //! pointer to the connected ESD track
babd135a	145	ClassDef(AliITStrackV2,3) //ITS reconstructed track
006b5f7f	146	};
	147
	148	inline
	149	void AliITStrackV2::GetExternalParameters(Double_t& xr, Double_t x[5]) const {
	150	//---------------------------------------------------------------------
83d73500	151	// This function return external ITS track representation
006b5f7f	152	//---------------------------------------------------------------------
	153	xr=fX;
	154	x[0]=GetY(); x[1]=GetZ(); x[2]=GetSnp(); x[3]=GetTgl(); x[4]=Get1Pt();
	155	}
a9a2d814	156
23efe5f1	157	inline
	158	void AliITStrackV2::SetSampledEdx(Float_t q, Int_t i) {
	159	//----------------------------------------------------------------------
a9a2d814	160	// This function stores dEdx sample corrected for the track segment length
a9a2d814	161	// Origin: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
23efe5f1	162	//----------------------------------------------------------------------
a9a2d814	163	if (i<0) return;
a9a2d814	164	if (i>3) return;
23efe5f1	165	Double_t s=GetSnp(), t=GetTgl();
23efe5f1	166	q = TMath::Sqrt((1-ss)/(1+t*t));
23efe5f1	167	fdEdxSample[i]=q;
23efe5f1	168	}
006b5f7f	169	#endif
	170
	171