mirror of https://github.com/ppy/osu
59 lines
1.9 KiB
C#
59 lines
1.9 KiB
C#
// Copyright (c) 2007-2018 ppy Pty Ltd <contact@ppy.sh>.
|
|
// Licensed under the MIT Licence - https://raw.githubusercontent.com/ppy/osu/master/LICENCE
|
|
|
|
using osu.Game.Rulesets.Objects;
|
|
using System.Collections.Generic;
|
|
using osu.Game.Rulesets.Mods;
|
|
using osu.Framework.Timing;
|
|
using System.Linq;
|
|
using osu.Framework.Extensions.IEnumerableExtensions;
|
|
|
|
namespace osu.Game.Beatmaps
|
|
{
|
|
public abstract class DifficultyCalculator
|
|
{
|
|
protected double TimeRate = 1;
|
|
|
|
public abstract double Calculate(Dictionary<string, double> categoryDifficulty = null);
|
|
}
|
|
|
|
public abstract class DifficultyCalculator<T> : DifficultyCalculator where T : HitObject
|
|
{
|
|
protected readonly Beatmap<T> Beatmap;
|
|
protected readonly Mod[] Mods;
|
|
|
|
protected DifficultyCalculator(Beatmap beatmap, Mod[] mods = null)
|
|
{
|
|
Beatmap = CreateBeatmapConverter(beatmap).Convert(beatmap);
|
|
Mods = mods ?? new Mod[0];
|
|
|
|
ApplyMods(Mods);
|
|
|
|
PreprocessHitObjects();
|
|
}
|
|
|
|
protected virtual void ApplyMods(Mod[] mods)
|
|
{
|
|
var clock = new StopwatchClock();
|
|
mods.OfType<IApplicableToClock>().ForEach(m => m.ApplyToClock(clock));
|
|
TimeRate = clock.Rate;
|
|
|
|
foreach (var mod in Mods.OfType<IApplicableToDifficulty>())
|
|
mod.ApplyToDifficulty(Beatmap.BeatmapInfo.BaseDifficulty);
|
|
|
|
foreach (var mod in mods.OfType<IApplicableToHitObject<T>>())
|
|
foreach (var obj in Beatmap.HitObjects)
|
|
mod.ApplyToHitObject(obj);
|
|
|
|
foreach (var h in Beatmap.HitObjects)
|
|
h.ApplyDefaults(Beatmap.ControlPointInfo, Beatmap.BeatmapInfo.BaseDifficulty);
|
|
}
|
|
|
|
protected virtual void PreprocessHitObjects()
|
|
{
|
|
}
|
|
|
|
protected abstract BeatmapConverter<T> CreateBeatmapConverter(Beatmap beatmap);
|
|
}
|
|
}
|