PWG2/FEMTOSCOPY/AliFemto/AliFemtoTrack.cxx

   1 ///////////////////////////////////////////////////////////////////////////
   2 //                                                                       //
   3 // AliFemtoTrack: main class holding all the necessary information       //
   4 // about a track (before the identification) that is required during     //
   5 // femtoscopic analysis. This class is filled with information from the  //
   6 // input stream by the reader. A particle has a link back to the Track   //
   7 // it was created from, so we do not copy the information.               //
   8 //                                                                       //
   9 ///////////////////////////////////////////////////////////////////////////
  10
  11 #include "AliFemtoTrack.h"
  12 #ifndef __NO_STAR_DEPENDENCE_ALLOWED__
  13 #ifdef __ROOT__
  14 #include "StEvent/StEnumerations.h"
  15 #include "AliFemtoAihongPid.h"
  16 #include "StEventUtilities/StuProbabilityPidAlgorithm.h"
  17 #endif
  18 #endif
  19
  20 #include "SystemOfUnits.h"   // has "tesla" in it
  21 //#include "AliFemtoParticleTypes.h"
  22 //#include "AliFemtoTTreeEvent.h"
  23 //#include "AliFemtoTTreeTrack.h"
  24
  25 AliFemtoTrack::AliFemtoTrack():
  26   fCharge(0),
  27   fPidProbElectron(0),
  28   fPidProbPion(0),
  29   fPidProbKaon(0),
  30   fPidProbProton(0),
  31   fPidProbMuon(0),
  32   fTrackId(0),
  33   fP(0,0,0),
  34   fPt(0),
  35   fHelix(),
  36   fFlags(0),
  37   fLabel(0),
  38   fImpactD(0),
  39   fImpactZ(0),
  40   fCdd(0),
  41   fCdz(0),
  42   fCzz(0),
  43   fITSchi2(0),
  44   fITSncls(0),
  45   fTPCchi2(0),
  46   fTPCncls(0),
  47   fTPCnclsF(0),
  48   fTPCsignalN(0),
  49   fTPCsignalS(0),
  50   fSigmaToVertex(0),
  51   fClusters(159),
  52   fShared(159),
  53   fNominalTpcEntrancePoint(0,0,0),
  54   fNominalTpcExitPoint(0,0,0),
  55   fHiddenInfo(0)
  56 {
  57   // Default constructor
  58   fHiddenInfo = NULL;
  59   fKinkIndexes[0] = 0;
  60   fKinkIndexes[1] = 0;
  61   fKinkIndexes[2] = 0;
  62   //  cout << "Created track " << this << endl;
  63 }
  64
  65
  66 AliFemtoTrack::AliFemtoTrack(const AliFemtoTrack& t) :
  67   fCharge(0),
  68   fPidProbElectron(0),
  69   fPidProbPion(0),
  70   fPidProbKaon(0),
  71   fPidProbProton(0),
  72   fPidProbMuon(0),
  73   fTrackId(0),
  74   fP(0,0,0),
  75   fPt(0),
  76   fHelix(),
  77   fFlags(0),
  78   fLabel(0),
  79   fImpactD(0),
  80   fImpactZ(0),
  81   fCdd(0),
  82   fCdz(0),
  83   fCzz(0),
  84   fITSchi2(0),
  85   fITSncls(0),
  86   fTPCchi2(0),
  87   fTPCncls(0),
  88   fTPCnclsF(0),
  89   fTPCsignalN(0),
  90   fTPCsignalS(0),
  91   fSigmaToVertex(0),
  92   fClusters(159),
  93   fShared(159),
  94   fNominalTpcEntrancePoint(0,0,0),
  95   fNominalTpcExitPoint(0,0,0),
  96   fHiddenInfo(0)
  97  {
  98    // copy constructor
  99   fCharge = t.fCharge;
 100   fPidProbElectron = t.fPidProbElectron;
 101   fPidProbPion = t.fPidProbPion;
 102   fPidProbKaon = t.fPidProbKaon;
 103   fPidProbProton = t.fPidProbProton;
 104   fPidProbMuon=t.fPidProbMuon;
 105   fP = t.fP;
 106   fPt = t.fPt;
 107   fHelix = t.fHelix;
 108   fTrackId = t.fTrackId;
 109   fFlags=t.fFlags;
 110   fLabel=t.fLabel;
 111   fImpactD=t.fImpactD;
 112   fImpactZ=t.fImpactZ;
 113   fCdd=t.fCdd;
 114   fCdz=t.fCdz;
 115   fCzz=t.fCzz;
 116   fITSchi2=t.fITSchi2;
 117   fITSncls=t.fITSncls;
 118   fTPCchi2=t.fTPCchi2;
 119   fTPCncls=t.fTPCncls;
 120   fTPCnclsF=t.fTPCnclsF;
 121   fTPCsignalN=t.fTPCsignalN;
 122   fTPCsignalS=t.fTPCsignalS;
 123   fSigmaToVertex=t.fSigmaToVertex;
 124   fClusters=t.fClusters;
 125   fShared=t.fShared;
 126   fNominalTpcEntrancePoint=t.fNominalTpcEntrancePoint;
 127   fNominalTpcExitPoint=t.fNominalTpcExitPoint;
 128   if (t.ValidHiddenInfo())
 129     fHiddenInfo = t.GetHiddenInfo()->Clone();
 130   else
 131     fHiddenInfo = NULL;
 132   fKinkIndexes[0] = t.fKinkIndexes[0];
 133   fKinkIndexes[1] = t.fKinkIndexes[1];
 134   fKinkIndexes[2] = t.fKinkIndexes[2];
 135   //  cout << "Created track " << this << endl;
 136 }
 137
 138 AliFemtoTrack& AliFemtoTrack::operator=(const AliFemtoTrack& aTrack)
 139 {
 140   // assignment operator
 141   if (this == &aTrack)
 142     return *this;
 143   fCharge = aTrack.fCharge;
 144   fPidProbElectron = aTrack.fPidProbElectron;
 145   fPidProbPion = aTrack.fPidProbPion;
 146   fPidProbKaon = aTrack.fPidProbKaon;
 147   fPidProbProton = aTrack.fPidProbProton;
 148   fPidProbMuon=aTrack.fPidProbMuon;
 149   fP = aTrack.fP;
 150   fPt = aTrack.fPt;
 151   fHelix = aTrack.fHelix;
 152   fTrackId = aTrack.fTrackId;
 153   fFlags=aTrack.fFlags;
 154   fLabel=aTrack.fLabel;
 155   fImpactD=aTrack.fImpactD;
 156   fImpactZ=aTrack.fImpactZ;
 157   fCdd=aTrack.fCdd;
 158   fCdz=aTrack.fCdz;
 159   fCzz=aTrack.fCzz;
 160   fITSchi2=aTrack.fITSchi2;
 161   fITSncls=aTrack.fITSncls;
 162   fTPCchi2=aTrack.fTPCchi2;
 163   fTPCncls=aTrack.fTPCncls;
 164   fTPCnclsF=aTrack.fTPCnclsF;
 165   fTPCsignalN=aTrack.fTPCsignalN;
 166   fTPCsignalS=aTrack.fTPCsignalS;
 167   fClusters=aTrack.fClusters;
 168   fShared=aTrack.fShared;
 169   fNominalTpcEntrancePoint=aTrack.fNominalTpcEntrancePoint;
 170   fNominalTpcExitPoint=aTrack.fNominalTpcExitPoint;
 171   fKinkIndexes[0] = aTrack.fKinkIndexes[0];
 172   fKinkIndexes[1] = aTrack.fKinkIndexes[1];
 173   fKinkIndexes[2] = aTrack.fKinkIndexes[2];
 174   if (ValidHiddenInfo())
 175     delete fHiddenInfo;
 176   if (aTrack.ValidHiddenInfo())
 177     fHiddenInfo = aTrack.GetHiddenInfo()->Clone();
 178   else
 179     fHiddenInfo = NULL;
 180
 181   return *this;
 182 }
 183
 184 void AliFemtoTrack::SetCharge(const short& ch){fCharge=ch;}
 185
 186 void AliFemtoTrack::SetPidProbElectron(const float& x){fPidProbElectron = x;}
 187 void AliFemtoTrack::SetPidProbPion(const float& x){fPidProbPion = x;}
 188 void AliFemtoTrack::SetPidProbKaon(const float& x){fPidProbKaon = x;}
 189 void AliFemtoTrack::SetPidProbProton(const float& x){fPidProbProton = x;}
 190 void AliFemtoTrack::SetPidProbMuon(const float& x){fPidProbMuon = x;}
 191
 192 void AliFemtoTrack::SetP(const AliFemtoThreeVector& p){fP = p;}
 193 void AliFemtoTrack::SetPt(const float& pt){fPt = pt;}
 194 void AliFemtoTrack::SetHelix(const AliFmPhysicalHelixD& h){fHelix = h;}
 195 void AliFemtoTrack::SetTrackId(const short & id) { fTrackId=id;}
 196 void AliFemtoTrack::SetFlags(const long int &flags) {fFlags=flags;}
 197 void AliFemtoTrack::SetLabel(const int &label) {fLabel=label;}
 198 void AliFemtoTrack::SetImpactD(const float& aImpactD){fImpactD=aImpactD;}
 199 void AliFemtoTrack::SetImpactZ(const float& aImpactZ){fImpactZ=aImpactZ;}
 200 void AliFemtoTrack::SetCdd(const float& aCdd){fCdd=aCdd;}
 201 void AliFemtoTrack::SetCdz(const float& aCdz){fCdz=aCdz;}
 202 void AliFemtoTrack::SetCzz(const float& aCzz){fCzz=aCzz;}
 203 void AliFemtoTrack::SetITSchi2(const float& aITSchi2){fITSchi2=aITSchi2;}
 204 void AliFemtoTrack::SetITSncls(const int& aITSncls){fITSncls=aITSncls;}
 205 void AliFemtoTrack::SetTPCchi2(const float& aTPCchi2){fTPCchi2=aTPCchi2;}
 206 void AliFemtoTrack::SetTPCncls(const int& aTPCncls){fTPCncls=aTPCncls;}
 207 void AliFemtoTrack::SetTPCnclsF(const short& aTPCnclsF){fTPCnclsF=aTPCnclsF;}
 208 void AliFemtoTrack::SetTPCsignalN(const short& aTPCsignalN){fTPCsignalN=aTPCsignalN;}
 209 void AliFemtoTrack::SetTPCsignalS(const float& aTPCsignalS){fTPCsignalS=aTPCsignalS;}
 210 void AliFemtoTrack::SetSigmaToVertex(const float& aSigma){fSigmaToVertex=aSigma;}
 211
 212
 213 short AliFemtoTrack::Charge() const {return fCharge;}
 214 AliFemtoThreeVector AliFemtoTrack::P() const {return fP;}
 215 float AliFemtoTrack::Pt() const {return fPt;}
 216 const AliFmPhysicalHelixD& AliFemtoTrack::Helix() const {return fHelix;}
 217 short AliFemtoTrack::TrackId() const { return fTrackId; }
 218 long int AliFemtoTrack::Flags() const {return fFlags;}
 219 int AliFemtoTrack::Label()const {return fLabel;}
 220 float AliFemtoTrack::ImpactD()const{return fImpactD;}
 221 float AliFemtoTrack::ImpactZ()const{return fImpactZ;}
 222 float AliFemtoTrack::Cdd() const{return fCdd;}
 223 float AliFemtoTrack::Cdz() const{return fCdz;}
 224 float AliFemtoTrack::Czz() const{return fCzz;}
 225 float AliFemtoTrack::ITSchi2() const{return fITSchi2;}
 226 int   AliFemtoTrack::ITSncls() const{return fITSncls;}
 227 float AliFemtoTrack::TPCchi2() const{return fTPCchi2;}
 228 int   AliFemtoTrack::TPCncls() const{return fTPCncls;}
 229 short AliFemtoTrack::TPCnclsF() const{return fTPCnclsF;}
 230 short AliFemtoTrack::TPCsignalN() const{return fTPCsignalN;}
 231 float AliFemtoTrack::TPCsignalS() const{return fTPCsignalS;}
 232 float AliFemtoTrack::SigmaToVertex() const{return fSigmaToVertex;}
 233
 234 void AliFemtoTrack::SetHiddenInfo(AliFemtoHiddenInfo* aHiddenInfo) {fHiddenInfo=aHiddenInfo;}
 235 bool AliFemtoTrack::ValidHiddenInfo() const { if (fHiddenInfo) return true; else return false; }
 236 AliFemtoHiddenInfo* AliFemtoTrack::GetHiddenInfo() const {return fHiddenInfo;}
 237
 238 AliFemtoTrack::~AliFemtoTrack()
 239 {
 240   // destructor
 241   if (fHiddenInfo)
 242     delete fHiddenInfo;
 243   //  cout << "Deleted track " << this << endl;
 244 }
 245
 246 const TBits& AliFemtoTrack::TPCclusters() const {return fClusters;}
 247 const TBits& AliFemtoTrack::TPCsharing()  const {return fShared;}
 248
 249 void AliFemtoTrack::SetTPCcluster(const short& aNBit, const Bool_t& aValue)
 250 {
 251   fClusters.SetBitNumber(aNBit, aValue);
 252 }
 253
 254 void AliFemtoTrack::SetTPCshared(const short& aNBit, const Bool_t& aValue)
 255 {
 256   fShared.SetBitNumber(aNBit, aValue);
 257 }
 258
 259 void AliFemtoTrack::SetTPCClusterMap(const TBits& aBits)
 260 {
 261   fClusters = aBits;
 262 }
 263 void AliFemtoTrack::SetTPCSharedMap(const TBits& aBits)
 264 {
 265   fShared = aBits;
 266 }
 267
 268 void AliFemtoTrack::SetKinkIndexes(int points[3])
 269 {
 270   // Transfer the Kink indices
 271   fKinkIndexes[0] = points[0];
 272   fKinkIndexes[1] = points[1];
 273   fKinkIndexes[2] = points[2];
 274 }
 275
 276 int  AliFemtoTrack::KinkIndex(int aIndex) const
 277 {
 278   // Return Kink index
 279   if ((aIndex <3) && (aIndex>=0))
 280     return fKinkIndexes[aIndex];
 281   else
 282     return 0;
 283 }
 284
 285 // void AliFemtoTrack::SetXTPC(const AliFemtoThreeVector& aXTPC)
 286 // {
 287 //   fXTPC = aXTPC;
 288 // }
 289
 290 // void AliFemtoTrack::SetXTPC(double *aXTPC)
 291 // {
 292 //   fXTPC.setX(aXTPC[0]);
 293 //   fXTPC.setY(aXTPC[1]);
 294 //   fXTPC.setZ(aXTPC[2]);
 295 // }
 296
 297 // AliFemtoThreeVector AliFemtoTrack::XTPC() const
 298 // {
 299 //   return fXTPC;
 300 // }
 301
 302 const AliFemtoThreeVector& AliFemtoTrack::NominalTpcExitPoint() const
 303 {
 304   return fNominalTpcExitPoint;
 305 }
 306 const AliFemtoThreeVector& AliFemtoTrack::NominalTpcEntrancePoint() const
 307 {
 308   return fNominalTpcEntrancePoint;
 309 }
 310
 311 void AliFemtoTrack::SetNominalTPCEntrancePoint(const AliFemtoThreeVector& aXTPC)
 312 {
 313   fNominalTpcEntrancePoint = aXTPC;
 314 }
 315 void AliFemtoTrack::SetNominalTPCEntrancePoint(double *aXTPC)
 316 {
 317   // Store the nominal TPC entrance point
 318   fNominalTpcEntrancePoint.SetX(aXTPC[0]);
 319   fNominalTpcEntrancePoint.SetY(aXTPC[1]);
 320   fNominalTpcEntrancePoint.SetZ(aXTPC[2]);
 321 }
 322
 323 void AliFemtoTrack::SetNominalTPCExitPoint(const AliFemtoThreeVector& aXTPC)
 324 {
 325   fNominalTpcExitPoint = aXTPC;
 326 }
 327 void AliFemtoTrack::SetNominalTPCExitPoint(double *aXTPC)
 328 {
 329   // Store the nominal TPC exit point
 330   fNominalTpcExitPoint.SetX(aXTPC[0]);
 331   fNominalTpcExitPoint.SetY(aXTPC[1]);
 332   fNominalTpcExitPoint.SetZ(aXTPC[2]);
 333 }