# HG changeset patch # User matthijs # Date 2006-05-01 09:34:23 # Node ID fbc996c74eb0a96eed995a18ba0a2a9454b78365 # Parent 3b732a391a2763b95188b24bed017f8777d28968 (svn r4640) - Add: IsValidDiagDirection, IsValidDirection and IsValidAxis functions. - Codechange: Use IsValidDiagDirection() in CmdBuildRoadStop(). diff --git a/direction.h b/direction.h --- a/direction.h +++ b/direction.h @@ -93,7 +93,8 @@ static inline Direction DiagDirToDir(Dia /* the 2 axis */ typedef enum Axis { AXIS_X = 0, - AXIS_Y = 1 + AXIS_Y = 1, + AXIS_END } Axis; @@ -112,4 +113,19 @@ static inline DiagDirection AxisToDiagDi return (DiagDirection)(2 - a); } +static inline bool IsValidDiagDirection(DiagDirection d) +{ + return d < DIAGDIR_END; +} + +static inline bool IsValidDirection(DiagDirection d) +{ + return d < DIR_END; +} + +static inline bool IsValidAxis(DiagDirection d) +{ + return d < AXIS_END; +} + #endif diff --git a/station_cmd.c b/station_cmd.c --- a/station_cmd.c +++ b/station_cmd.c @@ -1360,7 +1360,7 @@ int32 CmdBuildRoadStop(TileIndex tile, u bool type = !!p2; /* Saveguard the parameters */ - if (p1 > 3) return CMD_ERROR; + if (!IsValidDiagDirection(p1)) return CMD_ERROR; SET_EXPENSES_TYPE(EXPENSES_CONSTRUCTION);