From 7e609984b47a211d4b5b85c9714495e926df718f Mon Sep 17 00:00:00 2001
From: Wojtek Figat <wojtek@figat.pl>
Date: Thu, 28 Apr 2022 22:44:56 +0200
Subject: [PATCH] Add `TypeReference` attribute to actors/scripts searching
 utilities for easier type picking in Visual Script

---
 Source/Engine/Level/Actor.h | 12 ++++++------
 Source/Engine/Level/Level.h |  8 ++++----
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/Source/Engine/Level/Actor.h b/Source/Engine/Level/Actor.h
index 85dccfe13..ef4e0fc36 100644
--- a/Source/Engine/Level/Actor.h
+++ b/Source/Engine/Level/Actor.h
@@ -208,7 +208,7 @@ public:
     /// </summary>
     /// <param name="type">Type of the actor to search for. Includes any actors derived from the type.</param>
     /// <returns>The child actor or null.</returns>
-    API_FUNCTION() Actor* GetChild(const MClass* type) const;
+    API_FUNCTION() Actor* GetChild(API_PARAM(Attributes="TypeReference(typeof(Actor))") const MClass* type) const;
 
     /// <summary>
     /// Gets the child actor of the given type.
@@ -225,7 +225,7 @@ public:
     /// </summary>
     /// <param name="type">Type of the actor to search for. Includes any actors derived from the type.</param>
     /// <returns>The child actors.</returns>
-    API_FUNCTION() Array<Actor*> GetChildren(const MClass* type) const;
+    API_FUNCTION() Array<Actor*> GetChildren(API_PARAM(Attributes="TypeReference(typeof(Actor))") const MClass* type) const;
 
     /// <summary>
     /// Gets the child actors of the given type.
@@ -271,7 +271,7 @@ public:
     /// </summary>
     /// <param name="type">Type of the script to search for. Includes any scripts derived from the type.</param>
     /// <returns>The script or null.</returns>
-    API_FUNCTION() Script* GetScript(const MClass* type) const;
+    API_FUNCTION() Script* GetScript(API_PARAM(Attributes="TypeReference(typeof(Script))") const MClass* type) const;
 
     /// <summary>
     /// Gets the script of the given type from this actor.
@@ -288,7 +288,7 @@ public:
     /// </summary>
     /// <param name="type">Type of the script to search for. Includes any scripts derived from the type.</param>
     /// <returns>The scripts.</returns>
-    API_FUNCTION() Array<Script*> GetScripts(const MClass* type) const;
+    API_FUNCTION() Array<Script*> GetScripts(API_PARAM(Attributes="TypeReference(typeof(Script))") const MClass* type) const;
 
     /// <summary>
     /// Gets the scripts of the given type from this actor.
@@ -712,7 +712,7 @@ public:
     /// </summary>
     /// <param name="type">Type of the actor to search for. Includes any actors derived from the type.</param>
     /// <returns>Actor instance if found, null otherwise.</returns>
-    API_FUNCTION() Actor* FindActor(const MClass* type) const;
+    API_FUNCTION() Actor* FindActor(API_PARAM(Attributes="TypeReference(typeof(Actor))") const MClass* type) const;
 
     /// <summary>
     /// Tries to find the actor of the given type in this actor hierarchy (checks this actor and all children hierarchy).
@@ -729,7 +729,7 @@ public:
     /// </summary>
     /// <param name="type">Type of the actor to search for. Includes any actors derived from the type.</param>
     /// <returns>Script instance if found, null otherwise.</returns>
-    API_FUNCTION() Script* FindScript(const MClass* type) const;
+    API_FUNCTION() Script* FindScript(API_PARAM(Attributes="TypeReference(typeof(Script))") const MClass* type) const;
 
     /// <summary>
     /// Tries to find the script of the given type in this actor hierarchy (checks this actor and all children hierarchy).
diff --git a/Source/Engine/Level/Level.h b/Source/Engine/Level/Level.h
index 34691c474..8a6e53aae 100644
--- a/Source/Engine/Level/Level.h
+++ b/Source/Engine/Level/Level.h
@@ -362,7 +362,7 @@ public:
     /// </summary>
     /// <param name="type">Type of the actor to search for. Includes any actors derived from the type.</param>
     /// <returns>Found actor or null.</returns>
-    API_FUNCTION() static Actor* FindActor(const MClass* type);
+    API_FUNCTION() static Actor* FindActor(API_PARAM(Attributes="TypeReference(typeof(Actor))") const MClass* type);
 
     /// <summary>
     /// Tries to find the actor of the given type in all the loaded scenes.
@@ -379,7 +379,7 @@ public:
     /// </summary>
     /// <param name="type">Type of the script to search for. Includes any scripts derived from the type.</param>
     /// <returns>Found script or null.</returns>
-    API_FUNCTION() static Script* FindScript(const MClass* type);
+    API_FUNCTION() static Script* FindScript(API_PARAM(Attributes="TypeReference(typeof(Script))") const MClass* type);
 
     /// <summary>
     /// Tries to find the script of the given type in all the loaded scenes.
@@ -396,14 +396,14 @@ public:
     /// </summary>
     /// <param name="type">Type of the actor to search for. Includes any actors derived from the type.</param>
     /// <returns>Found actors list.</returns>
-    API_FUNCTION() static Array<Actor*> GetActors(const MClass* type);
+    API_FUNCTION() static Array<Actor*> GetActors(API_PARAM(Attributes="TypeReference(typeof(Actor))") const MClass* type);
 
     /// <summary>
     /// Finds all the scripts of the given type in all the loaded scenes.
     /// </summary>
     /// <param name="type">Type of the script to search for. Includes any scripts derived from the type.</param>
     /// <returns>Found scripts list.</returns>
-    API_FUNCTION() static Array<Script*> GetScripts(const MClass* type);
+    API_FUNCTION() static Array<Script*> GetScripts(API_PARAM(Attributes="TypeReference(typeof(Script))") const MClass* type);
 
     /// <summary>
     /// Tries to find scene with given ID.