find_child_of_type

Author: Ryan-000

modules/gdscript/gdscript_utility_functions.cpp

@@ -609,6 +609,21 @@ struct GDScriptUtilityFunctionsDefinitions {
 	}
 };
 
+// expose is_instance_of to C++
+bool GDScriptUtilityFunctions::is_instance_of(const Variant &p_value, const Variant &p_type) {
+	Variant ret;
+	const Variant *args[2] = { &p_value, &p_type };
+	Callable::CallError error;
+	GDScriptUtilityFunctionsDefinitions::is_instance_of(&ret, args, 2, error);
+
+	// if there was an error, return false
+	if (error.error != Callable::CallError::CALL_OK) {
+		return false;
+	}
+
+	return ret;
+}
+
 struct GDScriptUtilityFunctionInfo {
 	GDScriptUtilityFunctions::FunctionPtr function = nullptr;
 	MethodInfo info;

modules/gdscript/gdscript_utility_functions.h

@@ -56,6 +56,7 @@ class GDScriptUtilityFunctions {
 
 	static void register_functions();
 	static void unregister_functions();
+	static bool is_instance_of(const Variant &p_value, const Variant &p_type);
 };
 
 #endif // GDSCRIPT_UTILITY_FUNCTIONS_H

scene/main/node.cpp

@@ -42,6 +42,7 @@
 #include "scene/main/window.h"
 #include "scene/resources/packed_scene.h"
 #include "viewport.h"
+#include "modules/gdscript/gdscript_utility_functions.h"
 
 #include <stdint.h>
 
@@ -1827,6 +1828,30 @@ bool Node::has_node(const NodePath &p_path) const {
 	return get_node_or_null(p_path) != nullptr;
 }
 
+
+Node *Node::find_child_of_type(const Variant &p_type) const {
+	ERR_THREAD_GUARD_V(nullptr);
+	_update_children_cache();
+	Node *const *cptr = data.children_cache.ptr();
+	int ccount = data.children_cache.size();
+	for (int i = 0; i < ccount; i++) {
+		Node* node = cptr[i];
+
+		// if p_type is a string, check if the node is an instance of the class
+		if (p_type.get_type() == Variant::STRING || p_type.get_type() == Variant::STRING_NAME) {
+			if (node->is_class(p_type)) {
+				return node;
+			}
+		} else {
+			// if p_type is a class, check if the node is an instance of the class
+			if (GDScriptUtilityFunctions::is_instance_of(node, p_type)) {
+				return node;
+			}
+		}
+	}
+	return nullptr;
+}
+
 // Finds the first child node (in tree order) whose name matches the given pattern.
 // Can be recursive or not, and limited to owned nodes.
 Node *Node::find_child(const String &p_pattern, bool p_recursive, bool p_owned) const {
@@ -3536,6 +3561,7 @@ void Node::_bind_methods() {
 	ClassDB::bind_method(D_METHOD("get_node", "path"), &Node::get_node);
 	ClassDB::bind_method(D_METHOD("get_node_or_null", "path"), &Node::get_node_or_null);
 	ClassDB::bind_method(D_METHOD("get_parent"), &Node::get_parent);
+	ClassDB::bind_method(D_METHOD("find_child_of_type", "type"), &Node::find_child_of_type);
 	ClassDB::bind_method(D_METHOD("find_child", "pattern", "recursive", "owned"), &Node::find_child, DEFVAL(true), DEFVAL(true));
 	ClassDB::bind_method(D_METHOD("find_children", "pattern", "type", "recursive", "owned"), &Node::find_children, DEFVAL(""), DEFVAL(true), DEFVAL(true));
 	ClassDB::bind_method(D_METHOD("find_parent", "pattern"), &Node::find_parent);
@@ -3973,4 +3999,5 @@ bool Node::is_connected(const StringName &p_signal, const Callable &p_callable)
 	return Object::is_connected(p_signal, p_callable);
 }
 
+
 #endif

scene/main/node.h

@@ -455,6 +455,7 @@ class Node : public Object {
 	bool has_node(const NodePath &p_path) const;
 	Node *get_node(const NodePath &p_path) const;
 	Node *get_node_or_null(const NodePath &p_path) const;
+	Node *find_child_of_type(const Variant &p_type) const;
 	Node *find_child(const String &p_pattern, bool p_recursive = true, bool p_owned = true) const;
 	TypedArray<Node> find_children(const String &p_pattern, const String &p_type = "", bool p_recursive = true, bool p_owned = true) const;
 	bool has_node_and_resource(const NodePath &p_path) const;