New class AliHMPIDPid added. Now HMPID could run PID even on ESD within the same...
[u/mrichter/AliRoot.git] / HMPID / AliHMPIDReconstructor.cxx
index 2265284322c5ec171afa5ab68d2f26832d04a14d..e040af2da0f5f6e99577b7ebcee349f3bc75ac71 100644 (file)
@@ -20,6 +20,7 @@
 #include "AliHMPIDReconstructor.h" //class header
 #include "AliHMPID.h"              //Reconstruct() 
 #include "AliHMPIDCluster.h"       //Dig2Clu()
+#include "AliHMPIDPid.h"           //FillEsd() 
 #include "AliHMPIDParam.h"         //FillEsd() 
 #include <AliCDBEntry.h>           //ctor
 #include <AliCDBManager.h>         //ctor
@@ -193,39 +194,12 @@ void AliHMPIDReconstructor::ConvertDigits(AliRawReader *pRR,TTree *pDigTree)cons
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 void AliHMPIDReconstructor::FillESD(TTree */*digitsTree*/, TTree */*clustersTree*/, AliESDEvent *pESD) const
 {
-// Calculates probability to be a electron-muon-pion-kaon-proton
-// from the given Cerenkov angle and momentum assuming no initial particle composition
-// (i.e. apriory probability to be the particle of the given sort is the same for all sorts)
+// Fill ESD with all the infos from HMPID
+// Probability vector from AliHMPIDPid
+//...
+  
+  AliHMPIDPid *pPid = new AliHMPIDPid();
+  pPid->FindPid(pESD);
+  delete pPid;
 
-  AliPID ppp; //needed
-  Double_t pid[AliPID::kSPECIES],h[AliPID::kSPECIES];
-   
-  for(Int_t iTrk=0;iTrk<pESD->GetNumberOfTracks();iTrk++){//ESD tracks loop
-    AliESDtrack *pTrk = pESD->GetTrack(iTrk);// get next reconstructed track
-    if(pTrk->GetHMPIDsignal()<=0){//HMPID does not find anything reasonable for this track, assign 0.2 for all species
-      for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++) pid[iPart]=1.0/AliPID::kSPECIES;
-      pTrk->SetHMPIDpid(pid);
-      continue;
-    } 
-    Double_t pmod = pTrk->GetP();
-    Double_t hTot=0;
-    for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++){
-      Double_t mass = AliPID::ParticleMass(iPart);
-      Double_t cosThetaTh = TMath::Sqrt(mass*mass+pmod*pmod)/(AliHMPIDParam::Instance()->MeanIdxRad()*pmod);
-      if(cosThetaTh<1) //calculate the height of theoretical theta ckov on the gaus of experimental one
-        h[iPart] =TMath::Gaus(TMath::ACos(cosThetaTh),pTrk->GetHMPIDsignal(),TMath::Sqrt(pTrk->GetHMPIDchi2()),kTRUE);      
-      else             //beta < 1/ref. idx. => no light at all  
-        h[iPart] =0 ;       
-      hTot    +=h[iPart]; //total height of all theoretical heights for normalization
-    }//species loop
-     
-    Double_t hMin=TMath::Gaus(pTrk->GetHMPIDsignal()-4*TMath::Sqrt(pTrk->GetHMPIDchi2()),pTrk->GetHMPIDsignal(),TMath::Sqrt(pTrk->GetHMPIDchi2()),kTRUE);//5 sigma protection
-    
-    for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++)//species loop to assign probabilities
-      if(hTot>hMin)  
-        pid[iPart]=h[iPart]/hTot;
-      else                               //all theoretical values are far away from experemental one
-        pid[iPart]=1.0/AliPID::kSPECIES; 
-    pTrk->SetHMPIDpid(pid);
-  }//ESD tracks loop
 }//FillESD()