SongBrowserModel.cs 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748
  1. using SongBrowser.DataAccess;
  2. using SongCore.Utilities;
  3. using System;
  4. using System.Collections.Concurrent;
  5. using System.Collections.Generic;
  6. using System.Diagnostics;
  7. using System.IO;
  8. using System.Linq;
  9. using System.Text.RegularExpressions;
  10. using UnityEngine;
  11. using Logger = SongBrowser.Logging.Logger;
  12. namespace SongBrowser
  13. {
  14. public class SongBrowserModel
  15. {
  16. public static readonly string FilteredSongsCollectionName = CustomLevelLoader.kCustomLevelPackPrefixId + "SongBrowser_FilteredSongPack";
  17. public static readonly string PlaylistSongsCollectionName = "SongBrowser_PlaylistPack";
  18. private readonly String CUSTOM_SONGS_DIR = Path.Combine("Beat Saber_Data", "CustomLevels");
  19. private readonly DateTime EPOCH = new DateTime(1970, 1, 1);
  20. // song_browser_settings.xml
  21. private SongBrowserSettings _settings;
  22. // song list management
  23. private double _customSongDirLastWriteTime = 0;
  24. private readonly Dictionary<String, double> _cachedLastWriteTimes;
  25. private Dictionary<string, int> _levelIdToPlayCount;
  26. public BeatmapCharacteristicSO CurrentBeatmapCharacteristicSO;
  27. public static Func<IAnnotatedBeatmapLevelCollection, List<IPreviewBeatmapLevel>> CustomFilterHandler;
  28. public static Func<List<IPreviewBeatmapLevel>, List<IPreviewBeatmapLevel>> CustomSortHandler;
  29. public static Action<ConcurrentDictionary<string, CustomPreviewBeatmapLevel>> didFinishProcessingSongs;
  30. public bool SortWasMissingData { get; private set; } = false;
  31. /// <summary>
  32. /// Get the settings the model is using.
  33. /// </summary>
  34. public SongBrowserSettings Settings
  35. {
  36. get
  37. {
  38. return _settings;
  39. }
  40. }
  41. /// <summary>
  42. /// Get the last selected (stored in settings) level id.
  43. /// </summary>
  44. public String LastSelectedLevelId
  45. {
  46. get
  47. {
  48. return _settings.currentLevelId;
  49. }
  50. set
  51. {
  52. _settings.currentLevelId = value;
  53. _settings.Save();
  54. }
  55. }
  56. /// <summary>
  57. /// Constructor.
  58. /// </summary>
  59. public SongBrowserModel()
  60. {
  61. _cachedLastWriteTimes = new Dictionary<String, double>();
  62. _levelIdToPlayCount = new Dictionary<string, int>();
  63. }
  64. /// <summary>
  65. /// Init this model.
  66. /// </summary>
  67. /// <param name="songSelectionMasterView"></param>
  68. /// <param name="songListViewController"></param>
  69. public void Init()
  70. {
  71. _settings = SongBrowserSettings.Load();
  72. Logger.Info("Settings loaded, sorting mode is: {0}", _settings.sortMode);
  73. }
  74. /// <summary>
  75. /// Easy invert of toggling.
  76. /// </summary>
  77. public void ToggleInverting()
  78. {
  79. this.Settings.invertSortResults = !this.Settings.invertSortResults;
  80. }
  81. /// <summary>
  82. /// Get the song cache from the game.
  83. /// </summary>
  84. public void UpdateLevelRecords()
  85. {
  86. Stopwatch timer = new Stopwatch();
  87. timer.Start();
  88. // Calculate some information about the custom song dir
  89. String customSongsPath = Path.Combine(Environment.CurrentDirectory, CUSTOM_SONGS_DIR);
  90. String revSlashCustomSongPath = customSongsPath.Replace('\\', '/');
  91. double currentCustomSongDirLastWriteTIme = (File.GetLastWriteTimeUtc(customSongsPath) - EPOCH).TotalMilliseconds;
  92. bool customSongDirChanged = false;
  93. if (_customSongDirLastWriteTime != currentCustomSongDirLastWriteTIme)
  94. {
  95. customSongDirChanged = true;
  96. _customSongDirLastWriteTime = currentCustomSongDirLastWriteTIme;
  97. }
  98. if (!Directory.Exists(customSongsPath))
  99. {
  100. Logger.Error("CustomSong directory is missing...");
  101. return;
  102. }
  103. // Map some data for custom songs
  104. Regex r = new Regex(@"(\d+-\d+)", RegexOptions.IgnoreCase);
  105. Stopwatch lastWriteTimer = new Stopwatch();
  106. lastWriteTimer.Start();
  107. foreach (KeyValuePair<string, CustomPreviewBeatmapLevel> level in SongCore.Loader.CustomLevels)
  108. {
  109. // If we already know this levelID, don't both updating it.
  110. // SongLoader should filter duplicates but in case of failure we don't want to crash
  111. if (!_cachedLastWriteTimes.ContainsKey(level.Value.levelID) || customSongDirChanged)
  112. {
  113. double lastWriteTime = GetSongUserDate(level.Value);
  114. _cachedLastWriteTimes[level.Value.levelID] = lastWriteTime;
  115. }
  116. }
  117. lastWriteTimer.Stop();
  118. Logger.Info("Determining song download time and determining mappings took {0}ms", lastWriteTimer.ElapsedMilliseconds);
  119. // Update song Infos, directory tree, and sort
  120. this.UpdatePlayCounts();
  121. // Signal complete
  122. if (SongCore.Loader.CustomLevels.Count > 0)
  123. {
  124. didFinishProcessingSongs?.Invoke(SongCore.Loader.CustomLevels);
  125. }
  126. timer.Stop();
  127. Logger.Info("Updating songs infos took {0}ms", timer.ElapsedMilliseconds);
  128. }
  129. /// <summary>
  130. /// Try to get the date from the cover file, likely the most reliable.
  131. /// Fall back on the folders creation date.
  132. /// </summary>
  133. /// <param name="level"></param>
  134. /// <returns></returns>
  135. private double GetSongUserDate(CustomPreviewBeatmapLevel level)
  136. {
  137. var coverPath = Path.Combine(level.customLevelPath, level.standardLevelInfoSaveData.coverImageFilename);
  138. DateTime lastTime;
  139. if (File.Exists(coverPath))
  140. {
  141. var lastWriteTime = File.GetLastWriteTimeUtc(coverPath);
  142. var lastCreateTime = File.GetCreationTimeUtc(coverPath);
  143. lastTime = lastWriteTime > lastCreateTime ? lastWriteTime : lastCreateTime;
  144. }
  145. else
  146. {
  147. var lastCreateTime = File.GetCreationTimeUtc(level.customLevelPath);
  148. lastTime = lastCreateTime;
  149. }
  150. return (lastTime - EPOCH).TotalMilliseconds;
  151. }
  152. /// <summary>
  153. /// SongLoader doesn't fire event when we delete a song.
  154. /// </summary>
  155. /// <param name="levelCollection"></param>
  156. /// <param name="levelId"></param>
  157. public void RemoveSongFromLevelCollection(IAnnotatedBeatmapLevelCollection levelCollection, String levelId)
  158. {
  159. levelCollection.beatmapLevelCollection.beatmapLevels.ToList().RemoveAll(x => x.levelID == levelId);
  160. }
  161. /// <summary>
  162. /// Update the gameplay play counts.
  163. /// </summary>
  164. /// <param name="gameplayMode"></param>
  165. private void UpdatePlayCounts()
  166. {
  167. // Reset current playcounts
  168. _levelIdToPlayCount = new Dictionary<string, int>();
  169. // Build a map of levelId to sum of all playcounts and sort.
  170. PlayerDataModel playerData = Resources.FindObjectsOfTypeAll<PlayerDataModel>().FirstOrDefault();
  171. foreach (var levelData in playerData.playerData.levelsStatsData)
  172. {
  173. if (!_levelIdToPlayCount.ContainsKey(levelData.levelID))
  174. {
  175. _levelIdToPlayCount.Add(levelData.levelID, 0);
  176. }
  177. _levelIdToPlayCount[levelData.levelID] += levelData.playCount;
  178. }
  179. }
  180. /// <summary>
  181. /// Sort the song list based on the settings.
  182. /// </summary>
  183. public void ProcessSongList(IAnnotatedBeatmapLevelCollection selectedBeatmapCollection, LevelSelectionNavigationController navController)
  184. {
  185. Logger.Trace("ProcessSongList()");
  186. List<IPreviewBeatmapLevel> unsortedSongs = null;
  187. List<IPreviewBeatmapLevel> filteredSongs = null;
  188. List<IPreviewBeatmapLevel> sortedSongs = null;
  189. // Abort
  190. if (selectedBeatmapCollection == null)
  191. {
  192. Logger.Debug("Cannot process songs yet, no level collection selected...");
  193. return;
  194. }
  195. Logger.Debug($"Using songs from level collection: {selectedBeatmapCollection.collectionName} [num={selectedBeatmapCollection.beatmapLevelCollection.beatmapLevels.Length}");
  196. unsortedSongs = selectedBeatmapCollection.beatmapLevelCollection.beatmapLevels.ToList();
  197. // filter
  198. Logger.Debug($"Starting filtering songs by {_settings.filterMode}");
  199. Stopwatch stopwatch = Stopwatch.StartNew();
  200. switch (_settings.filterMode)
  201. {
  202. case SongFilterMode.Favorites:
  203. filteredSongs = FilterFavorites(unsortedSongs);
  204. break;
  205. case SongFilterMode.Search:
  206. filteredSongs = FilterSearch(unsortedSongs);
  207. break;
  208. case SongFilterMode.Ranked:
  209. filteredSongs = FilterRanked(unsortedSongs, true, false);
  210. break;
  211. case SongFilterMode.Unranked:
  212. filteredSongs = FilterRanked(unsortedSongs, false, true);
  213. break;
  214. case SongFilterMode.Custom:
  215. Logger.Info("Song filter mode set to custom. Deferring filter behaviour to another mod.");
  216. filteredSongs = CustomFilterHandler != null ? CustomFilterHandler.Invoke(selectedBeatmapCollection) : unsortedSongs;
  217. break;
  218. case SongFilterMode.None:
  219. default:
  220. Logger.Info("No song filter selected...");
  221. filteredSongs = unsortedSongs;
  222. break;
  223. }
  224. stopwatch.Stop();
  225. Logger.Info("Filtering songs took {0}ms", stopwatch.ElapsedMilliseconds);
  226. // sort
  227. Logger.Debug($"Starting to sort songs by {_settings.sortMode}");
  228. stopwatch = Stopwatch.StartNew();
  229. SortWasMissingData = false;
  230. switch (_settings.sortMode)
  231. {
  232. case SongSortMode.Original:
  233. sortedSongs = SortOriginal(filteredSongs);
  234. break;
  235. case SongSortMode.Newest:
  236. sortedSongs = SortNewest(filteredSongs);
  237. break;
  238. case SongSortMode.Author:
  239. sortedSongs = SortAuthor(filteredSongs);
  240. break;
  241. case SongSortMode.UpVotes:
  242. sortedSongs = SortUpVotes(filteredSongs);
  243. break;
  244. case SongSortMode.PlayCount:
  245. sortedSongs = SortBeatSaverPlayCount(filteredSongs);
  246. break;
  247. case SongSortMode.Rating:
  248. sortedSongs = SortBeatSaverRating(filteredSongs);
  249. break;
  250. case SongSortMode.Heat:
  251. sortedSongs = SortBeatSaverHeat(filteredSongs);
  252. break;
  253. case SongSortMode.YourPlayCount:
  254. sortedSongs = SortPlayCount(filteredSongs);
  255. break;
  256. case SongSortMode.PP:
  257. sortedSongs = SortPerformancePoints(filteredSongs);
  258. break;
  259. case SongSortMode.Stars:
  260. sortedSongs = SortStars(filteredSongs);
  261. break;
  262. case SongSortMode.Random:
  263. sortedSongs = SortRandom(filteredSongs);
  264. break;
  265. case SongSortMode.Custom:
  266. sortedSongs = CustomSortHandler != null ? CustomSortHandler.Invoke(filteredSongs) : filteredSongs;
  267. break;
  268. case SongSortMode.Default:
  269. default:
  270. sortedSongs = SortSongName(filteredSongs);
  271. break;
  272. }
  273. if (this.Settings.invertSortResults && _settings.sortMode != SongSortMode.Random)
  274. {
  275. sortedSongs.Reverse();
  276. }
  277. stopwatch.Stop();
  278. Logger.Info("Sorting songs took {0}ms", stopwatch.ElapsedMilliseconds);
  279. // Still hacking in a custom level pack
  280. // Asterisk the pack name so it is identifable as filtered.
  281. var packName = selectedBeatmapCollection.collectionName;
  282. if (packName == null)
  283. {
  284. packName = "";
  285. }
  286. if (!packName.EndsWith("*") && _settings.filterMode != SongFilterMode.None)
  287. {
  288. packName += "*";
  289. }
  290. // Some level categories have a null cover image, supply something, it won't show it anyway
  291. var coverImage = selectedBeatmapCollection.coverImage;
  292. if (coverImage == null)
  293. {
  294. coverImage = BeatSaberMarkupLanguage.Utilities.ImageResources.BlankSprite;
  295. }
  296. Logger.Debug("Creating filtered level pack...");
  297. BeatmapLevelPack levelPack = new BeatmapLevelPack(SongBrowserModel.FilteredSongsCollectionName, packName, selectedBeatmapCollection.collectionName, coverImage, new BeatmapLevelCollection(sortedSongs.ToArray()));
  298. /*
  299. public virtual void SetData(
  300. IAnnotatedBeatmapLevelCollection annotatedBeatmapLevelCollection,
  301. bool showPackHeader, bool showPlayerStats, bool showPracticeButton,
  302. string actionButtonText,
  303. GameObject noDataInfoPrefab, BeatmapDifficultyMask allowedBeatmapDifficultyMask, BeatmapCharacteristicSO[] notAllowedCharacteristics);
  304. */
  305. Logger.Debug("Acquiring necessary fields to call SetData(pack)...");
  306. LevelCollectionNavigationController lcnvc = navController.GetPrivateField<LevelCollectionNavigationController>("_levelCollectionNavigationController");
  307. var _showPlayerStatsInDetailView = navController.GetPrivateField<bool>("_showPlayerStatsInDetailView");
  308. var _hidePracticeButton = navController.GetPrivateField<bool>("_hidePracticeButton");
  309. var _actionButtonText = navController.GetPrivateField<string>("_actionButtonText");
  310. var _allowedBeatmapDifficultyMask = navController.GetPrivateField<BeatmapDifficultyMask>("_allowedBeatmapDifficultyMask");
  311. var _notAllowedCharacteristics = navController.GetPrivateField<BeatmapCharacteristicSO[]>("_notAllowedCharacteristics");
  312. Logger.Debug("Calling lcnvc.SetData...");
  313. lcnvc.SetData(levelPack,
  314. true,
  315. _showPlayerStatsInDetailView,
  316. !_hidePracticeButton,
  317. _actionButtonText,
  318. null,
  319. _allowedBeatmapDifficultyMask,
  320. _notAllowedCharacteristics);
  321. //_sortedSongs.ForEach(x => Logger.Debug(x.levelID));
  322. }
  323. /// <summary>
  324. /// Filter songs based on playerdata favorites.
  325. /// </summary>
  326. private List<IPreviewBeatmapLevel> FilterFavorites(List<IPreviewBeatmapLevel> levels)
  327. {
  328. Logger.Info("Filtering song list as favorites playlist...");
  329. PlayerDataModel playerData = Resources.FindObjectsOfTypeAll<PlayerDataModel>().FirstOrDefault();
  330. return levels.Where(x => playerData.playerData.favoritesLevelIds.Contains(x.levelID)).ToList();
  331. }
  332. /// <summary>
  333. /// Filter for a search query.
  334. /// </summary>
  335. /// <param name="levels"></param>
  336. /// <returns></returns>
  337. private List<IPreviewBeatmapLevel> FilterSearch(List<IPreviewBeatmapLevel> levels)
  338. {
  339. // Make sure we can actually search.
  340. if (this._settings.searchTerms.Count <= 0)
  341. {
  342. Logger.Error("Tried to search for a song with no valid search terms...");
  343. SortSongName(levels);
  344. return levels;
  345. }
  346. string searchTerm = this._settings.searchTerms[0];
  347. if (String.IsNullOrEmpty(searchTerm))
  348. {
  349. Logger.Error("Empty search term entered.");
  350. SortSongName(levels);
  351. return levels;
  352. }
  353. Logger.Info("Filtering song list by search term: {0}", searchTerm);
  354. var terms = searchTerm.Split(' ');
  355. foreach (var term in terms)
  356. {
  357. levels = levels.Intersect(
  358. levels
  359. .Where(x => $"{x.songName} {x.songSubName} {x.songAuthorName} {x.levelAuthorName}".ToLower().Contains(term.ToLower()))
  360. .ToList(
  361. )
  362. ).ToList();
  363. }
  364. return levels;
  365. }
  366. /// <summary>
  367. /// Filter songs based on ranked or unranked status.
  368. /// </summary>
  369. /// <param name="levels"></param>
  370. /// <param name="includeRanked"></param>
  371. /// <param name="includeUnranked"></param>
  372. /// <returns></returns>
  373. private List<IPreviewBeatmapLevel> FilterRanked(List<IPreviewBeatmapLevel> levels, bool includeRanked, bool includeUnranked)
  374. {
  375. return levels.Where(x =>
  376. {
  377. var hash = SongBrowserModel.GetSongHash(x.levelID);
  378. double maxPP = 0.0;
  379. if (SongDataCore.Plugin.Songs.Data.Songs.ContainsKey(hash))
  380. {
  381. maxPP = SongDataCore.Plugin.Songs.Data.Songs[hash].diffs.Max(y => y.pp);
  382. }
  383. if (maxPP > 0f)
  384. {
  385. return includeRanked;
  386. }
  387. else
  388. {
  389. return includeUnranked;
  390. }
  391. }).ToList();
  392. }
  393. /// <summary>
  394. /// Sorting returns original list.
  395. /// </summary>
  396. /// <param name="levels"></param>
  397. /// <returns></returns>
  398. private List<IPreviewBeatmapLevel> SortOriginal(List<IPreviewBeatmapLevel> levels)
  399. {
  400. Logger.Info("Sorting song list as original");
  401. return levels;
  402. }
  403. /// <summary>
  404. /// Sorting by newest (file time, creation+modified).
  405. /// </summary>
  406. /// <param name="levels"></param>
  407. /// <returns></returns>
  408. private List<IPreviewBeatmapLevel> SortNewest(List<IPreviewBeatmapLevel> levels)
  409. {
  410. Logger.Info("Sorting song list as newest.");
  411. return levels
  412. .OrderByDescending(x => _cachedLastWriteTimes.ContainsKey(x.levelID) ? _cachedLastWriteTimes[x.levelID] : int.MinValue)
  413. .ToList();
  414. }
  415. /// <summary>
  416. /// Sorting by the song author.
  417. /// </summary>
  418. /// <param name="levelIds"></param>
  419. /// <returns></returns>
  420. private List<IPreviewBeatmapLevel> SortAuthor(List<IPreviewBeatmapLevel> levelIds)
  421. {
  422. Logger.Info("Sorting song list by author");
  423. return levelIds
  424. .OrderBy(x => x.songAuthorName)
  425. .ThenBy(x => x.songName)
  426. .ToList();
  427. }
  428. /// <summary>
  429. /// Sorting by song users play count.
  430. /// </summary>
  431. /// <param name="levels"></param>
  432. /// <returns></returns>
  433. private List<IPreviewBeatmapLevel> SortPlayCount(List<IPreviewBeatmapLevel> levels)
  434. {
  435. Logger.Info("Sorting song list by playcount");
  436. return levels
  437. .OrderByDescending(x => _levelIdToPlayCount.ContainsKey(x.levelID) ? _levelIdToPlayCount[x.levelID] : 0)
  438. .ThenBy(x => x.songName)
  439. .ToList();
  440. }
  441. /// <summary>
  442. /// Sorting by PP.
  443. /// </summary>
  444. /// <param name="levels"></param>
  445. /// <returns></returns>
  446. private List<IPreviewBeatmapLevel> SortPerformancePoints(List<IPreviewBeatmapLevel> levels)
  447. {
  448. Logger.Info("Sorting song list by performance points...");
  449. if (!SongDataCore.Plugin.Songs.IsDataAvailable())
  450. {
  451. SortWasMissingData = true;
  452. return levels;
  453. }
  454. return levels
  455. .OrderByDescending(x =>
  456. {
  457. var hash = SongBrowserModel.GetSongHash(x.levelID);
  458. if (SongDataCore.Plugin.Songs.Data.Songs.ContainsKey(hash))
  459. {
  460. return SongDataCore.Plugin.Songs.Data.Songs[hash].diffs.Max(y => y.pp);
  461. }
  462. else
  463. {
  464. return 0;
  465. }
  466. })
  467. .ToList();
  468. }
  469. /// <summary>
  470. /// Sorting by star rating.
  471. /// </summary>
  472. /// <param name="levels"></param>
  473. /// <returns></returns>
  474. private List<IPreviewBeatmapLevel> SortStars(List<IPreviewBeatmapLevel> levels)
  475. {
  476. Logger.Info("Sorting song list by star points...");
  477. if (!SongDataCore.Plugin.Songs.IsDataAvailable())
  478. {
  479. SortWasMissingData = true;
  480. return levels;
  481. }
  482. return levels
  483. .OrderByDescending(x =>
  484. {
  485. var hash = SongBrowserModel.GetSongHash(x.levelID);
  486. var stars = 0.0;
  487. if (SongDataCore.Plugin.Songs.Data.Songs.ContainsKey(hash))
  488. {
  489. var diffs = SongDataCore.Plugin.Songs.Data.Songs[hash].diffs;
  490. stars = diffs.Max(y => y.star);
  491. }
  492. //Logger.Debug("Stars={0}", stars);
  493. if (stars != 0)
  494. {
  495. return stars;
  496. }
  497. if (_settings.invertSortResults)
  498. {
  499. return double.MaxValue;
  500. }
  501. else
  502. {
  503. return double.MinValue;
  504. }
  505. })
  506. .ToList();
  507. }
  508. /// <summary>
  509. /// Randomize the sorting.
  510. /// </summary>
  511. /// <param name="levelIds"></param>
  512. /// <returns></returns>
  513. private List<IPreviewBeatmapLevel> SortRandom(List<IPreviewBeatmapLevel> levelIds)
  514. {
  515. Logger.Info("Sorting song list by random (seed={0})...", Settings.randomSongSeed);
  516. System.Random rnd = new System.Random(Settings.randomSongSeed);
  517. return levelIds
  518. .OrderBy(x => x.songName)
  519. .OrderBy(x => rnd.Next())
  520. .ToList();
  521. }
  522. /// <summary>
  523. /// Sorting by the song name.
  524. /// </summary>
  525. /// <param name="levels"></param>
  526. /// <returns></returns>
  527. private List<IPreviewBeatmapLevel> SortSongName(List<IPreviewBeatmapLevel> levels)
  528. {
  529. Logger.Info("Sorting song list as default (songName)");
  530. return levels
  531. .OrderBy(x => x.songName)
  532. .ThenBy(x => x.songAuthorName)
  533. .ToList();
  534. }
  535. /// <summary>
  536. /// Sorting by BeatSaver UpVotes.
  537. /// </summary>
  538. /// <param name="levelIds"></param>
  539. /// <returns></returns>
  540. private List<IPreviewBeatmapLevel> SortUpVotes(List<IPreviewBeatmapLevel> levelIds)
  541. {
  542. Logger.Info("Sorting song list by BeatSaver UpVotes");
  543. // Do not always have data when trying to sort by UpVotes
  544. if (!SongDataCore.Plugin.Songs.IsDataAvailable())
  545. {
  546. SortWasMissingData = true;
  547. return levelIds;
  548. }
  549. return levelIds
  550. .OrderByDescending(x =>
  551. {
  552. var hash = SongBrowserModel.GetSongHash(x.levelID);
  553. if (SongDataCore.Plugin.Songs.Data.Songs.ContainsKey(hash))
  554. {
  555. return SongDataCore.Plugin.Songs.Data.Songs[hash].upVotes;
  556. }
  557. else
  558. {
  559. return int.MinValue;
  560. }
  561. })
  562. .ToList();
  563. }
  564. /// <summary>
  565. /// Sorting by BeatSaver playcount stat.
  566. /// </summary>
  567. /// <param name="levelIds"></param>
  568. /// <returns></returns>
  569. private List<IPreviewBeatmapLevel> SortBeatSaverPlayCount(List<IPreviewBeatmapLevel> levelIds)
  570. {
  571. Logger.Info("Sorting song list by BeatSaver PlayCount");
  572. return levelIds;
  573. // Do not always have data when trying to sort by UpVotes
  574. /*if (!SongDataCore.Plugin.Songs.IsDataAvailable())
  575. {
  576. SortWasMissingData = true;
  577. return levelIds;
  578. }
  579. return levelIds
  580. .OrderByDescending(x => {
  581. var hash = SongBrowserModel.GetSongHash(x.levelID);
  582. if (SongDataCore.Plugin.Songs.Data.Songs.ContainsKey(hash))
  583. {
  584. return SongDataCore.Plugin.Songs.Data.Songs[hash].plays;
  585. }
  586. else
  587. {
  588. return int.MinValue;
  589. }
  590. })
  591. .ToList();*/
  592. }
  593. /// <summary>
  594. /// Sorting by BeatSaver rating stat.
  595. /// </summary>
  596. /// <param name="levelIds"></param>
  597. /// <returns></returns>
  598. private List<IPreviewBeatmapLevel> SortBeatSaverRating(List<IPreviewBeatmapLevel> levelIds)
  599. {
  600. Logger.Info("Sorting song list by BeatSaver Rating!");
  601. // Do not always have data when trying to sort by rating
  602. if (!SongDataCore.Plugin.Songs.IsDataAvailable())
  603. {
  604. SortWasMissingData = true;
  605. return levelIds;
  606. }
  607. return levelIds
  608. .OrderByDescending(x =>
  609. {
  610. var hash = SongBrowserModel.GetSongHash(x.levelID);
  611. if (SongDataCore.Plugin.Songs.Data.Songs.ContainsKey(hash))
  612. {
  613. return SongDataCore.Plugin.Songs.Data.Songs[hash].rating;
  614. }
  615. else
  616. {
  617. return int.MinValue;
  618. }
  619. })
  620. .ToList();
  621. }
  622. /// <summary>
  623. /// Sorting by BeatSaver heat stat.
  624. /// </summary>
  625. /// <param name="levelIds"></param>
  626. /// <returns></returns>
  627. private List<IPreviewBeatmapLevel> SortBeatSaverHeat(List<IPreviewBeatmapLevel> levelIds)
  628. {
  629. Logger.Info("Sorting song list by BeatSaver Heat!");
  630. // Do not always have data when trying to sort by heat
  631. if (!SongDataCore.Plugin.Songs.IsDataAvailable())
  632. {
  633. SortWasMissingData = true;
  634. return levelIds;
  635. }
  636. return levelIds
  637. .OrderByDescending(x =>
  638. {
  639. var hash = SongBrowserModel.GetSongHash(x.levelID);
  640. if (SongDataCore.Plugin.Songs.Data.Songs.ContainsKey(hash))
  641. {
  642. return SongDataCore.Plugin.Songs.Data.Songs[hash].heat;
  643. }
  644. else
  645. {
  646. return int.MinValue;
  647. }
  648. })
  649. .ToList();
  650. }
  651. #region Song helpers
  652. /// <summary>
  653. /// Get the song hash from a levelID
  654. /// </summary>
  655. /// <param name="levelId"></param>
  656. /// <returns></returns>
  657. public static string GetSongHash(string levelId)
  658. {
  659. var split = levelId.Split('_');
  660. return split.Length > 2 ? split[2] : levelId;
  661. }
  662. #endregion
  663. }
  664. }