cache.hpp

#ifndef ENTT_RESOURCE_CACHE_HPP
#define ENTT_RESOURCE_CACHE_HPP
 
 
#include <memory>
#include <utility>
#include <type_traits>
#include <unordered_map>
#include "../config/config.h"
#include "handle.hpp"
#include "loader.hpp"
#include "fwd.hpp"
 
 
namespace entt {
 
 
template<typename Resource>
struct cache {
    using size_type = std::size_t;
    using resource_type = Resource;
    using id_type = ENTT_ID_TYPE;
 
    cache() = default;
 
    cache(cache &&) = default;
 
    cache & operator=(cache &&) = default;
 
    size_type size() const ENTT_NOEXCEPT {
        return resources.size();
    }
 
    bool empty() const ENTT_NOEXCEPT {
        return resources.empty();
    }
 
    void clear() ENTT_NOEXCEPT {
        resources.clear();
    }
 
    template<typename Loader, typename... Args>
    entt::handle<Resource> load(const id_type id, Args &&... args) {
        static_assert(std::is_base_of_v<loader<Loader, Resource>, Loader>);
        entt::handle<Resource> resource{};
 
        if(auto it = resources.find(id); it == resources.cend()) {
            if(auto instance = Loader{}.get(std::forward<Args>(args)...); instance) {
                resources[id] = instance;
                resource = std::move(instance);
            }
        } else {
            resource = it->second;
        }
 
        return resource;
    }
 
    template<typename Loader, typename... Args>
    entt::handle<Resource> reload(const id_type id, Args &&... args) {
        return (discard(id), load<Loader>(id, std::forward<Args>(args)...));
    }
 
    template<typename Loader, typename... Args>
    entt::handle<Resource> temp(Args &&... args) const {
        return { Loader{}.get(std::forward<Args>(args)...) };
    }
 
    entt::handle<Resource> handle(const id_type id) const {
        auto it = resources.find(id);
        return { it == resources.end() ? nullptr : it->second };
    }
 
    bool contains(const id_type id) const ENTT_NOEXCEPT {
        return (resources.find(id) != resources.cend());
    }
 
    void discard(const id_type id) ENTT_NOEXCEPT {
        if(auto it = resources.find(id); it != resources.end()) {
            resources.erase(it);
        }
    }
 
    template <typename Func>
    void each(Func func) const {
        auto begin = resources.begin();
        auto end = resources.end();
 
        while(begin != end) {
            auto curr = begin++;
 
            if constexpr(std::is_invocable_v<Func, id_type>) {
                func(curr->first);
            } else if constexpr(std::is_invocable_v<Func, entt::handle<Resource>>) {
                func(entt::handle{ curr->second });
            } else {
                func(curr->first, entt::handle{ curr->second });
            }
        }
    }
 
private:
    std::unordered_map<id_type, std::shared_ptr<Resource>> resources;
};
 
 
}
 
 
#endif // ENTT_RESOURCE_CACHE_HPP