(svn r11636) -Fix: it was possible to build/remove road/tram in tunnel/bridge when there was vehicle in it
authorsmatz
Sat, 15 Dec 2007 13:59:38 +0000
changeset 8571 052b7794484d
parent 8570 3312a80e0ebd
child 8572 e17b345c9ab6
(svn r11636) -Fix: it was possible to build/remove road/tram in tunnel/bridge when there was vehicle in it
src/road_cmd.cpp
--- a/src/road_cmd.cpp	Sat Dec 15 03:42:41 2007 +0000
+++ b/src/road_cmd.cpp	Sat Dec 15 13:59:38 2007 +0000
@@ -121,16 +121,27 @@
 	switch (GetTileType(tile)) {
 		case MP_ROAD:
 			if (_game_mode != GM_EDITOR && GetRoadOwner(tile, rt) == OWNER_TOWN) t = GetTownByTile(tile);
+			if (!EnsureNoVehicleOnGround(tile)) return CMD_ERROR;
 			break;
 
 		case MP_STATION:
 			if (!IsDriveThroughStopTile(tile)) return CMD_ERROR;
+			if (!EnsureNoVehicleOnGround(tile)) return CMD_ERROR;
 			break;
 
 		case MP_TUNNELBRIDGE:
-			if ((IsTunnel(tile) && GetTunnelTransportType(tile) != TRANSPORT_ROAD) ||
-					(IsBridge(tile) && GetBridgeTransportType(tile) != TRANSPORT_ROAD)) return CMD_ERROR;
-			break;
+			{
+				TileIndex endtile;
+				if (IsTunnel(tile)) {
+					if (GetTunnelTransportType(tile) != TRANSPORT_ROAD) return CMD_ERROR;
+					endtile = GetOtherTunnelEnd(tile);
+				} else {
+					if (GetBridgeTransportType(tile) != TRANSPORT_ROAD) return CMD_ERROR;
+					endtile = GetOtherBridgeEnd(tile);
+				}
+
+				if (GetVehicleTunnelBridge(tile, endtile) != NULL) return CMD_ERROR;
+			} break;
 
 		default:
 			return CMD_ERROR;
@@ -143,8 +154,6 @@
 
 	if (!CheckAllowRemoveRoad(tile, pieces, &edge_road, rt)) return CMD_ERROR;
 
-	if (!EnsureNoVehicleOnGround(tile)) return CMD_ERROR;
-
 	/* check if you're allowed to remove the street owned by a town
 	 * removal allowance depends on difficulty setting */
 	if (!CheckforTownRating(flags, t, ROAD_REMOVE)) return CMD_ERROR;
@@ -502,12 +511,19 @@
 			break;
 
 		case MP_TUNNELBRIDGE:
-			if ((IsTunnel(tile) && GetTunnelTransportType(tile) != TRANSPORT_ROAD) ||
-					(IsBridge(tile) && GetBridgeTransportType(tile) != TRANSPORT_ROAD)) return CMD_ERROR;
-			if (HasBit(GetRoadTypes(tile), rt)) return_cmd_error(STR_1007_ALREADY_BUILT);
-			/* Don't allow "upgrading" the bridge/tunnel when vehicles are already driving on it */
-			if (!EnsureNoVehicleOnGround(tile)) return CMD_ERROR;
-			break;
+			{
+				TileIndex endtile;
+				if (IsTunnel(tile)) {
+					if (GetTunnelTransportType(tile) != TRANSPORT_ROAD) return CMD_ERROR;
+					endtile = GetOtherTunnelEnd(tile);
+				} else {
+					if (GetBridgeTransportType(tile) != TRANSPORT_ROAD) return CMD_ERROR;
+					endtile = GetOtherBridgeEnd(tile);
+				}
+				if (HasBit(GetRoadTypes(tile), rt)) return_cmd_error(STR_1007_ALREADY_BUILT);
+				/* Don't allow "upgrading" the bridge/tunnel when vehicles are already driving on it */
+				if (GetVehicleTunnelBridge(tile, endtile) != NULL) return CMD_ERROR;
+			} break;
 
 		default:
 do_clear:;