/*
  ファイルの読み込み操作用関数群
*/
#ifndef __LOADFILE_H
#define __LOADFILE_H


#include <iostream>
#include <fstream>

#include <string>
#include <vector>


using namespace std;

/*
  perl でいうところの split() っぽい関数のC++版実装
*/
int split(string delimiter, string source, vector<string> &storage) 
{
  storage.clear();
  string::iterator delim_It, srcIt;

  string word = "";
  int wordnum=0;

  for( srcIt=source.begin(); srcIt!=source.end(); srcIt++){
    int flg_delim=0;

    // 複数のデリミタについて走査,デリミタがあったらフラグを立てる
    for( delim_It=delimiter.begin(); delim_It!=delimiter.end(); delim_It++ )
      if( (*srcIt)==(*delim_It) ) flg_delim=1;

    // デリミタフラグが立っているなら単語として格納
    if( flg_delim!=1 ) word.push_back(*srcIt);
    else if( !(word.empty()) ){
      storage.push_back(word);
      wordnum++;
      word="";
    }
  }

  // 入力行末がデリミタまたはwordが空なら関数を抜ける
  for( delim_It=delimiter.begin(); delim_It!=delimiter.end(); delim_It++ )
    if( (*(source.end()-1)==(*delim_It))||(word.empty()) ) return(wordnum);

  storage.push_back(word);
  wordnum++;
  return(wordnum);
}


/*
  split()をつかってテーブル形式のデータファイルから二次元ベクタに格納．
*/
int LoadData(string delimiter, string filename, vector< vector<double> > &output_vector)
{
  string str;
  ifstream infile;
  vector<string> lines;
  vector<string>::iterator linesIt;
  vector<string> tmpline;
  vector<string>::iterator tmplineIt;
  vector<double> tmpline_to_double;

  lines.clear();
  output_vector.clear();
  

  // ファイルを読み込んでstring型ベクタに格納する
  infile.open(filename.c_str(),ios::in);
  if(!infile){
    cout << "Can't open file:["<<filename<<"]"<< endl;
    return(0);
  }
  int errcnt=0;
  while(1) {
    getline(infile,str);
    if (infile.fail()){
      // 行の取得に失敗したら中止
      cout<<" FAIL:read lines error."<<endl;
      return(0);
    }
    if (str[0] == '!') break;     // 終了行がきたら読むのを止めて格納処理に移る    
    if (str[0] == '#') continue;  // コメント行は読み飛ばす

    lines.push_back(str);

    errcnt++;
    if(errcnt>10000){
      // データ行が大き過ぎる（想定外）ならエラー終了
      cout<<" FAIL:terminater [!] doesnt exist."<<endl;
      return(0); 
    }
  }
  infile.close();

  // string型ベクタをdouble型二次元なベクタに変換する
  for(linesIt=lines.begin(); linesIt!=lines.end(); linesIt++){
    tmpline.clear();
    tmpline_to_double.clear();

    // 実装はsplit.cpp
    split(delimiter, (*linesIt), tmpline);

    for(tmplineIt=tmpline.begin(); tmplineIt!=tmpline.end(); tmplineIt++)
      tmpline_to_double.push_back( atof( (*tmplineIt).c_str() ) );
    output_vector.push_back(tmpline_to_double);
  }
  return(1);
}


/*
  テーブル形式のデータファイルから二次元配列に格納．
*/
int LoadData(string delimiter, string filename,
	     int &size_i, int &size_j, double** &storage)
{
  vector< vector<double> > tmpvect;

  // テーブル形式のデータファイルから二次元に格納
  LoadData(delimiter, filename, tmpvect);

  // arr[i][j]について最大のjを取得する(動的に配列を確保する時のオーバーフロー防止)
  // 計算領域を矩形として扱うためにすべてのjを揃える
  size_i = tmpvect.size();
  for(int i=0; i<size_i; i++)
    if( size_j < (int)tmpvect[i].size() ) size_j = tmpvect[i].size();

  // double型の二次元配列を動的に確保し,初期化する
  storage = new (double*)[size_i];
  for(int i=0; i<size_i; i++) { storage[i] = new (double)[size_j]; }
  for(int i=0; i<size_i; i++) {
    for(int j=0; j<size_j; j++){
      storage[i][j]=0;
    }
  }

  // 二元ベクタ→二元配列
  {
    vector< vector<double> >::iterator i_It;
    vector<double>::iterator j_It;

    int i=0;
    for( i_It=tmpvect.begin(); i_It!=tmpvect.end(); i_It++ ){
      int j=0;
      for( j_It=(*i_It).begin(); j_It!=(*i_It).end(); j_It++){
	storage[i][j]=(*j_It);
	j++;
      }
      i++;
    }
  }

  return(1);
}

#endif