osu-lazer/osu.Game.Rulesets.Osu.Tests/TestSceneOsuAnalysisContainer.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.

#nullable disable

using System;
using System.Collections.Generic;
using System.Linq;
using NUnit.Framework;
using osu.Framework.Allocation;
using osu.Game.Replays;
using osu.Game.Rulesets.Osu.Beatmaps;
using osu.Game.Rulesets.Osu.Replays;
using osu.Game.Rulesets.Osu.UI;
using osu.Game.Rulesets.Replays;
using osu.Game.Rulesets.UI;
using osu.Game.Tests.Visual;
using osuTK;

namespace osu.Game.Rulesets.Osu.Tests
{
    public partial class TestSceneOsuAnalysisContainer : OsuTestScene
    {
        private TestOsuAnalysisContainer analysisContainer;

        [BackgroundDependencyLoader]
        private void load()
        {
            Child = analysisContainer = createAnalysisContainer();
        }

        [Test]
        public void TestHitMarkers()
        {
            AddStep("enable hit markers", () => analysisContainer.AnalysisSettings.HitMarkersEnabled.Value = true);
            AddAssert("hit markers visible", () => analysisContainer.HitMarkersVisible);
            AddStep("disable hit markers", () => analysisContainer.AnalysisSettings.HitMarkersEnabled.Value = false);
            AddAssert("hit markers not visible", () => !analysisContainer.HitMarkersVisible);
        }

        [Test]
        public void TestAimMarker()
        {
            AddStep("enable aim markers", () => analysisContainer.AnalysisSettings.AimMarkersEnabled.Value = true);
            AddAssert("aim markers visible", () => analysisContainer.AimMarkersVisible);
            AddStep("disable aim markers", () => analysisContainer.AnalysisSettings.AimMarkersEnabled.Value = false);
            AddAssert("aim markers not visible", () => !analysisContainer.AimMarkersVisible);
        }

        [Test]
        public void TestAimLines()
        {
            AddStep("enable aim lines", () => analysisContainer.AnalysisSettings.AimLinesEnabled.Value = true);
            AddAssert("aim lines visible", () => analysisContainer.AimLinesVisible);
            AddStep("disable aim lines", () => analysisContainer.AnalysisSettings.AimLinesEnabled.Value = false);
            AddAssert("aim lines not visible", () => !analysisContainer.AimLinesVisible);
        }

        private TestOsuAnalysisContainer createAnalysisContainer()
        {
            var replay = new Replay();
            var ruleset = new OsuRuleset();
            var beatmap = new OsuBeatmap();
            var drawableRuleset = new DrawableOsuRuleset(ruleset, beatmap);
            // Load playfield cursor to avoid errors
            Add(drawableRuleset);

            return new TestOsuAnalysisContainer(replay, drawableRuleset);
        }

        private partial class TestOsuAnalysisContainer : OsuAnalysisContainer
        {
            public TestOsuAnalysisContainer(Replay replay, DrawableRuleset drawableRuleset)
                : base(replay, drawableRuleset)
            {
            }

            [BackgroundDependencyLoader]
            private void load()
            {
                Replay = fabricateReplay();
                LoadReplay();

                makeReplayLoop();
            }

            private void makeReplayLoop()
            {
                Scheduler.AddDelayed(() =>
                {
                    Replay = fabricateReplay();

                    HitMarkers.Clear();
                    AimMarkers.Clear();
                    AimLines.Clear();

                    LoadReplay();

                    makeReplayLoop();
                }, 15000);
            }

            public bool HitMarkersVisible => HitMarkers.Alpha > 0 && HitMarkers.Entries.Any();

            public bool AimMarkersVisible => AimMarkers.Alpha > 0 && AimMarkers.Entries.Any();

            public bool AimLinesVisible => AimLines.Alpha > 0 && AimLines.Vertices.Count > 1;

            private Replay fabricateReplay()
            {
                var frames = new List<ReplayFrame>();
                var random = new Random();
                int posX = 250;
                int posY = 250;
                bool leftOrRight = false;

                for (int i = 0; i < 1000; i++)
                {
                    posX = Math.Clamp(posX + random.Next(-10, 11), 0, 500);
                    posY = Math.Clamp(posY + random.Next(-10, 11), 0, 500);

                    var actions = new List<OsuAction>();

                    if (i % 20 == 0)
                    {
                        actions.Add(leftOrRight ? OsuAction.LeftButton : OsuAction.RightButton);
                        leftOrRight = !leftOrRight;
                    }

                    frames.Add(new OsuReplayFrame
                    {
                        Time = Time.Current + i * 15,
                        Position = new Vector2(posX, posY),
                        Actions = actions
                    });
                }

                return new Replay { Frames = frames };
            }
        }
    }
}