diff --git a/osu.Game/OsuGame.cs b/osu.Game/OsuGame.cs
index 1b2fd658f4..113e9bbe24 100644
--- a/osu.Game/OsuGame.cs
+++ b/osu.Game/OsuGame.cs
@@ -315,8 +315,12 @@ namespace osu.Game
/// The user should have already requested this interactively.
///
/// The beatmap to select.
- public void PresentBeatmap(BeatmapSetInfo beatmap)
+ /// Predicate used to find a difficulty to select
+ public void PresentBeatmap(BeatmapSetInfo beatmap, Predicate findPredicate = null)
{
+ // Use this predicate if non was provided. This will try to find some difficulty from current ruleset so we wouldn't have to change rulesets
+ findPredicate ??= b => b.Ruleset.Equals(Ruleset.Value);
+
var databasedSet = beatmap.OnlineBeatmapSetID != null
? BeatmapManager.QueryBeatmapSet(s => s.OnlineBeatmapSetID == beatmap.OnlineBeatmapSetID)
: BeatmapManager.QueryBeatmapSet(s => s.Hash == beatmap.Hash);
@@ -334,13 +338,13 @@ namespace osu.Game
menuScreen.LoadToSolo();
// we might even already be at the song
- if (Beatmap.Value.BeatmapSetInfo.Hash == databasedSet.Hash)
+ if (Beatmap.Value.BeatmapSetInfo.Hash == databasedSet.Hash && findPredicate(Beatmap.Value.BeatmapInfo))
{
return;
}
- // Use first beatmap available for current ruleset, else switch ruleset.
- var first = databasedSet.Beatmaps.Find(b => b.Ruleset.Equals(Ruleset.Value)) ?? databasedSet.Beatmaps.First();
+ // Find first beatmap that matches our predicate.
+ var first = databasedSet.Beatmaps.Find(findPredicate) ?? databasedSet.Beatmaps.First();
Ruleset.Value = first.Ruleset;
Beatmap.Value = BeatmapManager.GetWorkingBeatmap(first);
diff --git a/osu.Game/Overlays/BeatmapSet/Header.cs b/osu.Game/Overlays/BeatmapSet/Header.cs
index 29c259b7f8..a60b9e04b1 100644
--- a/osu.Game/Overlays/BeatmapSet/Header.cs
+++ b/osu.Game/Overlays/BeatmapSet/Header.cs
@@ -277,7 +277,8 @@ namespace osu.Game.Overlays.BeatmapSet
downloadButtonsContainer.Child = new PanelDownloadButton(BeatmapSet.Value)
{
Width = 50,
- RelativeSizeAxes = Axes.Y
+ RelativeSizeAxes = Axes.Y,
+ CurrentBeatmap = Picker.Beatmap.GetBoundCopy()
};
break;
diff --git a/osu.Game/Overlays/Direct/PanelDownloadButton.cs b/osu.Game/Overlays/Direct/PanelDownloadButton.cs
index 1b3657f010..eae2f3353c 100644
--- a/osu.Game/Overlays/Direct/PanelDownloadButton.cs
+++ b/osu.Game/Overlays/Direct/PanelDownloadButton.cs
@@ -1,7 +1,9 @@
// Copyright (c) ppy Pty Ltd . Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.
+using System;
using osu.Framework.Allocation;
+using osu.Framework.Bindables;
using osu.Framework.Graphics;
using osu.Game.Beatmaps;
using osu.Game.Graphics.Containers;
@@ -16,6 +18,8 @@ namespace osu.Game.Overlays.Direct
private readonly bool noVideo;
+ public Bindable CurrentBeatmap = new Bindable();
+
private readonly ShakeContainer shakeContainer;
private readonly DownloadButton button;
@@ -62,7 +66,11 @@ namespace osu.Game.Overlays.Direct
break;
case DownloadState.LocallyAvailable:
- game?.PresentBeatmap(BeatmapSet.Value);
+ Predicate findPredicate = null;
+ if (CurrentBeatmap.Value != null)
+ findPredicate = b => b.OnlineBeatmapID == CurrentBeatmap.Value.OnlineBeatmapID;
+
+ game?.PresentBeatmap(BeatmapSet.Value, findPredicate);
break;
default: