| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205 |
- #ifndef ENTT_ENTITY_REGISTRY_HPP
- #define ENTT_ENTITY_REGISTRY_HPP
- #include <algorithm>
- #include <cstddef>
- #include <functional>
- #include <iterator>
- #include <memory>
- #include <tuple>
- #include <type_traits>
- #include <utility>
- #include <vector>
- #include "../config/config.h"
- #include "../container/dense_map.hpp"
- #include "../core/algorithm.hpp"
- #include "../core/any.hpp"
- #include "../core/fwd.hpp"
- #include "../core/iterator.hpp"
- #include "../core/memory.hpp"
- #include "../core/type_info.hpp"
- #include "../core/type_traits.hpp"
- #include "../core/utility.hpp"
- #include "entity.hpp"
- #include "fwd.hpp"
- #include "group.hpp"
- #include "mixin.hpp"
- #include "sparse_set.hpp"
- #include "storage.hpp"
- #include "view.hpp"
- namespace entt {
- /**
- * @cond TURN_OFF_DOXYGEN
- * Internal details not to be documented.
- */
- namespace internal {
- template<typename It>
- class registry_storage_iterator final {
- template<typename Other>
- friend class registry_storage_iterator;
- using mapped_type = std::remove_reference_t<decltype(std::declval<It>()->second)>;
- public:
- using value_type = std::pair<id_type, constness_as_t<typename mapped_type::element_type, mapped_type> &>;
- using pointer = input_iterator_pointer<value_type>;
- using reference = value_type;
- using difference_type = std::ptrdiff_t;
- using iterator_category = std::input_iterator_tag;
- using iterator_concept = std::random_access_iterator_tag;
- constexpr registry_storage_iterator() noexcept
- : it{} {}
- constexpr registry_storage_iterator(It iter) noexcept
- : it{iter} {}
- template<typename Other, typename = std::enable_if_t<!std::is_same_v<It, Other> && std::is_constructible_v<It, Other>>>
- constexpr registry_storage_iterator(const registry_storage_iterator<Other> &other) noexcept
- : registry_storage_iterator{other.it} {}
- constexpr registry_storage_iterator &operator++() noexcept {
- return ++it, *this;
- }
- constexpr registry_storage_iterator operator++(int) noexcept {
- registry_storage_iterator orig = *this;
- return ++(*this), orig;
- }
- constexpr registry_storage_iterator &operator--() noexcept {
- return --it, *this;
- }
- constexpr registry_storage_iterator operator--(int) noexcept {
- registry_storage_iterator orig = *this;
- return operator--(), orig;
- }
- constexpr registry_storage_iterator &operator+=(const difference_type value) noexcept {
- it += value;
- return *this;
- }
- constexpr registry_storage_iterator operator+(const difference_type value) const noexcept {
- registry_storage_iterator copy = *this;
- return (copy += value);
- }
- constexpr registry_storage_iterator &operator-=(const difference_type value) noexcept {
- return (*this += -value);
- }
- constexpr registry_storage_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, *it[value].second};
- }
- [[nodiscard]] constexpr reference operator*() const noexcept {
- return {it->first, *it->second};
- }
- [[nodiscard]] constexpr pointer operator->() const noexcept {
- return operator*();
- }
- template<typename Lhs, typename Rhs>
- friend constexpr std::ptrdiff_t operator-(const registry_storage_iterator<Lhs> &, const registry_storage_iterator<Rhs> &) noexcept;
- template<typename Lhs, typename Rhs>
- friend constexpr bool operator==(const registry_storage_iterator<Lhs> &, const registry_storage_iterator<Rhs> &) noexcept;
- template<typename Lhs, typename Rhs>
- friend constexpr bool operator<(const registry_storage_iterator<Lhs> &, const registry_storage_iterator<Rhs> &) noexcept;
- private:
- It it;
- };
- template<typename Lhs, typename Rhs>
- [[nodiscard]] constexpr std::ptrdiff_t operator-(const registry_storage_iterator<Lhs> &lhs, const registry_storage_iterator<Rhs> &rhs) noexcept {
- return lhs.it - rhs.it;
- }
- template<typename Lhs, typename Rhs>
- [[nodiscard]] constexpr bool operator==(const registry_storage_iterator<Lhs> &lhs, const registry_storage_iterator<Rhs> &rhs) noexcept {
- return lhs.it == rhs.it;
- }
- template<typename Lhs, typename Rhs>
- [[nodiscard]] constexpr bool operator!=(const registry_storage_iterator<Lhs> &lhs, const registry_storage_iterator<Rhs> &rhs) noexcept {
- return !(lhs == rhs);
- }
- template<typename Lhs, typename Rhs>
- [[nodiscard]] constexpr bool operator<(const registry_storage_iterator<Lhs> &lhs, const registry_storage_iterator<Rhs> &rhs) noexcept {
- return lhs.it < rhs.it;
- }
- template<typename Lhs, typename Rhs>
- [[nodiscard]] constexpr bool operator>(const registry_storage_iterator<Lhs> &lhs, const registry_storage_iterator<Rhs> &rhs) noexcept {
- return rhs < lhs;
- }
- template<typename Lhs, typename Rhs>
- [[nodiscard]] constexpr bool operator<=(const registry_storage_iterator<Lhs> &lhs, const registry_storage_iterator<Rhs> &rhs) noexcept {
- return !(lhs > rhs);
- }
- template<typename Lhs, typename Rhs>
- [[nodiscard]] constexpr bool operator>=(const registry_storage_iterator<Lhs> &lhs, const registry_storage_iterator<Rhs> &rhs) noexcept {
- return !(lhs < rhs);
- }
- template<typename Allocator>
- class registry_context {
- using alloc_traits = std::allocator_traits<Allocator>;
- using allocator_type = typename alloc_traits::template rebind_alloc<std::pair<const id_type, basic_any<0u>>>;
- public:
- explicit registry_context(const allocator_type &allocator)
- : ctx{allocator} {}
- template<typename Type, typename... Args>
- Type &emplace_as(const id_type id, Args &&...args) {
- return any_cast<Type &>(ctx.try_emplace(id, std::in_place_type<Type>, std::forward<Args>(args)...).first->second);
- }
- template<typename Type, typename... Args>
- Type &emplace(Args &&...args) {
- return emplace_as<Type>(type_id<Type>().hash(), std::forward<Args>(args)...);
- }
- template<typename Type>
- Type &insert_or_assign(const id_type id, Type &&value) {
- return any_cast<std::remove_cv_t<std::remove_reference_t<Type>> &>(ctx.insert_or_assign(id, std::forward<Type>(value)).first->second);
- }
- template<typename Type>
- Type &insert_or_assign(Type &&value) {
- return insert_or_assign(type_id<Type>().hash(), std::forward<Type>(value));
- }
- template<typename Type>
- bool erase(const id_type id = type_id<Type>().hash()) {
- const auto it = ctx.find(id);
- return it != ctx.end() && it->second.type() == type_id<Type>() ? (ctx.erase(it), true) : false;
- }
- template<typename Type>
- [[nodiscard]] const Type &get(const id_type id = type_id<Type>().hash()) const {
- return any_cast<const Type &>(ctx.at(id));
- }
- template<typename Type>
- [[nodiscard]] Type &get(const id_type id = type_id<Type>().hash()) {
- return any_cast<Type &>(ctx.at(id));
- }
- template<typename Type>
- [[nodiscard]] const Type *find(const id_type id = type_id<Type>().hash()) const {
- const auto it = ctx.find(id);
- return it != ctx.cend() ? any_cast<const Type>(&it->second) : nullptr;
- }
- template<typename Type>
- [[nodiscard]] Type *find(const id_type id = type_id<Type>().hash()) {
- const auto it = ctx.find(id);
- return it != ctx.end() ? any_cast<Type>(&it->second) : nullptr;
- }
- template<typename Type>
- [[nodiscard]] bool contains(const id_type id = type_id<Type>().hash()) const {
- const auto it = ctx.find(id);
- return it != ctx.cend() && it->second.type() == type_id<Type>();
- }
- private:
- dense_map<id_type, basic_any<0u>, identity, std::equal_to<id_type>, allocator_type> ctx;
- };
- } // namespace internal
- /**
- * Internal details not to be documented.
- * @endcond
- */
- /**
- * @brief Fast and reliable entity-component system.
- * @tparam Entity A valid entity type.
- * @tparam Allocator Type of allocator used to manage memory and elements.
- */
- template<typename Entity, typename Allocator>
- class basic_registry {
- using base_type = basic_sparse_set<Entity, Allocator>;
- using alloc_traits = std::allocator_traits<Allocator>;
- static_assert(std::is_same_v<typename alloc_traits::value_type, Entity>, "Invalid value type");
- // std::shared_ptr because of its type erased allocator which is useful here
- using pool_container_type = dense_map<id_type, std::shared_ptr<base_type>, identity, std::equal_to<id_type>, typename alloc_traits::template rebind_alloc<std::pair<const id_type, std::shared_ptr<base_type>>>>;
- using group_container_type = dense_map<id_type, std::shared_ptr<internal::group_descriptor>, identity, std::equal_to<id_type>, typename alloc_traits::template rebind_alloc<std::pair<const id_type, std::shared_ptr<internal::group_descriptor>>>>;
- template<typename Type>
- [[nodiscard]] auto &assure([[maybe_unused]] const id_type id = type_hash<Type>::value()) {
- if constexpr(std::is_same_v<Type, entity_type>) {
- return entities;
- } else {
- static_assert(std::is_same_v<Type, std::decay_t<Type>>, "Non-decayed types not allowed");
- auto &cpool = pools[id];
- if(!cpool) {
- using storage_type = storage_for_type<Type>;
- using alloc_type = typename storage_type::allocator_type;
- if constexpr(std::is_void_v<Type> && !std::is_constructible_v<alloc_type, allocator_type>) {
- // std::allocator<void> has no cross constructors (waiting for C++20)
- cpool = std::allocate_shared<storage_type>(get_allocator(), alloc_type{});
- } else {
- cpool = std::allocate_shared<storage_type>(get_allocator(), get_allocator());
- }
- cpool->bind(forward_as_any(*this));
- }
- ENTT_ASSERT(cpool->type() == type_id<Type>(), "Unexpected type");
- return static_cast<storage_for_type<Type> &>(*cpool);
- }
- }
- template<typename Type>
- [[nodiscard]] const auto *assure([[maybe_unused]] const id_type id = type_hash<Type>::value()) const {
- if constexpr(std::is_same_v<Type, entity_type>) {
- return &entities;
- } else {
- static_assert(std::is_same_v<Type, std::decay_t<Type>>, "Non-decayed types not allowed");
- if(const auto it = pools.find(id); it != pools.cend()) {
- ENTT_ASSERT(it->second->type() == type_id<Type>(), "Unexpected type");
- return static_cast<const storage_for_type<Type> *>(it->second.get());
- }
- return static_cast<const storage_for_type<Type> *>(nullptr);
- }
- }
- void rebind() {
- entities.bind(forward_as_any(*this));
- for(auto &&curr: pools) {
- curr.second->bind(forward_as_any(*this));
- }
- }
- public:
- /*! @brief Entity traits. */
- using traits_type = typename base_type::traits_type;
- /*! @brief Allocator type. */
- using allocator_type = Allocator;
- /*! @brief Underlying entity identifier. */
- using entity_type = typename traits_type::value_type;
- /*! @brief Underlying version type. */
- using version_type = typename traits_type::version_type;
- /*! @brief Unsigned integer type. */
- using size_type = std::size_t;
- /*! @brief Common type among all storage types. */
- using common_type = base_type;
- /*! @brief Context type. */
- using context = internal::registry_context<allocator_type>;
- /**
- * @copybrief storage_for
- * @tparam Type Storage value type, eventually const.
- */
- template<typename Type>
- using storage_for_type = typename storage_for<Type, Entity, typename alloc_traits::template rebind_alloc<std::remove_const_t<Type>>>::type;
- /*! @brief Default constructor. */
- basic_registry()
- : basic_registry{allocator_type{}} {}
- /**
- * @brief Constructs an empty registry with a given allocator.
- * @param allocator The allocator to use.
- */
- explicit basic_registry(const allocator_type &allocator)
- : basic_registry{0u, allocator} {}
- /**
- * @brief Allocates enough memory upon construction to store `count` pools.
- * @param count The number of pools to allocate memory for.
- * @param allocator The allocator to use.
- */
- basic_registry(const size_type count, const allocator_type &allocator = allocator_type{})
- : vars{allocator},
- pools{allocator},
- groups{allocator},
- entities{allocator} {
- pools.reserve(count);
- rebind();
- }
- /**
- * @brief Move constructor.
- * @param other The instance to move from.
- */
- basic_registry(basic_registry &&other) noexcept
- : vars{std::move(other.vars)},
- pools{std::move(other.pools)},
- groups{std::move(other.groups)},
- entities{std::move(other.entities)} {
- rebind();
- }
- /**
- * @brief Move assignment operator.
- * @param other The instance to move from.
- * @return This registry.
- */
- basic_registry &operator=(basic_registry &&other) noexcept {
- vars = std::move(other.vars);
- pools = std::move(other.pools);
- groups = std::move(other.groups);
- entities = std::move(other.entities);
- rebind();
- return *this;
- }
- /**
- * @brief Exchanges the contents with those of a given registry.
- * @param other Registry to exchange the content with.
- */
- void swap(basic_registry &other) {
- using std::swap;
- swap(vars, other.vars);
- swap(pools, other.pools);
- swap(groups, other.groups);
- swap(entities, other.entities);
- rebind();
- other.rebind();
- }
- /**
- * @brief Returns the associated allocator.
- * @return The associated allocator.
- */
- [[nodiscard]] constexpr allocator_type get_allocator() const noexcept {
- return entities.get_allocator();
- }
- /**
- * @brief Returns an iterable object to use to _visit_ a registry.
- *
- * The iterable object returns a pair that contains the name and a reference
- * to the current storage.
- *
- * @return An iterable object to use to _visit_ the registry.
- */
- [[nodiscard]] auto storage() noexcept {
- return iterable_adaptor{internal::registry_storage_iterator{pools.begin()}, internal::registry_storage_iterator{pools.end()}};
- }
- /*! @copydoc storage */
- [[nodiscard]] auto storage() const noexcept {
- return iterable_adaptor{internal::registry_storage_iterator{pools.cbegin()}, internal::registry_storage_iterator{pools.cend()}};
- }
- /**
- * @brief Finds the storage associated with a given name, if any.
- * @param id Name used to map the storage within the registry.
- * @return A pointer to the storage if it exists, a null pointer otherwise.
- */
- [[nodiscard]] common_type *storage(const id_type id) {
- return const_cast<common_type *>(std::as_const(*this).storage(id));
- }
- /**
- * @brief Finds the storage associated with a given name, if any.
- * @param id Name used to map the storage within the registry.
- * @return A pointer to the storage if it exists, a null pointer otherwise.
- */
- [[nodiscard]] const common_type *storage(const id_type id) const {
- const auto it = pools.find(id);
- return it == pools.cend() ? nullptr : it->second.get();
- }
- /**
- * @brief Returns the storage for a given component type.
- * @tparam Type Type of component of which to return the storage.
- * @param id Optional name used to map the storage within the registry.
- * @return The storage for the given component type.
- */
- template<typename Type>
- storage_for_type<Type> &storage(const id_type id = type_hash<Type>::value()) {
- return assure<Type>(id);
- }
- /**
- * @brief Returns the storage for a given component type, if any.
- * @tparam Type Type of component of which to return the storage.
- * @param id Optional name used to map the storage within the registry.
- * @return The storage for the given component type.
- */
- template<typename Type>
- const storage_for_type<Type> *storage(const id_type id = type_hash<Type>::value()) const {
- return assure<Type>(id);
- }
- /**
- * @brief Checks if an identifier refers to a valid entity.
- * @param entt An identifier, either valid or not.
- * @return True if the identifier is valid, false otherwise.
- */
- [[nodiscard]] bool valid(const entity_type entt) const {
- return entities.contains(entt) && (entities.index(entt) < entities.free_list());
- }
- /**
- * @brief Returns the actual version for an identifier.
- * @param entt A valid identifier.
- * @return The version for the given identifier if valid, the tombstone
- * version otherwise.
- */
- [[nodiscard]] version_type current(const entity_type entt) const {
- return entities.current(entt);
- }
- /**
- * @brief Creates a new entity or recycles a destroyed one.
- * @return A valid identifier.
- */
- [[nodiscard]] entity_type create() {
- return entities.emplace();
- }
- /**
- * @copybrief create
- *
- * If the requested entity isn't in use, the suggested identifier is used.
- * Otherwise, a new identifier is generated.
- *
- * @param hint Required identifier.
- * @return A valid identifier.
- */
- [[nodiscard]] entity_type create(const entity_type hint) {
- return entities.emplace(hint);
- }
- /**
- * @brief Assigns each element in a range an identifier.
- *
- * @sa create
- *
- * @tparam It Type of forward iterator.
- * @param first An iterator to the first element of the range to generate.
- * @param last An iterator past the last element of the range to generate.
- */
- template<typename It>
- void create(It first, It last) {
- entities.insert(std::move(first), std::move(last));
- }
- /**
- * @brief Destroys an entity and releases its identifier.
- *
- * @warning
- * Adding or removing components to an entity that is being destroyed can
- * result in undefined behavior.
- *
- * @param entt A valid identifier.
- * @return The version of the recycled entity.
- */
- version_type destroy(const entity_type entt) {
- for(size_type pos = pools.size(); pos; --pos) {
- pools.begin()[pos - 1u].second->remove(entt);
- }
- entities.erase(entt);
- return entities.current(entt);
- }
- /**
- * @brief Destroys an entity and releases its identifier.
- *
- * The suggested version or the valid version closest to the suggested one
- * is used instead of the implicitly generated version.
- *
- * @sa destroy
- *
- * @param entt A valid identifier.
- * @param version A desired version upon destruction.
- * @return The version actually assigned to the entity.
- */
- version_type destroy(const entity_type entt, const version_type version) {
- destroy(entt);
- const auto elem = traits_type::construct(traits_type::to_entity(entt), version);
- return entities.bump((elem == tombstone) ? traits_type::next(elem) : elem);
- }
- /**
- * @brief Destroys all entities in a range and releases their identifiers.
- *
- * @sa destroy
- *
- * @tparam It Type of input iterator.
- * @param first An iterator to the first element of the range of entities.
- * @param last An iterator past the last element of the range of entities.
- */
- template<typename It>
- void destroy(It first, It last) {
- const auto from = entities.each().cbegin().base();
- const auto to = from + entities.pack(first, last);
- for(size_type pos = pools.size(); pos; --pos) {
- pools.begin()[pos - 1u].second->remove(from, to);
- }
- entities.erase(from, to);
- }
- /**
- * @brief Assigns the given component to an entity.
- *
- * The component must have a proper constructor or be of aggregate type.
- *
- * @warning
- * Attempting to assign a component to an entity that already owns it
- * results in undefined behavior.
- *
- * @tparam Type Type of component to create.
- * @tparam Args Types of arguments to use to construct the component.
- * @param entt A valid identifier.
- * @param args Parameters to use to initialize the component.
- * @return A reference to the newly created component.
- */
- template<typename Type, typename... Args>
- decltype(auto) emplace(const entity_type entt, Args &&...args) {
- return assure<Type>().emplace(entt, std::forward<Args>(args)...);
- }
- /**
- * @brief Assigns each entity in a range the given component.
- *
- * @sa emplace
- *
- * @tparam Type Type of component to create.
- * @tparam It Type of input iterator.
- * @param first An iterator to the first element of the range of entities.
- * @param last An iterator past the last element of the range of entities.
- * @param value An instance of the component to assign.
- */
- template<typename Type, typename It>
- void insert(It first, It last, const Type &value = {}) {
- assure<Type>().insert(std::move(first), std::move(last), value);
- }
- /**
- * @brief Assigns each entity in a range the given components.
- *
- * @sa emplace
- *
- * @tparam Type Type of component to create.
- * @tparam EIt Type of input iterator.
- * @tparam CIt Type of input iterator.
- * @param first An iterator to the first element of the range of entities.
- * @param last An iterator past the last element of the range of entities.
- * @param from An iterator to the first element of the range of components.
- */
- template<typename Type, typename EIt, typename CIt, typename = std::enable_if_t<std::is_same_v<typename std::iterator_traits<CIt>::value_type, Type>>>
- void insert(EIt first, EIt last, CIt from) {
- assure<Type>().insert(first, last, from);
- }
- /**
- * @brief Assigns or replaces the given component for an entity.
- *
- * @sa emplace
- * @sa replace
- *
- * @tparam Type Type of component to assign or replace.
- * @tparam Args Types of arguments to use to construct the component.
- * @param entt A valid identifier.
- * @param args Parameters to use to initialize the component.
- * @return A reference to the newly created component.
- */
- template<typename Type, typename... Args>
- decltype(auto) emplace_or_replace(const entity_type entt, Args &&...args) {
- if(auto &cpool = assure<Type>(); cpool.contains(entt)) {
- return cpool.patch(entt, [&args...](auto &...curr) { ((curr = Type{std::forward<Args>(args)...}), ...); });
- } else {
- return cpool.emplace(entt, std::forward<Args>(args)...);
- }
- }
- /**
- * @brief Patches the given component for an entity.
- *
- * The signature of the function should be equivalent to the following:
- *
- * @code{.cpp}
- * void(Type &);
- * @endcode
- *
- * @warning
- * Attempting to patch a component of an entity that doesn't own it
- * results in undefined behavior.
- *
- * @tparam Type Type of component to patch.
- * @tparam Func Types of the function objects to invoke.
- * @param entt A valid identifier.
- * @param func Valid function objects.
- * @return A reference to the patched component.
- */
- template<typename Type, typename... Func>
- decltype(auto) patch(const entity_type entt, Func &&...func) {
- return assure<Type>().patch(entt, std::forward<Func>(func)...);
- }
- /**
- * @brief Replaces the given component for an entity.
- *
- * The component must have a proper constructor or be of aggregate type.
- *
- * @warning
- * Attempting to replace a component of an entity that doesn't own it
- * results in undefined behavior.
- *
- * @tparam Type Type of component to replace.
- * @tparam Args Types of arguments to use to construct the component.
- * @param entt A valid identifier.
- * @param args Parameters to use to initialize the component.
- * @return A reference to the component being replaced.
- */
- template<typename Type, typename... Args>
- decltype(auto) replace(const entity_type entt, Args &&...args) {
- return patch<Type>(entt, [&args...](auto &...curr) { ((curr = Type{std::forward<Args>(args)...}), ...); });
- }
- /**
- * @brief Removes the given components from an entity.
- * @tparam Type Type of component to remove.
- * @tparam Other Other types of components to remove.
- * @param entt A valid identifier.
- * @return The number of components actually removed.
- */
- template<typename Type, typename... Other>
- size_type remove(const entity_type entt) {
- return (assure<Type>().remove(entt) + ... + assure<Other>().remove(entt));
- }
- /**
- * @brief Removes the given components from all the entities in a range.
- *
- * @sa remove
- *
- * @tparam Type Type of component to remove.
- * @tparam Other Other types of components to remove.
- * @tparam It Type of input iterator.
- * @param first An iterator to the first element of the range of entities.
- * @param last An iterator past the last element of the range of entities.
- * @return The number of components actually removed.
- */
- template<typename Type, typename... Other, typename It>
- size_type remove(It first, It last) {
- size_type count{};
- if constexpr(std::is_same_v<It, typename common_type::iterator>) {
- common_type *cpools[sizeof...(Other) + 1u]{&assure<Type>(), &assure<Other>()...};
- for(size_type pos{}, len = sizeof...(Other) + 1u; pos < len; ++pos) {
- if constexpr(sizeof...(Other) != 0u) {
- if(cpools[pos]->data() == first.data()) {
- std::swap(cpools[pos], cpools[sizeof...(Other)]);
- }
- }
- count += cpools[pos]->remove(first, last);
- }
- } else {
- for(auto cpools = std::forward_as_tuple(assure<Type>(), assure<Other>()...); first != last; ++first) {
- count += std::apply([entt = *first](auto &...curr) { return (curr.remove(entt) + ... + 0u); }, cpools);
- }
- }
- return count;
- }
- /**
- * @brief Erases the given components from an entity.
- *
- * @warning
- * Attempting to erase a component from an entity that doesn't own it
- * results in undefined behavior.
- *
- * @tparam Type Types of components to erase.
- * @tparam Other Other types of components to erase.
- * @param entt A valid identifier.
- */
- template<typename Type, typename... Other>
- void erase(const entity_type entt) {
- (assure<Type>().erase(entt), (assure<Other>().erase(entt), ...));
- }
- /**
- * @brief Erases the given components from all the entities in a range.
- *
- * @sa erase
- *
- * @tparam Type Types of components to erase.
- * @tparam Other Other types of components to erase.
- * @tparam It Type of input iterator.
- * @param first An iterator to the first element of the range of entities.
- * @param last An iterator past the last element of the range of entities.
- */
- template<typename Type, typename... Other, typename It>
- void erase(It first, It last) {
- if constexpr(std::is_same_v<It, typename common_type::iterator>) {
- common_type *cpools[sizeof...(Other) + 1u]{&assure<Type>(), &assure<Other>()...};
- for(size_type pos{}, len = sizeof...(Other) + 1u; pos < len; ++pos) {
- if constexpr(sizeof...(Other) != 0u) {
- if(cpools[pos]->data() == first.data()) {
- std::swap(cpools[pos], cpools[sizeof...(Other)]);
- }
- }
- cpools[pos]->erase(first, last);
- }
- } else {
- for(auto cpools = std::forward_as_tuple(assure<Type>(), assure<Other>()...); first != last; ++first) {
- std::apply([entt = *first](auto &...curr) { (curr.erase(entt), ...); }, cpools);
- }
- }
- }
- /**
- * @brief Erases components satisfying specific criteria from an entity.
- *
- * The function type is equivalent to:
- *
- * @code{.cpp}
- * void(const id_type, typename basic_registry<Entity>::base_type &);
- * @endcode
- *
- * Only storage where the entity exists are passed to the function.
- *
- * @tparam Func Type of the function object to invoke.
- * @param entt A valid identifier.
- * @param func A valid function object.
- */
- template<typename Func>
- void erase_if(const entity_type entt, Func func) {
- for(auto [id, cpool]: storage()) {
- if(cpool.contains(entt) && func(id, std::as_const(cpool))) {
- cpool.erase(entt);
- }
- }
- }
- /**
- * @brief Removes all tombstones from a registry or only the pools for the
- * given components.
- * @tparam Type Types of components for which to clear all tombstones.
- */
- template<typename... Type>
- void compact() {
- if constexpr(sizeof...(Type) == 0u) {
- for(auto &&curr: pools) {
- curr.second->compact();
- }
- } else {
- (assure<Type>().compact(), ...);
- }
- }
- /**
- * @brief Check if an entity is part of all the given storage.
- * @tparam Type Type of storage to check for.
- * @param entt A valid identifier.
- * @return True if the entity is part of all the storage, false otherwise.
- */
- template<typename... Type>
- [[nodiscard]] bool all_of([[maybe_unused]] const entity_type entt) const {
- if constexpr(sizeof...(Type) == 1u) {
- auto *cpool = assure<std::remove_const_t<Type>...>();
- return cpool && cpool->contains(entt);
- } else {
- return (all_of<Type>(entt) && ...);
- }
- }
- /**
- * @brief Check if an entity is part of at least one given storage.
- * @tparam Type Type of storage to check for.
- * @param entt A valid identifier.
- * @return True if the entity is part of at least one storage, false
- * otherwise.
- */
- template<typename... Type>
- [[nodiscard]] bool any_of([[maybe_unused]] const entity_type entt) const {
- return (all_of<Type>(entt) || ...);
- }
- /**
- * @brief Returns references to the given components for an entity.
- *
- * @warning
- * Attempting to get a component from an entity that doesn't own it results
- * in undefined behavior.
- *
- * @tparam Type Types of components to get.
- * @param entt A valid identifier.
- * @return References to the components owned by the entity.
- */
- template<typename... Type>
- [[nodiscard]] decltype(auto) get([[maybe_unused]] const entity_type entt) const {
- if constexpr(sizeof...(Type) == 1u) {
- return (assure<std::remove_const_t<Type>>()->get(entt), ...);
- } else {
- return std::forward_as_tuple(get<Type>(entt)...);
- }
- }
- /*! @copydoc get */
- template<typename... Type>
- [[nodiscard]] decltype(auto) get([[maybe_unused]] const entity_type entt) {
- if constexpr(sizeof...(Type) == 1u) {
- return (static_cast<storage_for_type<Type> &>(assure<std::remove_const_t<Type>>()).get(entt), ...);
- } else {
- return std::forward_as_tuple(get<Type>(entt)...);
- }
- }
- /**
- * @brief Returns a reference to the given component for an entity.
- *
- * In case the entity doesn't own the component, the parameters provided are
- * used to construct it.
- *
- * @sa get
- * @sa emplace
- *
- * @tparam Type Type of component to get.
- * @tparam Args Types of arguments to use to construct the component.
- * @param entt A valid identifier.
- * @param args Parameters to use to initialize the component.
- * @return Reference to the component owned by the entity.
- */
- template<typename Type, typename... Args>
- [[nodiscard]] decltype(auto) get_or_emplace(const entity_type entt, Args &&...args) {
- if(auto &cpool = assure<Type>(); cpool.contains(entt)) {
- return cpool.get(entt);
- } else {
- return cpool.emplace(entt, std::forward<Args>(args)...);
- }
- }
- /**
- * @brief Returns pointers to the given components for an entity.
- *
- * @note
- * The registry retains ownership of the pointed-to components.
- *
- * @tparam Type Types of components to get.
- * @param entt A valid identifier.
- * @return Pointers to the components owned by the entity.
- */
- template<typename... Type>
- [[nodiscard]] auto try_get([[maybe_unused]] const entity_type entt) const {
- if constexpr(sizeof...(Type) == 1u) {
- const auto *cpool = assure<std::remove_const_t<Type>...>();
- return (cpool && cpool->contains(entt)) ? std::addressof(cpool->get(entt)) : nullptr;
- } else {
- return std::make_tuple(try_get<Type>(entt)...);
- }
- }
- /*! @copydoc try_get */
- template<typename... Type>
- [[nodiscard]] auto try_get([[maybe_unused]] const entity_type entt) {
- if constexpr(sizeof...(Type) == 1u) {
- return (const_cast<Type *>(std::as_const(*this).template try_get<Type>(entt)), ...);
- } else {
- return std::make_tuple(try_get<Type>(entt)...);
- }
- }
- /**
- * @brief Clears a whole registry or the pools for the given components.
- * @tparam Type Types of components to remove from their entities.
- */
- template<typename... Type>
- void clear() {
- if constexpr(sizeof...(Type) == 0u) {
- for(size_type pos = pools.size(); pos; --pos) {
- pools.begin()[pos - 1u].second->clear();
- }
- const auto iterable = entities.each();
- entities.erase(iterable.begin().base(), iterable.end().base());
- } else {
- (assure<Type>().clear(), ...);
- }
- }
- /**
- * @brief Checks if an entity has components assigned.
- * @param entt A valid identifier.
- * @return True if the entity has no components assigned, false otherwise.
- */
- [[nodiscard]] bool orphan(const entity_type entt) const {
- return std::none_of(pools.cbegin(), pools.cend(), [entt](auto &&curr) { return curr.second->contains(entt); });
- }
- /**
- * @brief Returns a sink object for the given component.
- *
- * Use this function to receive notifications whenever a new instance of the
- * given component is created and assigned to an entity.<br/>
- * The function type for a listener is equivalent to:
- *
- * @code{.cpp}
- * void(basic_registry<Entity> &, Entity);
- * @endcode
- *
- * Listeners are invoked **after** assigning the component to the entity.
- *
- * @sa sink
- *
- * @tparam Type Type of component of which to get the sink.
- * @param id Optional name used to map the storage within the registry.
- * @return A temporary sink object.
- */
- template<typename Type>
- [[nodiscard]] auto on_construct(const id_type id = type_hash<Type>::value()) {
- return assure<Type>(id).on_construct();
- }
- /**
- * @brief Returns a sink object for the given component.
- *
- * Use this function to receive notifications whenever an instance of the
- * given component is explicitly updated.<br/>
- * The function type for a listener is equivalent to:
- *
- * @code{.cpp}
- * void(basic_registry<Entity> &, Entity);
- * @endcode
- *
- * Listeners are invoked **after** updating the component.
- *
- * @sa sink
- *
- * @tparam Type Type of component of which to get the sink.
- * @param id Optional name used to map the storage within the registry.
- * @return A temporary sink object.
- */
- template<typename Type>
- [[nodiscard]] auto on_update(const id_type id = type_hash<Type>::value()) {
- return assure<Type>(id).on_update();
- }
- /**
- * @brief Returns a sink object for the given component.
- *
- * Use this function to receive notifications whenever an instance of the
- * given component is removed from an entity and thus destroyed.<br/>
- * The function type for a listener is equivalent to:
- *
- * @code{.cpp}
- * void(basic_registry<Entity> &, Entity);
- * @endcode
- *
- * Listeners are invoked **before** removing the component from the entity.
- *
- * @sa sink
- *
- * @tparam Type Type of component of which to get the sink.
- * @param id Optional name used to map the storage within the registry.
- * @return A temporary sink object.
- */
- template<typename Type>
- [[nodiscard]] auto on_destroy(const id_type id = type_hash<Type>::value()) {
- return assure<Type>(id).on_destroy();
- }
- /**
- * @brief Returns a view for the given components.
- * @tparam Type Type of component used to construct the view.
- * @tparam Other Other types of components used to construct the view.
- * @tparam Exclude Types of components used to filter the view.
- * @return A newly created view.
- */
- template<typename Type, typename... Other, typename... Exclude>
- [[nodiscard]] basic_view<get_t<storage_for_type<const Type>, storage_for_type<const Other>...>, exclude_t<storage_for_type<const Exclude>...>>
- view(exclude_t<Exclude...> = exclude_t{}) const {
- const auto cpools = std::make_tuple(assure<std::remove_const_t<Type>>(), assure<std::remove_const_t<Other>>()..., assure<std::remove_const_t<Exclude>>()...);
- basic_view<get_t<storage_for_type<const Type>, storage_for_type<const Other>...>, exclude_t<storage_for_type<const Exclude>...>> elem{};
- std::apply([&elem](const auto *...curr) { ((curr ? elem.storage(*curr) : void()), ...); }, cpools);
- return elem;
- }
- /*! @copydoc view */
- template<typename Type, typename... Other, typename... Exclude>
- [[nodiscard]] basic_view<get_t<storage_for_type<Type>, storage_for_type<Other>...>, exclude_t<storage_for_type<Exclude>...>>
- view(exclude_t<Exclude...> = exclude_t{}) {
- return {assure<std::remove_const_t<Type>>(), assure<std::remove_const_t<Other>>()..., assure<std::remove_const_t<Exclude>>()...};
- }
- /**
- * @brief Returns a group for the given components.
- * @tparam Owned Types of storage _owned_ by the group.
- * @tparam Get Types of storage _observed_ by the group, if any.
- * @tparam Exclude Types of storage used to filter the group, if any.
- * @return A newly created group.
- */
- template<typename... Owned, typename... Get, typename... Exclude>
- basic_group<owned_t<storage_for_type<Owned>...>, get_t<storage_for_type<Get>...>, exclude_t<storage_for_type<Exclude>...>>
- group(get_t<Get...> = get_t{}, exclude_t<Exclude...> = exclude_t{}) {
- using handler_type = typename basic_group<owned_t<storage_for_type<Owned>...>, get_t<storage_for_type<Get>...>, exclude_t<storage_for_type<Exclude>...>>::handler;
- if(auto it = groups.find(type_hash<handler_type>::value()); it != groups.cend()) {
- return {*std::static_pointer_cast<handler_type>(it->second)};
- }
- std::shared_ptr<handler_type> handler{};
- if constexpr(sizeof...(Owned) == 0u) {
- handler = std::allocate_shared<handler_type>(get_allocator(), get_allocator(), assure<std::remove_const_t<Get>>()..., assure<std::remove_const_t<Exclude>>()...);
- } else {
- handler = std::allocate_shared<handler_type>(get_allocator(), assure<std::remove_const_t<Owned>>()..., assure<std::remove_const_t<Get>>()..., assure<std::remove_const_t<Exclude>>()...);
- [[maybe_unused]] const id_type elem[]{type_hash<std::remove_const_t<Owned>>::value()..., type_hash<std::remove_const_t<Get>>::value()..., type_hash<std::remove_const_t<Exclude>>::value()...};
- ENTT_ASSERT(std::all_of(groups.cbegin(), groups.cend(), [&elem](const auto &data) { return data.second->owned(elem, sizeof...(Owned)) == 0u; }), "Conflicting groups");
- }
- groups.emplace(type_hash<handler_type>::value(), handler);
- return {*handler};
- }
- /*! @copydoc group */
- template<typename... Owned, typename... Get, typename... Exclude>
- basic_group<owned_t<storage_for_type<const Owned>...>, get_t<storage_for_type<const Get>...>, exclude_t<storage_for_type<const Exclude>...>>
- group_if_exists(get_t<Get...> = get_t{}, exclude_t<Exclude...> = exclude_t{}) const {
- using handler_type = typename basic_group<owned_t<storage_for_type<const Owned>...>, get_t<storage_for_type<const Get>...>, exclude_t<storage_for_type<const Exclude>...>>::handler;
- if(auto it = groups.find(type_hash<handler_type>::value()); it != groups.cend()) {
- return {*std::static_pointer_cast<handler_type>(it->second)};
- }
- return {};
- }
- /**
- * @brief Checks whether the given components belong to any group.
- * @tparam Type Type of component in which one is interested.
- * @tparam Other Other types of components in which one is interested.
- * @return True if the pools of the given components are _free_, false
- * otherwise.
- */
- template<typename Type, typename... Other>
- [[nodiscard]] bool owned() const {
- const id_type elem[]{type_hash<std::remove_const_t<Type>>::value(), type_hash<std::remove_const_t<Other>>::value()...};
- return std::any_of(groups.cbegin(), groups.cend(), [&elem](auto &&data) { return data.second->owned(elem, 1u + sizeof...(Other)); });
- }
- /**
- * @brief Sorts the elements of a given component.
- *
- * The comparison function object returns `true` if the first element is
- * _less_ than the second one, `false` otherwise. Its signature is also
- * equivalent to one of the following:
- *
- * @code{.cpp}
- * bool(const Entity, const Entity);
- * bool(const Type &, const Type &);
- * @endcode
- *
- * Moreover, it shall induce a _strict weak ordering_ on the values.<br/>
- * The sort function object offers an `operator()` that accepts:
- *
- * * An iterator to the first element of the range to sort.
- * * An iterator past the last element of the range to sort.
- * * A comparison function object to use to compare the elements.
- *
- * The comparison function object hasn't necessarily the type of the one
- * passed along with the other parameters to this member function.
- *
- * @warning
- * Pools of components owned by a group cannot be sorted.
- *
- * @tparam Type Type of components to sort.
- * @tparam Compare Type of comparison function object.
- * @tparam Sort Type of sort function object.
- * @tparam Args Types of arguments to forward to the sort function object.
- * @param compare A valid comparison function object.
- * @param algo A valid sort function object.
- * @param args Arguments to forward to the sort function object, if any.
- */
- template<typename Type, typename Compare, typename Sort = std_sort, typename... Args>
- void sort(Compare compare, Sort algo = Sort{}, Args &&...args) {
- ENTT_ASSERT(!owned<Type>(), "Cannot sort owned storage");
- auto &cpool = assure<Type>();
- if constexpr(std::is_invocable_v<Compare, decltype(cpool.get({})), decltype(cpool.get({}))>) {
- auto comp = [&cpool, compare = std::move(compare)](const auto lhs, const auto rhs) { return compare(std::as_const(cpool.get(lhs)), std::as_const(cpool.get(rhs))); };
- cpool.sort(std::move(comp), std::move(algo), std::forward<Args>(args)...);
- } else {
- cpool.sort(std::move(compare), std::move(algo), std::forward<Args>(args)...);
- }
- }
- /**
- * @brief Sorts two pools of components in the same way.
- *
- * Entities and components in `To` which are part of both storage are sorted
- * internally with the order they have in `From`. The others follow in no
- * particular order.
- *
- * @warning
- * Pools of components owned by a group cannot be sorted.
- *
- * @tparam To Type of components to sort.
- * @tparam From Type of components to use to sort.
- */
- template<typename To, typename From>
- void sort() {
- ENTT_ASSERT(!owned<To>(), "Cannot sort owned storage");
- const base_type &cpool = assure<From>();
- assure<To>().sort_as(cpool.begin(), cpool.end());
- }
- /**
- * @brief Returns the context object, that is, a general purpose container.
- * @return The context object, that is, a general purpose container.
- */
- context &ctx() noexcept {
- return vars;
- }
- /*! @copydoc ctx */
- const context &ctx() const noexcept {
- return vars;
- }
- private:
- context vars;
- pool_container_type pools;
- group_container_type groups;
- storage_for_type<entity_type> entities;
- };
- } // namespace entt
- #endif
|