]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWG1/TRD/AliTRDcheckESD.cxx
update task to read results from AnalysisResults.root file
[u/mrichter/AliRoot.git] / PWG1 / TRD / AliTRDcheckESD.cxx
index 3f0e0df5e974854e0770c988d824aecee6834937..4a1be79f224a0792718ff5d3fad0b78f35f3331d 100644 (file)
@@ -61,10 +61,18 @@ ClassImp(AliTRDcheckESD)
 
 const Float_t AliTRDcheckESD::fgkxTPC = 290.;
 const Float_t AliTRDcheckESD::fgkxTOF = 365.;
-const UChar_t AliTRDcheckESD::fgkNgraph[kNrefs] ={
-0, 4, 2, 10};
+const UChar_t AliTRDcheckESD::fgkNgraph[AliTRDcheckESD::kNrefs] ={
+8, 4, 2, 20};
 FILE* AliTRDcheckESD::fgFile = NULL;
 
+const Float_t AliTRDcheckESD::fgkEvVertexZ = 15.;
+const Int_t   AliTRDcheckESD::fgkEvVertexN = 1;
+const Float_t AliTRDcheckESD::fgkTrkDCAxy  = 40.;
+const Float_t AliTRDcheckESD::fgkTrkDCAz   = 15.;
+const Int_t   AliTRDcheckESD::fgkNclTPC    = 100;
+const Float_t AliTRDcheckESD::fgkPt        = 0.2;
+const Float_t AliTRDcheckESD::fgkEta       = 0.9;
+
 //____________________________________________________________________
 AliTRDcheckESD::AliTRDcheckESD():
   AliAnalysisTaskSE()
@@ -120,8 +128,6 @@ void AliTRDcheckESD::UserCreateOutputObjects()
   //
   // Create Output Containers (TObjectArray containing 1D histograms)
   //
-  //OpenFile(0, "RECREATE");  
-
   Histos();
 }
 
@@ -136,30 +142,52 @@ Bool_t AliTRDcheckESD::GetRefFigure(Int_t ifig)
   if(!gPad){
     AliWarning("Please provide a canvas to draw results.");
     return kFALSE;
+  } else {
+    gPad->SetLogx(0);gPad->SetLogy(0);
+    gPad->SetMargin(0.125, 0.015, 0.1, 0.015);
   }
 
   const Char_t *title[20];
   TH1 *hF(NULL);
   if((hF=(TH1S*)gROOT->FindObject("hFcheckESD"))) delete hF;
   TLegend *leg(NULL);
+  TList *l(NULL); TVirtualPad *pad(NULL);
   TGraphErrors *g(NULL);TGraphAsymmErrors *ga(NULL);
   TObjArray *arr(NULL);
   switch(ifig){
   case kNCl: // number of clusters/track
-    ((TH1I*)fResults->At(kNCl))->Draw("c");
+    if(!(arr = (TObjArray*)fResults->At(kNCl))) return kFALSE;
+
+    leg = new TLegend(.83, .7, .99, .96);
+    leg->SetHeader("Species");
+    leg->SetBorderSize(0); leg->SetFillStyle(0);
+    for(Int_t ig(0); ig<fgkNgraph[kNCl]; ig++){
+      if(!(g = (TGraphErrors*)arr->At(ig))) return kFALSE;
+      if(!g->GetN()) continue;
+      g->Draw(ig?"pc":"apc"); leg->AddEntry(g, g->GetTitle(), "pl");
+      if(ig) continue;
+      hF=g->GetHistogram();
+      hF->SetXTitle("no of clusters");
+      hF->SetYTitle("entries"); 
+      hF->GetYaxis()->CenterTitle(1);
+      hF->GetYaxis()->SetTitleOffset(1.2);
+      hF->SetMinimum(5);
+    }
+    leg->Draw(); gPad->SetLogy();
     break;
   case kTRDstat: // Efficiency
     if(!(arr = (TObjArray*)fResults->At(kTRDstat))) return kFALSE;
-    leg = new TLegend(.65, .7, .95, .99);
+    leg = new TLegend(.62, .77, .98, .98);
     leg->SetHeader("TRD Efficiency");
-    leg->SetBorderSize(1); leg->SetFillColor(0);
+    leg->SetBorderSize(0); leg->SetFillStyle(0);
     title[0] = "Geometrical (TRDin/TPCout)";
     title[1] = "Tracking (TRDout/TRDin)";
     title[2] = "PID (TRDpid/TRDin)";
     title[3] = "Refit (TRDrefit/TRDin)";
     hF = new TH1S("hFcheckESD", ";p [GeV/c];Efficiency", 10, 0.1, 10.);
-    hF->SetMaximum(1.3);
+    hF->SetMaximum(1.4);
     hF->GetXaxis()->SetMoreLogLabels();
+    hF->GetYaxis()->CenterTitle(1);
     hF->Draw("p");
     for(Int_t ig(0); ig<fgkNgraph[kTRDstat]; ig++){
       if(!(g = (TGraphErrors*)arr->At(ig))) return kFALSE;
@@ -189,27 +217,49 @@ Bool_t AliTRDcheckESD::GetRefFigure(Int_t ifig)
     break;
   case kPtRes: // Pt resolution @ vertex
     if(!(arr = (TObjArray*)fResults->At(kPtRes))) return kFALSE;
-    gPad->SetMargin(0.1, 0.22, 0.1, 0.023);
-    leg = new TLegend(.78, .1, .99, .98);
-    leg->SetHeader("P_{t} @ DCA");
-    leg->SetBorderSize(1); leg->SetFillColor(0);
-    leg->SetTextAlign(22);
-    leg->SetTextFont(12);
-    leg->SetTextSize(0.03813559);
-    hF = new TH1S("hFcheckESD", ";p_{t} [GeV/c];#Delta p_{t} / p_{t} [%]", 10, 0.2, 10.);
+    gPad->Divide(2, 1, 1.e-5, 1.e-5); l=gPad->GetListOfPrimitives(); 
+    pad = ((TVirtualPad*)l->At(0)); pad->cd(); pad->SetLogx();
+    pad->SetMargin(0.1, 0.022, 0.1, 0.023);
+    hF = new TH1S("hFcheckESD", "ITS+TPC+TRD;p_{t} [GeV/c];#Delta p_{t} / p_{t} [%]", 10, 0.2, 10.);
     hF->SetMaximum(10.);hF->SetMinimum(-3.);
     hF->GetXaxis()->SetMoreLogLabels();
     hF->GetXaxis()->SetTitleOffset(1.2);
     hF->GetYaxis()->CenterTitle();
     hF->Draw("p");
-    for(Int_t ig(0); ig<fgkNgraph[kPtRes]; ig++){
-      if(!(g = (TGraphErrors*)arr->At(ig))) continue;//return kFALSE;
+    //for(Int_t ig(0); ig<fgkNgraph[kPtRes]/2; ig++){
+    for(Int_t ig(2); ig<6; ig++){
+      if(!(g = (TGraphErrors*)arr->At(ig))) continue;
       if(!g->GetN()) continue;
-      g->Draw("pl"); leg->AddEntry(g, g->GetTitle(), "pl");
+      g->Draw("pl");
       //PutTrendValue(name[id], g->GetMean(2));
       //PutTrendValue(Form("%sRMS", name[id]), g->GetRMS(2));
     }
-    leg->Draw();gPad->SetLogx();
+    pad = ((TVirtualPad*)l->At(1)); pad->cd(); pad->SetLogx();
+    pad->SetMargin(0.1, 0.22, 0.1, 0.023);
+    hF = (TH1*)hF->Clone("hFcheckESD1");
+    hF->SetTitle("ITS+TPC");
+    hF->SetMaximum(10.);hF->SetMinimum(-3.);
+    hF->Draw("p");
+    leg = new TLegend(.78, .1, .99, .98);
+    leg->SetHeader("P_{t} @ DCA");
+    leg->SetBorderSize(1); leg->SetFillColor(0);
+    leg->SetTextAlign(22);
+    leg->SetTextFont(12);
+    leg->SetTextSize(0.03813559);
+    {
+      Int_t nPlots(0);
+      //for(Int_t ig(fgkNgraph[kPtRes]/2); ig<fgkNgraph[kPtRes]; ig++){
+      for(Int_t ig(12); ig<16; ig++){
+        if(!(g = (TGraphErrors*)arr->At(ig))) continue;
+        if(!g->GetN()) continue;
+        nPlots++;
+        g->Draw("pl"); leg->AddEntry(g, g->GetTitle(), "pl");
+        //PutTrendValue(name[id], g->GetMean(2));
+        //PutTrendValue(Form("%sRMS", name[id]), g->GetRMS(2));
+      }
+      if(nPlots) leg->Draw();
+    }
+
     break;
   }
   return kTRUE;
@@ -222,7 +272,7 @@ void AliTRDcheckESD::UserExec(Option_t *){
   //
   fESD = dynamic_cast<AliESDEvent*>(InputEvent());
   fMC = MCEvent();
-  
+
   if(!fESD){
     AliError("ESD event missing.");
     return;
@@ -249,8 +299,42 @@ void AliTRDcheckESD::UserExec(Option_t *){
 
     // track status
     ULong_t status = esdTrack->GetStatus(); //PrintStatus(status);
-    if(!Bool_t(status & AliESDtrack::kTPCout)) continue;
-    if(esdTrack->GetKinkIndex(0) > 0) continue;
+
+    // track selection
+    Bool_t selected(kTRUE);
+    if(esdTrack->Pt() < fgkPt){ 
+      AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Pt[%5.2f]", fESD->GetEventNumberInFile(), itrk, esdTrack->Pt()));
+      selected = kFALSE;
+    }
+    if(TMath::Abs(esdTrack->Eta()) > fgkEta){
+      AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Eta[%5.2f]", fESD->GetEventNumberInFile(), itrk, TMath::Abs(esdTrack->Eta())));
+      selected = kFALSE;
+    }
+    if(!Bool_t(status & AliESDtrack::kTPCout)){
+      AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] !TPCout", fESD->GetEventNumberInFile(), itrk));
+      selected = kFALSE;
+    }
+    if(esdTrack->GetKinkIndex(0) > 0){
+      AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] Kink", fESD->GetEventNumberInFile(), itrk));
+      selected = kFALSE;
+    }
+    if(esdTrack->GetTPCNcls() < fgkNclTPC){ 
+      AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] NclTPC[%d]", fESD->GetEventNumberInFile(), itrk, esdTrack->GetTPCNcls()));
+      selected = kFALSE;
+    }
+    Float_t par[2], cov[3];
+    esdTrack->GetImpactParameters(par, cov);
+    if(IsCollision()){ // cuts on DCA
+      if(TMath::Abs(par[0]) > fgkTrkDCAxy){ 
+        AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] DCAxy[%f]", fESD->GetEventNumberInFile(), itrk, TMath::Abs(par[0])));
+        selected = kFALSE;
+      }
+      if(TMath::Abs(par[1]) > fgkTrkDCAz){ 
+        AliDebug(2, Form("Reject Ev[%4d] Trk[%3d] DCAz[%f]", fESD->GetEventNumberInFile(), itrk, TMath::Abs(par[1])));
+        selected = kFALSE;
+      }
+    }
+    if(!selected) continue;
 
     //Int_t nTPC(esdTrack->GetNcls(1));
     Int_t nTRD(esdTrack->GetNcls(2));
@@ -325,18 +409,19 @@ void AliTRDcheckESD::UserExec(Option_t *){
       h = (TH2I*)fHistos->At(kTRDstat);
       if(status & AliESDtrack::kTPCout) h->Fill(ptTRD, kTPCout);
       if(status & AliESDtrack::kTRDin) h->Fill(ptTRD, kTRDin);
-      if(kBarrel && (status & AliESDtrack::kTRDout)){ 
-        ((TH1*)fHistos->At(kNCl))->Fill(nTRD);
-        h->Fill(ptTRD, kTRDout);
-      }
+      if(kBarrel && (status & AliESDtrack::kTRDout)) h->Fill(ptTRD, kTRDout);
       if(kBarrel && (status & AliESDtrack::kTRDpid)) h->Fill(ptTRD, kTRDpid);
       if(kBarrel && (status & AliESDtrack::kTRDrefit)) h->Fill(ptTRD, kTRDref);
     }
-    if(HasMC() && kBarrel && (status & AliESDtrack::kTRDout)) {
+    Int_t idx(HasMC() ? Pdg2Idx(TMath::Abs(mcParticle->PdgCode())): 0)
+         ,sgn(esdTrack->Charge()<0?0:1);
+    if(kBarrel && kPhysPrim) {
       TH3 *h3 = (TH3S*)fHistos->At(kPtRes);
-      Int_t sgn = mcParticle->Charge()>0?1:-1;
-      h3->Fill(pt0, 1.e2*pt/pt0-1.e2, sgn*Pdg2Idx(TMath::Abs(mcParticle->PdgCode())));
+      Int_t offset = (status & AliESDtrack::kTRDrefit) ? 0 : 10; 
+      h3->Fill(pt0, 1.e2*(pt/pt0-1.), 
+        offset + 2*idx + sgn);
     }
+    ((TH1*)fHistos->At(kNCl))->Fill(nTRD, 2*idx + sgn);
     if(ip){
       h = (TH2I*)fHistos->At(kTRDmom);
       Float_t pTRD(0.);
@@ -361,17 +446,26 @@ TObjArray* AliTRDcheckESD::Histos()
 
   TH1 *h = NULL;
 
-  // clusters per tracklet
-  if(!(h = (TH1I*)gROOT->FindObject("hNCl"))){
-    h = new TH1I("hNCl", "Clusters per TRD track;N_{cl}^{TRD};entries", 100, 0., 200.);
+  // clusters per track
+  const Int_t kNpt(30);
+  Float_t Pt(0.2);
+  Float_t binsPt[kNpt+1];
+  for(Int_t i=0;i<kNpt+1; i++,Pt+=(TMath::Exp(i*i*.001)-1.)) binsPt[i]=Pt;
+  if(!(h = (TH2S*)gROOT->FindObject("hNCl"))){
+    h = new TH2S("hNCl", "Clusters per TRD track;N_{cl}^{TRD};SPECIES;entries", 60, 0., 180., 10, -0.5, 9.5);
+    TAxis *ay(h->GetYaxis());
+    ay->SetLabelOffset(0.015);
+    for(Int_t i(0); i<AliPID::kSPECIES; i++){
+      ay->SetBinLabel(2*i+1, Form("%s^{-}", AliPID::ParticleLatexName(i)));
+      ay->SetBinLabel(2*i+2, Form("%s^{+}", AliPID::ParticleLatexName(i)));
+    }
   } else h->Reset();
   fHistos->AddAt(h, kNCl); fNRefFigures++;
 
   // status bits histogram
-  const Int_t kNpt(10), kNbits(5);
-  Float_t Pt(0.1), Bits(.5);
-  Float_t binsPt[kNpt+1], binsBits[kNbits+1];
-  for(Int_t i=0;i<kNpt+1; i++,Pt+=(TMath::Exp(i*i*.015)-1.)) binsPt[i]=Pt;
+  const Int_t kNbits(5);
+  Float_t Bits(.5);
+  Float_t binsBits[kNbits+1];
   for(Int_t i=0; i<kNbits+1; i++,Bits+=1.) binsBits[i]=Bits;
   if(!(h = (TH2I*)gROOT->FindObject("hTRDstat"))){
     h = new TH2I("hTRDstat", "TRD status bits;p_{t} @ TRD [GeV/c];status;entries", kNpt, binsPt, kNbits, binsBits);
@@ -392,17 +486,20 @@ TObjArray* AliTRDcheckESD::Histos()
   if(!HasMC()) return fHistos;
 
   // pt resolution
-  const Int_t kNdpt(100), kNspec(2*AliPID::kSPECIES+1);
-  Float_t DPt(-3.), Spec(-AliPID::kSPECIES-0.5);
+  const Int_t kNdpt(100), kNspec(4*AliPID::kSPECIES);
+  Float_t DPt(-3.), Spec(-0.5);
   Float_t binsDPt[kNdpt+1], binsSpec[kNspec+1];
   for(Int_t i=0; i<kNdpt+1; i++,DPt+=6.e-2) binsDPt[i]=DPt;
   for(Int_t i=0; i<kNspec+1; i++,Spec+=1.) binsSpec[i]=Spec;
   if(!(h = (TH3S*)gROOT->FindObject("hPtRes"))){
     h = new TH3S("hPtRes", "P_{t} resolution @ DCA;p_{t}^{MC} [GeV/c];#Delta p_{t}/p_{t}^{MC} [%];SPECIES", kNpt, binsPt, kNdpt, binsDPt, kNspec, binsSpec);
     TAxis *az(h->GetZaxis());
+    az->SetLabelOffset(0.015);
     for(Int_t i(0); i<AliPID::kSPECIES; i++){
-      az->SetBinLabel(5-i, AliPID::ParticleLatexName(i));
-      az->SetBinLabel(7+i, AliPID::ParticleLatexName(i));
+      az->SetBinLabel(2*i+1, Form("%s^{-}", AliPID::ParticleLatexName(i)));
+      az->SetBinLabel(2*i+2, Form("%s^{+}", AliPID::ParticleLatexName(i)));
+      az->SetBinLabel(10+2*i+1, Form("%s^{-}", AliPID::ParticleLatexName(i)));
+      az->SetBinLabel(10+2*i+2, Form("%s^{+}", AliPID::ParticleLatexName(i)));
     }
   } else h->Reset();
   fHistos->AddAt(h, kPtRes);
@@ -411,22 +508,28 @@ TObjArray* AliTRDcheckESD::Histos()
 }
 
 //____________________________________________________________________
-Bool_t AliTRDcheckESD::Load(const Char_t *filename, const Char_t *name)
+Bool_t AliTRDcheckESD::Load(const Char_t *file, const Char_t *dir, const Char_t *name)
 {
 // Load data from performance file
 
-  if(!TFile::Open(filename)){
-    AliWarning(Form("Couldn't open file %s.", filename));
+  if(!TFile::Open(file)){
+    AliWarning(Form("Couldn't open file %s.", file));
     return kFALSE;
   }
-  TObjArray *o = NULL;
-  if(!(o = (TObjArray*)gFile->Get(name ? name : GetName()))){
-    AliWarning("Missing histogram container.");
+  if(dir){
+    if(!gFile->cd(dir)){
+      AliWarning(Form("Couldn't cd to %s in %s.", dir, file));
+      return kFALSE;
+    }
+  }
+  TObjArray *o(NULL);
+  const Char_t *tn=(name ? name : GetName());
+  if(!(o = (TObjArray*)gDirectory->Get(tn))){
+    AliWarning(Form("Missing histogram container %s.", tn));
     return kFALSE;
   }
   fHistos = (TObjArray*)o->Clone(GetName());
   gFile->Close();
-  SETBIT(fStatus, kLoad);
   return kTRUE;
 }
 
@@ -446,7 +549,7 @@ Bool_t AliTRDcheckESD::PutTrendValue(const Char_t *name, Double_t val)
 void AliTRDcheckESD::Terminate(Option_t *)
 {
 // Steer post-processing 
-  if(!IsLoad()){
+  if(!fHistos){
     fHistos = dynamic_cast<TObjArray *>(GetOutputData(1));
     if(!fHistos){
       AliError("Histogram container not found in output");
@@ -466,6 +569,21 @@ void AliTRDcheckESD::Terminate(Option_t *)
       fResults->AddAt(arr = new TObjArray(fgkNgraph[iref]), iref);
       arr->SetName(name[iref]);  arr->SetOwner();
       switch(iref){
+      case kNCl:
+        for(Int_t ig(0); ig<fgkNgraph[iref]; ig++){
+          arr->AddAt(g = new TGraphErrors(), ig);
+          g->SetLineColor(ig+1); 
+          g->SetMarkerColor(ig+1); 
+          g->SetMarkerStyle(ig+20); 
+          g->SetName(Form("s%d", ig));
+          switch(ig){
+          case 0: g->SetTitle("ALL"); break;
+          case 1: g->SetTitle("NEG"); break;
+          case 2: g->SetTitle("POS"); break;
+          default: g->SetTitle(AliPID::ParticleLatexName(ig-3)); break;
+          };
+        }
+        break;
       case kTRDmom:
         for(Int_t ig(0); ig<fgkNgraph[iref]; ig++){
           arr->AddAt(g = new TGraphAsymmErrors(), ig);
@@ -475,17 +593,30 @@ void AliTRDcheckESD::Terminate(Option_t *)
         }
         break;
       case kPtRes:
-        for(Int_t im(fgkNgraph[iref]/2), ig(im), idx(0); ig--; idx+=ig){
-         arr->AddAt(g = new TGraphErrors(), ig);
+        for(Int_t idx(0); idx<AliPID::kSPECIES; idx++){
+          Int_t ig(2*idx);
+          arr->AddAt(g = new TGraphErrors(), ig);
           g->SetLineColor(kRed-idx); 
           g->SetMarkerColor(kRed-idx); 
-          g->SetMarkerStyle(20+ig); 
-          g->SetNameTitle(Form("s%d", ig), Form("res %s", AliPID::ParticleLatexName(ig)));
-          arr->AddAt(g = new TGraphErrors(), im+ig);
+          g->SetMarkerStyle(20+idx); 
+          g->SetNameTitle(Form("s%d", ig), Form("res %s", AliPID::ParticleLatexName(idx)));
+          arr->AddAt(g = new TGraphErrors(), ig+1);
           g->SetLineColor(kBlue-idx); 
           g->SetMarkerColor(kBlue-idx); 
-          g->SetMarkerStyle(20+ig); 
-          g->SetNameTitle(Form("m%d", ig), Form("sys %s", AliPID::ParticleLatexName(ig)));
+          g->SetMarkerStyle(20+idx); 
+          g->SetNameTitle(Form("m%d", ig+1), Form("sys %s", AliPID::ParticleLatexName(idx)));
+
+          ig+=10;
+          arr->AddAt(g = new TGraphErrors(), ig);
+          g->SetLineColor(kRed-idx); 
+          g->SetMarkerColor(kRed-idx); 
+          g->SetMarkerStyle(20+idx); 
+          g->SetNameTitle(Form("s%d", ig), Form("sigma %s", AliPID::ParticleLatexName(idx)));
+          arr->AddAt(g = new TGraphErrors(), ig+1);
+          g->SetLineColor(kBlue-idx); 
+          g->SetMarkerColor(kBlue-idx); 
+          g->SetMarkerStyle(20+idx); 
+          g->SetNameTitle(Form("m%d", ig+1), Form("mean %s", AliPID::ParticleLatexName(idx)));
         }
         break;
       default:
@@ -499,14 +630,54 @@ void AliTRDcheckESD::Terminate(Option_t *)
       }
     }
   }
+  TH1 *h1[2] = {NULL, NULL};
+  TH2I *h2(NULL);
+  TAxis *ax(NULL);
+
+  // No of clusters
+  if(!(h2 = (TH2I*)fHistos->At(kNCl))) return;
+  ax = h2->GetXaxis();
+  arr = (TObjArray*)fResults->At(kNCl);
+  // All tracks
+  h1[0] = h2->ProjectionX("Ncl_px");
+  TGraphErrors *ge=(TGraphErrors*)arr->At(0);
+  for(Int_t ib=2; ib<=ax->GetNbins(); ib++){
+    ge->SetPoint(ib-2, ax->GetBinCenter(ib), h1[0]->GetBinContent(ib));
+  }
+  // All charged tracks
+  TH1 *hNclCh[2] = {(TH1D*)h1[0]->Clone("NEG"), (TH1D*)h1[0]->Clone("POS")};
+  hNclCh[0]->Reset();hNclCh[1]->Reset();
+  for(Int_t is(1); is<=AliPID::kSPECIES; is++){
+    hNclCh[0]->Add(h2->ProjectionX("Ncl_px", 2*is-1, 2*is-1)); // neg
+    hNclCh[1]->Add(h2->ProjectionX("Ncl_px", 2*is, 2*is));     // pos
+  }
+  if(Int_t(hNclCh[0]->GetEntries())){
+    ge=(TGraphErrors*)arr->At(1);
+    for(Int_t ib=2; ib<=ax->GetNbins(); ib++){
+      ge->SetPoint(ib-2, ax->GetBinCenter(ib), hNclCh[0]->GetBinContent(ib));
+    }
+  }
+  if(Int_t(hNclCh[1]->GetEntries())){
+    ge=(TGraphErrors*)arr->At(2);
+    for(Int_t ib=2; ib<=ax->GetNbins(); ib++){
+      ge->SetPoint(ib-2, ax->GetBinCenter(ib), hNclCh[1]->GetBinContent(ib));
+    }
+  }
+  // Species wise
+  for(Int_t is(1); is<=AliPID::kSPECIES; is++){
+    h1[0] = h2->ProjectionX("Ncl_px", 2*is-1, 2*is);
+    if(!Int_t(h1[0]->GetEntries())) continue;
+    ge=(TGraphErrors*)arr->At(2+is);
+    for(Int_t ib=2; ib<=ax->GetNbins(); ib++){
+      ge->SetPoint(ib-2, ax->GetBinCenter(ib), h1[0]->GetBinContent(ib));
+    }
+  }
   fNRefFigures = 1;
 
   // EFFICIENCY
   // geometrical efficiency
-  TH2I *h2(NULL);
   if(!(h2 = (TH2I*)fHistos->At(kTRDstat))) return;
   arr = (TObjArray*)fResults->At(kTRDstat);
-  TH1 *h1[2] = {NULL, NULL};
   h1[0] = h2->ProjectionX("checkESDx0", kTPCout, kTPCout);
   h1[1] = h2->ProjectionX("checkESDx1", kTRDin, kTRDin);
   Process(h1, (TGraphErrors*)arr->At(0));
@@ -530,7 +701,7 @@ void AliTRDcheckESD::Terminate(Option_t *)
   if(!(h2 = dynamic_cast<TH2I*>(fHistos->At(kTRDmom)))) return;
   arr = (TObjArray*)fResults->At(kTRDmom);
   TGraphAsymmErrors *g06 = (TGraphAsymmErrors*)arr->At(0), *g09 = (TGraphAsymmErrors*)arr->At(1);
-  TAxis *ax=h2->GetXaxis();
+  ax=h2->GetXaxis();
   const Int_t nq(4);
   const Double_t xq[nq] = {0.05, 0.2, 0.8, 0.95};
   Double_t yq[nq];
@@ -554,12 +725,17 @@ void AliTRDcheckESD::Terminate(Option_t *)
   arr = (TObjArray*)fResults->At(kPtRes);
   TAxis *az(h3->GetZaxis());
   for(Int_t i(0); i<AliPID::kSPECIES; i++){
-    az->SetRange(5-i, 5-i); 
-    gg[1] = (TGraphErrors*)arr->At(4-i);
-    gg[0] = (TGraphErrors*)arr->At(9-i);
+    Int_t idx(2*i);
+    az->SetRange(idx+1, idx+2); 
+    gg[1] = (TGraphErrors*)arr->At(idx);
+    gg[0] = (TGraphErrors*)arr->At(idx+1);
+    Process2D((TH2*)h3->Project3D("yx"), gg);
+
+    idx+=10;
+    az->SetRange(idx+1, idx+2); 
+    gg[1] = (TGraphErrors*)arr->At(idx);
+    gg[0] = (TGraphErrors*)arr->At(idx+1);
     Process2D((TH2*)h3->Project3D("yx"), gg);
-    //az->SetRange(7+i, 7+i);
-    //Process2D((TH2*)h3->Project3D("yx"), gg);
   }
   fNRefFigures++;
 }
@@ -568,13 +744,13 @@ void AliTRDcheckESD::Terminate(Option_t *)
 Int_t AliTRDcheckESD::Pdg2Idx(Int_t pdg)
 {
   switch(pdg){
-  case kElectron: return AliPID::kElectron+1;  
-  case kMuonMinus: return AliPID::kMuon+1;  
-  case kPiPlus: return AliPID::kPion+1;  
-  case kKPlus: return AliPID::kKaon+1;
-  case kProton: return AliPID::kProton+1;
+  case kElectron: return AliPID::kElectron;  
+  case kMuonMinus: return AliPID::kMuon;  
+  case kPiPlus: return AliPID::kPion;  
+  case kKPlus: return AliPID::kKaon;
+  case kProton: return AliPID::kProton;
   } 
-  return 0;
+  return -1;
 }
 
 //____________________________________________________________________