[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
//-------------------------------------------------------------------------

#include <AliKalmanTrack.h>
#include "AliITSRecoParam.h"
#include "AliITSgeomTGeo.h"

/* $Id$ */

class AliESDtrack;
class AliESDVertex;
class AliTracker;

//_____________________________________________________________________________
class AliITStrackV2 : public AliKalmanTrack {
public:
  AliITStrackV2();
  AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
  AliITStrackV2(const AliITStrackV2& t);
  ~AliITStrackV2(){fESDtrack=0;}

  void   SetCheckInvariant(Bool_t check=kTRUE) {fCheckInvariant=check;}
  Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho,
				Bool_t anglecorr=kFALSE) {
    return AliExternalTrackParam::CorrectForMeanMaterial(xOverX0,xTimesRho,GetMass(),anglecorr);
  }
  Bool_t CorrectForMaterial(Double_t d, Double_t x0=AliITSRecoParam::GetX0Air()) {
    // deprecated: use CorrectForMeanMaterial instead
    return AliExternalTrackParam::CorrectForMaterial(d,x0,GetMass());
  }
  Bool_t PropagateTo(Double_t xr, Double_t d, Double_t x0=AliITSRecoParam::GetX0Air());
  Bool_t PropagateToTGeo(Double_t xToGo, Int_t nstep, Double_t &xOverX0, Double_t &xTimesRho, Bool_t addTime=kTRUE);
  Bool_t PropagateToTGeo(Double_t xToGo, Int_t nstep=1, Bool_t addTime=kTRUE) {
    Double_t dummy1,dummy2; return PropagateToTGeo(xToGo,nstep,dummy1,dummy2,addTime);
  }
  Double_t GetPredictedChi2(const AliCluster *cluster) const;
  Bool_t Update(const AliCluster *cl, Double_t chi2, Int_t i);

  Bool_t PropagateToVertex(const AliESDVertex *v,Double_t d=0.,Double_t x0=0.);
  Bool_t Propagate(Double_t alpha, Double_t xr);
  Bool_t Propagate(Double_t xr) { return Propagate(GetAlpha(),xr); }
  Bool_t MeanBudgetToPrimVertex(Double_t xyz[3], Double_t step, Double_t &d) const;
  Bool_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);
  Float_t GetSampledEdx(Int_t i) const {return fdEdxSample[i];}
  void CookdEdx(Double_t low=0., Double_t up=0.51);
  void SetDetectorIndex(Int_t i) {SetLabel(i);}
  void ResetClusters();
  void UpdateESDtrack(ULong_t flags) const;
  
  AliESDtrack *GetESDtrack() const {return fESDtrack;}

  Int_t GetDetectorIndex() const {return GetLabel();}
  Double_t GetdEdx() const {return fdEdx;}
  Double_t GetPIDsignal() const {return GetdEdx();}
  using AliExternalTrackParam::GetC;
  Double_t GetC() const {return AliExternalTrackParam::GetC(GetBz());}
  Double_t GetD(Double_t x, Double_t y) const {
    return AliExternalTrackParam::GetD(x,y,GetBz());
  }
  void GetDZ(Double_t xv, Double_t yv, Double_t zv, Float_t dz[2]) const {
    return AliExternalTrackParam::GetDZ(xv,yv,zv,GetBz(),dz);
  }

  Bool_t GetGlobalXYZat(Double_t xloc,Double_t &x,Double_t &y,Double_t &z) const;
  Bool_t GetPhiZat(Double_t r,Double_t &phi,Double_t &z) const;
  Bool_t GetLocalXat(Double_t r,Double_t &xloc) const;

  Int_t Compare(const TObject *o) const;
  Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
  void  SetModuleIndex(Int_t ilayer,Int_t idx) {fModule[ilayer]=idx;}
  Int_t GetModuleIndex(Int_t ilayer) const {return fModule[ilayer];}
  void  SetModuleIndexInfo(Int_t ilayer,Int_t idet,Int_t status=1,Float_t xloc=0,Float_t zloc=0);
  Bool_t GetModuleIndexInfo(Int_t ilayer,Int_t &idet,Int_t &status,Float_t &xloc,Float_t &zloc) const;
  Bool_t Invariant() const;

  void  SetExtraCluster(Int_t ilayer, Int_t idx) {fIndex[AliITSgeomTGeo::kNLayers+ilayer]=idx;}
  Int_t GetExtraCluster(Int_t ilayer) const {return fIndex[AliITSgeomTGeo::kNLayers+ilayer];}

  void  SetExtraModule(Int_t ilayer, Int_t idx) {fModule[AliITSgeomTGeo::kNLayers+ilayer]=idx;}
  Int_t GetExtraModule(Int_t ilayer) const {return fModule[AliITSgeomTGeo::kNLayers+ilayer];}

protected:

  Bool_t fCheckInvariant;  // check the covariance matrix

  Double_t fdEdx;            // dE/dx

  static const Int_t fgkWARN; //! used for debugging purposes
  Float_t fdEdxSample[4];    // array of dE/dx samples b.b.

  Int_t fIndex[2*AliITSgeomTGeo::kNLayers]; // indices of associated clusters 

  Int_t fModule[2*AliITSgeomTGeo::kNLayers]; // indices of crossed modules: 
                                             // see SetModuleIndexInfo()
  AliESDtrack *fESDtrack;    //! pointer to the connected ESD track

private:
  AliITStrackV2 &operator=(const AliITStrackV2 &tr);
  ClassDef(AliITStrackV2,8)  //ITS reconstructed track
};

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;
  return;
}

inline void  AliITStrackV2::SetModuleIndexInfo(Int_t ilayer,Int_t idet,Int_t status,
					       Float_t xloc,Float_t zloc) {
  //----------------------------------------------------------------------
  // This function encodes in the module number also the status of cluster association
  // "status" can have the following values: 
  // 1 "found" (cluster is associated), 
  // 2 "dead" (module is dead from OCDB), 
  // 3 "skipped" (module or layer forced to be skipped),
  // 4 "outinz" (track out of z acceptance), 
  // 5 "nocls" (no clusters in the road), 
  // 6 "norefit" (cluster rejected during refit) 
  // 7 "deadzspd" (holes in z in SPD)
  // WARNING: THIS METHOD HAS TO BE SYNCHRONIZED WITH AliESDtrack::GetITSModuleIndexInfo()!
  //----------------------------------------------------------------------

  if(idet<0) {
    idet=0;
  } else {
    // same detector numbering as in AliITSCalib classes
    if(ilayer==1) idet+=AliITSgeomTGeo::GetNLadders(1)*AliITSgeomTGeo::GetNDetectors(1);
    if(ilayer==3) idet+=AliITSgeomTGeo::GetNLadders(3)*AliITSgeomTGeo::GetNDetectors(3);
    if(ilayer==5) idet+=AliITSgeomTGeo::GetNLadders(5)*AliITSgeomTGeo::GetNDetectors(5);
  }

  Int_t xInt = Int_t(xloc*10.);
  Int_t zInt = Int_t(zloc*10.);

  if(TMath::Abs(xloc*10.-(Float_t)xInt)>0.5){
    if(zloc>0) { 
      xInt++; 
    } 
    else { 
      xInt--; 
    }
  }
  if(TMath::Abs(zloc*10.-(Float_t)zInt)>0.5){
    if(zloc>0) { 
      zInt++; 
    } 
    else { 
      zInt--; 
    }
  }
  Int_t signs=0;
  if(xInt>=0 && zInt>=0) signs=10000;
  if(xInt>=0 && zInt<0)  signs=20000;
  if(xInt<0 && zInt>=0)  signs=30000;
  if(xInt<0 && zInt<0)   signs=40000;

  Int_t modindex = signs;
  
  modindex += TMath::Abs(zInt);
  modindex += TMath::Abs(xInt)*100;

  modindex += status*100000;

  modindex += idet*1000000;

  SetModuleIndex(ilayer,modindex);
  return;
}

inline Bool_t AliITStrackV2::GetModuleIndexInfo(Int_t ilayer,Int_t &idet,Int_t &status,
					       Float_t &xloc,Float_t &zloc) const {
  //----------------------------------------------------------------------
  // This function encodes in the module number also the status of cluster association
  // "status" can have the following values: 
  // 1 "found" (cluster is associated), 
  // 2 "dead" (module is dead from OCDB), 
  // 3 "skipped" (module or layer forced to be skipped),
  // 4 "outinz" (track out of z acceptance), 
  // 5 "nocls" (no clusters in the road), 
  // 6 "norefit" (cluster rejected during refit), 
  // 7 "deadzspd" (holes in z in SPD)
  // Also given are the coordinates of the crossing point of track and module
  // (in the local module ref. system)
  // WARNING: THIS METHOD HAS TO BE SYNCHRONIZED WITH AliESDtrack::GetITSModuleIndexInfo()!
  //----------------------------------------------------------------------


  if(fModule[ilayer]==-1) {
    AliError("fModule was not set !");
    idet = -1;
    status=0;
    xloc=-99.; zloc=-99.;
    return kFALSE;
  }

  Int_t module = fModule[ilayer];

  idet = Int_t(module/1000000);

  module -= idet*1000000;

  status = Int_t(module/100000);

  module -= status*100000;

  Int_t signs = Int_t(module/10000);

  module-=signs*10000;

  Int_t xInt = Int_t(module/100);
  module -= xInt*100;

  Int_t zInt = module;

  if(signs==1) { xInt*=1; zInt*=1; }
  if(signs==2) { xInt*=1; zInt*=-1; }
  if(signs==3) { xInt*=-1; zInt*=1; }
  if(signs==4) { xInt*=-1; zInt*=-1; }

  xloc = 0.1*(Float_t)xInt;
  zloc = 0.1*(Float_t)zInt;

  if(status==4) idet = -1;

  return kTRUE;
}


#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	//-------------------------------------------------------------------------
006b5f7f	12
006b5f7f	13	#include <AliKalmanTrack.h>
44347160	14	#include "AliITSRecoParam.h"
e50912db	15	#include "AliITSgeomTGeo.h"
006b5f7f	16
ae00569a	17	/* $Id$ */
ae00569a	18
83d73500	19	class AliESDtrack;
58e536c5	20	class AliESDVertex;
f7a1cc68	21	class AliTracker;
006b5f7f	22
006b5f7f	23	//_____________________________________________________________________________
8676d691	24	class AliITStrackV2 : public AliKalmanTrack {
006b5f7f	25	public:
22b13da0	26	AliITStrackV2();
67c3dcbe	27	AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE) throw (const Char_t *);
006b5f7f	28	AliITStrackV2(const AliITStrackV2& t);
43d2197f	29	~AliITStrackV2(){fESDtrack=0;}
6c94f330	30
545af9ef	31	void SetCheckInvariant(Bool_t check=kTRUE) {fCheckInvariant=check;}
e50912db	32	Bool_t CorrectForMeanMaterial(Double_t xOverX0, Double_t xTimesRho,
	33	Bool_t anglecorr=kFALSE) {
	34	return AliExternalTrackParam::CorrectForMeanMaterial(xOverX0,xTimesRho,GetMass(),anglecorr);
	35	}
	36	Bool_t CorrectForMaterial(Double_t d, Double_t x0=AliITSRecoParam::GetX0Air()) {
	37	// deprecated: use CorrectForMeanMaterial instead
6c94f330	38	return AliExternalTrackParam::CorrectForMaterial(d,x0,GetMass());
6c94f330	39	}
e50912db	40	Bool_t PropagateTo(Double_t xr, Double_t d, Double_t x0=AliITSRecoParam::GetX0Air());
db355ee7	41	Bool_t PropagateToTGeo(Double_t xToGo, Int_t nstep, Double_t &xOverX0, Double_t &xTimesRho, Bool_t addTime=kTRUE);
	42	Bool_t PropagateToTGeo(Double_t xToGo, Int_t nstep=1, Bool_t addTime=kTRUE) {
	43	Double_t dummy1,dummy2; return PropagateToTGeo(xToGo,nstep,dummy1,dummy2,addTime);
afd25725	44	}
6c94f330	45	Double_t GetPredictedChi2(const AliCluster *cluster) const;
	46	Bool_t Update(const AliCluster *cl, Double_t chi2, Int_t i);
	47
e50912db	48	Bool_t PropagateToVertex(const AliESDVertex *v,Double_t d=0.,Double_t x0=0.);
6c94f330	49	Bool_t Propagate(Double_t alpha, Double_t xr);
9be1d1c7	50	Bool_t Propagate(Double_t xr) { return Propagate(GetAlpha(),xr); }
afd25725	51	Bool_t MeanBudgetToPrimVertex(Double_t xyz[3], Double_t step, Double_t &d) const;
6c94f330	52	Bool_t Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]);
6c94f330	53
7f6ddf58	54	void SetdEdx(Double_t dedx) {fdEdx=dedx;}
a9a2d814	55	void SetSampledEdx(Float_t q, Int_t i);
b9671574	56	Float_t GetSampledEdx(Int_t i) const {return fdEdxSample[i];}
a9a2d814	57	void CookdEdx(Double_t low=0., Double_t up=0.51);
006b5f7f	58	void SetDetectorIndex(Int_t i) {SetLabel(i);}
7ac23097	59	void ResetClusters();
15dd636f	60	void UpdateESDtrack(ULong_t flags) const;
e43c066c	61
1aedd96e	62	AliESDtrack *GetESDtrack() const {return fESDtrack;}
e43c066c	63
006b5f7f	64	Int_t GetDetectorIndex() const {return GetLabel();}
7f6ddf58	65	Double_t GetdEdx() const {return fdEdx;}
83d73500	66	Double_t GetPIDsignal() const {return GetdEdx();}
df37c864	67	using AliExternalTrackParam::GetC;
6c94f330	68	Double_t GetC() const {return AliExternalTrackParam::GetC(GetBz());}
	69	Double_t GetD(Double_t x, Double_t y) const {
	70	return AliExternalTrackParam::GetD(x,y,GetBz());
	71	}
	72	void GetDZ(Double_t xv, Double_t yv, Double_t zv, Float_t dz[2]) const {
	73	return AliExternalTrackParam::GetDZ(xv,yv,zv,GetBz(),dz);
	74	}
	75
e50912db	76	Bool_t GetGlobalXYZat(Double_t xloc,Double_t &x,Double_t &y,Double_t &z) const;
8602c008	77	Bool_t GetPhiZat(Double_t r,Double_t &phi,Double_t &z) const;
e50912db	78	Bool_t GetLocalXat(Double_t r,Double_t &xloc) const;
6c94f330	79
006b5f7f	80	Int_t Compare(const TObject *o) const;
006b5f7f	81	Int_t GetClusterIndex(Int_t i) const {return fIndex[i];}
ae00569a	82	void SetModuleIndex(Int_t ilayer,Int_t idx) {fModule[ilayer]=idx;}
	83	Int_t GetModuleIndex(Int_t ilayer) const {return fModule[ilayer];}
	84	void SetModuleIndexInfo(Int_t ilayer,Int_t idet,Int_t status=1,Float_t xloc=0,Float_t zloc=0);
	85	Bool_t GetModuleIndexInfo(Int_t ilayer,Int_t &idet,Int_t &status,Float_t &xloc,Float_t &zloc) const;
6c94f330	86	Bool_t Invariant() const;
ef7253ac	87
ae00569a	88	void SetExtraCluster(Int_t ilayer, Int_t idx) {fIndex[AliITSgeomTGeo::kNLayers+ilayer]=idx;}
	89	Int_t GetExtraCluster(Int_t ilayer) const {return fIndex[AliITSgeomTGeo::kNLayers+ilayer];}
	90
	91	void SetExtraModule(Int_t ilayer, Int_t idx) {fModule[AliITSgeomTGeo::kNLayers+ilayer]=idx;}
	92	Int_t GetExtraModule(Int_t ilayer) const {return fModule[AliITSgeomTGeo::kNLayers+ilayer];}
c84a5e9e	93
6c94f330	94	protected:
545af9ef	95
	96	Bool_t fCheckInvariant; // check the covariance matrix
	97
6c94f330	98	Double_t fdEdx; // dE/dx
8602c008	99
8602c008	100	static const Int_t fgkWARN; //! used for debugging purposes
6c94f330	101	Float_t fdEdxSample[4]; // array of dE/dx samples b.b.
15dd636f	102
e50912db	103	Int_t fIndex[2*AliITSgeomTGeo::kNLayers]; // indices of associated clusters
15dd636f	104
ae00569a	105	Int_t fModule[2*AliITSgeomTGeo::kNLayers]; // indices of crossed modules:
ae00569a	106	// see SetModuleIndexInfo()
ef7253ac	107	AliESDtrack *fESDtrack; //! pointer to the connected ESD track
006b5f7f	108
6c94f330	109	private:
6c94f330	110	AliITStrackV2 &operator=(const AliITStrackV2 &tr);
545af9ef	111	ClassDef(AliITStrackV2,8) //ITS reconstructed track
6c94f330	112	};
a9a2d814	113
ae00569a	114	inline void AliITStrackV2::SetSampledEdx(Float_t q, Int_t i) {
23efe5f1	115	//----------------------------------------------------------------------
a9a2d814	116	// This function stores dEdx sample corrected for the track segment length
a9a2d814	117	// Origin: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
23efe5f1	118	//----------------------------------------------------------------------
a9a2d814	119	if (i<0) return;
a9a2d814	120	if (i>3) return;
23efe5f1	121	Double_t s=GetSnp(), t=GetTgl();
23efe5f1	122	q = TMath::Sqrt((1-ss)/(1+t*t));
23efe5f1	123	fdEdxSample[i]=q;
ae00569a	124	return;
23efe5f1	125	}
c84a5e9e	126
ae00569a	127	inline void AliITStrackV2::SetModuleIndexInfo(Int_t ilayer,Int_t idet,Int_t status,
	128	Float_t xloc,Float_t zloc) {
	129	//----------------------------------------------------------------------
	130	// This function encodes in the module number also the status of cluster association
	131	// "status" can have the following values:
	132	// 1 "found" (cluster is associated),
	133	// 2 "dead" (module is dead from OCDB),
	134	// 3 "skipped" (module or layer forced to be skipped),
	135	// 4 "outinz" (track out of z acceptance),
	136	// 5 "nocls" (no clusters in the road),
	137	// 6 "norefit" (cluster rejected during refit)
	138	// 7 "deadzspd" (holes in z in SPD)
	139	// WARNING: THIS METHOD HAS TO BE SYNCHRONIZED WITH AliESDtrack::GetITSModuleIndexInfo()!
	140	//----------------------------------------------------------------------
	141
	142	if(idet<0) {
	143	idet=0;
	144	} else {
	145	// same detector numbering as in AliITSCalib classes
	146	if(ilayer==1) idet+=AliITSgeomTGeo::GetNLadders(1)*AliITSgeomTGeo::GetNDetectors(1);
	147	if(ilayer==3) idet+=AliITSgeomTGeo::GetNLadders(3)*AliITSgeomTGeo::GetNDetectors(3);
	148	if(ilayer==5) idet+=AliITSgeomTGeo::GetNLadders(5)*AliITSgeomTGeo::GetNDetectors(5);
	149	}
	150
	151	Int_t xInt = Int_t(xloc*10.);
	152	Int_t zInt = Int_t(zloc*10.);
	153
5af4a2d0	154	if(TMath::Abs(xloc*10.-(Float_t)xInt)>0.5){
	155	if(zloc>0) {
	156	xInt++;
	157	}
	158	else {
	159	xInt--;
	160	}
	161	}
	162	if(TMath::Abs(zloc*10.-(Float_t)zInt)>0.5){
	163	if(zloc>0) {
	164	zInt++;
	165	}
	166	else {
	167	zInt--;
	168	}
	169	}
ae00569a	170	Int_t signs=0;
	171	if(xInt>=0 && zInt>=0) signs=10000;
	172	if(xInt>=0 && zInt<0) signs=20000;
	173	if(xInt<0 && zInt>=0) signs=30000;
	174	if(xInt<0 && zInt<0) signs=40000;
	175
	176	Int_t modindex = signs;
	177
	178	modindex += TMath::Abs(zInt);
	179	modindex += TMath::Abs(xInt)*100;
	180
	181	modindex += status*100000;
	182
	183	modindex += idet*1000000;
	184
	185	SetModuleIndex(ilayer,modindex);
	186	return;
	187	}
	188
	189	inline Bool_t AliITStrackV2::GetModuleIndexInfo(Int_t ilayer,Int_t &idet,Int_t &status,
	190	Float_t &xloc,Float_t &zloc) const {
	191	//----------------------------------------------------------------------
	192	// This function encodes in the module number also the status of cluster association
	193	// "status" can have the following values:
	194	// 1 "found" (cluster is associated),
	195	// 2 "dead" (module is dead from OCDB),
	196	// 3 "skipped" (module or layer forced to be skipped),
	197	// 4 "outinz" (track out of z acceptance),
	198	// 5 "nocls" (no clusters in the road),
	199	// 6 "norefit" (cluster rejected during refit),
	200	// 7 "deadzspd" (holes in z in SPD)
	201	// Also given are the coordinates of the crossing point of track and module
	202	// (in the local module ref. system)
	203	// WARNING: THIS METHOD HAS TO BE SYNCHRONIZED WITH AliESDtrack::GetITSModuleIndexInfo()!
	204	//----------------------------------------------------------------------
	205
	206
	207	if(fModule[ilayer]==-1) {
	208	AliError("fModule was not set !");
	209	idet = -1;
	210	status=0;
	211	xloc=-99.; zloc=-99.;
	212	return kFALSE;
	213	}
	214
	215	Int_t module = fModule[ilayer];
	216
	217	idet = Int_t(module/1000000);
	218
	219	module -= idet*1000000;
	220
	221	status = Int_t(module/100000);
	222
	223	module -= status*100000;
	224
	225	Int_t signs = Int_t(module/10000);
	226
	227	module-=signs*10000;
	228
	229	Int_t xInt = Int_t(module/100);
	230	module -= xInt*100;
	231
	232	Int_t zInt = module;
	233
234	if(signs==1) { xInt=1; zInt=1; }
235	if(signs==2) { xInt=1; zInt=-1; }
236	if(signs==3) { xInt=-1; zInt=1; }
237	if(signs==4) { xInt=-1; zInt=-1; }
238
239	xloc = 0.1*(Float_t)xInt;
240	zloc = 0.1*(Float_t)zInt;
241
242	if(status==4) idet = -1;
243
244	return kTRUE;
245	}
246
247
006b5f7f	248	#endif
	249
	250