mirror of https://github.com/ppy/osu
109 lines
4.4 KiB
C#
109 lines
4.4 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.
|
|
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using osu.Framework.Extensions.ObjectExtensions;
|
|
using osu.Framework.Logging;
|
|
using osu.Framework.Platform;
|
|
using osu.Framework.Threading;
|
|
using osu.Game.Database;
|
|
using osu.Game.Online.API;
|
|
using osu.Game.Rulesets.Objects.Types;
|
|
|
|
namespace osu.Game.Beatmaps
|
|
{
|
|
public class BeatmapUpdater : IBeatmapUpdater
|
|
{
|
|
private readonly IWorkingBeatmapCache workingBeatmapCache;
|
|
|
|
private readonly BeatmapDifficultyCache difficultyCache;
|
|
|
|
private readonly BeatmapUpdaterMetadataLookup metadataLookup;
|
|
|
|
private const int update_queue_request_concurrency = 4;
|
|
|
|
private readonly ThreadedTaskScheduler updateScheduler = new ThreadedTaskScheduler(update_queue_request_concurrency, nameof(BeatmapUpdaterMetadataLookup));
|
|
|
|
public BeatmapUpdater(IWorkingBeatmapCache workingBeatmapCache, BeatmapDifficultyCache difficultyCache, IAPIProvider api, Storage storage)
|
|
{
|
|
this.workingBeatmapCache = workingBeatmapCache;
|
|
this.difficultyCache = difficultyCache;
|
|
|
|
metadataLookup = new BeatmapUpdaterMetadataLookup(api, storage);
|
|
}
|
|
|
|
public void Queue(Live<BeatmapSetInfo> beatmapSet, MetadataLookupScope lookupScope = MetadataLookupScope.LocalCacheFirst)
|
|
{
|
|
Logger.Log($"Queueing change for local beatmap {beatmapSet}");
|
|
Task.Factory.StartNew(() => beatmapSet.PerformRead(b => Process(b, lookupScope)), default, TaskCreationOptions.HideScheduler | TaskCreationOptions.RunContinuationsAsynchronously,
|
|
updateScheduler);
|
|
}
|
|
|
|
public void Process(BeatmapSetInfo beatmapSet, MetadataLookupScope lookupScope = MetadataLookupScope.LocalCacheFirst)
|
|
{
|
|
beatmapSet.Realm!.Write(_ =>
|
|
{
|
|
// Before we use below, we want to invalidate.
|
|
workingBeatmapCache.Invalidate(beatmapSet);
|
|
|
|
if (lookupScope != MetadataLookupScope.None)
|
|
metadataLookup.Update(beatmapSet, lookupScope == MetadataLookupScope.OnlineFirst);
|
|
|
|
foreach (var beatmap in beatmapSet.Beatmaps)
|
|
{
|
|
difficultyCache.Invalidate(beatmap);
|
|
|
|
var working = workingBeatmapCache.GetWorkingBeatmap(beatmap);
|
|
var ruleset = working.BeatmapInfo.Ruleset.CreateInstance();
|
|
|
|
Debug.Assert(ruleset != null);
|
|
|
|
var calculator = ruleset.CreateDifficultyCalculator(working);
|
|
|
|
beatmap.StarRating = calculator.Calculate().StarRating;
|
|
beatmap.Length = working.Beatmap.CalculatePlayableLength();
|
|
beatmap.BPM = 60000 / working.Beatmap.GetMostCommonBeatLength();
|
|
beatmap.EndTimeObjectCount = working.Beatmap.HitObjects.Count(h => h is IHasDuration);
|
|
beatmap.TotalObjectCount = working.Beatmap.HitObjects.Count;
|
|
}
|
|
|
|
// And invalidate again afterwards as re-fetching the most up-to-date database metadata will be required.
|
|
workingBeatmapCache.Invalidate(beatmapSet);
|
|
});
|
|
}
|
|
|
|
public void ProcessObjectCounts(BeatmapInfo beatmapInfo, MetadataLookupScope lookupScope = MetadataLookupScope.LocalCacheFirst)
|
|
{
|
|
beatmapInfo.Realm!.Write(_ =>
|
|
{
|
|
// Before we use below, we want to invalidate.
|
|
workingBeatmapCache.Invalidate(beatmapInfo);
|
|
|
|
var working = workingBeatmapCache.GetWorkingBeatmap(beatmapInfo);
|
|
var beatmap = working.Beatmap;
|
|
|
|
beatmapInfo.EndTimeObjectCount = beatmap.HitObjects.Count(h => h is IHasDuration);
|
|
beatmapInfo.TotalObjectCount = beatmap.HitObjects.Count;
|
|
|
|
// And invalidate again afterwards as re-fetching the most up-to-date database metadata will be required.
|
|
workingBeatmapCache.Invalidate(beatmapInfo);
|
|
});
|
|
}
|
|
|
|
#region Implementation of IDisposable
|
|
|
|
public void Dispose()
|
|
{
|
|
if (metadataLookup.IsNotNull())
|
|
metadataLookup.Dispose();
|
|
|
|
if (updateScheduler.IsNotNull())
|
|
updateScheduler.Dispose();
|
|
}
|
|
|
|
#endregion
|
|
}
|
|
}
|