config.h

Go to the documentation of this file.

#ifndef DMLC_CONFIG_H_
#define DMLC_CONFIG_H_

#include <cstring>
#include <iostream>
#include <iterator>
#include <map>
#include <vector>
#include <utility>
#include <string>
#include <sstream>

namespace dmlc {

class Config {
 public:
  typedef std::pair<std::string, std::string> ConfigEntry;

  class ConfigIterator;

  explicit Config(bool multi_value = false);
  explicit Config(std::istream& is, bool multi_value = false);  // NOLINT(*)
  void Clear(void);
  void LoadFromStream(std::istream& is);  // NOLINT(*)
  template<class T>
  void SetParam(const std::string& key, const T& value, bool is_string = false);

  const std::string& GetParam(const std::string& key) const;

  bool IsGenuineString(const std::string& key) const;

  std::string ToProtoString(void) const;

  ConfigIterator begin() const;

  ConfigIterator end() const;

 public:
  class ConfigIterator : public std::iterator< std::input_iterator_tag, ConfigEntry > {
    friend class Config;
   public:
    ConfigIterator(const ConfigIterator& other);
    ConfigIterator& operator++();
    ConfigIterator operator++(int);  // NOLINT(*)
    bool operator == (const ConfigIterator& rhs) const;
    bool operator != (const ConfigIterator& rhs) const;
    ConfigEntry operator * () const;

   private:
    ConfigIterator(size_t index, const Config* config);
    void FindNextIndex();

   private:
    size_t index_;
    const Config* config_;
  };

 private:
  struct ConfigValue {
    std::vector<std::string> val;
    std::vector<size_t> insert_index;
    bool is_string;
  };
  void Insert(const std::string& key, const std::string& value, bool is_string);

 private:
  std::map<std::string, ConfigValue> config_map_;
  std::vector<std::pair<std::string, size_t> > order_;
  const bool multi_value_;
};

template<class T>
void Config::SetParam(const std::string& key, const T& value, bool is_string) {
  std::ostringstream oss;
  oss << value;
  Insert(key, oss.str(), is_string);
}

}  // namespace dmlc

#endif  // DMLC_CONFIG_H_