include/xbt/config.hpp

   1 /* Copyright (c) 2016-2020. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #ifndef XBT_CONFIG_HPP
   7 #define XBT_CONFIG_HPP
   8
   9 #include <xbt/base.h>
  10
  11 #include <cstdlib>
  12
  13 #include <functional>
  14 #include <initializer_list>
  15 #include <map>
  16 #include <stdexcept>
  17 #include <string>
  18 #include <type_traits>
  19 #include <utility>
  20
  21 #include <xbt/base.h>
  22 #include <xbt/sysdep.h>
  23
  24 namespace simgrid {
  25 namespace config {
  26
  27 class Config;
  28
  29 template<class T> inline
  30 std::string to_string(T&& value)
  31 {
  32   return std::to_string(std::forward<T>(value));
  33 }
  34 inline std::string const& to_string(std::string& value)
  35 {
  36   return value;
  37 }
  38 inline std::string const& to_string(std::string const& value)
  39 {
  40   return value;
  41 }
  42 inline std::string to_string(std::string&& value)
  43 {
  44   return std::move(value);
  45 }
  46
  47 // Set default
  48
  49 template <class T> XBT_PUBLIC void set_default(const char* name, T value);
  50
  51 extern template XBT_PUBLIC void set_default<int>(const char* name, int value);
  52 extern template XBT_PUBLIC void set_default<double>(const char* name, double value);
  53 extern template XBT_PUBLIC void set_default<bool>(const char* name, bool value);
  54 extern template XBT_PUBLIC void set_default<std::string>(const char* name, std::string value);
  55
  56 XBT_PUBLIC bool is_default(const char* name);
  57
  58 // Set config
  59
  60 template <class T> XBT_PUBLIC void set_value(const char* name, T value);
  61
  62 extern template XBT_PUBLIC void set_value<int>(const char* name, int value);
  63 extern template XBT_PUBLIC void set_value<double>(const char* name, double value);
  64 extern template XBT_PUBLIC void set_value<bool>(const char* name, bool value);
  65 extern template XBT_PUBLIC void set_value<std::string>(const char* name, std::string value);
  66
  67 XBT_PUBLIC void set_as_string(const char* name, const std::string& value);
  68 XBT_PUBLIC void set_parse(const std::string& options);
  69
  70 // Get config
  71
  72 template <class T> XBT_PUBLIC T const& get_value(const std::string& name);
  73
  74 extern template XBT_PUBLIC int const& get_value<int>(const std::string& name);
  75 extern template XBT_PUBLIC double const& get_value<double>(const std::string& name);
  76 extern template XBT_PUBLIC bool const& get_value<bool>(const std::string& name);
  77 extern template XBT_PUBLIC std::string const& get_value<std::string>(const std::string& name);
  78
  79 // Register:
  80
  81 /** Register a configuration flag
  82  *
  83  *  @param name        name of the option
  84  *  @param description Description of the option
  85  *  @param value       Initial/default value
  86  *  @param callback    called with the option value
  87  */
  88 template <class T>
  89 XBT_PUBLIC void declare_flag(const std::string& name, const std::string& description, T value,
  90                              std::function<void(const T&)> callback = std::function<void(const T&)>());
  91
  92 extern template XBT_PUBLIC void declare_flag(const std::string& name, const std::string& description, int value,
  93                                              std::function<void(int const&)> callback);
  94 extern template XBT_PUBLIC void declare_flag(const std::string& name, const std::string& description, double value,
  95                                              std::function<void(double const&)> callback);
  96 extern template XBT_PUBLIC void declare_flag(const std::string& name, const std::string& description, bool value,
  97                                              std::function<void(bool const&)> callback);
  98 extern template XBT_PUBLIC void declare_flag(const std::string& name, const std::string& description, std::string value,
  99                                              std::function<void(std::string const&)> callback);
 100
 101 // ***** alias *****
 102
 103 XBT_PUBLIC void alias(const char* realname, std::initializer_list<const char*> aliases);
 104
 105 /** Bind a variable to configuration flag
 106  *
 107  *  @param value Bound variable
 108  *  @param name  Flag name
 109  *  @param description Option description
 110  */
 111 template <class T> void bind_flag(T& value, const char* name, const char* description)
 112 {
 113   declare_flag<T>(name, description, value, [&value](T const& val) { value = val; });
 114 }
 115
 116 template <class T>
 117 void bind_flag(T& value, const char* name, std::initializer_list<const char*> aliases, const char* description)
 118 {
 119   bind_flag(value, name, description);
 120   alias(name, aliases);
 121 }
 122
 123 /** Bind a variable to configuration flag
 124  *
 125  *  <pre><code>
 126  *  static int x;
 127  *  simgrid::config::bind_flag(a, "x", [](int x) {
 128  *    if (x < x_min || x => x_max)
 129  *      throw std::range_error("must be in [x_min, x_max)")
 130  *  });
 131  *  </code></pre>
 132  */
 133 // F is a checker, F : T& -> ()
 134 template <class T, class F>
 135 typename std::enable_if<std::is_same<void, decltype(std::declval<F>()(std::declval<const T&>()))>::value, void>::type
 136 bind_flag(T& value, const char* name, const char* description, F callback)
 137 {
 138   declare_flag(name, description, value, std::function<void(const T&)>([&value, callback](const T& val) {
 139                  callback(val);
 140                  value = std::move(val);
 141                }));
 142 }
 143
 144 template <class T, class F>
 145 typename std::enable_if<std::is_same<void, decltype(std::declval<F>()(std::declval<const T&>()))>::value, void>::type
 146 bind_flag(T& value, const char* name, std::initializer_list<const char*> aliases, const char* description, F callback)
 147 {
 148   bind_flag(value, name, description, std::move(callback));
 149   alias(name, aliases);
 150 }
 151
 152 template <class T, class F>
 153 typename std::enable_if<std::is_same<void, decltype(std::declval<F>()(std::declval<const T&>()))>::value, void>::type
 154 bind_flag(T& value, const char* name, const char* description, const std::map<T, std::string>& valid_values, F callback)
 155 {
 156   declare_flag(name, description, value,
 157                std::function<void(const T&)>([&value, name, valid_values, callback](const T& val) {
 158                  callback(val);
 159                  if (valid_values.find(val) != valid_values.end()) {
 160                    value = std::move(val);
 161                    return;
 162                  }
 163                  std::string mesg = "\n";
 164                  if (std::string(val) == "help")
 165                    mesg += std::string("Possible values for option ") + name + ":\n";
 166                  else
 167                    mesg += std::string("Invalid value '") + val + "' for option " + name + ". Possible values:\n";
 168                  for (auto const& kv : valid_values)
 169                    mesg += "  - '" + std::string(kv.first) + "': " + kv.second +
 170                            (kv.first == value ? "  <=== DEFAULT" : "") + "\n";
 171                  xbt_die("%s", mesg.c_str());
 172                }));
 173 }
 174 template <class T, class F>
 175 typename std::enable_if<std::is_same<void, decltype(std::declval<F>()(std::declval<const T&>()))>::value, void>::type
 176 bind_flag(T& value, const char* name, std::initializer_list<const char*> aliases, const char* description,
 177           const std::map<T, std::string>& valid_values, F callback)
 178 {
 179   bind_flag(value, name, description, valid_values, std::move(callback));
 180   alias(name, aliases);
 181 }
 182
 183 /** Bind a variable to configuration flag
 184  *
 185  *  <pre><code>
 186  *  static int x;
 187  *  simgrid::config::bind_flag(a, "x", [](int x) { return x > 0; });
 188  *  </code></pre>
 189  */
 190 // F is a predicate, F : T const& -> bool
 191 template <class T, class F>
 192 typename std::enable_if<std::is_same<bool, decltype(std::declval<F>()(std::declval<const T&>()))>::value, void>::type
 193 bind_flag(T& value, const char* name, const char* description, F callback)
 194 {
 195   declare_flag(name, description, value, std::function<void(const T&)>([&value, callback](const T& val) {
 196                  if (not callback(val))
 197                    throw std::range_error("invalid value.");
 198                  value = std::move(val);
 199                }));
 200 }
 201
 202 /** A variable bound to a CLI option
 203  *
 204  *  <pre><code>
 205  *  static simgrid::config::flag<int> answer("answer", "Expected answer", 42);
 206  *  static simgrid::config::flag<std::string> name("name", "Ford Perfect", "John Doe");
 207  *  static simgrid::config::flag<double> gamma("gamma", "Gamma factor", 1.987);
 208  *  </code></pre>
 209  */
 210 template<class T>
 211 class Flag {
 212   T value_;
 213   std::string name_;
 214
 215 public:
 216   /** Constructor
 217    *
 218    *  @param name  Flag name
 219    *  @param desc  Flag description
 220    *  @param value Flag initial/default value
 221    */
 222   Flag(const char* name, const char* desc, T value) : value_(value), name_(name)
 223   {
 224     simgrid::config::bind_flag(value_, name, desc);
 225   }
 226
 227   /** Constructor taking also an array of aliases for name */
 228   Flag(const char* name, std::initializer_list<const char*> aliases, const char* desc, T value)
 229       : value_(value), name_(name)
 230   {
 231     simgrid::config::bind_flag(value_, name, aliases, desc);
 232   }
 233
 234   /* A constructor accepting a callback that will be passed the parameter.
 235    * It can either return a boolean (informing whether the parameter is valid), or returning void.
 236    */
 237   template <class F> Flag(const char* name, const char* desc, T value, F callback) : value_(value), name_(name)
 238   {
 239     simgrid::config::bind_flag(value_, name, desc, std::move(callback));
 240   }
 241
 242   template <class F>
 243   Flag(const char* name, std::initializer_list<const char*> aliases, const char* desc, T value, F callback)
 244       : value_(value), name_(name)
 245   {
 246     simgrid::config::bind_flag(value_, name, aliases, desc, std::move(callback));
 247   }
 248
 249   /* A constructor accepting a map of valid values -> their description,
 250    * and producing an informative error message when an invalid value is passed, or when help is passed as a value.
 251    */
 252   template <class F>
 253   Flag(const char* name, const char* desc, T value, const std::map<T, std::string>& valid_values, F callback)
 254       : value_(value), name_(name)
 255   {
 256     simgrid::config::bind_flag(value_, name, desc, std::move(valid_values), std::move(callback));
 257   }
 258
 259   /* A constructor with everything */
 260   template <class F>
 261   Flag(const char* name, std::initializer_list<const char*> aliases, const char* desc, T value,
 262        const std::map<T, std::string>& valid_values, F callback)
 263       : value_(value), name_(name)
 264   {
 265     simgrid::config::bind_flag(value_, name, aliases, desc, valid_values, std::move(callback));
 266   }
 267
 268   // No copy:
 269   Flag(Flag const&) = delete;
 270   Flag& operator=(Flag const&) = delete;
 271
 272   // Get the underlying value:
 273   T& get() { return value_; }
 274   T const& get() const { return value_; }
 275
 276   const std::string& get_name() const { return name_; }
 277   // Implicit conversion to the underlying type:
 278   operator T&() { return value_; }
 279   operator T const&() const{ return value_; }
 280
 281   // Basic interop with T:
 282   template<class U>
 283   Flag& operator=(U const& that) { value_ = that; return *this; }
 284   template<class U>
 285   Flag& operator=(U&& that) { value_ = std::forward<U>(that); return *this; }
 286   template<class U>
 287   bool operator==(U const& that) const { return value_ == that; }
 288   template<class U>
 289   bool operator!=(U const& that) const { return value_ != that; }
 290   template<class U>
 291   bool operator<(U const& that) const { return value_ < that; }
 292   template<class U>
 293   bool operator>(U const& that) const { return value_ > that; }
 294   template<class U>
 295   bool operator<=(U const& that) const { return value_ <= that; }
 296   template<class U>
 297   bool operator>=(U const& that) const { return value_ >= that; }
 298 };
 299
 300 XBT_PUBLIC void finalize();
 301 XBT_PUBLIC void show_aliases();
 302 XBT_PUBLIC void help();
 303 } // namespace config
 304 } // namespace simgrid
 305
 306 #endif