#ifndef APPLY_PLOT_STYLE

#define APPLY_PLOT_STYLE

#include "TLegend.h"

#include "TF1.h"

#include <sstream>

#include <iostream>

#include "TFile.h"

#include "TTree.h"

#include <fstream>

#include <TROOT.h>

#include <TStyle.h>

#include <TH1F.h>

#include <TH2F.h>

#include <TLine.h>

#include <THStack.h>

#include <TMath.h>

#include <TRandom3.h>

#include <TPaletteAxis.h>

class ApplyPlotStyle{

public:

    ApplyPlotStyle();

    void Hist1D(TH1F* data_hist, TH1F* mc_hist, string saveAs="output.pdf", double cut_lower=TMath::Infinity(), double cut_upper=TMath::Infinity(), bool normalise=true, string legend1="Data", string legend2="Recon MC");

    void Hist1D_single(TH1F* hist, string saveAs="output.pdf", double cut_lower=TMath::Infinity(), double cut_upper=TMath::Infinity(), string legend="Data", bool isData=true);

    void Hist2D(TH2F* data_hist, TH2F* mc_hist, string saveAs="output.pdf", double cut_a_lower=TMath::Infinity(), double cut_a_upper=TMath::Infinity(), double cut_b_lower=TMath::Infinity(), double cut_b_upper=TMath::Infinity(), int legend_location=0,bool normalise=true, string legend1="Data", string legend2="Recon MC");

    void Hist2D_single(TH2F* data_hist, string saveAs="output.pdf", double cut_a_lower=TMath::Infinity(), double cut_a_upper=TMath::Infinity(), double cut_b_lower=TMath::Infinity(), double cut_b_upper=TMath::Infinity(), int legend_location=0,bool normalise=true, string legend1="Data");

    void Hist_TOFvTKU(TH2F* data_hist, TH2F* mc_hist, string saveAs="output.pdf", double mean_dP=20.0, double min_dP=-2.0, double max_dP=42.0, int legend_location=0,bool normalise=true, string legend1="Data", string legend2="Recon MC");

    void test();

private:

    static Double_t tofFunc(Double_t *x, Double_t *par);

};

#endif