osu-lazer/osu.Game.Tests/Visual/UserInterface/TestSceneModIcon.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 System.Collections.Generic;
using System.Linq;
using NUnit.Framework;
using osu.Framework.Graphics;
using osu.Framework.Graphics.Containers;
using osu.Framework.Testing;
using osu.Framework.Utils;
using osu.Game.Rulesets.Mods;
using osu.Game.Rulesets.Osu;
using osu.Game.Rulesets.Osu.Mods;
using osu.Game.Rulesets.UI;
using osu.Game.Screens.Play.HUD;

namespace osu.Game.Tests.Visual.UserInterface
{
    public partial class TestSceneModIcon : OsuTestScene
    {
        private FillFlowContainer spreadOutFlow = null!;
        private ModDisplay modDisplay = null!;

        [SetUpSteps]
        public void SetUpSteps()
        {
            AddStep("create flows", () =>
            {
                Child = new GridContainer
                {
                    RelativeSizeAxes = Axes.Both,
                    RowDimensions = new[]
                    {
                        new Dimension(GridSizeMode.Relative, 0.5f),
                        new Dimension(GridSizeMode.Relative, 0.5f),
                    },
                    Content = new[]
                    {
                        new Drawable[]
                        {
                            modDisplay = new ModDisplay
                            {
                                Anchor = Anchor.Centre,
                                Origin = Anchor.Centre,
                            }
                        },
                        new Drawable[]
                        {
                            spreadOutFlow = new FillFlowContainer
                            {
                                Anchor = Anchor.Centre,
                                Origin = Anchor.Centre,
                                RelativeSizeAxes = Axes.X,
                                AutoSizeAxes = Axes.Y,
                                Direction = FillDirection.Full,
                            }
                        }
                    }
                };
            });
        }

        private void addRange(IEnumerable<IMod> mods)
        {
            spreadOutFlow.AddRange(mods.Select(m => new ModIcon(m)));
            modDisplay.Current.Value = modDisplay.Current.Value.Concat(mods.OfType<Mod>()).ToList();
        }

        [Test]
        public void TestShowAllMods()
        {
            AddStep("create mod icons", () =>
            {
                addRange(Ruleset.Value.CreateInstance().CreateAllMods());
            });

            AddStep("toggle selected", () =>
            {
                foreach (var icon in this.ChildrenOfType<ModIcon>())
                    icon.Selected.Toggle();
            });
        }

        [Test]
        public void TestShowRateAdjusts()
        {
            AddStep("create mod icons", () =>
            {
                var rateAdjustMods = Ruleset.Value.CreateInstance().CreateAllMods()
                                            .OfType<ModRateAdjust>();

                addRange(rateAdjustMods.SelectMany(m =>
                {
                    List<Mod> mods = new List<Mod> { m };

                    for (double i = m.SpeedChange.MinValue; i < m.SpeedChange.MaxValue; i += m.SpeedChange.Precision * 10)
                    {
                        m = (ModRateAdjust)m.DeepClone();
                        m.SpeedChange.Value = i;
                        mods.Add(m);
                    }

                    return mods;
                }));
            });

            AddStep("adjust rates", () =>
            {
                foreach (var icon in this.ChildrenOfType<ModIcon>())
                {
                    if (icon.Mod is ModRateAdjust rateAdjust)
                    {
                        rateAdjust.SpeedChange.Value = RNG.NextDouble() > 0.9
                            ? rateAdjust.SpeedChange.Default
                            : RNG.NextDouble(rateAdjust.SpeedChange.MinValue, rateAdjust.SpeedChange.MaxValue);
                    }
                }
            });
        }

        [Test]
        public void TestChangeModType()
        {
            AddStep("create mod icon", () => addRange([new OsuModDoubleTime()]));
            AddStep("change mod", () =>
            {
                foreach (var modIcon in this.ChildrenOfType<ModIcon>())
                    modIcon.Mod = new OsuModEasy();
            });
        }

        [Test]
        public void TestInterfaceModType()
        {
            var ruleset = new OsuRuleset();

            AddStep("create mod icon", () => addRange([ruleset.AllMods.First(m => m.Acronym == "DT")]));
            AddStep("change mod", () =>
            {
                foreach (var modIcon in this.ChildrenOfType<ModIcon>())
                    modIcon.Mod = ruleset.AllMods.First(m => m.Acronym == "EZ");
            });
        }
    }
}