#include <iostream>
#include <string>
#include <sstream>
#include <vector>
#include "PHITStoRooTracker.hh"

using namespace std;
void usage();
void TestFormat(string);
void Deposit(string);
void ICEDUSTFormat(string);
static int bunch = 1000;
static int event = 1000;

int main(int argc, char** argv) {
  
  if (argc<1) cerr << "Error: there is no input file or options!" << endl;

  vector<string> input;
  for (int i=1; i<argc; i++) input.push_back(string(argv[i]));
  
  // option:
  for (int i=0; i<input.size(); i++) {
    if (input[i]=="-b" || input[i]=="--bunch") {
      istringstream bch(input[i+1]);
      bch >> bunch;
    }
    else if (input[i]=="-e" || input[i]=="--event") {
      istringstream evt(input[i+1]);
      evt >> event;
    }
  }
  
  for (int i=0; i<input.size(); i++) {
    if (input[i]=="-t" || input[i]=="--testoutput")    TestFormat(input[i+1]);
    else if (input[i]=="-i" || input[i]=="--icedust")  ICEDUSTFormat(input[i+1]);
    else if (input[i]=="-d" || input[i]=="--deposit")  Deposit(input[i+1]);
    else if (input[i]=="-h" || input[i]=="--help")     usage();
  }
  
  /* read multi files*/
  /*for (int i=0; i<input.size(); i++) {
    if (input[i]=="-h" || input[i]=="--help")   usage();
  else if (input[i]=="-m" || input[i]=="--my") {
      for (int j=i+1; j<input.size(); j++) TestFormat(input[j]);
  }
  else if (input[i]=="-d" || input[i]=="--deposit") {
      for (int j=i+1; j<input.size(); j++) Deposit(input[j]);
  }
  else if (input[i]=="-i" || input[i]=="--icedust") {
      for (int j=i+1; j<input.size(); j++) ICEDUSTFormat(input[j]);
  }
  else if (!input[i].find("-") && !input[i].find("=")) {
      for (int j=i+1; j<input.size(); j++) TestFormat(input[j]);
  }
  }*/

  return 0;
}


void usage() {
  cout << "\nUsage\n";
  cout << "       SimPHITSToRooTracker.exe  <option>  [input file]" << endl;
  cout << "\n  RooTracker is a simple program to convert PHITS dump file into a root file," << endl;
  cout << "then it will be read out by GEANT4 to simulate physics.\n";
  cout << "\nOptions:\n";
  cout << "\n  -h, --help              help" << endl;
  cout << "  -b, --bunch             number of bunch" << endl;
  cout << "  -e, --event             number of event" << endl;
  cout << "  -t, --testoutput        a simple format used to check the data" << endl;
  cout << "  -d, --deposit           convert energy depostion data to root file" << endl;
  cout << "  -i, --icedust           convert dump file to icedust format\n" << endl;
}


void TestFormat(string file) {
  cout << "\n file: " << file << " --> convert to my format" << endl;
  cout << "reading... " << endl;
  SimPHITSToRooTracker* rt = new SimPHITSToRooTracker(file);
  rt->WriteToTestFormat(bunch, event);
}


void Deposit(string file) {
  cout << "\n file: " << file << " --> convert to root file" << endl;
  cout << "reading..." << endl;
  SimPHITSToRooTracker* rt = new SimPHITSToRooTracker(file);
  rt->ReadEnergyDeposition(bunch, event);
}


void ICEDUSTFormat(string file) {
  cout << "\n file: " << file << " --> convert to icedust format" << endl;
  cout << "reading..." << endl;
  SimPHITSToRooTracker* rt = new SimPHITSToRooTracker(file);
  rt->WriteToICEDUSTFormat(bunch, event);
}