From b8352b0fbe7c3d61cd0452900dea9ea963400520 Mon Sep 17 00:00:00 2001
From: lethal-guitar <lethal_guitar128@web.de>
Date: Wed, 28 Dec 2016 15:50:28 +0100
Subject: Add dereference operator to ComponentHandle

This makes it possible to write code like this:

  auto& componentRef = *someEntity.component<Position>();

This was already possible previously, but required calling .get() on the
component handle.
---
 entityx/Entity.h       | 15 +++++++++++++++
 entityx/Entity_test.cc | 10 ++++++++++
 2 files changed, 25 insertions(+)

diff --git a/entityx/Entity.h b/entityx/Entity.h
index e4a3871..7c69e0a 100644
--- a/entityx/Entity.h
+++ b/entityx/Entity.h
@@ -193,6 +193,9 @@ public:
   C *operator -> ();
   const C *operator -> () const;
 
+  C &operator * ();
+  const C &operator * () const;
+
   C *get();
   const C *get() const;
 
@@ -1045,6 +1048,18 @@ inline const C *ComponentHandle<C, EM>::operator -> () const {
   return manager_->template get_component_ptr<C>(id_);
 }
 
+template <typename C, typename EM>
+inline C &ComponentHandle<C, EM>::operator * () {
+  assert(valid());
+  return *manager_->template get_component_ptr<C>(id_);
+}
+
+template <typename C, typename EM>
+inline const C &ComponentHandle<C, EM>::operator * () const {
+  assert(valid());
+  return *manager_->template get_component_ptr<C>(id_);
+}
+
 template <typename C, typename EM>
 inline C *ComponentHandle<C, EM>::get() {
   assert(valid());
diff --git a/entityx/Entity_test.cc b/entityx/Entity_test.cc
index 9e75ee3..f985794 100644
--- a/entityx/Entity_test.cc
+++ b/entityx/Entity_test.cc
@@ -668,3 +668,13 @@ TEST_CASE_METHOD(EntityManagerFixture, "TestViewEach") {
   });
   REQUIRE(count == 1);
 }
+
+TEST_CASE_METHOD(EntityManagerFixture, "TestComponentDereference") {
+  Entity a = em.create();
+  a.assign<Position>(10, 5);
+  auto& positionRef = *a.component<Position>();
+  REQUIRE(positionRef.x == 10);
+  REQUIRE(positionRef.y == 5);
+  positionRef.y = 20;
+  REQUIRE(a.component<Position>()->y == 20);
+}
-- 
cgit v1.2.3