FitterHelper.hpp

 
#ifndef GlauberFitterHelper_H
#define GlauberFitterHelper_H 1
 
#include "TCanvas.h"
#include "TFile.h"
#include "TH1.h"
#include "TLegend.h"
#include "TPad.h"
 
#include "Fitter.hpp"
 
namespace Glauber {
inline void DrawHistos(const Fitter& fit, Bool_t isSim, Bool_t isData, Bool_t isGlauber, Bool_t isNBD) {
  std::unique_ptr<TCanvas> c1{new TCanvas("c1", "canvas", 1500, 900)};
 
  c1->Divide(2, 2);
 
  std::unique_ptr<TPad> c1_1{(TPad*) c1->GetListOfPrimitives()->FindObject("c1_1")};
  std::unique_ptr<TPad> c1_2{(TPad*) c1->GetListOfPrimitives()->FindObject("c1_2")};
  std::unique_ptr<TPad> c1_4{(TPad*) c1->GetListOfPrimitives()->FindObject("c1_4")};
 
  c1_1->SetLogy(1);
  c1_2->SetLogy(1);
  c1_4->SetLogy(1);
 
  /*const*/ TH1F hGlaub = fit.GetGlauberFitHisto();
  /*const*/ TH1F hData = fit.GetDataHisto();
  /*const*/ TH1F hNBD = fit.GetNBDHisto();
  /*const*/ TH1F hNcoll = fit.GetNcollHisto();
  /*const*/ TH1F hNpart = fit.GetNpartHisto();
  /*const*/ TH1F hBestFit = fit.GetBestFiHisto();
 
  std::unique_ptr<TFile> fOut{TFile::Open("glauber_qa.root", "recreate")};
 
  if (isSim) {
    c1->cd(1);
    hNcoll.SetLineColor(2);
 
    hNcoll.Draw();
    hNpart.Draw("same");
 
    std::unique_ptr<TLegend> legSim{new TLegend(0.6, 0.75, 0.75, 0.83)};
    legSim->AddEntry(&hNpart, "Npart", "l");
    legSim->AddEntry(&hNcoll, "hNcoll", "l");
    legSim->Draw("same");
 
    hNcoll.Write();
    hNpart.Write();
  }
 
  if (isData) {
    c1->cd(2);
    hData.Draw();
    hData.Write();
    if (isGlauber) {
      hBestFit.SetLineColor(kRed);
      hBestFit.Draw("same");
 
      std::unique_ptr<TLegend> legData{new TLegend(0.6, 0.75, 0.75, 0.83)};
      legData->AddEntry(&hBestFit, "Fit", "l");
      legData->AddEntry(&hData, "Data", "l");
      legData->Draw("same");
      hBestFit.Write();
    }
  }
 
  if (isNBD) {
    c1->cd(3);
    hNBD.Draw();
    hNBD.Write();
  }
 
  if (isGlauber) {
    c1->cd(4);
    hBestFit.Draw();
  }
 
  c1->Write();
  c1->SaveAs("glauber.pdf");
  fOut->Close();
}
 
}// namespace Glauber
 
#endif