75 lines
2.3 KiB
C#
75 lines
2.3 KiB
C#
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using UnityEditor;
|
|
using UnityEngine;
|
|
|
|
namespace NaughtyAttributes.Editor
|
|
{
|
|
[CustomPropertyDrawer(typeof(InputAxisAttribute))]
|
|
public class InputAxisPropertyDrawer : PropertyDrawerBase
|
|
{
|
|
private static readonly string AssetPath = Path.Combine("ProjectSettings", "InputManager.asset");
|
|
private const string AxesPropertyPath = "m_Axes";
|
|
private const string NamePropertyPath = "m_Name";
|
|
|
|
protected override float GetPropertyHeight_Internal(SerializedProperty property, GUIContent label)
|
|
{
|
|
return (property.propertyType == SerializedPropertyType.String)
|
|
? GetPropertyHeight(property)
|
|
: GetPropertyHeight(property) + GetHelpBoxHeight();
|
|
}
|
|
|
|
protected override void OnGUI_Internal(Rect rect, SerializedProperty property, GUIContent label)
|
|
{
|
|
if (property.propertyType == SerializedPropertyType.String)
|
|
{
|
|
var inputManagerAsset = AssetDatabase.LoadAssetAtPath(AssetPath, typeof(object));
|
|
var inputManager = new SerializedObject(inputManagerAsset);
|
|
|
|
var axesProperty = inputManager.FindProperty(AxesPropertyPath);
|
|
var axesSet = new HashSet<string>();
|
|
axesSet.Add("(None)");
|
|
|
|
for (var i = 0; i < axesProperty.arraySize; i++)
|
|
{
|
|
var axis = axesProperty.GetArrayElementAtIndex(i).FindPropertyRelative(NamePropertyPath).stringValue;
|
|
axesSet.Add(axis);
|
|
}
|
|
|
|
var axes = axesSet.ToArray();
|
|
|
|
string propertyString = property.stringValue;
|
|
int index = 0;
|
|
// check if there is an entry that matches the entry and get the index
|
|
// we skip index 0 as that is a special custom case
|
|
for (int i = 1; i < axes.Length; i++)
|
|
{
|
|
if (axes[i] == propertyString)
|
|
{
|
|
index = i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Draw the popup box with the current selected index
|
|
var newIndex = EditorGUI.Popup(rect, label.text, index, axes);
|
|
|
|
// Adjust the actual string value of the property based on the selection
|
|
if (newIndex > 0)
|
|
{
|
|
property.stringValue = axes[newIndex];
|
|
}
|
|
else
|
|
{
|
|
property.stringValue = string.Empty;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
string message = string.Format("{0} supports only string fields", typeof(InputAxisAttribute).Name);
|
|
DrawDefaultPropertyAndHelpBox(rect, property, message, MessageType.Warning);
|
|
}
|
|
}
|
|
}
|
|
}
|