Adding functionality to create AOD.par files.

[u/mrichter/AliRoot.git] / STEER / AliESDComparison.C

diff --git a/STEER/AliESDComparison.C b/STEER/AliESDComparison.C
index 707ae6be6f0232b4d52799057c8fe7c25194b115..5759e2923a58a8c47ed7945c383972d75659ec4c 100644 (file)
--- a/STEER/AliESDComparison.C
+++ b/STEER/AliESDComparison.C
@@ -13,7 +13,8 @@
   #include <TParticle.h>
   #include <TCanvas.h>
   #include <TBenchmark.h>
-  #include <TKey.h>
+  #include <TFile.h>
+  #include <TTree.h>
   #include <TROOT.h>
 
   #include <AliStack.h>
@@ -116,23 +117,30 @@ Int_t AliESDComparison(const Char_t *dir=".") {
       delete rl;
       return 1;
    }
-   TKey *key=0;
-   TIter next(ef->GetListOfKeys());
+   AliESD* event = new AliESD;
+   TTree* tree = (TTree*) ef->Get("esdTree");
+   if (!tree) {
+      ::Error("AliESDComparison.C", "no ESD tree found");
+      delete rl;
+      return 1;
+   }
+   tree->SetBranchAddress("ESD", &event);
 
    //****** Tentative particle type "concentrations"
    Double_t c[5]={0.01, 0.01, 0.85, 0.10, 0.05};
+   //Double_t c[5]={0.2, 0.2, 0.2, 0.2, 0.2};
+   AliPID::SetPriors(c);
+
 
    //******* The loop over events
    Int_t e=0;
-   while ((key=(TKey*)next())!=0) {
+   while (tree->GetEvent(e)) {
       cout<<endl<<endl<<"********* Processing event number: "<<e<<"*******\n";
 
-      rl->GetEvent(e); ef->cd();
+      rl->GetEvent(e);
  
       e++;
 
-      AliESD *event=(AliESD*)key->ReadObj();
-
       Int_t ntrk=event->GetNumberOfTracks();
       cerr<<"Number of ESD tracks : "<<ntrk<<endl; 
 
@@ -146,6 +154,7 @@ Int_t AliESDComparison(const Char_t *dir=".") {
          UInt_t status=AliESDtrack::kESDpid;
          status|=AliESDtrack::kITSpid; 
          status|=AliESDtrack::kTPCpid; 
+         status|=AliESDtrack::kTRDpid; 
          status|=AliESDtrack::kTOFpid; 
 
         if ((t->GetStatus()&status) == status) {
@@ -160,15 +169,18 @@ Int_t AliESDComparison(const Char_t *dir=".") {
            Int_t code=part->GetPdgCode();
 
            Double_t r[10]; t->GetESDpid(r);
+           //t->GetTRDpid(r);
+           //t->GetTPCpid(r);
 
-           Double_t rcc=0.;
-           Int_t i;
-           for (i=0; i<AliESDtrack::kSPECIES; i++) rcc+=(c[i]*r[i]);
-           if (rcc==0.) continue;
+           AliPID pid(r);
 
-          //Here we apply Bayes' formula
            Double_t w[10];
-           for (i=0; i<AliESDtrack::kSPECIES; i++) w[i]=c[i]*r[i]/rcc;
+           w[0]=pid.GetProbability(AliPID::kElectron);
+           w[1]=pid.GetProbability(AliPID::kMuon);
+           w[2]=pid.GetProbability(AliPID::kPion);
+           w[3]=pid.GetProbability(AliPID::kKaon);
+           w[4]=pid.GetProbability(AliPID::kProton);
+
 
            if (TMath::Abs(code)==2212) prR->Fill(p);
            if (w[4]>w[3] && w[4]>w[2] && w[4]>w[1] && w[4]>w[0]) {//proton
@@ -185,14 +197,13 @@ Int_t AliESDComparison(const Char_t *dir=".") {
            }
 
           if (TMath::Abs(code)==211) piR->Fill(p);
-          if (w[2]>w[3] && w[2]>w[4] && w[2]>w[1] && w[2]>w[0]) {//pion
+          if (w[2]>w[4] && w[2]>w[3] && w[2]>w[0] && w[2]>w[1]) {//pion
              pisel++;
              piG->Fill(p);
              if (TMath::Abs(code)!=211) piF->Fill(p);
            }
        }
       }
-      delete event;
       cout<<"Number of selected ESD tracks : "<<nsel<<endl;
       cout<<"Number of selected pion ESD tracks : "<<pisel<<endl;
       cout<<"Number of selected kaon ESD tracks : "<<kasel<<endl;
@@ -202,6 +213,7 @@ Int_t AliESDComparison(const Char_t *dir=".") {
 
    } // ***** End of the loop over events
 
+   delete event;
    ef->Close();
 
    TCanvas *c1=(TCanvas*)gROOT->FindObject("c1");