#include "TString.h"
#include "TROOT.h"


#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <iostream>
using namespace std;

#include "pluto_embedded.C" // Int_t pluto_embedded(TString outdir = "./", TString outfile = "test",TString type = "pi0",Int_t nEvents = 1000, Int_t source = 1,Float_t Energy = 1.58, TString datbasefile = "database_mar19.dat", TString vertexntuple = "")

struct parameters {
    TString  outdir;
    TString  outfile;
    TString  type;
    Int_t    nEvents;
    Int_t    source;
    Float_t  Energy;
    TString  datbasefile;
    TString  vertexntuple;
    Int_t    help;

    void print(){

	cout<<"---------------------------------------------------"<<endl;
	cout<<"Params :"<<endl;
        cout<<"outdir     = "<<outdir<<endl;
        cout<<"outfile    = "<<outfile<<endl;
        cout<<"type       = "<<type<<endl;
        cout<<"nEvents    = "<<nEvents<<endl;
        cout<<"source     = "<<source<<endl;
        cout<<"Energy     = "<<Energy<<endl;
        cout<<"dataBase   = "<<datbasefile<<endl;
        cout<<"vertexntup = "<<vertexntuple<<endl;
	cout<<"---------------------------------------------------"<<endl;
    }


};

void help(TString progname)
{

    cout<<"---------------------------------------------------"<<endl;
    cout<<"USAGE : thermal pi0 (optional using vertexentuple) : "<<progname<<" -d outdir -f outfile -t pi0 -s 1 -n 1000 -e 1.58 -b database_mar19.dat -r vertex.root"<<endl;
    cout<<"USAGE : white   pi0 (optional using vertexentuple) : "<<progname<<" -d outdir -f outfile -t pi0 -s 0 -n 1000 -r vertex.root"<<endl;
    cout<<"--help       , -h  : print this help"<<endl;
    cout<<"--outdir     , -d  : output directory"<<endl;
    cout<<"--outfile    , -f  : output file"<<endl;
    cout<<"--type       , -t  : particle type (thermal source :  keyword from database, white source: pluto particle name)"<<endl;
    cout<<"--nevents    , -n  : number of events"<<endl;
    cout<<"--source     , -s  : 0 white, 1 thermal fireball"<<endl;
    cout<<"--energy     , -e  : Beam energy [GeV] (optional, required for thermal source)"<<endl;
    cout<<"--database   , -b  : data base file    (optional, required for thermal source)"<<endl;
    cout<<"--rootntuple , -r  : root vertex ntuple(optional) : ROOT file containing TNtuple vertex : Float_t vX,vY,vZ, Int_t seqNumber per event"<<endl;
    cout<<"---------------------------------------------------"<<endl;

}


#ifndef __CINT__
int main(int argc, char **argv)
{
    TROOT Analysis("Analysis","compiled analysis macro");
    TString progname = argv[0];
    parameters params;


    //---------------------------------------------------------------------------------
    params.nEvents          =  0;
    params.source           = -1;
    params.Energy           =  0;
    params.datbasefile      = "";
    params.vertexntuple     = "";
    params.help             =  0;

    while (1)
    {
	static struct option long_options[] =
	{
	    // These options set a flag.
	    {"help"       ,  no_argument      , 0, 'h'},
	    {"outdir"     ,  required_argument, 0, 'd'},
	    {"outfile"    ,  required_argument, 0, 'f'},
	    {"type"       ,  required_argument, 0, 't'},
	    {"nevents"    ,  required_argument, 0, 'n'},
	    {"source"     ,  required_argument, 0, 's'},
	    {"energy"     ,  required_argument, 0, 'e'},
	    {"database"   ,  required_argument, 0, 'b'},
	    {"rootntuple" ,  required_argument, 0, 'r'},
	    {0, 0, 0, 0}
	};
	// getopt_long stores the option index here.
	Int_t option_index = 0;

	Int_t c = getopt_long (argc, argv, "d:f:t:n:s:e:b:r:h", long_options, &option_index);

	// Detect the end of the options.
	if (c == -1)
	    break;

	switch (c)
	{
	case 0:
	    // If this option set a flag, do nothing else now.
	    if (long_options[option_index].flag != 0)
		break;
	    printf ("option %s", long_options[option_index].name);
	    if (optarg) printf (" with arg %s", optarg);
	    printf ("\n");
	    break;

	case 'd':
	    params.outdir = optarg;
	    break;

	case 'f':
	    params.outfile = optarg;
	    break;

	case 't':
	    params.type = optarg;
	    break;

	case 'n':
	    params.nEvents  = atoi(optarg);
	    break;

	case 's':
	    params.source = atoi(optarg);
	    break;

	case 'h':
	    params.help = 1;
	    break;

	case 'e':
	    params.Energy = atof(optarg);
	    break;

	case 'r':
	    params.vertexntuple = optarg;
	    break;

	case 'b':
	    params.datbasefile =  optarg;
	    break;

	case '?':
	    // getopt_long already printed an error message.
	    break;

	default:
	    return 1;
	}
    }

    if(params.help){
	help(progname);
        return 0;
    }


    params.print();

    if(params.source == 1 && params.datbasefile == ""){
	cerr<<"ERROR: option --sources == 1, require database  file"<<endl;
        help(progname);
	return 1;
    }

    if(params.source == 1 && params.Energy == 0){
	cerr<<"ERROR: option --sources == 1, require Energy !"<<endl;
        help(progname);
        return 1;
    }

    if(params.outfile == ""){
	cerr<<"ERROR: option --outfile not set , required !"<<endl;
        help(progname);
        return 1;
    }

    if(params.outfile == ""){
	cerr<<"ERROR: option --outdir not set , required !"<<endl;
        help(progname);
        return 1;
    }

    if(params.type == ""){
	cerr<<"ERROR: option --type not set , required !"<<endl;
        help(progname);
        return 1;
    }

    if(params.source == -1){
	cerr<<"ERROR: option --source not set , required !"<<endl;
        help(progname);
        return 1;
    }



    // Print any remaining command line arguments (not options).
    if (optind < argc)
    {
	cerr<<"ERROR: non-option ARGV-elements: ";
	while (optind < argc)
	    cerr <<" "<< argv[optind++];
	cerr<<endl;
        help(progname);
	return 1;
    }

    //---------------------------------------------------------------------------------

    return pluto_embedded(params.outdir,params.outfile,params.type,params.nEvents,params.source,params.Energy,params.datbasefile,params.vertexntuple);

}
#endif