PWGCF/Correlations/macros/underlyingevent/plots.C

   1 void loadlibs()
   2 {
   3   gSystem->Load("libANALYSIS");
   4   gSystem->Load("libANALYSISalice");
   5   gSystem->Load("libCORRFW");
   6   gSystem->Load("libJETAN");
   7   gSystem->Load("libPWG4JetTasks");
   8 }
   9
  10 void PlotTrackingEfficiencies()
  11 {
  12   TGrid::Connect("alien://");
  13
  14   loadlibs();
  15
  16   ifstream in;
  17   in.open("list");
  18
  19   Int_t i=0;
  20
  21   legend = new TLegend(0.5, 0.1, 0.9, 0.5);
  22   legend->SetNColumns(2);
  23
  24   TString line;
  25   while (in.good())
  26   {
  27     in >> line;
  28
  29     if (line.Length() == 0)
  30       continue;
  31
  32     TString fileName;
  33     //fileName.Form("%s/%s/PWG4_JetTasksOutput.root", "alien:///alice/cern.ch/user/k/kleinb/analysisESD/LHC10d4/output_pwg4train_LHC10d4_101001a", line.Data());
  34     fileName.Form("%s/%s/PWG4_JetTasksOutput.root", "maps", line.Data());
  35     Printf("%s", fileName.Data());
  36     file = TFile::Open(fileName);
  37     if (!file)
  38       continue;
  39     list = (TList*) file->Get("PWG4_LeadingTrackUE/histosLeadingTrackUE");
  40     AliUEHistograms* h = (AliUEHistograms*) list->FindObject("AliUEHistograms");
  41     h->SetEtaRange(-0.79, 0.79);
  42
  43     TH1::AddDirectory(0);
  44     eff = (TH1*) h->GetNumberDensitypT()->GetTrackingEfficiency(1)->Clone(Form("trackingefficiency%s", line.Data()));
  45
  46     delete h;
  47     //delete list;
  48     file->Close();
  49
  50     eff->SetLineColor(i+1);
  51     eff->GetXaxis()->SetRangeUser(0, 10);
  52     eff->GetYaxis()->SetRangeUser(0.6, 0.8);
  53     eff->Draw((i == 0) ? "" : "SAME");
  54
  55     legend->AddEntry(eff, line);
  56
  57     i++;
  58     //if (i > 1)
  59     //  break;
  60   }
  61
  62   legend->Draw();
  63
  64   in.close();
  65 }
  66
  67 void TrackingEfficiencySpecies(const char* fileName = "PWG4_JetTasksOutput.root")
  68 {
  69   loadlibs();
  70
  71   file = TFile::Open(fileName);
  72   list = (TList*) file->Get("PWG4_LeadingTrackUE/histosLeadingTrackUE");
  73   AliUEHistograms* h = (AliUEHistograms*) list->FindObject("AliUEHistograms");
  74   h->SetEtaRange(-0.79, 0.79);
  75
  76   eff = (TH2*) h->GetNumberDensitypT()->GetTrackEfficiency(AliUEHist::kCFStepAnaTopology, AliUEHist::kCFStepTrackedOnlyPrim, 1, 2);
  77   eff->Draw("colz");
  78
  79   for (Int_t i=0; i<4; i++)
  80   {
  81     proj = eff->ProjectionX(Form("p%d", i), i+1, i+1);
  82     proj->SetLineColor(i+1);
  83     proj->Draw((i==0) ? "" : "SAME");
  84   }
  85 }
  86
  87 void CheckTrackingEfficiency(const char* reference = 0, const char* fileName = "PWG4_JetTasksOutput.root", Bool_t all = kFALSE)
  88 {
  89   loadlibs();
  90
  91   file = TFile::Open(fileName);
  92   list = (TList*) file->Get("PWG4_LeadingTrackUE/histosLeadingTrackUE");
  93   tree = (TTree*) list->FindObject("UEAnalysisSettings");
  94
  95   if (reference)
  96   {
  97     file2 = TFile::Open(reference);
  98     refHist = (TH1D*) file2->Get("trackingefficiency");
  99   }
 100   else
 101   {
 102     AliUEHistograms* h = (AliUEHistograms*) list->FindObject("AliUEHistograms");
 103     h->SetEtaRange(-0.79, 0.79);
 104     refHist = h->GetNumberDensitypT()->GetTrackingEfficiency(1);
 105   }
 106
 107   refHist->SetLineWidth(3);
 108   refHist->GetListOfFunctions()->Clear();
 109   refHist->Draw();
 110
 111   TH1D* hist = 0;
 112
 113   tree->SetBranchAddress("fkTrackingEfficiency", &hist);
 114
 115   for (Int_t i=0; i<tree->GetEntries(); i++)
 116   {
 117     tree->GetEntry(i);
 118     hist->GetListOfFunctions()->Clear();
 119     hist->SetLineColor(i+2);
 120     hist->DrawCopy("SAME");
 121     if (!all)
 122       break;
 123     Printf("%d", i);
 124   }
 125 }
 126
 127 void PlotSingleTrackingEfficiency(const char* fileName, Int_t what = 0)
 128 {
 129   loadlibs();
 130
 131   file = TFile::Open(fileName);
 132   list = (TList*) file->Get("PWG4_LeadingTrackUE/histosLeadingTrackUE");
 133   AliUEHistograms* h = (AliUEHistograms*) list->FindObject("AliUEHistograms");
 134   h->SetEtaRange(-0.79, 0.79);
 135
 136   if (what == 0)
 137     eff = (TH2*) h->GetNumberDensitypT()->GetTrackEfficiency(AliUEHist::kCFStepAnaTopology, AliUEHist::kCFStepTrackedOnlyPrim, 0, 1);
 138   else
 139     eff = (TH2*) h->GetNumberDensitypT()->GetTrackingContamination();
 140
 141   eff->Draw("colz");
 142 }
 143
 144 void ExtendTrackingEfficiency(const char* fileName)
 145 {
 146   loadlibs();
 147
 148   file = TFile::Open(fileName);
 149   list = (TList*) file->Get("PWG4_LeadingTrackUE/histosLeadingTrackUE");
 150   AliUEHistograms* h = (AliUEHistograms*) list->FindObject("AliUEHistograms");
 151   h->SetEtaRange(-0.79, 0.79);
 152
 153   h->GetUEHist(0)->ExtendTrackingEfficiency(1);
 154 }
 155
 156 void PlotSystUncertainties()
 157 {
 158   //
 159 }
 160