path: root/VRCSDK3Avatars/Assets/Resources/GestureManager/Scripts/Editor/Modules/Vrc3/RadialMenuUtility.cs

#if VRC_SDK_VRCSDK3
using System;
using System.Linq;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using GestureManager.Scripts.Core.Editor;
using GestureManager.Scripts.Core.VisualElements;
using GestureManager.Scripts.Editor.Modules.Vrc3.Params;
using GestureManager.Scripts.Editor.Modules.Vrc3.RadialButtons;
using UnityEditor;
using UnityEngine;
using UnityEngine.Animations;
using UnityEngine.UIElements;
using UnityEngine.Playables;
using VRC.SDK3.Avatars.ScriptableObjects;
using UIEPosition = UnityEngine.UIElements.Position;
using ControlType = VRC.SDK3.Avatars.ScriptableObjects.VRCExpressionsMenu.Control.ControlType;

namespace GestureManager.Scripts.Editor.Modules.Vrc3
{
    public static class RadialMenuUtility
    {
        internal static Vector2 DataVector = new Vector2(-50, -60);

        public static class Colors
        {
            internal static class Default
            {
                internal static readonly Color Main = new Color(0.14f, 0.18f, 0.2f);
                internal static readonly Color Border = new Color(0.1f, 0.35f, 0.38f);
                internal static readonly Color Selected = new Color(0.07f, 0.55f, 0.58f);
            }

            private const string MainKeyName = "GM3 Main Color";
            private const string BorderKeyName = "GM3 Border Color";
            private const string SelectedKeyName = "GM3 Selected Color";

            private static Color PrefColor(string name, Color defaultColor) => ColorUtility.TryParseHtmlString(EditorPrefs.GetString(name), out var color) ? color : defaultColor;

            internal static readonly Color RadialTextBackground = new Color(0.11f, 0.11f, 0.11f, 0.49f);
            internal static readonly Color RadialSelColor = new Color(0.06f, 0.2f, 0.22f);
            internal static readonly Color RadialCenter = new Color(0.06f, 0.27f, 0.29f);
            internal static readonly Color RadialInner = new Color(0.21f, 0.24f, 0.27f);
            internal static readonly Color RestartButton = new Color(1f, 0.72f, 0.41f);
            internal static readonly Color SubIcon = new Color(0.22f, 0.24f, 0.27f);

            internal static Color Cursor => new Color(CustomMain.r, CustomMain.g, CustomMain.b, CustomMain.a - 0.3f);
            internal static Color CursorBorder => new Color(CustomBorder.r, CustomBorder.g, CustomBorder.b, CustomBorder.a - 0.5f);
            internal static Color ProgressBorder => CustomSelected * 1.5f;

            internal static Color CustomMain = PrefColor(MainKeyName, Default.Main);
            internal static Color CustomBorder = PrefColor(BorderKeyName, Default.Border);
            internal static Color CustomSelected = PrefColor(SelectedKeyName, Default.Selected);

            public static void SaveColors(Color main, Color border, Color selected)
            {
                CustomMain = main;
                CustomBorder = border;
                CustomSelected = selected;
                EditorPrefs.SetString(MainKeyName, $"#{ColorUtility.ToHtmlStringRGBA(main)}");
                EditorPrefs.SetString(BorderKeyName, $"#{ColorUtility.ToHtmlStringRGBA(border)}");
                EditorPrefs.SetString(SelectedKeyName, $"#{ColorUtility.ToHtmlStringRGBA(selected)}");
            }
        }

        public static class Prefabs
        {
            internal static VisualElement NewBorder(float size)
            {
                return new VisualElement
                {
                    pickingMode = PickingMode.Ignore,
                    style =
                    {
                        width = size,
                        height = 2,
                        backgroundColor = Colors.CustomBorder,
                        position = UIEPosition.Absolute
                    }
                };
            }

            internal static VisualElement NewBorder(float size, float euler)
            {
                var element = NewBorder(size);
                element.transform.rotation = Quaternion.Euler(0, 0, euler);
                return element;
            }

            internal static VisualElement NewCircleBorder(float size, Color border, UIEPosition position = default) => SetCircleBorder(new VisualElement(), size, border, position);

            private static VisualElement SetCircleBorder(VisualElement element, float size, Color border, UIEPosition position = default)
            {
                element.MyBorder(2, size, border);
                element.pickingMode = PickingMode.Ignore;
                element.style.position = position;
                element.style.alignItems = Align.Center;
                element.style.width = element.style.height = size;
                element.style.justifyContent = Justify.Center;
                return element;
            }

            internal static VisualElement NewIconText(float x, float y, float size, Texture2D icon, string text) => NewIconText(x, y, size, size / 2, icon, text);

            private static VisualElement NewIconText(float x, float y, float size, float hSize, Texture2D icon, string text)
            {
                return new VisualElement
                {
                    pickingMode = PickingMode.Ignore,
                    style =
                    {
                        width = size,
                        height = size,
                        left = x - hSize,
                        top = y - hSize,
                        backgroundImage = icon,
                        position = UIEPosition.Absolute
                    }
                }.With(new TextElement
                {
                    text = text,
                    pickingMode = PickingMode.Ignore,
                    style =
                    {
                        color = Color.white,
                        fontSize = 8,
                        unityTextAlign = TextAnchor.MiddleCenter,
                        top = size
                    }
                });
            }

            internal static VisualElement NewData(float width, float height)
            {
                return new VisualElement
                {
                    pickingMode = PickingMode.Ignore,
                    style =
                    {
                        width = width,
                        height = height,
                        left = -width / 2,
                        top = -height / 2 - 10,
                        backgroundColor = Color.clear,
                        position = UIEPosition.Absolute,
                        alignItems = Align.Center,
                        justifyContent = Justify.Center
                    }
                };
            }

            internal static GmgCircleElement NewSlice(float size, Color centerColor, Color color, Color border, UIEPosition position = UIEPosition.Absolute) => SetSlice(new GmgCircleElement(), size, centerColor, color, border, position);

            private static GmgCircleElement SetSlice(GmgCircleElement element, float size, Color centerColor, Color color, Color border, UIEPosition position = default)
            {
                element.BorderWidth = 2;
                element.VertexColor = color;
                element.BorderColor = border;
                element.CenterColor = centerColor;
                element.pickingMode = PickingMode.Ignore;
                element.style.position = position;
                element.style.alignItems = Align.Center;
                element.style.width = element.style.height = size;
                element.style.justifyContent = Justify.Center;
                element.style.left = element.style.top = -size / 2;
                return element;
            }

            internal static GmgCircleElement NewCircle(float size, Color color, Color border, UIEPosition position = default) => SetCircle(new GmgCircleElement(), size, color, color, border, position);

            internal static GmgCircleElement NewCircle(float size, Color centerColor, Color color, Color border, UIEPosition position = default) => SetCircle(new GmgCircleElement(), size, centerColor, color, border, position);

            internal static GmgCircleElement SetCircle(GmgCircleElement element, float size, Color color, Color border, UIEPosition position = default) => SetCircle(element, size, color, color, border, position);

            private static GmgCircleElement SetCircle(GmgCircleElement element, float size, Color centerColor, Color color, Color border, UIEPosition position = default)
            {
                element.BorderWidth = 2;
                element.VertexColor = color;
                element.BorderColor = border;
                element.CenterColor = centerColor;
                element.pickingMode = PickingMode.Ignore;
                element.style.position = position;
                element.style.alignItems = Align.Center;
                element.style.width = element.style.height = size;
                element.style.justifyContent = Justify.Center;
                return element;
            }

            internal static VisualElement NewRadialText(out TextElement text, int top, UIEPosition position = default)
            {
                return SetRadialText(new VisualElement(), out text, top, position);
            }

            internal static VisualElement SetRadialText(VisualElement element, out TextElement text, int top, UIEPosition position = default)
            {
                element.style.width = 50;
                element.style.height = 20;
                element.pickingMode = PickingMode.Ignore;
                element.style.backgroundColor = Colors.RadialTextBackground;
                element.style.borderTopLeftRadius = 10;
                element.style.borderTopRightRadius = 10;
                element.style.borderBottomLeftRadius = 10;
                element.style.borderBottomRightRadius = 10;
                element.style.position = position;
                if (top != 0) element.style.top = top;
                text = element.MyAdd(new TextElement { pickingMode = PickingMode.Ignore, style = { height = 20, unityTextAlign = TextAnchor.MiddleCenter, color = Color.white, fontSize = 14 } });
                return element;
            }

            internal static VisualElement NewSubIcon(Texture2D texture)
            {
                return new VisualElement
                {
                    pickingMode = PickingMode.Ignore,
                    style =
                    {
                        width = 25,
                        height = 25,
                        borderTopLeftRadius = 15,
                        borderTopRightRadius = 15,
                        borderBottomLeftRadius = 15,
                        borderBottomRightRadius = 15,
                        left = 60,
                        top = 50,
                        backgroundColor = Colors.SubIcon,
                        position = UIEPosition.Absolute,
                        justifyContent = Justify.Center,
                        alignItems = Align.Center
                    }
                }.With(new VisualElement
                {
                    pickingMode = PickingMode.Ignore,
                    style =
                    {
                        width = 20,
                        height = 20,
                        backgroundImage = texture
                    }
                });
            }
        }

        public static class Buttons
        {
            public static RadialMenuControl ToggleFromParam(RadialMenu menu, string name, Vrc3Param param)
            {
                return ParamStateToggle(menu, name, param, 1f);
            }

            public static RadialMenuControl ParamStateToggle(RadialMenu menu, string name, Vrc3Param param, float activeValue)
            {
                return new RadialMenuControl(menu, name, null, ControlType.Toggle, activeValue, param, Array.Empty<Vrc3Param>(), null, null);
            }

            public static RadialMenuItem RadialFromParam(RadialMenu menu, string name, Vrc3Param param, float amplify = 1f)
            {
                return new RadialMenuControl(menu, name, null, ControlType.RadialPuppet, 1f, null, new[] { param }, null, null, amplify);
            }

            public static RadialMenuControl AxisFromParams(RadialMenu menu, string name, Vrc3Param xParam, Vrc3Param yParam, float amplify = 1f)
            {
                var subLabels = new VRCExpressionsMenu.Control.Label[4];
                return new RadialMenuControl(menu, name, null, ControlType.TwoAxisPuppet, 1f, null, new[] { xParam, yParam }, null, subLabels, amplify);
            }
        }

        public static Texture2D GetSubIcon(ControlType type)
        {
            switch (type)
            {
                case ControlType.Button:
                    return null;
                case ControlType.Toggle:
                    return ModuleVrc3Styles.Toggle;
                case ControlType.SubMenu:
                    return ModuleVrc3Styles.Option;
                case ControlType.TwoAxisPuppet:
                    return ModuleVrc3Styles.TwoAxis;
                case ControlType.FourAxisPuppet:
                    return ModuleVrc3Styles.FourAxis;
                case ControlType.RadialPuppet:
                    return ModuleVrc3Styles.Radial;
                default: throw new ArgumentOutOfRangeException(nameof(type), type, null);
            }
        }

        public static IEnumerable<AnimatorControllerParameter> GetParameters(AnimatorControllerPlayable playable)
        {
            if (playable.IsNull()) yield break;
            for (var i = 0; i < playable.GetParameterCount(); i++)
                yield return playable.GetParameter(i);
        }

        public static Vrc3ParamController CreateParamFromController(Animator animator, AnimatorControllerParameter parameter, AnimatorControllerPlayable controller)
        {
            return new Vrc3ParamController(parameter.type, parameter.name, controller, animator);
        }

        public static Vrc3Param CreateParamFromNothing(VRCExpressionParameters.Parameter parameter)
        {
            return new Vrc3ParamExternal(parameter.name, ModuleVrc3Styles.Data.TypeOf[parameter.valueType]);
        }

        public static Vrc3Param CreateParamFromNothing(string name, AnimatorControllerParameterType type)
        {
            return new Vrc3ParamExternal(name, type);
        }

        public static int RadialPercentage(float value)
        {
            return RadialPercentage(value, out _);
        }

        public static int RadialPercentage(float value, out float clamp)
        {
            clamp = Mathf.Clamp(value, 0f, 1f);
            return (int)(clamp * 100);
        }

        private static void AppendMenus(VRCExpressionsMenu menu, ICollection<VRCExpressionsMenu> menus)
        {
            if (!menu || menus.Contains(menu)) return;
            menus.Add(menu);
            foreach (var control in menu.controls) AppendMenus(control.subMenu, menus);
        }

        private static IEnumerable<VRCExpressionsMenu> GetMenus(VRCExpressionsMenu menu)
        {
            var menus = new List<VRCExpressionsMenu>();
            AppendMenus(menu, menus);
            return menus;
        }

        private static IEnumerable<string> GetParams(IEnumerable<VRCExpressionsMenu> menus)
        {
            var paramList = new HashSet<string>();
            foreach (var menu in menus)
            foreach (var control in menu.controls)
            {
                paramList.Add(control.name);
                if (control.subParameters == null) continue;
                foreach (var subParameter in control.subParameters)
                    paramList.Add(subParameter.name);
            }

            return paramList;
        }

        public static IEnumerable<string> CheckErrors(VRCExpressionsMenu menu, VRCExpressionParameters param)
        {
            var errors = new List<string>();
            if (!menu && !param) return errors;
            if (!menu || !param) errors.Add(!menu ? "- No menu defined when parameters are defined!" : "- No parameters defined when menu is defined!");
            return errors;
        }

        [SuppressMessage("ReSharper", "UnusedMember.Global")]
        public static IEnumerable<string> CheckWarnings(VRCExpressionsMenu menu, VRCExpressionParameters param)
        {
            return from paramName in GetParams(GetMenus(menu))
                let isDefined = string.IsNullOrEmpty(paramName) || param.FindParameter(paramName) != null
                where !isDefined
                select "Menu uses a parameter that is not defined: " + paramName;
        }

        [SuppressMessage("ReSharper", "CompareOfFloatsByEqualityOperator")]
        public static bool Is(float get, float value)
        {
            return get == value;
        }
    }
}
#endif