PWG0/dNdEta/drawSystematicsNew.C

   1
   2 Int_t markers[] = {20,21,22,23,28,29};
   3 Int_t colors[]  = {1,2,3,4,6,8,102};
   4
   5 void DrawpiKpAndCombinedZOnly(Float_t upperPtLimit=0.99)
   6 {
   7   //gROOT->ProcessLine(".L drawPlots.C");
   8   gSystem->Load("libPWG0base");
   9
  10   const char* fileNames[] = { "systematics.root", "systematics.root", "systematics.root", "correction_map.root" };
  11   const char* folderNames[] = { "correction_0", "correction_1", "correction_2", "dndeta_correction" };
  12   const char* legendNames[] = { "#pi", "K", "p", "standard" };
  13   Int_t folderCount = 3;
  14
  15
  16
  17   TH2F* h2DCorrections[4];
  18   TH1F* h1DCorrections[4];
  19   for (Int_t i=0; i<4; i++) {
  20     TFile::Open(fileNames[i]);
  21     AlidNdEtaCorrection* correctionTmp = new AlidNdEtaCorrection(folderNames[i],folderNames[i]);
  22     correctionTmp->LoadHistograms();
  23
  24     //    h2DCorrections[i] = correctionTmp->GetTrack2ParticleCorrection()->GetTrackCorrection()->Get2DCorrectionHistogram("yz",-1,1);
  25
  26     h1DCorrections[i] = correctionTmp->GetTrack2ParticleCorrection()->GetTrackCorrection()->Get1DCorrectionHistogram("z",-10,10,0.8,0.8);
  27   }
  28
  29   TH2F* null = new TH2F("","",100,0.1,10,100,0.5,9.99);
  30   null->SetXTitle("p_{T} (GeV/c)");
  31   null->SetXTitle("Correction");
  32
  33   null->Draw();
  34
  35   //h1DCorrections[0]->SetMaximum(5);
  36   //h1DCorrections[0]->Draw();
  37   //  h2DCorrections[0]->Draw("colz");
  38   for (Int_t i=1; i<4; i++) {
  39     h1DCorrections[i]->Draw("same");
  40   }
  41
  42 }
  43
  44
  45 void DrawEffectOfChangeInCrossSection() {
  46
  47   //get the data
  48   TFile* fin = TFile::Open("systematics_xsections.root");
  49
  50   const Char_t* changes[]  = {"pythia","ddmore","ddless","sdmore","sdless", "dmore", "dless"};
  51   Int_t colors[] = {1,2,103,102,4,6,1};
  52
  53   TH1F* hRatios[7];
  54   for(Int_t i=0; i<7; i++) {
  55     hRatios[i] = (TH1F*)fin->Get(Form("ratio_vertexReco_triggerBias_%s",changes[i]));
  56     hRatios[i]->SetLineWidth(2);
  57     hRatios[i]->SetLineColor(colors[i]);
  58     hRatios[i]->SetMarkerStyle(22);
  59     hRatios[i]->SetMarkerSize(0.8);
  60   }
  61
  62   TPad* p = DrawCanvasAndPad("syst_changeInXsection",700,400);
  63   p->SetRightMargin(0.2);
  64   p->SetLeftMargin(0.13);
  65
  66   TH2F* null = new TH2F("","",100,-1.05,1.05,100,0.93,1.06);
  67   null->GetXaxis()->SetTitle("#eta");
  68   null->GetYaxis()->SetTitle("Ratio pythia/modified cross-sections");
  69   null->Draw();
  70
  71   TLatex* text[7];
  72
  73   for(Int_t i=1; i<7; i++) {
  74     hRatios[i]->Draw("same");
  75
  76     TString str(hRatios[i]->GetTitle());
  77     str.Remove(0,str.First("DD"));
  78     str.Remove(str.First(")"),1);
  79     text[i] = new TLatex(1.08,hRatios[i]->GetBinContent(hRatios[i]->FindBin(0.))-0.002,str.Data());
  80     text[i]->SetTextColor(colors[i]);
  81     text[i]->Draw();
  82   }
  83 }
  84
  85
  86 void DrawEffectOfChangeInComposition() {
  87
  88   //get the data
  89   TFile* fin = TFile::Open("systematics_composition.root");
  90
  91   Int_t colors[] = {1,2,103,102,4,6,1};
  92
  93   TH1F* hRatios[6];
  94
  95   Float_t etaRange = 0.899;
  96
  97   for (Int_t i=0; i<6; i++) {
  98     hRatios[i] = (TH1F*)fin->Get(Form("ratio_%d",i));
  99
 100     hRatios[i]->SetLineWidth(2);
 101     hRatios[i]->SetLineColor(colors[i]);
 102     hRatios[i]->SetMarkerStyle(22);
 103     hRatios[i]->SetMarkerSize(0.8);
 104
 105     hRatios[i]->GetXaxis()->SetRangeUser(-etaRange, etaRange);
 106   }
 107
 108   TPad* p = DrawCanvasAndPad("syst_changeOfComposition",700,400);
 109   p->SetRightMargin(0.2);
 110   p->SetLeftMargin(0.13);
 111
 112   TH2F* null = new TH2F("","",100,-1.05,1.05,100,0.97,1.03);
 113   null->GetXaxis()->SetTitle("#eta");
 114   null->GetYaxis()->SetTitle("Ratio pythia/modified composition");
 115   null->Draw();
 116
 117   TLatex* text[6];
 118
 119   for(Int_t i=0; i<6; i++) {
 120     hRatios[i]->Draw("same");
 121
 122     TString str(hRatios[i]->GetTitle());
 123     str.Remove(0,16);
 124     text[i] = new TLatex(1.08,hRatios[i]->GetBinContent(hRatios[i]->FindBin(0))-0.002,str.Data());
 125     text[i]->SetTextColor(colors[i]);
 126     text[i]->SetTextSize(0.053);
 127
 128     text[i]->Draw();
 129   }
 130
 131
 132 }
 133
 134 TPad* DrawCanvasAndPad(const Char_t* name, Int_t sizeX=600, Int_t sizeY=500) {
 135
 136   gStyle->SetOptStat(0);
 137   gStyle->SetOptTitle(0);
 138   gStyle->SetOptFit(0);
 139
 140   gStyle->SetTextSize(0.04);
 141   gStyle->SetTitleSize(0.05,"xyz");
 142   //gStyle->SetTitleFont(133, "xyz");
 143   //gStyle->SetLabelFont(133, "xyz");
 144   //gStyle->SetLabelSize(17, "xyz");
 145   gStyle->SetLabelOffset(0.01, "xyz");
 146
 147   gStyle->SetTitleOffset(1.1, "y");
 148   gStyle->SetTitleOffset(1.1, "x");
 149   gStyle->SetEndErrorSize(0.0);
 150
 151   //##############################################
 152
 153   //making canvas and pads
 154   TCanvas *c = new TCanvas(name,name,sizeX,sizeY);
 155
 156   TPad* p1 = new TPad("pad1","", 0, 0.0, 1.0, 1.0, 0, 0, 0);
 157
 158   p1->SetBottomMargin(0.15);
 159   p1->SetTopMargin(0.03);
 160   p1->SetLeftMargin(0.15);
 161   p1->SetRightMargin(0.03);
 162
 163   p1->SetGridx();
 164   p1->SetGridy();
 165
 166   p1->Draw();
 167   p1->cd();
 168
 169   return p1;
 170 }
 171
 172 void MisalignmentShowRawTrackPlots()
 173 {
 174   gSystem->Load("libPWG0base");
 175   TFile* file = TFile::Open("resC-resC/analysis_esd_raw.root");
 176   dNdEtaAnalysis* fdNdEtaAnalysis = new dNdEtaAnalysis("dndeta", "dndeta");
 177   fdNdEtaAnalysis->LoadHistograms("dndeta");
 178
 179   TFile* file2 = TFile::Open("resC-fullA/analysis_esd_raw.root");
 180   dNdEtaAnalysis* fdNdEtaAnalysis2 = new dNdEtaAnalysis("dndeta", "dndeta");
 181   fdNdEtaAnalysis2->LoadHistograms("dndeta");
 182
 183   TH3* track1 = fdNdEtaAnalysis->GetData()->GetTrackCorrection()->GetMeasuredHistogram()->Clone("track1");
 184   TH3* track2 = fdNdEtaAnalysis2->GetData()->GetTrackCorrection()->GetMeasuredHistogram()->Clone("track2");
 185
 186   // normalize to number of events;
 187   TH2* event1 = fdNdEtaAnalysis->GetData()->GetEventCorrection()->GetMeasuredHistogram();
 188   TH2* event2 = fdNdEtaAnalysis2->GetData()->GetEventCorrection()->GetMeasuredHistogram();
 189   Int_t event1Count = event1->Integral();
 190   Int_t event2Count = event2->Integral();
 191   track1->Scale(1.0 / event1Count);
 192   track2->Scale(1.0 / event2Count);
 193
 194   const Float_t innerLimit = 0.49;
 195   const Float_t outerLimit = 0.99;
 196
 197   track1->GetYaxis()->SetRangeUser(-outerLimit, outerLimit);
 198   track2->GetYaxis()->SetRangeUser(-outerLimit, outerLimit);
 199   AliPWG0Helper::CreateDividedProjections(track1, track2, "ze1");
 200   TH1* fullRange = gROOT->FindObject("track1_ze1_div_track2_ze1");
 201
 202   track1->GetYaxis()->SetRangeUser(-innerLimit, innerLimit);
 203   track2->GetYaxis()->SetRangeUser(-innerLimit, innerLimit);
 204   AliPWG0Helper::CreateDividedProjections(track1, track2, "ze2");
 205   TH1* central = gROOT->FindObject("track1_ze2_div_track2_ze2");
 206   central->SetLineColor(1);
 207   central->SetMarkerStyle(21);
 208
 209   for (Int_t x=1; x<track1->GetXaxis()->GetNbins(); ++x)
 210     for (Int_t y=track1->GetYaxis()->FindBin(-innerLimit); y<track1->GetYaxis()->FindBin(innerLimit); ++y)
 211       for (Int_t z=1; z<track1->GetZaxis()->GetNbins(); ++z)
 212       {
 213         track1->SetBinContent(x, y, z, 0);
 214         track1->SetBinError(x, y, z, 0);
 215         track2->SetBinContent(x, y, z, 0);
 216         track2->SetBinError(x, y, z, 0);
 217       }
 218
 219   track1->GetYaxis()->SetRangeUser(-outerLimit, outerLimit);
 220   track2->GetYaxis()->SetRangeUser(-outerLimit, outerLimit);
 221   AliPWG0Helper::CreateDividedProjections(track1, track2, "ze3");
 222   TH1* peripheral = gROOT->FindObject("track1_ze3_div_track2_ze3");
 223   peripheral->SetLineColor(2);
 224   peripheral->SetMarkerStyle(22);
 225   peripheral->SetMarkerColor(2);
 226
 227   TH2* tmp = new TH2F("tmp", ";p_{T} [GeV/c]      ;#frac{tracks residual misalignment}{tracks full misalignment}", 1, 0.1, 10, 1, 0.9, 1.3);
 228
 229   tmp->SetStats(kFALSE);
 230   //tmp->GetXaxis()->SetNoExponent();
 231
 232   Float_t ptStart = 0.1;
 233
 234   fullRange->GetXaxis()->SetRangeUser(ptStart, 9.9);
 235   central->GetXaxis()->SetRangeUser(ptStart, 9.9);
 236   peripheral->GetXaxis()->SetRangeUser(ptStart, 9.9);
 237
 238   TCanvas* canvas = new TCanvas("MisalignmentShowRawTrackPlots", "MisalignmentShowRawTrackPlots", 700, 400);
 239   gPad->SetLogx();
 240   gPad->SetGridx();
 241   gPad->SetGridy();
 242
 243   TLegend* legend = new TLegend(0.2, 0.7, 0.4, 0.8);
 244
 245   legend->AddEntry(central, "|#eta| < 0.5");
 246   legend->AddEntry(peripheral, "0.5 < |#eta| < 1.0");
 247
 248   legend->SetFillColor(0);
 249
 250   tmp->Draw();
 251   //fullRange->Draw("SAME");
 252   central->Draw("SAME");
 253   peripheral->Draw("SAME");
 254
 255   legend->Draw();
 256
 257   canvas->SaveAs("syst_mis_ntracks.eps");
 258 }
 259
 260
 261 void drawdNdEtaRatios(const char* canvasName, Int_t n, const char** files, const char** dirs, const char** names, Int_t* histID)
 262 {
 263   gSystem->Load("libPWG0base");
 264
 265   TCanvas* canvas = new TCanvas(canvasName, canvasName, 1000, 500);
 266   canvas->Divide(2, 1);
 267
 268   TLegend* legend = new TLegend(0.63, 0.73, 0.98, 0.98);
 269   legend->SetFillColor(0);
 270
 271   TH1* base = 0;
 272
 273   for (Int_t i = 0; i < n; ++i)
 274   {
 275     TFile::Open(files[i]);
 276
 277     dNdEtaAnalysis* tmp = new dNdEtaAnalysis(dirs[i], dirs[i]);
 278     tmp->LoadHistograms();
 279
 280     TH1* hist = tmp->GetdNdEtaPtCutOffCorrectedHistogram(histID[i]);
 281
 282     if (i == 0)
 283       base = hist;
 284
 285     legend->AddEntry(hist, names[i]);
 286
 287     hist->SetMarkerColor(colors[i]);
 288     hist->SetMarkerStyle(markers[i]);
 289
 290     canvas->cd(1);
 291     hist->DrawCopy((i == 0) ? "" : "SAME");
 292
 293     if (i != 0)
 294     {
 295       canvas->cd(2);
 296       hist->Divide(hist, base, 1, 1, "B");
 297       hist->GetYaxis()->SetRangeUser(0.98, 1.02);
 298       hist->DrawCopy((i == 1) ? "" : "SAME");
 299     }
 300   }
 301
 302   legend->Draw();
 303
 304   canvas->SaveAs(Form("%s.eps", canvas->GetName()));
 305 }
 306
 307 void drawdNdEtaRatios(const char* canvasName, Int_t n, const char** files1, const char** files2, const char** dirs1, const char** dirs2, const char** names, Int_t* histID)
 308 {
 309   gSystem->Load("libPWG0base");
 310
 311   TCanvas* canvas = new TCanvas(canvasName, canvasName, 700, 400);
 312   canvas->SetLeftMargin(0.12);
 313
 314   TLegend* legend = new TLegend(0.35, 0.7, 0.65, 0.85);
 315   legend->SetFillColor(0);
 316
 317   for (Int_t i = 0; i < n; ++i)
 318   {
 319     TFile::Open(files1[i]);
 320
 321     dNdEtaAnalysis* tmp1 = new dNdEtaAnalysis(dirs1[i], dirs1[i]);
 322     tmp1->LoadHistograms();
 323
 324     TFile::Open(files2[i]);
 325
 326     dNdEtaAnalysis* tmp2 = new dNdEtaAnalysis(dirs2[i], dirs2[i]);
 327     tmp2->LoadHistograms();
 328
 329     TH1* hist1 = tmp1->GetdNdEtaPtCutOffCorrectedHistogram(histID[i]);
 330     TH1* hist2 = tmp2->GetdNdEtaPtCutOffCorrectedHistogram(histID[i]);
 331
 332     TH1* division = hist1->Clone();
 333
 334     division->Divide(hist1, hist2, 1, 1, "B");
 335
 336     division->SetMarkerColor(colors[i]);
 337     division->SetMarkerStyle(markers[i]);
 338
 339     legend->AddEntry(division, names[i]);
 340
 341     division->SetTitle("");
 342     division->GetYaxis()->SetTitle("#frac{dN_{ch}/d#eta using MC vtx}{dN_{ch}/d#eta using ESD vtx}");
 343     division->SetStats(kFALSE);
 344     division->GetYaxis()->SetTitleOffset(1.3);
 345     division->GetXaxis()->SetRangeUser(-0.99, 0.99);
 346     division->GetYaxis()->SetRangeUser(0.981, 1.02);
 347     division->DrawCopy((i == 0) ? "" : "SAME");
 348   }
 349
 350   gPad->SetGridx();
 351   gPad->SetGridy();
 352
 353   legend->Draw();
 354
 355   canvas->SaveAs(Form("%s.eps", canvas->GetName()));
 356 }
 357
 358 void vertexShiftStudy(Int_t histID)
 359 {
 360   const char* files[] = { "maps/idealA/mc-vertex/analysis_esd.root", "results/idealC-idealA/analysis_esd.root", "maps/idealA/mc-vertex-shift-0.05/analysis_esd.root", "maps/idealA/mc-vertex-shift-0.1/analysis_esd.root", "maps/idealA/mc-vertex-shift-dep/analysis_esd.root" };
 361   const char* dirs[] = { "dndeta", "dndeta", "dndeta", "dndeta", "dndeta" };
 362   const char* names[] = { "mc vtx", "esd vtx", "+ 0.05 cm", "+ 0.1 cm", "old vtx shift" };
 363   Int_t hist[] = { histID, histID, histID, histID, histID };
 364
 365   drawdNdEtaRatios("syst_vertex_shift1", 5, files, dirs, names, hist);
 366
 367   const char* files1[] = { "maps/idealA/mc-vertex/analysis_esd.root", "maps/idealA/mc-vertex/analysis_esd.root", "maps/idealA/mc-vertex/analysis_esd.root" };
 368   const char* files2[] = { "results/idealC-idealA/analysis_esd.root", "results/idealC-idealA/analysis_esd.root", "results/idealC-idealA/analysis_esd.root" };
 369   const char* dirs1[] = { "dndeta", "dndeta", "dndeta"};
 370   const char* names[] = { "|vtx-z| < 10 cm", "-10 cm < vtx-z < 0 cm", "0 cm < vtx-z < 10 cm" };
 371   Int_t hist[] = { 0, 1, 2 };
 372
 373   drawdNdEtaRatios("syst_vertex_shift2", 3, files1, files2, dirs, dirs, names, hist);
 374 }
 375
 376 void vertexShift()
 377 {
 378   TFile::Open("vertex.root");
 379
 380   TH2* hist = gFile->Get("fVertexCorr");
 381   TProfile* prof = hist->ProfileX();
 382
 383   prof->SetStats(kFALSE);
 384   prof->SetTitle(";MC vtx-z [cm];mean (ESD vtx-z - MC vtx-z) [cm]");
 385   prof->GetYaxis()->SetTitleOffset(1.2);
 386
 387   prof->SetLineWidth(2);
 388
 389   TCanvas* canvas = new TCanvas("syst_vertex_shift", "syst_vertex_shift", 700, 400);
 390
 391   gPad->SetGridx();
 392   gPad->SetGridy();
 393
 394   prof->Draw();
 395
 396   canvas->SaveAs(Form("%s.eps", canvas->GetName()));
 397 }
 398