mirror of
https://github.com/ppy/osu
synced 2024-12-09 08:30:36 +00:00
298 lines
9.9 KiB
C#
298 lines
9.9 KiB
C#
// 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.Linq;
|
|
using NUnit.Framework;
|
|
using osu.Framework.Graphics;
|
|
using osu.Framework.Graphics.Containers;
|
|
using osu.Framework.Testing;
|
|
using osu.Framework.Timing;
|
|
using osu.Framework.Utils;
|
|
using osu.Game.Beatmaps;
|
|
using osu.Game.Beatmaps.ControlPoints;
|
|
using osu.Game.Rulesets.Osu.Objects;
|
|
using osu.Game.Rulesets.Osu.Objects.Drawables;
|
|
using osu.Game.Rulesets.Osu.Objects.Drawables.Connections;
|
|
using osu.Game.Tests.Visual;
|
|
using osuTK;
|
|
|
|
namespace osu.Game.Rulesets.Osu.Tests
|
|
{
|
|
public class TestSceneFollowPoints : OsuTestScene
|
|
{
|
|
private Container<DrawableOsuHitObject> hitObjectContainer;
|
|
private FollowPointRenderer followPointRenderer;
|
|
|
|
[SetUp]
|
|
public void Setup() => Schedule(() =>
|
|
{
|
|
Children = new Drawable[]
|
|
{
|
|
hitObjectContainer = new TestHitObjectContainer { RelativeSizeAxes = Axes.Both },
|
|
followPointRenderer = new FollowPointRenderer { RelativeSizeAxes = Axes.Both }
|
|
};
|
|
});
|
|
|
|
[Test]
|
|
public void TestAddObject()
|
|
{
|
|
addObjectsStep(() => new OsuHitObject[] { new HitCircle { Position = new Vector2(100, 100) } });
|
|
|
|
assertGroups();
|
|
}
|
|
|
|
[Test]
|
|
public void TestRemoveObject()
|
|
{
|
|
addObjectsStep(() => new OsuHitObject[] { new HitCircle { Position = new Vector2(100, 100) } });
|
|
|
|
removeObjectStep(() => getObject(0));
|
|
|
|
assertGroups();
|
|
}
|
|
|
|
[Test]
|
|
public void TestAddMultipleObjects()
|
|
{
|
|
addMultipleObjectsStep();
|
|
|
|
assertGroups();
|
|
}
|
|
|
|
[Test]
|
|
public void TestRemoveEndObject()
|
|
{
|
|
addMultipleObjectsStep();
|
|
|
|
removeObjectStep(() => getObject(4));
|
|
|
|
assertGroups();
|
|
}
|
|
|
|
[Test]
|
|
public void TestRemoveStartObject()
|
|
{
|
|
addMultipleObjectsStep();
|
|
|
|
removeObjectStep(() => getObject(0));
|
|
|
|
assertGroups();
|
|
}
|
|
|
|
[Test]
|
|
public void TestRemoveMiddleObject()
|
|
{
|
|
addMultipleObjectsStep();
|
|
|
|
removeObjectStep(() => getObject(2));
|
|
|
|
assertGroups();
|
|
}
|
|
|
|
[Test]
|
|
public void TestMoveObject()
|
|
{
|
|
addMultipleObjectsStep();
|
|
|
|
AddStep("move hitobject", () =>
|
|
{
|
|
var manualClock = new ManualClock();
|
|
followPointRenderer.Clock = new FramedClock(manualClock);
|
|
|
|
manualClock.CurrentTime = getObject(1).HitObject.StartTime;
|
|
followPointRenderer.UpdateSubTree();
|
|
|
|
getObject(2).HitObject.Position = new Vector2(300, 100);
|
|
});
|
|
|
|
assertGroups();
|
|
assertDirections();
|
|
}
|
|
|
|
[TestCase(0, 0)] // Start -> Start
|
|
[TestCase(0, 2)] // Start -> Middle
|
|
[TestCase(0, 5)] // Start -> End
|
|
[TestCase(2, 0)] // Middle -> Start
|
|
[TestCase(1, 3)] // Middle -> Middle (forwards)
|
|
[TestCase(3, 1)] // Middle -> Middle (backwards)
|
|
[TestCase(4, 0)] // End -> Start
|
|
[TestCase(4, 2)] // End -> Middle
|
|
[TestCase(4, 4)] // End -> End
|
|
public void TestReorderObjects(int startIndex, int endIndex)
|
|
{
|
|
addMultipleObjectsStep();
|
|
|
|
reorderObjectStep(startIndex, endIndex);
|
|
|
|
assertGroups();
|
|
}
|
|
|
|
[Test]
|
|
public void TestStackedObjects()
|
|
{
|
|
addObjectsStep(() => new OsuHitObject[]
|
|
{
|
|
new HitCircle { Position = new Vector2(300, 100) },
|
|
new HitCircle
|
|
{
|
|
Position = new Vector2(300, 300),
|
|
StackHeight = 20
|
|
},
|
|
});
|
|
|
|
assertDirections();
|
|
}
|
|
|
|
private void addMultipleObjectsStep() => addObjectsStep(() => new OsuHitObject[]
|
|
{
|
|
new HitCircle { Position = new Vector2(100, 100) },
|
|
new HitCircle { Position = new Vector2(200, 200) },
|
|
new HitCircle { Position = new Vector2(300, 300) },
|
|
new HitCircle { Position = new Vector2(400, 400) },
|
|
new HitCircle { Position = new Vector2(500, 500) },
|
|
});
|
|
|
|
private void addObjectsStep(Func<OsuHitObject[]> ctorFunc)
|
|
{
|
|
AddStep("add hitobjects", () =>
|
|
{
|
|
var objects = ctorFunc();
|
|
|
|
for (int i = 0; i < objects.Length; i++)
|
|
{
|
|
objects[i].StartTime = Time.Current + 1000 + 500 * (i + 1);
|
|
objects[i].ApplyDefaults(new ControlPointInfo(), new BeatmapDifficulty());
|
|
|
|
DrawableOsuHitObject drawableObject = null;
|
|
|
|
switch (objects[i])
|
|
{
|
|
case HitCircle circle:
|
|
drawableObject = new DrawableHitCircle(circle);
|
|
break;
|
|
|
|
case Slider slider:
|
|
drawableObject = new DrawableSlider(slider);
|
|
break;
|
|
|
|
case Spinner spinner:
|
|
drawableObject = new DrawableSpinner(spinner);
|
|
break;
|
|
}
|
|
|
|
hitObjectContainer.Add(drawableObject);
|
|
followPointRenderer.AddFollowPoints(objects[i]);
|
|
}
|
|
});
|
|
}
|
|
|
|
private void removeObjectStep(Func<DrawableOsuHitObject> getFunc)
|
|
{
|
|
AddStep("remove hitobject", () =>
|
|
{
|
|
var drawableObject = getFunc.Invoke();
|
|
|
|
hitObjectContainer.Remove(drawableObject, false);
|
|
followPointRenderer.RemoveFollowPoints(drawableObject.HitObject);
|
|
});
|
|
}
|
|
|
|
private void reorderObjectStep(int startIndex, int endIndex)
|
|
{
|
|
AddStep($"move object {startIndex} to {endIndex}", () =>
|
|
{
|
|
DrawableOsuHitObject toReorder = getObject(startIndex);
|
|
|
|
double targetTime;
|
|
if (endIndex < hitObjectContainer.Count)
|
|
targetTime = getObject(endIndex).HitObject.StartTime - 1;
|
|
else
|
|
targetTime = getObject(hitObjectContainer.Count - 1).HitObject.StartTime + 1;
|
|
|
|
hitObjectContainer.Remove(toReorder, false);
|
|
toReorder.HitObject.StartTime = targetTime;
|
|
hitObjectContainer.Add(toReorder);
|
|
});
|
|
}
|
|
|
|
private void assertGroups()
|
|
{
|
|
AddAssert("has correct group count", () => followPointRenderer.Entries.Count == hitObjectContainer.Count);
|
|
AddAssert("group endpoints are correct", () =>
|
|
{
|
|
for (int i = 0; i < hitObjectContainer.Count; i++)
|
|
{
|
|
DrawableOsuHitObject expectedStart = getObject(i);
|
|
DrawableOsuHitObject expectedEnd = i < hitObjectContainer.Count - 1 ? getObject(i + 1) : null;
|
|
|
|
if (getEntry(i).Start != expectedStart.HitObject)
|
|
throw new AssertionException($"Object {i} expected to be the start of group {i}.");
|
|
|
|
if (getEntry(i).End != expectedEnd?.HitObject)
|
|
throw new AssertionException($"Object {(expectedEnd == null ? "null" : i.ToString())} expected to be the end of group {i}.");
|
|
}
|
|
|
|
return true;
|
|
});
|
|
}
|
|
|
|
private void assertDirections()
|
|
{
|
|
AddAssert("group directions are correct", () =>
|
|
{
|
|
for (int i = 0; i < hitObjectContainer.Count; i++)
|
|
{
|
|
DrawableOsuHitObject expectedStart = getObject(i);
|
|
DrawableOsuHitObject expectedEnd = i < hitObjectContainer.Count - 1 ? getObject(i + 1) : null;
|
|
|
|
if (expectedEnd == null)
|
|
continue;
|
|
|
|
var manualClock = new ManualClock();
|
|
followPointRenderer.Clock = new FramedClock(manualClock);
|
|
|
|
manualClock.CurrentTime = expectedStart.HitObject.StartTime;
|
|
followPointRenderer.UpdateSubTree();
|
|
|
|
var points = getGroup(i).ChildrenOfType<FollowPoint>().ToArray();
|
|
if (points.Length == 0)
|
|
continue;
|
|
|
|
float expectedDirection = MathF.Atan2(expectedStart.Position.Y - expectedEnd.Position.Y, expectedStart.Position.X - expectedEnd.Position.X);
|
|
float realDirection = MathF.Atan2(expectedStart.Position.Y - points[^1].Position.Y, expectedStart.Position.X - points[^1].Position.X);
|
|
|
|
if (!Precision.AlmostEquals(expectedDirection, realDirection))
|
|
throw new AssertionException($"Expected group {i} in direction {expectedDirection}, but was {realDirection}.");
|
|
}
|
|
|
|
return true;
|
|
});
|
|
}
|
|
|
|
private DrawableOsuHitObject getObject(int index) => hitObjectContainer[index];
|
|
|
|
private FollowPointLifetimeEntry getEntry(int index) => followPointRenderer.Entries[index];
|
|
|
|
private FollowPointConnection getGroup(int index) => followPointRenderer.ChildrenOfType<FollowPointConnection>().Single(c => c.Entry == getEntry(index));
|
|
|
|
private class TestHitObjectContainer : Container<DrawableOsuHitObject>
|
|
{
|
|
protected override int Compare(Drawable x, Drawable y)
|
|
{
|
|
var osuX = (DrawableOsuHitObject)x;
|
|
var osuY = (DrawableOsuHitObject)y;
|
|
|
|
int compare = osuX.HitObject.StartTime.CompareTo(osuY.HitObject.StartTime);
|
|
|
|
if (compare == 0)
|
|
return base.Compare(x, y);
|
|
|
|
return compare;
|
|
}
|
|
}
|
|
}
|
|
}
|