Fix Redo not working for FlattenGroundMutation

CnCNet · Dec 18, 2024 · f1eb35a · f1eb35a
1 parent d619907
commit f1eb35a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/TSMapEditor/Mutations/Classes/HeightMutations/AlterElevationMutationBase.cs b/src/TSMapEditor/Mutations/Classes/HeightMutations/AlterElevationMutationBase.cs
@@ -33,6 +33,14 @@ protected bool IsCellMorphable(MapTile cell)
             return Map.TheaterInstance.Theater.TileSets[Map.TheaterInstance.GetTileSetId(cell.TileIndex)].Morphable;
         }
 
+        protected void Clear()
+        {
+            cellsToProcess.Clear();
+            processedCellsThisIteration.Clear();
+            totalProcessedCells.Clear();
+            undoData.Clear();
+        }
+
         /// <summary>
         /// Adds a cell's data to the undo data structure.
         /// Does nothing if the cell has already been added to the undo data structure.

diff --git a/src/TSMapEditor/Mutations/Classes/HeightMutations/FlattenGroundMutation.cs b/src/TSMapEditor/Mutations/Classes/HeightMutations/FlattenGroundMutation.cs
@@ -29,6 +29,8 @@ public FlattenGroundMutation(IMutationTarget mutationTarget, Point2D originCell,
 
         private void FlattenGround()
         {
+            Clear();
+
             int xSize = BrushSize.Width;
             int ySize = BrushSize.Height;