111

Assets/Scripts/Menu/MenuInformationRecorder.cs

@@ -23,17 +23,30 @@ namespace Ichni.Menu
                 Destroy(gameObject);
             }
         }
+
+        public SongSelectionRecord GetRecordOfThisChapter()
+        {
+            ChapterSelectionUnit currentChapter = ChapterSelectionManager.instance.currentChapter;
+            if (songSelectionRecords.TryGetValue(currentChapter, out SongSelectionRecord record))
+            {
+                return record;
+            }
+            else
+            {
+                return new SongSelectionRecord(currentChapter.songs[0], currentChapter.songs[0].difficultyDataList[0]);
+            }
+        }
     }
 
-    public struct SongSelectionRecord
+    public class SongSelectionRecord
     {
         public SongItemData song;
-        public DifficultyData difficulty;
+        public int difficultyIndex;
         
         public SongSelectionRecord(SongItemData song, DifficultyData difficulty)
         {
             this.song = song;
-            this.difficulty = difficulty;
+            difficultyIndex = song.difficultyDataList.IndexOf(difficulty);
         }
     }
 }