cache.hpp

#ifndef ENTT_RESOURCE_RESOURCE_CACHE_HPP
#define ENTT_RESOURCE_RESOURCE_CACHE_HPP
 
#include <compare>
#include "../container/dense_map.hpp"
#include "../core/compressed_pair.hpp"
#include "../core/fwd.hpp"
#include "../core/iterator.hpp"
#include "../stl/concepts.hpp"
#include "../stl/cstddef.hpp"
#include "../stl/functional.hpp"
#include "../stl/iterator.hpp"
#include "../stl/memory.hpp"
#include "../stl/tuple.hpp"
#include "../stl/type_traits.hpp"
#include "../stl/utility.hpp"
#include "fwd.hpp"
#include "loader.hpp"
#include "resource.hpp"
 
namespace entt {

namespace internal {
 
template<typename Type, typename It>
class resource_cache_iterator final {
    template<typename, typename>
    friend class resource_cache_iterator;
 
public:
    using value_type = stl::pair<id_type, resource<Type>>;
    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 resource_cache_iterator() noexcept = default;
 
    constexpr resource_cache_iterator(const It iter) noexcept
        : it{iter} {}
 
    template<typename Other>
    requires (!stl::same_as<It, Other> && stl::constructible_from<It, Other>)
    constexpr resource_cache_iterator(const resource_cache_iterator<stl::remove_const_t<Type>, Other> &other) noexcept
        : it{other.it} {}
 
    constexpr resource_cache_iterator &operator++() noexcept {
        return ++it, *this;
    }
 
    constexpr resource_cache_iterator operator++(int) noexcept {
        const resource_cache_iterator orig = *this;
        return ++(*this), orig;
    }
 
    constexpr resource_cache_iterator &operator--() noexcept {
        return --it, *this;
    }
 
    constexpr resource_cache_iterator operator--(int) noexcept {
        const resource_cache_iterator orig = *this;
        return operator--(), orig;
    }
 
    constexpr resource_cache_iterator &operator+=(const difference_type value) noexcept {
        it += value;
        return *this;
    }
 
    constexpr resource_cache_iterator operator+(const difference_type value) const noexcept {
        resource_cache_iterator copy = *this;
        return (copy += value);
    }
 
    constexpr resource_cache_iterator &operator-=(const difference_type value) noexcept {
        return (*this += -value);
    }
 
    constexpr resource_cache_iterator operator-(const difference_type value) const noexcept {
        return (*this + -value);
    }
 
    [[nodiscard]] constexpr reference operator[](const difference_type value) const noexcept {
        return {it[value].first, resource<Type>{it[value].second}};
    }
 
    [[nodiscard]] constexpr reference operator*() const noexcept {
        return operator[](0);
    }
 
    [[nodiscard]] constexpr pointer operator->() const noexcept {
        return operator*();
    }
 
    template<typename... Args>
    [[nodiscard]] constexpr stl::ptrdiff_t operator-(const resource_cache_iterator<Args...> &other) const noexcept {
        return it - other.it;
    }
 
    template<typename... Args>
    [[nodiscard]] constexpr bool operator==(const resource_cache_iterator<Args...> &other) const noexcept {
        return it == other.it;
    }
 
    template<typename... Args>
    [[nodiscard]] constexpr auto operator<=>(const resource_cache_iterator<Args...> &other) const noexcept {
        return it <=> other.it;
    }
 
private:
    It it;
};
 
} // namespace internal

template<typename Type, typename Loader, typename Allocator>
class resource_cache {
    using alloc_traits = stl::allocator_traits<Allocator>;
    static_assert(stl::is_same_v<typename alloc_traits::value_type, Type>, "Invalid value type");
    using container_allocator = alloc_traits::template rebind_alloc<stl::pair<const id_type, typename Loader::result_type>>;
    using container_type = dense_map<id_type, typename Loader::result_type, stl::identity, stl::equal_to<>, container_allocator>;
 
public:
    using allocator_type = Allocator;
    using value_type = Type;
    using size_type = stl::size_t;
    using loader_type = Loader;
    using iterator = internal::resource_cache_iterator<Type, typename container_type::iterator>;
    using const_iterator = internal::resource_cache_iterator<const Type, typename container_type::const_iterator>;

    resource_cache()
        : resource_cache{loader_type{}} {}

    explicit resource_cache(const allocator_type &allocator)
        : resource_cache{loader_type{}, allocator} {}

    explicit resource_cache(const loader_type &callable, const allocator_type &allocator = allocator_type{})
        : pool{container_type{allocator}, callable} {}

    resource_cache(const resource_cache &) = default;

    resource_cache(const resource_cache &other, const allocator_type &allocator)
        : pool{stl::piecewise_construct, stl::forward_as_tuple(other.pool.first(), allocator), stl::forward_as_tuple(other.pool.second())} {}

    resource_cache(resource_cache &&) noexcept = default;

    resource_cache(resource_cache &&other, const allocator_type &allocator)
        : pool{stl::piecewise_construct, stl::forward_as_tuple(stl::move(other.pool.first()), allocator), stl::forward_as_tuple(stl::move(other.pool.second()))} {}

    ~resource_cache() = default;

    resource_cache &operator=(const resource_cache &) = default;

    resource_cache &operator=(resource_cache &&) noexcept = default;

    [[nodiscard]] constexpr allocator_type get_allocator() const noexcept {
        return pool.first().get_allocator();
    }

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

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

    [[nodiscard]] iterator begin() noexcept {
        return pool.first().begin();
    }

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

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

    [[nodiscard]] iterator end() noexcept {
        return pool.first().end();
    }

    [[nodiscard]] bool empty() const noexcept {
        return pool.first().empty();
    }

    [[nodiscard]] size_type size() const noexcept {
        return pool.first().size();
    }

    void clear() noexcept {
        pool.first().clear();
    }

    template<typename... Args>
    stl::pair<iterator, bool> load(const id_type id, Args &&...args) {
        if(auto it = pool.first().find(id); it != pool.first().end()) {
            return {it, false};
        }
 
        return pool.first().emplace(id, pool.second()(stl::forward<Args>(args)...));
    }

    template<typename... Args>
    stl::pair<iterator, bool> force_load(const id_type id, Args &&...args) {
        return {pool.first().insert_or_assign(id, pool.second()(stl::forward<Args>(args)...)).first, true};
    }

    [[nodiscard]] resource<const value_type> operator[](const id_type id) const {
        if(auto it = pool.first().find(id); it != pool.first().cend()) {
            return resource<const value_type>{it->second};
        }
 
        return {};
    }

    [[nodiscard]] resource<value_type> operator[](const id_type id) {
        if(auto it = pool.first().find(id); it != pool.first().end()) {
            return resource<value_type>{it->second};
        }
 
        return {};
    }

    [[nodiscard]] bool contains(const id_type id) const {
        return pool.first().contains(id);
    }

    iterator erase(const_iterator pos) {
        const auto it = pool.first().begin();
        return pool.first().erase(it + (pos - const_iterator{it}));
    }

    iterator erase(const_iterator first, const_iterator last) {
        const auto it = pool.first().begin();
        return pool.first().erase(it + (first - const_iterator{it}), it + (last - const_iterator{it}));
    }

    size_type erase(const id_type id) {
        return pool.first().erase(id);
    }

    [[nodiscard]] loader_type loader() const {
        return pool.second();
    }
 
private:
    compressed_pair<container_type, loader_type> pool;
};
 
} // namespace entt
 
#endif