From f182f6e6753f96e1b3ac4bcab2caa48bfb29bced Mon Sep 17 00:00:00 2001 From: michi_cc Date: Thu, 7 Jul 2011 14:16:16 +0000 Subject: [PATCH] (svn r22638) -Fix: Keep the previous owner of the upper and lower lock parts if they are built on existing water. --- src/water_cmd.cpp | 4 ++-- src/water_map.h | 6 ++++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/src/water_cmd.cpp b/src/water_cmd.cpp index 3ce93333ba..b37636ccb9 100644 --- a/src/water_cmd.cpp +++ b/src/water_cmd.cpp @@ -279,8 +279,8 @@ static CommandCost RemoveLock(TileIndex tile, DoCommandFlag flags) if (flags & DC_EXEC) { DoClearSquare(tile); - MakeWaterKeepingClass(tile + delta, GetTileOwner(tile)); - MakeWaterKeepingClass(tile - delta, GetTileOwner(tile)); + MakeWaterKeepingClass(tile + delta, GetTileOwner(tile + delta)); + MakeWaterKeepingClass(tile - delta, GetTileOwner(tile - delta)); MarkCanalsAndRiversAroundDirty(tile - delta); MarkCanalsAndRiversAroundDirty(tile + delta); } diff --git a/src/water_map.h b/src/water_map.h index d26b93210a..02838a6841 100644 --- a/src/water_map.h +++ b/src/water_map.h @@ -421,8 +421,10 @@ static inline void MakeLock(TileIndex t, Owner o, DiagDirection d, WaterClass wc TileIndexDiff delta = TileOffsByDiagDir(d); MakeLockTile(t, o, LOCK_MIDDLE + d, WATER_CLASS_CANAL); - MakeLockTile(t - delta, o, LOCK_LOWER + d, wc_lower); - MakeLockTile(t + delta, o, LOCK_UPPER + d, wc_upper); + /* Keep the current owner for the upper and lower part if it is a + * water tile so we can restore the owner after deleting the lock. */ + MakeLockTile(t - delta, IsWaterTile(t - delta) ? GetTileOwner(t - delta) : o, LOCK_LOWER + d, wc_lower); + MakeLockTile(t + delta, IsWaterTile(t + delta) ? GetTileOwner(t + delta) : o, LOCK_UPPER + d, wc_upper); } #endif /* WATER_MAP_H */