PWGLF/totEt/macros/emEt/PlotEmEtDistribution.C

   1 void SetStyles(TH1 *histo,int marker, int color){
   2   histo->Sumw2();
   3   histo->SetMarkerStyle(marker);
   4   histo->SetMarkerColor(color);
   5   histo->SetLineColor(color);
   6   //histo->GetXaxis()->SetTitle(xtitle);
   7   //histo->GetYaxis()->SetTitle(ytitle);
   8 }
   9 void PlotEmEtDistribution(Bool_t isPhos = kFALSE){
  10   gStyle->SetOptTitle(0);
  11   gStyle->SetOptStat(0);
  12   gStyle->SetOptFit(0);
  13   TString filename, detname;
  14   if(isPhos){
  15     detname = "PHOS";
  16     filename = "rootFiles/LHC11a10a_bis/Et.ESD.simPbPb.PHOS.LHC11a10a_bis.Run139465.root";
  17   }
  18   else{
  19     filename = "rootFiles/LHC11a10a_bis/Et.ESD.simPbPb.EMCal.LHC11a10a_bis.Run139465.root";
  20     detname = "EMCal";
  21   }
  22
  23   TFile *f = TFile::Open(filename, "READ");
  24   TList *l = dynamic_cast<TList*>(f->Get("out1"));
  25   TH1F *fHistAllEnergy = l->FindObject("fHistAllEnergy");
  26   TH1F *fHistSignalEnergy = l->FindObject("fHistSignalEnergy");
  27   TH1F *fHistNeutronEnergy = l->FindObject("fHistNeutronEnergy");
  28   TH1F *fHistKaonEnergy = l->FindObject("fHistKaonEnergy");
  29   TH1F *fHistHadronEnergy = l->FindObject("fHistHadronEnergy");
  30   TH1F *fHistSecondaryEnergy = l->FindObject("fHistSecondaryEnergy");
  31   TH1F *fHistSignalEnergyFraction = fHistSignalEnergy->Clone("fHistSignalEnergyFraction");
  32   TH1F *fHistNeutronEnergyFraction = fHistNeutronEnergy->Clone("fHistNeutronEnergyFraction");
  33   TH1F *fHistKaonEnergyFraction = fHistKaonEnergy->Clone("fHistKaonEnergyFraction");
  34   TH1F *fHistHadronEnergyFraction = fHistHadronEnergy->Clone("fHistHadronEnergyFraction");
  35   TH1F *fHistSecondaryEnergyFraction = fHistSecondaryEnergy->Clone("fHistSecondaryEnergyFraction");
  36   SetStyles(fHistSignalEnergyFraction,20,TColor::kRed);
  37   SetStyles(fHistNeutronEnergyFraction,21,TColor::kYellow);
  38   SetStyles(fHistSecondaryEnergyFraction,22,TColor::kGreen);
  39   SetStyles(fHistHadronEnergyFraction,23,1);
  40   SetStyles(fHistKaonEnergyFraction,24,TColor::kBlue);
  41  fHistSignalEnergyFraction->Divide(fHistAllEnergy);
  42  fHistNeutronEnergyFraction->Divide(fHistAllEnergy);
  43  fHistSecondaryEnergyFraction->Divide(fHistAllEnergy);
  44  fHistHadronEnergyFraction->Divide(fHistAllEnergy);
  45  fHistKaonEnergyFraction->Divide(fHistAllEnergy);
  46  TLegend *leg = new TLegend(0.67953,0.36631,0.800336,0.590909);
  47  leg->SetFillStyle(0);
  48  leg->SetFillColor(0);
  49  leg->SetBorderSize(0);
  50  leg->SetTextSize(0.03);
  51  leg->SetTextSize(0.038682);
  52  leg->AddEntry(fHistSignalEnergyFraction,"Signal");
  53  leg->AddEntry(fHistHadronEnergyFraction,"Hadron");
  54  leg->AddEntry(fHistSecondaryEnergyFraction,"Secondary");
  55  leg->AddEntry(fHistKaonEnergyFraction,"Kaon");
  56  leg->AddEntry(fHistNeutronEnergyFraction,"Neutron");
  57
  58     TCanvas *c1 = new TCanvas("c1","Simulation",600,400);
  59     c1->SetTopMargin(0.02);
  60     c1->SetRightMargin(0.03);
  61     c1->SetLeftMargin(0.11745);
  62     c1->SetBottomMargin(0.11745);
  63     c1->SetBorderSize(0);
  64     c1->SetFillColor(0);
  65     c1->SetFillColor(0);
  66     c1->SetBorderMode(0);
  67     c1->SetFrameFillColor(0);
  68     c1->SetFrameBorderMode(0);
  69     fHistSignalEnergyFraction->SetMaximum(0.6);
  70     fHistSignalEnergyFraction->SetMinimum(0.0);
  71     fHistSignalEnergyFraction->GetXaxis()->SetTitle("Centrality bin");
  72     fHistSignalEnergyFraction->GetYaxis()->SetTitle("fraction");
  73     fHistSignalEnergyFraction->GetYaxis()->SetLabelSize(0.06);
  74     fHistSignalEnergyFraction->GetXaxis()->SetLabelSize(0.06);
  75     fHistSignalEnergyFraction->GetYaxis()->SetTitleSize(0.06);
  76     fHistSignalEnergyFraction->GetXaxis()->SetTitleSize(0.06);
  77     fHistSignalEnergyFraction->Draw();
  78     fHistSignalEnergyFraction->Draw("same");
  79     fHistNeutronEnergyFraction->Draw("same");
  80     fHistSecondaryEnergyFraction->Draw("same");
  81     fHistHadronEnergyFraction->Draw("same");
  82     fHistKaonEnergyFraction->Draw("same");
  83
  84     leg->Draw();
  85
  86     TString outfile = "/tmp/EmEtDistribution"+detname+".png";
  87     c1->SaveAs(outfile.Data());
  88 }