]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGPP/analysisQA/processHFEQAtask.C
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
Analysis QA post-processing macros and script
[u/mrichter/AliRoot.git] / PWGPP / analysisQA / processHFEQAtask.C
diff --git a/PWGPP/analysisQA/processHFEQAtask.C b/PWGPP/analysisQA/processHFEQAtask.C
new file mode 100644 (file)
index 0000000..22588fd
--- /dev/null
+++ b/PWGPP/analysisQA/processHFEQAtask.C
@@ -0,0 +1,290 @@
+/***************************************************************
+ processHFEQAtask:
+
+ Post Processing of HF QA task in Analysis QA train 
+ 
+ Modification Done // sjena
+ To save file and putting an unique naming convention
+
+***************************************************************/
+
+void processHFEQAtask(const char *fnamedata = "AnalysisResults.root", 
+                     TString suffix = "eps",
+                     const char * outfile = "HFEQAtask_output.root") {
+
+  const char *fnamedir  = "t131073f16s0p1TPC110r60p80ITS4Pi2DCAr100z200TOF30TPCe50V0D2er8i0t-20t50";
+  const char *fnamelist = "list_t131073f16s0p1TPC110r60p80ITS4Pi2DCAr100z200TOF30TPCe50V0D2er8i0t-20t50";
+
+
+  gSystem->Load("libANALYSIS.so");
+  gSystem->Load("libANALYSISalice.so");
+  gSystem->Load("libCORRFW.so");
+  gSystem->Load("libTENDER.so");
+  gSystem->Load("libPWGPP.so");
+  gSystem->Load("libPWGflowBase.so");
+  gSystem->Load("libPWGflowTasks.so");
+  gSystem->Load("libPWGHFhfe.so");
+
+  
+  gStyle->SetPalette(1);
+  gStyle->SetOptStat(1111);
+  gStyle->SetPadBorderMode(0);
+  gStyle->SetCanvasColor(10);
+  gStyle->SetPadLeftMargin(0.13);
+  gStyle->SetPadRightMargin(0.13);
+  gStyle->SetPadBottomMargin(0.13);
+  gStyle->SetPadTopMargin(0.13);
+
+  
+  //////////////////////
+  // Data
+  //////////////////////
+  TFile *indata = TFile::Open(fnamedata);
+  indata->ls();
+  TDirectoryFile *d = (TDirectoryFile *) indata->Get(fnamedir);
+
+  if(!d) {
+    Printf("FATAL: %s :  Not found", fnamedir);
+    Printf("Exiting <<<<<<<<<<<<<<<< processHFEQAtask >>>>>>>>>>>>>");
+    return;
+  }
+
+  TList *qadata = (TList *) d->Get(fnamelist);
+
+  
+  if(!qadata) {
+    Printf("FATAL: %s :  Not found", fnamelist);
+    Printf("Exiting <<<<<<<<<<<<<<<< processHFEQAtask >>>>>>>>>>>>>");
+    return;
+  }
+
+
+
+  TList *qaadata = (TList *)qadata->FindObject("HFEpidQA");
+ 
+  if(!qaadata) {
+    Printf("FATAL: HFEpidQA   Not found");
+    Printf("Exiting <<<<<<<<<<<<<<<< processHFEQAtask >>>>>>>>>>>>>");
+    return;
+  }
+
+
+  // Make Plots for TPC
+  AliHFEtpcPIDqa *tpcqadata = (AliHFEtpcPIDqa *)qaadata->FindObject("TPCQA");
+  AliHFEcollection *collectiontpc = tpcqadata->GetHistograms();
+  THnSparseF *tpcsparseF = (THnSparseF *) collectiontpc->Get("tpcnSigma");
+ 
+  // Make Plots for TOF
+  AliHFEtofPIDqa *tofqadata = (AliHFEtofPIDqa *)qaadata->FindObject("TOFQA");
+  AliHFEcollection *collectiontof = tofqadata->GetHistoCollection();
+  THnSparseF *tofsparseF = (THnSparseF *) collectiontof->Get("tofnSigma");
+  THnSparseF *toftpcsparseF = (THnSparseF *) collectiontof->Get("tofMonitorTPC");
+  
+
+  ////////////////
+  // Projection
+  /////////////////
+
+  TAxis *pidaxistpc = tpcsparseF->GetAxis(0);
+  TAxis *pidaxistof = tofsparseF->GetAxis(0);
+  TAxis *pidaxistoftpc = toftpcsparseF->GetAxis(0);
+
+  TAxis *stepaxistpc = tpcsparseF->GetAxis(3);
+  TAxis *stepaxistof = tofsparseF->GetAxis(3);
+  TAxis *stepaxistoftpc = toftpcsparseF->GetAxis(3);
+
+  TAxis *centralityaxistpc = tpcsparseF->GetAxis(4);
+  TAxis *centralityaxistof = tofsparseF->GetAxis(4);
+  TAxis *centralityaxistoftpc = toftpcsparseF->GetAxis(4);
+
+  stepaxistpc->SetRange(1,1);
+  stepaxistof->SetRange(1,1);
+  stepaxistoftpc->SetRange(2,2);
+
+  Int_t nbinsc = centralityaxistpc->GetNbins();
+  printf("There are %d centrality bins \n",nbinsc);
+  
+
+  //
+  centralityaxistpc->SetRange(8,9);
+  centralityaxistof->SetRange(8,9);
+  centralityaxistoftpc->SetRange(8,9);
+  stepaxistoftpc->SetRange(1,1);  
+  TH2D *toftpc2Dsumper = toftpcsparseF->Projection(2,1);
+  toftpc2Dsumper->SetName("toftpc2Dsumper_21"); 
+  TH2D *tof2Dsumper = tofsparseF->Projection(2,1);
+  tof2Dsumper->SetName("tof2Dsumper_21");
+
+  stepaxistof->SetRange(2,2);
+  TH2D *tof2Dsumbper = tofsparseF->Projection(2,1);
+  tof2Dsumbper->SetName("tof2Dsumbper_21");
+
+  TH2D *tpc2Dsumper = tpcsparseF->Projection(2,1);
+  tpc2Dsumper->SetName("toftpc2Dsumper_21");
+
+  stepaxistpc->SetRange(2,2);
+  TH2D *tpc2Dsumbper = tpcsparseF->Projection(2,1);
+  tpc2Dsumbper->SetName("tpc2Dsumbper_21");
+
+  //
+  centralityaxistpc->SetRange(0,1);
+  centralityaxistof->SetRange(0,1);
+  centralityaxistoftpc->SetRange(0,1);
+  stepaxistpc->SetRange(1,1);
+  stepaxistof->SetRange(1,1);
+  stepaxistoftpc->SetRange(1,1);
+  TH2D *toftpc2Dsumc = toftpcsparseF->Projection(2,1);
+  toftpc2Dsumc->SetName("toftpc2Dsumc_21");
+
+  TH2D *tof2Dsumc = tofsparseF->Projection(2,1);
+  tof2Dsumc->SetName("tof2Dsumc_21");
+
+  stepaxistof->SetRange(2,2);
+
+  TH2D *tof2Dsumbc = tofsparseF->Projection(2,1);
+  tof2Dsumbc->SetName("tof2Dsumbc_21");
+
+  TH2D *tpc2Dsumc = tpcsparseF->Projection(2,1);
+  tpc2Dsumc->SetName("tpc2Dsumc_21");
+
+  stepaxistpc->SetRange(2,2);
+  TH2D *tpc2Dsumbc = tpcsparseF->Projection(2,1);
+  tpc2Dsumbc->SetName("tpc2Dsumbc_21");
+
+  
+  ///////////////
+  // Plots
+  ///////////////
+
+  //added by sjena 
+  TFile *fout = TFile::Open(outfile,"UPDATE");
+  fout->ls();
+  
+  TDirectoryFile *cdd = NULL;
+  cdd = (TDirectoryFile*)fout->Get("HF");
+  if(!cdd) {
+    Printf("Warning: HF <dir> doesn't exist, creating a new one");
+    cdd = (TDirectoryFile*)fout->mkdir("HF");
+  }
+  cdd->cd();
+  cdd->ls();
+
+
+
+  // Draw Plots
+  TCanvas *cElectronsTPCper = new TCanvas("ElectronTPCPID_70-90%", "ElectronTPCPID_70-90%");
+  cElectronsTPCper->Divide(3,1);
+  cElectronsTPCper->cd(1);
+  gPad->SetLogz();
+  gPad->SetLogx();
+  gPad->SetTicks();
+  tof2Dsumper->SetStats(0);
+  tof2Dsumper->SetTitle("TOF 70-90% Pb-Pb");
+  tof2Dsumper->GetXaxis()->SetTitle("p_{t} (GeV/c)");
+  tof2Dsumper->GetYaxis()->SetTitle("TOF time_{e} - expected time|_{e}  (#sigma)");
+  tof2Dsumper->GetXaxis()->SetRangeUser(0.3,10.0);
+  tof2Dsumper->GetXaxis()->SetTitleSize(0.05);
+  tof2Dsumper->GetYaxis()->SetTitleSize(0.05);
+  tof2Dsumper->Draw("colz");
+  
+  //Added By Satya
+  tof2Dsumper->SetName(Form("fig_hf_1_%s", tof2Dsumper->GetName()));
+  tof2Dsumper->Write();
+
+  cElectronsTPCper->cd(2);
+  gPad->SetLogz();
+  gPad->SetLogx();
+  gPad->SetTicks();
+  tof2Dsumbper->SetStats(0);
+  tof2Dsumbper->SetTitle("TOF cut 70-90% Pb-Pb");
+  tof2Dsumbper->GetXaxis()->SetTitle("p_{t} (GeV/c)");
+  tof2Dsumbper->GetYaxis()->SetTitle("TOF time_{e} - expected time|_{e}  (#sigma)");
+  tof2Dsumbper->GetXaxis()->SetRangeUser(0.3,10.0);
+  tof2Dsumbper->GetXaxis()->SetTitleSize(0.05);
+  tof2Dsumbper->GetYaxis()->SetTitleSize(0.05);
+  tof2Dsumbper->Draw("colz");
+
+
+  //Added By Satya
+  tof2Dsumbper->SetName(Form("fig_hf_2_%s", tof2Dsumbper->GetName()));
+  tof2Dsumbper->Write();
+
+  cElectronsTPCper->cd(3);
+  gPad->SetLogz();
+  gPad->SetLogx();
+  gPad->SetTicks();
+  tpc2Dsumper->SetStats(0);
+  tpc2Dsumper->SetTitle("TPC 70-90% Pb-Pb");
+  tpc2Dsumper->GetXaxis()->SetTitle("p_{t} (GeV/c)");
+  tpc2Dsumper->GetYaxis()->SetTitle("TPC dE/dx - expected dE/dx|_{e}  (#sigma)");
+  tpc2Dsumper->GetXaxis()->SetRangeUser(0.3,10.0);
+  tpc2Dsumper->GetXaxis()->SetTitleSize(0.05);
+  tpc2Dsumper->GetYaxis()->SetTitleSize(0.05);
+  tpc2Dsumper->Draw("colz");
+  //Added by satya
+
+  
+  tpc2Dsumper->SetName(Form("fig_hf_3_%s", tpc2Dsumper->GetName()));
+  tpc2Dsumper->Write();
+
+  cElectronsTPCper->SaveAs(Form("fig_hf_ElectronTPCPID_70-90.%s",suffix.Data()));
+
+
+
+  TCanvas *cElectronsTPCc = new TCanvas("ElectronTPCPID_0-10%", "ElectronTPCPID_0-10%");
+  cElectronsTPCc->Divide(3,1);
+  cElectronsTPCc->cd(1);
+  gPad->SetLogz();
+  gPad->SetLogx();
+  gPad->SetTicks();
+  tof2Dsumc->SetStats(0);
+  tof2Dsumc->SetTitle("TOF 0-10% Pb-Pb");
+  tof2Dsumc->GetXaxis()->SetTitle("p_{t} (GeV/c)");
+  tof2Dsumc->GetYaxis()->SetTitle("TOF time_{e} - expected time|_{e}  (#sigma)");
+  tof2Dsumc->GetXaxis()->SetRangeUser(0.3,10.0);
+  tof2Dsumc->GetXaxis()->SetTitleSize(0.05);
+  tof2Dsumc->GetYaxis()->SetTitleSize(0.05);
+  tof2Dsumc->Draw("colz");
+
+  tof2Dsumc->SetName(Form("fig_hf_4_%s", tof2Dsumc->GetName()));
+  tof2Dsumc->Write();
+
+  cElectronsTPCc->cd(2);
+  gPad->SetLogz();
+  gPad->SetLogx();
+  gPad->SetTicks();
+  tof2Dsumbc->SetStats(0);
+  tof2Dsumbc->SetTitle("TOF cut 0-10% Pb-Pb");
+  tof2Dsumbc->GetXaxis()->SetTitle("p_{t} (GeV/c)");
+  tof2Dsumbc->GetYaxis()->SetTitle("TOF time_{e} - expected time|_{e}  (#sigma)");
+  tof2Dsumbc->GetXaxis()->SetRangeUser(0.3,10.0);
+  tof2Dsumbc->GetXaxis()->SetTitleSize(0.05);
+  tof2Dsumbc->GetYaxis()->SetTitleSize(0.05);
+  tof2Dsumbc->Draw("colz");
+
+  tof2Dsumbc->SetName(Form("fig_hf_5_%s", tof2Dsumbc->GetName()));
+  tof2Dsumbc->Write();
+
+  cElectronsTPCc->cd(3);
+  gPad->SetLogz();
+  gPad->SetLogx();
+  gPad->SetTicks();
+  tpc2Dsumc->SetStats(0);
+  tpc2Dsumc->SetTitle("TPC 0-10% Pb-Pb");
+  tpc2Dsumc->GetXaxis()->SetTitle("p_{t} (GeV/c)");
+  tpc2Dsumc->GetYaxis()->SetTitle("TPC dE/dx - expected dE/dx|_{e}  (#sigma)");
+  tpc2Dsumc->GetXaxis()->SetRangeUser(0.3,10.0);
+  tpc2Dsumc->GetXaxis()->SetTitleSize(0.05);
+  tpc2Dsumc->GetYaxis()->SetTitleSize(0.05);
+  tpc2Dsumc->Draw("colz");
+  // Added by satya
+
+  tpc2Dsumc->SetName(Form("fig_hf_6_%s", tpc2Dsumc->GetName()));
+  tpc2Dsumc->Write();
+
+  cElectronsTPCc->SaveAs(Form("fig_hf_ElectronTPCPID_0-10.%s",suffix.Data()));
+
+  fout->Close();
+
+
+}