table.hpp

#ifndef ENTT_CONTAINER_TABLE_HPP
#define ENTT_CONTAINER_TABLE_HPP
 
#include "../config/config.h"
#include "../core/iterator.hpp"
#include "../stl/concepts.hpp"
#include "../stl/cstddef.hpp"
#include "../stl/iterator.hpp"
#include "../stl/tuple.hpp"
#include "../stl/utility.hpp"
#include "fwd.hpp"
 
namespace entt {

namespace internal {
 
template<typename... It>
class table_iterator {
    template<typename...>
    friend class table_iterator;
 
public:
    using value_type = decltype(stl::forward_as_tuple(*stl::declval<It>()...));
    using pointer = input_iterator_pointer<value_type>;
    using reference = value_type;
    using difference_type = stl::ptrdiff_t;
    using iterator_category = stl::input_iterator_tag;
    using iterator_concept = stl::random_access_iterator_tag;
 
    constexpr table_iterator() noexcept
        : it{} {}
 
    constexpr table_iterator(It... from) noexcept
        : it{from...} {}
 
    template<typename... Other>
    requires (stl::constructible_from<It, Other> && ...)
    constexpr table_iterator(const table_iterator<Other...> &other) noexcept
        : table_iterator{stl::get<Other>(other.it)...} {}
 
    constexpr table_iterator &operator++() noexcept {
        return (++stl::get<It>(it), ...), *this;
    }
 
    constexpr table_iterator operator++(int) noexcept {
        const table_iterator orig = *this;
        return ++(*this), orig;
    }
 
    constexpr table_iterator &operator--() noexcept {
        return (--stl::get<It>(it), ...), *this;
    }
 
    constexpr table_iterator operator--(int) noexcept {
        const table_iterator orig = *this;
        return operator--(), orig;
    }
 
    constexpr table_iterator &operator+=(const difference_type value) noexcept {
        return ((stl::get<It>(it) += value), ...), *this;
    }
 
    constexpr table_iterator operator+(const difference_type value) const noexcept {
        table_iterator copy = *this;
        return (copy += value);
    }
 
    constexpr table_iterator &operator-=(const difference_type value) noexcept {
        return (*this += -value);
    }
 
    constexpr table_iterator operator-(const difference_type value) const noexcept {
        return (*this + -value);
    }
 
    [[nodiscard]] constexpr reference operator[](const difference_type value) const noexcept {
        return stl::forward_as_tuple(stl::get<It>(it)[value]...);
    }
 
    [[nodiscard]] constexpr pointer operator->() const noexcept {
        return {operator[](0)};
    }
 
    [[nodiscard]] constexpr reference operator*() const noexcept {
        return operator[](0);
    }
 
    template<typename... Other>
    [[nodiscard]] constexpr stl::ptrdiff_t operator-(const table_iterator<Other...> &other) const noexcept {
        return stl::get<0>(it) - stl::get<0>(other.it);
    }
 
    template<typename... Other>
    [[nodiscard]] constexpr bool operator==(const table_iterator<Other...> &other) const noexcept {
        return stl::get<0>(it) == stl::get<0>(other.it);
    }
 
    template<typename... Other>
    [[nodiscard]] constexpr auto operator<=>(const table_iterator<Other...> &other) const noexcept {
        return stl::get<0>(it) <=> stl::get<0>(other.it);
    }
 
private:
    stl::tuple<It...> it;
};
 
} // namespace internal

template<typename... Container>
class basic_table {
    using container_type = stl::tuple<Container...>;
 
public:
    using size_type = stl::size_t;
    using difference_type = stl::ptrdiff_t;
    using iterator = internal::table_iterator<typename Container::iterator...>;
    using const_iterator = internal::table_iterator<typename Container::const_iterator...>;
    using reverse_iterator = internal::table_iterator<typename Container::reverse_iterator...>;
    using const_reverse_iterator = internal::table_iterator<typename Container::const_reverse_iterator...>;

    basic_table()
        : payload{} {
    }

    explicit basic_table(const Container &...container) noexcept
        : payload{container...} {
        ENTT_ASSERT((((stl::get<Container>(payload).size() * sizeof...(Container)) == (stl::get<Container>(payload).size() + ...)) && ...), "Unexpected container size");
    }

    explicit basic_table(Container &&...container) noexcept
        : payload{stl::move(container)...} {
        ENTT_ASSERT((((stl::get<Container>(payload).size() * sizeof...(Container)) == (stl::get<Container>(payload).size() + ...)) && ...), "Unexpected container size");
    }

    basic_table(const basic_table &) = delete;

    basic_table(basic_table &&other) noexcept
        : payload{stl::move(other.payload)} {}

    explicit basic_table(const auto &allocator)
        : payload{Container{allocator}...} {}

    template<class Allocator>
    basic_table(const Container &...container, const Allocator &allocator) noexcept
        : payload{Container{container, allocator}...} {
        ENTT_ASSERT((((stl::get<Container>(payload).size() * sizeof...(Container)) == (stl::get<Container>(payload).size() + ...)) && ...), "Unexpected container size");
    }

    template<class Allocator>
    basic_table(Container &&...container, const Allocator &allocator) noexcept
        : payload{Container{stl::move(container), allocator}...} {
        ENTT_ASSERT((((stl::get<Container>(payload).size() * sizeof...(Container)) == (stl::get<Container>(payload).size() + ...)) && ...), "Unexpected container size");
    }

    template<class Allocator>
    basic_table(basic_table &&other, const Allocator &allocator)
        : payload{Container{stl::move(stl::get<Container>(other.payload)), allocator}...} {}

    ~basic_table() = default;

    basic_table &operator=(const basic_table &) = delete;

    basic_table &operator=(basic_table &&other) noexcept {
        swap(other);
        return *this;
    }

    void swap(basic_table &other) noexcept {
        using stl::swap;
        swap(payload, other.payload);
    }

    void reserve(const size_type cap) {
        (stl::get<Container>(payload).reserve(cap), ...);
    }

    [[nodiscard]] size_type capacity() const noexcept {
        return stl::get<0>(payload).capacity();
    }

    void shrink_to_fit() {
        (stl::get<Container>(payload).shrink_to_fit(), ...);
    }

    [[nodiscard]] size_type size() const noexcept {
        return stl::get<0>(payload).size();
    }

    [[nodiscard]] bool empty() const noexcept {
        return stl::get<0>(payload).empty();
    }

    [[nodiscard]] const_iterator cbegin() const noexcept {
        return {stl::get<Container>(payload).cbegin()...};
    }

    [[nodiscard]] const_iterator begin() const noexcept {
        return cbegin();
    }

    [[nodiscard]] iterator begin() noexcept {
        return {stl::get<Container>(payload).begin()...};
    }

    [[nodiscard]] const_iterator cend() const noexcept {
        return {stl::get<Container>(payload).cend()...};
    }

    [[nodiscard]] const_iterator end() const noexcept {
        return cend();
    }

    [[nodiscard]] iterator end() noexcept {
        return {stl::get<Container>(payload).end()...};
    }

    [[nodiscard]] const_reverse_iterator crbegin() const noexcept {
        return {stl::get<Container>(payload).crbegin()...};
    }

    [[nodiscard]] const_reverse_iterator rbegin() const noexcept {
        return crbegin();
    }

    [[nodiscard]] reverse_iterator rbegin() noexcept {
        return {stl::get<Container>(payload).rbegin()...};
    }

    [[nodiscard]] const_reverse_iterator crend() const noexcept {
        return {stl::get<Container>(payload).crend()...};
    }

    [[nodiscard]] const_reverse_iterator rend() const noexcept {
        return crend();
    }

    [[nodiscard]] reverse_iterator rend() noexcept {
        return {stl::get<Container>(payload).rend()...};
    }

    template<typename... Args>
    stl::tuple<typename Container::value_type &...> emplace(Args &&...args) {
        if constexpr(sizeof...(Args) == 0u) {
            return stl::forward_as_tuple(stl::get<Container>(payload).emplace_back()...);
        } else {
            return stl::forward_as_tuple(stl::get<Container>(payload).emplace_back(stl::forward<Args>(args))...);
        }
    }

    iterator erase(const_iterator pos) {
        const auto diff = pos - begin();
        return {stl::get<Container>(payload).erase(stl::get<Container>(payload).begin() + diff)...};
    }

    void erase(const size_type pos) {
        ENTT_ASSERT(pos < size(), "Index out of bounds");
        erase(begin() + static_cast<difference_type>(pos));
    }

    [[nodiscard]] stl::tuple<const typename Container::value_type &...> operator[](const size_type pos) const {
        ENTT_ASSERT(pos < size(), "Index out of bounds");
        return stl::forward_as_tuple(stl::get<Container>(payload)[pos]...);
    }

    [[nodiscard]] stl::tuple<typename Container::value_type &...> operator[](const size_type pos) {
        ENTT_ASSERT(pos < size(), "Index out of bounds");
        return stl::forward_as_tuple(stl::get<Container>(payload)[pos]...);
    }

    void clear() {
        (stl::get<Container>(payload).clear(), ...);
    }
 
private:
    container_type payload;
};
 
} // namespace entt

#include <utility>
 
namespace std {
 
template<typename... Container, typename Allocator>
struct uses_allocator<entt::basic_table<Container...>, Allocator>
    : entt::stl::bool_constant<(entt::stl::uses_allocator_v<Container, Allocator> && ...)> {};
 
} // namespace std
 
#endif