dataFrame.hpp

#ifndef MARIAN_DATAFRAME_HPP
#define MARIAN_DATAFRAME_HPP

#include <string>
#include <vector>
#include <iostream>
#include <map>

namespace marian {
  class DataEntryClerk;

  class DataFrame {
  public:
    DataFrame(){};
    DataFrame(std::string file_name, char delimiter, bool first_row_is_column_names, int primery_key = 0);

    std::string operator()(const int col, const  int row);
    std::string operator()(const std::string col, const int row);
    std::string operator()(const std::string col, const std::string row);

    std::vector<std::string> operator()(int col);
    std::vector<std::string> operator()(std::string col);

    double getNumber(const int col, const  int row);
    double getNumber(const std::string col, const int row);
    double getNumber(const std::string col, const std::string row);

    std::vector<std::string> getColumnNames() const;
    std::vector<std::string> getPrimaryKeys() const;
    std::map<std::string, std::string> getRow(int row) const;
    
    void append(const DataEntryClerk&);
    void append(const DataFrame&);
    
    void print(int n_rows = -1);
    void printToCsv(std::string file_name, char delimiter = ';');
    
    int getNumberOfRows() const;
    int getNumberOfColumns() const;
    
    friend std::ostream& operator<<(std::ostream&, DataFrame&);
  private:
    void parseFile(std::string file_name, char delimiter);
    void createColumnNames(bool first_row);
    
    std::vector<std::vector<std::string> > data_; 
    std::map<std::string,int> column_names_;      
    std::map<std::string,int> primary_key_;       
    int ncols_ = 0;                               
    int nrows_ = 0;                               
  };

  class DataEntryClerk {
  public:

    DataEntryClerk(){};
    void add(std::string, std::string);
    void add(std::string, double);
    void add(std::string, int);

    template<typename T>
    void add(const std::map<std::string, T>& input) {
      for (auto item : input)
    add(item.first, item.second);
    }
    
    friend class DataFrame;
  private:
    std::map<std::string, std::string> data_; 
  };
}  // namespace marian

#endif /* MARIAN_DATAFRAME_HPP */