/*
 * plot_all.C
 * @author Jennifer Puetz (j.puetz@fz-juelich.de)
 * @date 2016
 * @brief script to display all histograms
 * @details This script should display all histograms from the PAWIAN output file.
 */

#include "TString.h"

void plot_all(TString input=""){

	TFile * inFile = new TFile(input, "READ");

	TString names [3] ={"Xibarp", "Lambda", "Km"};
	TString params[2] ={"Theta", "Phi"};
	TString mass [3] = {"LambdaKm", "XibarpLambda", "XibarpKm"};

	TCanvas * c = new TCanvas("c", "all plots", 0,0, 1500,1000);
	c->Divide(3,3);

	int c_counter=1;

	gStyle->SetOptStat(0);

	for (int name=0; name<3; name++){
		for(int par=0; par<2; par++){
			c->cd(c_counter);

			TString data = TString::Format("Data%s%s_HeliXibarpLambdaKm", params[par].Data(), names[name].Data());
			TString fit = TString::Format("Fit%s%s_HeliXibarpLambdaKm", params[par].Data(), names[name].Data());
			TString title =  TString::Format("%s (%s_HeliXibarpLambdaKm)", params[par].Data(), names[name].Data());

			TH1F * h_data = (TH1F*) inFile->Get(data);
			TH1F * h_fit = (TH1F*) inFile->Get(fit);

			h_data->SetTitle(title);
			h_fit->SetLineColor(kRed);



			TLegend * l = new TLegend(0.745,0.727,0.927,0.886);

			l->AddEntry(h_data ,"Data", "lep");
			l->AddEntry(h_fit ,"Fit");

			h_fit->Draw("HIST");
			h_data->Draw("SAME");
			l->Draw("SAME");
			c_counter++;
		}
	}

	for (int m=0; m<3; m++){

		c->cd(c_counter);

		TString m_data = TString::Format("Data%s", mass[m].Data());
		TString m_fit = TString::Format("Fit%s", mass[m].Data());

		TH1F * h_mdata = (TH1F*) inFile->Get(m_data);
		TH1F * h_mfit = (TH1F*) inFile->Get(m_fit);

		h_mdata->SetTitle("M("+mass[m]+")");
		h_mfit->SetLineColor(kRed);

		TLegend * l2 = new TLegend(0.745,0.727,0.927,0.886);

		l2->AddEntry(h_mdata ,"Data", "lep");
		l2->AddEntry(h_mfit ,"Fit");

		h_mfit->Draw("HIST");
		h_mdata->Draw("SAME");
		l2->Draw("SAME");
		c_counter++;

	}

//	c->Print("all_plots.pdf");
//	c->Print("all_plots.png");
//	c->Print("all_plots.root");

}