1

Assets/Dreamteck/Utilities/Editor/FindDerivedClasses.cs

@@ -0,0 +1,66 @@
+namespace Dreamteck
+{
+    using UnityEngine;
+    using System;
+    using System.Reflection;
+    using System.Collections.Generic;
+    public static class FindDerivedClasses
+    {
+        public static List<Type> GetAllDerivedClasses(this Type aBaseClass, string[] aExcludeAssemblies)
+        {
+            List<Type> result = new List<Type>();
+            foreach (Assembly A in AppDomain.CurrentDomain.GetAssemblies())
+            {
+                bool exclude = false;
+                foreach (string S in aExcludeAssemblies)
+                {
+                    if (A.GetName().FullName.StartsWith(S))
+                    {
+                        exclude = true;
+                        break;
+                    }
+                }
+                if (exclude)
+                {
+                    continue;
+                }
+                try
+                {
+                    if (aBaseClass.IsInterface)
+                    {
+                        foreach (Type C in A.GetExportedTypes())
+                        {
+                            foreach (Type I in C.GetInterfaces())
+                            {
+                                if (aBaseClass == I)
+                                {
+                                    result.Add(C);
+                                    break;
+                                }
+                            }
+                        }
+                    }
+                    else
+                    {
+                        foreach (Type C in A.GetExportedTypes())
+                        {
+                            if (C.IsSubclassOf(aBaseClass))
+                            {
+                                result.Add(C);
+                            }
+                        }
+                    }
+                } catch
+                {
+                    Debug.LogWarning("Dreamteck was unable to scan " + A.FullName + " for derived classes");
+                }
+            }
+            return result;
+        }
+
+        public static List<Type> GetAllDerivedClasses(this Type aBaseClass)
+        {
+            return GetAllDerivedClasses(aBaseClass, new string[0]);
+        }
+    }
+}