#include <iostream>
#include <sstream>
#include <string>
#include <fstream>
#include <map>
#include <set>

#include "PlotDrawer/DCanvasInstructions.h"
#include "PlotDrawer/DPadInstructions.h"
#include "PlotDrawer/DPlotDrawer.h"

#include "PlotFitter/DPlotFitter.h"
#include "PlotFitter/DFitControl.h"
#include "PlotFitter/DFitUtilities.h"
#include "PlotFitter/DFitFunctors_Miscellaneous.h"
#include "PlotFitter/DFitFunctors_Gaussians.h"
#include "PlotFitter/DFitFunctors_BreitWigners.h"

#include "PlotData/DStatUtilities.h"

#include "../particleType.h"

/*
.x ../Load.C
.L ~/Code/Work/gluex/ROOT_Scripts/strange_v2/Plot_Strangeness.C+
Plot_kpi0sigma0("~/Scratch/gluex/strange_v2/kpi0sigma0.root");
*/
void Plot_kpi0sigma0(string locInputFileName)
{
	TFile* locInputFile = new TFile(locInputFileName.c_str(), "READ");

	DCanvasInstructions* locCanvasInstructions = new DCanvasInstructions("Heya");
	locCanvasInstructions->Set_TargetCanvasAspectRatio(16.0/9.0);
	locCanvasInstructions->dTargetPadAspectRatio = 16.0/9.0;
	locCanvasInstructions->dMinPadWidth = 400;
	locCanvasInstructions->dStyle->SetOptFit(102);

	TH2* locDalitzHist = (TH2*)locInputFile->Get("Hist_Dalitz_KinFit_PostKinFitCut/Dalitz");
	locDalitzHist->Rebin2D(5, 5);
	locDalitzHist->GetXaxis()->SetRangeUser(0.0, 6.0);
	locDalitzHist->GetYaxis()->SetRangeUser(0.0, 12.0);

	DPlotDrawer locPlotDrawer;
	locPlotDrawer.Draw_Object(locDalitzHist, locCanvasInstructions);
}

/*
.x ../Load.C
.L ~/Code/Work/gluex/ROOT_Scripts/strange_v2/Plot_Strangeness.C+
Plot_kpisigmaplus("~/Scratch/gluex/strange_v2/kpisigmaplus.root");
*/
void Plot_kpisigmaplus(string locInputFileName)
{
	TFile* locInputFile = new TFile(locInputFileName.c_str(), "READ");

	DCanvasInstructions* locCanvasInstructions = new DCanvasInstructions("Heya");
	locCanvasInstructions->Set_TargetCanvasAspectRatio(16.0/9.0);
	locCanvasInstructions->dTargetPadAspectRatio = 16.0/9.0;
	locCanvasInstructions->dMinPadWidth = 400;
	locCanvasInstructions->dStyle->SetOptFit(102);

	TH2* locDalitzHist = (TH2*)locInputFile->Get("Hist_Dalitz_KinFit_PostKinFitCut/Dalitz");
	locDalitzHist->Rebin2D(2, 5);
	locDalitzHist->GetXaxis()->SetRangeUser(0.0, 9.0);
	locDalitzHist->GetYaxis()->SetRangeUser(0.0, 16.0);

	DPlotDrawer locPlotDrawer;
	locPlotDrawer.Draw_Object(locDalitzHist, locCanvasInstructions);

	TH1* locMassHist = (TH1*)locInputFile->Get("Hist_InvariantMass_YStarPlus_KinFit_PostKinFitCut/InvariantMass");
	locPlotDrawer.Draw_Object(locMassHist, locCanvasInstructions);
}

/*
.x ../Load.C
.L ~/Code/Work/gluex/ROOT_Scripts/strange_v2/Plot_Strangeness.C+
Plot_k2pilambda("~/Scratch/gluex/strange_v2/k2pilambda.root");
*/
void Plot_k2pilambda(string locInputFileName)
{
	TFile* locInputFile = new TFile(locInputFileName.c_str(), "READ");

	DCanvasInstructions* locCanvasInstructions = new DCanvasInstructions("Heya");
	locCanvasInstructions->Set_TargetCanvasAspectRatio(16.0/9.0);
	locCanvasInstructions->dTargetPadAspectRatio = 16.0/9.0;
	locCanvasInstructions->dMinPadWidth = 400;
	locCanvasInstructions->dStyle->SetOptFit(102);
	locCanvasInstructions->dStyle->SetOptStat(10);

	TH2* locDalitzHist = (TH2*)locInputFile->Get("Hist_2DInvariantMass_SigmaStarPlus_KinFit_PostKinFitCut/2DInvariantMass");
	locDalitzHist->Rebin2D(5, 5);
	locDalitzHist->GetXaxis()->SetRangeUser(0.5, 2.5);
	locDalitzHist->GetYaxis()->SetRangeUser(1.2, 3.5);

	DPlotDrawer locPlotDrawer;
	locPlotDrawer.Draw_Object(locDalitzHist, locCanvasInstructions);
}