source/core/variant.h

   1 #ifndef MSP_CORE_VARIANT_H_
   2 #define MSP_CORE_VARIANT_H_
   3
   4 #include <stdexcept>
   5 #include <type_traits>
   6 #include <typeinfo>
   7 #include "meta.h"
   8
   9 namespace Msp {
  10
  11 class type_mismatch: public std::runtime_error
  12 {
  13 public:
  14         type_mismatch(const std::type_info &, const std::type_info &);
  15 };
  16
  17
  18 class Variant
  19 {
  20 private:
  21         struct StoreBase
  22         {
  23                 virtual ~StoreBase() { }
  24
  25                 virtual const std::type_info &type_id() const = 0;
  26                 virtual StoreBase *clone() const = 0;
  27                 virtual bool type_equals(const StoreBase &) const = 0;
  28                 virtual bool value_equals(const StoreBase &) const = 0;
  29         };
  30
  31         template<typename T>
  32         struct Store: public StoreBase
  33         {
  34                 T data;
  35
  36                 Store(const T &d): data(d) { }
  37
  38                 const std::type_info &type_id() const override { return typeid(T); }
  39                 StoreBase *clone() const override { return new Store<T>(data); }
  40                 bool type_equals(const StoreBase &s) const override { return dynamic_cast<const Store<T> *>(&s); }
  41                 bool value_equals(const StoreBase &s) const override { return _value_equals<T>(s); }
  42
  43                 template<typename U>
  44                 typename std::enable_if<IsEqualityComparable<U>::value, bool>::type _value_equals(const StoreBase &s) const
  45                 { const Store<T> *t = dynamic_cast<const Store<T> *>(&s); return (t && t->data==data); }
  46
  47                 template<typename U>
  48                 typename std::enable_if<!IsEqualityComparable<U>::value, bool>::type _value_equals(const StoreBase &) const
  49                 { return false; }
  50         };
  51
  52         StoreBase *store = nullptr;
  53
  54 public:
  55         Variant() = default;
  56         template<typename T>
  57         Variant(const T &v): store(new Store<typename std::remove_cv<T>::type>(v)) { }
  58         Variant(const Variant &v): store(v.store ? v.store->clone() : nullptr) { }
  59         ~Variant() { delete store; }
  60
  61         template<typename T>
  62         Variant &operator=(const T &v)
  63         {
  64                 delete store;
  65                 store = new Store<typename std::remove_cv<T>::type>(v);
  66                 return *this;
  67         }
  68
  69         Variant &operator=(const Variant &v)
  70         {
  71                 if(&v==this)
  72                         return *this;
  73
  74                 delete store;
  75                 store = (v.store ? v.store->clone() : nullptr);
  76                 return *this;
  77         }
  78
  79 private:
  80         template<typename T>
  81         Store<typename std::remove_cv<T>::type> *get_typed_store() const
  82         {
  83                 typedef typename std::remove_cv<T>::type NCT;
  84                 Store<NCT> *s = dynamic_cast<Store<NCT> *>(store);
  85                 if(!s)
  86                         throw type_mismatch(typeid(T), (store ? store->type_id() : typeid(void)));
  87                 return s;
  88         }
  89
  90 public:
  91         template<typename T>
  92         T &value()
  93         {
  94                 return get_typed_store<T>()->data;
  95         }
  96
  97         template<typename T>
  98         const T &value() const
  99         {
 100                 return get_typed_store<T>()->data;
 101         }
 102
 103         template<typename T>
 104         bool check_type() const
 105         {
 106                 return dynamic_cast<Store<typename std::remove_cv<T>::type> *>(store);
 107         }
 108
 109         bool check_same_type(const Variant &v) const
 110         { return store && v.store && store->type_equals(*v.store); }
 111
 112         bool operator==(const Variant &v) const
 113         { return store && v.store && store->value_equals(*v.store); }
 114
 115         bool operator!=(const Variant &v) const
 116         { return !(operator==(v)); }
 117
 118         template<typename T>
 119         operator T() const
 120         { return value<T>(); }
 121 };
 122
 123 } // namespace Msp
 124
 125 #endif