HMPID information in AliAODPid

Sona Pochybova <sona@rmki.kfki.hu>

diff --git a/ANALYSIS/AliAnalysisTaskESDfilter.cxx b/ANALYSIS/AliAnalysisTaskESDfilter.cxx
index 4cfb453ee0d2fc654553a6de744ed63ddf74f560..a5f6d8749d61a0744cedf7f047e107079f83c027 100644 (file)
--- a/ANALYSIS/AliAnalysisTaskESDfilter.cxx
+++ b/ANALYSIS/AliAnalysisTaskESDfilter.cxx
@@ -1772,6 +1772,10 @@ void AliAnalysisTaskESDfilter::SetDetectorRawSignals(AliAODPid *aodpid, AliESDtr
  aodpid->SetEMCALPosition(emcpos);\r
  aodpid->SetEMCALMomentum(emcmom);\r
 \r
+ Double_t hmpPid[5] = {0};\r
+ track->GetHMPIDpid(hmpPid);\r
+ aodpid->SetHMPIDprobs(hmpPid);\r
+\r
  AliExternalTrackParam *outerparam = (AliExternalTrackParam*)track->GetOuterParam();\r
  if(!outerparam) return;\r
 \r

diff --git a/STEER/AliAODPid.cxx b/STEER/AliAODPid.cxx
index 53195e11d4a8890d0aa32a6d723d228f8030d2fe..1f462715e428c2f33be18af0cad63935e554699b 100644 (file)
--- a/STEER/AliAODPid.cxx
+++ b/STEER/AliAODPid.cxx
@@ -40,6 +40,7 @@ AliAODPid::AliAODPid():
 {
   // default constructor
     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]   = 0; 
+    for(Int_t i=0; i<5; i++) fHMPIDprobs[i] = 0.;
     for(Int_t i=0; i<3; i++) fEMCALPosition[i]    = 0.;
     for(Int_t i=0; i<5; i++) fTOFpidResolution[i] = 0.;
     for(Int_t i=0; i<6; i++) fTRDmomentum[i]      = 0.;
@@ -72,6 +73,7 @@ AliAODPid::AliAODPid(const AliAODPid& pid) :
     fTRDslices = new Double32_t[fTRDnSlices];
     for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
+    for(Int_t i=0; i<5; i++) fHMPIDprobs[i] = pid.fHMPIDprobs[i];
     for(Int_t i=0; i<3; i++) {
       fEMCALPosition[i]=pid.fEMCALPosition[i];
       fEMCALMomentum[i]=pid.fEMCALMomentum[i];
@@ -102,6 +104,7 @@ AliAODPid& AliAODPid::operator=(const AliAODPid& pid)
     fTOFesdsignal=pid.fTOFesdsignal;
     fHMPIDsignal=pid.fHMPIDsignal;
     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
+    for(Int_t i=0; i<5; i++) fHMPIDprobs[i] = pid.fHMPIDprobs[i];
     for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
     for(Int_t i=0; i<3; i++) {
       fEMCALPosition[i]=pid.fEMCALPosition[i];
@@ -160,3 +163,19 @@ void AliAODPid::GetTOFpidResolution(Double_t tofRes[5]) const
   for (Int_t i=0; i<5; i++) tofRes[i]=fTOFpidResolution[i];
 }
 
+//______________________________________________________________________________
+void AliAODPid::SetHMPIDprobs(Double_t hmpPid[5]) 
+{
+  //
+  // Set the HMPID PID probablities that are read from ESD
+  //  
+  for(Int_t i = 0; i < 5; i++ ) fHMPIDprobs[i] =  hmpPid[i];
+}
+//______________________________________________________________________________
+void AliAODPid::GetHMPIDprobs(Double_t *p) 
+{
+  //
+  // Set the HMPID PID probablities that are read from ESD
+  //  
+  for(Int_t i = 0; i < AliPID::kSPECIES; i++ ) p[i] =  fHMPIDprobs[i];
+}

diff --git a/STEER/AliAODPid.h b/STEER/AliAODPid.h
index 5893aa1dec4496ee424fe51db7d7eeafaed27694..3b86cb503efc8eeaeff4edee66fe15f9bbeea5ef 100644 (file)
--- a/STEER/AliAODPid.h
+++ b/STEER/AliAODPid.h
@@ -33,6 +33,7 @@ class AliAODPid : public TObject {
   void      SetTOFpidResolution(Double_t tofPIDres[5]);
   void      SetIntegratedTimes(Double_t timeint[5]);
   void      SetHMPIDsignal(Double_t hmpid)                     {fHMPIDsignal=hmpid;}
+  void      SetHMPIDprobs(Double_t hmpPid[5]);
   void      SetEMCALPosition(Double_t emcalpos[3]);
   void      SetEMCALMomentum(Double_t emcalmom[3]);
 
@@ -45,6 +46,7 @@ class AliAODPid : public TObject {
   Float_t*  GetTRDmomentum()           {return  fTRDmomentum;}
   Double_t  GetTOFsignal()       const {return  fTOFesdsignal;}
   Double_t  GetHMPIDsignal()     const {return  fHMPIDsignal;}
+  void  GetHMPIDprobs(Double_t *p);            
 
   void      GetIntegratedTimes(Double_t timeint[5])  const; 
   void      GetEMCALPosition  (Double_t emcalpos[3]) const;
@@ -63,12 +65,13 @@ class AliAODPid : public TObject {
   Double32_t  fTOFpidResolution[5]; // TOF pid resolution for each mass hypotesys 
   Double32_t  fIntTime[5];       // track time hypothesis
   Double32_t  fHMPIDsignal;      // detector raw signal
+  Double32_t  fHMPIDprobs[5];    // detector pid probabilities
   Double32_t  fEMCALPosition[3]; // global position of track
                                 // extrapolated to EMCAL surface
   Double32_t  fEMCALMomentum[3]; // momentum of track
                                 // extrapolated to EMCAL surface
 
-  ClassDef(AliAODPid, 5);
+  ClassDef(AliAODPid, 6);
 };
 
 #endif