#include <stdlib.h>
#include <string.h>

#include <iostream>
#include <string>
using namespace std;


// These are defined in copytoplusplus.cc
extern string INFILE;
extern string OUTFILE;
extern bool POSTSMEAR;
extern string MCSMEAROPTS;
extern bool DELETEUNSMEARED;

// Defined in calibDB.cc
extern string HDDS_XML;

// Declare routines callable from FORTRAN
extern "C" int hdgeant_(void); // define in hdgeant_f.F
extern "C" const char* GetMD5Geom(void);

void Usage(void);

// Get access to FORTRAN common block with some control flags
#include "controlparams.h"

//------------------
// main
//------------------
int main(int narg, char *argv[])
{
	// Set some defaults. Note that most defaults related to the
	// simulation are set in uginit.F
	controlparams_.runtime_geom = 0;

	// Parse command line parameters
	bool print_xml_md5_checksum = false;
	for(int i=1; i<narg; i++){
		string arg = argv[i];
		string next = i<(narg+1) ? argv[i]:"";
		
		if(arg=="-h" || arg=="--help")Usage();
		if(arg=="-checksum" || arg=="--checksum")print_xml_md5_checksum = true;
		if(arg.find("-xml")==0){
			controlparams_.runtime_geom = 1;
			if(arg.find("=")!=string::npos){
				HDDS_XML = arg.substr(arg.find("=")+1);
			}
		}
	}
	
	// If user specified printing the checksum then 
	// do that and quit
	if(print_xml_md5_checksum){
	
		cout << "HDDS Geometry MD5 Checksum: " << GetMD5Geom() << endl;

		return 0;
	}

	// Run hdgeant proper
	int res = hdgeant_();

	// Optionally smear the resulting output file
	if(POSTSMEAR && res == 0){
		string cmd = "mcsmear "+MCSMEAROPTS+" "+OUTFILE;
		cout<<endl;
		cout<<"Smearing data with:"<<endl;
		cout<<endl;
		cout<<cmd<<endl;
		cout<<endl;
		res = system(cmd.c_str());
		
		if(DELETEUNSMEARED && res == 0){
			cmd = "rm -f "+OUTFILE;
			cout<<endl;
			cout<<"Deleting unsmeared file:"<<endl;
			cout<<"   "<<cmd<<endl;
			res = system(cmd.c_str());
                }else if(DELETEUNSMEARED){
			cout<<endl;
			cout<<"Not deleting unsmeared file ";
			cout<<"because of problems with the smearing.";
			cout<<endl;
		}
	}else if(POSTSMEAR){
		cout<<endl;
		cout<<"Skipping smearing step because of problems ";
                cout<<"with the hdgeant simulation"<<endl;
		cout<<endl;
	}else{
		if(DELETEUNSMEARED){
			cerr<<endl;
			cerr<<"The DELETEUNSMEARED flag is set indicating that you want the output file"<<endl;
			cerr<<"\""<<OUTFILE<<"\" to be deleted. However, the POSTSMEAR flag is not set"<<endl;
			cerr<<"so no smeared data file was created so deleting the unsmeared file is"<<endl;
			cerr<<"most likely NOT what you want to do. Therefore, I'm ignoring the"<<endl;
			cerr<<"DELETEUNSMEARED flag. "<<__FILE__<<":"<<__LINE__<<endl;
		}
	}

	return res;
}
//------------------
// main
//------------------
void Usage(void)
{
	cout<<endl;
	cout<<"Usage:"<<endl;
	cout<<"   hdgeant [options]"<<endl;
	cout<<endl;
	cout<<" Hall-D Monte Carlo simulation based on GEANT3."<<endl;
	cout<<"Most configurable options are set using a control.in"<<endl;
	cout<<"file. A well-annotated example control.in can be"<<endl;
	cout<<"found in the HDGeant source code directory with."<<endl;
	cout<<endl;
	cout<<" options:"<<endl;
	cout<<"    -h or --help          Print this usage statement"<<endl;
	cout<<"    -xml[=main_HDDS.xml]  Dynamically generate geometry"<<endl;
	cout<<"    -checksum             Print the MD5 checksum of the "<<endl;
	cout<<"                          geometry and exit"<<endl;
	cout<<endl;
	cout<<"If the -xml option is given and no file is specified,"<<endl;
	cout<<"then a value of: "<<HDDS_XML<<endl;
	cout<<"is used."<<endl;
	cout<<endl;

	exit(0);
}