mirror of https://github.com/ppy/osu
353 lines
14 KiB
C#
353 lines
14 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.IO;
|
|
using System.Linq;
|
|
using JetBrains.Annotations;
|
|
using osu.Framework.Audio;
|
|
using osu.Framework.Audio.Track;
|
|
using osu.Framework.Graphics.Rendering;
|
|
using osu.Framework.Graphics.Rendering.Dummy;
|
|
using osu.Framework.Graphics.Textures;
|
|
using osu.Framework.IO.Stores;
|
|
using osu.Framework.Lists;
|
|
using osu.Framework.Logging;
|
|
using osu.Framework.Platform;
|
|
using osu.Framework.Statistics;
|
|
using osu.Game.Beatmaps.Formats;
|
|
using osu.Game.Database;
|
|
using osu.Game.IO;
|
|
using osu.Game.Skinning;
|
|
using osu.Game.Storyboards;
|
|
|
|
namespace osu.Game.Beatmaps
|
|
{
|
|
public class WorkingBeatmapCache : IBeatmapResourceProvider, IWorkingBeatmapCache
|
|
{
|
|
private readonly WeakList<BeatmapManagerWorkingBeatmap> workingCache = new WeakList<BeatmapManagerWorkingBeatmap>();
|
|
|
|
/// <summary>
|
|
/// Beatmap files may specify this filename to denote that they don't have an audio track.
|
|
/// </summary>
|
|
private const string virtual_track_filename = @"virtual";
|
|
|
|
/// <summary>
|
|
/// A default representation of a WorkingBeatmap to use when no beatmap is available.
|
|
/// </summary>
|
|
public readonly WorkingBeatmap DefaultBeatmap;
|
|
|
|
private readonly AudioManager audioManager;
|
|
private readonly IResourceStore<byte[]> resources;
|
|
private readonly LargeTextureStore largeTextureStore;
|
|
private readonly LargeTextureStore beatmapPanelTextureStore;
|
|
private readonly ITrackStore trackStore;
|
|
private readonly IResourceStore<byte[]> files;
|
|
|
|
[CanBeNull]
|
|
private readonly GameHost host;
|
|
|
|
public WorkingBeatmapCache(ITrackStore trackStore, AudioManager audioManager, IResourceStore<byte[]> resources, IResourceStore<byte[]> files, WorkingBeatmap defaultBeatmap = null,
|
|
GameHost host = null)
|
|
{
|
|
DefaultBeatmap = defaultBeatmap;
|
|
|
|
this.audioManager = audioManager;
|
|
this.resources = resources;
|
|
this.host = host;
|
|
this.files = files;
|
|
largeTextureStore = new LargeTextureStore(host?.Renderer ?? new DummyRenderer(), host?.CreateTextureLoaderStore(files));
|
|
beatmapPanelTextureStore = new LargeTextureStore(host?.Renderer ?? new DummyRenderer(), new BeatmapPanelBackgroundTextureLoaderStore(host?.CreateTextureLoaderStore(files)));
|
|
this.trackStore = trackStore;
|
|
}
|
|
|
|
public void Invalidate(BeatmapSetInfo info)
|
|
{
|
|
foreach (var b in info.Beatmaps)
|
|
Invalidate(b);
|
|
}
|
|
|
|
public void Invalidate(BeatmapInfo info)
|
|
{
|
|
lock (workingCache)
|
|
{
|
|
var working = workingCache.FirstOrDefault(w => info.Equals(w.BeatmapInfo));
|
|
|
|
if (working != null)
|
|
{
|
|
Logger.Log($"Invalidating working beatmap cache for {info}");
|
|
workingCache.Remove(working);
|
|
OnInvalidated?.Invoke(working);
|
|
}
|
|
}
|
|
}
|
|
|
|
public event Action<WorkingBeatmap> OnInvalidated;
|
|
|
|
public virtual WorkingBeatmap GetWorkingBeatmap(BeatmapInfo beatmapInfo)
|
|
{
|
|
if (beatmapInfo?.BeatmapSet == null)
|
|
return DefaultBeatmap;
|
|
|
|
lock (workingCache)
|
|
{
|
|
var working = workingCache.FirstOrDefault(w => beatmapInfo.Equals(w.BeatmapInfo));
|
|
|
|
if (working != null)
|
|
return working;
|
|
|
|
beatmapInfo = beatmapInfo.Detach();
|
|
|
|
workingCache.Add(working = new BeatmapManagerWorkingBeatmap(beatmapInfo, this));
|
|
|
|
// best effort; may be higher than expected.
|
|
GlobalStatistics.Get<int>("Beatmaps", $"Cached {nameof(WorkingBeatmap)}s").Value = workingCache.Count();
|
|
|
|
return working;
|
|
}
|
|
}
|
|
|
|
#region IResourceStorageProvider
|
|
|
|
TextureStore IBeatmapResourceProvider.LargeTextureStore => largeTextureStore;
|
|
TextureStore IBeatmapResourceProvider.BeatmapPanelTextureStore => beatmapPanelTextureStore;
|
|
ITrackStore IBeatmapResourceProvider.Tracks => trackStore;
|
|
IRenderer IStorageResourceProvider.Renderer => host?.Renderer ?? new DummyRenderer();
|
|
AudioManager IStorageResourceProvider.AudioManager => audioManager;
|
|
RealmAccess IStorageResourceProvider.RealmAccess => null;
|
|
IResourceStore<byte[]> IStorageResourceProvider.Files => files;
|
|
IResourceStore<byte[]> IStorageResourceProvider.Resources => resources;
|
|
IResourceStore<TextureUpload> IStorageResourceProvider.CreateTextureLoaderStore(IResourceStore<byte[]> underlyingStore) => host?.CreateTextureLoaderStore(underlyingStore);
|
|
|
|
#endregion
|
|
|
|
private class BeatmapManagerWorkingBeatmap : WorkingBeatmap
|
|
{
|
|
[NotNull]
|
|
private readonly IBeatmapResourceProvider resources;
|
|
|
|
public BeatmapManagerWorkingBeatmap(BeatmapInfo beatmapInfo, [NotNull] IBeatmapResourceProvider resources)
|
|
: base(beatmapInfo, resources.AudioManager)
|
|
{
|
|
this.resources = resources;
|
|
}
|
|
|
|
protected override IBeatmap GetBeatmap()
|
|
{
|
|
if (BeatmapInfo.Path == null)
|
|
return new Beatmap { BeatmapInfo = BeatmapInfo };
|
|
|
|
try
|
|
{
|
|
string fileStorePath = BeatmapSetInfo.GetPathForFile(BeatmapInfo.Path);
|
|
|
|
// TODO: check validity of file
|
|
|
|
var stream = GetStream(fileStorePath);
|
|
|
|
if (stream == null)
|
|
{
|
|
Logger.Log($"Beatmap failed to load (file {BeatmapInfo.Path} not found on disk at expected location {fileStorePath}).", level: LogLevel.Error);
|
|
return null;
|
|
}
|
|
|
|
using (var reader = new LineBufferedReader(stream))
|
|
return Decoder.GetDecoder<Beatmap>(reader).Decode(reader);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.Error(e, "Beatmap failed to load");
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public override Texture GetPanelBackground() => getBackgroundFromStore(resources.BeatmapPanelTextureStore);
|
|
|
|
public override Texture GetBackground() => getBackgroundFromStore(resources.LargeTextureStore);
|
|
|
|
private Texture getBackgroundFromStore(TextureStore store)
|
|
{
|
|
if (string.IsNullOrEmpty(Metadata?.BackgroundFile))
|
|
return null;
|
|
|
|
try
|
|
{
|
|
string fileStorePath = BeatmapSetInfo.GetPathForFile(Metadata.BackgroundFile);
|
|
var texture = store.Get(fileStorePath);
|
|
|
|
if (texture == null)
|
|
{
|
|
Logger.Log($"Beatmap background failed to load (file {Metadata.BackgroundFile} not found on disk at expected location {fileStorePath}).");
|
|
return null;
|
|
}
|
|
|
|
return texture;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.Error(e, "Background failed to load");
|
|
return null;
|
|
}
|
|
}
|
|
|
|
protected override Track GetBeatmapTrack()
|
|
{
|
|
if (string.IsNullOrEmpty(Metadata?.AudioFile))
|
|
return null;
|
|
|
|
if (Metadata.AudioFile == virtual_track_filename)
|
|
return null;
|
|
|
|
try
|
|
{
|
|
string fileStorePath = BeatmapSetInfo.GetPathForFile(Metadata.AudioFile);
|
|
var track = resources.Tracks.Get(fileStorePath);
|
|
|
|
if (track == null)
|
|
{
|
|
Logger.Log($"Beatmap failed to load (file {Metadata.AudioFile} not found on disk at expected location {fileStorePath}).", level: LogLevel.Error);
|
|
return null;
|
|
}
|
|
|
|
return track;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.Error(e, "Track failed to load");
|
|
return null;
|
|
}
|
|
}
|
|
|
|
protected override Waveform GetWaveform()
|
|
{
|
|
if (string.IsNullOrEmpty(Metadata?.AudioFile))
|
|
return null;
|
|
|
|
if (Metadata.AudioFile == virtual_track_filename)
|
|
return null;
|
|
|
|
try
|
|
{
|
|
string fileStorePath = BeatmapSetInfo.GetPathForFile(Metadata.AudioFile);
|
|
|
|
var trackData = GetStream(fileStorePath);
|
|
|
|
if (trackData == null)
|
|
{
|
|
Logger.Log($"Beatmap waveform failed to load (file {Metadata.AudioFile} not found on disk at expected location {fileStorePath}).", level: LogLevel.Error);
|
|
return null;
|
|
}
|
|
|
|
return new Waveform(trackData);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.Error(e, "Waveform failed to load");
|
|
return null;
|
|
}
|
|
}
|
|
|
|
protected override Storyboard GetStoryboard()
|
|
{
|
|
Storyboard storyboard;
|
|
|
|
if (BeatmapInfo.Path == null)
|
|
return new Storyboard();
|
|
|
|
try
|
|
{
|
|
string fileStorePath = BeatmapSetInfo.GetPathForFile(BeatmapInfo.Path);
|
|
var beatmapFileStream = GetStream(fileStorePath);
|
|
|
|
if (beatmapFileStream == null)
|
|
{
|
|
Logger.Log($"Beatmap failed to load (file {BeatmapInfo.Path} not found on disk at expected location {fileStorePath})", level: LogLevel.Error);
|
|
return new Storyboard();
|
|
}
|
|
|
|
using (var reader = new LineBufferedReader(beatmapFileStream))
|
|
{
|
|
var decoder = Decoder.GetDecoder<Storyboard>(reader);
|
|
|
|
Stream storyboardFileStream = null;
|
|
|
|
string mainStoryboardFilename = getMainStoryboardFilename(BeatmapSetInfo.Metadata);
|
|
|
|
if (BeatmapSetInfo?.Files.FirstOrDefault(f => f.Filename.Equals(mainStoryboardFilename, StringComparison.OrdinalIgnoreCase))?.Filename is string
|
|
storyboardFilename)
|
|
{
|
|
string storyboardFileStorePath = BeatmapSetInfo?.GetPathForFile(storyboardFilename);
|
|
storyboardFileStream = GetStream(storyboardFileStorePath);
|
|
|
|
if (storyboardFileStream == null)
|
|
Logger.Log($"Storyboard failed to load (file {storyboardFilename} not found on disk at expected location {storyboardFileStorePath})", level: LogLevel.Error);
|
|
}
|
|
|
|
if (storyboardFileStream != null)
|
|
{
|
|
// Stand-alone storyboard was found, so parse in addition to the beatmap's local storyboard.
|
|
using (var secondaryReader = new LineBufferedReader(storyboardFileStream))
|
|
storyboard = decoder.Decode(reader, secondaryReader);
|
|
}
|
|
else
|
|
storyboard = decoder.Decode(reader);
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.Error(e, "Storyboard failed to load");
|
|
storyboard = new Storyboard();
|
|
}
|
|
|
|
storyboard.BeatmapInfo = BeatmapInfo;
|
|
|
|
return storyboard;
|
|
}
|
|
|
|
protected internal override ISkin GetSkin()
|
|
{
|
|
try
|
|
{
|
|
return new LegacyBeatmapSkin(BeatmapInfo, resources);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.Error(e, "Skin failed to load");
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public override Stream GetStream(string storagePath) => resources.Files.GetStream(storagePath);
|
|
|
|
private string getMainStoryboardFilename(IBeatmapMetadataInfo metadata)
|
|
{
|
|
// Matches stable implementation, because it's probably simpler than trying to do anything else.
|
|
// This may need to be reconsidered after we begin storing storyboards in the new editor.
|
|
return windowsFilenameStrip(
|
|
(metadata.Artist.Length > 0 ? metadata.Artist + @" - " + metadata.Title : Path.GetFileNameWithoutExtension(metadata.AudioFile))
|
|
+ (metadata.Author.Username.Length > 0 ? @" (" + metadata.Author.Username + @")" : string.Empty)
|
|
+ @".osb");
|
|
|
|
string windowsFilenameStrip(string entry)
|
|
{
|
|
// Inlined from Path.GetInvalidFilenameChars() to ensure the windows characters are used (to match stable).
|
|
char[] invalidCharacters =
|
|
{
|
|
'\x00', '\x01', '\x02', '\x03', '\x04', '\x05', '\x06', '\x07',
|
|
'\x08', '\x09', '\x0A', '\x0B', '\x0C', '\x0D', '\x0E', '\x0F', '\x10', '\x11', '\x12',
|
|
'\x13', '\x14', '\x15', '\x16', '\x17', '\x18', '\x19', '\x1A', '\x1B', '\x1C', '\x1D',
|
|
'\x1E', '\x1F', '\x22', '\x3C', '\x3E', '\x7C', ':', '*', '?', '\\', '/'
|
|
};
|
|
|
|
foreach (char c in invalidCharacters)
|
|
entry = entry.Replace(c.ToString(), string.Empty);
|
|
|
|
return entry;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|