[u/mrichter/AliRoot.git] / STEER / AliExternalTrackParam.h

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

/* $Id$ */

/*****************************************************************************
 *              "External" track parametrisation class                       *
 *                                                                           *
 *      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))                                  *
 *                                                                           *
 * The parameters are estimated at an exact position x in a local coord.     *
 * system rotated by angle alpha with respect to the global coord.system.    *
 *        Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch                     *
 *****************************************************************************/
#include "TMath.h"

#include "AliVTrack.h"

const Double_t kVeryBig=1./kAlmost0;
const Double_t kMostProbablePt=0.35;

class AliVVertex;
class TPolyMarker3D; 

class AliExternalTrackParam: public AliVTrack {
 public:
  AliExternalTrackParam();
  AliExternalTrackParam(const AliExternalTrackParam &);
  AliExternalTrackParam& operator=(const AliExternalTrackParam & trkPar);
  AliExternalTrackParam(Double_t x, Double_t alpha, 
			const Double_t param[5], const Double_t covar[15]);
  AliExternalTrackParam(const AliVTrack *vTrack);
  AliExternalTrackParam(Double_t xyz[3],Double_t pxpypz[3],
			Double_t cv[21],Short_t sign);
  virtual ~AliExternalTrackParam(){}

  template <typename T>
  void Set(T x, T alpha, const T param[5], const T covar[15]) {
    //  Sets the parameters
    fX=x; fAlpha=alpha;
    for (Int_t i = 0; i < 5; i++)  fP[i] = param[i];
    for (Int_t i = 0; i < 15; i++) fC[i] = covar[i];
  }

  void Set(Double_t xyz[3],Double_t pxpypz[3],Double_t cv[21],Short_t sign);

  static void SetMostProbablePt(Double_t pt) { fgMostProbablePt=pt; }
  static Double_t GetMostProbablePt() { return fgMostProbablePt; }

  void Reset();
  void ResetCovariance(Double_t s2);
  void AddCovariance(const Double_t cov[15]);

  const Double_t *GetParameter() const {return fP;}
  const Double_t *GetCovariance() const {return fC;}

  Double_t GetAlpha() const {return fAlpha;}
  Double_t GetX() const {return fX;}
  Double_t GetY()    const {return fP[0];}
  Double_t GetZ()    const {return fP[1];}
  Double_t GetSnp()  const {return fP[2];}
  Double_t GetTgl()  const {return fP[3];}
  Double_t GetSigned1Pt()  const {return fP[4];}

  Double_t GetSigmaY2() const {return fC[0];}
  Double_t GetSigmaZY() const {return fC[1];}
  Double_t GetSigmaZ2() const {return fC[2];}
  Double_t GetSigmaSnpY() const {return fC[3];}
  Double_t GetSigmaSnpZ() const {return fC[4];}
  Double_t GetSigmaSnp2() const {return fC[5];}
  Double_t GetSigmaTglY() const {return fC[6];}
  Double_t GetSigmaTglZ() const {return fC[7];}
  Double_t GetSigmaTglSnp() const {return fC[8];}
  Double_t GetSigmaTgl2() const {return fC[9];}
  Double_t GetSigma1PtY() const {return fC[10];}
  Double_t GetSigma1PtZ() const {return fC[11];}
  Double_t GetSigma1PtSnp() const {return fC[12];}
  Double_t GetSigma1PtTgl() const {return fC[13];}
  Double_t GetSigma1Pt2() const {return fC[14];}

  // additional functions for AliVParticle
  Double_t Px() const;
  Double_t Py() const;
  Double_t Pz() const;
  Double_t Pt() const { return TMath::Abs(GetSignedPt()); }
  Double_t P() const { return GetP(); }
  Bool_t   PxPyPz(Double_t p[3]) const { return GetPxPyPz(p); }
  
  Double_t Xv() const;
  Double_t Yv() const;
  Double_t Zv() const;
  Bool_t   XvYvZv(Double_t x[3]) const { return GetXYZ(x); }

  Double_t OneOverPt() const { return 1./Pt(); }
  Double_t Phi() const;
  Double_t Theta() const;
  virtual Double_t E() const;
  virtual Double_t M() const;
  Double_t Eta() const;
  virtual Double_t Y() const;
  virtual Short_t  Charge() const { return (Short_t)GetSign(); }
  virtual const Double_t *PID() const { return 0x0; }

  // additional functions from AliVTrack
  virtual Int_t    GetID() const { return -999; }
  virtual UChar_t  GetITSClusterMap() const {return 0; }
  virtual ULong_t  GetStatus() const { return 0; }

  Double_t GetSign() const {return (fP[4]>0) ? 1 : -1;}
  Double_t GetP() const;
  Double_t GetSignedPt() const {
    return (TMath::Abs(fP[4])>kAlmost0) ? 1./fP[4]:TMath::Sign(kVeryBig,fP[4]);
  }
  Double_t Get1P() const;
  virtual Double_t GetC(Double_t b) const {return fP[4]*b*kB2C;}
  void GetDZ(Double_t x,Double_t y,Double_t z,Double_t b,Float_t dz[2]) const; 
  Double_t GetD(Double_t xv, Double_t yv, Double_t b) const; 
  Double_t GetLinearD(Double_t xv, Double_t yv) const; 
  Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho, 
        Double_t mass,  Bool_t anglecorr=kFALSE,
	Double_t (*f)(Double_t)=AliExternalTrackParam::BetheBlochSolid);

  //
  // Bethe-Bloch formula parameterizations
  //
  static Double_t BetheBlochAleph(Double_t bg,
                                  Double_t kp1=0.76176e-1,
                                  Double_t kp2=10.632,
                                  Double_t kp3=0.13279e-4,
                                  Double_t kp4=1.8631,
                                  Double_t kp5=1.9479
				  );
  static Double_t BetheBlochGeant(Double_t bg,
                                  Double_t kp0=2.33,
                                  Double_t kp1=0.20,
                                  Double_t kp2=3.00,
                                  Double_t kp3=173e-9,
                                  Double_t kp4=0.49848
				  );
    
  static Double_t BetheBlochSolid(Double_t bg);
  static Double_t BetheBlochGas(Double_t bg);

  Double_t GetPredictedChi2(Double_t p[2],Double_t cov[3]) const;

  Double_t 
    GetPredictedChi2(Double_t p[3],Double_t covyz[3],Double_t covxyz[3]) const;
  Bool_t 
    PropagateTo(Double_t p[3],Double_t covyz[3],Double_t covxyz[3],Double_t b);

  Double_t *GetResiduals(Double_t *p,Double_t *cov,Bool_t updated=kTRUE) const;
  Bool_t Update(Double_t p[2],Double_t cov[3]);
  Bool_t Rotate(Double_t alpha);
  Bool_t PropagateTo(Double_t x, Double_t b);
  Bool_t Propagate(Double_t alpha, Double_t x, Double_t b);
  void   Propagate(Double_t len,Double_t x[3],Double_t p[3],Double_t bz) const;
  Bool_t Intersect(Double_t pnt[3], Double_t norm[3], Double_t bz) const;

  void GetHelixParameters(Double_t h[6], Double_t b) const;
  Double_t GetDCA(const AliExternalTrackParam *p, Double_t b,
    Double_t &xthis,Double_t &xp) const;
  Double_t PropagateToDCA(AliExternalTrackParam *p, Double_t b);
  Bool_t PropagateToDCA(const AliVVertex *vtx, Double_t b, Double_t maxd,
                        Double_t dz[2]=0, Double_t cov[3]=0);
  
  void GetDirection(Double_t d[3]) const;
  Bool_t GetPxPyPz(Double_t *p) const;
  Bool_t GetXYZ(Double_t *p) const;
  Bool_t GetCovarianceXYZPxPyPz(Double_t cv[21]) const;
  Bool_t GetPxPyPzAt(Double_t x, Double_t b, Double_t p[3]) const;
  Bool_t GetXYZAt(Double_t x, Double_t b, Double_t r[3]) const;
  Bool_t GetYAt(Double_t x,  Double_t b,  Double_t &y) const;
  Bool_t GetZAt(Double_t x,  Double_t b,  Double_t &z) const;
  void Print(Option_t* option = "") const;
  Double_t GetSnpAt(Double_t x,Double_t b) const;

  //Deprecated
  Bool_t CorrectForMaterial(Double_t d, Double_t x0, Double_t mass,
	 Double_t (*f)(Double_t)=AliExternalTrackParam::BetheBlochSolid);

  Bool_t GetDistance(AliExternalTrackParam *param2, Double_t x, Double_t dist[3], Double_t b);
  Int_t GetIndex(Int_t i, Int_t j) const;
  Int_t GetLabel() const {return -1;} 
  //
  // visualization (M. Ivanov)
  //
  virtual void FillPolymarker(TPolyMarker3D *pol, Float_t magf, Float_t minR, Float_t maxR, Float_t stepR);
  virtual void DrawTrack(Float_t magF, Float_t minR, Float_t maxR, Float_t stepR);
 protected:
  Double_t &Par(Int_t i) {return fP[i];}
  Double_t &Cov(Int_t i) {return fC[i];}
  
 private:
  Double32_t           fX;     // X coordinate for the point of parametrisation
  Double32_t           fAlpha; // Local <-->global coor.system rotation angle
  Double32_t           fP[5];  // The track parameters
  Double32_t           fC[15]; // The track parameter covariance matrix

  static Double32_t    fgMostProbablePt; // "Most probable" pt
                                         // (to be used if Bz=0)
  ClassDef(AliExternalTrackParam, 8)
};

inline void AliExternalTrackParam::ResetCovariance(Double_t s2) {
  //
  // Reset the covarince matrix to "something big"
  //
    fC[0]*= s2;
    fC[1] = 0.;  fC[2]*= s2;
    fC[3] = 0.;  fC[4] = 0.;  fC[5]*= s2;
    fC[6] = 0.;  fC[7] = 0.;  fC[8] = 0.;  fC[9]*= s2;
    fC[10]= 0.;  fC[11]= 0.;  fC[12]= 0.;  fC[13]= 0.;  fC[14]*=s2;
}


#endif
Commit	Line	Data
51ad6848	1	#ifndef ALIEXTERNALTRACKPARAM_H
	2	#define ALIEXTERNALTRACKPARAM_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
49d13e89	8	/*****************************************************************************
	9	* "External" track parametrisation class *
	10	* *
	11	* external param0: local Y-coordinate of a track (cm) *
	12	* external param1: local Z-coordinate of a track (cm) *
	13	* external param2: local sine of the track momentum azimuthal angle *
	14	* external param3: tangent of the track momentum dip angle *
	15	* external param4: 1/pt (1/(GeV/c)) *
	16	* *
	17	* The parameters are estimated at an exact position x in a local coord. *
	18	* system rotated by angle alpha with respect to the global coord.system. *
	19	* Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch *
	20	*****************************************************************************/
6c94f330	21	#include "TMath.h"
a5e407e9	22
4f6e22bd	23	#include "AliVTrack.h"
def9660e	24
49d13e89	25	const Double_t kVeryBig=1./kAlmost0;
7b5ef2e6	26	const Double_t kMostProbablePt=0.35;
5773defd	27
58e536c5	28	class AliVVertex;
0c19adf7	29	class TPolyMarker3D;
c683ddc2	30
4f6e22bd	31	class AliExternalTrackParam: public AliVTrack {
51ad6848	32	public:
51ad6848	33	AliExternalTrackParam();
6c94f330	34	AliExternalTrackParam(const AliExternalTrackParam &);
def9660e	35	AliExternalTrackParam& operator=(const AliExternalTrackParam & trkPar);
51ad6848	36	AliExternalTrackParam(Double_t x, Double_t alpha,
51ad6848	37	const Double_t param[5], const Double_t covar[15]);
4f6e22bd	38	AliExternalTrackParam(const AliVTrack *vTrack);
da4e3deb	39	AliExternalTrackParam(Double_t xyz[3],Double_t pxpypz[3],
da4e3deb	40	Double_t cv[21],Short_t sign);
1530f89c	41	virtual ~AliExternalTrackParam(){}
ac3eaff4	42
1443179a	43	template <typename T>
	44	void Set(T x, T alpha, const T param[5], const T covar[15]) {
	45	// Sets the parameters
	46	fX=x; fAlpha=alpha;
	47	for (Int_t i = 0; i < 5; i++) fP[i] = param[i];
	48	for (Int_t i = 0; i < 15; i++) fC[i] = covar[i];
	49	}
	50
4f6e22bd	51	void Set(Double_t xyz[3],Double_t pxpypz[3],Double_t cv[21],Short_t sign);
ed5f2849	52
	53	static void SetMostProbablePt(Double_t pt) { fgMostProbablePt=pt; }
	54	static Double_t GetMostProbablePt() { return fgMostProbablePt; }
	55
c9ec41e8	56	void Reset();
ed5f2849	57	void ResetCovariance(Double_t s2);
3775b0ca	58	void AddCovariance(const Double_t cov[15]);
c9ec41e8	59
49d13e89	60	const Double_t *GetParameter() const {return fP;}
49d13e89	61	const Double_t *GetCovariance() const {return fC;}
6c94f330	62
6c94f330	63	Double_t GetAlpha() const {return fAlpha;}
c7bafca9	64	Double_t GetX() const {return fX;}
2d57349e	65	Double_t GetY() const {return fP[0];}
2d57349e	66	Double_t GetZ() const {return fP[1];}
6c94f330	67	Double_t GetSnp() const {return fP[2];}
6c94f330	68	Double_t GetTgl() const {return fP[3];}
def9660e	69	Double_t GetSigned1Pt() const {return fP[4];}
6c94f330	70
	71	Double_t GetSigmaY2() const {return fC[0];}
	72	Double_t GetSigmaZY() const {return fC[1];}
	73	Double_t GetSigmaZ2() const {return fC[2];}
	74	Double_t GetSigmaSnpY() const {return fC[3];}
	75	Double_t GetSigmaSnpZ() const {return fC[4];}
	76	Double_t GetSigmaSnp2() const {return fC[5];}
	77	Double_t GetSigmaTglY() const {return fC[6];}
	78	Double_t GetSigmaTglZ() const {return fC[7];}
	79	Double_t GetSigmaTglSnp() const {return fC[8];}
	80	Double_t GetSigmaTgl2() const {return fC[9];}
	81	Double_t GetSigma1PtY() const {return fC[10];}
	82	Double_t GetSigma1PtZ() const {return fC[11];}
	83	Double_t GetSigma1PtSnp() const {return fC[12];}
	84	Double_t GetSigma1PtTgl() const {return fC[13];}
	85	Double_t GetSigma1Pt2() const {return fC[14];}
	86
def9660e	87	// additional functions for AliVParticle
	88	Double_t Px() const;
	89	Double_t Py() const;
	90	Double_t Pz() const;
	91	Double_t Pt() const { return TMath::Abs(GetSignedPt()); }
	92	Double_t P() const { return GetP(); }
c683ddc2	93	Bool_t PxPyPz(Double_t p[3]) const { return GetPxPyPz(p); }
	94
	95	Double_t Xv() const;
	96	Double_t Yv() const;
	97	Double_t Zv() const;
	98	Bool_t XvYvZv(Double_t x[3]) const { return GetXYZ(x); }
	99
def9660e	100	Double_t OneOverPt() const { return 1./Pt(); }
	101	Double_t Phi() const;
	102	Double_t Theta() const;
	103	virtual Double_t E() const;
	104	virtual Double_t M() const;
	105	Double_t Eta() const;
	106	virtual Double_t Y() const;
2258e165	107	virtual Short_t Charge() const { return (Short_t)GetSign(); }
67be2d29	108	virtual const Double_t *PID() const { return 0x0; }
def9660e	109
4f6e22bd	110	// additional functions from AliVTrack
	111	virtual Int_t GetID() const { return -999; }
	112	virtual UChar_t GetITSClusterMap() const {return 0; }
	113	virtual ULong_t GetStatus() const { return 0; }
	114
c9ec41e8	115	Double_t GetSign() const {return (fP[4]>0) ? 1 : -1;}
c9ec41e8	116	Double_t GetP() const;
def9660e	117	Double_t GetSignedPt() const {
6c94f330	118	return (TMath::Abs(fP[4])>kAlmost0) ? 1./fP[4]:TMath::Sign(kVeryBig,fP[4]);
6c94f330	119	}
1d99986f	120	Double_t Get1P() const;
2258e165	121	virtual Double_t GetC(Double_t b) const {return fP[4]bkB2C;}
1530f89c	122	void GetDZ(Double_t x,Double_t y,Double_t z,Double_t b,Float_t dz[2]) const;
c7bafca9	123	Double_t GetD(Double_t xv, Double_t yv, Double_t b) const;
49d13e89	124	Double_t GetLinearD(Double_t xv, Double_t yv) const;
116b445b	125	Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho,
7dded1d5	126	Double_t mass, Bool_t anglecorr=kFALSE,
d46683db	127	Double_t (*f)(Double_t)=AliExternalTrackParam::BetheBlochSolid);
d46683db	128
9c56b409	129	//
	130	// Bethe-Bloch formula parameterizations
	131	//
	132	static Double_t BetheBlochAleph(Double_t bg,
	133	Double_t kp1=0.76176e-1,
	134	Double_t kp2=10.632,
	135	Double_t kp3=0.13279e-4,
	136	Double_t kp4=1.8631,
	137	Double_t kp5=1.9479
	138	);
	139	static Double_t BetheBlochGeant(Double_t bg,
	140	Double_t kp0=2.33,
	141	Double_t kp1=0.20,
	142	Double_t kp2=3.00,
	143	Double_t kp3=173e-9,
	144	Double_t kp4=0.49848
	145	);
	146
d46683db	147	static Double_t BetheBlochSolid(Double_t bg);
	148	static Double_t BetheBlochGas(Double_t bg);
	149
49d13e89	150	Double_t GetPredictedChi2(Double_t p[2],Double_t cov[3]) const;
1e023a36	151
4b189f98	152	Double_t
4b189f98	153	GetPredictedChi2(Double_t p[3],Double_t covyz[3],Double_t covxyz[3]) const;
1e023a36	154	Bool_t
	155	PropagateTo(Double_t p[3],Double_t covyz[3],Double_t covxyz[3],Double_t b);
	156
e23a38cb	157	Double_t GetResiduals(Double_t p,Double_t *cov,Bool_t updated=kTRUE) const;
49d13e89	158	Bool_t Update(Double_t p[2],Double_t cov[3]);
	159	Bool_t Rotate(Double_t alpha);
	160	Bool_t PropagateTo(Double_t x, Double_t b);
9f2bec63	161	Bool_t Propagate(Double_t alpha, Double_t x, Double_t b);
052daaff	162	void Propagate(Double_t len,Double_t x[3],Double_t p[3],Double_t bz) const;
052daaff	163	Bool_t Intersect(Double_t pnt[3], Double_t norm[3], Double_t bz) const;
49d13e89	164
c7bafca9	165	void GetHelixParameters(Double_t h[6], Double_t b) const;
	166	Double_t GetDCA(const AliExternalTrackParam *p, Double_t b,
	167	Double_t &xthis,Double_t &xp) const;
	168	Double_t PropagateToDCA(AliExternalTrackParam *p, Double_t b);
58e536c5	169	Bool_t PropagateToDCA(const AliVVertex *vtx, Double_t b, Double_t maxd,
e99a34df	170	Double_t dz[2]=0, Double_t cov[3]=0);
e99a34df	171
b1149664	172	void GetDirection(Double_t d[3]) const;
c9ec41e8	173	Bool_t GetPxPyPz(Double_t *p) const;
	174	Bool_t GetXYZ(Double_t *p) const;
	175	Bool_t GetCovarianceXYZPxPyPz(Double_t cv[21]) const;
	176	Bool_t GetPxPyPzAt(Double_t x, Double_t b, Double_t p[3]) const;
	177	Bool_t GetXYZAt(Double_t x, Double_t b, Double_t r[3]) const;
7cf7bb6c	178	Bool_t GetYAt(Double_t x, Double_t b, Double_t &y) const;
6c94f330	179	Bool_t GetZAt(Double_t x, Double_t b, Double_t &z) const;
c9ec41e8	180	void Print(Option_t* option = "") const;
c194ba83	181	Double_t GetSnpAt(Double_t x,Double_t b) const;
6c94f330	182
116b445b	183	//Deprecated
116b445b	184	Bool_t CorrectForMaterial(Double_t d, Double_t x0, Double_t mass,
d46683db	185	Double_t (*f)(Double_t)=AliExternalTrackParam::BetheBlochSolid);
116b445b	186
bf00ebb8	187	Bool_t GetDistance(AliExternalTrackParam *param2, Double_t x, Double_t dist[3], Double_t b);
0e8460af	188	Int_t GetIndex(Int_t i, Int_t j) const;
3c43fb2b	189	Int_t GetLabel() const {return -1;}
0c19adf7	190	//
	191	// visualization (M. Ivanov)
	192	//
	193	virtual void FillPolymarker(TPolyMarker3D *pol, Float_t magf, Float_t minR, Float_t maxR, Float_t stepR);
	194	virtual void DrawTrack(Float_t magF, Float_t minR, Float_t maxR, Float_t stepR);
	195	protected:
6c94f330	196	Double_t &Par(Int_t i) {return fP[i];}
6c94f330	197	Double_t &Cov(Int_t i) {return fC[i];}
0c19adf7	198
0c19adf7	199	private:
bad5ac2c	200	Double32_t fX; // X coordinate for the point of parametrisation
	201	Double32_t fAlpha; // Local <-->global coor.system rotation angle
	202	Double32_t fP[5]; // The track parameters
	203	Double32_t fC[15]; // The track parameter covariance matrix
c9ec41e8	204
ed5f2849	205	static Double32_t fgMostProbablePt; // "Most probable" pt
ed5f2849	206	// (to be used if Bz=0)
4f6e22bd	207	ClassDef(AliExternalTrackParam, 8)
51ad6848	208	};
51ad6848	209
ed5f2849	210	inline void AliExternalTrackParam::ResetCovariance(Double_t s2) {
	211	//
	212	// Reset the covarince matrix to "something big"
	213	//
	214	fC[0]*= s2;
	215	fC[1] = 0.; fC[2]*= s2;
	216	fC[3] = 0.; fC[4] = 0.; fC[5]*= s2;
	217	fC[6] = 0.; fC[7] = 0.; fC[8] = 0.; fC[9]*= s2;
	218	fC[10]= 0.; fC[11]= 0.; fC[12]= 0.; fC[13]= 0.; fC[14]*=s2;
	219	}
	220
bf00ebb8	221
	222
	223
51ad6848	224	#endif