Add missing cell height checks

CnCNet · Jun 19, 2024 · 4cab9da · 4cab9da
1 parent 669c9b9
commit 4cab9da
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/TSMapEditor/Mutations/Classes/HeightMutations/LowerCellsMutation.cs b/src/TSMapEditor/Mutations/Classes/HeightMutations/LowerCellsMutation.cs
@@ -45,6 +45,10 @@ public override void Perform()
             else
             {
                 var targetTile = MutationTarget.Map.GetTile(targetCellCoords);
+
+                if (targetTile == null || targetTile.Level <= 0)
+                    return;
+
                 var tilesToProcess = Helpers.GetFillAreaTiles(targetTile, MutationTarget.Map, MutationTarget.TheaterGraphics);
 
                 // Process tiles

diff --git a/src/TSMapEditor/Mutations/Classes/HeightMutations/RaiseCellsMutation.cs b/src/TSMapEditor/Mutations/Classes/HeightMutations/RaiseCellsMutation.cs
@@ -46,6 +46,10 @@ public override void Perform()
             else
             {
                 var targetTile = MutationTarget.Map.GetTile(targetCellCoords);
+
+                if (targetTile == null || targetTile.Level >= Constants.MaxMapHeightLevel)
+                    return;
+
                 var tilesToProcess = Helpers.GetFillAreaTiles(targetTile, MutationTarget.Map, MutationTarget.TheaterGraphics);
 
                 // Process tiles