From 7b44e4f20b91a64b778bcc8b2f00602d4ba13bd2 Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Sun, 23 Mar 2025 18:46:10 +0000
Subject: Support using typed collections for subclass filtering

---
 lib/collection.h | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

(limited to 'lib')

diff --git a/lib/collection.h b/lib/collection.h
index d32cff1..b786e2f 100644
--- a/lib/collection.h
+++ b/lib/collection.h
@@ -235,13 +235,22 @@ protected:
 				...);
 	}
 
+	template<typename T>
+		requires((std::is_convertible_v<T *, Others *> || ...))
+	[[nodiscard]] consteval static bool
+	idx()
+	{
+		size_t typeIdx = 0;
+		return ((typeIdx++ && std::is_convertible_v<Others *, T *>) || ...);
+	}
+
 	template<typename T>
 	[[nodiscard]]
-	const auto &
+	constexpr const auto &
 	containerFor() const
 	{
 		if constexpr ((std::is_convertible_v<T *, Others *> || ...)) {
-			return std::get<OtherObjects<T>>(otherObjects);
+			return std::get<idx<T>()>(otherObjects);
 		}
 		else {
 			return objects;
@@ -260,7 +269,7 @@ protected:
 		}
 		else {
 			return std::count_if(begin, end, [&m, &params...](auto && op) {
-				if (auto o = dynamic_cast<T *>(op.get())) {
+				if (auto o = dynamic_cast<T *>(std::to_address(op))) {
 					std::invoke(m, o, std::forward<Params>(params)...);
 					return true;
 				}
@@ -280,7 +289,7 @@ protected:
 		}
 		else {
 			return std::find_if(begin, end, [&m, &params...](auto && op) {
-				if (auto o = dynamic_cast<T *>(op.get())) {
+				if (auto o = dynamic_cast<T *>(std::to_address(op))) {
 					return std::invoke(m, o, std::forward<Params>(params)...);
 				}
 				return false;
-- 
cgit v1.2.3