From 5d2fae5896ba2caee0ad460c1914a5771e055029 Mon Sep 17 00:00:00 2001
From: Wojtek Figat <wojtek@figat.pl>
Date: Thu, 31 Aug 2023 14:01:30 +0200
Subject: [PATCH] Add `GetColliders` utility to `RigidBody`

---
 Source/Engine/Physics/Actors/RigidBody.h | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/Source/Engine/Physics/Actors/RigidBody.h b/Source/Engine/Physics/Actors/RigidBody.h
index 8108256c6..e203e70fc 100644
--- a/Source/Engine/Physics/Actors/RigidBody.h
+++ b/Source/Engine/Physics/Actors/RigidBody.h
@@ -519,6 +519,17 @@ public:
     /// </summary>
     void UpdateScale();
 
+    template<typename ColliderType = Collider, typename AllocationType = HeapAllocation>
+    void GetColliders(Array<ColliderType*, AllocationType>& result) const
+    {
+        for (int32 i = 0; i < Children.Count(); i++)
+        {
+            const auto collider = Cast<ColliderType>(Children.Get()[i]);
+            if (collider && collider->GetAttachedRigidBody() == this)
+                result.Add(collider);
+        }
+    }
+
 public:
     // [Actor]
     void Serialize(SerializeStream& stream, const void* otherObj) override;