PWGPP/analysisQA/processCFv2vsPt.C

   1 /***************************************************************
   2  processCFv2vsPt.C
   3  Post Processing of Flow QA task in Analysis QA train
   4  Clone copy of v2vsPt_0d.C from Ante
   5
   6  Modification Done // sjena
   7  To save file and putting ani unique naming convention
   8
   9 ***************************************************************/
  10
  11 // Name of the merged, large statistics file obtained with the merging macros:
  12 TString mergedFileName = "AnalysisResults.root";
  13
  14 // Final output file name holding correct final results for large statistics sample:
  15 TString outputFileName = "AnalysisResults.root";
  16
  17 void processCFv2vsPt(const char *filePath = "AnalysisResults.root",
  18                      TString suffix="eps",
  19                      const char *outfile="CFv2vsPt_output.root") {
  20  // Load flow libraries:
  21  gSystem->Load("libPWGflowBase");
  22
  23  // Let's go:
  24  TString mergedFileFullPathName(filePath);
  25  TFile *mergedFile = NULL;
  26  if(gSystem->AccessPathName(mergedFileFullPathName.Data(),kFileExists))
  27  {
  28   cout<<endl;
  29   cout<<" WARNING: Couldn't find a file: "<<mergedFileFullPathName.Data()<<endl;
  30   cout<<endl;
  31   exit(0);
  32  } else
  33    {
  34     // Create temporarily copy of <mergedFileName> if neccessary:
  35     if(!(mergedFileName == outputFileName))
  36     {
  37      TSystemFile *fileTemp = new TSystemFile(mergedFileFullPathName.Data(),".");
  38      fileTemp->Copy("mergedAnalysisResultsTemp.root");
  39      delete fileTemp;
  40     }
  41     // Access <mergedFileName>:
  42     mergedFile = TFile::Open(mergedFileFullPathName.Data(),"UPDATE");
  43   }
  44
  45  // Accessing the file and updating it:
  46  TList* mergedFileKeys = mergedFile->GetListOfKeys();
  47  for(Int_t i=0; i<mergedFileKeys->GetEntries(); i++)
  48  {
  49   TDirectory* directory = dynamic_cast<TDirectory*>(mergedFile->Get(mergedFileKeys->At(i)->GetName()));
  50   if (!directory) continue;
  51   TList* listTemp = directory->GetListOfKeys();
  52   if(!listTemp) continue;
  53   for (Int_t icent=0; icent<listTemp->GetEntries(); icent++)
  54   {
  55    TList* list = dynamic_cast<TList*>(directory->Get(listTemp->At(icent)->GetName()));
  56    if (!list) continue;
  57    // QC:
  58    if(TString(list->GetName()).Contains("QC"))
  59    {
  60     AliFlowAnalysisWithQCumulants* qc = new AliFlowAnalysisWithQCumulants();
  61     qc->GetOutputHistograms(list);
  62     Bool_t bApplyCorrectionForNUA = kFALSE; // apply correction for non-uniform acceptance
  63     qc->GetIntFlowFlags()->SetBinContent(3,(Int_t)bApplyCorrectionForNUA);
  64     qc->Finish();
  65     directory->Add(list,kTRUE);
  66     directory->Write(directory->GetName(),TObject::kSingleKey+TObject::kWriteDelete);
  67    } // if(TString(list->GetName()).Contains("QC"))
  68   } // for (Int_t icent=0; icent<listTemp->GetEntries(); icent++)
  69  } // for(Int_t i=0; i<mergedFileKeys->GetEntries(); i++)
  70
  71  // Close the final output file:
  72  delete mergedFile;
  73
  74  // Changing the final name, if neccessary:
  75  if(!(mergedFileName == outputFileName))
  76  {
  77   TSystemFile *outputFileFinal = new TSystemFile(mergedFileName.Data(),".");
  78   outputFileFinal->Rename(outputFileName.Data());
  79   delete outputFileFinal;
  80   TSystemFile *mergedFileFinal = new TSystemFile("mergedAnalysisResultsTemp.root",".");
  81   mergedFileFinal->Rename(mergedFileName.Data());
  82   delete mergedFileFinal;
  83  } // end of if(!(mergedFileName == outputFileName))
  84
  85  // Finally, make some nice plots...:
  86  // Access "v2 vs. pT" histogram from the specified output file:
  87   TFile *f = TFile::Open(filePath,"READ");
  88
  89  TDirectoryFile *df = (TDirectoryFile*)f->FindObjectAny("outputQCanalysisTPCstandalone");
  90  TList *l = NULL;
  91  TList *listTemp = df->GetListOfKeys();
  92  if(listTemp && listTemp->GetEntries() == 1)
  93  {
  94   TString listName = listTemp->At(0)->GetName();
  95   df->GetObject(listName.Data(),l);
  96  }
  97  TList *diffFlowList = dynamic_cast<TList*> l->FindObject("Differential Flow");
  98  diffFlowList = dynamic_cast<TList*> diffFlowList->FindObject("Results");
  99  diffFlowList = dynamic_cast<TList*> diffFlowList->FindObject("Differential flow (POI, p_{T})");
 100
 101  // v2{2} vs. pT:
 102  TH1D *v22 = dynamic_cast<TH1D*> diffFlowList->FindObject("fDiffFlow, POI, p_{T}, v'{2}");
 103  v22->SetMarkerStyle(kFullCircle);
 104  v22->SetMarkerColor(kBlue);
 105  v22->SetLineColor(kBlue);
 106
 107  // v2{4} vs. pT:
 108  TH1D *v24 = dynamic_cast<TH1D*> diffFlowList->FindObject("fDiffFlow, POI, p_{T}, v'{4}");
 109  v24->SetMarkerStyle(kFullSquare);
 110  v24->SetMarkerColor(kRed);
 111  v24->SetLineColor(kRed);
 112
 113  // Legend:
 114  TLegend *legend = new TLegend(0.15,0.67,0.37,0.87);
 115  legend->SetFillStyle(0); // white legend background
 116  legend->SetTextSize(0.04);
 117  legend->SetBorderSize(0.0);
 118  //legend->SetTextFont(22);
 119  legend->SetMargin(0.1);
 120  //legend->AddEntry(inputValues_g,"theoretical/input values","p");
 121  legend->AddEntry(v22,"v_{2}{2}","p");
 122  legend->AddEntry(v24,"v_{2}{4}","p");
 123
 124  // Style histogram:
 125  TH1D *hist = new TH1D("","",10,0.,10.);
 126  hist->SetStats(kFALSE);
 127  hist->SetTitle("");
 128  hist->GetXaxis()->SetTitle("p_{T}");
 129  hist->GetYaxis()->SetRangeUser(0.0,0.3044);
 130
 131  // Final drawing:
 132  // v2 vs. pT:
 133  TCanvas *c1 = new TCanvas("c1","v2 vs. pT");
 134  hist->Draw();
 135  legend->Draw("same");
 136  v22->Draw("psame");
 137  v24->Draw("psame");
 138
 139   c1->SaveAs(Form("fig_cf_flow_fDiffFlow.%s",suffix.Data()));
 140
 141  // Bias from non-uniform azimuthal acceptance (NUA):
 142  TList *intFlowList = dynamic_cast<TList*> l->FindObject("Integrated Flow");
 143  intFlowList = dynamic_cast<TList*> intFlowList->FindObject("Results");
 144  TH1D *fIntFlowDetectorBias = dynamic_cast<TH1D*> intFlowList->FindObject("fIntFlowDetectorBias");
 145  fIntFlowDetectorBias->SetStats(kFALSE);
 146  fIntFlowDetectorBias->SetTitle("Quantifying bias from non-uniform acceptance");
 147  fIntFlowDetectorBias->SetFillColor(kBlue-10);
 148  fIntFlowDetectorBias->GetXaxis()->SetRangeUser(0.,2.);
 149  fIntFlowDetectorBias->GetYaxis()->SetRangeUser(0.8044,1.0544);
 150  fIntFlowDetectorBias->SetBinError(1,0.); // stat. errors need to be reimplemented
 151  fIntFlowDetectorBias->SetBinError(2,0.); // stat. errors need to be reimplemented
 152
 153  // Final drawing:
 154  TCanvas *c2 = new TCanvas("c2","NUA");
 155  fIntFlowDetectorBias->Draw();
 156
 157  c2->SaveAs(Form("fig_cf_flow_IntFlowDetectorBias.%s",suffix.Data()));
 158
 159  f->Close();
 160
 161  cout<<endl;
 162
 163  //Added by sjena
 164  TFile *fout = TFile::Open(outfile,"UPDATE");
 165  fout->ls();
 166
 167  TDirectoryFile *cdd = NULL;
 168  cdd = (TDirectoryFile*)fout->Get("CF");
 169  if(!cdd) {
 170     Printf("Warning: CF <dir> doesn't exist, creating a new one");
 171     cdd = (TDirectoryFile*)fout->mkdir("CF");
 172  }
 173  cdd->cd();
 174  cdd->ls();
 175
 176
 177  v22->SetName(Form("fig_cf_flow_22", v22->GetName()));
 178  // v22->SetName(Form("fig_cf_flow_fDiffFlow22"));
 179  v22->Write();
 180
 181  v24->SetName(Form("fig_cf_flow_24", v24->GetName()));
 182  // v24->SetName(Form("fig_cf_flow_fDiffFlow24"));
 183  v24->Write();
 184
 185  fIntFlowDetectorBias->SetName(Form("fig_cf_flow_IntFlowDetectorBias", fIntFlowDetectorBias->GetName()));
 186  fIntFlowDetectorBias->Write();
 187
 188  fout->cd();
 189  fout->Close();
 190
 191
 192
 193 } // void void v2vsPt_0d()
 194