MinhHai
60 lines
1.8 KiB
C#
60 lines
1.8 KiB
C#
using UnityEditor;
|
|
using UnityEngine.UIElements;
|
|
|
|
namespace EditorAttributes.Editor
|
|
{
|
|
[CustomPropertyDrawer(typeof(ProgressBarAttribute))]
|
|
public class ProgressBarDrawer : PropertyDrawerBase
|
|
{
|
|
public override VisualElement CreatePropertyGUI(SerializedProperty property)
|
|
{
|
|
var root = new VisualElement();
|
|
|
|
if (property.propertyType == SerializedPropertyType.Integer || property.propertyType == SerializedPropertyType.Float)
|
|
{
|
|
var progressBarAttribute = attribute as ProgressBarAttribute;
|
|
|
|
var progressBar = new ProgressBar { highValue = progressBarAttribute.MaxValue, tooltip = property.tooltip };
|
|
|
|
// Make the visual elements in the progress bar grow
|
|
progressBar.ElementAt(0).style.flexGrow = 1f;
|
|
progressBar.ElementAt(0).ElementAt(0).style.flexGrow = 1f;
|
|
|
|
progressBar.style.height = progressBarAttribute.BarHeight;
|
|
|
|
root.Add(progressBar);
|
|
|
|
ExecuteLater(progressBar, () =>
|
|
{
|
|
if (CanApplyGlobalColor)
|
|
progressBar.Q(className: AbstractProgressBar.progressUssClassName).style.backgroundColor = EditorExtension.GLOBAL_COLOR / 2f;
|
|
});
|
|
|
|
UpdateVisualElement(progressBar, () =>
|
|
{
|
|
var propertyValue = GetPropertyValue(property);
|
|
|
|
progressBar.value = propertyValue;
|
|
progressBar.title = $"{property.displayName}: {propertyValue}/{progressBarAttribute.MaxValue}";
|
|
}, 30L);
|
|
}
|
|
else
|
|
{
|
|
root.Add(new HelpBox("The ProgressBar Attribute can only be attached to an int or float", HelpBoxMessageType.Error));
|
|
}
|
|
|
|
return root;
|
|
}
|
|
|
|
private float GetPropertyValue(SerializedProperty property)
|
|
{
|
|
return property.propertyType switch
|
|
{
|
|
SerializedPropertyType.Integer => property.intValue,
|
|
SerializedPropertyType.Float => property.floatValue,
|
|
_ => 0f
|
|
};
|
|
}
|
|
}
|
|
}
|