forked from Bananymous/banan-os
BAN: Variant now has variadic template types
This commit is contained in:
parent
5494e2c125
commit
04ac23b67c
|
@ -72,7 +72,7 @@ namespace BAN
|
||||||
: m_data(move(error))
|
: m_data(move(error))
|
||||||
{}
|
{}
|
||||||
|
|
||||||
bool is_error() const { return m_data.template is<Error>(); }
|
bool is_error() const { return m_data.template has<Error>(); }
|
||||||
const Error& error() const { return m_data.template get<Error>(); }
|
const Error& error() const { return m_data.template get<Error>(); }
|
||||||
Error& error() { return m_data.template get<Error>(); }
|
Error& error() { return m_data.template get<Error>(); }
|
||||||
const T& value() const { return m_data.template get<T>(); }
|
const T& value() const { return m_data.template get<T>(); }
|
||||||
|
|
|
@ -7,141 +7,226 @@
|
||||||
namespace BAN
|
namespace BAN
|
||||||
{
|
{
|
||||||
|
|
||||||
template<typename T1, typename T2>
|
namespace detail
|
||||||
|
{
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
constexpr size_t max_size() { return sizeof(T); }
|
||||||
|
template<typename T0, typename T1, typename... Ts>
|
||||||
|
constexpr size_t max_size() { return sizeof(T0) > sizeof(T1) ? max_size<T0, Ts...>() : max_size<T1, Ts...>(); }
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
constexpr size_t max_align() { return alignof(T); }
|
||||||
|
template<typename T0, typename T1, typename... Ts>
|
||||||
|
constexpr size_t max_align() { return alignof(T0) > alignof(T1) ? max_align<T0, Ts...>() : max_align<T1, Ts...>(); }
|
||||||
|
|
||||||
|
template<typename T, typename T0, typename... Ts>
|
||||||
|
constexpr size_t index()
|
||||||
|
{
|
||||||
|
if constexpr(is_same_v<T, T0>)
|
||||||
|
return 0;
|
||||||
|
else if constexpr(sizeof...(Ts) == 0)
|
||||||
|
return 1;
|
||||||
|
else
|
||||||
|
return index<T, Ts...>() + 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, typename... Ts>
|
||||||
|
void destruct(size_t index, uint8_t* data)
|
||||||
|
{
|
||||||
|
if (index == 0)
|
||||||
|
reinterpret_cast<T*>(data)->~T();
|
||||||
|
else if constexpr(sizeof...(Ts) > 0)
|
||||||
|
destruct<Ts...>(index - 1, data);
|
||||||
|
else
|
||||||
|
ASSERT_NOT_REACHED();
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, typename... Ts>
|
||||||
|
void move_construct(size_t index, uint8_t* source, uint8_t* target)
|
||||||
|
{
|
||||||
|
if (index == 0)
|
||||||
|
new (target) T(move(*reinterpret_cast<T*>(source)));
|
||||||
|
else if constexpr(sizeof...(Ts) > 0)
|
||||||
|
move_construct<Ts...>(index - 1, source, target);
|
||||||
|
else
|
||||||
|
ASSERT_NOT_REACHED();
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, typename... Ts>
|
||||||
|
void copy_construct(size_t index, const uint8_t* source, uint8_t* target)
|
||||||
|
{
|
||||||
|
if (index == 0)
|
||||||
|
new (target) T(*reinterpret_cast<const T*>(source));
|
||||||
|
else if constexpr(sizeof...(Ts) > 0)
|
||||||
|
copy_construct<Ts...>(index - 1, source, target);
|
||||||
|
else
|
||||||
|
ASSERT_NOT_REACHED();
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, typename... Ts>
|
||||||
|
void move_assign(size_t index, uint8_t* source, uint8_t* target)
|
||||||
|
{
|
||||||
|
if (index == 0)
|
||||||
|
*reinterpret_cast<T*>(target) = move(*reinterpret_cast<T*>(source));
|
||||||
|
else if constexpr(sizeof...(Ts) > 0)
|
||||||
|
move_assign<Ts...>(index - 1, source, target);
|
||||||
|
else
|
||||||
|
ASSERT_NOT_REACHED();
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, typename... Ts>
|
||||||
|
void copy_assign(size_t index, const uint8_t* source, uint8_t* target)
|
||||||
|
{
|
||||||
|
if (index == 0)
|
||||||
|
*reinterpret_cast<T*>(target) = *reinterpret_cast<const T*>(source);
|
||||||
|
else if constexpr(sizeof...(Ts) > 0)
|
||||||
|
copy_assign<Ts...>(index - 1, source, target);
|
||||||
|
else
|
||||||
|
ASSERT_NOT_REACHED();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename... Ts>
|
||||||
|
requires (!is_lvalue_reference_v<Ts> && ...)
|
||||||
class Variant
|
class Variant
|
||||||
{
|
{
|
||||||
|
private:
|
||||||
|
template<typename T>
|
||||||
|
static constexpr bool can_have() { return detail::index<T, Ts...>() != invalid_index(); }
|
||||||
|
static constexpr size_t invalid_index() { return sizeof...(Ts); }
|
||||||
|
|
||||||
public:
|
public:
|
||||||
static_assert(!is_same_v<T1, T2>);
|
Variant()
|
||||||
|
: m_index(invalid_index())
|
||||||
|
{ }
|
||||||
|
|
||||||
Variant() = default;
|
Variant(Variant&& other)
|
||||||
|
: m_index(other.m_index)
|
||||||
|
{
|
||||||
|
detail::move_construct<Ts...>(other.m_index, other.m_storage, m_storage);
|
||||||
|
other.clear();
|
||||||
|
}
|
||||||
|
|
||||||
Variant(const T1& value) { set(value); }
|
Variant(const Variant& other)
|
||||||
Variant(T1&& value) { set(move(value)); }
|
: m_index(other.m_index)
|
||||||
Variant(const T2& value) { set(value); }
|
{
|
||||||
Variant(T2&& value) { set(move(value)); }
|
detail::copy_construct<Ts...>(other.m_index, other.m_storage, m_storage);
|
||||||
|
}
|
||||||
|
|
||||||
Variant(const Variant<T1, T2>& other) { *this = other; }
|
template<typename T>
|
||||||
Variant(Variant<T1, T2>&& other) { *this = move(other); }
|
Variant(T&& value) requires (can_have<T>())
|
||||||
|
: m_index(detail::index<T, Ts...>())
|
||||||
|
{
|
||||||
|
new (m_storage) T(move(value));
|
||||||
|
}
|
||||||
|
|
||||||
~Variant() { clear(); }
|
template<typename T>
|
||||||
|
Variant(const T& value) requires (can_have<T>())
|
||||||
|
: m_index(detail::index<T, Ts...>())
|
||||||
|
{
|
||||||
|
new (m_storage) T(value);
|
||||||
|
}
|
||||||
|
|
||||||
Variant<T1, T2>& operator=(const Variant<T1, T2>& other);
|
~Variant()
|
||||||
Variant<T1, T2>& operator=(Variant<T1, T2>&& other);
|
{
|
||||||
|
clear();
|
||||||
|
}
|
||||||
|
|
||||||
template<typename U>
|
Variant& operator=(Variant&& other)
|
||||||
bool is() const;
|
{
|
||||||
|
if (m_index == other.m_index)
|
||||||
|
{
|
||||||
|
detail::move_assign<Ts...>(m_index, other.m_storage, m_storage);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
clear();
|
||||||
|
detail::move_construct<Ts...>(other.m_index, other.m_storage, m_storage);
|
||||||
|
m_index = other.m_index;
|
||||||
|
}
|
||||||
|
other.clear();
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
template<typename U>
|
Variant& operator=(const Variant& other)
|
||||||
void set(U&&);
|
{
|
||||||
template<typename U>
|
if (m_index == other.m_index)
|
||||||
void set(const U& value) { set(move(U(value))); }
|
{
|
||||||
|
detail::copy_assign<Ts...>(m_index, other.m_storage, m_storage);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
clear();
|
||||||
|
detail::copy_construct<Ts...>(other.m_index, other.m_storage, m_storage);
|
||||||
|
m_index = other.m_index;
|
||||||
|
}
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
template<typename U>
|
template<typename T>
|
||||||
const U& get() const;
|
Variant& operator=(T&& value) requires (can_have<T>())
|
||||||
template<typename U>
|
{
|
||||||
U& get();
|
set(move(value));
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
void clear();
|
template<typename T>
|
||||||
|
Variant& operator=(const T& value) requires (can_have<T>())
|
||||||
|
{
|
||||||
|
set(value);
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
bool has() const requires (can_have<T>())
|
||||||
|
{
|
||||||
|
return m_index == detail::index<T, Ts...>();
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
void set(T&& value) requires (can_have<T>())
|
||||||
|
{
|
||||||
|
clear();
|
||||||
|
m_index = detail::index<T, Ts...>();
|
||||||
|
new (m_storage) T(move(value));
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
void set(const T& value) requires (can_have<T>())
|
||||||
|
{
|
||||||
|
clear();
|
||||||
|
m_index = detail::index<T, Ts...>();
|
||||||
|
new (m_storage) T(value);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
T& get() requires (can_have<T>())
|
||||||
|
{
|
||||||
|
ASSERT(has<T>());
|
||||||
|
return (T&)m_storage;
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
const T& get() const requires (can_have<T>())
|
||||||
|
{
|
||||||
|
ASSERT(has<T>());
|
||||||
|
return (const T&)m_storage;
|
||||||
|
}
|
||||||
|
|
||||||
|
void clear()
|
||||||
|
{
|
||||||
|
if (m_index != invalid_index())
|
||||||
|
{
|
||||||
|
detail::destruct<Ts...>(m_index, m_storage);
|
||||||
|
m_index = invalid_index();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
static constexpr uint32_t m_size = Math::max(sizeof(T1), sizeof(T2));
|
alignas(detail::max_align<Ts...>()) uint8_t m_storage[detail::max_size<Ts...>()] {};
|
||||||
alignas(Math::max(alignof(T1), alignof(T2))) uint8_t m_storage[m_size] = {};
|
size_t m_index { invalid_index() };
|
||||||
uint32_t m_index = 0;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
template<typename T1, typename T2>
|
|
||||||
Variant<T1, T2>& Variant<T1, T2>::operator=(const Variant<T1, T2>& other)
|
|
||||||
{
|
|
||||||
clear();
|
|
||||||
if (other.is<T1>())
|
|
||||||
set(other.get<T1>());
|
|
||||||
if (other.is<T2>())
|
|
||||||
set(other.get<T2>());
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
template<typename T1, typename T2>
|
|
||||||
Variant<T1, T2>& Variant<T1, T2>::operator=(Variant<T1, T2>&& other)
|
|
||||||
{
|
|
||||||
clear();
|
|
||||||
if (other.is<T1>())
|
|
||||||
set(move(other.get<T1>()));
|
|
||||||
if (other.is<T2>())
|
|
||||||
set(move(other.get<T2>()));
|
|
||||||
other.clear();
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
template<typename T1, typename T2>
|
|
||||||
template<typename U>
|
|
||||||
bool Variant<T1, T2>::is() const
|
|
||||||
{
|
|
||||||
if constexpr(is_same_v<T1, U>)
|
|
||||||
return m_index == 1;
|
|
||||||
if constexpr(is_same_v<T2, U>)
|
|
||||||
return m_index == 2;
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
template<typename T1, typename T2>
|
|
||||||
template<typename U>
|
|
||||||
void Variant<T1, T2>::set(U&& value)
|
|
||||||
{
|
|
||||||
static_assert(is_same_v<T1, U> || is_same_v<T2, U>);
|
|
||||||
clear();
|
|
||||||
if constexpr(is_same_v<T1, U>)
|
|
||||||
{
|
|
||||||
new (m_storage) T1(move(value));
|
|
||||||
m_index = 1;
|
|
||||||
}
|
|
||||||
if constexpr(is_same_v<T2, U>)
|
|
||||||
{
|
|
||||||
new (m_storage) T2(move(value));
|
|
||||||
m_index = 2;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template<typename T1, typename T2>
|
|
||||||
template<typename U>
|
|
||||||
const U& Variant<T1, T2>::get() const
|
|
||||||
{
|
|
||||||
static_assert(is_same_v<T1, U> || is_same_v<T2, U>);
|
|
||||||
if constexpr(is_same_v<T1, U>)
|
|
||||||
{
|
|
||||||
ASSERT(m_index == 1);
|
|
||||||
return *(T1*)m_storage;
|
|
||||||
}
|
|
||||||
if constexpr(is_same_v<T2, U>)
|
|
||||||
{
|
|
||||||
ASSERT(m_index == 2);
|
|
||||||
return *(T2*)m_storage;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template<typename T1, typename T2>
|
|
||||||
template<typename U>
|
|
||||||
U& Variant<T1, T2>::get()
|
|
||||||
{
|
|
||||||
static_assert(is_same_v<T1, U> || is_same_v<T2, U>);
|
|
||||||
if constexpr(is_same_v<T1, U>)
|
|
||||||
{
|
|
||||||
ASSERT(m_index == 1);
|
|
||||||
return *(T1*)m_storage;
|
|
||||||
}
|
|
||||||
if constexpr(is_same_v<T2, U>)
|
|
||||||
{
|
|
||||||
ASSERT(m_index == 2);
|
|
||||||
return *(T2*)m_storage;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template<typename T1, typename T2>
|
|
||||||
void Variant<T1, T2>::clear()
|
|
||||||
{
|
|
||||||
if (is<T1>()) ((T1*)m_storage)->~T1();
|
|
||||||
if (is<T2>()) ((T2*)m_storage)->~T2();
|
|
||||||
m_index = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
Loading…
Reference in New Issue