Merge pull request #79 from sansumbrella/component_replace

Entity::replace<Component> method.

1 files changed, 16 insertions, 0 deletions

diff --git a/entityx/Entity.h b/entityx/Entity.h
index fa88726..85033c9 100644
--- a/entityx/Entity.h
+++ b/entityx/Entity.h
@@ -135,6 +135,9 @@ public:
   template <typename C>
   ComponentHandle<C> assign_from_copy(const C &component);
 
+  template <typename C, typename ... Args>
+  ComponentHandle<C> replace(Args && ... args);
+
   template <typename C>
   void remove();
 
@@ -865,6 +868,19 @@ ComponentHandle<C> Entity::assign_from_copy(const C &component) {
   return manager_->assign<C>(id_, std::forward<const C &>(component));
 }
 
+template <typename C, typename ... Args>
+ComponentHandle<C> Entity::replace(Args && ... args) {
+  assert(valid());
+  auto handle = component<C>();
+  if (handle) {
+    *(handle.get()) = C(std::forward<Args>(args) ...);
+  }
+  else {
+    handle = manager_->assign<C>(id_, std::forward<Args>(args) ...);
+  }
+  return handle;
+}
+
 template <typename C>
 void Entity::remove() {
   assert(valid() && has_component<C>());