Possibility to specify TypeGrouping for Reference pickers

Author: arimger

Assets/Editor Toolbox/Editor/Drawers/Toolbox/PropertySelf/ReferencePickerAttributeDrawer.cs

@@ -10,9 +10,16 @@ namespace Toolbox.Editor.Drawers
 
     public class ReferencePickerAttributeDrawer : ToolboxSelfPropertyDrawer<ReferencePickerAttribute>
     {
-        private readonly TypeField typeField = new TypeField(new TypeConstraintReference(null));
+        private static readonly TypeConstraintContext sharedConstraint = new TypeConstraintReference(null);
+        private static readonly TypeAppearanceContext sharedAppearance = new TypeAppearanceContext(sharedConstraint, TypeGrouping.None, true);
+        private static readonly TypeField typeField = new TypeField(sharedConstraint, sharedAppearance);
 
 
+        private void UpdateContexts(ReferencePickerAttribute attribute)
+        {
+            sharedAppearance.TypeGrouping = attribute.TypeGrouping;
+        }
+
         private void CreateTypeProperty(SerializedProperty property, Type parentType)
         {
             TypeUtilities.TryGetTypeFromManagedReferenceFullTypeName(property.managedReferenceFullTypename, out var currentType);
@@ -72,7 +79,6 @@ private Type GetParentType(SerializedProperty property, ReferencePickerAttribute
             return propertyType;
         }
 
-
         protected override void OnGuiSafe(SerializedProperty property, GUIContent label, ReferencePickerAttribute attribute)
         {
             using (var propertyScope = new PropertyScope(property, label))
@@ -82,8 +88,10 @@ protected override void OnGuiSafe(SerializedProperty property, GUIContent label,
                     return;
                 }
 
-                EditorGUI.indentLevel++;
+                UpdateContexts(attribute);
                 var parentType = GetParentType(property, attribute);
+
+                EditorGUI.indentLevel++;
                 CreateTypeProperty(property, parentType);
                 ToolboxEditorGui.DrawPropertyChildren(property);
                 EditorGUI.indentLevel--;

Assets/Editor Toolbox/Runtime/Attributes/Toolbox/PropertySelfAttributes/ReferencePickerAttribute.cs

@@ -16,12 +16,21 @@ public class ReferencePickerAttribute : ToolboxSelfPropertyAttribute
         public ReferencePickerAttribute()
         { }
 
-        public ReferencePickerAttribute(Type parentType)
+        public ReferencePickerAttribute(Type parentType) : this(parentType, TypeGrouping.None)
+        { }
+
+        public ReferencePickerAttribute(Type parentType, TypeGrouping typeGrouping)
         {
             ParentType = parentType;
+            TypeGrouping = typeGrouping;
         }
 
         public Type ParentType { get; set; }
+        /// <summary>
+        /// Gets or sets grouping of selectable classes.
+        /// Defaults to <see cref="TypeGrouping.None"/> unless explicitly specified.
+        /// </summary>
+        public TypeGrouping TypeGrouping { get; set; } = TypeGrouping.None;
     }
 }
 #endif

Assets/Examples/Scripts/SampleBehaviour6.cs

@@ -7,7 +7,7 @@
 public class SampleBehaviour6 : MonoBehaviour
 {
 #if UNITY_2019_3_OR_NEWER
-    [SerializeReference, ReferencePicker]
+    [SerializeReference, ReferencePicker(TypeGrouping = TypeGrouping.ByFlatName)]
     public Interface1 var1;
     [SerializeReference, ReferencePicker]
     public ClassWithInterfaceBase var2;