osu-lazer/osu.Game/Overlays/Mods/ModSection.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using osuTK;
using osuTK.Input;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Game.Graphics.Sprites;
using osu.Game.Rulesets.Mods;
using System;
using System.Linq;
using System.Collections.Generic;
using System.Threading;
using Humanizer;
using osu.Framework.Input.Events;
using osu.Game.Graphics;

namespace osu.Game.Overlays.Mods
{
    public class ModSection : CompositeDrawable
    {
        private readonly Drawable header;

        public FillFlowContainer<ModButtonEmpty> ButtonsContainer { get; }

        public Action<Mod> Action;

        public Key[] ToggleKeys;

        public readonly ModType ModType;

        public IEnumerable<Mod> SelectedMods => buttons.Select(b => b.SelectedMod).Where(m => m != null);

        private CancellationTokenSource modsLoadCts;

        protected bool SelectionAnimationRunning => pendingSelectionOperations.Count > 0;

        /// <summary>
        /// True when all mod icons have completed loading.
        /// </summary>
        public bool ModIconsLoaded { get; private set; } = true;

        public IEnumerable<Mod> Mods
        {
            set
            {
                var modContainers = value.Select(m =>
                {
                    if (m == null)
                        return new ModButtonEmpty();

                    return new ModButton(m)
                    {
                        SelectionChanged = mod =>
                        {
                            ModButtonStateChanged(mod);
                            Action?.Invoke(mod);
                        },
                    };
                }).ToArray();

                modsLoadCts?.Cancel();

                if (modContainers.Length == 0)
                {
                    ModIconsLoaded = true;
                    header.Hide();
                    Hide();
                    return;
                }

                ModIconsLoaded = false;

                LoadComponentsAsync(modContainers, c =>
                {
                    ModIconsLoaded = true;
                    ButtonsContainer.ChildrenEnumerable = c;
                }, (modsLoadCts = new CancellationTokenSource()).Token);

                buttons = modContainers.OfType<ModButton>().ToArray();

                header.FadeIn(200);
                this.FadeIn(200);
            }
        }

        protected virtual void ModButtonStateChanged(Mod mod)
        {
        }

        private ModButton[] buttons = Array.Empty<ModButton>();

        protected override bool OnKeyDown(KeyDownEvent e)
        {
            if (e.ControlPressed) return false;

            if (ToggleKeys != null)
            {
                var index = Array.IndexOf(ToggleKeys, e.Key);
                if (index > -1 && index < buttons.Length)
                    buttons[index].SelectNext(e.ShiftPressed ? -1 : 1);
            }

            return base.OnKeyDown(e);
        }

        private const double initial_multiple_selection_delay = 100;

        private readonly Queue<Action> pendingSelectionOperations = new Queue<Action>();

        protected override void LoadComplete()
        {
            base.LoadComplete();

            Scheduler.AddDelayed(() =>
            {
                if (pendingSelectionOperations.TryDequeue(out var dequeuedAction))
                    dequeuedAction();
            }, initial_multiple_selection_delay, true);
        }

        /// <summary>
        /// Selects all mods.
        /// </summary>
        public void SelectAll()
        {
            pendingSelectionOperations.Clear();

            foreach (var button in buttons.Where(b => !b.Selected))
                pendingSelectionOperations.Enqueue(() => button.SelectAt(0));
        }

        /// <summary>
        /// Deselects all mods.
        /// </summary>
        public void DeselectAll() => DeselectTypes(buttons.Select(b => b.SelectedMod?.GetType()).Where(t => t != null));

        /// <summary>
        /// Deselect one or more mods in this section.
        /// </summary>
        /// <param name="modTypes">The types of <see cref="Mod"/>s which should be deselected.</param>
        public void DeselectTypes(IEnumerable<Type> modTypes)
        {
            pendingSelectionOperations.Clear();

            foreach (var button in buttons)
            {
                if (button.SelectedMod == null) continue;

                foreach (var type in modTypes)
                {
                    if (type.IsInstanceOfType(button.SelectedMod))
                        pendingSelectionOperations.Enqueue(button.Deselect);
                }
            }
        }

        /// <summary>
        /// Updates all buttons with the given list of selected mods.
        /// </summary>
        /// <param name="newSelectedMods">The new list of selected mods to select.</param>
        public void UpdateSelectedButtons(IReadOnlyList<Mod> newSelectedMods)
        {
            foreach (var button in buttons)
                updateButtonSelection(button, newSelectedMods);
        }

        private void updateButtonSelection(ModButton button, IReadOnlyList<Mod> newSelectedMods)
        {
            foreach (var mod in newSelectedMods)
            {
                var index = Array.FindIndex(button.Mods, m1 => mod.GetType() == m1.GetType());
                if (index < 0)
                    continue;

                var buttonMod = button.Mods[index];
                buttonMod.CopyFrom(mod);
                button.SelectAt(index);
                return;
            }

            button.Deselect();
        }

        public ModSection(ModType type)
        {
            ModType = type;

            AutoSizeAxes = Axes.Y;
            RelativeSizeAxes = Axes.X;

            Origin = Anchor.TopCentre;
            Anchor = Anchor.TopCentre;

            InternalChildren = new[]
            {
                header = CreateHeader(type.Humanize(LetterCasing.Title)),
                ButtonsContainer = new FillFlowContainer<ModButtonEmpty>
                {
                    AutoSizeAxes = Axes.Y,
                    RelativeSizeAxes = Axes.X,
                    Origin = Anchor.BottomLeft,
                    Anchor = Anchor.BottomLeft,
                    Spacing = new Vector2(50f, 0f),
                    Margin = new MarginPadding
                    {
                        Top = 20,
                    },
                    AlwaysPresent = true
                },
            };
        }

        protected virtual Drawable CreateHeader(string text) => new OsuSpriteText
        {
            Font = OsuFont.GetFont(weight: FontWeight.Bold),
            Text = text
        };
    }
}