Update APPlayState.hx

Author: Yuta12342

source/archipelago/APPlayState.hx

@@ -1496,7 +1496,24 @@ class APPlayState extends PlayState {
     {
         super.generateSong();
         if (PlayState.SONG == null) return;
-        archipelago.APNote.replaceNotes(unspawnNotes, apGame.excludeCheckedLocations(apGame.noteData(PlayState.SONG.song, currentMod)));
+        var apNotes = archipelago.APNote.replaceNotes(unspawnNotes, apGame.excludeCheckedLocations(apGame.noteData(PlayState.SONG.song, currentMod)));
+        for (playfield in playfields) {
+            for (column in playfield.noteQueue)
+                for (note in column)
+            {
+                if (!unspawnNotes.contains(note))
+                {playfield.unqueue(note);}
+            
+            for (note in apNotes) {
+                if (note.field == playfield) {
+                    playfield.queue(note);
+                    note.field = playfield;
+                    note.fieldIndex = playfield.playerId;
+                }
+            }
+        }
+    }
+
     }
 
 	// override public function generateNotes(song:SwagSong, AI:Array<Array<Float>>):Void