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   fClusters(159),
  51   fShared(159),
  52   fHiddenInfo(0)
  53 {
  54   // Default constructor
  55   fHiddenInfo = NULL;
  56   //  cout << "Created track " << this << endl;
  57 }
  58
  59
  60 AliFemtoTrack::AliFemtoTrack(const AliFemtoTrack& t) :
  61   fCharge(0),
  62   fPidProbElectron(0),
  63   fPidProbPion(0),
  64   fPidProbKaon(0),
  65   fPidProbProton(0),
  66   fPidProbMuon(0),
  67   fTrackId(0),
  68   fP(0,0,0),
  69   fPt(0),
  70   fHelix(),
  71   fFlags(0),
  72   fLabel(0),
  73   fImpactD(0),
  74   fImpactZ(0),
  75   fCdd(0),
  76   fCdz(0),
  77   fCzz(0),
  78   fITSchi2(0),
  79   fITSncls(0),
  80   fTPCchi2(0),
  81   fTPCncls(0),
  82   fTPCnclsF(0),
  83   fTPCsignalN(0),
  84   fTPCsignalS(0),
  85   fClusters(159),
  86   fShared(159),
  87   fHiddenInfo(0)
  88  {
  89    // copy constructor
  90   fCharge = t.fCharge;
  91   fPidProbElectron = t.fPidProbElectron;
  92   fPidProbPion = t.fPidProbPion;
  93   fPidProbKaon = t.fPidProbKaon;
  94   fPidProbProton = t.fPidProbProton;
  95   fPidProbMuon=t.fPidProbMuon;
  96   fP = t.fP;
  97   fPt = t.fPt;
  98   fHelix = t.fHelix;
  99   fTrackId = t.fTrackId;
 100   fFlags=t.fFlags;
 101   fLabel=t.fLabel;
 102   fImpactD=t.fImpactD;
 103   fImpactZ=t.fImpactZ;
 104   fCdd=t.fCdd;
 105   fCdz=t.fCdz;
 106   fCzz=t.fCzz;
 107   fITSchi2=t.fITSchi2;
 108   fITSncls=t.fITSncls;
 109   fTPCchi2=t.fTPCchi2;
 110   fTPCncls=t.fTPCncls;
 111   fTPCnclsF=t.fTPCnclsF;
 112   fTPCsignalN=t.fTPCsignalN;
 113   fTPCsignalS=t.fTPCsignalS;
 114   fClusters=t.fClusters;
 115   fShared=t.fShared;
 116   if (t.ValidHiddenInfo())
 117     fHiddenInfo = t.GetHiddenInfo()->Clone();
 118   else
 119     fHiddenInfo = NULL;
 120   //  cout << "Created track " << this << endl;
 121 };
 122
 123 AliFemtoTrack& AliFemtoTrack::operator=(const AliFemtoTrack& aTrack)
 124 {
 125   // assignment operator
 126   if (this == &aTrack)
 127     return *this;
 128   fCharge = aTrack.fCharge;
 129   fPidProbElectron = aTrack.fPidProbElectron;
 130   fPidProbPion = aTrack.fPidProbPion;
 131   fPidProbKaon = aTrack.fPidProbKaon;
 132   fPidProbProton = aTrack.fPidProbProton;
 133   fPidProbMuon=aTrack.fPidProbMuon;
 134   fP = aTrack.fP;
 135   fPt = aTrack.fPt;
 136   fHelix = aTrack.fHelix;
 137   fTrackId = aTrack.fTrackId;
 138   fFlags=aTrack.fFlags;
 139   fLabel=aTrack.fLabel;
 140   fImpactD=aTrack.fImpactD;
 141   fImpactZ=aTrack.fImpactZ;
 142   fCdd=aTrack.fCdd;
 143   fCdz=aTrack.fCdz;
 144   fCzz=aTrack.fCzz;
 145   fITSchi2=aTrack.fITSchi2;
 146   fITSncls=aTrack.fITSncls;
 147   fTPCchi2=aTrack.fTPCchi2;
 148   fTPCncls=aTrack.fTPCncls;
 149   fTPCnclsF=aTrack.fTPCnclsF;
 150   fTPCsignalN=aTrack.fTPCsignalN;
 151   fTPCsignalS=aTrack.fTPCsignalS;
 152   fClusters=aTrack.fClusters;
 153   fShared=aTrack.fShared;
 154   if (ValidHiddenInfo())
 155     delete fHiddenInfo;
 156   if (aTrack.ValidHiddenInfo())
 157     fHiddenInfo = aTrack.GetHiddenInfo()->Clone();
 158   else
 159     fHiddenInfo = NULL;
 160
 161   return *this;
 162 }
 163
 164 void AliFemtoTrack::SetCharge(const short& ch){fCharge=ch;}
 165
 166 void AliFemtoTrack::SetPidProbElectron(const float& x){fPidProbElectron = x;}
 167 void AliFemtoTrack::SetPidProbPion(const float& x){fPidProbPion = x;}
 168 void AliFemtoTrack::SetPidProbKaon(const float& x){fPidProbKaon = x;}
 169 void AliFemtoTrack::SetPidProbProton(const float& x){fPidProbProton = x;}
 170 void AliFemtoTrack::SetPidProbMuon(const float& x){fPidProbMuon = x;}
 171
 172 void AliFemtoTrack::SetP(const AliFemtoThreeVector& p){fP = p;}
 173 void AliFemtoTrack::SetPt(const float& pt){fPt = pt;}
 174 void AliFemtoTrack::SetHelix(const AliFmPhysicalHelixD& h){fHelix = h;}
 175 void AliFemtoTrack::SetTrackId(const short & id) { fTrackId=id;}
 176 void AliFemtoTrack::SetFlags(const long int &flags) {fFlags=flags;}
 177 void AliFemtoTrack::SetLabel(const int &label) {fLabel=label;}
 178 void AliFemtoTrack::SetImpactD(const float& ImpactD){fImpactD=ImpactD;}
 179 void AliFemtoTrack::SetImpactZ(const float& ImpactZ){fImpactZ=ImpactZ;}
 180 void AliFemtoTrack::SetCdd(const float& Cdd){fCdd=Cdd;}
 181 void AliFemtoTrack::SetCdz(const float& Cdz){fCdz=Cdz;}
 182 void AliFemtoTrack::SetCzz(const float& Czz){fCzz=Czz;}
 183 void AliFemtoTrack::SetITSchi2(const float& ITSchi2){fITSchi2=ITSchi2;}
 184 void AliFemtoTrack::SetITSncls(const int& ITSncls){fITSncls=ITSncls;}
 185 void AliFemtoTrack::SetTPCchi2(const float& TPCchi2){fTPCchi2=TPCchi2;}
 186 void AliFemtoTrack::SetTPCncls(const int& TPCncls){fTPCncls=TPCncls;}
 187 void AliFemtoTrack::SetTPCnclsF(const short& TPCnclsF){fTPCnclsF=TPCnclsF;}
 188 void AliFemtoTrack::SetTPCsignalN(const short& TPCsignalN){fTPCsignalN=TPCsignalN;}
 189 void AliFemtoTrack::SetTPCsignalS(const float& TPCsignalS){fTPCsignalS=TPCsignalS;}
 190
 191
 192 short AliFemtoTrack::Charge() const {return fCharge;}
 193 AliFemtoThreeVector AliFemtoTrack::P() const {return fP;}
 194 float AliFemtoTrack::Pt() const {return fPt;}
 195 const AliFmPhysicalHelixD& AliFemtoTrack::Helix() const {return fHelix;}
 196 short AliFemtoTrack::TrackId() const { return fTrackId; }
 197 long int AliFemtoTrack::Flags() const {return fFlags;}
 198 int AliFemtoTrack::Label()const {return fLabel;}
 199 float AliFemtoTrack::ImpactD()const{return fImpactD;}
 200 float AliFemtoTrack::ImpactZ()const{return fImpactZ;}
 201 float AliFemtoTrack::Cdd() const{return fCdd;}
 202 float AliFemtoTrack::Cdz() const{return fCdz;}
 203 float AliFemtoTrack::Czz() const{return fCzz;}
 204 float AliFemtoTrack::ITSchi2() const{return fITSchi2;}
 205 int   AliFemtoTrack::ITSncls() const{return fITSncls;}
 206 float AliFemtoTrack::TPCchi2() const{return fTPCchi2;}
 207 int   AliFemtoTrack::TPCncls() const{return fTPCncls;}
 208 short AliFemtoTrack::TPCnclsF() const{return fTPCnclsF;}
 209 short AliFemtoTrack::TPCsignalN() const{return fTPCsignalN;}
 210 float AliFemtoTrack::TPCsignalS() const{return fTPCsignalS;}
 211
 212 void AliFemtoTrack::SetHiddenInfo(AliFemtoHiddenInfo* aHiddenInfo) {fHiddenInfo=aHiddenInfo;}
 213 bool AliFemtoTrack::ValidHiddenInfo() const { if (fHiddenInfo) return true; else return false; }
 214 AliFemtoHiddenInfo* AliFemtoTrack::GetHiddenInfo() const {return fHiddenInfo;}
 215
 216 AliFemtoTrack::~AliFemtoTrack()
 217 {
 218   if (fHiddenInfo)
 219     delete fHiddenInfo;
 220   //  cout << "Deleted track " << this << endl;
 221 }
 222
 223 const TBits& AliFemtoTrack::TPCclusters() const {return fClusters;}
 224 const TBits& AliFemtoTrack::TPCsharing()  const {return fShared;}
 225
 226 void AliFemtoTrack::SetTPCcluster(const short& aNBit, const Bool_t& aValue)
 227 {
 228   fClusters.SetBitNumber(aNBit, aValue);
 229 }
 230
 231 void AliFemtoTrack::SetTPCshared(const short& aNBit, const Bool_t& aValue)
 232 {
 233   fShared.SetBitNumber(aNBit, aValue);
 234 }
 235