BlackMATov/ecs.hpp
1
0
Fork 0
mirror of https://github.com/BlackMATov/ecs.hpp.git synced 2025-12-13 10:35:39 +07:00
Code Issues Packages Projects Releases Wiki Activity

registry: for_each_entity

Browse Source
This commit is contained in:
Matvey Cherevko
2018-12-29 03:03:24 +07:00
parent 8f87d23c9b
commit f9c6ab4e9d
2 changed files with 49 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
ecs.hpp
View File

@@ -761,6 +761,11 @@ namespace ecs_hpp
template < typename... Ts >
std::tuple<const Ts*...> find_components(const entity& ent) const noexcept;
template < typename F >
void for_each_entity(F&& f);
template < typename F >
void for_each_entity(F&& f) const;
template < typename T, typename F >
void for_each_component(F&& f);
template < typename T, typename F >
@@ -1071,6 +1076,20 @@ namespace ecs_hpp
return std::make_tuple(find_component<Ts>(ent)...);
}
template < typename F >
void registry::for_each_entity(F&& f) {
for ( const auto id : entity_ids_ ) {
f(entity(*this, id));
}
}
template < typename F >
void registry::for_each_entity(F&& f) const {
for ( const auto id : entity_ids_ ) {
f(entity(const_cast<registry&>(*this), id));
}
}
template < typename T, typename F >
void registry::for_each_component(F&& f) {
detail::component_storage<T>* storage = find_storage_<T>();
@@ -1117,8 +1136,9 @@ namespace ecs_hpp
detail::component_storage<T>* registry::find_storage_() noexcept {
const auto family = detail::type_family<T>::id();
using raw_storage_ptr = detail::component_storage<T>*;
return storages_.has(family)
? static_cast<raw_storage_ptr>(storages_.get(family).get())
const storage_uptr* storage_uptr_ptr = storages_.find(family);
return storage_uptr_ptr && *storage_uptr_ptr
? static_cast<raw_storage_ptr>(storage_uptr_ptr->get())
: nullptr;
}
@@ -1126,8 +1146,9 @@ namespace ecs_hpp
const detail::component_storage<T>* registry::find_storage_() const noexcept {
const auto family = detail::type_family<T>::id();
using raw_storage_ptr = const detail::component_storage<T>*;
return storages_.has(family)
? static_cast<raw_storage_ptr>(storages_.get(family).get())
const storage_uptr* storage_uptr_ptr = storages_.find(family);
return storage_uptr_ptr && *storage_uptr_ptr
? static_cast<raw_storage_ptr>(storage_uptr_ptr->get())
: nullptr;
}

24
ecs_tests.cpp
View File

@@ -565,6 +565,30 @@ TEST_CASE("registry") {
REQUIRE(e1.get_component<velocity_c>().y == 40);
}
}
SECTION("for_each_entity") {
{
ecs::registry w;
auto e1 = w.create_entity();
auto e2 = w.create_entity();
{
ecs::entity_id acc1 = 0;
w.for_each_entity([&acc1](const ecs::entity& e){
acc1 += e.id();
});
REQUIRE(acc1 == e1.id() + e2.id());
}
{
const ecs::registry& ww = w;
ecs::entity_id acc1 = 0;
ww.for_each_entity([&acc1](const ecs::entity& e){
acc1 += e.id();
});
REQUIRE(acc1 == e1.id() + e2.id());
}
}
}
SECTION("for_each_component") {
{
ecs::registry w;