X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/1847d1441271d076b3de449c8853031ea208ce8f..91d05079e37bf2fdd32ee099cff06e2cae7c74cd:/include/xbt/signal.hpp

diff --git a/include/xbt/signal.hpp b/include/xbt/signal.hpp
index d5ad826fa2..36bd87f2dc 100644
--- a/include/xbt/signal.hpp
+++ b/include/xbt/signal.hpp
@@ -1,4 +1,4 @@
-/* Copyright (c) 2014-2017. The SimGrid Team. All rights reserved.          */
+/* Copyright (c) 2014-2020. The SimGrid Team. All rights reserved.          */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
@@ -7,8 +7,8 @@
 #define SIMGRID_XBT_SIGNAL_HPP
 
 #include <functional>
+#include <map>
 #include <utility>
-#include <vector>
 
 namespace simgrid {
 namespace xbt {
@@ -23,24 +23,24 @@ namespace xbt {
   */
   template<class R, class... P>
   class signal<R(P...)> {
-  private:
-    typedef std::function<R(P...)> callback_type;
-    std::vector<callback_type> handlers_;
+    using callback_type = std::function<R(P...)>;
+    std::map<unsigned int, callback_type> handlers_;
+    unsigned int callback_sequence_id = 0;
+
   public:
-    template<class U>
-    void connect(U slot)
+    template <class U> unsigned int connect(U slot)
     {
-      handlers_.push_back(std::move(slot));
+      handlers_.insert({callback_sequence_id, std::move(slot)});
+      return callback_sequence_id++;
     }
     R operator()(P... args) const
     {
       for (auto const& handler : handlers_)
-        handler(args...);
-    }
-    void disconnect_all_slots()
-    {
-      handlers_.clear();
+        handler.second(args...);
     }
+    void disconnect(unsigned int id) { handlers_.erase(id); }
+    void disconnect_slots() { handlers_.clear(); }
+    int get_slot_count() { return handlers_.size(); }
   };
 
 }