//
// File: DataTable.h
// Created by: Julien Dutheil
// Created on: Aug 2005
//

/*
   Copyright or © or Copr. Bio++ Development Team, (November 17, 2004)

   This software is a computer program whose purpose is to provide classes
   for numerical calculus.

   This software is governed by the CeCILL  license under French law and
   abiding by the rules of distribution of free software.  You can  use,
   modify and/ or redistribute the software under the terms of the CeCILL
   license as circulated by CEA, CNRS and INRIA at the following URL
   "http://www.cecill.info".

   As a counterpart to the access to the source code and  rights to copy,
   modify and redistribute granted by the license, users are provided only
   with a limited warranty  and the software's author,  the holder of the
   economic rights,  and the successive licensors  have only  limited
   liability.

   In this respect, the user's attention is drawn to the risks associated
   with loading,  using,  modifying and/or developing or reproducing the
   software by the user in light of its specific status of free software,
   that may mean  that it is complicated to manipulate,  and  that  also
   therefore means  that it is reserved for developers  and  experienced
   professionals having in-depth computer knowledge. Users are therefore
   encouraged to load and test the software's suitability as regards their
   requirements in conditions enabling the security of their systems and/or
   data to be ensured and,  more generally, to use and operate it in the
   same conditions as regards security.

   The fact that you are presently reading this means that you have had
   knowledge of the CeCILL license and that you accept its terms.
 */

#ifndef _DataTable_H_
#define _DataTable_H_

#include "VectorTools.h"
#include "TableExceptions.h"
#include "../Text/TextTools.h"
#include "../Clonable.h"

// From the STL:
#include <string>
#include <vector>
#include <map>

namespace bpp
{
/**
 * @brief This class corresponds to a 'dataset', <i>i.e.</i> a table with data by rows
 * and variable by columns.
 *
 * Data are stored as string objects, by column.
 * A DataTable object is hence similar to a ColMatrix<string>.object.
 * (NB: actually, ColMatrix does not exist yet...)
 */
class DataTable :
  public Clonable
{
protected:
  size_t nRow_, nCol_;
  std::vector< std::vector<std::string> > data_;
  std::vector<std::string>* rowNames_;
  std::vector<std::string>* colNames_;

public:
  /**
   * @brief Build a new void DataTable object with nRow rows and nCol columns.
   *
   * @param nRow The number of rows of the DataTable.
   * @param nCol The number of columns of the DataTable.
   */
  DataTable(size_t nRow, size_t nCol);

  /**
   * @brief Build a new void DataTable object with nRow rows and named columns.
   *
   * @param nRow The number of rows of the DataTable.
   * @param colNames The names of the columns of the DataTable.
   * @throw DuplicatedTableColumnNameException If colnames contains identical names.
   */
  DataTable(size_t nRow, const std::vector<std::string>& colNames);

  /**
   * @brief Build a new void DataTable object with nCol columns.
   *
   * @param nCol The number of columns of the DataTable.
   */
  DataTable(size_t nCol);

  /**
   * @brief Build a new void DataTable object with named columns.
   *
   * @param colNames The names of the columns of the DataTable.
   * @throw DuplicatedTableColumnNameException If colnames contains identical names.
   */
  DataTable(const std::vector<std::string>& colNames);

  DataTable(const DataTable& table);

  DataTable& operator=(const DataTable& table);

  DataTable* clone() const { return new DataTable(*this); }

  virtual ~DataTable();

public:
  /**
   * @return The element at a given position.
   * @param rowIndex Row number.
   * @param colIndex Column number.
   * @throw IndexOutOfBoundsException If one of the index is greater or equal to the corresponding number of columns/rows.
   */
  std::string& operator()(size_t rowIndex, size_t colIndex);

  /**
   * @return The element at a given position.
   * @param rowIndex Row number.
   * @param colIndex Column number.
   * @throw IndexOutOfBoundsException If one of the index is greater or equal to the corresponding number of columns/rows.
   */
  const std::string& operator()(size_t rowIndex, size_t colIndex) const;

  /**
   * @return The element at a given position.
   * @param rowName Row name.
   * @param colName Column name.
   * @throw NoTableRowNamesException If the table does not have names associated to rows.
   * @throw NoTableColumnNamesException If the table does not have names associated to columns.
   * @throw TableNameNotFoundException If one of rowName or colName do not match existing names.
   */
  std::string& operator()(const std::string& rowName, const std::string& colName);

  /**
   * @return The element at a given position.
   * @param rowName Row name.
   * @param colName Column name.
   * @throw NoTableRowNamesException If the table does not have names associated to rows.
   * @throw NoTableColumnNamesException If the table does not have names associated to columns.
   * @throw TableNameNotFoundException If one of rowName or colName do not match existing names.
   */
  const std::string& operator()(const std::string& rowName, const std::string& colName) const;

  /**
   * @return The element at a given position.
   * @param rowName Row name.
   * @param colIndex Column number.
   * @throw NoTableRowNamesException If the table does not have names associated to rows.
   * @throw IndexOutOfBoundsException If the index is greater or equal to the number of columns.
   * @throw TableNameNotFoundException If rowName do not match existing names.
   */
  std::string& operator()(const std::string& rowName, size_t colIndex);

  /**
   * @return The element at a given position.
   * @param rowName Row name.
   * @param colIndex Column number.
   * @throw NoTableRowNamesException If the table does not have names associated to rows.
   * @throw IndexOutOfBoundsException If the index is greater or equal to the number of columns.
   * @throw TableNameNotFoundException If rowName do not match existing names.
   */
  const std::string& operator()(const std::string& rowName, size_t colIndex) const;

  /**
   * @return The element at a given position.
   * @param rowIndex Row number.
   * @param colName Column name.
   * @throw IndexOutOfBoundsException If the index is greater or equal to the number of rows.
   * @throw NoTableColumnNamesException If the table does not have names associated to columns.
   * @throw TableNameNotFoundException If colName do not match existing names.
   */
  std::string& operator()(size_t rowIndex, const std::string& colName);

  /**
   * @return The element at a given position.
   * @param rowIndex Row number.
   * @param colName Column name.
   * @throw IndexOutOfBoundsException If the index is greater or equal to the number of rows.
   * @throw NoTableColumnNamesException If the table does not have names associated to columns.
   * @throw TableNameNotFoundException If colName do not match existing names.
   */
  const std::string& operator()(size_t rowIndex, const std::string& colName) const;

  /**
   * @name Work on columns.
   *
   * @{
   */

  /**
   * @return The number of columns in this table.
   */
  size_t getNumberOfColumns() const { return nCol_; }

  /**
   * @brief Set the column names of this table.
   *
   * @param colNames The row names.
   * @throw DimensionException If the number of names do not match the number of columns in the table.
   * @throw DuplicatedTableColumnNameException If names are not unique.
   */
  void setColumnNames(const std::vector<std::string>& colNames);
  /**
   * @brief Get the column names of this table.
   *
   * @return The column names of this table.
   * @throw NoTableColumnNamesException If no column names are associated to this table.
   */
  std::vector<std::string> getColumnNames() const;
  /**
   * @brief Get a given column name.
   *
   * @param index The index of the column.
   * @return The column name associated to the given column.
   * @throw NoTableColumnNamesException If no column names are associated to this table.
   * @throw IndexOutOfBoundsException If index is >= number of columns.
   */
  std::string getColumnName(size_t index) const;

  /**
   * @return true If column names are associated to this table.
   */
  bool hasColumnNames() const { return colNames_ != 0; }

  /**
   * @return The values in the given column.
   * @param index The index of the column.
   * @throw IndexOutOfBoundsException If index is >= number of columns.
   */
  std::vector<std::string>& getColumn(size_t index);
  /**
   * @return The values in the given column.
   * @param index The index of the column.
   * @throw IndexOutOfBoundsException If index is >= number of columns.
   */
  const std::vector<std::string>& getColumn(size_t index) const;

  /**
   * @return The values in the given column.
   * @param colName The name of the column.
   * @throw NoTableColumnNamesException If no column names are associated to this table.
   * @throw TableColumnNameNotFoundException If colName do not match existing column names.
   */
  std::vector<std::string>& getColumn(const std::string& colName);
  /**
   * @return The values in the given column.
   * @param colName The name of the column.
   * @throw NoTableColumnNamesException If no column names are associated to this table.
   * @throw TableColumnNameNotFoundException If colName do not match existing column names.
   */
  const std::vector<std::string>& getColumn(const std::string& colName) const;

  /**
   * @brief Tell is a given column exists.
   *
   * @param colName The name of the column to look for.
   * @return true if the column was found, false if not or if there are no column names.
   */
  bool hasColumn(const std::string& colName) const;

  /**
   * @brief Delete the given column.
   *
   * @param index The index of the column.
   * @throw IndexOutOfBoundsException If index is >= number of columns.
   */
  void deleteColumn(size_t index);

  /**
   * @brief Delete the given column.
   *
   * @param colName The name of the column.
   * @throw NoTableColumnNamesException If no column names are associated to this table.
   * @throw TableColumnNameNotFoundException If colName do not match existing column names.
   */
  void deleteColumn(const std::string& colName);

  /**
   * @brief Add a new column.
   *
   * @param newColumn The new column values.
   * @throw DimensionException If the number of values does not match the number of rows.
   * @throw TableColumnNamesException If the table has row names.
   */
  void addColumn(const std::vector<std::string>& newColumn);
  /**
   * @brief Add a new column.
   *
   * @param colName   The name of the column.
   * @param newColumn The new column values.
   * @throw DimensionException If the number of values does not match the number of rows.
   * @throw NoTableColumnNamesException If the table does not have row names.
   * @throw DuplicatedTableColumnNameException If colName is already used.
   */
  void addColumn(const std::string& colName, const std::vector<std::string>& newColumn);
  /** @} */

  /**
   * @name Work on rows.
   *
   * @{
   */

  /**
   * @return The number of rows in this table.
   */
  size_t getNumberOfRows() const { return nRow_; }

  /**
   * @brief Set the row names of this table.
   *
   * @param rowNames The row names.
   * @throw DimensionException If the number of names do not match the number of rows in the table.
   * @throw DuplicatedTableRowNameException If names are not unique.
   */
  void setRowNames(const std::vector<std::string>& rowNames);

  /**
   * @brief Set the row name of a row with a given index.
   *
   * @param rowIndex the index of the row
   * @param rowName The row name.
   * @throw DimensionException If the index does not match.
   * @throw DuplicatedTableRowNameException If name is already given.
   */
  
  void setRowName(size_t rowIndex, const std::string& rowName);

  /**
   * @brief Get the row names of this table.
   *
   * @return The row names of this table.
   * @throw NoTableRowNamesException If no row names are associated to this table.
   */
  std::vector<std::string> getRowNames() const;

  /**
   * @brief Tell is a given row exists.
   *
   * @param rowName The name of the row to look for.
   * @return true if the row was found, false if not or if there are no row names.
   */
  bool hasRow(const std::string& rowName) const;

  /**
   * @brief Get a given row name.
   *
   * @param index The index of the row.
   * @return The row name associated to the given row.
   * @throw NoTableRowNamesException If no row names are associated to this table.
   * @throw IndexOutOfBoundsException If index is >= number of rows.
   */
  std::string getRowName(size_t index) const;

  /**
   * @return true If row names are associated to this table.
   */
  bool hasRowNames() const { return rowNames_ != 0; }

  /**
   * @return A vector which contains a copy  in the given row.
   * @param index The index of the row.
   * @throw IndexOutOfBoundsException If index is >= number of rows.
   */
  std::vector<std::string> getRow(size_t index) const;

  /**
   * @return A vector which contains a copy  in the given row.
   * @param rowName The name of the row.
   * @throw NoTableRowNamesException If no row names are associated to this table.
   * @throw TableRowNameNotFoundException If rowName do not match existing row names.
   */
  std::vector<std::string> getRow(const std::string& rowName) const;

  /**
   * @brief Delete the given row.
   *
   * @param index The index of the row.
   * @throw IndexOutOfBoundsException If index is >= number of row.
   */
  void deleteRow(size_t index);

  /**
   * @brief Delete the given row.
   *
   * @param rowName The name of the row.
   * @throw NoTableRowNamesException If no row names are associated to this table.
   * @throw TableRowNameNotFoundException If rowName do not match existing column names.
   */
  void deleteRow(const std::string& rowName);

  /**
   * @brief Add a new row.
   *
   * @param newRow The new row values.
   * @throw DimensionException If the number of values does not match the number of columns.
   * @throw TableRowNamesException If the table has column names.
   */
  void addRow(const std::vector<std::string>& newRow);

  /**
   * @brief Add a new row.
   *
   * @param rowName   The name of the row.
   * @param newRow    The new row values.
   * @throw DimensionException If the number of values does not match the number of columns.
   * @throw NoTableRowNamesException If the table does not have column names.
   * @throw DuplicatedTableRowNameException If rowName is already used.
   */
  
  void addRow(const std::string& rowName, const std::vector<std::string>& newRow);

  /**
   * @brief Sets an existing with a given index.
   *
   * @param rowIndex  The index of the row.
   * @param newRow    The new row values.
   */
  
  void setRow(const size_t rowIndex, const std::vector<std::string>& newRow);


/** @} */

public:
  /**
   * @brief Read a table form a stream in CSV-like format.
   *
   * The number of rows is given by the second line in the file.
   * By default, if the first line as one column less than the second one,
   * the first line is taken as column names, and the first column as row names.
   * Otherwise, no column names and no row names are specified, unless
   * explicitely precised by the user.
   *
   * @param in       The input stream.
   * @param sep      The column delimiter.
   * @param header   Tell if the first line must be used as column names, otherwise use default.
   * @param rowNames Use a column as rowNames. If positive, use the specified column to compute rownames, otherwise use default;
   * @return         A pointer toward a new DataTable object.
   */
  static DataTable* read(std::istream& in, const std::string& sep = "\t", bool header = true, int rowNames = -1);

  /**
   * @brief Write a DataTable object to stream in CVS-like format.
   *
   * @param data         The table to write.
   * @param out          The output stream.
   * @param sep          The column delimiter.
   * @param alignHeaders If true, add a delimiter before the first column header if there is row names.
   */
  static void write(const DataTable& data, std::ostream& out, const std::string& sep = "\t", bool alignHeaders = false);
  static void write(const DataTable& data, bpp::OutputStream& out, const std::string& sep = "\t", bool alignHeaders = false);
};
} // end of namespace bpp.

#endif // _DataTable_H_

