ITS/ITSrec/AliITStrackV2.h

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