osu/osu.Game/Rulesets/Mods/ModAdaptiveSpeed.cs

197 lines
7.3 KiB
C#
Raw Normal View History

2022-03-01 13:12:06 +00:00
// 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.
2022-03-03 05:09:29 +00:00
2022-03-01 13:12:06 +00:00
using System;
using System.Collections.Generic;
using System.Linq;
using osu.Framework.Audio;
using osu.Framework.Audio.Track;
using osu.Framework.Bindables;
using osu.Framework.Graphics.Audio;
using osu.Framework.Utils;
2022-03-01 13:12:06 +00:00
using osu.Game.Beatmaps;
using osu.Game.Configuration;
2022-03-03 02:43:04 +00:00
using osu.Game.Rulesets.Judgements;
2022-03-01 13:12:06 +00:00
using osu.Game.Rulesets.Objects;
using osu.Game.Rulesets.Objects.Drawables;
using osu.Game.Rulesets.Scoring;
using osu.Game.Rulesets.UI;
2022-03-01 13:12:06 +00:00
namespace osu.Game.Rulesets.Mods
{
public class ModAdaptiveSpeed : Mod, IApplicableToRate, IApplicableToDrawableHitObject, IApplicableToBeatmap, IUpdatableByPlayfield
2022-03-01 13:12:06 +00:00
{
/// <summary>
/// Adjust track rate using the average speed of the last x hits
/// </summary>
2022-03-02 13:07:57 +00:00
private const int average_count = 6;
2022-03-01 13:12:06 +00:00
public override string Name => "Adaptive Speed";
public override string Acronym => "AS";
public override string Description => "Let track speed adapt to you.";
public override ModType Type => ModType.Fun;
public override double ScoreMultiplier => 1;
public override Type[] IncompatibleMods => new[] { typeof(ModRateAdjust), typeof(ModTimeRamp) };
2022-03-01 13:50:17 +00:00
[SettingSource("Initial rate", "The starting speed of the track")]
public BindableNumber<double> InitialRate { get; } = new BindableDouble
{
MinValue = 0.5,
MaxValue = 2,
Default = 1,
Value = 1,
Precision = 0.01
2022-03-01 13:50:17 +00:00
};
2022-03-01 13:12:06 +00:00
[SettingSource("Adjust pitch", "Should pitch be adjusted with speed")]
public BindableBool AdjustPitch { get; } = new BindableBool
{
Default = true,
Value = true
};
public BindableNumber<double> SpeedChange { get; } = new BindableDouble
{
MinValue = min_allowable_rate,
MaxValue = max_allowable_rate,
2022-03-01 13:12:06 +00:00
Default = 1,
Value = 1
2022-03-01 13:12:06 +00:00
};
// The two constants below denote the maximum allowable range of rates that `SpeedChange` can take.
// The range is purposefully wider than the range of values that `InitialRate` allows
// in order to give some leeway for change even when extreme initial rates are chosen.
private const double min_allowable_rate = 0.4f;
private const double max_allowable_rate = 2.5f;
2022-03-01 13:12:06 +00:00
private ITrack track;
private double targetRate = 1d;
2022-03-01 13:12:06 +00:00
2022-03-03 02:18:36 +00:00
private readonly List<double> recentRates = Enumerable.Repeat(1d, average_count).ToList();
2022-03-01 13:12:06 +00:00
2022-03-03 05:03:53 +00:00
/// <summary>
/// Rate for a hit is calculated using the end time of another hit object earlier in time,
/// caching them here for easy access
/// </summary>
2022-03-01 13:12:06 +00:00
private readonly Dictionary<HitObject, double> previousEndTimes = new Dictionary<HitObject, double>();
2022-03-03 05:03:53 +00:00
/// <summary>
/// Record the value removed from <see cref="recentRates"/> when an object is hit for rewind support
/// </summary>
private readonly Dictionary<HitObject, double> dequeuedRates = new Dictionary<HitObject, double>();
2022-03-01 13:12:06 +00:00
public ModAdaptiveSpeed()
{
InitialRate.BindValueChanged(val =>
{
SpeedChange.Value = val.NewValue;
targetRate = val.NewValue;
});
AdjustPitch.BindValueChanged(adjustPitchChanged);
2022-03-01 13:12:06 +00:00
}
public void ApplyToTrack(ITrack track)
{
this.track = track;
2022-03-01 13:50:17 +00:00
InitialRate.TriggerChange();
2022-03-01 13:12:06 +00:00
AdjustPitch.TriggerChange();
2022-03-01 13:50:17 +00:00
recentRates.Clear();
2022-03-03 02:18:36 +00:00
recentRates.AddRange(Enumerable.Repeat(InitialRate.Value, average_count));
2022-03-01 13:12:06 +00:00
}
public void ApplyToSample(DrawableSample sample)
{
sample.AddAdjustment(AdjustableProperty.Frequency, SpeedChange);
}
public void Update(Playfield playfield)
{
SpeedChange.Value = Interpolation.DampContinuously(SpeedChange.Value, targetRate, 50, playfield.Clock.ElapsedFrameTime);
}
public double ApplyToRate(double time, double rate = 1) => rate * InitialRate.Value;
2022-03-01 13:12:06 +00:00
public void ApplyToDrawableHitObject(DrawableHitObject drawable)
{
drawable.OnNewResult += (o, result) =>
{
if (dequeuedRates.ContainsKey(result.HitObject)) return;
2022-03-03 02:43:04 +00:00
if (!shouldProcessResult(result)) return;
2022-03-01 13:12:06 +00:00
double prevEndTime = previousEndTimes[result.HitObject];
recentRates.Add(Math.Clamp((result.HitObject.GetEndTime() - prevEndTime) / (result.TimeAbsolute - prevEndTime) * SpeedChange.Value, min_allowable_rate, max_allowable_rate));
dequeuedRates.Add(result.HitObject, recentRates[0]);
recentRates.RemoveAt(0);
targetRate = recentRates.Average();
};
drawable.OnRevertResult += (o, result) =>
{
if (!dequeuedRates.ContainsKey(result.HitObject)) return;
2022-03-03 02:43:04 +00:00
if (!shouldProcessResult(result)) return;
recentRates.Insert(0, dequeuedRates[result.HitObject]);
recentRates.RemoveAt(recentRates.Count - 1);
dequeuedRates.Remove(result.HitObject);
2022-03-01 13:12:06 +00:00
targetRate = recentRates.Average();
2022-03-01 13:12:06 +00:00
};
}
public void ApplyToBeatmap(IBeatmap beatmap)
{
var hitObjects = getAllApplicableHitObjects(beatmap.HitObjects).ToList();
2022-03-03 03:21:20 +00:00
var endTimes = hitObjects.Select(x => x.GetEndTime()).OrderBy(x => x).Distinct().ToList();
foreach (HitObject hitObject in hitObjects)
2022-03-01 13:12:06 +00:00
{
2022-03-03 03:21:20 +00:00
int index = endTimes.BinarySearch(hitObject.GetEndTime());
if (index < 0) index = ~index; // BinarySearch returns the next larger element in bitwise complement if there's no exact match
index -= 1;
2022-03-03 03:21:20 +00:00
if (index >= 0)
previousEndTimes.Add(hitObject, endTimes[index]);
}
}
2022-03-03 05:09:29 +00:00
private void adjustPitchChanged(ValueChangedEvent<bool> adjustPitchSetting)
{
track?.RemoveAdjustment(adjustmentForPitchSetting(adjustPitchSetting.OldValue), SpeedChange);
track?.AddAdjustment(adjustmentForPitchSetting(adjustPitchSetting.NewValue), SpeedChange);
}
private AdjustableProperty adjustmentForPitchSetting(bool adjustPitchSettingValue)
=> adjustPitchSettingValue ? AdjustableProperty.Frequency : AdjustableProperty.Tempo;
private IEnumerable<HitObject> getAllApplicableHitObjects(IEnumerable<HitObject> hitObjects)
{
foreach (var hitObject in hitObjects)
{
if (!(hitObject.HitWindows is HitWindows.EmptyHitWindows))
yield return hitObject;
2022-03-01 13:12:06 +00:00
foreach (HitObject nested in getAllApplicableHitObjects(hitObject.NestedHitObjects))
yield return nested;
2022-03-01 13:12:06 +00:00
}
}
2022-03-03 02:43:04 +00:00
private bool shouldProcessResult(JudgementResult result)
{
if (!result.IsHit) return false;
if (!result.Type.AffectsAccuracy()) return false;
if (!previousEndTimes.ContainsKey(result.HitObject)) return false;
return true;
}
2022-03-01 13:12:06 +00:00
}
}