#include "StandardLabels.C"

void pythia_W(void)
{
	TColor::CreateColorWheel();

	TFile *f = new TFile("hd_root.root");
	TH1D *Wplus_minus = (TH1D*)gROOT->FindObject("Wplus_minus");
	TH1D *Wpipi = (TH1D*)gROOT->FindObject("Wpipi");
	TH1D *Wplus_minus_coherent = (TH1D*)gROOT->FindObject("Wplus_minus_coherent");
	TH1D *Wpipi_coherent = (TH1D*)gROOT->FindObject("Wpipi_coherent");

	TCanvas *c1 = new TCanvas("c1");
	c1->SetTicks();
	c1->SetGridy();
	
	Wplus_minus_coherent->SetLineColor(kRed+2);
	Wplus_minus_coherent->SetFillColor(kRed+2);
	Wplus_minus_coherent->SetFillStyle(3004);

	Wpipi_coherent->SetLineColor(kBlue+2);
	Wpipi_coherent->SetFillColor(kBlue+2);
	Wpipi_coherent->SetFillStyle(3005);
	
	Wpipi->GetXaxis()->SetRangeUser(0.2, 2.5);
	Wpipi->SetTitle("PYTHIA generated background");
	
	Wpipi->Draw();
	Wplus_minus->Draw("same");
	Wpipi_coherent->Draw("same");
	Wplus_minus_coherent->Draw("same");
	
	TLegend *leg = new TLegend(0.40, 0.6, 0.88, 0.85);
	leg->SetFillColor(kWhite);
	leg->AddEntry(Wplus_minus, "perfect PID");
	leg->AddEntry(Wpipi, "No PID (pion assumption)");
	leg->AddEntry(Wplus_minus_coherent, "perfect PID 8.5#leqE_{#gamma}#leq9.5");
	leg->AddEntry(Wpipi_coherent, "No PID (pion assumption) 8.5#leqE_{#gamma}#leq9.5");
	leg->Draw();
	
	StandardLabels1D(Wpipi, "0.5^{o} #leq #theta_{#pm} #leq 10^{o}", "#gamma p #rightarrow p K^{+} K^{-}", "generated events only", "");
	
	c1->SaveAs("pythia_W.pdf");
	c1->SaveAs("pythia_W.png");
}