/***************************************************************

    Class for parsing inputs

    Written by k-fleak

***************************************************************/

#include <iostream>
#include <fstream> 
#include <sstream>
#include <iterator>
#include <boost/lexical_cast.hpp>
#include <string>
#include <vector>
#include <map>

#ifndef __PARSEINPUT_CPP
#define __PARSEINPUT_CPP


class ParseInput{
private:
  std::ifstream input;
  std::map<std::string, std::vector<std::string> > dataArray;
  void checkName(std::string name);
public:
  ParseInput(char* fileName);
  ~ParseInput();
  template<class dataType> void getData(char* dataName, dataType &dataOut);
  template<class dataType> void getData(char* dataName, std::vector<dataType> &dataOut);
  template<class dataType> void setDefault(char* dataName, dataType &defaults);
  void printData();
};


ParseInput::ParseInput(char* filename){
  input.open(filename);
  std::string line;
  
  while(!input.eof()){
    std::getline(input, line);
    std::stringstream str;
    str << line;
    std::istream_iterator<std::string> iss_iter (str);
    std::istream_iterator<std::string> iss_end;
    std::string dataName(*iss_iter);
    std::vector<std::string> dataIn;
    std::advance( iss_iter, 2 );
    while (iss_iter != iss_end){
      dataIn.push_back(*iss_iter);
      ++iss_iter;
    }
    dataArray[dataName] = dataIn;
    dataIn.clear();
  }
  input.close();
}

ParseInput::~ParseInput(){}

template<class dataType> void ParseInput::getData(char* dataName, dataType &dataOut){
  checkName(dataName);
  dataOut = boost::lexical_cast<dataType>(dataArray[dataName][0]);
}

template<class dataType> void ParseInput::getData(char* dataName, std::vector<dataType> &dataOut){
  checkName(dataName);
  for ( int i = 0; i < (dataArray[dataName]).size(); ++i){
    dataOut.push_back(boost::lexical_cast<dataType>(dataArray[dataName][i]));
  }
} 

template<class dataType> void ParseInput::setDefault(char* dataName, dataType &defaults){
  std::map<std::string, std::vector<std::string> >::iterator it;
  it = dataArray.find(dataName);
  std::vector<std::string> vtmp;
  vtmp.push_back(boost::lexical_cast<std::string>(defaults));
  if ( it == dataArray.end()){  
    dataArray[dataName] = vtmp;
  }
}


void ParseInput::checkName(std::string name){
  std::map<std::string, std::vector<std::string> >::iterator it;
  it = dataArray.find(name);
  if ( it == dataArray.end()){
    std::cout << "Error while reading input files. Please check the name: " << name << std::endl;
    exit(0);
  }
}







#endif