parametertree.hh

Go to the documentation of this file.

// -*- tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
// vi: set et ts=4 sw=2 sts=2:
#ifndef DUNE_PARAMETERTREE_HH
#define DUNE_PARAMETERTREE_HH
 
#include <cstddef>
#include <iostream>
#include <istream>
#include <iterator>
#include <map>
#include <ostream>
#include <sstream>
#include <string>
#include <typeinfo>
#include <vector>
#include <algorithm>
#include <cctype>
 
#include <dune/common/array.hh>
#include <dune/common/exceptions.hh>
#include <dune/common/fvector.hh>
#include <dune/common/classname.hh>
 
namespace Dune {
 
  class ParameterTree
  {
    // class providing a single static parse() function, used by the
    // generic get() method
    template<typename T>
    struct Parser;
 
  public:
 
    typedef std::vector<std::string> KeyVector;
 
    ParameterTree();
 
 
    bool hasKey(const std::string& key) const;
 
 
    bool hasSub(const std::string& sub) const;
 
 
    std::string& operator[] (const std::string& key);
 
 
    const std::string& operator[] (const std::string& key) const;
 
 
    void report(std::ostream& stream = std::cout,
                const std::string& prefix = "") const;
 
 
    ParameterTree& sub(const std::string& sub);
 
 
    const ParameterTree& sub(const std::string& sub) const;
 
 
    std::string get(const std::string& key, const std::string& defaultValue) const;
 
    std::string get(const std::string& key, const char* defaultValue) const;
 
 
    int get(const std::string& key, int defaultValue) const;
 
 
    double get(const std::string& key, double defaultValue) const;
 
 
    template<typename T>
    T get(const std::string& key, const T& defaultValue) const {
      if(hasKey(key))
        return get<T>(key);
      else
        return defaultValue;
    }
 
    template <class T>
    T get(const std::string& key) const {
      if(not hasKey(key))
        DUNE_THROW(RangeError, "Key '" << key << "' not found in parameter "
                   "file!");
      try {
        return Parser<T>::parse((*this)[key]);
      }
      catch(const RangeError&) {
        DUNE_THROW(RangeError, "Cannot parse value \"" <<
                   (*this)[key] << "\" for key \"" << key << "\" as a " <<
                   className<T>());
      }
    }
 
    const KeyVector& getValueKeys() const;
 
 
    const KeyVector& getSubKeys() const;
 
  protected:
    KeyVector valueKeys;
    KeyVector subKeys;
 
    std::map<std::string, std::string> values;
    std::map<std::string, ParameterTree> subs;
    static std::string ltrim(const std::string& s);
    static std::string rtrim(const std::string& s);
    static std::vector<std::string> split(const std::string & s);
 
    // parse into a fixed-size range of iterators
    template<class Iterator>
    static void parseRange(const std::string &str,
                           Iterator it, const Iterator &end)
    {
      typedef typename std::iterator_traits<Iterator>::value_type Value;
      std::istringstream s(str);
      std::size_t n = 0;
      for(; it != end; ++it, ++n) {
        s >> *it;
        if(!s)
          DUNE_THROW(RangeError, "Cannot parse value \"" << str << "\" as a "
                     "range of items of type " << className<Value>() << " "
                     "(" << n << " items were extracted successfully)");
      }
      Value dummy;
      s >> dummy;
      // now extraction should have failed, and eof should be set
      if(not s.fail() or not s.eof())
        DUNE_THROW(RangeError, "Cannot parse value \"" << str << "\" as a "
                   "range of " << n << " items of type "
                                                       << className<Value>() << " (more items than the range "
                   "can hold)");
    }
  };
 
  template<typename T>
  struct ParameterTree::Parser {
    static T parse(const std::string& str) {
      T val;
      std::istringstream s(str);
      s >> val;
      if(!s)
        DUNE_THROW(RangeError, "Cannot parse value \"" << str << "\" as a " <<
                   className<T>());
      T dummy;
      s >> dummy;
      // now extraction should have failed, and eof should be set
      if(not s.fail() or not s.eof())
        DUNE_THROW(RangeError, "Cannot parse value \"" << str << "\" as a " <<
                   className<T>());
      return val;
    }
  };
 
  // "How do I convert a string into a wstring in C++?"  "Why, that very simple
  // son. You just need a these hundred lines of code."
  // Instead im gonna restrict myself to string with charT=char here
  template<typename traits, typename Allocator>
  struct ParameterTree::Parser<std::basic_string<char, traits, Allocator> > {
    static std::basic_string<char, traits, Allocator>
    parse(const std::string& str) {
      std::string trimmed = ltrim(rtrim(str));
      return std::basic_string<char, traits, Allocator>(trimmed.begin(),
                                                        trimmed.end());
    }
  };
 
  template<>
  struct ParameterTree::Parser< bool > {
    struct ToLower {
      int operator()(int c)
      {
        return std::tolower(c);
      }
    };
 
    static bool
    parse(const std::string& str) {
      std::string ret = str;
 
      std::transform(ret.begin(), ret.end(), ret.begin(), ToLower());
 
      if (ret == "yes" || ret == "true")
        return true;
 
      if (ret == "no" || ret == "false")
        return false;
 
      return (Parser<int>::parse(ret) != 0);
    }
  };
 
  template<typename T, int n>
  struct ParameterTree::Parser<FieldVector<T, n> > {
    static FieldVector<T, n>
    parse(const std::string& str) {
      FieldVector<T, n> val;
      parseRange(str, val.begin(), val.end());
      return val;
    }
  };
 
  template<typename T, std::size_t n>
  struct ParameterTree::Parser<array<T, n> > {
    static array<T, n>
    parse(const std::string& str) {
      array<T, n> val;
      parseRange(str, val.begin(), val.end());
      return val;
    }
  };
 
  template<typename T, typename A>
  struct ParameterTree::Parser<std::vector<T, A> > {
    static std::vector<T, A>
    parse(const std::string& str) {
      std::vector<std::string> sub = split(str);
      std::vector<T, A> vec;
      for (unsigned int i=0; i<sub.size(); ++i) {
        T val = ParameterTree::Parser<T>::parse(sub[i]);
        vec.push_back(val);
      }
      return vec;
    }
  };
 
} // end namespace Dune
 
#endif // DUNE_PARAMETERTREE_HH