From 7c49ab4f044dfe81c8552ce044c8d05c02e525e9 Mon Sep 17 00:00:00 2001
From: Stephen Seo <seo.disparate@gmail.com>
Date: Tue, 20 Sep 2016 20:07:28 +0900
Subject: [PATCH] Added capability to store functions

EC::Manager can now store functions similar to functions
given to EC::Manager::forMatchingSignature.
---
 src/EC/Manager.hpp  | 48 +++++++++++++++++++++++++++++++++++++++++++++
 src/test/ECTest.cpp | 32 ++++++++++++++++++++++++++++++
 2 files changed, 80 insertions(+)

diff --git a/src/EC/Manager.hpp b/src/EC/Manager.hpp
index aa44a79..2338c8d 100644
--- a/src/EC/Manager.hpp
+++ b/src/EC/Manager.hpp
@@ -13,6 +13,7 @@
 #include <vector>
 #include <tuple>
 #include <utility>
+#include <functional>
 
 #include "Meta/Combine.hpp"
 #include "Meta/Matching.hpp"
@@ -226,6 +227,12 @@ namespace EC
                     ctype.template getEntityData<Types>(entityID)...
                 );
             }
+
+            template <typename CType, typename Function>
+            void callInstance(std::size_t entityID, CType& ctype, Function&& function) const
+            {
+                ForMatchingSignatureHelper<Types...>::call(entityID, ctype, std::forward<Function>(function));
+            }
         };
 
     public:
@@ -250,6 +257,47 @@ namespace EC
             }
         }
 
+    private:
+        std::vector<std::function<void()> > forMatchingFunctions;
+
+    public:
+        template <typename Signature, typename Function>
+        void addForMatchingFunction(Function&& function)
+        {
+            using SignatureComponents = typename EC::Meta::Matching<Signature, ComponentsList>::type;
+            using Helper = EC::Meta::Morph<SignatureComponents, ForMatchingSignatureHelper<> >;
+
+            Helper helper;
+            BitsetType signatureBitset = BitsetType::template generateBitset<Signature>();
+
+            forMatchingFunctions.emplace_back( [function, signatureBitset, helper, this] () {
+                for(std::size_t i = 0; i < this->currentSize; ++i)
+                {
+                    if(!std::get<bool>(this->entities[i]))
+                    {
+                        continue;
+                    }
+                    if((signatureBitset & std::get<BitsetType>(this->entities[i])) == signatureBitset)
+                    {
+                        helper.callInstance(i, *this, function);
+                    }
+                }
+            });
+        }
+
+        void callForMatchingFunctions()
+        {
+            for(auto functionIter = forMatchingFunctions.begin(); functionIter != forMatchingFunctions.end(); ++functionIter)
+            {
+                (*functionIter)();
+            }
+        }
+
+        void clearForMatchingFunctions()
+        {
+            forMatchingFunctions.clear();
+        }
+
     };
 }
 
diff --git a/src/test/ECTest.cpp b/src/test/ECTest.cpp
index ee72cd2..e93d929 100644
--- a/src/test/ECTest.cpp
+++ b/src/test/ECTest.cpp
@@ -242,3 +242,35 @@ TEST(EC, DeletedEntities)
     }
 }
 
+TEST(EC, FunctionStorage)
+{
+    EC::Manager<ListComponentsAll, ListTagsAll> manager;
+    auto eid = manager.addEntity();
+    manager.addComponent<C0>(eid);
+
+    manager.addForMatchingFunction<EC::Meta::TypeList<C0> >( [] (std::size_t eid, C0& c0) {
+        c0.x = 1;
+        c0.y = 2;
+    });
+
+    manager.callForMatchingFunctions();
+
+    {
+        auto c0 = manager.getEntityData<C0>(eid);
+
+        EXPECT_EQ(c0.x, 1);
+        EXPECT_EQ(c0.y, 2);
+    }
+
+    manager.clearForMatchingFunctions();
+
+    manager.callForMatchingFunctions();
+
+    {
+        auto c0 = manager.getEntityData<C0>(eid);
+
+        EXPECT_EQ(c0.x, 1);
+        EXPECT_EQ(c0.y, 2);
+    }
+}
+