/**
 * @file dataset.h
 *
 * Generic datasets and associated utilities.
 */

#ifndef DATA_DATASET_H
#define DATA_DATASET_H

#include "col/string.h"
#include "la/matrix.h"
#include "math/discrete.h"

class TextLineReader;
class TextWriter;

/**
 * Metadata about a particular dataset feature (attribute).
 *
 * Supports nominal, continuous, and integer values.
 */
class DatasetFeature {
 public:
  /**
   * Feature types supported.
   */
  enum Type {
      /** Real-valued data. */
      CONTINUOUS,
      /** Integer valued data. */
      INTEGER,
      /** Discrete data, each of which has a "name". */
      NOMINAL
  };
  
 private:
  /** Name of the feature. */
  String name_;
  /** Type of data this feature represents. */
  Type type_;
  /** If nominal, the names of each numbered value. */
  ArrayList<String> value_names_;
  
  OT_DEF(DatasetFeature) {
    OT_MY_OBJECT(name_);
    OT_MY_OBJECT(type_);
    OT_MY_OBJECT(value_names_);
  }
  
 public:
  DatasetFeature() {}
  ~DatasetFeature() {}
  
  /**
   * Initialize to be a continuous feature.
   *
   * @param name_in the name of the feature
   */
  void InitContinuous(const char *name_in) {
    name_.Copy(name_in);
    type_ = CONTINUOUS;
    value_names_.Init();
  }

  /**
   * Initializes to an integer type.
   *
   * @param name_in the name of the feature
   */
  void InitInteger(const char *name_in) {
    name_.Copy(name_in);
    type_ = CONTINUOUS;
    value_names_.Init();
  }

  /**
   * Initializes to a nominal type.
   *
   * The value_names list starts empty, so you need to add the name of
   * each feature to this.  (The dataset reading functions will do this
   * for you).
   *
   * @name_in the name of the feature
   */
  void InitNominal(const char *name_in) {
    name_.Copy(name_in);
    type_ = NOMINAL;
    value_names_.Init();
  }
  
  /**
   * Creates a text version of the value based on the type.
   *
   * Continuous parameters are printed in floating point, and integers
   * are shown as integers.  For nominal, the value_name(int(value)) is
   * shown.  NaN (missing data) is always shown as '?'.
   *
   * @param value the value to format
   * @param result this will be initialized to the formatted text
   */
  void Format(double value, String *result) const;
  
  /**
   * Parses a string into the particular value.
   *
   * Integers and continuous are parsed using the normal functions.
   * For nominal, the entry 
   *
   * If an invalid parse occurs, such as a mal-formatted number or
   * a nominal value not in the list, SUCCESS_FAIL will be returned.
   *
   * @param str the string to parse
   * @param d where to store the result
   */
  success_t Parse(const char *str, double *d) const;
  
  /**
   * Gets what the feature is named.
   *
   * @return the name of the feature; for point, "Age" or "X Position"
   */
  const String& name() const {
    return name_;
  }
  
  /**
   * Identifies the type of feature.
   *
   * @return whether this is DatasetFeature::CONTINUOUS, INTEGER, or NOMINAL
   */
  Type type() const {
    return type_;
  }
  
  /**
   * Returns the name of a particular nominal value, given its index.
   *
   * The first nominal value is 0, the second is 1, etc.
   *
   * @param value the number of the value
   */
  const String& value_name(int value) const {
    DEBUG_ASSERT(type_ == NOMINAL);
    return value_names_[value];
  }
  
  /**
   * The number of nominal values.
   *
   * The values 0 to n_values() - 1 are valid.
   * This will return zero for CONTINUOUS and INTEGER types.
   *
   * @return the number of nominal values
   */
  index_t n_values() const {
    return value_names_.size();
  }
  
  /**
   * Gets the array of value names.
   *
   * Useful for creating a nominal feature yourself.
   *
   * @return a mutable array of value names
   */
  ArrayList<String>& value_names() {
    return value_names_;
  }
};

/**
 * Information describing a dataset and its features.
 */
class DatasetInfo {
  FORBID_COPY(DatasetInfo);
 
 private:
  String name_;
  ArrayList<DatasetFeature> features_;
  
  OT_DEF(DatasetInfo) {
    OT_MY_OBJECT(name_);
    OT_MY_OBJECT(features_);
  }
  
 public:
  DatasetInfo() {}
  ~DatasetInfo() {}
  
  /** Gets a mutable list of all features. */
  ArrayList<DatasetFeature>& features() {
    return features_;
  }
  
  /** Gets information about a particular feature. */
  const DatasetFeature& feature(index_t attrib_num) const {
    return features_[attrib_num];
  }
  
  /** Gets the number of features. */
  index_t n_features() const {
    return features_.size();
  }
  
  /** Gets the title of the data set. */
  const char *name() const {
    return name_;
  }
  
  /** Sets the title of the data set. */
  void set_name(const char *name_in) {
    name_.Destruct();
    name_.Copy(name_in);
  }

  /**
   * Checks if all parameters are continuous.
   */
  bool is_all_continuous() const;

  /**
   * Initialize an all-continuous dataset;
   *
   * @param n_features the number of continuous features
   * @param name_in the dataset title
   */
  void InitContinuous(index_t n_features,
      const char *name_in = "dataset");

  /**
   * Initialize a custom dataset.
   *
   * This assumes you will eventually use the features() to add features.
   *
   * @param name_in the dataset title
   */
  void Init(const char *name_in = "dataset");

  /**
   * Writes the header for an ARFF file.
   */
  void WriteArffHeader(TextWriter *writer) const;
  
  /**
   * Writes header for CSV file.
   *
   * @param sep the value separator (use ",\t" for CSV)
   */
  void WriteCsvHeader(const char *sep, TextWriter *writer) const;

  /**
   * Writes the contents of a matrix to a file.
   *
   * @param matrix the matrix
   * @param sep the separator (use ",\t" for CSV)
   * @param writer the writer to write to
   */
  void WriteMatrix(const Matrix& matrix, const char *sep,
      TextWriter *writer) const;

  /**
   * Initialize explicitly from an ARFF file.
   *
   * ARFF LIMITATIONS: Values cannot have spaces or commas, even with quotes;
   * 'string' data type not supported (nominal is supported).
   *
   * You might just use InitFromFile, which will guess the type for you.
   *
   * This will read only the header information and leave the reader at the
   * first line of data.
   */
  success_t InitFromArff(TextLineReader *reader,
      const char *filename = "dataset");
  
  /**
   * Initialize from a CSV-like file with numbers only, inferring
   * automatically that if the first row has non-numeric characters, it is
   * a header.
   *
   * InitFromFile will automatically detect this.
   */
  success_t InitFromCsv(TextLineReader *reader,
      const char *filename = "dataset");

  /**
   * Initializes the header from a file, either CSV or ARFF.
   *
   * All header lines will be gobbled, so the reader's position will be
   * left at the first line of actual data.
   * You can then read the data with matrix.
   */
  success_t InitFromFile(TextLineReader *reader,
      const char *filename = "dataset");
  /**
   * Populate a matrix from a file, given the internal data model.
   *
   * ARFF LIMITATIONS: Values cannot have spaces or commas, even with quotes;
   * 'string' data type not supported (nominal is supported).
   *
   * @param reader the reader to get lines from
   * @param matrix the matrix to store text into
   */
  success_t ReadMatrix(TextLineReader *reader, Matrix *matrix) const;
  
  /**
   * Initializes to be a copy.
   * @param other data set feature information to copy
   */
  void Copy(const DatasetInfo& other) {
    name_.Copy(other.name_);
    features_.Copy(other.features_);
  }

 private:

  char *SkipSpace_(char *s);

  char *SkipNonspace_(char *s);

  void SkipBlanks_(TextLineReader *reader);

};


/**
 * Most generic dataset type.
 *
 * To allow polymorphic usage, everything is internally stored as doubles.
 * If your data are discrete, the integer values are stored in the doubles.
 *
 * (Implementation note: To allow efficient polymorphic use of mixed data,
 * we found it is far more efficient to store these as doubles than to
 * introduce any sort of casting or pointer arithmetic overhead.  Space
 * may be an issue, especially if this is Boolean data.)
 */
class Dataset {
  FORBID_COPY(Dataset);
  
 private:
  Matrix matrix_;
  DatasetInfo info_;
  
  OT_DEF(Dataset) {
    OT_MY_OBJECT(matrix_);
    OT_MY_OBJECT(info_);
  }
  
 public:
  Dataset() {}
  ~Dataset() {}
  
  /**
   * Metadata about the feature types and names for the dataset.
   *
   * Each dataset has a name, and each feature is described by a type
   * (continuous, etc) and a name, modelled loosely over the ARFF format.
   * See the DatasetInfo documentation for more information.
   * 
   * @return metadata about the dataset
   */
  const DatasetInfo& info() const {
    return info_;
  }
  
  /**
   * Metadata about the feature types and names for the dataset.
   * @return a mutable reference to the dataset
   */
  DatasetInfo& info() {
    return info_;
  }
  
  /**
   * Gets the number of features/attributes the dataset has.
   *
   * This corresponds to the number of <i>rows</i> in the matrix.
   *
   * @return the number of features, or variables, in the dataset
   */
  index_t n_features() const {
    return matrix_.n_rows();
  }
  
  /**
   * Gets the number of points/instances in the dataset.
   *
   * This corresponds to the number of <i>columns</i> in the matrix.
   *
   * @return the number of points in the dataset
   */
  index_t n_points() const {
    return matrix_.n_cols();
  }
  
  /**
   * Gets the numeric value of a particular feature and point.
   *
   * @param feature the feature index
   * @param point the point index
   */
  double get(index_t feature, index_t point) const {
    return matrix_.get(feature, point);
  }
  
  /**
   * Gets the integer value of a particular feature and point.
   */
  int get_int(index_t feature, index_t point) const {
    double d = get(feature, point);
    int i = int(d);
    DEBUG_ASSERT(d == double(i));
    return i;
  }
  
  /**
   * Modifies a value in the dataset.
   *
   * @param feature the feature number
   * @param point the point index
   * @param d the numeric value to set it to
   */
  void set(index_t feature, index_t point, double d) {
    matrix_.set(feature, point, d);
  }
  
  /**
   * Gets the "raw" form of a particular point.
   *
   * @return a C-like array of the values of a particular point
   */
  const double *point(index_t point) const {
    return matrix_.GetColumnPtr(point);
  }
  /**
   * Gets the "raw" form of a particular point.
   *
   * @return a C-like array of the values of a particular point
   */
  double *point(index_t point) {
    return matrix_.GetColumnPtr(point);
  }
  
  /**
   * Returns the matrix that stores all the data.
   */
  const Matrix& matrix() const {
    return matrix_;
  }
  /**
   * Returns the matrix that stores all the data
   * (can be used for modification).
   */
  Matrix& matrix() {
    return matrix_;
  }
  
  /**
   * Formats as text a particular location of the data set.
   *
   * @param feature the feature number
   * @param point the point index
   * @param result string that will be initialized to the formatted text
   */
  void Format(index_t feature, index_t point, String *result) const {
    info_.feature(feature).Format(get(feature, point), result);
  }
  
  /**
   * Initializer that omits the matrix and info - you must initialize
   * these yourself.
   *
   * (Although this currently does nothing, this is to future-proof your code
   * against possible changes.)
   */
  void InitBlank() {
  }
  
  /**
   * Reads in an ARFF or CSV/WSV file.
   *
   * ARFF LIMITATIONS: Values cannot have spaces or commas, even with quotes;
   * 'string' data type not supported (nominal is supported).
   *
   * @param fname the name of an ARFF, CSV, or whitespace-separated
   */
  success_t InitFromFile(const char *fname);
  
  /**
   * Reads in an ARFF or CSV/WSV file.
   *
   * ARFF LIMITATIONS: Values cannot have spaces or commas, even with quotes;
   * 'string' data type not supported (nominal is supported).
   *
   * @param reader a line reader opened on a CSV or WSV or ARFF file
   */
  success_t InitFromFile(TextLineReader *reader,
      const char *filename = "dataset");
  
  /**
   * Writes to a CSV file.
   *
   * @param fname name of the file
   * @param header whether to include a first line which is the titles of the
   *               data
   */
  success_t WriteCsv(const char *fname, bool header = false) const;

  /**
   * Writes to an ARFF file.
   *
   * @param fname name of the file
   */
  success_t WriteArff(const char *fname) const;

  /**
   * Initializes from a matrix copying all contents, assuming all features
   * are continuous.
   *
   * @param matrix_in data where rows are features, columns are points
   */
  void CopyMatrix(const Matrix& matrix_in) {
    InitBlank();
    matrix_.Copy(matrix_in);
    info_.InitContinuous(matrix_.n_rows());
  }
  
  /**
   * Initializes by becoming the owner of an existing matrix, assuming
   * all features are continuous.
   *
   * By becoming the owner of the matrix, it means that the matrix will be
   * freed when the Dataset falls out of scope.  See Matrix class for
   * details about the Own function.
   *
   * @param matrix_in data where rows are features, columns are points
   */
  void OwnMatrix(Matrix* matrix_in) {
    InitBlank();
    matrix_.Own(matrix_in);
    info_.InitContinuous(matrix_.n_rows());
  }
  
  /**
   * Initializes as an alias or mirror of an existing matrix, assuming
   * all features are continuous.
   *
   * This does not copy the matrix but instead refers to an existing
   * matrix, and changes in one will be reflected in the other.  Make
   * sure the other matrix does not fall out of scope and get freed!
   *
   * @param matrix_in data where rows are features, columns are points
   */
  void AliasMatrix(const Matrix& matrix_in) {
    InitBlank();
    matrix_.Alias(matrix_in);
    info_.InitContinuous(matrix_.n_rows());
  }
  
  //--- Cross-validation features ---

  /**
   * Creates a training and test dataset for k-fold cross validation.
   *
   * The test set will be approximately n_points() / folds, and the
   * training set will be all remaining points.  This takes as an argument
   * a permutation to allow use of consistent random permutations.  If
   * an identity permutation is used, the split will be performed strided.
   *
   * @param folds the number of folds being used
   * @param fold_number the fold number, 0 to folds - 1
   * @param permutation the permutation to use, the same size as n_points()
   *        (use math::MakeIdentiyPermutation or math::MakeRandomPermutation)
   * @param train the training set
   * @param test the test set
   */
  void SplitTrainTest(int folds, int fold_number,
      const ArrayList<index_t>& permutation,
      Dataset *train, Dataset *test) const;
};

/**
 * Miscellaneous dataset-related routines.
 */
namespace data {
  /**
   * Loads a matrix from a file.
   *
   * This supports any type the Dataset class supports with the
   * InitFromFile function: CSV and ARFF.
   *
   * @code
   * Matrix A;
   * data::Load("foo.csv", &A);
   * @endcode
   *
   * @param fname the file name to load
   * @param matrix a pointer to an uninitialized matrix to load
   */
  success_t Load(const char *fname, Matrix *matrix);
  /**
   * Saves a matrix to a file.
   *
   * This saves in CSV format that MATLAB and Excel can handle.
   *
   * @code
   * Matrix matrix_to_save;
   * ... matrix_to_save contains the values you want to save
   * data::Save("mymatrix.csv", matrix_to_save);
   * @endcode
   *
   * @param fname the file name to load
   * @param matrix a pointer to an uninitialized matrix to load
   */
  success_t Save(const char *fname, const Matrix& matrix);
};

#endif