GeneralCSVLoader.h

 
#ifndef CONDITIONS_GENERALCSVLOADER_H_
#define CONDITIONS_GENERALCSVLOADER_H_
 
#include <istream>
#include <string>
#include <vector>
 
namespace conditions {
 
class GeneralCSVLoader {
 public:
  virtual ~GeneralCSVLoader() {}
 
  std::vector<std::string> columnNames() const { return colNames_; }
 
  bool nextRow();
 
  const std::string& get(const std::string& colname,
                         bool ignore_case = true) const;
 
  int getInteger(const std::string& colname, bool ignore_case = true) const;
 
 protected:
  GeneralCSVLoader() {}
 
  virtual std::string getNextLine() = 0;
 
 private:
  std::vector<std::string> colNames_;
  std::vector<std::string> rowData_;
};
 
class StringCSVLoader : public GeneralCSVLoader {
 public:
  StringCSVLoader(const std::string& source,
                  const std::string lineseparators = "\n");
 
 protected:
  virtual std::string getNextLine();
 
 private:
  const std::string& source_;
  const std::string linesep_;
  std::string::size_type rowBegin_, rowEnd_;
};
 
class StreamCSVLoader : public GeneralCSVLoader {
 public:
  StreamCSVLoader(const std::string& filename);
  StreamCSVLoader(std::istream& stream);
 
  virtual ~StreamCSVLoader();
 
 protected:
  virtual std::string getNextLine();
 
 private:
  std::istream* source_;
  bool ownStream_;
  std::string line_;
};
 
}  // namespace conditions
#endif  // CONDITIONS_GENERALCSVLOADER_H_