/*
 * Copyright (C) 2004 Laurent Mazet
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

/* 2004-04-08
 *   initial release
 */

#include <fstream>

#include <octave/oct.h>
#include <octave/Cell.h>

#define MAXSTRINGLENGTH 4096

DEFUN_DLD (csv2cell, args, nargout, 
           "-*- texinfo -*-\n"
           "@deftypefn {Function File} {@var{c} = } csv2cell (@var{file})\n"
	   "@deftypefnx {Function File} {@var{c} = } csv2cell (@var{file}, @var{sep})\n"
	   "@deftypefnx {Function File} {@var{c} = } csv2cell (@var{file}, @var{sep}, @var{prot})\n"
	   "\n"
	   "Read a CSV (Comma Separated Values) file and convert it into a "
	   "cell. "
	   "@var{sep} changes the character used to separate two fields. By "
	   "default, two fields are expected to be separated by a coma "
	   "(@code{,}). @var{prot} changes the character used to protect a string. "
	   "By default it's a double quote (@code{\"}).\n"
	   "@end deftypefn") {

  /* Get arguments */
  std::string file = args(0).string_value();

  std::string _sep = (args.length() > 1) ? args(1).string_value() : ",";
  if (_sep.length() != 1) {
    error("only on charactere need as separator\n");
    return octave_value();
  }
  char sep = _sep[0];

  std::string _prot = (args.length() > 2) ? args(2).string_value() : "\"";
  if (_prot.length() != 1) {
    error("only on charactere need as protector\n");
    return octave_value();
  }
  char prot = _prot[0];

  /* Open file */
  std::ifstream fd(file.c_str());
  if (!fd.is_open()) {
    error("cannot read %s\n", file.c_str());
    return octave_value();
  }
  fd.seekg (0, std::ios::end);
  long fdend = fd.tellg();
  fd.seekg (0, std::ios::beg);

  /* Buffers */
  char line[MAXSTRINGLENGTH];
  std::string str, word;
  bool inside = false;

  /* Read a line */
  str = "";
  fd.getline(line, MAXSTRINGLENGTH);
  while (fd.fail()) {
    fd.clear();
    str += line;
    fd.getline(line, MAXSTRINGLENGTH);
  }
  str += line;

  /* Parse first to get number of columns */
  int nbcolumns = 0;
  for (int i=0, len=str.length(); i<=len; i++)
    if (((i==len) || (str[i] == sep)) && (!inside))
      nbcolumns++;
    else if ((inside) && (str[i]==prot) && ((i<len) && (str[i+1]==prot)))
      ++i;
    else if (str[i] == prot)
      inside = !inside;
  
  /* Read all the file to get number of rows */
  int nbrows = 1;
  while (fd.tellg() < fdend) {
    fd.getline(line, MAXSTRINGLENGTH);
    while (fd.fail()) {
      fd.clear();
      fd.getline(line, MAXSTRINGLENGTH);
    }
    nbrows++;
  }

  /* Rewind */
  fd.seekg (0, std::ios::beg);
  if (!fd.good()) {
    error("cannot reread %s\n", file.c_str());
    return octave_value();
  }

  /* Read all the file until the end */
  Cell c(nbrows, nbcolumns);
  for (int i=0; i<nbrows; i++) {
    
    /* Read a line */
    str = "";
    fd.getline(line, MAXSTRINGLENGTH);
    while (fd.fail()) {
      fd.clear();
      str += line;
      fd.getline(line, MAXSTRINGLENGTH);
    }
    str += line;

    /* Explode a line into a sub cell */
    word = "";
    inside = false;
    int j = 0;
    for (int k=0, len=str.length(); k<=len; k++) {
      if (((k==len) || (str[k] == sep)) && (!inside)) {

	/* Check number of columns */
	if (j == nbcolumns) {
	  fd.close();
	  error("incorrect CSV file, line %d too long\n", i+1);
	  return octave_value();
	}

	/* Check if scalar */
	const char *word_str=word.c_str();
	char *err;
	double val = strtod (word_str, &err);

	/* Store into the cell */
	c(i, j++) = ((word == "") || (err != word_str+word.length())) ?
	  octave_value(word) : octave_value(val);
	word = "";
      }
      else if ((inside) && (str[k]==prot) && ((k<len) && (str[k+1]==prot))) {

	/* Insisde a string */
	word += prot;
	++k;
      }
      else if (str[k] == prot)
	/* Changing */
	inside = !inside;
      else
	word += str[k];
    }

    /* Check number of columns */
    if (j != nbcolumns) {
      fd.close();
      error("incorrect CSV file, line %d too short\n", i+1);
      return octave_value();
    }
  }

  /* Close file */
  fd.close();

  return octave_value(c);
}