bsnes/nall/maybe.hpp

118 lines
4.6 KiB
C++

#ifndef NALL_MAYBE_HPP
#define NALL_MAYBE_HPP
namespace nall {
struct nothing_t {};
static nothing_t nothing;
template<typename T, typename = void>
class maybe {
T* value = nullptr;
public:
maybe() {}
maybe(nothing_t) {}
maybe(const T& source) { operator=(source); }
maybe(const maybe& source) { operator=(source); }
maybe(maybe&& source) { operator=(move(source)); }
~maybe() { reset(); }
auto operator=(nothing_t) -> maybe& { reset(); return *this; }
auto operator=(const T& source) -> maybe& { reset(); value = new T(source); return *this; }
auto operator=(const maybe& source) -> maybe& { reset(); if(source) value = new T(source()); return *this; }
auto operator=(maybe&& source) -> maybe& { reset(); value = source.value; source.value = nullptr; return *this; }
auto operator==(const maybe& source) const -> bool {
if(value && source.value) return *value == *source.value;
return !value && !source.value;
}
auto operator!=(const maybe& source) const -> bool { return !operator==(source); }
explicit operator bool() const { return value; }
auto operator->() -> T* { assert(value); return value; }
auto operator->() const -> T* { assert(value); return value; }
auto operator*() -> T& { assert(value); return *value; }
auto operator*() const -> T& { assert(value); return *value; }
auto operator()() -> T& { assert(value); return *value; }
auto operator()() const -> T& { assert(value); return *value; }
auto operator()(const T& invalid) const -> T& { if(value) return *value; return invalid; }
auto empty() const -> bool { return value == nullptr; }
auto reset() -> void { if(value) { delete value; value = nullptr; } }
auto swap(maybe& source) -> void { std::swap(value, source.value); }
};
template<typename T>
class maybe<T&> {
T* value = nullptr;
public:
maybe() {}
maybe(nothing_t) {}
maybe(const T& source) { operator=(source); }
maybe(const maybe& source) { operator=(source); }
maybe(maybe&& source) { operator=(move(source)); }
auto operator=(nothing_t) -> maybe& { value = nullptr; return *this; }
auto operator=(const T& source) -> maybe& { value = (T*)&source; return *this; }
auto operator=(const maybe& source) -> maybe& { value = source.value; return *this; }
auto operator=(maybe&& source) -> maybe& { value = source.value; source.value = nullptr; return *this; }
auto operator==(const maybe& source) const -> bool {
if(value && source.value) return *value == *source.value;
return !value && !source.value;
}
auto operator!=(const maybe& source) const -> bool { return !operator==(source); }
explicit operator bool() const { return value; }
auto operator->() -> T* { assert(value); return value; }
auto operator->() const -> T* { assert(value); return *value; }
auto operator*() -> T& { assert(value); return *value; }
auto operator*() const -> T& { assert(value); return *value; }
auto operator()() -> T& { assert(value); return *value; }
auto operator()() const -> T& { assert(value); return *value; }
auto operator()(const T& invalid) const -> T& { if(value) return *value; return invalid; }
auto empty() const -> bool { return value == nullptr; }
auto reset() -> void { value = nullptr; }
auto swap(maybe& source) -> void { std::swap(value, source.value); }
};
template<typename T>
class maybe<T, enable_if<is_integral<T>>> {
T value = 0;
bool valid = false;
public:
maybe() {}
maybe(nothing_t) {}
maybe(const T& source) { operator=(source); }
maybe(const maybe& source) { operator=(source); }
maybe(maybe&& source) { operator=(move(source)); }
auto operator=(nothing_t) -> maybe& { valid = false; return *this; }
auto operator=(const T& source) -> maybe& { valid = true; value = source; return *this; }
auto operator=(const maybe& source) -> maybe& { valid = source.valid; value = source.value; return *this; }
auto operator=(maybe&& source) -> maybe& { valid = source.valid; value = source.value; source.valid = false; return *this; }
auto operator==(const maybe& source) const -> bool {
if(valid && source.valid) return value == source.value;
return !valid && !source.valid;
}
auto operator!=(const maybe& source) const -> bool { return !operator==(source); }
explicit operator bool() const { return valid; }
auto operator*() const -> T { assert(valid); return value; }
auto operator()() const -> T { assert(valid); return value; }
auto operator()(const T& invalid) const -> T { if(valid) return value; return invalid; }
auto empty() const -> bool { return !valid; }
auto reset() -> void { valid = false; }
auto swap(maybe& source) -> void { std::swap(valid, source.valid); std::swap(value, source.value); }
};
}
#endif