ROOT logo
#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"
#include "AliESDtrack.h"

/* $Id$ */

class AliESDVertex;
class AliTracker;

//_____________________________________________________________________________
class AliITStrackV2 : public AliKalmanTrack {
public:
  AliITStrackV2();
  AliITStrackV2(AliESDtrack& t,Bool_t c=kFALSE);
  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]);
  Bool_t ImproveKalman(Double_t xyz[3],Double_t ers[3], const Double_t* xlMS, const Double_t* x2X0MS, Int_t nMS);
  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;}
  virtual ULong_t  GetStatus() const {
    if(fESDtrack){return fESDtrack->GetStatus();} 
    else { AliWarning("null ESD track pointer - status 0"); return 0;} }

  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());
  }
  using AliExternalTrackParam::GetDZ;
  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;
  void    SetSharedWeight(Int_t ilayer,Float_t w) {fSharedWeight[ilayer]=w;}
  Float_t GetSharedWeight(Int_t ilayer) const {return fSharedWeight[ilayer];}
  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()

  Float_t fSharedWeight[AliITSgeomTGeo::kNLayers]; // probability of sharing

  AliESDtrack *fESDtrack;    //! pointer to the connected ESD track

private:
  AliITStrackV2 &operator=(const AliITStrackV2 &tr);
  ClassDef(AliITStrackV2,9)  //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


 AliITStrackV2.h:1
 AliITStrackV2.h:2
 AliITStrackV2.h:3
 AliITStrackV2.h:4
 AliITStrackV2.h:5
 AliITStrackV2.h:6
 AliITStrackV2.h:7
 AliITStrackV2.h:8
 AliITStrackV2.h:9
 AliITStrackV2.h:10
 AliITStrackV2.h:11
 AliITStrackV2.h:12
 AliITStrackV2.h:13
 AliITStrackV2.h:14
 AliITStrackV2.h:15
 AliITStrackV2.h:16
 AliITStrackV2.h:17
 AliITStrackV2.h:18
 AliITStrackV2.h:19
 AliITStrackV2.h:20
 AliITStrackV2.h:21
 AliITStrackV2.h:22
 AliITStrackV2.h:23
 AliITStrackV2.h:24
 AliITStrackV2.h:25
 AliITStrackV2.h:26
 AliITStrackV2.h:27
 AliITStrackV2.h:28
 AliITStrackV2.h:29
 AliITStrackV2.h:30
 AliITStrackV2.h:31
 AliITStrackV2.h:32
 AliITStrackV2.h:33
 AliITStrackV2.h:34
 AliITStrackV2.h:35
 AliITStrackV2.h:36
 AliITStrackV2.h:37
 AliITStrackV2.h:38
 AliITStrackV2.h:39
 AliITStrackV2.h:40
 AliITStrackV2.h:41
 AliITStrackV2.h:42
 AliITStrackV2.h:43
 AliITStrackV2.h:44
 AliITStrackV2.h:45
 AliITStrackV2.h:46
 AliITStrackV2.h:47
 AliITStrackV2.h:48
 AliITStrackV2.h:49
 AliITStrackV2.h:50
 AliITStrackV2.h:51
 AliITStrackV2.h:52
 AliITStrackV2.h:53
 AliITStrackV2.h:54
 AliITStrackV2.h:55
 AliITStrackV2.h:56
 AliITStrackV2.h:57
 AliITStrackV2.h:58
 AliITStrackV2.h:59
 AliITStrackV2.h:60
 AliITStrackV2.h:61
 AliITStrackV2.h:62
 AliITStrackV2.h:63
 AliITStrackV2.h:64
 AliITStrackV2.h:65
 AliITStrackV2.h:66
 AliITStrackV2.h:67
 AliITStrackV2.h:68
 AliITStrackV2.h:69
 AliITStrackV2.h:70
 AliITStrackV2.h:71
 AliITStrackV2.h:72
 AliITStrackV2.h:73
 AliITStrackV2.h:74
 AliITStrackV2.h:75
 AliITStrackV2.h:76
 AliITStrackV2.h:77
 AliITStrackV2.h:78
 AliITStrackV2.h:79
 AliITStrackV2.h:80
 AliITStrackV2.h:81
 AliITStrackV2.h:82
 AliITStrackV2.h:83
 AliITStrackV2.h:84
 AliITStrackV2.h:85
 AliITStrackV2.h:86
 AliITStrackV2.h:87
 AliITStrackV2.h:88
 AliITStrackV2.h:89
 AliITStrackV2.h:90
 AliITStrackV2.h:91
 AliITStrackV2.h:92
 AliITStrackV2.h:93
 AliITStrackV2.h:94
 AliITStrackV2.h:95
 AliITStrackV2.h:96
 AliITStrackV2.h:97
 AliITStrackV2.h:98
 AliITStrackV2.h:99
 AliITStrackV2.h:100
 AliITStrackV2.h:101
 AliITStrackV2.h:102
 AliITStrackV2.h:103
 AliITStrackV2.h:104
 AliITStrackV2.h:105
 AliITStrackV2.h:106
 AliITStrackV2.h:107
 AliITStrackV2.h:108
 AliITStrackV2.h:109
 AliITStrackV2.h:110
 AliITStrackV2.h:111
 AliITStrackV2.h:112
 AliITStrackV2.h:113
 AliITStrackV2.h:114
 AliITStrackV2.h:115
 AliITStrackV2.h:116
 AliITStrackV2.h:117
 AliITStrackV2.h:118
 AliITStrackV2.h:119
 AliITStrackV2.h:120
 AliITStrackV2.h:121
 AliITStrackV2.h:122
 AliITStrackV2.h:123
 AliITStrackV2.h:124
 AliITStrackV2.h:125
 AliITStrackV2.h:126
 AliITStrackV2.h:127
 AliITStrackV2.h:128
 AliITStrackV2.h:129
 AliITStrackV2.h:130
 AliITStrackV2.h:131
 AliITStrackV2.h:132
 AliITStrackV2.h:133
 AliITStrackV2.h:134
 AliITStrackV2.h:135
 AliITStrackV2.h:136
 AliITStrackV2.h:137
 AliITStrackV2.h:138
 AliITStrackV2.h:139
 AliITStrackV2.h:140
 AliITStrackV2.h:141
 AliITStrackV2.h:142
 AliITStrackV2.h:143
 AliITStrackV2.h:144
 AliITStrackV2.h:145
 AliITStrackV2.h:146
 AliITStrackV2.h:147
 AliITStrackV2.h:148
 AliITStrackV2.h:149
 AliITStrackV2.h:150
 AliITStrackV2.h:151
 AliITStrackV2.h:152
 AliITStrackV2.h:153
 AliITStrackV2.h:154
 AliITStrackV2.h:155
 AliITStrackV2.h:156
 AliITStrackV2.h:157
 AliITStrackV2.h:158
 AliITStrackV2.h:159
 AliITStrackV2.h:160
 AliITStrackV2.h:161
 AliITStrackV2.h:162
 AliITStrackV2.h:163
 AliITStrackV2.h:164
 AliITStrackV2.h:165
 AliITStrackV2.h:166
 AliITStrackV2.h:167
 AliITStrackV2.h:168
 AliITStrackV2.h:169
 AliITStrackV2.h:170
 AliITStrackV2.h:171
 AliITStrackV2.h:172
 AliITStrackV2.h:173
 AliITStrackV2.h:174
 AliITStrackV2.h:175
 AliITStrackV2.h:176
 AliITStrackV2.h:177
 AliITStrackV2.h:178
 AliITStrackV2.h:179
 AliITStrackV2.h:180
 AliITStrackV2.h:181
 AliITStrackV2.h:182
 AliITStrackV2.h:183
 AliITStrackV2.h:184
 AliITStrackV2.h:185
 AliITStrackV2.h:186
 AliITStrackV2.h:187
 AliITStrackV2.h:188
 AliITStrackV2.h:189
 AliITStrackV2.h:190
 AliITStrackV2.h:191
 AliITStrackV2.h:192
 AliITStrackV2.h:193
 AliITStrackV2.h:194
 AliITStrackV2.h:195
 AliITStrackV2.h:196
 AliITStrackV2.h:197
 AliITStrackV2.h:198
 AliITStrackV2.h:199
 AliITStrackV2.h:200
 AliITStrackV2.h:201
 AliITStrackV2.h:202
 AliITStrackV2.h:203
 AliITStrackV2.h:204
 AliITStrackV2.h:205
 AliITStrackV2.h:206
 AliITStrackV2.h:207
 AliITStrackV2.h:208
 AliITStrackV2.h:209
 AliITStrackV2.h:210
 AliITStrackV2.h:211
 AliITStrackV2.h:212
 AliITStrackV2.h:213
 AliITStrackV2.h:214
 AliITStrackV2.h:215
 AliITStrackV2.h:216
 AliITStrackV2.h:217
 AliITStrackV2.h:218
 AliITStrackV2.h:219
 AliITStrackV2.h:220
 AliITStrackV2.h:221
 AliITStrackV2.h:222
 AliITStrackV2.h:223
 AliITStrackV2.h:224
 AliITStrackV2.h:225
 AliITStrackV2.h:226
 AliITStrackV2.h:227
 AliITStrackV2.h:228
 AliITStrackV2.h:229
 AliITStrackV2.h:230
 AliITStrackV2.h:231
 AliITStrackV2.h:232
 AliITStrackV2.h:233
 AliITStrackV2.h:234
 AliITStrackV2.h:235
 AliITStrackV2.h:236
 AliITStrackV2.h:237
 AliITStrackV2.h:238
 AliITStrackV2.h:239
 AliITStrackV2.h:240
 AliITStrackV2.h:241
 AliITStrackV2.h:242
 AliITStrackV2.h:243
 AliITStrackV2.h:244
 AliITStrackV2.h:245
 AliITStrackV2.h:246
 AliITStrackV2.h:247
 AliITStrackV2.h:248
 AliITStrackV2.h:249
 AliITStrackV2.h:250
 AliITStrackV2.h:251
 AliITStrackV2.h:252
 AliITStrackV2.h:253
 AliITStrackV2.h:254
 AliITStrackV2.h:255
 AliITStrackV2.h:256
 AliITStrackV2.h:257
 AliITStrackV2.h:258
 AliITStrackV2.h:259