From 845cdb3097ce2781c2575f0c8277900852fcccf5 Mon Sep 17 00:00:00 2001
From: tsrk <tsrk@tsrk.me>
Date: Wed, 11 Jan 2023 14:03:49 +0100
Subject: [PATCH] test: add visual tests for SegmentedGraph class

---
 .../UserInterface/TestSceneSegmentedGraph.cs  | 96 +++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 osu.Game.Tests/Visual/UserInterface/TestSceneSegmentedGraph.cs

diff --git a/osu.Game.Tests/Visual/UserInterface/TestSceneSegmentedGraph.cs b/osu.Game.Tests/Visual/UserInterface/TestSceneSegmentedGraph.cs
new file mode 100644
index 0000000000..ab7078d1b7
--- /dev/null
+++ b/osu.Game.Tests/Visual/UserInterface/TestSceneSegmentedGraph.cs
@@ -0,0 +1,96 @@
+// 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;
+using System.Linq;
+using osu.Framework.Graphics;
+using osu.Game.Graphics.UserInterface;
+using osuTK;
+using Vector4 = System.Numerics.Vector4;
+
+namespace osu.Game.Tests.Visual.UserInterface
+{
+    public partial class TestSceneSegmentedGraph : OsuTestScene
+    {
+        private readonly SegmentedGraph<int> graph;
+
+        public TestSceneSegmentedGraph()
+        {
+            Children = new Drawable[]
+            {
+                graph = new SegmentedGraph<int>(10)
+                {
+                    RelativeSizeAxes = Axes.Both,
+                    Anchor = Anchor.Centre,
+                    Origin = Anchor.Centre,
+                    Size = new Vector2(1, 0.5f),
+                },
+            };
+
+            for (int i = 0; i < graph.TierColours.Length; i++)
+            {
+                graph.TierColours[i] =
+                    new Colour4(Vector4.Lerp(Colour4.Blue.Vector, Colour4.White.Vector, i * 1f / (graph.TierColours.Length - 1)));
+            }
+
+            AddStep("values from 1-10", () => graph.Values = Enumerable.Range(1, 10).ToArray());
+            AddStep("values from 1-100", () => graph.Values = Enumerable.Range(1, 100).ToArray());
+            AddStep("values from 1-500", () => graph.Values = Enumerable.Range(1, 500).ToArray());
+            AddStep("sin() function of size 100", () => sinFunction());
+            AddStep("sin() function of size 500", () => sinFunction(500));
+            AddStep("bumps of size 100", () => bumps());
+            AddStep("bumps of size 500", () => bumps(500));
+            AddStep("100 random values", () => randomValues());
+            AddStep("500 random values", () => randomValues(500));
+            AddStep("reversed values from 1-10", () => graph.Values = Enumerable.Range(1, 10).Reverse().ToArray());
+        }
+
+        private void sinFunction(int size = 100)
+        {
+            const int max_value = 255;
+            int[] values = new int[size];
+
+            float step = 2 * MathF.PI / size;
+            float x = 0;
+
+            for (int i = 0; i < size; i++)
+            {
+                values[i] = (int)(max_value * MathF.Sin(x));
+                x += step;
+            }
+
+            graph.Values = values;
+        }
+
+        private void bumps(int size = 100)
+        {
+            const int max_value = 255;
+            int[] values = new int[size];
+
+            float step = 2 * MathF.PI / size;
+            float x = 0;
+
+            for (int i = 0; i < size; i++)
+            {
+                values[i] = (int)(max_value * Math.Abs(MathF.Sin(x)));
+                x += step;
+            }
+
+            graph.Values = values;
+        }
+
+        private void randomValues(int size = 100)
+        {
+            Random rng = new Random();
+
+            int[] values = new int[size];
+
+            for (int i = 0; i < size; i++)
+            {
+                values[i] = rng.Next(255);
+            }
+
+            graph.Values = values;
+        }
+    }
+}