/*
 * Example program. This code shows how to use the train procedure, in
 * order to generate weight file to be used by the KNN based
 * classifier.
 */
#include <sstream>

#include "PndKnnTrain.h"
#include "TRandom3.h"

int str2int (const std::string &str)
{
  std::stringstream ss(str);
  int n;
  ss >> n;
  return n;
}

std::string int2str (int n)
{
  std::stringstream ss;
  ss << n;
  return ss.str();
}


int main(int argc, char** argv)
{
  if(argc < 3){
    std::cerr << "\t<ERROR> Usage\n"
	      <<"\t./train <NumEvents> <OutFile>"
	      << std::endl;
    return 1;
  }
  
  int totEvtNum = 0;
  std::string numstr = argv[1];
  std::istringstream buff(numstr);
  buff >> totEvtNum;
  
  std::string ot   = argv[2];
  
  std::vector<std::string> clas;
  std::vector<std::string> nam;
  
  // Class names
  clas.push_back("Elec"); clas.push_back("Pion"); 
  clas.push_back("Kaon"); clas.push_back("Gamma"); 
  clas.push_back("Muon"); clas.push_back("Prot");
  
  // Parameter names
  nam.push_back("emc"); nam.push_back("tof"); nam.push_back("mvd");
  nam.push_back("p");   nam.push_back("stt"); nam.push_back("tpc");

  const char* file = ot.c_str();
  
  PndKnnTrain bla (file, clas, nam);
  
  std::vector<float> evt;
  
  TRandom3 myran(987629);
  
  //El events
  for(int i = 0; i < totEvtNum; i++){
    evt.clear();
    for(unsigned int j = 0; j < nam.size(); j++){
      evt.push_back(myran.Uniform(-1,1));
    }
    bla.Train(evt,"Elec");
  }
  
  //Pi events
  for(int i = 0; i < totEvtNum; i++){
    evt.clear();
    for(unsigned int j = 0; j < nam.size(); j++){
      evt.push_back(myran.Uniform(1,3));
    }
    bla.Train(evt,"Pion");
  }
  
  // /*
  //Ka events
  for(int i = 0; i < totEvtNum; i++){
    evt.clear();
    for(unsigned int j = 0; j < nam.size(); j++){
      evt.push_back(myran.Uniform(3,4));
    }
    bla.Train(evt,"Kaon");
  }
  
  //Gam events
  for(int i = 0; i < totEvtNum; i++){ 
    evt.clear();
    for(unsigned int j = 0; j < nam.size(); j++){
      evt.push_back(myran.Uniform(4,6));
    }
    bla.Train(evt,"Gamma");
  }
  
  //mu events
  for(int i = 0; i < totEvtNum; i++){
    evt.clear();
    for(unsigned int j = 0; j < nam.size(); j++){
      evt.push_back(myran.Uniform(6,8));
    }
    bla.Train(evt,"Muon");
  }

  // proton events
  for(int i = 0; i < totEvtNum; i++){
    evt.clear();
    for(unsigned int j = 0; j < nam.size(); j++){
      evt.push_back(myran.Uniform(10,15));
    }
    bla.Train(evt,"Prot");
  }
  // */
  std::cout << "Done generating events.\n"
	    <<"Output is saved in " << ot
	    << std::endl;
  return 0;
}