From 347cb0a1b5bdaf64f90a4dfc699a1dd2b30d5856 Mon Sep 17 00:00:00 2001
From: Hanamuke <antoine.houdayer@laposte.net>
Date: Fri, 14 Sep 2018 19:41:29 +0200
Subject: [PATCH] Check for type conversion, mais the update recursive

---
 osu.Game/Beatmaps/BeatmapProcessor.cs | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/osu.Game/Beatmaps/BeatmapProcessor.cs b/osu.Game/Beatmaps/BeatmapProcessor.cs
index 84bf6ada18..1a767942eb 100644
--- a/osu.Game/Beatmaps/BeatmapProcessor.cs
+++ b/osu.Game/Beatmaps/BeatmapProcessor.cs
@@ -44,13 +44,24 @@ namespace osu.Game.Beatmaps
 
         public virtual void PostProcess()
         {
+            void UpdateNestedCombo(Rulesets.Objects.HitObject obj, int comboIndex, int indexInCurrentCombo)
+            {
+                if (obj is IHasComboInformation)
+                {
+                    var objectComboInfo = (IHasComboInformation)obj;
+                    objectComboInfo.ComboIndex = comboIndex;
+                    objectComboInfo.IndexInCurrentCombo = indexInCurrentCombo;
+                    foreach (var nestedObjet in obj.NestedHitObjects)
+                        UpdateNestedCombo(nestedObjet, comboIndex, indexInCurrentCombo);
+                }
+            }
             foreach (var hitObject in Beatmap.HitObjects)
             {
-                var objectComboInfo = (IHasComboInformation)hitObject;
-                foreach (var obj in hitObject.NestedHitObjects.OfType<IHasComboInformation>())
+                if (hitObject is IHasComboInformation)
                 {
-                    obj.IndexInCurrentCombo = objectComboInfo.IndexInCurrentCombo;
-                    obj.ComboIndex = objectComboInfo.ComboIndex;
+                    var objectComboInfo = (IHasComboInformation)hitObject;
+                    foreach (var nested in hitObject.NestedHitObjects)
+                        UpdateNestedCombo(nested, objectComboInfo.ComboIndex, objectComboInfo.IndexInCurrentCombo);
                 }
             }
         }