component access

This commit is contained in:
2018-12-25 20:38:17 +07:00
parent e29aa3d4eb
commit 078c6c0a4c
2 changed files with 176 additions and 3 deletions

130
ecs.hpp
View File

@@ -14,6 +14,8 @@
#include <memory>
#include <limits>
#include <utility>
#include <exception>
#include <stdexcept>
#include <functional>
#include <type_traits>
#include <unordered_set>
@@ -100,6 +102,8 @@ namespace ecs_hpp
void assign(entity_id id, Args&&... args);
bool remove(entity_id id) noexcept override;
bool exists(entity_id id) const noexcept override;
T* find(entity_id id) noexcept;
const T* find(entity_id id) const noexcept;
private:
std::unordered_map<entity_id, T> components_;
};
@@ -119,9 +123,40 @@ namespace ecs_hpp
bool component_storage<T>::exists(entity_id id) const noexcept {
return components_.find(id) != components_.end();
}
template < typename T >
T* component_storage<T>::find(entity_id id) noexcept {
const auto iter = components_.find(id);
return iter != components_.end()
? &iter->second
: nullptr;
}
template < typename T >
const T* component_storage<T>::find(entity_id id) const noexcept {
const auto iter = components_.find(id);
return iter != components_.end()
? &iter->second
: nullptr;
}
}
}
// -----------------------------------------------------------------------------
//
// exceptions
//
// -----------------------------------------------------------------------------
namespace ecs_hpp
{
class basic_exception : public std::logic_error {
public:
basic_exception(const char* msg)
: std::logic_error(msg) {}
};
}
// -----------------------------------------------------------------------------
//
// entity
@@ -151,6 +186,18 @@ namespace ecs_hpp
bool exists_component() const noexcept;
std::size_t remove_all_components() noexcept;
template < typename T >
T& get_component();
template < typename T >
const T& get_component() const;
template < typename T >
T* find_component() noexcept;
template < typename T >
const T* find_component() const noexcept;
private:
world& owner_;
entity_id id_{0u};
@@ -199,9 +246,24 @@ namespace ecs_hpp
bool exists_component(const entity& ent) const noexcept;
std::size_t remove_all_components(const entity& ent) const noexcept;
template < typename T >
T& get_component(const entity& ent);
template < typename T >
const T& get_component(const entity& ent) const;
template < typename T >
T* find_component(const entity& ent) noexcept;
template < typename T >
const T* find_component(const entity& ent) const noexcept;
private:
template < typename T >
detail::component_storage<T>* find_storage_() const;
detail::component_storage<T>* find_storage_() noexcept;
template < typename T >
const detail::component_storage<T>* find_storage_() const noexcept;
template < typename T >
detail::component_storage<T>& get_or_create_storage_();
@@ -271,6 +333,26 @@ namespace ecs_hpp
return owner_.remove_all_components(*this);
}
template < typename T >
T& entity::get_component() {
return owner_.get_component<T>(*this);
}
template < typename T >
const T& entity::get_component() const {
return owner_.get_component<T>(*this);
}
template < typename T >
T* entity::find_component() noexcept {
return owner_.find_component<T>(*this);
}
template < typename T >
const T* entity::find_component() const noexcept {
return owner_.find_component<T>(*this);
}
inline bool operator==(const entity& l, const entity& r) noexcept {
return &l.owner() == &r.owner()
&& l.id() == r.id();
@@ -353,7 +435,41 @@ namespace ecs_hpp
}
template < typename T >
detail::component_storage<T>* world::find_storage_() const {
T& world::get_component(const entity& ent) {
T* component = find_component<T>(ent);
if ( component ) {
return *component;
}
throw basic_exception("component not found");
}
template < typename T >
const T& world::get_component(const entity& ent) const {
const T* component = find_component<T>(ent);
if ( component ) {
return *component;
}
throw basic_exception("component not found");
}
template < typename T >
T* world::find_component(const entity& ent) noexcept {
detail::component_storage<T>* storage = find_storage_<T>();
return storage
? storage->find(ent.id())
: nullptr;
}
template < typename T >
const T* world::find_component(const entity& ent) const noexcept {
const detail::component_storage<T>* storage = find_storage_<T>();
return storage
? storage->find(ent.id())
: nullptr;
}
template < typename T >
detail::component_storage<T>* world::find_storage_() noexcept {
const auto family = detail::type_family<T>::id();
const auto iter = storages_.find(family);
if ( iter != storages_.end() ) {
@@ -362,6 +478,16 @@ namespace ecs_hpp
return nullptr;
}
template < typename T >
const detail::component_storage<T>* world::find_storage_() const noexcept {
const auto family = detail::type_family<T>::id();
const auto iter = storages_.find(family);
if ( iter != storages_.end() ) {
return static_cast<const detail::component_storage<T>*>(iter->second.get());
}
return nullptr;
}
template < typename T >
detail::component_storage<T>& world::get_or_create_storage_() {
detail::component_storage<T>* storage = find_storage_<T>();

View File

@@ -79,7 +79,7 @@ TEST_CASE("world") {
REQUIRE_FALSE(w.destroy_entity(e2));
}
}
SECTION("components") {
SECTION("component_assigning") {
{
ecs::world w;
ecs::entity e1 = w.create_entity();
@@ -147,4 +147,51 @@ TEST_CASE("world") {
REQUIRE_FALSE(w.exists_component<position_c>(e1));
}
}
SECTION("component_accessing") {
{
ecs::world w;
auto e1 = w.create_entity();
auto e2 = w.create_entity();
REQUIRE_FALSE(e1.find_component<position_c>());
REQUIRE_FALSE(e2.find_component<velocity_c>());
e1.assign_component<position_c>(1, 2);
e2.assign_component<velocity_c>(3, 4);
REQUIRE(e1.get_component<position_c>().x == 1);
REQUIRE(e1.get_component<position_c>().y == 2);
REQUIRE(e2.get_component<velocity_c>().dx == 3);
REQUIRE(e2.get_component<velocity_c>().dy == 4);
REQUIRE_THROWS_AS(e1.get_component<velocity_c>(), ecs::basic_exception);
REQUIRE_THROWS_AS(e2.get_component<position_c>(), ecs::basic_exception);
}
{
ecs::world w;
const auto e1 = w.create_entity();
const auto e2 = w.create_entity();
REQUIRE_FALSE(e1.find_component<position_c>());
REQUIRE_FALSE(e2.find_component<velocity_c>());
w.assign_component<position_c>(e1, 1, 2);
w.assign_component<velocity_c>(e2, 3, 4);
{
const ecs::world& ww = w;
REQUIRE(ww.get_component<position_c>(e1).x == 1);
REQUIRE(ww.get_component<position_c>(e1).y == 2);
REQUIRE(ww.get_component<velocity_c>(e2).dx == 3);
REQUIRE(ww.get_component<velocity_c>(e2).dy == 4);
REQUIRE_THROWS_AS(ww.get_component<velocity_c>(e1), ecs::basic_exception);
REQUIRE_THROWS_AS(ww.get_component<position_c>(e2), ecs::basic_exception);
}
}
}
}