From dc27330752293bb58ebe47b8d575b22002386294 Mon Sep 17 00:00:00 2001
From: Haoyu Qiu <timothyqiu32@gmail.com>
Date: Fri, 13 Jan 2023 18:19:26 +0800
Subject: [PATCH] Make CollisionShape selection box use shape AABB

---
 scene/3d/collision_shape.cpp | 12 ++++++++++++
 scene/3d/collision_shape.h   |  4 ++++
 2 files changed, 16 insertions(+)

diff --git a/scene/3d/collision_shape.cpp b/scene/3d/collision_shape.cpp
index c136313a8..0820fc539 100644
--- a/scene/3d/collision_shape.cpp
+++ b/scene/3d/collision_shape.cpp
@@ -143,6 +143,18 @@ String CollisionShape::get_configuration_warning() const {
 	return warning;
 }
 
+#ifdef TOOLS_ENABLED
+AABB CollisionShape::get_fallback_gizmo_aabb() const {
+	if (shape.is_null()) {
+		return Spatial::get_fallback_gizmo_aabb();
+	}
+
+	// get_debug_mesh() is not const because the mesh is lazy initialized and cached.
+	// It would be better if we can mark the cache mutable and make get_debug_mesh() const.
+	return const_cast<CollisionShape *>(this)->shape->get_debug_mesh()->get_aabb();
+}
+#endif
+
 void CollisionShape::_bind_methods() {
 	//not sure if this should do anything
 	ClassDB::bind_method(D_METHOD("resource_changed", "resource"), &CollisionShape::resource_changed);
diff --git a/scene/3d/collision_shape.h b/scene/3d/collision_shape.h
index 1f2b9e942..806a1eedc 100644
--- a/scene/3d/collision_shape.h
+++ b/scene/3d/collision_shape.h
@@ -66,6 +66,10 @@ public:
 
 	String get_configuration_warning() const;
 
+#ifdef TOOLS_ENABLED
+	virtual AABB get_fallback_gizmo_aabb() const;
+#endif
+
 	CollisionShape();
 	~CollisionShape();
 };