Fix TR3 no-secret levels

Regression from #681.
LostArtefacts · May 30, 2024 · dc07dc1 · dc07dc1
1 parent bb10647
commit dc07dc1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/TRRandomizerCore/Randomizers/Shared/SecretArtefactPlacer.cs b/TRRandomizerCore/Randomizers/Shared/SecretArtefactPlacer.cs
@@ -80,7 +80,13 @@ public void InitialiseLevel(TRGameVersion version,
         _sectorGetter = sectorGetter;
         _sectorEntityTester = sectorEntityTester;
 
-        foreach (TRRoomSector sector in _rooms.SelectMany(r => r.Sectors))
+        RemoveDefaultSecrets(_rooms, _floorData);
+    }
+
+    public void RemoveDefaultSecrets<R>(List<R> rooms, FDControl floorData)
+        where R : TRRoom
+    {
+        foreach (TRRoomSector sector in rooms.SelectMany(r => r.Sectors))
         {
             if (sector.FDIndex == 0)
             {

diff --git a/TRRandomizerCore/Randomizers/TR3/Classic/TR3SecretRandomizer.cs b/TRRandomizerCore/Randomizers/TR3/Classic/TR3SecretRandomizer.cs
@@ -501,6 +501,10 @@ internal void ApplyRandomization()
                     // Convert the placeholder reward room into an actual room now that secrets are positioned.
                     _outer.ActualiseRewardRoom(level, rewardRoom);
                 }
+                else if (!level.HasSecrets)
+                {
+                    _outer._placer.RemoveDefaultSecrets(level.Data.Rooms, level.Data.FloorData);
+                }
 
                 _outer.SaveLevel(level);