PWG3/vertexingHF/AliAODPidHF.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  *                                                                        *
   5  * Author: The ALICE Off-line Project.                                    *
   6  * Contributors are mentioned in the code where appropriate.              *
   7  *
   8  * Permission to use, copy, modify and distribute this software and its   *
   9  * documentation strictly for non-commercial purposes is hereby granted   *
  10  * without fee, provided that the above copyright notice appears in all   *
  11  * copies and that both the copyright notice and this permission notice   *
  12  * appear in the supporting documentation. The authors make no claims     *
  13  * about the suitability of this software for any purpose. It is          *
  14  * provided "as is" without express or implied warranty.                  *
  15  * *************************************************************************/
  16
  17 //***********************************************************
  18 // Class AliAODPidHF
  19 // class for PID with AliAODRecoDecayHF
  20 // Authors: D. Caffarri caffarri@bo.infn.it, A.Dainese andrea.dainese@pd.infn.it, S. Dash dash@to.infn.it, F. Prino prino@to.infn.it, R. Romita r.romita@gsi.de, Y. Wang yifei@pi0.physi.uni-heidelberg.de
  21 //***********************************************************
  22 #include "AliAODPidHF.h"
  23 #include "AliAODPid.h"
  24 #include "AliAODTrack.h"
  25 #include "AliPID.h"
  26 #include "AliTPCPIDResponse.h"
  27 #include "AliITSPIDResponse.h"
  28 #include "AliTOFPIDResponse.h"
  29 #include "AliAODpidUtil.h"
  30 #include "AliESDtrack.h"
  31
  32
  33 ClassImp(AliAODPidHF)
  34
  35 //------------------------------
  36 AliAODPidHF::AliAODPidHF():
  37   fSigma(3.),
  38   fPriors()
  39 {
  40  //
  41  // Default constructor
  42  //
  43
  44 }
  45 //----------------------
  46 AliAODPidHF::~AliAODPidHF()
  47 {
  48       // destructor
  49 }
  50 //------------------------
  51 AliAODPidHF::AliAODPidHF(const AliAODPidHF& pid) :
  52   AliAODPid(pid),
  53   fSigma(pid.fSigma),
  54   fPriors(pid.fPriors)
  55   {
  56
  57   for(Int_t i=0;i<5;i++){
  58     fPriors[i]=pid.fPriors[i];
  59   }
  60
  61   }
  62
  63 //----------------------
  64 Int_t AliAODPidHF::RawSignalPID(AliAODTrack *track, TString detector){
  65
  66    Int_t specie=-1;
  67    if(detector.Contains("ITS")) return ApplyPidITSRaw(track,specie);
  68    if(detector.Contains("TPC")) return ApplyPidTPCRaw(track,specie);
  69    if(detector.Contains("TOF")) return ApplyPidTOFRaw(track,specie);
  70
  71   return specie;
  72
  73 }
  74 //---------------------------
  75 Bool_t AliAODPidHF::IsKaonRaw (AliAODTrack *track, TString detector){
  76
  77  Int_t specie=0;
  78
  79  if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,3);
  80  if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,3);
  81  if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,3);
  82
  83  if(specie==3) return kTRUE;
  84  return kFALSE;
  85 }
  86 //---------------------------
  87 Bool_t AliAODPidHF::IsPionRaw (AliAODTrack *track, TString detector){
  88
  89  Int_t specie=0;
  90
  91  if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,2);
  92  if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,2);
  93  if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,2);
  94
  95  if(specie==2) return kTRUE;
  96  return kFALSE;
  97 }
  98 //---------------------------
  99 Bool_t AliAODPidHF::IsProtonRaw (AliAODTrack *track, TString detector){
 100
 101  Int_t specie=0;
 102  if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,4);
 103  if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,4);
 104  if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,4);
 105
 106  if(specie==4) return kTRUE;
 107
 108  return kFALSE;
 109 }
 110 //--------------------------
 111 Bool_t AliAODPidHF::IsElectronRaw (AliAODTrack *track, TString detector){
 112
 113  Int_t specie=-1;
 114  if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,0);
 115  if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,0);
 116  if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,0);
 117
 118  if(specie==0) return kTRUE;
 119
 120  return kFALSE;
 121 }
 122 //--------------------------
 123 Int_t AliAODPidHF::ApplyPidTPCRaw(AliAODTrack *track,Int_t specie){
 124
 125   if(!CheckStatus(track,"TPC")) return -1;
 126   AliAODPid *pidObj = track->GetDetPid();
 127
 128   Double_t dedx=pidObj->GetTPCsignal();
 129   Double_t mom = pidObj->GetTPCmomentum();
 130   AliTPCPIDResponse tpcResponse;
 131   Int_t pid=-1;
 132   if(specie<0){  // from RawSignalPID : should return the particle specie to wich the de/dx is closer to the bethe-block curve -> performance to be checked
 133    Double_t nsigmaMax=fSigma;
 134    for(Int_t ipart=0;ipart<5;ipart++){
 135     AliPID::EParticleType type=AliPID::EParticleType(ipart);
 136     Double_t nsigma = TMath::Abs(tpcResponse.GetNumberOfSigmas(mom,dedx,track->GetTPCNcls(),type));
 137     if((nsigma<nsigmaMax) && (nsigma<fSigma)) {
 138      pid=ipart;
 139      nsigmaMax=nsigma;
 140     }
 141    }
 142   }else{ // asks only for one particle specie
 143    AliPID::EParticleType type=AliPID::EParticleType(specie);
 144     Double_t nsigma = TMath::Abs(tpcResponse.GetNumberOfSigmas(mom,dedx,track->GetTPCNcls(),type));
 145    if (nsigma>fSigma) {
 146     pid=-1;
 147    }else{
 148     pid=specie;
 149    }
 150   }
 151
 152  return pid;
 153
 154 }
 155 //----------------------------
 156 Int_t AliAODPidHF::ApplyPidITSRaw(AliAODTrack *track,Int_t specie){
 157
 158   if(!CheckStatus(track,"ITS")) return -1;
 159
 160   Double_t mom=track->P();
 161   AliAODPid *pidObj = track->GetDetPid();
 162
 163   Double_t dedx=pidObj->GetITSsignal();
 164   AliITSPIDResponse itsResponse;
 165   Int_t pid=-1;
 166   if(specie<0){  // from RawSignalPID : should return the particle specie to wich the de/dx is closer to the bethe-block curve -> performance to be checked
 167    Double_t nsigmaMax=fSigma;
 168    for(Int_t ipart=0;ipart<5;ipart++){
 169     AliPID::EParticleType type=AliPID::EParticleType(ipart);
 170     Double_t nsigma = TMath::Abs(itsResponse.GetNumberOfSigmas(mom,dedx,type));
 171     if((nsigma<nsigmaMax) && (nsigma<fSigma)) {
 172      pid=ipart;
 173      nsigmaMax=nsigma;
 174     }
 175    }
 176   }else{ // asks only for one particle specie
 177    AliPID::EParticleType type=AliPID::EParticleType(specie);
 178     Double_t nsigma = TMath::Abs(itsResponse.GetNumberOfSigmas(mom,dedx,type));
 179    if (nsigma>fSigma) {
 180     pid=-1;
 181    }else{
 182     pid=specie;
 183    }
 184   }
 185  return pid;
 186 }
 187 //----------------------------
 188 Int_t AliAODPidHF::ApplyPidTOFRaw(AliAODTrack *track,Int_t specie){
 189
 190  if(!CheckStatus(track,"TOF")) return -1;
 191
 192  Double_t time[AliPID::kSPECIESN];
 193  AliAODPid *pidObj = track->GetDetPid();
 194  pidObj->GetIntegratedTimes(time);
 195  AliTOFPIDResponse tofResponse;
 196  Int_t pid=-1;
 197
 198   if(specie<0){  // from RawSignalPID : should return the particle specie to wich the de/dx is closer to the bethe-block curve -> performance to be checked
 199    Double_t nsigmaMax=fSigma;
 200    for(Int_t ipart=0;ipart<5;ipart++){
 201     AliPID::EParticleType type=AliPID::EParticleType(ipart);
 202     Double_t nsigma = tofResponse.GetExpectedSigma(track->P(),time[type],AliPID::ParticleMass(type));
 203     if((nsigma<nsigmaMax) && (nsigma<fSigma)) {
 204      pid=ipart;
 205      nsigmaMax=nsigma;
 206     }
 207    }
 208   }else{ // asks only for one particle specie
 209    AliPID::EParticleType type=AliPID::EParticleType(specie);
 210    Double_t nsigma = TMath::Abs(tofResponse.GetExpectedSigma(track->P(),time[type],AliPID::ParticleMass(type)));
 211    if (nsigma>fSigma) {
 212     pid=-1;
 213    }else{
 214     pid=specie;
 215    }
 216   }
 217  return pid;
 218
 219 }
 220 //------------------------------
 221 void AliAODPidHF::CombinedProbability(AliAODTrack *track,Bool_t *type){
 222
 223  const Double_t *pid=track->PID();
 224  Float_t max=0.;
 225  Int_t k=-1;
 226  for (Int_t i=0; i<10; i++) {
 227    if (pid[i]>max) {k=i; max=pid[i];}
 228  }
 229
 230  if(k==2) type[0]=kTRUE;
 231  if(k==3) type[1]=kTRUE;
 232  if(k==4) type[2]=kTRUE;
 233
 234  return;
 235 }
 236 //--------------------
 237 void AliAODPidHF::BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid){
 238
 239   if(detectors.Contains("ITS")) {BayesianProbabilityITS(track,pid);return;}
 240   if(detectors.Contains("TPC")) {BayesianProbabilityTPC(track,pid);return;}
 241   if(detectors.Contains("TOF")) {BayesianProbabilityTOF(track,pid);return;}
 242
 243   if(detectors.Contains("All")) {
 244     Double_t probITS[5]={0.,0.,0.,0.,0.};
 245     Double_t probTPC[5]={0.,0.,0.,0.,0.};
 246     Double_t probTOF[5]={0.,0.,0.,0.,0.};
 247     BayesianProbabilityITS(track,probITS);
 248     BayesianProbabilityTPC(track,probTPC);
 249     BayesianProbabilityTOF(track,probTOF);
 250     Double_t probTot[5]={0.,0.,0.,0.,0.};
 251     for(Int_t i=0;i<5;i++){
 252      probTot[i]=probITS[i]*probTPC[i]*probTOF[i];
 253     }
 254     for(Int_t i2=0;i2<5;i2++){
 255      pid[i2]=probTot[i2]*fPriors[i2]/(probTot[0]*fPriors[0]+probTot[1]*fPriors[1]+probTot[2]*fPriors[2]+probTot[3]*fPriors[3]+probTot[4]*fPriors[4]);
 256     }
 257    }
 258
 259  return;
 260
 261 }
 262 //------------------------------------
 263 void AliAODPidHF::BayesianProbabilityITS(AliAODTrack *track,Double_t *prob){
 264
 265  AliAODpidUtil pid;
 266  Double_t itspid[AliPID::kSPECIES];
 267  pid.MakeITSPID(track,itspid);
 268  for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
 269   prob[ind]=itspid[ind]*fPriors[ind]/(itspid[0]*fPriors[0]+itspid[1]*fPriors[1]+itspid[2]*fPriors[2]+itspid[3]*fPriors[3]+itspid[4]*fPriors[4]);
 270  }
 271  return;
 272
 273 }
 274 //------------------------------------
 275 void AliAODPidHF::BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob){
 276
 277  AliAODpidUtil pid;
 278  Double_t tpcpid[AliPID::kSPECIES];
 279  pid.MakeTPCPID(track,tpcpid);
 280  for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
 281   if(tpcpid[ind]>0.) {
 282    prob[ind]=tpcpid[ind]*fPriors[ind]/(tpcpid[0]*fPriors[0]+tpcpid[1]*fPriors[1]+tpcpid[2]*fPriors[2]+tpcpid[3]*fPriors[3]+tpcpid[4]*fPriors[4]);
 283   }else{
 284    prob[ind]=0.;
 285   }
 286  }
 287  return;
 288
 289 }
 290 //------------------------------------
 291 void AliAODPidHF::BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob){
 292
 293  AliAODpidUtil pid;
 294  Double_t tofpid[AliPID::kSPECIES];
 295  //pid.MakeTOFPID(track,tofpid);
 296  for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
 297   prob[ind]=tofpid[ind]*fPriors[ind]/(tofpid[0]*fPriors[0]+tofpid[1]*fPriors[1]+tofpid[2]*fPriors[2]+tofpid[3]*fPriors[3]+tofpid[4]*fPriors[4]);
 298  }
 299  return;
 300
 301 }
 302 //---------------------------------
 303 void AliAODPidHF::BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob){
 304
 305  AliAODpidUtil pid;
 306  Double_t trdpid[AliPID::kSPECIES];
 307  pid.MakeTRDPID(track,trdpid);
 308  for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
 309   if(trdpid[ind]>0.) {
 310    prob[ind]=trdpid[ind]*fPriors[ind]/(trdpid[0]*fPriors[0]+trdpid[1]*fPriors[1]+trdpid[2]*fPriors[2]+trdpid[3]*fPriors[3]+trdpid[4]*fPriors[4]);
 311   }else{
 312    prob[ind]=0.;
 313   }
 314  }
 315   return;
 316
 317  }
 318 //--------------------------------
 319 Bool_t AliAODPidHF::CheckStatus(AliAODTrack *track,TString detectors){
 320
 321
 322  if(detectors.Contains("ITS")){
 323   if ((track->GetStatus()&AliESDtrack::kITSin)==0) return kFALSE;
 324   UChar_t clumap=track->GetITSClusterMap();
 325   Int_t nPointsForPid=0;
 326   for(Int_t i=2; i<6; i++){
 327    if(clumap&(1<<i)) ++nPointsForPid;
 328   }
 329   if(nPointsForPid<3) return kFALSE;// track not to be used for PID purposes
 330  }
 331
 332  if(detectors.Contains("TPC")){
 333    if ((track->GetStatus()&AliESDtrack::kTPCin )==0) return kFALSE;
 334    UShort_t nTPCClus=track->GetTPCClusterMap().CountBits();
 335    if (nTPCClus<70) return kFALSE;
 336  }
 337
 338  if(detectors.Contains("TOF")){
 339    if ((track->GetStatus()&AliESDtrack::kTOFout )==0)    return kFALSE;
 340    if ((track->GetStatus()&AliESDtrack::kTIME )==0)     return kFALSE;
 341    if ((track->GetStatus()&AliESDtrack::kTOFpid )==0)   return kFALSE;
 342  }
 343
 344
 345  if(detectors.Contains("TRD")){
 346   if ((track->GetStatus()&AliESDtrack::kTRDout )==0)   return kFALSE;
 347   AliAODPid *pidObj = track->GetDetPid();
 348   Float_t *mom=pidObj->GetTRDmomentum();
 349   Int_t ntracklets=0;
 350   for(Int_t iPl=0;iPl<6;iPl++){
 351    if(mom[iPl]>0.) ntracklets++;
 352   }
 353    if(ntracklets<4) return kFALSE;
 354  }
 355
 356  return kTRUE;
 357 }