Rasagar/Library/PackageCache/com.unity.inputsystem/InputSystem/Editor/PropertyDrawers/InputActionReferencePropertyDrawer.cs
2024-08-26 23:07:20 +03:00

51 lines
2.1 KiB
C#

// Note: If not UNITY_INPUT_SYSTEM_PROJECT_WIDE_ACTIONS we do not use a custom property drawer and
// picker for InputActionReferences but rather rely on default (classic) object picker.
#if UNITY_EDITOR && UNITY_INPUT_SYSTEM_PROJECT_WIDE_ACTIONS
using UnityEditor;
using UnityEditor.Search;
namespace UnityEngine.InputSystem.Editor
{
/// <summary>
/// Custom property drawer in order to use the "Advanced Picker" from UnityEditor.Search.
/// </summary>
[CustomPropertyDrawer(typeof(InputActionReference))]
internal sealed class InputActionReferencePropertyDrawer : PropertyDrawer
{
private readonly SearchContext m_Context = UnityEditor.Search.SearchService.CreateContext(new[]
{
InputActionReferenceSearchProviders.CreateInputActionReferenceSearchProviderForAssets(),
InputActionReferenceSearchProviders.CreateInputActionReferenceSearchProviderForProjectWideActions(),
}, string.Empty, SearchConstants.PickerSearchFlags);
public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
{
// Sets the property to null if the action is not found in the asset.
ValidatePropertyWithDanglingInputActionReferences(property);
ObjectField.DoObjectField(position, property, typeof(InputActionReference), label,
m_Context, SearchConstants.PickerViewFlags);
}
static void ValidatePropertyWithDanglingInputActionReferences(SerializedProperty property)
{
if (property?.objectReferenceValue is InputActionReference reference)
{
// Check only if the reference is a project-wide action.
if (reference?.asset == InputSystem.actions)
{
var action = reference?.asset?.FindAction(reference.action.id);
if (action is null)
{
property.objectReferenceValue = null;
property.serializedObject.ApplyModifiedProperties();
}
}
}
}
}
}
#endif