#include "veto_psim.h"

void ParseCommandLineArgs(int narg, char* argv[]);
void Usage(void);

int Ngen = 0;
int Ndet = 0;
int N0veto = 0;
int N1veto = 0;
int N2veto = 0;
long int rSeed = 1;

//-----------------------------------------------------------------
// main()
//-----------------------------------------------------------------
int main(int narg, char* argv[])
{	
  // Proccess command line
  ParseCommandLineArgs(narg, argv);

  srand48(rSeed); // initialize the random seed

  // Create a new PEvent_Loop object
  PEvent_Loop event_loop = PEvent_Loop(narg, argv);
  
  // Have the PEvent_Loop call my hist_book() and hist_end routines when the run number changes
  event_loop.RegisterBRUNProc(hist_book);
  event_loop.RegisterERUNProc(hist_end);
  
  // Process all events. Have the PEvent_Loop call my hist_fill routine every event
  // The PROCESS_ALL flag actually specifies which packages to run (tagger, ps, ...)
  event_loop.Run(hist_fill, PROCESS_ALL);
  
  cout<<"veto_psim Statistics"<<endl;
  cout<<"=========================================================="<<endl;
  cout<<"Generated =\t"<<Ngen<<endl;
  cout<<"Detected =\t"<<Ndet<<endl;
  cout<<"Veto 0 hits=\t"<<N0veto<<endl;
  cout<<"Veto 1 hits=\t"<<N1veto<<endl;
  cout<<"Veto 2 hits=\t"<<N2veto<<endl;

  return 0;
}


//-----------------------------------------------------------------
// Parse command line arguments
//-----------------------------------------------------------------
void ParseCommandLineArgs(int narg, char* argv[])
{
  char *ptr;
  
  if(narg==1)Usage();
  
  for(int i=1;i<narg;i++){
    ptr = argv[i];
    if(*(ptr++) == '-'){
      switch(*(ptr++)){
      case 'h':
	Usage();
	break;
      case 'R':
	break; // -R is processed in PEvent_Loop
      case 'S':
	rSeed = atoi(ptr);
	cout<<"Using random seed "<<rSeed<<endl;
	break;
      default:
	cerr<<endl<<"Unknown option \""<<argv[i]<<"\""<<endl<<endl;
	exit(-1);
      }
    }
  }
}

//-----------------------------------------------------------------
// Print the Usage statement
//-----------------------------------------------------------------
void Usage(void)
{
  cout<<endl<<endl;
  cout<<"Usage:"<<endl;
  cout<<ansi_red<<ansi_bold;
  cout<<"\t veto_psim [options] filesource [filesource] [filesource] ..."<<endl<<endl;
  cout<<ansi_normal;

  cout<<"\"filesource\" can be either a file or an ET system for online use. Any number of"<<endl;
  cout<<"filesources may be specified, though it is not useful to specify more than one"<<endl;
  cout<<"ET system. Specify an ET system, as ";
  cout<<ansi_blue<<ansi_bold<<"ET:primex"<<ansi_normal<<". By default, the station C2HIST"<<endl;
  cout<<"is created (if it does not already exist). Alternative stations can be created"<<endl;
  cout<<"by specifying it in the file source. e.g. ";
  cout<<ansi_blue<<ansi_bold<<"ET:primex:C2HIST2"<<ansi_normal<<"."<<endl;
  
  cout<<endl<<"\t options:"<<endl;
  cout<<"\t\t-h        Print this usage statement"<<endl;
  cout<<"\t\t-S#       User supplied random seed(must be long int)"<<endl;
	
  exit(0);
}