/*
	EVBApp.cpp
	Class which represents the API of the event building environment. Wraps together the core concepts
	of the event builder, from conversion to plotting. Even intended to be able to archive data.
	Currently under development.

	Written by G.W. McCann Oct. 2020
*/
#include <cstdlib>
#include "EVBApp.h"
#include "RunCollector.h"
#include "CompassRun.h"
#include "SlowSort.h"
#include "FastSort.h"
#include "SFPAnalyzer.h"
#include "SFPPlotter.h"

namespace EventBuilder {
	
	EVBApp::EVBApp() :
		m_rmin(0), m_rmax(0), m_ZT(0), m_AT(0), m_ZP(0), m_AP(0), m_ZE(0), m_AE(0), m_ZR(0), m_AR(0),
		m_B(0), m_Theta(0), m_BKE(0), m_workspace("none"), m_mapfile("none"), m_shiftfile("none"),
		m_cutList("none"), m_SlowWindow(0), m_FastWindowIonCh(0), m_FastWindowSABRE(0), m_pb(nullptr)
	{
	}
	
	EVBApp::~EVBApp() 
	{
	}
	
	bool EVBApp::ReadConfigFile(const std::string& fullpath) 
	{
		std::cout<<"Reading in configuration from file: "<<fullpath<<std::endl;
		std::ifstream input(fullpath);
		if(!input.is_open()) 
		{
			std::cout<<"Read failed! Unable to open input file!"<<std::endl;
			return false;
		}
		std::string junk;
	
		std::getline(input, junk);
		input>>junk>>m_workspace;
		input>>junk;
		std::getline(input, junk);
		std::getline(input, junk);
		input>>junk>>m_mapfile;
		input>>junk>>m_scalerfile;
		input>>junk>>m_cutList;
		input>>junk>>m_ZT>>junk>>m_AT;
		input>>junk>>m_ZP>>junk>>m_AP;
		input>>junk>>m_ZE>>junk>>m_AE;
		input>>junk>>m_B;
		input>>junk>>m_BKE;
		input>>junk>>m_Theta;
		input>>junk;
		std::getline(input, junk);
		std::getline(input, junk);
		input>>junk>>m_shiftfile;
		input>>junk>>m_SlowWindow;
		input>>junk>>m_FastWindowIonCh;
		input>>junk>>m_FastWindowSABRE;
		input>>junk;
		std::getline(input, junk);
		std::getline(input, junk);
		input>>junk>>m_rmin;
		input>>junk>>m_rmax;
	
		input.close();
	
		std::cout<<"Completed."<<std::endl;
	
		return true;
	}
	
	void EVBApp::WriteConfigFile(const std::string& fullpath) 
	{
	
		std::cout<<"Writing out configuration to file: "<<fullpath<<std::endl;
		std::ofstream output(fullpath);
		if(!output.is_open()) 
		{
			std::cout<<"Write failed! Unable to open output file!"<<std::endl;
			return;
		}
	
		output<<"-------Data Location----------"<<std::endl;
		output<<"WorkspaceDirectory: "<<m_workspace<<std::endl;
		output<<"-------------------------------"<<std::endl;
		output<<"------Experimental Inputs------"<<std::endl;
		output<<"ChannelMapFile: "<<m_mapfile<<std::endl;
		output<<"ScalerFile: "<<m_scalerfile<<std::endl;
		output<<"CutListFile: "<<m_cutList<<std::endl;
		output<<"ZT: "<<m_ZT<<std::endl;
		output<<"AT: "<<m_AT<<std::endl;
		output<<"ZP: "<<m_ZP<<std::endl;
		output<<"AP: "<<m_AP<<std::endl;
		output<<"ZE: "<<m_ZE<<std::endl;
		output<<"AE: "<<m_AE<<std::endl;
		output<<"BField(G): "<<m_B<<std::endl;
		output<<"BeamKE(MeV): "<<m_BKE<<std::endl;
		output<<"Theta(deg): "<<m_Theta<<std::endl;
		output<<"-------------------------------"<<std::endl;
		output<<"-------Timing Information------"<<std::endl;
		output<<"BoardOffsetFile: "<<m_shiftfile<<std::endl;
		output<<"SlowCoincidenceWindow(ps): "<<m_SlowWindow<<std::endl;
		output<<"FastCoincidenceWindow_IonCh(ps): "<<m_FastWindowIonCh<<std::endl;
		output<<"FastCoincidenceWindow_SABRE(ps): "<<m_FastWindowSABRE<<std::endl;
		output<<"-------------------------------"<<std::endl;
		output<<"--------Run Information--------"<<std::endl;
		output<<"MinRun: "<<m_rmin<<std::endl;
		output<<"MaxRun: "<<m_rmax<<std::endl;
		output<<"-------------------------------"<<std::endl;
	
		output.close();
	
		std::cout<<"Completed."<<std::endl;
	
	}
	
	void EVBApp::PlotHistograms() 
	{
		std::string analyze_dir = m_workspace+"/analyzed/";
		std::string plot_file = m_workspace+"/histograms/run_"+std::to_string(m_rmin)+"_"+std::to_string(m_rmax)+".root";
		SFPPlotter grammer;
		grammer.ApplyCutlist(m_cutList);
		std::cout<<"-------------GWM Event Builder-------------"<<std::endl;
		std::cout<<"Generating a histogram file from analyzed files"<<std::endl;
		std::cout<<"Analyzed directory: "<<analyze_dir<<std::endl;
		std::cout<<"Min Run: "<<m_rmin<<" Max Run: "<<m_rmax<<std::endl;
		std::cout<<"Cut List File: "<<m_cutList<<std::endl;
		std::cout<<"Histogram File: "<<plot_file<<std::endl;
	
		if(m_pb) 
			grammer.AttachProgressBar(m_pb);
		grabber.SetSearchParams(analyze_dir, "", ".root", m_rmin, m_rmax);
		if(grabber.GrabFilesInRange()) 
		{
			std::cout<<"Working...";
			grammer.Run(grabber.GetFileList(), plot_file);
			std::cout<<" Complete."<<std::endl;
		} 
		else 
		{
			std::cout<<"Unable to find files at PlotHistograms"<<std::endl;
		}
		std::cout<<"-------------------------------------------"<<std::endl;
	
	}
	
	void EVBApp::Convert2RawRoot() 
	{
		int sys_return;
		std::string rawroot_dir = m_workspace+"/raw_root/";
		std::string unpack_dir = m_workspace+"/temp_binary/";
		std::string binary_dir = m_workspace+"/raw_binary/";
		std::cout<<"-------------GWM Event Builder-------------"<<std::endl;
		std::cout<<"Converting Binary file Archive to ROOT file"<<std::endl;
		std::cout<<"Binary Archive Directory: "<<binary_dir<<std::endl;
		std::cout<<"Temporary Unpack Directory: "<<unpack_dir<<std::endl;
		std::cout<<"Timestamp Shift File: "<<m_shiftfile<<std::endl;
		std::cout<<"Min Run: "<<m_rmin<<" Max Run: "<<m_rmax<<std::endl;
		
	
		grabber.SetSearchParams(binary_dir, "", ".tar.gz",0,1000);
	
		std::cout<<"Workspace Directory: "<<m_workspace<<std::endl;
	
		std::string rawfile, binfile;
		std::string unpack_command, wipe_command;
	
		CompassRun converter(unpack_dir);
		converter.SetShiftMap(m_shiftfile);
		converter.SetScalerInput(m_scalerfile);
		if(m_pb)
			converter.AttachProgressBar(m_pb);
	
		std::cout<<"Beginning conversion..."<<std::endl;
	
		for(int i=m_rmin; i<=m_rmax; i++) 
		{
			binfile = grabber.GrabFile(i);
			if(binfile == "") 
				continue;
			converter.SetRunNumber(i);
			std::cout<<"Converting file: "<<binfile<<std::endl;
	
			rawfile = rawroot_dir + "compass_run_"+ std::to_string(i) + ".root";
			unpack_command = "tar -xzf "+binfile+" --directory "+unpack_dir;
			wipe_command = "rm -r "+unpack_dir+"*.bin";
	
			sys_return = system(unpack_command.c_str());
			converter.Convert2RawRoot(rawfile);
			sys_return = system(wipe_command.c_str());
	
		}
		std::cout<<std::endl<<"Conversion complete."<<std::endl;
		std::cout<<"-------------------------------------------"<<std::endl;
	
	}
	
	void EVBApp::MergeROOTFiles() 
	{
		std::string merge_file = m_workspace+"/merged/run_"+std::to_string(m_rmin)+"_"+std::to_string(m_rmax)+".root";
		std::string file_dir = m_workspace+"/analyzed/";
		std::cout<<"-------------GWM Event Builder-------------"<<std::endl;
		std::cout<<"Merging ROOT files into single ROOT file"<<std::endl;
		std::cout<<"Workspace directory: "<<m_workspace<<std::endl;
		std::cout<<"Min Run: "<<m_rmin<<" Max Run: "<<m_rmax<<std::endl;
		std::cout<<"Output file: "<<merge_file<<std::endl;
		std::string prefix = "";
		std::string suffix = ".root";
		grabber.SetSearchParams(file_dir, prefix, suffix,m_rmin,m_rmax);
		std::cout<<"Beginning the merge...";
		if(!grabber.Merge_TChain(merge_file)) 
		{
			std::cout<<"Unable to find files at MergeROOTFiles"<<std::endl;
			return;
		}
		std::cout<<" Complete."<<std::endl;
		std::cout<<"-------------------------------------------"<<std::endl;
	}
	
	void EVBApp::Convert2SortedRoot() 
	{
		int sys_return;
		std::string sortroot_dir = m_workspace+"/sorted/";
		std::string unpack_dir = m_workspace+"/temp_binary/";
		std::string binary_dir = m_workspace+"/raw_binary/";
		std::cout<<"-------------GWM Event Builder-------------"<<std::endl;
		std::cout<<"Converting Binary file Archive to ROOT file"<<std::endl;
		std::cout<<"Binary Archive Directory: "<<binary_dir<<std::endl;
		std::cout<<"Temporary Unpack Directory: "<<unpack_dir<<std::endl;
		std::cout<<"Timestamp Shift File: "<<m_shiftfile<<std::endl;
		std::cout<<"Channel Map File: "<<m_mapfile<<std::endl;
		std::cout<<"Slow Coincidence Window(ps): "<<m_SlowWindow<<std::endl;
		std::cout<<"Min Run: "<<m_rmin<<" Max Run: "<<m_rmax<<std::endl;
	
		grabber.SetSearchParams(binary_dir,"",".tar.gz",m_rmin,m_rmax);
	
		std::cout<<"Workspace Directory: "<<m_workspace<<std::endl;
	
		std::string sortfile, binfile;
		std::string unpack_command, wipe_command;
	
		CompassRun converter(unpack_dir);
		converter.SetShiftMap(m_shiftfile);
		converter.SetScalerInput(m_scalerfile);
		if(m_pb) 
			converter.AttachProgressBar(m_pb);
	
		std::cout<<"Beginning conversion..."<<std::endl;
	
		for(int i=m_rmin; i<= m_rmax; i++) 
		{
			binfile = grabber.GrabFile(i);
			if(binfile == "") 
				continue;
			converter.SetRunNumber(i);
			std::cout<<"Converting file: "<<binfile<<std::endl;
	
			sortfile = sortroot_dir +"run_"+std::to_string(i)+ ".root";
			unpack_command = "tar -xzf "+binfile+" --directory "+unpack_dir;
			wipe_command = "rm -r "+unpack_dir+"*.bin";
	
			sys_return = system(unpack_command.c_str());
			converter.Convert2SortedRoot(sortfile, m_mapfile, m_SlowWindow);
			sys_return = system(wipe_command.c_str());
	
		}
		std::cout<<std::endl<<"Conversion complete."<<std::endl;
		std::cout<<"-------------------------------------------"<<std::endl;
	}
	
	void EVBApp::Convert2FastSortedRoot() {
		int sys_return;
		std::string sortroot_dir = m_workspace+"/fast/";
		std::string unpack_dir = m_workspace+"/temp_binary/";
		std::string binary_dir = m_workspace+"/raw_binary/";
		std::cout<<"-------------GWM Event Builder-------------"<<std::endl;
		std::cout<<"Converting Binary file Archive to ROOT file"<<std::endl;
		std::cout<<"Binary Archive Directory: "<<binary_dir<<std::endl;
		std::cout<<"Temporary Unpack Directory: "<<unpack_dir<<std::endl;
		std::cout<<"Timestamp Shift File: "<<m_shiftfile<<std::endl;
		std::cout<<"Channel Map File: "<<m_mapfile<<std::endl;
		std::cout<<"Slow Coincidence Window(ps): "<<m_SlowWindow<<std::endl;
		std::cout<<"Min Run: "<<m_rmin<<" Max Run: "<<m_rmax<<std::endl;
	
		grabber.SetSearchParams(binary_dir,"",".tar.gz",m_rmin,m_rmax);
	
		std::cout<<"Workspace Directory: "<<m_workspace<<std::endl;
	
		std::string sortfile, binfile;
		std::string unpack_command, wipe_command;
	
		CompassRun converter(unpack_dir);
		converter.SetShiftMap(m_shiftfile);
		converter.SetScalerInput(m_scalerfile);
		if(m_pb) 
			converter.AttachProgressBar(m_pb);
	
		std::cout<<"Beginning conversion..."<<std::endl;
	
		for(int i=m_rmin; i<=m_rmax; i++) 
		{
			binfile = grabber.GrabFile(i);
			if(binfile == "") 
				continue;
			converter.SetRunNumber(i);
			std::cout<<"Converting file: "<<binfile<<std::endl;
	
			sortfile = sortroot_dir + "run_" + std::to_string(i) + ".root";
			unpack_command = "tar -xzf "+binfile+" --directory "+unpack_dir;
			wipe_command = "rm -r "+unpack_dir+"*.bin";
	
			sys_return = system(unpack_command.c_str());
			converter.Convert2FastSortedRoot(sortfile, m_mapfile, m_SlowWindow, m_FastWindowSABRE, m_FastWindowIonCh);
			sys_return = system(wipe_command.c_str());
	
		}
		std::cout<<std::endl<<"Conversion complete."<<std::endl;
		std::cout<<"-------------------------------------------"<<std::endl;
	}
	
	void EVBApp::Convert2SlowAnalyzedRoot() {
		int sys_return;
		std::string sortroot_dir = m_workspace+"/analyzed/";
		std::string unpack_dir = m_workspace+"/temp_binary/";
		std::string binary_dir = m_workspace+"/raw_binary/";
		std::cout<<"-------------GWM Event Builder-------------"<<std::endl;
		std::cout<<"Converting Binary file Archive to ROOT file"<<std::endl;
		std::cout<<"Binary Archive Directory: "<<binary_dir<<std::endl;
		std::cout<<"Temporary Unpack Directory: "<<unpack_dir<<std::endl;
		std::cout<<"Timestamp Shift File: "<<m_shiftfile<<std::endl;
		std::cout<<"Channel Map File: "<<m_mapfile<<std::endl;
		std::cout<<"Slow Coincidence Window(ps): "<<m_SlowWindow<<std::endl;
		std::cout<<"Min Run: "<<m_rmin<<" Max Run: "<<m_rmax<<std::endl;
		
		grabber.SetSearchParams(binary_dir,"",".tar.gz",m_rmin, m_rmax);
		std::cout<<"Workspace Directory: "<<m_workspace<<std::endl;
	
		std::string sortfile, binfile;
		std::string unpack_command, wipe_command;
	
		CompassRun converter(unpack_dir);
		converter.SetShiftMap(m_shiftfile);
		converter.SetScalerInput(m_scalerfile);
		if(m_pb) 
			converter.AttachProgressBar(m_pb);
	
		std::cout<<"Beginning conversion..."<<std::endl;
	
		for(int i=m_rmin; i<=m_rmax; i++) 
		{
			binfile = grabber.GrabFile(i);
			if(binfile == "") 
				continue;
			converter.SetRunNumber(i);
			std::cout<<"Converting file: "<<binfile<<std::endl;
	
			sortfile = sortroot_dir + "run_" + std::to_string(i) + ".root";
			unpack_command = "tar -xzf "+binfile+" --directory "+unpack_dir;
			wipe_command = "rm -r "+unpack_dir+"*.bin";
	
			sys_return = system(unpack_command.c_str());
			converter.Convert2SlowAnalyzedRoot(sortfile, m_mapfile, m_SlowWindow, m_ZT, m_AT, m_ZP, m_AP, m_ZE, m_AE, m_BKE, m_B, m_Theta);
			sys_return = system(wipe_command.c_str());
	
		}
		std::cout<<std::endl<<"Conversion complete."<<std::endl;
		std::cout<<"-------------------------------------------"<<std::endl;
	}
	
	void EVBApp::Convert2FastAnalyzedRoot() 
	{
		int sys_return;
		std::string sortroot_dir = m_workspace+"/analyzed/";
		std::string unpack_dir = m_workspace+"/temp_binary/";
		std::string binary_dir = m_workspace+"/raw_binary/";
		std::cout<<"-------------GWM Event Builder-------------"<<std::endl;
		std::cout<<"Converting Binary file Archive to ROOT file"<<std::endl;
		std::cout<<"Binary Archive Directory: "<<binary_dir<<std::endl;
		std::cout<<"Temporary Unpack Directory: "<<unpack_dir<<std::endl;
		std::cout<<"Timestamp Shift File: "<<m_shiftfile<<std::endl;
		std::cout<<"Channel Map File: "<<m_mapfile<<std::endl;
		std::cout<<"Slow Coincidence Window(ps): "<<m_SlowWindow<<std::endl;
		std::cout<<"Fast Ion Chamber Coincidence Window(ps): "<<m_FastWindowIonCh<<std::endl;
		std::cout<<"Fast SABRE Coincidence Window(ps): "<<m_FastWindowSABRE<<std::endl;
		std::cout<<"Min Run: "<<m_rmin<<" Max Run: "<<m_rmax<<std::endl;
		
		grabber.SetSearchParams(binary_dir,"",".tar.gz",m_rmin,m_rmax);
	
		std::cout<<"Workspace Directory: "<<m_workspace<<std::endl;
	
		std::string sortfile, binfile;
		std::string unpack_command, wipe_command;
	
		CompassRun converter(unpack_dir);
		converter.SetShiftMap(m_shiftfile);
		converter.SetScalerInput(m_scalerfile);
		if(m_pb) 
			converter.AttachProgressBar(m_pb);
	
		std::cout<<"Beginning conversion..."<<std::endl;
	
		for(int i=m_rmin; i<=m_rmax; i++) 
		{
			binfile = grabber.GrabFile(i);
			if(binfile == "") 
				continue;
			converter.SetRunNumber(i);
			std::cout<<"Converting file: "<<binfile<<std::endl;
	
			sortfile = sortroot_dir + "run_" + std::to_string(i) + ".root";
			unpack_command = "tar -xzf "+binfile+" --directory "+unpack_dir;
			wipe_command = "rm -r "+unpack_dir+"*.bin";
	
			sys_return = system(unpack_command.c_str());
			converter.Convert2FastAnalyzedRoot(sortfile, m_mapfile, m_SlowWindow, m_FastWindowSABRE, m_FastWindowIonCh, m_ZT, m_AT, m_ZP, m_AP, m_ZE, m_AE, m_BKE, m_B, m_Theta);
			sys_return = system(wipe_command.c_str());
	
		}
		std::cout<<std::endl<<"Conversion complete."<<std::endl;
		std::cout<<"-------------------------------------------"<<std::endl;
	}
	
	bool EVBApp::SetKinematicParameters(int zt, int at, int zp, int ap, int ze, int ae, double b, double theta, double bke) 
	{
	
		if((at + ap - ae) < 0 || (zt + zp - ze) < 0) 
		{
			std::cout<<"Invalid Parameters at SetKinematicParameters"<<std::endl;
			return false;
		}
		m_ZT = zt; m_AT = at; m_ZP = zp; m_AP = ap; m_ZE = ze; m_AE = ae;
		m_B = b; m_Theta = theta; m_BKE = bke;
	
		m_ZR = (zt + zp - ze);
		m_AR = (at + ap - ae);
	
		return true;
	}

}