cannam@147: // Copyright (c) 2013-2014 Sandstorm Development Group, Inc. and contributors
cannam@147: // Licensed under the MIT License:
cannam@147: //
cannam@147: // Permission is hereby granted, free of charge, to any person obtaining a copy
cannam@147: // of this software and associated documentation files (the "Software"), to deal
cannam@147: // in the Software without restriction, including without limitation the rights
cannam@147: // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
cannam@147: // copies of the Software, and to permit persons to whom the Software is
cannam@147: // furnished to do so, subject to the following conditions:
cannam@147: //
cannam@147: // The above copyright notice and this permission notice shall be included in
cannam@147: // all copies or substantial portions of the Software.
cannam@147: //
cannam@147: // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
cannam@147: // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
cannam@147: // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
cannam@147: // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
cannam@147: // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
cannam@147: // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
cannam@147: // THE SOFTWARE.
cannam@147: 
cannam@147: #ifndef KJ_ONE_OF_H_
cannam@147: #define KJ_ONE_OF_H_
cannam@147: 
cannam@147: #if defined(__GNUC__) && !KJ_HEADER_WARNINGS
cannam@147: #pragma GCC system_header
cannam@147: #endif
cannam@147: 
cannam@147: #include "common.h"
cannam@147: 
cannam@147: namespace kj {
cannam@147: 
cannam@147: namespace _ {  // private
cannam@147: 
cannam@147: template <uint i, typename Key, typename First, typename... Rest>
cannam@147: struct TypeIndex_ { static constexpr uint value = TypeIndex_<i + 1, Key, Rest...>::value; };
cannam@147: template <uint i, typename Key, typename... Rest>
cannam@147: struct TypeIndex_<i, Key, Key, Rest...> { static constexpr uint value = i; };
cannam@147: 
cannam@147: }  // namespace _ (private)
cannam@147: 
cannam@147: template <typename... Variants>
cannam@147: class OneOf {
cannam@147:   template <typename Key>
cannam@147:   static inline constexpr uint typeIndex() { return _::TypeIndex_<1, Key, Variants...>::value; }
cannam@147:   // Get the 1-based index of Key within the type list Types.
cannam@147: 
cannam@147: public:
cannam@147:   inline OneOf(): tag(0) {}
cannam@147:   OneOf(const OneOf& other) { copyFrom(other); }
cannam@147:   OneOf(OneOf&& other) { moveFrom(other); }
cannam@147:   ~OneOf() { destroy(); }
cannam@147: 
cannam@147:   OneOf& operator=(const OneOf& other) { if (tag != 0) destroy(); copyFrom(other); return *this; }
cannam@147:   OneOf& operator=(OneOf&& other) { if (tag != 0) destroy(); moveFrom(other); return *this; }
cannam@147: 
cannam@147:   inline bool operator==(decltype(nullptr)) const { return tag == 0; }
cannam@147:   inline bool operator!=(decltype(nullptr)) const { return tag != 0; }
cannam@147: 
cannam@147:   template <typename T>
cannam@147:   bool is() const {
cannam@147:     return tag == typeIndex<T>();
cannam@147:   }
cannam@147: 
cannam@147:   template <typename T>
cannam@147:   T& get() {
cannam@147:     KJ_IREQUIRE(is<T>(), "Must check OneOf::is<T>() before calling get<T>().");
cannam@147:     return *reinterpret_cast<T*>(space);
cannam@147:   }
cannam@147:   template <typename T>
cannam@147:   const T& get() const {
cannam@147:     KJ_IREQUIRE(is<T>(), "Must check OneOf::is<T>() before calling get<T>().");
cannam@147:     return *reinterpret_cast<const T*>(space);
cannam@147:   }
cannam@147: 
cannam@147:   template <typename T, typename... Params>
cannam@147:   T& init(Params&&... params) {
cannam@147:     if (tag != 0) destroy();
cannam@147:     ctor(*reinterpret_cast<T*>(space), kj::fwd<Params>(params)...);
cannam@147:     tag = typeIndex<T>();
cannam@147:     return *reinterpret_cast<T*>(space);
cannam@147:   }
cannam@147: 
cannam@147: private:
cannam@147:   uint tag;
cannam@147: 
cannam@147:   static inline constexpr size_t maxSize(size_t a) {
cannam@147:     return a;
cannam@147:   }
cannam@147:   template <typename... Rest>
cannam@147:   static inline constexpr size_t maxSize(size_t a, size_t b, Rest... rest) {
cannam@147:     return maxSize(kj::max(a, b), rest...);
cannam@147:   }
cannam@147:   // Returns the maximum of all the parameters.
cannam@147:   // TODO(someday):  Generalize the above template and make it common.  I tried, but C++ decided to
cannam@147:   //   be difficult so I cut my losses.
cannam@147: 
cannam@147:   static constexpr auto spaceSize = maxSize(sizeof(Variants)...);
cannam@147:   // TODO(msvc):  This constant could just as well go directly inside space's bracket's, where it's
cannam@147:   // used, but MSVC suffers a parse error on `...`.
cannam@147: 
cannam@147:   union {
cannam@147:     byte space[spaceSize];
cannam@147: 
cannam@147:     void* forceAligned;
cannam@147:     // TODO(someday):  Use C++11 alignas() once we require GCC 4.8 / Clang 3.3.
cannam@147:   };
cannam@147: 
cannam@147:   template <typename... T>
cannam@147:   inline void doAll(T... t) {}
cannam@147: 
cannam@147:   template <typename T>
cannam@147:   inline bool destroyVariant() {
cannam@147:     if (tag == typeIndex<T>()) {
cannam@147:       tag = 0;
cannam@147:       dtor(*reinterpret_cast<T*>(space));
cannam@147:     }
cannam@147:     return false;
cannam@147:   }
cannam@147:   void destroy() {
cannam@147:     doAll(destroyVariant<Variants>()...);
cannam@147:   }
cannam@147: 
cannam@147:   template <typename T>
cannam@147:   inline bool copyVariantFrom(const OneOf& other) {
cannam@147:     if (other.is<T>()) {
cannam@147:       ctor(*reinterpret_cast<T*>(space), other.get<T>());
cannam@147:     }
cannam@147:     return false;
cannam@147:   }
cannam@147:   void copyFrom(const OneOf& other) {
cannam@147:     // Initialize as a copy of `other`.  Expects that `this` starts out uninitialized, so the tag
cannam@147:     // is invalid.
cannam@147:     tag = other.tag;
cannam@147:     doAll(copyVariantFrom<Variants>(other)...);
cannam@147:   }
cannam@147: 
cannam@147:   template <typename T>
cannam@147:   inline bool moveVariantFrom(OneOf& other) {
cannam@147:     if (other.is<T>()) {
cannam@147:       ctor(*reinterpret_cast<T*>(space), kj::mv(other.get<T>()));
cannam@147:     }
cannam@147:     return false;
cannam@147:   }
cannam@147:   void moveFrom(OneOf& other) {
cannam@147:     // Initialize as a copy of `other`.  Expects that `this` starts out uninitialized, so the tag
cannam@147:     // is invalid.
cannam@147:     tag = other.tag;
cannam@147:     doAll(moveVariantFrom<Variants>(other)...);
cannam@147:   }
cannam@147: };
cannam@147: 
cannam@147: }  // namespace kj
cannam@147: 
cannam@147: #endif  // KJ_ONE_OF_H_