Improvement of PID on ESD
authordibari <dibari@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 13 Dec 2007 17:47:28 +0000 (17:47 +0000)
committerdibari <dibari@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 13 Dec 2007 17:47:28 +0000 (17:47 +0000)
HMPID/AliHMPIDPid.cxx
HMPID/AliHMPIDPid.h
HMPID/AliHMPIDReconstructor.cxx

index 0715c35..5740d4e 100644 (file)
@@ -23,7 +23,7 @@
 
 #include "AliHMPIDPid.h"       //class header
 #include "AliHMPIDParam.h"     //class header
-#include <AliESDEvent.h>       //FindPid()
+#include <AliESDtrack.h>       //FindPid()
 
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 AliHMPIDPid::AliHMPIDPid():TTask("HMPIDrec","HMPIDPid")
@@ -33,42 +33,38 @@ AliHMPIDPid::AliHMPIDPid():TTask("HMPIDrec","HMPIDPid")
 //..
 }
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
-void AliHMPIDPid::FindPid(AliESDEvent *pESD)
+void AliHMPIDPid::FindPid(AliESDtrack *pTrk,Double_t *prob)
 {
 // 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)
 
   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
+  Double_t h[AliPID::kSPECIES];
+  
+  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++) prob[iPart]=1.0/AliPID::kSPECIES;
+    return;
+  } 
+
+  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) prob[iPart]=h[iPart]/hTot;
+    else prob[iPart]=1.0/AliPID::kSPECIES;            //all theoretical values are far away from experemental one
+  }
 }
 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
index b6b43b8..caea28f 100644 (file)
@@ -15,7 +15,7 @@
 
 #include <TTask.h>        //base class
 
-class AliESDEvent;
+class AliESDtrack;
 
 class AliHMPIDPid : public TTask 
 {
@@ -23,7 +23,7 @@ public :
              AliHMPIDPid();    //ctor
     virtual ~AliHMPIDPid() {;} //dtor
     
-    void     FindPid(AliESDEvent *pESD);             //Find PID for tracks
+    void FindPid(AliESDtrack *pESD,Double_t *prob);  //Find PID for tracks
 
 //
 protected:
index e040af2..fd9ae68 100644 (file)
@@ -197,9 +197,14 @@ void AliHMPIDReconstructor::FillESD(TTree */*digitsTree*/, TTree */*clustersTree
 // Fill ESD with all the infos from HMPID
 // Probability vector from AliHMPIDPid
 //...
+  AliHMPIDPid pID;
+  Double_t prob[5];
+  
+  for(Int_t iTrk=0;iTrk<pESD->GetNumberOfTracks();iTrk++){//ESD tracks loop
+    AliESDtrack *pTrk = pESD->GetTrack(iTrk);// get next reconstructed track
+    pID.FindPid(pTrk,prob);
+    Printf("Theta Cherenkov %5.3f |  e %5.1f%% | u %5.1f%% | K %5.1f%% | pi %5.1f%% | p %5.1f%%",
+          pTrk->GetHMPIDsignal(),prob[0]*100,prob[1]*100,prob[2]*100,prob[3]*100,prob[4]*100);
+  }//ESD tracks loop
   
-  AliHMPIDPid *pPid = new AliHMPIDPid();
-  pPid->FindPid(pESD);
-  delete pPid;
-
 }//FillESD()