1 files changed, 38 insertions, 50 deletions

diff --git a/include/libcamera/signal.h b/include/libcamera/signal.h
index 432d95d0..c13bb30f 100644
--- a/include/libcamera/signal.h
+++ b/include/libcamera/signal.h
@@ -7,6 +7,7 @@
 #ifndef __LIBCAMERA_SIGNAL_H__
 #define __LIBCAMERA_SIGNAL_H__
 
+#include <functional>
 #include <list>
 #include <type_traits>
 #include <vector>
@@ -19,23 +20,18 @@ namespace libcamera {
 class SignalBase
 {
 public:
-	template<typename T>
-	void disconnect(T *obj)
-	{
-		for (auto iter = slots_.begin(); iter != slots_.end(); ) {
-			BoundMethodBase *slot = *iter;
-			if (slot->match(obj)) {
-				iter = slots_.erase(iter);
-				delete slot;
-			} else {
-				++iter;
-			}
-		}
-	}
+	void disconnect(Object *object);
 
 protected:
-	friend class Object;
-	std::list<BoundMethodBase *> slots_;
+	using SlotList = std::list<BoundMethodBase *>;
+
+	void connect(BoundMethodBase *slot);
+	void disconnect(std::function<bool(SlotList::iterator &)> match);
+
+	SlotList slots();
+
+private:
+	SlotList slots_;
 };
 
 template<typename... Args>
@@ -45,12 +41,7 @@ public:
 	Signal() {}
 	~Signal()
 	{
-		for (BoundMethodBase *slot : slots_) {
-			Object *object = slot->object();
-			if (object)
-				object->disconnect(this);
-			delete slot;
-		}
+		disconnect();
 	}
 
 #ifndef __DOXYGEN__
@@ -59,8 +50,7 @@ public:
 		     ConnectionType type = ConnectionTypeAuto)
 	{
 		Object *object = static_cast<Object *>(obj);
-		object->connect(this);
-		slots_.push_back(new BoundMethodMember<T, void, Args...>(obj, object, func, type));
+		SignalBase::connect(new BoundMethodMember<T, void, Args...>(obj, object, func, type));
 	}
 
 	template<typename T, typename R, typename std::enable_if<!std::is_base_of<Object, T>::value>::type * = nullptr>
@@ -69,63 +59,62 @@ public:
 #endif
 	void connect(T *obj, R (T::*func)(Args...))
 	{
-		slots_.push_back(new BoundMethodMember<T, R, Args...>(obj, nullptr, func));
+		SignalBase::connect(new BoundMethodMember<T, R, Args...>(obj, nullptr, func));
 	}
 
 	template<typename R>
 	void connect(R (*func)(Args...))
 	{
-		slots_.push_back(new BoundMethodStatic<R, Args...>(func));
+		SignalBase::connect(new BoundMethodStatic<R, Args...>(func));
 	}
 
 	void disconnect()
 	{
-		for (BoundMethodBase *slot : slots_)
-			delete slot;
-		slots_.clear();
+		SignalBase::disconnect([](SlotList::iterator &iter) {
+			return true;
+		});
 	}
 
 	template<typename T>
 	void disconnect(T *obj)
 	{
-		SignalBase::disconnect(obj);
+		SignalBase::disconnect([obj](SlotList::iterator &iter) {
+			return (*iter)->match(obj);
+		});
 	}
 
 	template<typename T, typename R>
 	void disconnect(T *obj, R (T::*func)(Args...))
 	{
-		for (auto iter = slots_.begin(); iter != slots_.end(); ) {
+		SignalBase::disconnect([obj, func](SlotList::iterator &iter) {
 			BoundMethodArgs<R, Args...> *slot =
 				static_cast<BoundMethodArgs<R, Args...> *>(*iter);
+
+			if (!slot->match(obj))
+				return false;
+
 			/*
 			 * If the object matches the slot, the slot is
 			 * guaranteed to be a member slot, so we can safely
 			 * cast it to BoundMethodMember<T, Args...> to match
 			 * func.
 			 */
-			if (slot->match(obj) &&
-			    static_cast<BoundMethodMember<T, R, Args...> *>(slot)->match(func)) {
-				iter = slots_.erase(iter);
-				delete slot;
-			} else {
-				++iter;
-			}
-		}
+			return static_cast<BoundMethodMember<T, R, Args...> *>(slot)->match(func);
+		});
 	}
 
 	template<typename R>
 	void disconnect(R (*func)(Args...))
 	{
-		for (auto iter = slots_.begin(); iter != slots_.end(); ) {
-			BoundMethodArgs<R, Args...> *slot = *iter;
-			if (slot->match(nullptr) &&
-			    static_cast<BoundMethodStatic<R, Args...> *>(slot)->match(func)) {
-				iter = slots_.erase(iter);
-				delete slot;
-			} else {
-				++iter;
-			}
-		}
+		SignalBase::disconnect([func](SlotList::iterator &iter) {
+			BoundMethodArgs<R, Args...> *slot =
+				static_cast<BoundMethodArgs<R, Args...> *>(*iter);
+
+			if (!slot->match(nullptr))
+				return false;
+
+			return static_cast<BoundMethodStatic<R, Args...> *>(slot)->match(func);
+		});
 	}
 
 	void emit(Args... args)
@@ -134,8 +123,7 @@ public:
 		 * Make a copy of the slots list as the slot could call the
 		 * disconnect operation, invalidating the iterator.
 		 */
-		std::vector<BoundMethodBase *> slots{ slots_.begin(), slots_.end() };
-		for (BoundMethodBase *slot : slots)
+		for (BoundMethodBase *slot : slots())
 			static_cast<BoundMethodArgs<void, Args...> *>(slot)->activate(args...);
 	}
 };