# HG changeset patch
# User michi_cc <michi_cc@openttd.org>
# Date 2009-12-27 14:37:46
# Node ID 6395a7eff659ceb0d9a5416fd7ee3449ccb1fb5c
# Parent  6b9456d5d47552fe5fc1cc21f54f57b77b8c2137

(svn r18646) -Fix [FS#3418]: [YAPP] Don't extend the reserved path through a newly built path signal directly in front of a stopped or loading train. Also restore the reserved path in more cases after removing a signal.

diff --git a/src/rail_cmd.cpp b/src/rail_cmd.cpp
--- a/src/rail_cmd.cpp
+++ b/src/rail_cmd.cpp
@@ -973,7 +973,13 @@ CommandCost CmdBuildSingleSignal(TileInd
 		MarkTileDirtyByTile(tile);
 		AddTrackToSignalBuffer(tile, track, _current_company);
 		YapfNotifyTrackLayoutChange(tile, track);
-		if (v != NULL) TryPathReserve(v, true);
+		if (v != NULL) {
+			/* Extend the train's path if it's not stopped or loading, or not at a safe position. */
+			if (!(((v->vehstatus & VS_STOPPED) && v->cur_speed == 0) || v->current_order.IsType(OT_LOADING)) ||
+					!IsSafeWaitingPosition(v, v->tile, v->GetVehicleTrackdir(), true, _settings_game.pf.forbid_90_deg)) {
+				TryPathReserve(v, true);
+			}
+		}
 	}
 
 	return cost;
@@ -1214,6 +1220,18 @@ CommandCost CmdRemoveSingleSignal(TileIn
 		Train *v = NULL;
 		if (HasReservedTracks(tile, TrackToTrackBits(track))) {
 			v = GetTrainForReservation(tile, track);
+		} else if (IsPbsSignal(GetSignalType(tile, track))) {
+			/* PBS signal, might be the end of a path reservation. */
+			Trackdir td = TrackToTrackdir(track);
+			for (int i = 0; v == NULL && i < 2; i++, td = ReverseTrackdir(td)) {
+				/* Only test the active signal side. */
+				if (!HasSignalOnTrackdir(tile, ReverseTrackdir(td))) continue;
+				TileIndex next = TileAddByDiagDir(tile, TrackdirToExitdir(td));
+				TrackBits tracks = TrackdirBitsToTrackBits(TrackdirReachesTrackdirs(td));
+				if (HasReservedTracks(next, tracks)) {
+					v = GetTrainForReservation(next, TrackBitsToTrack(GetReservedTrackbits(next) & tracks));
+				}
+			}
 		}
 		SetPresentSignals(tile, GetPresentSignals(tile) & ~SignalOnTrack(track));