variant.hpp

Go to the documentation of this file.

// SPDX-License-Identifier: BSL-1.0
 
#ifndef TETL_VARIANT_VARIANT2_HPP
#define TETL_VARIANT_VARIANT2_HPP
 
#include <etl/_config/all.hpp>
 
#include <etl/_contracts/check.hpp>
#include <etl/_cstddef/size_t.hpp>
#include <etl/_functional/equal_to.hpp>
#include <etl/_functional/greater.hpp>
#include <etl/_functional/greater_equal.hpp>
#include <etl/_functional/less.hpp>
#include <etl/_functional/less_equal.hpp>
#include <etl/_memory/addressof.hpp>
#include <etl/_memory/construct_at.hpp>
#include <etl/_memory/destroy_at.hpp>
#include <etl/_meta/at.hpp>
#include <etl/_meta/count.hpp>
#include <etl/_meta/index_of.hpp>
#include <etl/_type_traits/add_pointer.hpp>
#include <etl/_type_traits/index_constant.hpp>
#include <etl/_type_traits/is_assignable.hpp>
#include <etl/_type_traits/is_constructible.hpp>
#include <etl/_type_traits/is_copy_assignable.hpp>
#include <etl/_type_traits/is_copy_constructible.hpp>
#include <etl/_type_traits/is_default_constructible.hpp>
#include <etl/_type_traits/is_move_assignable.hpp>
#include <etl/_type_traits/is_move_constructible.hpp>
#include <etl/_type_traits/is_nothrow_assignable.hpp>
#include <etl/_type_traits/is_nothrow_constructible.hpp>
#include <etl/_type_traits/is_nothrow_copy_assignable.hpp>
#include <etl/_type_traits/is_nothrow_copy_constructible.hpp>
#include <etl/_type_traits/is_nothrow_default_constructible.hpp>
#include <etl/_type_traits/is_nothrow_move_assignable.hpp>
#include <etl/_type_traits/is_nothrow_move_constructible.hpp>
#include <etl/_type_traits/is_same.hpp>
#include <etl/_type_traits/is_trivially_copy_assignable.hpp>
#include <etl/_type_traits/is_trivially_copy_constructible.hpp>
#include <etl/_type_traits/is_trivially_move_assignable.hpp>
#include <etl/_type_traits/is_trivially_move_constructible.hpp>
#include <etl/_type_traits/remove_cvref.hpp>
#include <etl/_type_traits/smallest_size_t.hpp>
#include <etl/_utility/forward.hpp>
#include <etl/_utility/in_place_index.hpp>
#include <etl/_utility/in_place_type.hpp>
#include <etl/_utility/unreachable.hpp>
#include <etl/_variant/variadic_union.hpp>
#include <etl/_variant/variant_alternative.hpp>
#include <etl/_variant/variant_alternative_selector.hpp>
#include <etl/_variant/variant_fwd.hpp>
#include <etl/_variant/variant_size.hpp>
#include <etl/_variant/visit.hpp>
 
namespace etl {
 
namespace detail {
 
template <typename T>
concept variant_copy_assignable = is_copy_constructible_v<T> and is_copy_assignable_v<T>;
 
template <typename T>
concept variant_trivially_copy_assignable = is_trivially_copy_constructible_v<T> and is_trivially_copy_assignable_v<T>;
 
template <typename T>
concept variant_move_assignable = is_move_constructible_v<T> and is_move_assignable_v<T>;
 
template <typename T>
concept variant_trivially_move_assignable = is_trivially_move_constructible_v<T> and is_trivially_move_assignable_v<T>;
 
template <typename T>
inline constexpr auto is_in_place_index = false;
 
template <size_t I>
inline constexpr auto is_in_place_index<in_place_index_t<I>> = true;
 
template <typename T>
inline constexpr auto is_in_place_type = false;
 
template <typename T>
inline constexpr auto is_in_place_type<in_place_type_t<T>> = true;
 
constexpr auto make_variant_compare_op(auto op)
{
    return [op](auto const& l, auto const& r) -> bool {
        if constexpr (etl::is_same_v<decltype(l), decltype(r)>) {
            return op(l, r);
        } else {
            etl::unreachable();
        }
    };
}
 
} // namespace detail

template <typename... Ts>
struct TETL_TRIVIAL_ABI variant {
private:
    // Avoid valueless_by_exception
    static_assert((is_nothrow_move_constructible_v<Ts> and ...));
 
    using index_type = smallest_size_t<sizeof...(Ts)>;
    using first_type = meta::at_t<0, meta::list<Ts...>>;
 
public:
    constexpr variant() noexcept(is_nothrow_default_constructible_v<first_type>)
        requires(is_default_constructible_v<first_type>)
        : variant(in_place_index<0>)
    {
    }
 
    // clang-format off
    template <typename T>
        requires (
                (sizeof...(Ts) > 0)
            and not is_same_v<remove_cvref_t<T>, variant>
            and not detail::is_in_place_index<remove_cvref_t<T>>
            and not detail::is_in_place_type<remove_cvref_t<T>>
            and is_constructible_v<detail::variant_alternative_selector_t<T, Ts...>, T>
        )
    // clang-format on
    constexpr variant(T&& t) noexcept(is_nothrow_constructible_v<detail::variant_alternative_selector_t<T, Ts...>, T>)
        : variant(in_place_type<detail::variant_alternative_selector_t<T, Ts...>>, etl::forward<T>(t))
    {
    }
 
    template <size_t I, typename... Args>
        requires((I < sizeof...(Ts)) and is_constructible_v<variant_alternative_t<I, variant>, Args...>)
    explicit constexpr variant(in_place_index_t<I> /*index*/, Args&&... args)
        : _index(static_cast<index_type>(I))
        , _union(index_v<I>, etl::forward<Args>(args)...)
    {
    }
 
    template <typename T, typename... Args>
        requires(is_constructible_v<T, Args...> and meta::count_v<remove_cvref_t<T>, meta::list<Ts...>> == 1)
    explicit constexpr variant(in_place_type_t<T> /*tag*/, Args&&... args)
        : variant(in_place_index<meta::index_of_v<T, meta::list<Ts...>>>, etl::forward<Args>(args)...)
    {
    }
 
    constexpr variant(variant const&) = default;
 
    constexpr variant(variant const& other) noexcept((... and etl::is_nothrow_copy_constructible_v<Ts>))
        requires((... and etl::is_copy_constructible_v<Ts>) and !(... and etl::is_trivially_copy_constructible_v<Ts>))
        : variant(other, copy_move_tag{})
    {
    }
 
    constexpr variant(variant&&) = default;
 
    constexpr variant(variant&& other) noexcept((... and is_nothrow_move_constructible_v<Ts>))
        requires((... and is_move_constructible_v<Ts>) and not(... and is_trivially_move_constructible_v<Ts>))
        : variant(etl::move(other), copy_move_tag{})
    {
    }
 
    constexpr auto operator=(variant const&) -> variant& = default;
 
    constexpr auto operator=(variant const& other
    ) noexcept((... and is_nothrow_copy_assignable_v<Ts>) and (... and is_nothrow_copy_constructible_v<Ts>))
        -> variant& requires(
            (... and detail::variant_copy_assignable<Ts>) and not(... and detail::variant_trivially_copy_assignable<Ts>)
        ) {
            assign(other);
            return *this;
        }
 
    constexpr auto operator=(variant&&) -> variant& = default;
 
    constexpr auto operator=(variant&& other
    ) noexcept((... and is_nothrow_move_assignable_v<Ts>) and (... and is_nothrow_move_constructible_v<Ts>))
        -> variant& requires(
            (... and detail::variant_move_assignable<Ts>) and !(... and detail::variant_trivially_move_assignable<Ts>)
        ) {
            assign(etl::move(other));
            return *this;
        }
 
    template <typename T>
        requires(not is_same_v<remove_cvref_t<T>, variant> and is_assignable_v<detail::variant_alternative_selector_t<T, Ts...>&, T> and is_assignable_v<detail::variant_alternative_selector_t<T, Ts...>, T>)
    constexpr auto operator=(T&& t) noexcept(
        (is_nothrow_assignable_v<detail::variant_alternative_selector_t<T, Ts...>&, T>
         and is_nothrow_constructible_v<detail::variant_alternative_selector_t<T, Ts...>, T>)
    ) -> variant&
    {
        emplace<detail::variant_alternative_selector_t<T, Ts...>>(etl::forward<T>(t));
        return *this;
    }
 
    ~variant()
        requires(... and is_trivially_destructible_v<Ts>)
    = default;
 
    constexpr ~variant() { destroy(); }

    [[nodiscard]] constexpr auto index() const noexcept -> size_t { return static_cast<size_t>(_index); }

    template <size_t I>
    constexpr auto operator[](index_constant<I> index) & -> auto&
    {
        static_assert(I < sizeof...(Ts));
        TETL_PRECONDITION(I == index());
        return _union[index];
    }

    template <size_t I>
    constexpr auto operator[](index_constant<I> index) const& -> auto const&
    {
        static_assert(I < sizeof...(Ts));
        TETL_PRECONDITION(I == index());
        return _union[index];
    }

    template <size_t I>
    constexpr auto operator[](index_constant<I> index) && -> auto&&
    {
        static_assert(I < sizeof...(Ts));
        TETL_PRECONDITION(I == index());
        return etl::move(_union)[index];
    }

    template <size_t I>
    constexpr auto operator[](index_constant<I> index) const&& -> auto const&&
    {
        static_assert(I < sizeof...(Ts));
        TETL_PRECONDITION(I == index());
        return etl::move(_union)[index];
    }
 
    template <typename T, typename... Args>
        requires(is_constructible_v<T, Args...> and meta::count_v<T, meta::list<Ts...>> == 1)
    constexpr auto emplace(Args&&... args) -> auto&
    {
        destroy();
        return replace(index_v<meta::index_of_v<T, meta::list<Ts...>>>, etl::forward<Args>(args)...);
    }
 
    template <size_t I, typename... Args>
        requires is_constructible_v<meta::at_t<I, meta::list<Ts...>>, Args...>
    constexpr auto emplace(Args&&... args) -> auto&
    {
        destroy();
        return replace(index_v<I>, etl::forward<Args>(args)...);
    }

    friend constexpr auto operator==(variant const& lhs, variant const& rhs) -> bool
    {
        if (lhs.index() != rhs.index()) {
            return false;
        }
        return etl::visit(detail::make_variant_compare_op(etl::equal_to()), lhs, rhs);
    }

    friend constexpr auto operator<(variant const& lhs, variant const& rhs) -> bool
    {
        if (lhs.index() < rhs.index()) {
            return true;
        }
        if (lhs.index() > rhs.index()) {
            return false;
        }
 
        return etl::visit(detail::make_variant_compare_op(etl::less()), lhs, rhs);
    }

    friend constexpr auto operator<=(variant const& lhs, variant const& rhs) -> bool
    {
        if (lhs.index() < rhs.index()) {
            return true;
        }
        if (lhs.index() > rhs.index()) {
            return false;
        }
 
        return etl::visit(detail::make_variant_compare_op(etl::less_equal()), lhs, rhs);
    }

    friend constexpr auto operator>(variant const& lhs, variant const& rhs) -> bool
    {
        if (lhs.index() > rhs.index()) {
            return true;
        }
        if (lhs.index() < rhs.index()) {
            return false;
        }
 
        return etl::visit(detail::make_variant_compare_op(etl::greater()), lhs, rhs);
    }

    friend constexpr auto operator>=(variant const& lhs, variant const& rhs) -> bool
    {
        if (lhs.index() > rhs.index()) {
            return true;
        }
        if (lhs.index() < rhs.index()) {
            return false;
        }
 
        return etl::visit(detail::make_variant_compare_op(etl::greater_equal()), lhs, rhs);
    }
 
private:
    struct copy_move_tag { };
 
    template <typename Other>
    constexpr variant(Other&& other, [[maybe_unused]] copy_move_tag tag)
        : _index()
        , _union(uninitialized_union())
    {
        etl::visit_with_index([&](auto param) {
            replace(param.index, etl::move(param).value());
        }, etl::forward<Other>(other));
    }
 
    template <typename Other>
    constexpr auto assign(Other&& other) -> void
    {
        etl::visit_with_index([&](auto lhs, auto rhs) {
            if constexpr (lhs.index == rhs.index) {
                lhs.value() = etl::move(rhs.value());
            } else {
                destroy();
                replace(rhs.index, etl::move(rhs.value()));
            }
        }, *this, etl::forward<Other>(other));
    }
 
    template <typename... Args>
    constexpr auto replace(auto index, Args&&... args) -> auto&
    {
        etl::construct_at(etl::addressof(_union), index, etl::forward<Args>(args)...);
        _index = static_cast<index_type>(index.value);
        return (*this)[index];
    }
 
    constexpr auto destroy() -> void
    {
        etl::visit([](auto& v) { etl::destroy_at(etl::addressof(v)); }, *this);
    }
 
    TETL_NO_UNIQUE_ADDRESS index_type _index;
    TETL_NO_UNIQUE_ADDRESS variadic_union<Ts...> _union;
};

template <typename T, typename... Ts>
constexpr auto holds_alternative(variant<Ts...> const& v) noexcept -> bool
{
    static_assert(meta::count_v<T, meta::list<Ts...>> == 1);
    return v.index() == meta::index_of_v<T, meta::list<Ts...>>;
}

template <size_t I, typename... Ts>
constexpr auto unchecked_get(variant<Ts...>& v) -> auto&
{
    static_assert(I < sizeof...(Ts));
    TETL_PRECONDITION(I == v.index());
    return v[index_v<I>];
}

template <size_t I, typename... Ts>
constexpr auto unchecked_get(variant<Ts...> const& v) -> auto const&
{
    static_assert(I < sizeof...(Ts));
    TETL_PRECONDITION(I == v.index());
    return v[index_v<I>];
}

template <size_t I, typename... Ts>
constexpr auto unchecked_get(variant<Ts...>&& v) -> auto&&
{
    static_assert(I < sizeof...(Ts));
    TETL_PRECONDITION(I == v.index());
    return etl::move(v)[index_v<I>];
}

template <size_t I, typename... Ts>
constexpr auto unchecked_get(variant<Ts...> const&& v) -> auto const&&
{
    static_assert(I < sizeof...(Ts));
    TETL_PRECONDITION(I == v.index());
    return etl::move(v)[index_v<I>];
}

template <size_t I, typename... Ts>
constexpr auto get_if(variant<Ts...>* pv) noexcept -> add_pointer_t<variant_alternative_t<I, variant<Ts...>>>
{
    static_assert(I < sizeof...(Ts));
    if (pv == nullptr or pv->index() != I) {
        return nullptr;
    }
    return etl::addressof(etl::unchecked_get<I>(*pv));
}

template <size_t I, typename... Ts>
constexpr auto get_if(variant<Ts...> const* pv) noexcept
    -> add_pointer_t<variant_alternative_t<I, variant<Ts...>> const>
{
    static_assert(I < sizeof...(Ts));
    if (pv == nullptr or pv->index() != I) {
        return nullptr;
    }
    return etl::addressof(etl::unchecked_get<I>(*pv));
}

template <typename T, typename... Ts>
constexpr auto get_if(variant<Ts...>* pv) noexcept -> add_pointer_t<T>
{
    return etl::get_if<meta::index_of_v<T, meta::list<Ts...>>>(pv);
}

template <typename T, typename... Ts>
constexpr auto get_if(variant<Ts...> const* pv) noexcept -> add_pointer_t<T const>
{
    return etl::get_if<meta::index_of_v<T, meta::list<Ts...>>>(pv);
}
 
} // namespace etl
 
#endif // TETL_VARIANT_VARIANT2_HPP