diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/direction_type.h | 4 | ||||
| -rw-r--r-- | src/water_cmd.cpp | 7 |
2 files changed, 7 insertions, 4 deletions
diff --git a/src/direction_type.h b/src/direction_type.h index d1f8d8920..89b527cb6 100644 --- a/src/direction_type.h +++ b/src/direction_type.h @@ -30,6 +30,9 @@ enum Direction { INVALID_DIR = 0xFF, ///< Flag for an invalid direction }; +/** Allow incrementing of Direction variables */ +DECLARE_POSTFIX_INCREMENT(Direction); + /** Define basic enum properties */ template <> struct EnumPropsT<Direction> : MakeEnumPropsT<Direction, byte, DIR_BEGIN, DIR_END, INVALID_DIR> {}; typedef TinyEnumT<Direction> DirectionByte; //typedefing-enumification of Direction @@ -77,6 +80,7 @@ enum DiagDirection { INVALID_DIAGDIR = 0xFF, ///< Flag for an invalid DiagDirection }; +/** Allow incrementing of DiagDirection variables */ DECLARE_POSTFIX_INCREMENT(DiagDirection); /** Define basic enum properties */ diff --git a/src/water_cmd.cpp b/src/water_cmd.cpp index 119432abe..b9f2f348a 100644 --- a/src/water_cmd.cpp +++ b/src/water_cmd.cpp @@ -644,10 +644,9 @@ static void TileLoopWaterHelper(TileIndex tile, const TileIndexDiffC *offs) MakeWater(target); MarkTileDirtyByTile(target); /* Mark surrounding canal tiles dirty too to avoid glitches */ - MarkTileDirtyIfCanal(target + TileDiffXY(0, 1)); - MarkTileDirtyIfCanal(target + TileDiffXY(1, 0)); - MarkTileDirtyIfCanal(target + TileDiffXY(0, -1)); - MarkTileDirtyIfCanal(target + TileDiffXY(-1, 0)); + for (Direction dir = DIR_BEGIN; dir < DIR_END; dir++) { + MarkTileDirtyIfCanal(target + TileOffsByDir(dir)); + } } } } |