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 {
    friend class Config;
   public:
    using iterator_category = std::input_iterator_tag;
    using value_type = ConfigEntry;
    using difference_type = std::ptrdiff_t;
    using pointer = ConfigEntry*;
    using reference = ConfigEntry&;
    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_