Add expanded state tracking to beatmap carousel v2

osu.Game.Tests/Visual/SongSelect/BeatmapCarouselV2TestScene.cs

@@ -153,7 +153,8 @@ protected void WaitForGroupSelection(int group, int panel)
                 var groupingFilter = Carousel.Filters.OfType<BeatmapCarouselFilterGrouping>().Single();
 
                 GroupDefinition g = groupingFilter.GroupItems.Keys.ElementAt(group);
-                CarouselItem item = groupingFilter.GroupItems[g].ElementAt(panel);
+                // offset by one because the group itself is included in the items list.
+                CarouselItem item = groupingFilter.GroupItems[g].ElementAt(panel + 1);
 
                 return ReferenceEquals(Carousel.CurrentSelection, item.Model);
             });

osu.Game.Tests/Visual/SongSelect/TestSceneBeatmapCarouselV2GroupSelection.cs

@@ -24,7 +24,7 @@ public override void SetUpSteps()
         }
 
         [Test]
-        public void TestOpenCloseGroupWithNoSelection()
+        public void TestOpenCloseGroupWithNoSelectionMouse()
         {
             AddBeatmaps(10, 5);
             WaitForDrawablePanels();
@@ -41,6 +41,29 @@ public void TestOpenCloseGroupWithNoSelection()
             CheckNoSelection();
         }
 
+        [Test]
+        public void TestOpenCloseGroupWithNoSelectionKeyboard()
+        {
+            AddBeatmaps(10, 5);
+            WaitForDrawablePanels();
+
+            AddAssert("no beatmaps visible", () => Carousel.ChildrenOfType<BeatmapPanel>().Count(p => p.Alpha > 0), () => Is.Zero);
+            CheckNoSelection();
+
+            SelectNextPanel();
+            Select();
+            AddUntilStep("some beatmaps visible", () => Carousel.ChildrenOfType<BeatmapPanel>().Count(p => p.Alpha > 0), () => Is.GreaterThan(0));
+            AddAssert("keyboard selected is expanded", () => getKeyboardSelectedPanel()?.Expanded.Value, () => Is.True);
+            CheckNoSelection();
+
+            Select();
+            AddUntilStep("no beatmaps visible", () => Carousel.ChildrenOfType<BeatmapPanel>().Count(p => p.Alpha > 0), () => Is.Zero);
+            AddAssert("keyboard selected is collapsed", () => getKeyboardSelectedPanel()?.Expanded.Value, () => Is.False);
+            CheckNoSelection();
+
+            GroupPanel? getKeyboardSelectedPanel() => Carousel.ChildrenOfType<GroupPanel>().SingleOrDefault(p => p.KeyboardSelected.Value);
+        }
+
         [Test]
         public void TestCarouselRemembersSelection()
         {

osu.Game/Screens/SelectV2/BeatmapCarousel.cs

@@ -105,12 +105,12 @@ protected override bool HandleItemSelected(object? model)
                     // Special case – collapsing an open group.
                     if (lastSelectedGroup == group)
                     {
-                        setVisibilityOfGroupItems(lastSelectedGroup, false);
+                        setExpansionStateOfGroup(lastSelectedGroup, false);
                         lastSelectedGroup = null;
                         return false;
                     }
 
-                    setVisibleGroup(group);
+                    setExpandedGroup(group);
                     return false;
 
                 case BeatmapSetInfo setInfo:
@@ -127,11 +127,11 @@ protected override bool HandleItemSelected(object? model)
                         GroupDefinition? group = grouping.GroupItems.SingleOrDefault(kvp => kvp.Value.Any(i => ReferenceEquals(i.Model, beatmapInfo))).Key;
 
                         if (group != null)
-                            setVisibleGroup(group);
+                            setExpandedGroup(group);
                     }
                     else
                     {
-                        setVisibleSet(beatmapInfo);
+                        setExpandedSet(beatmapInfo);
                     }
 
                     return true;
@@ -158,37 +158,47 @@ protected override bool CheckValidForGroupSelection(CarouselItem item)
             }
         }
 
-        private void setVisibleGroup(GroupDefinition group)
+        private void setExpandedGroup(GroupDefinition group)
         {
             if (lastSelectedGroup != null)
-                setVisibilityOfGroupItems(lastSelectedGroup, false);
+                setExpansionStateOfGroup(lastSelectedGroup, false);
             lastSelectedGroup = group;
-            setVisibilityOfGroupItems(group, true);
+            setExpansionStateOfGroup(group, true);
         }
 
-        private void setVisibilityOfGroupItems(GroupDefinition group, bool visible)
+        private void setExpansionStateOfGroup(GroupDefinition group, bool expanded)
         {
             if (grouping.GroupItems.TryGetValue(group, out var items))
             {
                 foreach (var i in items)
-                    i.IsVisible = visible;
+                {
+                    if (i.Model is GroupDefinition)
+                        i.IsExpanded = expanded;
+                    else
+                        i.IsVisible = expanded;
+                }
             }
         }
 
-        private void setVisibleSet(BeatmapInfo beatmapInfo)
+        private void setExpandedSet(BeatmapInfo beatmapInfo)
         {
             if (lastSelectedBeatmap != null)
-                setVisibilityOfSetItems(lastSelectedBeatmap.BeatmapSet!, false);
+                setExpansionStateOfSetItems(lastSelectedBeatmap.BeatmapSet!, false);
             lastSelectedBeatmap = beatmapInfo;
-            setVisibilityOfSetItems(beatmapInfo.BeatmapSet!, true);
+            setExpansionStateOfSetItems(beatmapInfo.BeatmapSet!, true);
         }
 
-        private void setVisibilityOfSetItems(BeatmapSetInfo set, bool visible)
+        private void setExpansionStateOfSetItems(BeatmapSetInfo set, bool expanded)
         {
             if (grouping.SetItems.TryGetValue(set, out var items))
             {
                 foreach (var i in items)
-                    i.IsVisible = visible;
+                {
+                    if (i.Model is BeatmapSetInfo)
+                        i.IsExpanded = expanded;
+                    else
+                        i.IsVisible = expanded;
+                }
             }
         }
 

osu.Game/Screens/SelectV2/BeatmapCarouselFilterGrouping.cs

@@ -65,7 +65,11 @@ public async Task<IEnumerable<CarouselItem>> Run(IEnumerable<CarouselItem> items
                         if (b.StarRating > starGroup)
                         {
                             starGroup = (int)Math.Floor(b.StarRating);
-                            newItems.Add(new CarouselItem(new GroupDefinition($"{starGroup} - {++starGroup} *")) { DrawHeight = GroupPanel.HEIGHT });
+                            var groupDefinition = new GroupDefinition($"{starGroup} - {++starGroup} *");
+                            var groupItem = new CarouselItem(groupDefinition) { DrawHeight = GroupPanel.HEIGHT };
+
+                            newItems.Add(groupItem);
+                            groupItems[groupDefinition] = new HashSet<CarouselItem> { groupItem };
                         }
 
                         newItems.Add(item);
@@ -91,14 +95,13 @@ public async Task<IEnumerable<CarouselItem>> Run(IEnumerable<CarouselItem> items
 
                         if (newBeatmapSet)
                         {
-                            newItems.Insert(i, new CarouselItem(beatmap.BeatmapSet!) { DrawHeight = BeatmapSetPanel.HEIGHT });
+                            var setItem = new CarouselItem(beatmap.BeatmapSet!) { DrawHeight = BeatmapSetPanel.HEIGHT };
+                            setItems[beatmap.BeatmapSet!] = new HashSet<CarouselItem> { setItem };
+                            newItems.Insert(i, setItem);
                             i++;
                         }
 
-                        if (!setItems.TryGetValue(beatmap.BeatmapSet!, out var related))
-                            setItems[beatmap.BeatmapSet!] = related = new HashSet<CarouselItem>();
-
-                        related.Add(item);
+                        setItems[beatmap.BeatmapSet!].Add(item);
                         item.IsVisible = false;
                     }
 
@@ -121,10 +124,7 @@ public async Task<IEnumerable<CarouselItem>> Run(IEnumerable<CarouselItem> items
 
                 if (lastGroup != null)
                 {
-                    if (!groupItems.TryGetValue(lastGroup, out var groupRelated))
-                        groupItems[lastGroup] = groupRelated = new HashSet<CarouselItem>();
-                    groupRelated.Add(item);
-
+                    groupItems[lastGroup].Add(item);
                     item.IsVisible = false;
                 }
             }

osu.Game/Screens/SelectV2/BeatmapPanel.cs

@@ -100,6 +100,7 @@ protected override bool OnClick(ClickEvent e)
 
         public CarouselItem? Item { get; set; }
         public BindableBool Selected { get; } = new BindableBool();
+        public BindableBool Expanded { get; } = new BindableBool();
         public BindableBool KeyboardSelected { get; } = new BindableBool();
 
         public double DrawYPosition { get; set; }

osu.Game/Screens/SelectV2/BeatmapSetPanel.cs

@@ -25,6 +25,7 @@ public partial class BeatmapSetPanel : PoolableDrawable, ICarouselPanel
         private BeatmapCarousel carousel { get; set; } = null!;
 
         private OsuSpriteText text = null!;
+        private Box box = null!;
 
         [BackgroundDependencyLoader]
         private void load()
@@ -34,7 +35,7 @@ private void load()
 
             InternalChildren = new Drawable[]
             {
-                new Box
+                box = new Box
                 {
                     Colour = Color4.Yellow.Darken(5),
                     Alpha = 0.8f,
@@ -48,6 +49,11 @@ private void load()
                 }
             };
 
+            Expanded.BindValueChanged(value =>
+            {
+                box.FadeColour(value.NewValue ? Color4.Yellow.Darken(2) : Color4.Yellow.Darken(5), 500, Easing.OutQuint);
+            });
+
             KeyboardSelected.BindValueChanged(value =>
             {
                 if (value.NewValue)
@@ -85,6 +91,7 @@ protected override bool OnClick(ClickEvent e)
 
         public CarouselItem? Item { get; set; }
         public BindableBool Selected { get; } = new BindableBool();
+        public BindableBool Expanded { get; } = new BindableBool();
         public BindableBool KeyboardSelected { get; } = new BindableBool();
 
         public double DrawYPosition { get; set; }

osu.Game/Screens/SelectV2/Carousel.cs

@@ -571,6 +571,7 @@ protected override void Update()
 
                 c.Selected.Value = c.Item == currentSelection?.CarouselItem;
                 c.KeyboardSelected.Value = c.Item == currentKeyboardSelection?.CarouselItem;
+                c.Expanded.Value = c.Item.IsExpanded;
             }
         }
 
@@ -674,6 +675,7 @@ private static void expirePanelImmediately(Drawable panel)
             carouselPanel.Item = null;
             carouselPanel.Selected.Value = false;
             carouselPanel.KeyboardSelected.Value = false;
+            carouselPanel.Expanded.Value = false;
         }
 
         #endregion

osu.Game/Screens/SelectV2/CarouselItem.cs

@@ -30,10 +30,15 @@ public sealed class CarouselItem : IComparable<CarouselItem>
         public float DrawHeight { get; set; } = DEFAULT_HEIGHT;
 
         /// <summary>
-        /// Whether this item is visible or collapsed (hidden).
+        /// Whether this item is visible or hidden.
         /// </summary>
         public bool IsVisible { get; set; } = true;
 
+        /// <summary>
+        /// Whether this item is expanded or not. Should only be used for headers of groups.
+        /// </summary>
+        public bool IsExpanded { get; set; }
+
         public CarouselItem(object model)
         {
             Model = model;

osu.Game/Screens/SelectV2/GroupPanel.cs

@@ -26,6 +26,8 @@ public partial class GroupPanel : PoolableDrawable, ICarouselPanel
         private Box activationFlash = null!;
         private OsuSpriteText text = null!;
 
+        private Box box = null!;
+
         [BackgroundDependencyLoader]
         private void load()
         {
@@ -34,7 +36,7 @@ private void load()
 
             InternalChildren = new Drawable[]
             {
-                new Box
+                box = new Box
                 {
                     Colour = Color4.DarkBlue.Darken(5),
                     Alpha = 0.8f,
@@ -60,6 +62,11 @@ private void load()
                 activationFlash.FadeTo(value.NewValue ? 0.2f : 0, 500, Easing.OutQuint);
             });
 
+            Expanded.BindValueChanged(value =>
+            {
+                box.FadeColour(value.NewValue ? Color4.SkyBlue : Color4.DarkBlue.Darken(5), 500, Easing.OutQuint);
+            });
+
             KeyboardSelected.BindValueChanged(value =>
             {
                 if (value.NewValue)
@@ -97,6 +104,7 @@ protected override bool OnClick(ClickEvent e)
 
         public CarouselItem? Item { get; set; }
         public BindableBool Selected { get; } = new BindableBool();
+        public BindableBool Expanded { get; } = new BindableBool();
         public BindableBool KeyboardSelected { get; } = new BindableBool();
 
         public double DrawYPosition { get; set; }

osu.Game/Screens/SelectV2/ICarouselPanel.cs

@@ -14,10 +14,15 @@ namespace osu.Game.Screens.SelectV2
     public interface ICarouselPanel
     {
         /// <summary>
-        /// Whether this item has selection. Should be read from to update the visual state.
+        /// Whether this item has selection (see <see cref="Carousel{T}.CurrentSelection"/>). Should be read from to update the visual state.
         /// </summary>
         BindableBool Selected { get; }
 
+        /// <summary>
+        /// Whether this item is expanded (see <see cref="CarouselItem.IsExpanded"/>). Should be read from to update the visual state.
+        /// </summary>
+        BindableBool Expanded { get; }
+
         /// <summary>
         /// Whether this item has keyboard selection. Should be read from to update the visual state.
         /// </summary>