#include "MicroNTuplePlotterClass.h"

void VertexingReweighting(){
	string path = "/data/users/kkennedy/DisplacedDiphotonAnalysis/MicroNTuples/v6.0/output_";
	string data_file = "data_97percent";
	vector<string> file = {data_file};
	string TreeName = "PostSel";
	TCut basic_selections = " abs(ph1_t) < 12 && abs(ph2_t) < 12 && abs(dEta_ph) > 0.1 && Hcand_M > 60 && MET_met < 20" ;
	TCut cuts_reweight = " abs(ph1_t) < 12 && abs(ph2_t) < 12 && abs(dEta_ph) > 0.1 && Hcand_M > 60 && MET_met > 20 && MET_met < 30" ;
	TCut ph1_isTight = "ph1_isTight";
	TCut ph1_isMedium = "ph1_isMedium";
	TCut ph1_isLoose = "ph1_isLoose";
	TCut ph2_isTight = "ph2_isTight";
	TCut ph2_isMedium = "ph2_isMedium";
	TCut ph2_isLoose = "ph2_isLoose";

	PlotParams quadratureSum_Vz_Vr = {"sqrt(2DSliceRZ_Vr**2 + 2DSliceRZ_Vz**2)", "quadratureSumVzVr", "sqrt(V_{r}^{2} + V_{z}^{2}) [mm]", 0., 5000.};
	vector<PlotParams> ParamsVertexing = {
		P_2DSliceRZ_Vr, P_2DSliceRZ_Vz, quadratureSum_Vz_Vr, P_ph1_maxEcell_E, P_ph2_maxEcell_E
	};

	vector<TCut> cuts_compare = {
		ph1_isTight && ph2_isTight, 
		ph1_isMedium && !ph1_isTight && ph2_isMedium && ph2_isTight || ph1_isMedium && ph1_isTight && ph2_isMedium && !ph2_isTight || ph1_isMedium && !ph1_isTight && ph2_isMedium && !ph2_isTight || ph1_isMedium && ph1_isTight && ph2_isMedium && ph2_isTight,
		ph1_isTight && ph2_isTight + ph1_isMedium && !ph1_isTight && ph2_isMedium && ph2_isTight || ph1_isMedium && ph1_isTight && ph2_isMedium && !ph2_isTight || ph1_isMedium && !ph1_isTight && ph2_isMedium && !ph2_isTight || ph1_isMedium && ph1_isTight && ph2_isMedium && ph2_isTight		
	};	

	class MicroNTuplePlotter plotter(file, path);
	plotter.SetCuts(basic_selections);
	plotter.SetPlots(ParamsVertexing);
	plotter.SetComparisonCuts(cuts_compare);
	plotter.Plot("ratio");	
	plotter.RunReweighting2D(P_ph1_maxEcell_E, P_ph2_maxEcell_E, "data_97percent", TreeName, cuts_reweight);
	plotter.Plot("ratio");
}