485 lines
12 KiB
C++
485 lines
12 KiB
C++
/* Option type implementation. Inspired by libc++.
|
|
*
|
|
* This file is part of OctaSTD. See COPYING.md for futher information.
|
|
*/
|
|
|
|
#ifndef OSTD_MAYBE_HH
|
|
#define OSTD_MAYBE_HH
|
|
|
|
#include "ostd/types.hh"
|
|
#include "ostd/type_traits.hh"
|
|
#include "ostd/memory.hh"
|
|
#include "ostd/utility.hh"
|
|
#include "ostd/initializer_list.hh"
|
|
#include "ostd/functional.hh"
|
|
|
|
namespace ostd {
|
|
|
|
struct InPlace {};
|
|
constexpr InPlace in_place = InPlace();
|
|
|
|
struct Nothing {
|
|
explicit constexpr Nothing(int) {}
|
|
};
|
|
constexpr Nothing nothing = Nothing(0);
|
|
|
|
namespace detail {
|
|
template<typename T, bool = IsTriviallyDestructible<T>>
|
|
class MaybeStorage {
|
|
protected:
|
|
using Value = T;
|
|
union {
|
|
char p_null_state;
|
|
Value p_value;
|
|
};
|
|
bool p_engaged = false;
|
|
|
|
constexpr MaybeStorage(): p_null_state('\0') {}
|
|
|
|
MaybeStorage(MaybeStorage const &v): p_engaged(v.p_engaged) {
|
|
if (p_engaged) {
|
|
::new(address_of(p_value)) Value(v.p_value);
|
|
}
|
|
}
|
|
|
|
MaybeStorage(MaybeStorage &&v): p_engaged(v.p_engaged) {
|
|
if (p_engaged) {
|
|
::new(address_of(p_value)) Value(move(v.p_value));
|
|
}
|
|
}
|
|
|
|
constexpr MaybeStorage(Value const &v): p_value(v), p_engaged(true) {}
|
|
constexpr MaybeStorage(Value &&v): p_value(move(v)), p_engaged(true) {}
|
|
|
|
template<typename ...A>
|
|
constexpr MaybeStorage(InPlace, A &&...args):
|
|
p_value(forward<A>(args)...), p_engaged(true)
|
|
{}
|
|
|
|
~MaybeStorage() {
|
|
if (p_engaged) {
|
|
p_value.~Value();
|
|
}
|
|
}
|
|
};
|
|
|
|
template<typename T>
|
|
class MaybeStorage<T, true> {
|
|
protected:
|
|
using Value = T;
|
|
union {
|
|
char p_null_state;
|
|
Value p_value;
|
|
};
|
|
bool p_engaged = false;
|
|
|
|
constexpr MaybeStorage(): p_null_state('\0') {}
|
|
|
|
MaybeStorage(MaybeStorage const &v): p_engaged(v.p_engaged) {
|
|
if (p_engaged) {
|
|
::new(address_of(p_value)) Value(v.p_value);
|
|
}
|
|
}
|
|
|
|
MaybeStorage(MaybeStorage &&v): p_engaged(v.p_engaged) {
|
|
if (p_engaged) {
|
|
::new(address_of(p_value)) Value(move(v.p_value));
|
|
}
|
|
}
|
|
|
|
constexpr MaybeStorage(Value const &v): p_value(v), p_engaged(true) {}
|
|
constexpr MaybeStorage(Value &&v): p_value(move(v)), p_engaged(true) {}
|
|
|
|
template<typename ...A>
|
|
constexpr MaybeStorage(InPlace, A &&...args):
|
|
p_value(forward<A>(args)...), p_engaged(true)
|
|
{}
|
|
};
|
|
}
|
|
|
|
template<typename T>
|
|
class Maybe: private detail::MaybeStorage<T> {
|
|
using Base = detail::MaybeStorage<T>;
|
|
public:
|
|
using Value = T;
|
|
|
|
static_assert(
|
|
!IsReference<T>,
|
|
"Initialization of Maybe with a reference type is not allowed."
|
|
);
|
|
static_assert(
|
|
!IsSame<RemoveCv<T>, InPlace>,
|
|
"Initialization of Maybe with InPlace is not allowed."
|
|
);
|
|
static_assert(
|
|
!IsSame<RemoveCv<T>, Nothing>,
|
|
"Initialization of Maybe with Nothing is not allowed."
|
|
);
|
|
static_assert(
|
|
IsObject<T>,
|
|
"Initialization of Maybe with non-object type is not allowed."
|
|
);
|
|
static_assert(
|
|
IsDestructible<T>,
|
|
"Initialization of Maybe with a non-destructible object is not allowed."
|
|
);
|
|
|
|
constexpr Maybe() {}
|
|
Maybe(Maybe const &) = default;
|
|
Maybe(Maybe &&) = default;
|
|
constexpr Maybe(Nothing) {}
|
|
constexpr Maybe(Value const &v): Base(v) {}
|
|
constexpr Maybe(Value &&v): Base(move(v)) {}
|
|
|
|
template<typename ...A, typename = EnableIf<IsConstructible<T, A...>>>
|
|
constexpr explicit Maybe(InPlace, A &&...args):
|
|
Base(in_place, forward<A>(args)...)
|
|
{}
|
|
|
|
template<typename U, typename ...A, typename = EnableIf<
|
|
IsConstructible<T, std::initializer_list<U> &, A...>>
|
|
>
|
|
constexpr explicit Maybe(InPlace, std::initializer_list<U> il, A &&...args):
|
|
Base(in_place, il, forward<A>(args)...)
|
|
{}
|
|
|
|
~Maybe() = default;
|
|
|
|
Maybe &operator=(Nothing) {
|
|
if (this->p_engaged) {
|
|
this->p_value.~Value();
|
|
this->p_engaged = false;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
Maybe &operator=(Maybe const &v) {
|
|
if (this->p_engaged == v.p_engaged) {
|
|
if (this->p_engaged) {
|
|
this->p_value = v.p_value;
|
|
}
|
|
} else {
|
|
if (this->p_engaged) {
|
|
this->p_value.~Value();
|
|
} else {
|
|
::new(address_of(this->p_value)) Value(v.p_value);
|
|
}
|
|
this->p_engaged = v.p_engaged;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
Maybe &operator=(Maybe &&v) {
|
|
if (this->p_engaged == v.p_engaged) {
|
|
if (this->p_engaged) {
|
|
this->p_value = move(v.p_value);
|
|
}
|
|
} else {
|
|
if (this->p_engaged) {
|
|
this->p_value.~Value();
|
|
} else {
|
|
::new(address_of(this->p_value)) Value(move(v.p_value));
|
|
}
|
|
this->p_engaged = v.p_engaged;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
template<typename U, typename = EnableIf<
|
|
IsSame<RemoveReference<U>, Value> &&
|
|
IsConstructible<Value, U> && IsAssignable<Value &, U>
|
|
>>
|
|
Maybe &operator=(U &&v) {
|
|
if (this->p_engaged) {
|
|
this->p_value = forward<U>(v);
|
|
} else {
|
|
::new(address_of(this->p_value)) Value(forward<U>(v));
|
|
this->p_engaged = true;
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
template<typename ...A, typename = EnableIf<IsConstructible<Value, A...>>>
|
|
void emplace(A &&...args) {
|
|
*this = nothing;
|
|
::new(address_of(this->p_value)) Value(forward<A>(args)...);
|
|
this->p_engaged = true;
|
|
}
|
|
|
|
template<typename U, typename ...A, typename = EnableIf<
|
|
IsConstructible<Value, std::initializer_list<U> &, A...>
|
|
>>
|
|
void emplace(std::initializer_list<U> il, A &&...args) {
|
|
*this = nothing;
|
|
::new(address_of(this->p_value))
|
|
Value(il, forward<A>(args)...);
|
|
this->p_engaged = true;
|
|
}
|
|
|
|
constexpr Value const *operator->() const {
|
|
return address_of(this->p_value);
|
|
}
|
|
|
|
constexpr Value *operator->() {
|
|
return address_of(this->p_value);
|
|
}
|
|
|
|
constexpr Value const &operator*() const & {
|
|
return this->p_value;
|
|
}
|
|
|
|
constexpr Value &operator*() & {
|
|
return this->p_value;
|
|
}
|
|
|
|
constexpr Value const &&operator*() const && {
|
|
return ostd::move(this->p_value);
|
|
}
|
|
|
|
constexpr Value &&operator*() && {
|
|
return ostd::move(this->p_value);
|
|
}
|
|
|
|
constexpr explicit operator bool() const { return this->p_engaged; }
|
|
|
|
constexpr Value const &value() const & {
|
|
return this->p_value;
|
|
}
|
|
|
|
constexpr Value &value() & {
|
|
return this->p_value;
|
|
}
|
|
|
|
constexpr Value const &&value() const && {
|
|
return ostd::move(this->p_value);
|
|
}
|
|
|
|
constexpr Value &&value() && {
|
|
return ostd::move(this->p_value);
|
|
}
|
|
|
|
template<typename U>
|
|
constexpr Value value_or(U &&v) const & {
|
|
static_assert(
|
|
IsCopyConstructible<Value>,
|
|
"Maybe<T>::value_or: T must be copy constructible"
|
|
);
|
|
static_assert(
|
|
IsConvertible<U, Value>,
|
|
"Maybe<T>::value_or: U must be convertible to T"
|
|
);
|
|
return this->p_engaged ? this->p_value : Value(forward<U>(v));
|
|
}
|
|
|
|
template<typename U>
|
|
constexpr Value value_or(U &&v) && {
|
|
static_assert(
|
|
IsMoveConstructible<Value>,
|
|
"Maybe<T>::value_or: T must be move constructible"
|
|
);
|
|
static_assert(
|
|
IsConvertible<U, Value>,
|
|
"Maybe<T>::value_or: U must be convertible to T"
|
|
);
|
|
return this->p_engaged ? move(this->p_value) : Value(forward<U>(v));
|
|
}
|
|
|
|
void swap(Maybe &v) {
|
|
if (this->p_engaged == v.p_engaged) {
|
|
if (this->p_engaged) {
|
|
detail::swap_adl(this->p_value, v.p_value);
|
|
}
|
|
} else {
|
|
if (this->p_engaged) {
|
|
::new(address_of(v.p_value)) Value(move(this->p_value));
|
|
this->p_value.~Value();
|
|
} else {
|
|
::new(address_of(this->p_value)) Value(move(v.p_value));
|
|
v.p_value.~Value();
|
|
}
|
|
detail::swap_adl(this->p_engaged, v.p_engaged);
|
|
}
|
|
}
|
|
|
|
Size to_hash() const {
|
|
return this->p_engaged ? ostd::ToHash<T>()(this->p_value) : 0;
|
|
}
|
|
};
|
|
|
|
/* maybe vs maybe */
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator==(Maybe<T> const &a, Maybe<T> const &b) {
|
|
return (bool(a) != bool(b)) ? false : (!bool(a) ? true : (*a == *b));
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator!=(Maybe<T> const &a, Maybe<T> const &b) {
|
|
return !(a == b);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<(Maybe<T> const &a, Maybe<T> const &b) {
|
|
return !bool(b) ? false : (!bool(a) ? true : (*a < *b));
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>(Maybe<T> const &a, Maybe<T> const &b) {
|
|
return b < a;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<=(Maybe<T> const &a, Maybe<T> const &b) {
|
|
return !(b < a);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>=(Maybe<T> const &a, Maybe<T> const &b) {
|
|
return !(a < b);
|
|
}
|
|
|
|
/* maybe vs nothing */
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator==(Maybe<T> const &v, Nothing) {
|
|
return !bool(v);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator==(Nothing, Maybe<T> const &v) {
|
|
return !bool(v);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator!=(Maybe<T> const &v, Nothing) {
|
|
return bool(v);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator!=(Nothing, Maybe<T> const &v) {
|
|
return bool(v);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<(Maybe<T> const &, Nothing) {
|
|
return false;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<(Nothing, Maybe<T> const &v) {
|
|
return bool(v);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<=(Maybe<T> const &v, Nothing) {
|
|
return !bool(v);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<=(Nothing, Maybe<T> const &) {
|
|
return true;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>(Maybe<T> const &v, Nothing) {
|
|
return bool(v);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>(Nothing, Maybe<T> const &) {
|
|
return false;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>=(Maybe<T> const &, Nothing) {
|
|
return true;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>=(Nothing, Maybe<T> const &v) {
|
|
return !bool(v);
|
|
}
|
|
|
|
/* maybe vs T */
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator==(Maybe<T> const &a, T const &b) {
|
|
return bool(a) ? (*a == b) : false;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator==(T const &b, Maybe<T> const &a) {
|
|
return bool(a) ? (*a == b) : false;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator!=(Maybe<T> const &a, T const &b) {
|
|
return bool(a) ? !(*a == b) : true;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator!=(T const &b, Maybe<T> const &a) {
|
|
return bool(a) ? !(*a == b) : true;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<(Maybe<T> const &a, T const &b) {
|
|
return bool(a) ? Less<T>()(*a, b) : true;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<(T const &b, Maybe<T> const &a) {
|
|
return bool(a) ? Less<T>()(b, *a) : false;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<=(Maybe<T> const &a, T const &b) {
|
|
return !(a > b);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator<=(T const &b, Maybe<T> const &a) {
|
|
return !(b > a);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>(Maybe<T> const &a, T const &b) {
|
|
return bool(a) ? (b < a) : true;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>(T const &b, Maybe<T> const &a) {
|
|
return bool(a) ? (a < b) : true;
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>=(Maybe<T> const &a, T const &b) {
|
|
return !(a < b);
|
|
}
|
|
|
|
template<typename T>
|
|
inline constexpr bool operator>=(T const &b, Maybe<T> const &a) {
|
|
return !(b < a);
|
|
}
|
|
|
|
/* make maybe */
|
|
|
|
template<typename T>
|
|
inline constexpr Maybe<Decay<T>> make_maybe(T &&v) {
|
|
return Maybe<Decay<T>>(forward<T>(v));
|
|
}
|
|
|
|
template<typename T, typename ...A>
|
|
inline constexpr Maybe<T> make_maybe(A &&...args) {
|
|
return Maybe<T>(in_place, forward<A>(args)...);
|
|
}
|
|
|
|
template<typename T, typename U, typename ...A>
|
|
inline constexpr Maybe<T> make_maybe(std::initializer_list<U> il, A &&...args) {
|
|
return Maybe<T>(in_place, il, forward<A>(args)...);
|
|
}
|
|
|
|
} /* namespace ostd */
|
|
|
|
#endif
|