New class for PID of HF candidates (R. Romita)

diff --git a/PWG3/PWG3vertexingHFLinkDef.h b/PWG3/PWG3vertexingHFLinkDef.h
index e7c0fc3a08d33202b4ae7bc2d3d3d174d0529a48..d0df8ded1e6cce764aa7e2876a3bf301d4ce6976 100644 (file)
--- a/PWG3/PWG3vertexingHFLinkDef.h
+++ b/PWG3/PWG3vertexingHFLinkDef.h
@@ -10,6 +10,7 @@
 #pragma link C++ class AliAODRecoDecayHF3Prong+;
 #pragma link C++ class AliAODRecoDecayHF4Prong+;
 #pragma link C++ class AliAODRecoCascadeHF+;
+#pragma link C++ class AliAODPidHF+;
 #pragma link C++ class AliRDHFCuts+;
 #pragma link C++ class AliRDHFCutsD0toKpi+;
 #pragma link C++ class AliRDHFCutsJpsitoee+;

diff --git a/PWG3/libPWG3vertexingHF.pkg b/PWG3/libPWG3vertexingHF.pkg
index 96dc43ad8e8d3edfbecb97f61bf260a368f2f4a1..d157a85809b39b0a806262a3c747469236633d76 100644 (file)
--- a/PWG3/libPWG3vertexingHF.pkg
+++ b/PWG3/libPWG3vertexingHF.pkg
@@ -4,6 +4,7 @@ SRCS:=   vertexingHF/AliAODRecoDecayHF.cxx \
        vertexingHF/AliAODRecoDecayHF2Prong.cxx  vertexingHF/AliAODRecoDecayHF3Prong.cxx \
        vertexingHF/AliAODRecoDecayHF4Prong.cxx \
        vertexingHF/AliAODRecoCascadeHF.cxx \
+       vertexingHF/AliAODPidHF.cxx \
        vertexingHF/AliRDHFCuts.cxx \
        vertexingHF/AliRDHFCutsD0toKpi.cxx \
        vertexingHF/AliRDHFCutsJpsitoee.cxx \

diff --git a/PWG3/vertexingHF/AliAODPidHF.cxx b/PWG3/vertexingHF/AliAODPidHF.cxx
new file mode 100644 (file)
index 0000000..6fa68ec
--- /dev/null
+++ b/PWG3/vertexingHF/AliAODPidHF.cxx
@@ -0,0 +1,282 @@
+/**************************************************************************
+ * Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
+ *                                                                        *
+ *                                                                        *
+ * Author: The ALICE Off-line Project.                                    *
+ * Contributors are mentioned in the code where appropriate.              *
+ *
+ * Permission to use, copy, modify and distribute this software and its   *
+ * documentation strictly for non-commercial purposes is hereby granted   *
+ * without fee, provided that the above copyright notice appears in all   *
+ * copies and that both the copyright notice and this permission notice   *
+ * appear in the supporting documentation. The authors make no claims     *
+ * about the suitability of this software for any purpose. It is          *
+ * provided "as is" without express or implied warranty.                  *
+ * *************************************************************************/
+
+//***********************************************************
+// Class AliAODPidHF
+// class for PID with AliAODRecoDecayHF
+// 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
+//***********************************************************
+#include "AliAODPidHF.h"
+#include "AliAODPid.h"
+#include "AliAODTrack.h"
+#include "AliPID.h"
+#include "AliTPCPIDResponse.h"
+#include "AliITSPIDResponse.h"
+#include "AliTOFPIDResponse.h"
+#include "AliAODpidUtil.h"
+
+
+ClassImp(AliAODPidHF)
+
+//------------------------------
+AliAODPidHF::AliAODPidHF():
+  fSigma(3.),
+  fPriors()
+{
+ //
+ // Default constructor
+ //
+
+}
+//----------------------
+AliAODPidHF::~AliAODPidHF()
+{
+      // destructor
+}
+//------------------------
+AliAODPidHF::AliAODPidHF(const AliAODPidHF& pid) :
+  AliAODPid(pid),
+  fSigma(pid.fSigma),
+  fPriors(pid.fPriors)
+  {
+  
+  for(Int_t i=0;i<5;i++){
+    fPriors[i]=pid.fPriors[i];
+  }
+  
+  }
+
+//----------------------
+Int_t AliAODPidHF::RawSignalPID(AliAODTrack *track, TString detector){
+
+   if(detector.Contains("ITS")) return ApplyPidITSRaw(track,0);
+   if(detector.Contains("TPC")) return ApplyPidTPCRaw(track,0);
+   if(detector.Contains("TOF")) return ApplyPidTOFRaw(track,0);
+
+  return 0;
+
+}
+//---------------------------
+Bool_t AliAODPidHF::IsKaonRaw (AliAODTrack *track, TString detector){
+
+ Int_t specie=0;
+
+ if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,3);
+ if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,3);
+ if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,3);
+
+ if(specie==3) return kTRUE;
+ return kFALSE;
+}
+//---------------------------
+Bool_t AliAODPidHF::IsPionRaw (AliAODTrack *track, TString detector){
+
+ Int_t specie=0;
+
+ if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,2);
+ if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,2);
+ if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,2);
+
+ if(specie==2) return kTRUE;
+ return kFALSE;
+}
+//---------------------------
+Bool_t AliAODPidHF::IsProtonRaw (AliAODTrack *track, TString detector){
+
+ Int_t specie=0;
+ if(detector.Contains("ITS")) specie=ApplyPidITSRaw(track,4);
+ if(detector.Contains("TPC")) specie=ApplyPidTPCRaw(track,4); 
+ if(detector.Contains("TOF")) specie=ApplyPidTOFRaw(track,4);
+
+ if(specie==4) return kTRUE;
+
+ return kFALSE;
+}
+//---------------------------
+Int_t AliAODPidHF::ApplyPidTPCRaw(AliAODTrack *track,Int_t specie){
+
+  if(track->P()>2.) printf("Warning: TPC resolution not good in this region");
+  AliAODPid *pidObj = track->GetDetPid();
+  
+  Double_t dedx=pidObj->GetTPCsignal();
+  Double_t mom = pidObj->GetTPCmomentum();
+  AliTPCPIDResponse tpcResponse;
+  Int_t pid=0;
+  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
+   Double_t nsigmaMax=fSigma;
+   for(Int_t ipart=2;ipart<5;ipart++){
+    AliPID::EParticleType type=AliPID::EParticleType(ipart);
+    Double_t nsigma = TMath::Abs(tpcResponse.GetNumberOfSigmas(mom,dedx,50,type));
+    if((nsigma<nsigmaMax) && (nsigma<fSigma)) {
+     pid=ipart;
+     nsigmaMax=nsigma;
+    }
+   }
+  }else{ // asks only for one particle specie
+   AliPID::EParticleType type=AliPID::EParticleType(specie);
+    Double_t nsigma = TMath::Abs(tpcResponse.GetNumberOfSigmas(mom,dedx,50,type));
+   if (nsigma>fSigma) {
+    pid=0; 
+   }else{
+    pid=specie;
+   }
+  }
+
+ return pid;
+
+}
+//----------------------------
+Int_t AliAODPidHF::ApplyPidITSRaw(AliAODTrack *track,Int_t specie){
+  Double_t mom=track->P();
+  AliAODPid *pidObj = track->GetDetPid();
+
+  Double_t dedx=pidObj->GetITSsignal();
+  AliITSPIDResponse itsResponse;
+  Int_t pid=0;
+  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
+   Double_t nsigmaMax=fSigma;
+   for(Int_t ipart=2;ipart<5;ipart++){
+    AliPID::EParticleType type=AliPID::EParticleType(ipart);
+    Double_t nsigma = TMath::Abs(itsResponse.GetNumberOfSigmas(mom,dedx,type));
+    if((nsigma<nsigmaMax) && (nsigma<fSigma)) {
+     pid=ipart;
+     nsigmaMax=nsigma;
+    }
+   }
+  }else{ // asks only for one particle specie
+   AliPID::EParticleType type=AliPID::EParticleType(specie);
+    Double_t nsigma = TMath::Abs(itsResponse.GetNumberOfSigmas(mom,dedx,type));
+   if (nsigma>fSigma) {
+    pid=0; 
+   }else{
+    pid=specie;
+   }
+  }
+ return pid; 
+}
+//----------------------------
+Int_t AliAODPidHF::ApplyPidTOFRaw(AliAODTrack *track,Int_t specie){
+ Double_t time[AliPID::kSPECIESN];
+ AliAODPid *pidObj = track->GetDetPid();
+ pidObj->GetIntegratedTimes(time);
+ AliTOFPIDResponse tofResponse;
+ Int_t pid=0;
+
+  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
+   Double_t nsigmaMax=fSigma;
+   for(Int_t ipart=2;ipart<5;ipart++){
+    AliPID::EParticleType type=AliPID::EParticleType(ipart);
+    Double_t nsigma = tofResponse.GetExpectedSigma(track->P(),time[type],AliPID::ParticleMass(type));
+    if((nsigma<nsigmaMax) && (nsigma<fSigma)) {
+     pid=ipart;
+     nsigmaMax=nsigma;
+    }
+   }
+  }else{ // asks only for one particle specie
+   AliPID::EParticleType type=AliPID::EParticleType(specie);
+   Double_t nsigma = TMath::Abs(tofResponse.GetExpectedSigma(track->P(),time[type],AliPID::ParticleMass(type)));
+   if (nsigma>fSigma) {
+    pid=0; 
+   }else{
+    pid=specie;
+   }
+  }
+ return pid; 
+
+}
+//------------------------------
+void AliAODPidHF::CombinedProbability(AliAODTrack *track,Bool_t *type){
+
+ const Double_t *pid=track->PID();
+ Float_t max=0.;
+ Int_t k=-1;
+ for (Int_t i=0; i<10; i++) {
+   if (pid[i]>max) {k=i; max=pid[i];}  
+ }
+
+ if(k==2) type[0]=kTRUE;
+ if(k==3) type[1]=kTRUE;
+ if(k==4) type[2]=kTRUE;
+
+ return;
+}
+//--------------------
+void AliAODPidHF::BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid){
+
+  if(detectors.Contains("ITS")) {BayesianProbabilityITS(track,pid);return;}
+  if(detectors.Contains("TPC")) {BayesianProbabilityTPC(track,pid);return;}
+  if(detectors.Contains("TOF")) {BayesianProbabilityTOF(track,pid);return;}
+
+  if(detectors.Contains("All")) {
+    Double_t probITS[5]={0.,0.,0.,0.,0.};
+    Double_t probTPC[5]={0.,0.,0.,0.,0.};
+    Double_t probTOF[5]={0.,0.,0.,0.,0.};
+    BayesianProbabilityITS(track,probITS);
+    BayesianProbabilityTPC(track,probTPC);
+    BayesianProbabilityTOF(track,probTOF);
+    Double_t probTot[5]={0.,0.,0.,0.,0.};
+    for(Int_t i=0;i<5;i++){
+     probTot[i]=probITS[i]*probTPC[i]*probTOF[i];
+    }
+    for(Int_t i2=0;i2<5;i2++){
+     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]);
+    }
+   }
+
+ return;
+
+}
+//------------------------------------
+void AliAODPidHF::BayesianProbabilityITS(AliAODTrack *track,Double_t *prob){
+
+ AliAODpidUtil pid;
+ Double_t itspid[AliPID::kSPECIES];
+ pid.MakeITSPID(track,itspid);
+ for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
+  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]);
+ }
+ return;
+
+}
+//------------------------------------
+void AliAODPidHF::BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob){
+
+ AliAODpidUtil pid;
+ Double_t tpcpid[AliPID::kSPECIES];
+ pid.MakeTPCPID(track,tpcpid);
+ for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
+  if(tpcpid[ind]>0.) {
+   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]);
+  }else{
+   prob[ind]=0.;
+  }
+ }
+ return;
+
+}
+//------------------------------------
+void AliAODPidHF::BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob){
+
+ AliAODpidUtil pid;
+ Double_t tofpid[AliPID::kSPECIES];
+ Float_t time=0.1; //needs to be changed!
+ pid.MakeTOFPID(track,time,tofpid);
+ for(Int_t ind=0;ind<AliPID::kSPECIES;ind++){
+  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]);
+ }
+ return;
+
+}

diff --git a/PWG3/vertexingHF/AliAODPidHF.h b/PWG3/vertexingHF/AliAODPidHF.h
new file mode 100644 (file)
index 0000000..8f64ab4
--- /dev/null
+++ b/PWG3/vertexingHF/AliAODPidHF.h
@@ -0,0 +1,55 @@
+#ifndef ALIAODPIDHF_H
+#define ALIAODPIDHF_H
+
+/* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
+ *  * See cxx source for full Copyright notice                               */
+
+//***********************************************************
+//// Class AliAODPidHF
+//// class for PID with AliAODRecoDecayHF 
+//// 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
+////***********************************************************
+
+#include "AliAODPid.h"
+#include "AliAODTrack.h"
+
+class AliAODPidHF : public AliAODPid{
+
+ public:
+
+ AliAODPidHF();
+ AliAODPidHF(const AliAODPidHF& pid);
+ AliAODPidHF& operator=(const AliAODPidHF& pid);
+ virtual ~AliAODPidHF();
+
+ //Setters
+ void SetSigma(Double_t sigma){fSigma=sigma;return;}
+ void SetPriors(Double_t *priors){fPriors=priors;return;}
+ 
+
+ Int_t RawSignalPID (AliAODTrack *track, TString detector);
+ Bool_t IsKaonRaw (AliAODTrack *track, TString detector);
+ Bool_t IsPionRaw (AliAODTrack *track, TString detector);
+ Bool_t IsProtonRaw (AliAODTrack *track, TString detector);
+ void BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid);
+ void CombinedProbability(AliAODTrack *track,Bool_t *type); //0 = pion, 1 = kaon, 2 = proton
+
+ protected:
+
+ Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie);
+ Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie);
+ Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie);
+ void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob);
+ void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob);
+ void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob);
+
+ private:
+ Double_t fSigma; // sigma for the raw signal PID 
+ Double_t *fPriors; // set of priors
+
+
+ ClassDef(AliAODPidHF,3)
+
+};
+
+#endif