Update for Beat Saber 1.37.0

PlaylistManager/AffinityPatches/AnnotatedBeatmapLevelCollectionsUIPatches.cs

@@ -0,0 +1,175 @@
+using System;
+using System.Collections.Generic;
+using System.Reflection.Emit;
+using HarmonyLib;
+using SiraUtil.Affinity;
+using SongCore.Utilities;
+using UnityEngine;
+
+namespace PlaylistManager.AffinityPatches
+{
+    internal class AnnotatedBeatmapLevelCollectionsUIPatches : IAffinity
+    {
+        private readonly AnnotatedBeatmapLevelCollectionsViewController _annotatedBeatmapLevelCollectionsViewController;
+        private readonly SelectLevelCategoryViewController _selectLevelCategoryViewController;
+
+        private int _originalColumnCount;
+        private Vector2 _originalScreenSize;
+        private bool _isGridResized;
+
+        private AnnotatedBeatmapLevelCollectionsUIPatches(AnnotatedBeatmapLevelCollectionsViewController annotatedBeatmapLevelCollectionsViewController, SelectLevelCategoryViewController selectLevelCategoryViewController)
+        {
+            _annotatedBeatmapLevelCollectionsViewController = annotatedBeatmapLevelCollectionsViewController;
+            _selectLevelCategoryViewController = selectLevelCategoryViewController;
+        }
+
+        [AffinityPatch(typeof(AnnotatedBeatmapLevelCollectionsGridView), nameof(AnnotatedBeatmapLevelCollectionsGridView.SetData))]
+        [AffinityPrefix]
+        private void ResizeGrid(AnnotatedBeatmapLevelCollectionsGridView __instance, IReadOnlyList<BeatmapLevelPack> annotatedBeatmapLevelCollections)
+        {
+            if (_originalColumnCount == default)
+            {
+                _originalColumnCount = __instance._gridView._columnCount;
+            }
+
+            var selectedLevelCategory = _selectLevelCategoryViewController.selectedLevelCategory;
+            if (selectedLevelCategory == SelectLevelCategoryViewController.LevelCategory.CustomSongs)
+            {
+                // Number of columns for max visible row count before it starts clipping with the ground.
+                __instance._gridView._columnCount = Math.Max(Mathf.CeilToInt((annotatedBeatmapLevelCollections?.Count ?? 0) / 5f), _originalColumnCount);
+
+                // Remove one column to make room for our buttons.
+                if (!_isGridResized)
+                {
+                    __instance._gridView._visibleColumnCount -= 1;
+
+                    var rectTransform = (RectTransform)__instance._gridView.transform;
+                    rectTransform.sizeDelta -= new Vector2(__instance._cellWidth, 0);
+                    rectTransform.anchoredPosition -= new Vector2(__instance._cellWidth / 2, 0);
+
+                    _isGridResized = true;
+                }
+            }
+            else if (selectedLevelCategory == SelectLevelCategoryViewController.LevelCategory.MusicPacks)
+            {
+                __instance._gridView._columnCount = _originalColumnCount;
+
+                // Restore the removed column since we don't want to show an empty cell.
+                if (_isGridResized)
+                {
+                    __instance._gridView._visibleColumnCount += 1;
+
+                    var rectTransform = (RectTransform)__instance._gridView.transform;
+                    rectTransform.sizeDelta += new Vector2(__instance._cellWidth, 0);
+                    rectTransform.anchoredPosition += new Vector2(__instance._cellWidth / 2, 0);
+
+                    _isGridResized = false;
+                }
+            }
+        }
+
+        [AffinityPatch(typeof(AnnotatedBeatmapLevelCollectionsGridView), nameof(AnnotatedBeatmapLevelCollectionsGridView.OnPointerEnter))]
+        [AffinityTranspiler]
+        private IEnumerable<CodeInstruction> ChangeInteractableConditionOnEnter(IEnumerable<CodeInstruction> instructions, ILGenerator il)
+        {
+            return ChangeInteractableCondition(instructions, il);
+        }
+
+        [AffinityPatch(typeof(AnnotatedBeatmapLevelCollectionsGridView), nameof(AnnotatedBeatmapLevelCollectionsGridView.OnPointerExit))]
+        [AffinityTranspiler]
+        private IEnumerable<CodeInstruction> ChangeInteractableConditionOnExit(IEnumerable<CodeInstruction> instructions, ILGenerator il)
+        {
+            return ChangeInteractableCondition(instructions, il);
+        }
+
+        [AffinityPatch(typeof(AnnotatedBeatmapLevelCollectionsGridView), nameof(AnnotatedBeatmapLevelCollectionsGridView.HandleCellSelectionDidChange))]
+        [AffinityTranspiler]
+        private IEnumerable<CodeInstruction> ChangeInteractableConditionOnSelect(IEnumerable<CodeInstruction> instructions, ILGenerator il)
+        {
+            return ChangeInteractableCondition(instructions, il);
+        }
+
+        private static IEnumerable<CodeInstruction> ChangeInteractableCondition(IEnumerable<CodeInstruction> instructions, ILGenerator il)
+        {
+            // Makes the grid interactable when there's 1 row or more, and when there's more collections to display than the number of visible columns.
+            var codeMatcher = new CodeMatcher(instructions, il)
+                .MatchStartForward(new CodeMatch(OpCodes.Ldc_I4_1))
+                .ThrowIfInvalid()
+                .SetOpcodeAndAdvance(OpCodes.Ldc_I4_0);
+            return codeMatcher
+                .InsertBranchAndAdvance(OpCodes.Ble_S, codeMatcher.Pos + 1)
+                .Insert(
+                    new CodeInstruction(OpCodes.Ldarg_0),
+                    new CodeInstruction(OpCodes.Ldfld, AccessTools.Field(typeof(AnnotatedBeatmapLevelCollectionsGridView), nameof(AnnotatedBeatmapLevelCollectionsGridView._annotatedBeatmapLevelCollections))),
+                    new CodeInstruction(OpCodes.Callvirt, AccessTools.PropertyGetter(typeof(IReadOnlyCollection<>).MakeGenericType(typeof(BeatmapLevelPack)), nameof(IReadOnlyList<BeatmapLevelPack>.Count))),
+                    new CodeInstruction(OpCodes.Ldarg_0),
+                    new CodeInstruction(OpCodes.Ldfld, AccessTools.Field(typeof(AnnotatedBeatmapLevelCollectionsGridView), nameof(AnnotatedBeatmapLevelCollectionsGridView._gridView))),
+                    new CodeInstruction(OpCodes.Callvirt, AccessTools.PropertyGetter(typeof(GridView), nameof(GridView.visibleColumnCount))))
+                .InstructionEnumeration();
+        }
+
+        [AffinityPatch(typeof(AnnotatedBeatmapLevelCollectionsGridViewAnimator), nameof(AnnotatedBeatmapLevelCollectionsGridViewAnimator.GetContentXOffset))]
+        private void RecalculateContentXOffsetBasedOnColumnCount(AnnotatedBeatmapLevelCollectionsGridViewAnimator __instance, ref float __result)
+        {
+            if (_annotatedBeatmapLevelCollectionsViewController._annotatedBeatmapLevelCollectionsGridView._annotatedBeatmapLevelCollections == null)
+            {
+                return;
+            }
+
+            if (_annotatedBeatmapLevelCollectionsViewController._annotatedBeatmapLevelCollectionsGridView._annotatedBeatmapLevelCollections.Count <= __instance._visibleColumnCount)
+            {
+                __result = __instance._columnWidth;
+
+                return;
+            }
+
+            var zeroOffset = (__instance._columnCount - 1) / 2f;
+            var maxMove = (__instance._columnCount - __instance._visibleColumnCount) / 2f;
+            var toMove = zeroOffset - __instance._selectedColumn;
+            if (__instance._visibleColumnCount % 2 == 0)
+            {
+                toMove -= 0.5f;
+            }
+
+            __result = Math.Clamp(toMove, -maxMove, maxMove) * __instance._columnWidth;
+        }
+
+        [AffinityPatch(typeof(AnnotatedBeatmapLevelCollectionsGridViewAnimator), nameof(AnnotatedBeatmapLevelCollectionsGridViewAnimator.AnimateOpen))]
+        private void RecalculateSizeBasedOnColumnCount(AnnotatedBeatmapLevelCollectionsGridViewAnimator __instance, bool animated)
+        {
+            var x = ((__instance._columnCount - __instance._visibleColumnCount) * 2 + __instance._visibleColumnCount) * __instance._columnWidth;
+            if (animated)
+            {
+                __instance._viewportSizeTween.toValue = new Vector2(x, __instance._viewportSizeTween.toValue.y);
+            }
+            else
+            {
+                __instance._viewportTransform.sizeDelta = new Vector2(x, __instance._viewportTransform.sizeDelta.y);
+            }
+
+            if (_isGridResized)
+            {
+                // It would otherwise fly away when setting the Screen size.
+                var rectTransform = (RectTransform)_selectLevelCategoryViewController.transform;
+
+                if (rectTransform.anchorMin.x == 0 || rectTransform.anchorMax.x == 0)
+                {
+                    var localPosition = rectTransform.localPosition;
+                    rectTransform.anchorMin = new Vector2(0.5f, rectTransform.anchorMin.y);
+                    rectTransform.anchorMax = new Vector2(0.5f, rectTransform.anchorMax.y);
+                    rectTransform.localPosition = localPosition;
+                }
+
+                rectTransform = (RectTransform)__instance.gameObject.GetComponentInParent<HMUI.Screen>().transform;
+
+                if (_originalScreenSize == default)
+                {
+                    _originalScreenSize = rectTransform.sizeDelta;
+                }
+
+                // Resizing Screen is needed to allow the hover hint to be shown when the GridView is larger.
+                rectTransform.sizeDelta = new Vector2(_originalScreenSize.x + (__instance._columnCount - __instance._visibleColumnCount - 1) * __instance._columnWidth * 2, _originalScreenSize.y);
+            }
+        }
+    }
+}