1644 lines
		
	
	
		
			64 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			1644 lines
		
	
	
		
			64 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
 * This file is part of OpenTTD.
 | 
						|
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 | 
						|
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 | 
						|
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 */
 | 
						|
 | 
						|
/** @file pbs.cpp PBS support routines */
 | 
						|
 | 
						|
#include "stdafx.h"
 | 
						|
#include "viewport_func.h"
 | 
						|
#include "vehicle_func.h"
 | 
						|
#include "newgrf_station.h"
 | 
						|
#include "pathfinder/follow_track.hpp"
 | 
						|
#include "tracerestrict.h"
 | 
						|
#include "newgrf_newsignals.h"
 | 
						|
#include "train_speed_adaptation.h"
 | 
						|
 | 
						|
#include "safeguards.h"
 | 
						|
 | 
						|
/**
 | 
						|
 * Get the reserved trackbits for any tile, regardless of type.
 | 
						|
 * @param t the tile
 | 
						|
 * @return the reserved trackbits. TRACK_BIT_NONE on nothing reserved or
 | 
						|
 *     a tile without rail.
 | 
						|
 */
 | 
						|
TrackBits GetReservedTrackbits(TileIndex t)
 | 
						|
{
 | 
						|
	switch (GetTileType(t)) {
 | 
						|
		case MP_RAILWAY:
 | 
						|
			if (IsRailDepot(t)) return GetDepotReservationTrackBits(t);
 | 
						|
			if (IsPlainRail(t)) return GetRailReservationTrackBits(t);
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_ROAD:
 | 
						|
			if (IsLevelCrossing(t)) return GetCrossingReservationTrackBits(t);
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_STATION:
 | 
						|
			if (HasStationRail(t)) return GetStationReservationTrackBits(t);
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_TUNNELBRIDGE:
 | 
						|
			if (GetTunnelBridgeTransportType(t) == TRANSPORT_RAIL) return GetTunnelBridgeReservationTrackBits(t);
 | 
						|
			break;
 | 
						|
 | 
						|
		default:
 | 
						|
			break;
 | 
						|
	}
 | 
						|
	return TRACK_BIT_NONE;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Set the reservation for a complete station platform.
 | 
						|
 * @pre IsRailStationTile(start)
 | 
						|
 * @param start starting tile of the platform
 | 
						|
 * @param dir the direction in which to follow the platform
 | 
						|
 * @param b the state the reservation should be set to
 | 
						|
 */
 | 
						|
void SetRailStationPlatformReservation(TileIndex start, DiagDirection dir, bool b)
 | 
						|
{
 | 
						|
	TileIndex     tile = start;
 | 
						|
	TileIndexDiff diff = TileOffsByDiagDir(dir);
 | 
						|
 | 
						|
	assert_tile(IsRailStationTile(start), start);
 | 
						|
	assert_tile(GetRailStationAxis(start) == DiagDirToAxis(dir), start);
 | 
						|
 | 
						|
	do {
 | 
						|
		SetRailStationReservation(tile, b);
 | 
						|
		MarkTileDirtyByTile(tile, VMDF_NOT_MAP_MODE);
 | 
						|
		tile = TILE_ADD(tile, diff);
 | 
						|
	} while (IsCompatibleTrainStationTile(tile, start));
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Try to reserve a specific track on a tile
 | 
						|
 * This also sets PBS signals to green if reserving through the facing track direction
 | 
						|
 * @param v the train performing the reservation
 | 
						|
 * @param tile the tile
 | 
						|
 * @param t the track
 | 
						|
 * @param trigger_stations whether to call station randomisation trigger
 | 
						|
 * @return \c true if reservation was successful, i.e. the track was
 | 
						|
 *     free and didn't cross any other reserved tracks.
 | 
						|
 */
 | 
						|
bool TryReserveRailTrackdir(const Train *v, TileIndex tile, Trackdir td, bool trigger_stations)
 | 
						|
{
 | 
						|
	bool success = TryReserveRailTrack(tile, TrackdirToTrack(td), trigger_stations);
 | 
						|
	if (success && HasPbsSignalOnTrackdir(tile, td)) {
 | 
						|
		SetSignalStateByTrackdir(tile, td, SIGNAL_STATE_GREEN);
 | 
						|
		MarkSingleSignalDirty(tile, td);
 | 
						|
		if (_extra_aspects > 0) {
 | 
						|
			SetSignalAspect(tile, TrackdirToTrack(td), 0);
 | 
						|
			UpdateAspectDeferredWithVehicle(v, tile, td, true);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return success;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Try to reserve a specific track on a tile
 | 
						|
 * @param tile the tile
 | 
						|
 * @param track the track
 | 
						|
 * @param trigger_stations whether to call station randomisation trigger
 | 
						|
 * @return \c true if reservation was successful, i.e. the track was
 | 
						|
 *     free and didn't cross any other reserved tracks.
 | 
						|
 */
 | 
						|
bool TryReserveRailTrack(TileIndex tile, Track track, bool trigger_stations)
 | 
						|
{
 | 
						|
	assert_msg_tile((TrackdirBitsToTrackBits(GetTileTrackdirBits(tile, TRANSPORT_RAIL, 0)) & TrackToTrackBits(track)) != 0, tile,
 | 
						|
			"%X, %X, %X", TrackdirBitsToTrackBits(GetTileTrackdirBits(tile, TRANSPORT_RAIL, 0)), track, TrackToTrackBits(track));
 | 
						|
 | 
						|
	if (_settings_client.gui.show_track_reservation) {
 | 
						|
		/* show the reserved rail if needed */
 | 
						|
		if (IsTileType(tile, MP_TUNNELBRIDGE)) {
 | 
						|
			MarkBridgeOrTunnelDirtyOnReservationChange(tile, VMDF_NOT_MAP_MODE);
 | 
						|
		} else {
 | 
						|
			MarkTileGroundDirtyByTile(tile, VMDF_NOT_MAP_MODE);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	switch (GetTileType(tile)) {
 | 
						|
		case MP_RAILWAY:
 | 
						|
			if (IsPlainRail(tile)) return TryReserveTrack(tile, track);
 | 
						|
			if (IsRailDepot(tile)) {
 | 
						|
				if (!HasDepotReservation(tile)) {
 | 
						|
					SetDepotReservation(tile, true);
 | 
						|
					MarkTileDirtyByTile(tile, VMDF_NOT_MAP_MODE); // some GRFs change their appearance when tile is reserved
 | 
						|
					return true;
 | 
						|
				}
 | 
						|
			}
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_ROAD:
 | 
						|
			if (IsLevelCrossing(tile) && !HasCrossingReservation(tile)) {
 | 
						|
				if (_settings_game.vehicle.safer_crossings) {
 | 
						|
					if (IsCrossingOccupiedByRoadVehicle(tile)) return false;
 | 
						|
					if (_settings_game.vehicle.adjacent_crossings) {
 | 
						|
						const Axis axis = GetCrossingRoadAxis(tile);
 | 
						|
						for (TileIndex t = tile; t < MapSize() && IsLevelCrossingTile(t) && GetCrossingRoadAxis(t) == axis; t = TileAddByDiagDir(t, AxisToDiagDir(GetCrossingRoadAxis(t)))) {
 | 
						|
							if (IsCrossingOccupiedByRoadVehicle(t)) return false;
 | 
						|
						}
 | 
						|
						for (TileIndex t = tile; t < MapSize() && IsLevelCrossingTile(t) && GetCrossingRoadAxis(t) == axis; t = TileAddByDiagDir(t, ReverseDiagDir(AxisToDiagDir(GetCrossingRoadAxis(t))))) {
 | 
						|
							if (IsCrossingOccupiedByRoadVehicle(t)) return false;
 | 
						|
						}
 | 
						|
					}
 | 
						|
				}
 | 
						|
				SetCrossingReservation(tile, true);
 | 
						|
				UpdateLevelCrossing(tile, false);
 | 
						|
				return true;
 | 
						|
			}
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_STATION:
 | 
						|
			if (HasStationRail(tile) && !HasStationReservation(tile)) {
 | 
						|
				SetRailStationReservation(tile, true);
 | 
						|
				if (trigger_stations && IsRailStation(tile)) TriggerStationRandomisation(nullptr, tile, SRT_PATH_RESERVATION);
 | 
						|
				MarkTileDirtyByTile(tile, VMDF_NOT_MAP_MODE); // some GRFs need redraw after reserving track
 | 
						|
				return true;
 | 
						|
			}
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_TUNNELBRIDGE:
 | 
						|
			if (GetTunnelBridgeTransportType(tile) == TRANSPORT_RAIL) {
 | 
						|
				if (IsTunnel(tile) && !HasTunnelReservation(tile)) {
 | 
						|
					SetTunnelReservation(tile, true);
 | 
						|
					MarkTileGroundDirtyByTile(tile, VMDF_NOT_MAP_MODE);
 | 
						|
					return true;
 | 
						|
				}
 | 
						|
				if (IsBridge(tile)) {
 | 
						|
					if (TryReserveRailBridgeHead(tile, track)) {
 | 
						|
						MarkBridgeOrTunnelDirtyOnReservationChange(tile, VMDF_NOT_MAP_MODE);
 | 
						|
						return true;
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
			break;
 | 
						|
 | 
						|
		default:
 | 
						|
			break;
 | 
						|
	}
 | 
						|
	return false;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Lift the reservation of a specific trackdir on a tile
 | 
						|
 * This also sets PBS signals to red if unreserving through the facing track direction
 | 
						|
 * @param tile the tile
 | 
						|
 * @param t the track
 | 
						|
 */
 | 
						|
void UnreserveRailTrackdir(TileIndex tile, Trackdir td)
 | 
						|
{
 | 
						|
	if (HasPbsSignalOnTrackdir(tile, td)) {
 | 
						|
		SetSignalStateByTrackdir(tile, td, SIGNAL_STATE_RED);
 | 
						|
		MarkSingleSignalDirty(tile, td);
 | 
						|
	}
 | 
						|
	UnreserveRailTrack(tile, TrackdirToTrack(td));
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Lift the reservation of a specific track on a tile
 | 
						|
 * @param tile the tile
 | 
						|
 * @param t the track
 | 
						|
 */
 | 
						|
void UnreserveRailTrack(TileIndex tile, Track t)
 | 
						|
{
 | 
						|
	assert_msg_tile(TrackdirBitsToTrackBits(GetTileTrackdirBits(tile, TRANSPORT_RAIL, 0)) & TrackToTrackBits(t), tile, "track: %u", t);
 | 
						|
 | 
						|
	if (_settings_client.gui.show_track_reservation) {
 | 
						|
		if (IsTileType(tile, MP_TUNNELBRIDGE)) {
 | 
						|
			MarkBridgeOrTunnelDirtyOnReservationChange(tile, VMDF_NOT_MAP_MODE);
 | 
						|
		} else {
 | 
						|
			MarkTileGroundDirtyByTile(tile, VMDF_NOT_MAP_MODE);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	switch (GetTileType(tile)) {
 | 
						|
		case MP_RAILWAY:
 | 
						|
			if (IsRailDepot(tile)) {
 | 
						|
				SetDepotReservation(tile, false);
 | 
						|
				MarkTileDirtyByTile(tile, VMDF_NOT_MAP_MODE);
 | 
						|
				break;
 | 
						|
			}
 | 
						|
			if (IsPlainRail(tile)) UnreserveTrack(tile, t);
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_ROAD:
 | 
						|
			if (IsLevelCrossing(tile)) {
 | 
						|
				SetCrossingReservation(tile, false);
 | 
						|
				UpdateLevelCrossing(tile);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_STATION:
 | 
						|
			if (HasStationRail(tile)) {
 | 
						|
				SetRailStationReservation(tile, false);
 | 
						|
				MarkTileDirtyByTile(tile, VMDF_NOT_MAP_MODE);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
 | 
						|
		case MP_TUNNELBRIDGE:
 | 
						|
			if (GetTunnelBridgeTransportType(tile) == TRANSPORT_RAIL) {
 | 
						|
				if (IsTunnel(tile)) {
 | 
						|
					SetTunnelReservation(tile, false);
 | 
						|
				} else {
 | 
						|
					UnreserveRailBridgeHeadTrack(tile, t);
 | 
						|
				}
 | 
						|
				if (IsTunnelBridgeSignalSimulationExit(tile) && IsTunnelBridgeEffectivelyPBS(tile) && IsTrackAcrossTunnelBridge(tile, t)) {
 | 
						|
					if (IsTunnelBridgePBS(tile)) {
 | 
						|
						SetTunnelBridgeExitSignalState(tile, SIGNAL_STATE_RED);
 | 
						|
						if (_extra_aspects > 0) PropagateAspectChange(tile, GetTunnelBridgeExitTrackdir(tile), 0);
 | 
						|
					} else {
 | 
						|
						UpdateSignalsOnSegment(tile, INVALID_DIAGDIR, GetTileOwner(tile));
 | 
						|
					}
 | 
						|
				}
 | 
						|
				MarkBridgeOrTunnelDirtyOnReservationChange(tile, VMDF_NOT_MAP_MODE);
 | 
						|
			}
 | 
						|
			break;
 | 
						|
 | 
						|
		default:
 | 
						|
			break;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Flags for FollowReservation */
 | 
						|
enum FollowReservationFlags {
 | 
						|
	FRF_NONE                 = 0,        ///< No flags
 | 
						|
	FRF_IGNORE_ONEWAY        = 0x01,     ///< Ignore one way signals in the opposite direction
 | 
						|
	FRF_TB_EXIT_FREE         = 0x02,     ///< Exit of starting tunnel/bridge is free
 | 
						|
};
 | 
						|
DECLARE_ENUM_AS_BIT_SET(FollowReservationFlags)
 | 
						|
 | 
						|
static void CheckCurveLookAhead(const Train *v, TrainReservationLookAhead *lookahead, int end_position, int z, RailType rt)
 | 
						|
{
 | 
						|
	/* Coarse filter: remove curves beyond train length */
 | 
						|
	while (!lookahead->curves.empty() && lookahead->curves.front().position < end_position - v->gcache.cached_total_length) {
 | 
						|
		lookahead->curves.pop_front();
 | 
						|
	}
 | 
						|
 | 
						|
	if (lookahead->curves.empty() || v->Next() == nullptr) return;
 | 
						|
 | 
						|
	static const int absolute_max_speed = UINT16_MAX;
 | 
						|
	int max_speed = absolute_max_speed;
 | 
						|
 | 
						|
	int curvecount[2] = {0, 0};
 | 
						|
 | 
						|
	/* first find the curve speed limit */
 | 
						|
	int numcurve = 0;
 | 
						|
	int sum = 0;
 | 
						|
	int pos = 0;
 | 
						|
	int lastpos = -1;
 | 
						|
	const Train *u = v->Next();
 | 
						|
	int veh_offset = v->CalcNextVehicleOffset();
 | 
						|
	for (auto iter = lookahead->curves.rbegin(); iter != lookahead->curves.rend(); ++iter) {
 | 
						|
		const TrainReservationLookAheadCurve &curve = *iter;
 | 
						|
		int delta = end_position - curve.position;
 | 
						|
		while (delta >= veh_offset) {
 | 
						|
			if (u->Next() != nullptr) {
 | 
						|
				veh_offset += u->CalcNextVehicleOffset();
 | 
						|
				u = u->Next();
 | 
						|
				pos++;
 | 
						|
			} else {
 | 
						|
				u = nullptr;
 | 
						|
				break;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		if (u == nullptr) break;
 | 
						|
 | 
						|
		if (curve.dir_diff == DIRDIFF_45LEFT) curvecount[0]++;
 | 
						|
		if (curve.dir_diff == DIRDIFF_45RIGHT) curvecount[1]++;
 | 
						|
		if (curve.dir_diff == DIRDIFF_45LEFT || curve.dir_diff == DIRDIFF_45RIGHT) {
 | 
						|
			if (lastpos != -1) {
 | 
						|
				numcurve++;
 | 
						|
				sum += pos - lastpos;
 | 
						|
				if (pos - lastpos == 1 && max_speed > 88) {
 | 
						|
					max_speed = 88;
 | 
						|
				}
 | 
						|
			}
 | 
						|
			lastpos = pos;
 | 
						|
		}
 | 
						|
 | 
						|
		/* if we have a 90 degree turn, fix the speed limit to 60 */
 | 
						|
		if (curve.dir_diff == DIRDIFF_90LEFT || curve.dir_diff == DIRDIFF_90RIGHT) {
 | 
						|
			max_speed = 61;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (numcurve > 0 && max_speed > 88) {
 | 
						|
		if (curvecount[0] == 1 && curvecount[1] == 1) {
 | 
						|
			max_speed = absolute_max_speed;
 | 
						|
		} else {
 | 
						|
			sum /= numcurve;
 | 
						|
			max_speed = 232 - (13 - Clamp(sum, 1, 12)) * (13 - Clamp(sum, 1, 12));
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (max_speed != absolute_max_speed) {
 | 
						|
		/* Apply the engine's rail type curve speed advantage, if it slowed by curves */
 | 
						|
		const RailTypeInfo *rti = GetRailTypeInfo(rt);
 | 
						|
		max_speed += (max_speed / 2) * rti->curve_speed;
 | 
						|
 | 
						|
		if (v->tcache.cached_tflags & TCF_TILT) {
 | 
						|
			/* Apply max_speed bonus of 20% for a tilting train */
 | 
						|
			max_speed += max_speed / 5;
 | 
						|
		}
 | 
						|
 | 
						|
		lookahead->AddCurveSpeedLimit(max_speed, 4, z);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static int LookaheadTileHeightForChunnel(int length, int offset)
 | 
						|
{
 | 
						|
	if (offset == 0) return 0;
 | 
						|
	if (offset < 3) return -1 * (int)TILE_HEIGHT;
 | 
						|
	if (offset < length - 3) return -2 * (int)TILE_HEIGHT;
 | 
						|
	if (offset < length) return -1 * (int)TILE_HEIGHT;
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static uint16_t ApplyTunnelBridgeLookaheadSignalSpeedRestriction(TileIndex tile, Trackdir trackdir, const Train *v,
 | 
						|
		uint16_t initial_speed_restriction, TrainReservationLookAhead *lookahead, int offset, int16_t z)
 | 
						|
{
 | 
						|
	uint16_t speed_restriction = initial_speed_restriction;
 | 
						|
 | 
						|
	if (v != nullptr && IsTunnelBridgeRestrictedSignal(tile)) {
 | 
						|
		if (trackdir == INVALID_TRACKDIR) {
 | 
						|
			trackdir = GetTunnelBridgeExitTrackdir(tile);
 | 
						|
		}
 | 
						|
		const TraceRestrictProgram *prog = GetExistingTraceRestrictProgram(tile, TrackdirToTrack(trackdir));
 | 
						|
		if (prog && prog->actions_used_flags & TRPAUF_SPEED_RESTRICTION) {
 | 
						|
			TraceRestrictProgramResult out;
 | 
						|
			TraceRestrictProgramInput input(tile, trackdir, nullptr, nullptr);
 | 
						|
			prog->Execute(v, input, out);
 | 
						|
			if (out.flags & TRPRF_SPEED_RESTRICTION_SET) {
 | 
						|
				int duration;
 | 
						|
				if (TrackdirEntersTunnelBridge(tile, trackdir)) {
 | 
						|
					duration = 4 + (IsDiagonalTrackdir(trackdir) ? 16 : 8);
 | 
						|
				} else {
 | 
						|
					duration = 4;
 | 
						|
				}
 | 
						|
				lookahead->AddSpeedRestriction(out.speed_restriction, offset, duration, z);
 | 
						|
				if (out.speed_restriction != 0 && (speed_restriction == 0 || out.speed_restriction < speed_restriction)) {
 | 
						|
					/* lower of the speed restrictions before or after the signal */
 | 
						|
					speed_restriction = out.speed_restriction;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return speed_restriction;
 | 
						|
}
 | 
						|
 | 
						|
static uint16_t GetTrainSpeedLimitForRailtype(const Train *v, RailType rt, TileIndex tile, Track track)
 | 
						|
{
 | 
						|
	uint16_t speed = GetRailTypeInfo(rt)->max_speed;
 | 
						|
	if (v->tcache.cached_tflags & TCF_SPD_RAILTYPE) {
 | 
						|
		for (const Train *u = v; u != nullptr; u = u->Next()) {
 | 
						|
			if (u->GetEngine()->callbacks_used & SGCU_CB36_SPEED_RAILTYPE) {
 | 
						|
				const TileIndex prev_tile = u->tile;
 | 
						|
				const TrackBits prev_track = u->track;
 | 
						|
				const_cast<Train *>(u)->tile = tile;
 | 
						|
				const_cast<Train *>(u)->track = TrackToTrackBits(track);
 | 
						|
				uint16_t cb_speed = GetVehicleProperty(u, PROP_TRAIN_SPEED, speed);
 | 
						|
				if (cb_speed != 0 && (cb_speed < speed || speed == 0)) speed = cb_speed;
 | 
						|
				const_cast<Train *>(u)->tile = prev_tile;
 | 
						|
				const_cast<Train *>(u)->track = prev_track;
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return speed;
 | 
						|
}
 | 
						|
 | 
						|
static void AddSignalToLookAhead(const Train *v, TrainReservationLookAhead *lookahead, uint16_t signal_speed, uint16_t signal_flags, TileIndex signal_tile, uint16_t signal_track, int offset, int16_t z_pos)
 | 
						|
{
 | 
						|
	lookahead->AddSignal(signal_speed, offset, z_pos, signal_flags);
 | 
						|
	if (_settings_game.vehicle.train_speed_adaptation) {
 | 
						|
		lookahead->AddSpeedAdaptation(signal_tile, signal_track, offset, z_pos);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Follow a reservation starting from a specific tile to the end. */
 | 
						|
static PBSTileInfo FollowReservation(Owner o, RailTypes rts, TileIndex tile, Trackdir trackdir, FollowReservationFlags flags, const Train *v, TrainReservationLookAhead *lookahead)
 | 
						|
{
 | 
						|
	TileIndex start_tile = tile;
 | 
						|
	Trackdir  start_trackdir = trackdir;
 | 
						|
	bool      first_loop = true;
 | 
						|
 | 
						|
	/* Start track not reserved? This can happen if two trains
 | 
						|
	 * are on the same tile. The reservation on the next tile
 | 
						|
	 * is not ours in this case, so exit. */
 | 
						|
	if (!(flags & FRF_TB_EXIT_FREE) && !HasReservedTracks(tile, TrackToTrackBits(TrackdirToTrack(trackdir)))) return PBSTileInfo(tile, trackdir, false);
 | 
						|
 | 
						|
	RailType rt = INVALID_RAILTYPE;
 | 
						|
	Direction dir = INVALID_DIR;
 | 
						|
	int z = 0;
 | 
						|
	auto update_z = [&](TileIndex t, Trackdir td, bool force) {
 | 
						|
		if (force || TrackdirToTrack(td) == TRACK_X || TrackdirToTrack(td) == TRACK_Y) {
 | 
						|
			if (IsBridgeTile(t) && TrackdirToExitdir(td) == GetTunnelBridgeDirection(t)) {
 | 
						|
				z = GetBridgePixelHeight(t);
 | 
						|
			} else {
 | 
						|
				int x = (TileX(t) * TILE_SIZE) + 8;
 | 
						|
				int y = (TileY(t) * TILE_SIZE) + 8;
 | 
						|
				if (!IsTunnelTile(tile)) {
 | 
						|
					switch (TrackdirToExitdir(td)) {
 | 
						|
						case DIAGDIR_NE: x -= 8; break;
 | 
						|
						case DIAGDIR_SE: y += 7; break;
 | 
						|
						case DIAGDIR_SW: x += 7; break;
 | 
						|
						case DIAGDIR_NW: y -= 8; break;
 | 
						|
						default: NOT_REACHED();
 | 
						|
					}
 | 
						|
				}
 | 
						|
				z = GetSlopePixelZ(x, y, true);
 | 
						|
			}
 | 
						|
		}
 | 
						|
	};
 | 
						|
 | 
						|
	if (lookahead != nullptr) {
 | 
						|
		rt = GetRailTypeByTrack(tile, TrackdirToTrack(trackdir));
 | 
						|
		dir = TrackdirToDirection(trackdir);
 | 
						|
		update_z(tile, trackdir, true);
 | 
						|
	}
 | 
						|
 | 
						|
	auto check_rail_type = [&](TileIndex t, Trackdir td, int offset) {
 | 
						|
		RailType new_rt = GetRailTypeByTrack(t, TrackdirToTrack(td));
 | 
						|
		if (new_rt != rt) {
 | 
						|
			uint16_t rail_speed = GetTrainSpeedLimitForRailtype(v, new_rt, t, TrackdirToTrack(td));
 | 
						|
			if (rail_speed > 0) lookahead->AddTrackSpeedLimit(rail_speed, offset, 4, z);
 | 
						|
			if (GetRailTypeInfo(rt)->curve_speed != GetRailTypeInfo(new_rt)->curve_speed) {
 | 
						|
				CheckCurveLookAhead(v, lookahead, lookahead->RealEndPosition() + 4 + offset, z, new_rt);
 | 
						|
			}
 | 
						|
			rt = new_rt;
 | 
						|
		}
 | 
						|
	};
 | 
						|
 | 
						|
	auto check_direction = [&](Direction new_dir, int offset, TileIndex tile) {
 | 
						|
		if (dir == new_dir) return;
 | 
						|
		DirDiff dirdiff = DirDifference(dir, new_dir);
 | 
						|
		int end = lookahead->RealEndPosition() + 4;
 | 
						|
		lookahead->curves.push_back({ end + offset, dirdiff });
 | 
						|
		dir = new_dir;
 | 
						|
		CheckCurveLookAhead(v, lookahead, end + offset, z, rt);
 | 
						|
	};
 | 
						|
 | 
						|
	/* Do not disallow 90 deg turns as the setting might have changed between reserving and now. */
 | 
						|
	CFollowTrackRail ft(o, rts);
 | 
						|
	auto check_tunnel_bridge = [&]() -> bool {
 | 
						|
		if (IsTunnelBridgeWithSignalSimulation(tile) && TrackdirEntersTunnelBridge(tile, trackdir)) {
 | 
						|
			if (_settings_game.vehicle.train_braking_model == TBM_REALISTIC && IsTunnelBridgeSignalSimulationEntrance(tile)) {
 | 
						|
				TileIndex end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
				if (HasAcrossTunnelBridgeReservation(end) && GetTunnelBridgeExitSignalState(end) == SIGNAL_STATE_GREEN &&
 | 
						|
						((flags & FRF_TB_EXIT_FREE) || TunnelBridgeIsFree(tile, end, nullptr, TBIFM_ACROSS_ONLY).Succeeded())) {
 | 
						|
					/* skip far end */
 | 
						|
					if (lookahead != nullptr) {
 | 
						|
						lookahead->reservation_end_position += (DistanceManhattan(tile, end) - 1) * TILE_SIZE;
 | 
						|
					}
 | 
						|
					Trackdir end_trackdir = GetTunnelBridgeExitTrackdir(end);
 | 
						|
					if (lookahead != nullptr) {
 | 
						|
						if ((flags & FRF_TB_EXIT_FREE) && GetTunnelBridgeLength(tile, end) > 1) {
 | 
						|
							/* middle part of bridge is in wormhole direction */
 | 
						|
							dir = DiagDirToDir(GetTunnelBridgeDirection(tile));
 | 
						|
						}
 | 
						|
						check_direction(TrackdirToDirection(end_trackdir), 0, end);
 | 
						|
						lookahead->reservation_end_position += (IsDiagonalTrackdir(end_trackdir) ? 16 : 8);
 | 
						|
						update_z(end, end_trackdir, false);
 | 
						|
					}
 | 
						|
					tile = end;
 | 
						|
					trackdir = end_trackdir;
 | 
						|
					return true;
 | 
						|
				}
 | 
						|
			}
 | 
						|
			if ((flags & FRF_IGNORE_ONEWAY) && _settings_game.vehicle.train_braking_model == TBM_REALISTIC && IsTunnelBridgeSignalSimulationExit(tile) &&
 | 
						|
					GetTunnelBridgeExitSignalState(tile) == SIGNAL_STATE_GREEN) {
 | 
						|
				TileIndex end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
				if (HasAcrossTunnelBridgeReservation(end) && TunnelBridgeIsFree(tile, end, nullptr, TBIFM_ACROSS_ONLY).Succeeded()) {
 | 
						|
					/* skip far end */
 | 
						|
					tile = end;
 | 
						|
					trackdir = GetTunnelBridgeExitTrackdir(tile);
 | 
						|
					return true;
 | 
						|
				}
 | 
						|
			}
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
		return true;
 | 
						|
	};
 | 
						|
	while (check_tunnel_bridge() && ft.Follow(tile, trackdir)) {
 | 
						|
		flags &= ~FRF_TB_EXIT_FREE;
 | 
						|
		TrackdirBits reserved = ft.m_new_td_bits & TrackBitsToTrackdirBits(GetReservedTrackbits(ft.m_new_tile));
 | 
						|
 | 
						|
		/* No reservation --> path end found */
 | 
						|
		if (reserved == TRACKDIR_BIT_NONE) {
 | 
						|
			if (ft.m_is_station) {
 | 
						|
				/* Check skipped station tiles as well, maybe our reservation ends inside the station. */
 | 
						|
				TileIndexDiff diff = TileOffsByDiagDir(ft.m_exitdir);
 | 
						|
				while (ft.m_tiles_skipped-- > 0) {
 | 
						|
					ft.m_new_tile -= diff;
 | 
						|
					if (HasStationReservation(ft.m_new_tile)) {
 | 
						|
						if (lookahead != nullptr) {
 | 
						|
							lookahead->AddStation(1 + ft.m_tiles_skipped, GetStationIndex(ft.m_new_tile), z);
 | 
						|
							lookahead->reservation_end_position += (1 + ft.m_tiles_skipped) * TILE_SIZE;
 | 
						|
						}
 | 
						|
						tile = ft.m_new_tile;
 | 
						|
						trackdir = DiagDirToDiagTrackdir(ft.m_exitdir);
 | 
						|
						break;
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		}
 | 
						|
 | 
						|
		/* Can't have more than one reserved trackdir */
 | 
						|
		Trackdir new_trackdir = FindFirstTrackdir(reserved);
 | 
						|
 | 
						|
		/* One-way signal against us. The reservation can't be ours as it is not
 | 
						|
		 * a safe position from our direction and we can never pass the signal. */
 | 
						|
		if (!(flags & FRF_IGNORE_ONEWAY) && HasOnewaySignalBlockingTrackdir(ft.m_new_tile, new_trackdir)) break;
 | 
						|
 | 
						|
		tile = ft.m_new_tile;
 | 
						|
		trackdir = new_trackdir;
 | 
						|
 | 
						|
		if (lookahead != nullptr) {
 | 
						|
			if (ft.m_tiles_skipped > 0) {
 | 
						|
				DiagDirection skip_dir = ReverseDiagDir(TrackdirToExitdir(ReverseTrackdir(trackdir)));
 | 
						|
				check_direction(DiagDirToDir(skip_dir), 0, tile);
 | 
						|
			}
 | 
						|
			if (ft.m_is_station) {
 | 
						|
				if (ft.m_tiles_skipped > 0) {
 | 
						|
					TileIndexDiff diff = TileOffsByDiagDir(TrackdirToExitdir(trackdir));
 | 
						|
					TileIndex start = tile - (diff * ft.m_tiles_skipped);
 | 
						|
					for (int i = 0; i < ft.m_tiles_skipped; i++) {
 | 
						|
						check_rail_type(start, trackdir, i * TILE_SIZE);
 | 
						|
						start += diff;
 | 
						|
					}
 | 
						|
				}
 | 
						|
				check_rail_type(tile, trackdir, ft.m_tiles_skipped * TILE_SIZE);
 | 
						|
				lookahead->AddStation(1 + ft.m_tiles_skipped, GetStationIndex(ft.m_new_tile), z);
 | 
						|
			} else {
 | 
						|
				check_rail_type(tile, trackdir, 0);
 | 
						|
			}
 | 
						|
			check_direction(TrackdirToDirection(trackdir), ft.m_tiles_skipped * TILE_SIZE, tile);
 | 
						|
			if (IsTileType(tile, MP_TUNNELBRIDGE) && TrackdirEntersTunnelBridge(tile, trackdir)) {
 | 
						|
				uint16_t bridge_speed = 0;
 | 
						|
				if (IsBridge(tile)) {
 | 
						|
					bridge_speed = GetBridgeSpec(GetBridgeType(tile))->speed;
 | 
						|
					lookahead->AddTrackSpeedLimit(bridge_speed, 0, 8, z);
 | 
						|
				}
 | 
						|
				const int start_offset = (IsDiagonalTrackdir(trackdir) ? 16 : 8);
 | 
						|
				const TileIndex end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
				const int length = GetTunnelBridgeLength(tile, end);
 | 
						|
				if (IsTunnelBridgeSignalSimulationEntrance(tile)) {
 | 
						|
					const int spacing = GetTunnelBridgeSignalSimulationSpacing(tile);
 | 
						|
					const int signals = length / spacing;
 | 
						|
 | 
						|
					uint16_t speed_restriction = ApplyTunnelBridgeLookaheadSignalSpeedRestriction(tile, trackdir, v, lookahead->speed_restriction, lookahead, 0, z);
 | 
						|
 | 
						|
					uint16_t signal_speed = GetRailTypeInfo(rt)->max_speed;
 | 
						|
					if (signal_speed == 0 || (speed_restriction != 0 && speed_restriction < signal_speed)) signal_speed = speed_restriction;
 | 
						|
					if (signal_speed == 0 || (bridge_speed != 0 && bridge_speed < signal_speed)) signal_speed = bridge_speed;
 | 
						|
 | 
						|
					const uint16_t signal_flags = GetTunnelBridgeSignalStyle(tile) << 8;
 | 
						|
 | 
						|
					/* Entrance signal */
 | 
						|
					AddSignalToLookAhead(v, lookahead, signal_speed, signal_flags, tile, TrackdirToTrack(trackdir), 0, z);
 | 
						|
 | 
						|
					update_z(tile, trackdir, false);
 | 
						|
 | 
						|
					if (length > 1) {
 | 
						|
						check_direction(DiagDirToDir(GetTunnelBridgeDirection(tile)), start_offset, tile);
 | 
						|
					}
 | 
						|
 | 
						|
					bool chunnel = IsTunnel(tile) && Tunnel::GetByTile(tile)->is_chunnel;
 | 
						|
 | 
						|
					/* Middle signals */
 | 
						|
					int offset = start_offset - TILE_SIZE;
 | 
						|
					for (int i = 0; i < signals; i++) {
 | 
						|
						offset += TILE_SIZE * spacing;
 | 
						|
						const int signal_z = chunnel ? LookaheadTileHeightForChunnel(length, i * spacing) : z;
 | 
						|
						AddSignalToLookAhead(v, lookahead, signal_speed, signal_flags, tile, 0x100 + i, offset, signal_z);
 | 
						|
					}
 | 
						|
 | 
						|
					/* Exit signal */
 | 
						|
					const int end_offset = start_offset + (TILE_SIZE * length);
 | 
						|
 | 
						|
					uint16_t exit_speed_restriction = ApplyTunnelBridgeLookaheadSignalSpeedRestriction(end, INVALID_TRACKDIR, v, lookahead->speed_restriction, lookahead, end_offset, z);
 | 
						|
					if (exit_speed_restriction != speed_restriction) {
 | 
						|
						speed_restriction = exit_speed_restriction;
 | 
						|
						signal_speed = GetRailTypeInfo(rt)->max_speed;
 | 
						|
						if (signal_speed == 0 || (speed_restriction != 0 && speed_restriction < signal_speed)) signal_speed = speed_restriction;
 | 
						|
						if (signal_speed == 0 || (bridge_speed != 0 && bridge_speed < signal_speed)) signal_speed = bridge_speed;
 | 
						|
					}
 | 
						|
 | 
						|
					AddSignalToLookAhead(v, lookahead, signal_speed, signal_flags, end, FindFirstTrack(GetAcrossTunnelBridgeTrackBits(end)), end_offset, z);
 | 
						|
 | 
						|
					lookahead->SetNextExtendPositionIfUnset();
 | 
						|
				} else {
 | 
						|
					update_z(tile, trackdir, false);
 | 
						|
					if (length > 1) {
 | 
						|
						check_direction(DiagDirToDir(GetTunnelBridgeDirection(tile)), start_offset, tile);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			if (IsTileType(tile, MP_RAILWAY) && HasSignalOnTrack(tile, TrackdirToTrack(trackdir))) {
 | 
						|
				TraceRestrictProgramActionsUsedFlags au_flags;
 | 
						|
				if (HasSignalOnTrackdir(tile, trackdir)) {
 | 
						|
					/* Passing through a signal from the front side */
 | 
						|
					au_flags = TRPAUF_SPEED_RESTRICTION;
 | 
						|
				} else {
 | 
						|
					/* Passing through a signal from the rear side */
 | 
						|
					au_flags = TRPAUF_SPEED_RESTRICTION | TRPAUF_REVERSE;
 | 
						|
				}
 | 
						|
				uint16_t speed_restriction = lookahead->speed_restriction;
 | 
						|
				if (v != nullptr) {
 | 
						|
					const TraceRestrictProgram *prog = GetExistingTraceRestrictProgram(tile, TrackdirToTrack(trackdir));
 | 
						|
					if (prog && prog->actions_used_flags & au_flags) {
 | 
						|
						TraceRestrictProgramResult out;
 | 
						|
						TraceRestrictProgramInput input(tile, trackdir, nullptr, nullptr);
 | 
						|
						prog->Execute(v, input, out);
 | 
						|
						if (out.flags & TRPRF_REVERSE && au_flags & TRPAUF_REVERSE) {
 | 
						|
							lookahead->AddReverse(z);
 | 
						|
						}
 | 
						|
						if (out.flags & TRPRF_SPEED_RESTRICTION_SET) {
 | 
						|
							lookahead->AddSpeedRestriction(out.speed_restriction, 0, 0, z);
 | 
						|
							if (out.speed_restriction != 0 && (speed_restriction == 0 || out.speed_restriction < speed_restriction)) {
 | 
						|
								/* lower of the speed restrictions before or after the signal */
 | 
						|
								speed_restriction = out.speed_restriction;
 | 
						|
							}
 | 
						|
						}
 | 
						|
					}
 | 
						|
				}
 | 
						|
				if (!(au_flags & TRPAUF_REVERSE)) {
 | 
						|
					/* Passing through a signal from the front side */
 | 
						|
					uint16_t signal_speed = GetRailTypeInfo(rt)->max_speed;
 | 
						|
					if (signal_speed == 0 || (speed_restriction != 0 && speed_restriction < signal_speed)) signal_speed = speed_restriction;
 | 
						|
					uint8_t signal_style = GetSignalStyle(tile, TrackdirToTrack(trackdir));
 | 
						|
					uint16_t signal_flags = signal_style << 8;
 | 
						|
					if (HasBit(_signal_style_masks.non_aspect_inc, signal_style)) {
 | 
						|
						SetBit(signal_flags, TRSLAI_NO_ASPECT_INC);
 | 
						|
					}
 | 
						|
					if (HasBit(_signal_style_masks.next_only, signal_style)) {
 | 
						|
						SetBit(signal_flags, TRSLAI_NEXT_ONLY);
 | 
						|
					}
 | 
						|
					if (HasBit(_signal_style_masks.combined_normal_shunt, signal_style)) {
 | 
						|
						SetBit(signal_flags, TRSLAI_COMBINED);
 | 
						|
						UpdateLookaheadCombinedNormalShuntSignalDeferred(tile, trackdir, lookahead->RealEndPosition());
 | 
						|
					}
 | 
						|
					AddSignalToLookAhead(v, lookahead, signal_speed, signal_flags, tile, TrackdirToTrack(trackdir), 0, z);
 | 
						|
					lookahead->SetNextExtendPositionIfUnset();
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			lookahead->reservation_end_position += (IsDiagonalTrackdir(trackdir) ? 16 : 8) + (ft.m_tiles_skipped * 16);
 | 
						|
			update_z(tile, trackdir, false);
 | 
						|
		}
 | 
						|
 | 
						|
		if (first_loop) {
 | 
						|
			/* Update the start tile after we followed the track the first
 | 
						|
			 * time. This is necessary because the track follower can skip
 | 
						|
			 * tiles (in stations for example) which means that we might
 | 
						|
			 * never visit our original starting tile again. */
 | 
						|
			start_tile = tile;
 | 
						|
			start_trackdir = trackdir;
 | 
						|
			first_loop = false;
 | 
						|
		} else {
 | 
						|
			/* Loop encountered? */
 | 
						|
			if (tile == start_tile && trackdir == start_trackdir) break;
 | 
						|
		}
 | 
						|
		/* Depot tile? Can't continue. */
 | 
						|
		if (IsRailDepotTile(tile)) {
 | 
						|
			if (lookahead != nullptr) SetBit(lookahead->flags, TRLF_DEPOT_END);
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		/* Non-pbs signal? Reservation can't continue. */
 | 
						|
		if (IsTileType(tile, MP_RAILWAY) && HasSignalOnTrackdir(tile, trackdir) && !IsPbsSignal(GetSignalType(tile, TrackdirToTrack(trackdir)))) break;
 | 
						|
	}
 | 
						|
 | 
						|
	if (lookahead != nullptr) lookahead->reservation_end_z = z;
 | 
						|
 | 
						|
	return PBSTileInfo(tile, trackdir, false);
 | 
						|
}
 | 
						|
 | 
						|
/** Follow a reservation starting from a specific tile to the end. */
 | 
						|
template <typename T>
 | 
						|
static void FollowReservationEnumerate(Owner o, RailTypes rts, TileIndex tile, Trackdir trackdir, FollowReservationFlags flags, T handler)
 | 
						|
{
 | 
						|
	TileIndex start_tile = tile;
 | 
						|
	Trackdir  start_trackdir = trackdir;
 | 
						|
	bool      first_loop = true;
 | 
						|
 | 
						|
	/* Start track not reserved? This can happen if two trains
 | 
						|
	 * are on the same tile. The reservation on the next tile
 | 
						|
	 * is not ours in this case, so exit. */
 | 
						|
	if (!(flags & FRF_TB_EXIT_FREE) && !HasReservedTracks(tile, TrackToTrackBits(TrackdirToTrack(trackdir)))) return;
 | 
						|
 | 
						|
	if (handler(start_tile, start_trackdir)) return;
 | 
						|
 | 
						|
	/* Do not disallow 90 deg turns as the setting might have changed between reserving and now. */
 | 
						|
	CFollowTrackRail ft(o, rts);
 | 
						|
	auto check_tunnel_bridge = [&]() -> bool {
 | 
						|
		if (IsTunnelBridgeWithSignalSimulation(tile) && TrackdirEntersTunnelBridge(tile, trackdir)) {
 | 
						|
			if (_settings_game.vehicle.train_braking_model == TBM_REALISTIC && IsTunnelBridgeSignalSimulationEntrance(tile)) {
 | 
						|
				TileIndex end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
				if (HasAcrossTunnelBridgeReservation(end) && GetTunnelBridgeExitSignalState(end) == SIGNAL_STATE_GREEN &&
 | 
						|
						((flags & FRF_TB_EXIT_FREE) || TunnelBridgeIsFree(tile, end, nullptr, TBIFM_ACROSS_ONLY).Succeeded())) {
 | 
						|
					/* skip far end */
 | 
						|
					Trackdir end_trackdir = GetTunnelBridgeExitTrackdir(end);
 | 
						|
					tile = end;
 | 
						|
					trackdir = end_trackdir;
 | 
						|
					if (handler(tile, trackdir)) return false;
 | 
						|
					return true;
 | 
						|
				}
 | 
						|
			}
 | 
						|
			if ((flags & FRF_IGNORE_ONEWAY) && _settings_game.vehicle.train_braking_model == TBM_REALISTIC && IsTunnelBridgeSignalSimulationExit(tile) &&
 | 
						|
					GetTunnelBridgeExitSignalState(tile) == SIGNAL_STATE_GREEN) {
 | 
						|
				TileIndex end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
				if (HasAcrossTunnelBridgeReservation(end) && TunnelBridgeIsFree(tile, end, nullptr, TBIFM_ACROSS_ONLY).Succeeded()) {
 | 
						|
					/* skip far end */
 | 
						|
					tile = end;
 | 
						|
					trackdir = GetTunnelBridgeExitTrackdir(tile);
 | 
						|
					if (handler(tile, trackdir)) return false;
 | 
						|
					return true;
 | 
						|
				}
 | 
						|
			}
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
		return true;
 | 
						|
	};
 | 
						|
	while (check_tunnel_bridge() && ft.Follow(tile, trackdir)) {
 | 
						|
		flags &= ~FRF_TB_EXIT_FREE;
 | 
						|
		TrackdirBits reserved = ft.m_new_td_bits & TrackBitsToTrackdirBits(GetReservedTrackbits(ft.m_new_tile));
 | 
						|
 | 
						|
		if (ft.m_is_station) {
 | 
						|
			/* Check skipped station tiles as well, maybe our reservation ends inside the station. */
 | 
						|
			TileIndexDiff diff = TileOffsByDiagDir(ft.m_exitdir);
 | 
						|
			TileIndex t = ft.m_new_tile - (ft.m_tiles_skipped * diff);
 | 
						|
			while (ft.m_tiles_skipped-- > 0) {
 | 
						|
				if (HasStationReservation(t)) {
 | 
						|
					if (handler(t, DiagDirToDiagTrackdir(ft.m_exitdir))) return;
 | 
						|
				} else {
 | 
						|
					break;
 | 
						|
				}
 | 
						|
				t += diff;
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		/* No reservation --> path end found */
 | 
						|
		if (reserved == TRACKDIR_BIT_NONE) {
 | 
						|
			break;
 | 
						|
		}
 | 
						|
 | 
						|
		/* Can't have more than one reserved trackdir */
 | 
						|
		Trackdir new_trackdir = FindFirstTrackdir(reserved);
 | 
						|
 | 
						|
		/* One-way signal against us. The reservation can't be ours as it is not
 | 
						|
		 * a safe position from our direction and we can never pass the signal. */
 | 
						|
		if (!(flags & FRF_IGNORE_ONEWAY) && HasOnewaySignalBlockingTrackdir(ft.m_new_tile, new_trackdir)) break;
 | 
						|
 | 
						|
		tile = ft.m_new_tile;
 | 
						|
		trackdir = new_trackdir;
 | 
						|
 | 
						|
		if (handler(tile, trackdir)) return;
 | 
						|
 | 
						|
		if (first_loop) {
 | 
						|
			/* Update the start tile after we followed the track the first
 | 
						|
			 * time. This is necessary because the track follower can skip
 | 
						|
			 * tiles (in stations for example) which means that we might
 | 
						|
			 * never visit our original starting tile again. */
 | 
						|
			start_tile = tile;
 | 
						|
			start_trackdir = trackdir;
 | 
						|
			first_loop = false;
 | 
						|
		} else {
 | 
						|
			/* Loop encountered? */
 | 
						|
			if (tile == start_tile && trackdir == start_trackdir) break;
 | 
						|
		}
 | 
						|
		/* Depot tile? Can't continue. */
 | 
						|
		if (IsRailDepotTile(tile)) {
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		/* Non-pbs signal? Reservation can't continue. */
 | 
						|
		if (IsTileType(tile, MP_RAILWAY) && HasSignalOnTrackdir(tile, trackdir) && !IsPbsSignal(GetSignalType(tile, TrackdirToTrack(trackdir)))) break;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Helper struct for finding the best matching vehicle on a specific track.
 | 
						|
 */
 | 
						|
struct FindTrainOnTrackInfo {
 | 
						|
	PBSTileInfo res; ///< Information about the track.
 | 
						|
	Train *best;     ///< The currently "best" vehicle we have found.
 | 
						|
 | 
						|
	/** Init the best location to nullptr always! */
 | 
						|
	FindTrainOnTrackInfo() : best(nullptr) {}
 | 
						|
};
 | 
						|
 | 
						|
/** Callback for Has/FindVehicleOnPos to find a train on a specific track. */
 | 
						|
static Vehicle *FindTrainOnTrackEnum(Vehicle *v, void *data)
 | 
						|
{
 | 
						|
	FindTrainOnTrackInfo *info = (FindTrainOnTrackInfo *)data;
 | 
						|
 | 
						|
	if ((v->vehstatus & VS_CRASHED)) return nullptr;
 | 
						|
 | 
						|
	Train *t = Train::From(v);
 | 
						|
	if (t->track & TRACK_BIT_WORMHOLE) {
 | 
						|
		/* Do not find trains inside signalled bridge/tunnels.
 | 
						|
		 * Trains on the ramp/entrance itself are found though.
 | 
						|
		 */
 | 
						|
		if (IsTileType(info->res.tile, MP_TUNNELBRIDGE) && IsTunnelBridgeWithSignalSimulation(info->res.tile) && info->res.tile != TileVirtXY(t->x_pos, t->y_pos)) {
 | 
						|
			return nullptr;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if (t->track & TRACK_BIT_WORMHOLE || HasBit((TrackBits)t->track, TrackdirToTrack(info->res.trackdir))) {
 | 
						|
		t = t->First();
 | 
						|
 | 
						|
		/* ALWAYS return the lowest ID (anti-desync!) */
 | 
						|
		if (info->best == nullptr || t->index < info->best->index) info->best = t;
 | 
						|
		return t;
 | 
						|
	}
 | 
						|
 | 
						|
	return nullptr;
 | 
						|
}
 | 
						|
 | 
						|
void TrainReservationLookAhead::SetNextExtendPosition()
 | 
						|
{
 | 
						|
	int32_t threshold = this->current_position + 24;
 | 
						|
	for (const TrainReservationLookAheadItem &item : this->items) {
 | 
						|
		if (item.type == TRLIT_SIGNAL && item.start > threshold) {
 | 
						|
			this->next_extend_position = item.start - 24;
 | 
						|
			return;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	this->next_extend_position = this->current_position;
 | 
						|
}
 | 
						|
 | 
						|
bool ValidateLookAhead(const Train *v)
 | 
						|
{
 | 
						|
	TileIndex tile = v->lookahead->reservation_end_tile;
 | 
						|
	Trackdir trackdir = v->lookahead->reservation_end_trackdir;
 | 
						|
 | 
						|
	if (HasBit(v->lookahead->flags, TRLF_TB_EXIT_FREE)) {
 | 
						|
		if (!likely(IsRailTunnelBridgeTile(tile) && TrackdirEntersTunnelBridge(tile, trackdir))) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if (HasBit(v->lookahead->flags, TRLF_DEPOT_END) && !IsRailDepotTile(tile)) return false;
 | 
						|
 | 
						|
	TrackdirBits trackdirbits = GetTileTrackdirBits(tile, TRANSPORT_RAIL, 0);
 | 
						|
	if (!HasTrackdir(trackdirbits, trackdir)) return false;
 | 
						|
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Follow a train reservation to the last tile.
 | 
						|
 *
 | 
						|
 * @param v the vehicle
 | 
						|
 * @param train_on_res Is set to a train we might encounter
 | 
						|
 * @returns The last tile of the reservation or the current train tile if no reservation present.
 | 
						|
 */
 | 
						|
PBSTileInfo FollowTrainReservation(const Train *v, Vehicle **train_on_res, FollowTrainReservationFlags flags)
 | 
						|
{
 | 
						|
	assert(v->type == VEH_TRAIN);
 | 
						|
 | 
						|
	TileIndex tile;
 | 
						|
	Trackdir  trackdir;
 | 
						|
 | 
						|
	if (!(flags & FTRF_IGNORE_LOOKAHEAD) && _settings_game.vehicle.train_braking_model == TBM_REALISTIC && v->lookahead != nullptr) {
 | 
						|
		tile = v->lookahead->reservation_end_tile;
 | 
						|
		trackdir = v->lookahead->reservation_end_trackdir;
 | 
						|
		if (HasBit(v->lookahead->flags, TRLF_DEPOT_END)) return PBSTileInfo(tile, trackdir, false);
 | 
						|
		if (HasBit(v->lookahead->flags, TRLF_TB_EXIT_FREE)) {
 | 
						|
			TileIndex exit_tile = GetOtherTunnelBridgeEnd(tile);
 | 
						|
			if (IsTunnelBridgeSignalSimulationExit(exit_tile) && GetTunnelBridgeExitSignalState(exit_tile) == SIGNAL_STATE_GREEN && HasAcrossTunnelBridgeReservation(exit_tile)) {
 | 
						|
				tile = exit_tile;
 | 
						|
				trackdir = GetTunnelBridgeExitTrackdir(exit_tile);
 | 
						|
			}
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		tile = v->tile;
 | 
						|
		trackdir = v->GetVehicleTrackdir();
 | 
						|
	}
 | 
						|
 | 
						|
	if (IsRailDepotTile(tile) && !GetDepotReservationTrackBits(tile)) return PBSTileInfo(tile, trackdir, false);
 | 
						|
 | 
						|
	FindTrainOnTrackInfo ftoti;
 | 
						|
	ftoti.res = FollowReservation(v->owner, GetRailTypeInfo(v->railtype)->all_compatible_railtypes, tile, trackdir, FRF_NONE, v, nullptr);
 | 
						|
	ftoti.res.okay = (flags & FTRF_OKAY_UNUSED) ? false : IsSafeWaitingPosition(v, ftoti.res.tile, ftoti.res.trackdir, true, _settings_game.pf.forbid_90_deg);
 | 
						|
	if (train_on_res != nullptr) {
 | 
						|
		FindVehicleOnPos(ftoti.res.tile, VEH_TRAIN, &ftoti, FindTrainOnTrackEnum);
 | 
						|
		if (ftoti.best != nullptr) *train_on_res = ftoti.best->First();
 | 
						|
		if (*train_on_res == nullptr && IsRailStationTile(ftoti.res.tile)) {
 | 
						|
			/* The target tile is a rail station. The track follower
 | 
						|
			 * has stopped on the last platform tile where we haven't
 | 
						|
			 * found a train. Also check all previous platform tiles
 | 
						|
			 * for a possible train. */
 | 
						|
			TileIndexDiff diff = TileOffsByDiagDir(TrackdirToExitdir(ReverseTrackdir(ftoti.res.trackdir)));
 | 
						|
			for (TileIndex st_tile = ftoti.res.tile + diff; *train_on_res == nullptr && IsCompatibleTrainStationTile(st_tile, ftoti.res.tile); st_tile += diff) {
 | 
						|
				FindVehicleOnPos(st_tile, VEH_TRAIN, &ftoti, FindTrainOnTrackEnum);
 | 
						|
				if (ftoti.best != nullptr) *train_on_res = ftoti.best->First();
 | 
						|
			}
 | 
						|
		}
 | 
						|
		if (*train_on_res == nullptr && IsTileType(ftoti.res.tile, MP_TUNNELBRIDGE) && IsTrackAcrossTunnelBridge(ftoti.res.tile, TrackdirToTrack(ftoti.res.trackdir)) && !IsTunnelBridgeWithSignalSimulation(ftoti.res.tile)) {
 | 
						|
			/* The target tile is a bridge/tunnel, also check the other end tile. */
 | 
						|
			FindVehicleOnPos(GetOtherTunnelBridgeEnd(ftoti.res.tile), VEH_TRAIN, &ftoti, FindTrainOnTrackEnum);
 | 
						|
			if (ftoti.best != nullptr) *train_on_res = ftoti.best->First();
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return ftoti.res;
 | 
						|
}
 | 
						|
 | 
						|
void ApplyAvailableFreeTunnelBridgeTiles(TrainReservationLookAhead *lookahead, int free_tiles, TileIndex tile, TileIndex end)
 | 
						|
{
 | 
						|
	SB(lookahead->flags, TRLF_TB_EXIT_FREE, 1, free_tiles == INT_MAX ? 1 : 0);
 | 
						|
	if (free_tiles == INT_MAX) {
 | 
						|
		/* whole tunnel/bridge is empty */
 | 
						|
		if (unlikely(end == INVALID_TILE)) end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
		free_tiles = DistanceManhattan(tile, end) - 1;
 | 
						|
	} else {
 | 
						|
		if (free_tiles > 0) {
 | 
						|
			int spacing = GetTunnelBridgeSignalSimulationSpacing(tile);
 | 
						|
			free_tiles = (((free_tiles - 1) / spacing) * spacing) - 1;
 | 
						|
		} else {
 | 
						|
			free_tiles = -1;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	lookahead->reservation_end_position += ((free_tiles - lookahead->tunnel_bridge_reserved_tiles) * TILE_SIZE);
 | 
						|
	lookahead->tunnel_bridge_reserved_tiles = free_tiles;
 | 
						|
	if (HasBit(lookahead->flags, TRLF_CHUNNEL)) {
 | 
						|
		if (unlikely(end == INVALID_TILE)) end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
		lookahead->reservation_end_z = LookaheadTileHeightForChunnel(GetTunnelBridgeLength(tile, end), free_tiles + 1);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void FillLookAheadCurveDataFromTrainPosition(Train *t)
 | 
						|
{
 | 
						|
	TileIndex tile = TileVirtXY(t->x_pos, t->y_pos);
 | 
						|
	Direction dir = t->direction;
 | 
						|
	int32_t current_pos = t->lookahead->reservation_end_position + 4 - ((dir & 1) ? 16 : 8);
 | 
						|
	for (Train *u = t->Next(); u != nullptr; u = u->Next()) {
 | 
						|
		TileIndex cur_tile = TileVirtXY(u->x_pos, u->y_pos);
 | 
						|
		if (cur_tile == tile) continue;
 | 
						|
		tile = cur_tile;
 | 
						|
		if (u->direction != dir) {
 | 
						|
			DirDiff dirdiff = DirDifference(u->direction, dir);
 | 
						|
			t->lookahead->curves.push_front({ current_pos, dirdiff });
 | 
						|
			dir = u->direction;
 | 
						|
		}
 | 
						|
		current_pos -= ((dir & 1) ? 16 : 8);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static int ScanTrainPositionForLookAheadStation(Train *t, TileIndex start_tile)
 | 
						|
{
 | 
						|
	StationID prev = INVALID_STATION;
 | 
						|
	int offset = 0;
 | 
						|
	int start_offset_tiles = 0;
 | 
						|
	TileIndex cur_tile = start_tile;
 | 
						|
	for (const Train *u = t; u != nullptr; u = u->Next()) {
 | 
						|
		if (u != t) {
 | 
						|
			TileIndex u_tile = TileVirtXY(u->x_pos, u->y_pos);
 | 
						|
			if (u_tile != cur_tile) {
 | 
						|
				offset += (IsDiagonalTrackdir(u->GetVehicleTrackdir()) ? 16 : 8);
 | 
						|
				cur_tile = u_tile;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		if (HasStationTileRail(u->tile)) {
 | 
						|
			StationID current = GetStationIndex(u->tile);
 | 
						|
			if (current != prev) {
 | 
						|
				/* Train is in a station, add that to the lookahead */
 | 
						|
				TileIndex tile = u->tile;
 | 
						|
				Trackdir trackdir = u->GetVehicleTrackdir();
 | 
						|
 | 
						|
				RailType rt = GetRailTypeByTrack(tile, TrackdirToTrack(trackdir));
 | 
						|
				int z = GetTileMaxPixelZ(tile);
 | 
						|
 | 
						|
				DiagDirection forward_dir = TrackdirToExitdir(trackdir);
 | 
						|
				TileIndexDiff diff = TileOffsByDiagDir(forward_dir);
 | 
						|
				uint forward_length = BaseStation::GetByTile(tile)->GetPlatformLength(tile, forward_dir);
 | 
						|
				uint reverse_length = BaseStation::GetByTile(tile)->GetPlatformLength(tile, ReverseDiagDir(forward_dir));
 | 
						|
 | 
						|
				if (u == t) {
 | 
						|
					for (uint i = 1; i < forward_length; i++) {
 | 
						|
						/* Check for mid platform rail type change */
 | 
						|
						TileIndex new_tile = tile + (i * diff);
 | 
						|
						RailType new_rt = GetRailTypeByTrack(new_tile, TrackdirToTrack(trackdir));
 | 
						|
						if (new_rt != rt) {
 | 
						|
							uint16_t rail_speed = GetTrainSpeedLimitForRailtype(t, new_rt, new_tile, TrackdirToTrack(trackdir));
 | 
						|
							if (rail_speed > 0) t->lookahead->AddTrackSpeedLimit(rail_speed, (i - 1) * TILE_SIZE, 4, z);
 | 
						|
							rt = new_rt;
 | 
						|
						}
 | 
						|
					}
 | 
						|
					start_offset_tiles = forward_length - 1;
 | 
						|
				}
 | 
						|
 | 
						|
				t->lookahead->AddStation(forward_length - 1, current, z);
 | 
						|
				t->lookahead->items.back().start -= offset + (reverse_length * TILE_SIZE);
 | 
						|
				t->lookahead->items.back().end -= offset;
 | 
						|
 | 
						|
				prev = current;
 | 
						|
			}
 | 
						|
		} else {
 | 
						|
			prev = INVALID_STATION;
 | 
						|
		}
 | 
						|
		if (!HasBit(u->flags, VRF_BEYOND_PLATFORM_END)) break;
 | 
						|
	}
 | 
						|
	return start_offset_tiles;
 | 
						|
}
 | 
						|
 | 
						|
void TryCreateLookAheadForTrainInTunnelBridge(Train *t)
 | 
						|
{
 | 
						|
	if (IsTunnelBridgeSignalSimulationExitOnly(t->tile)) return;
 | 
						|
	DiagDirection tb_dir = GetTunnelBridgeDirection(t->tile);
 | 
						|
	if (DirToDiagDirAlongAxis(t->direction, DiagDirToAxis(tb_dir)) == tb_dir) {
 | 
						|
		/* going in the right direction, allocate a new lookahead */
 | 
						|
		t->lookahead.reset(new TrainReservationLookAhead());
 | 
						|
		t->lookahead->reservation_end_tile = t->tile;
 | 
						|
		t->lookahead->reservation_end_trackdir = GetTunnelBridgeEntranceTrackdir(t->tile);
 | 
						|
		t->lookahead->reservation_end_z = t->z_pos;
 | 
						|
		t->lookahead->current_position = 0;
 | 
						|
		t->lookahead->next_extend_position = 0;
 | 
						|
		t->lookahead->tunnel_bridge_reserved_tiles = DistanceManhattan(t->tile, TileVirtXY(t->x_pos, t->y_pos));
 | 
						|
		t->lookahead->reservation_end_position = GetTileMarginInFrontOfTrain(t);
 | 
						|
		t->lookahead->flags = 0;
 | 
						|
		t->lookahead->speed_restriction = t->speed_restriction;
 | 
						|
		t->lookahead->cached_zpos = t->CalculateOverallZPos();
 | 
						|
		t->lookahead->zpos_refresh_remaining = t->GetZPosCacheUpdateInterval();
 | 
						|
		if (IsTunnel(t->tile) && Tunnel::GetByTile(t->tile)->is_chunnel) SetBit(t->lookahead->flags, TRLF_CHUNNEL);
 | 
						|
 | 
						|
		if (IsTunnelBridgeSignalSimulationEntrance(t->tile)) {
 | 
						|
			const uint16_t bridge_speed = IsBridge(t->tile) ? GetBridgeSpec(GetBridgeType(t->tile))->speed : 0;
 | 
						|
			const TileIndex end = GetOtherTunnelBridgeEnd(t->tile);
 | 
						|
			const int length = GetTunnelBridgeLength(t->tile, end);
 | 
						|
			const int spacing = GetTunnelBridgeSignalSimulationSpacing(t->tile);
 | 
						|
			const int signals = length / spacing;
 | 
						|
 | 
						|
			const RailType rt = GetRailTypeByTrack(t->tile, TrackdirToTrack(t->lookahead->reservation_end_trackdir));
 | 
						|
			uint16_t signal_speed = GetRailTypeInfo(rt)->max_speed;
 | 
						|
			if (signal_speed == 0 || (t->speed_restriction != 0 && t->speed_restriction < signal_speed)) signal_speed = t->speed_restriction;
 | 
						|
			if (signal_speed == 0 || (bridge_speed != 0 && bridge_speed < signal_speed)) signal_speed = bridge_speed;
 | 
						|
 | 
						|
			int z = IsBridge(t->tile) ? GetBridgeHeight(t->tile) : GetTilePixelZ(t->tile);
 | 
						|
 | 
						|
			const uint16_t signal_flags = GetTunnelBridgeSignalStyle(t->tile) << 8;
 | 
						|
 | 
						|
			/* Middle signals */
 | 
						|
			int offset = -(int)TILE_SIZE;
 | 
						|
			for (int i = 0; i < signals; i++) {
 | 
						|
				offset += TILE_SIZE * spacing;
 | 
						|
				const int signal_z = HasBit(t->lookahead->flags, TRLF_CHUNNEL) ? LookaheadTileHeightForChunnel(length, i * spacing) : z;
 | 
						|
				AddSignalToLookAhead(t, t->lookahead.get(), signal_speed, signal_flags, t->tile, 0x100 + i, offset, signal_z);
 | 
						|
			}
 | 
						|
 | 
						|
			/* Exit signal */
 | 
						|
			const int end_offset = TILE_SIZE * length;
 | 
						|
 | 
						|
			uint16_t exit_speed_restriction = ApplyTunnelBridgeLookaheadSignalSpeedRestriction(end, INVALID_TRACKDIR, t, t->speed_restriction, t->lookahead.get(), end_offset, z);
 | 
						|
			if (exit_speed_restriction != t->speed_restriction) {
 | 
						|
				signal_speed = GetRailTypeInfo(rt)->max_speed;
 | 
						|
				if (signal_speed == 0 || (exit_speed_restriction != 0 && exit_speed_restriction < signal_speed)) signal_speed = exit_speed_restriction;
 | 
						|
				if (signal_speed == 0 || (bridge_speed != 0 && bridge_speed < signal_speed)) signal_speed = bridge_speed;
 | 
						|
			}
 | 
						|
 | 
						|
			AddSignalToLookAhead(t, t->lookahead.get(), signal_speed, signal_flags, end, FindFirstTrack(GetAcrossTunnelBridgeTrackBits(end)), end_offset, z);
 | 
						|
 | 
						|
			t->lookahead->SetNextExtendPositionIfUnset();
 | 
						|
		}
 | 
						|
 | 
						|
		FillLookAheadCurveDataFromTrainPosition(t);
 | 
						|
		TileIndex end = GetOtherTunnelBridgeEnd(t->tile);
 | 
						|
		int raw_free_tiles = GetAvailableFreeTilesInSignalledTunnelBridgeWithStartOffset(t->tile, end, t->lookahead->tunnel_bridge_reserved_tiles + 1);
 | 
						|
		ApplyAvailableFreeTunnelBridgeTiles(t->lookahead.get(), raw_free_tiles, t->tile, end);
 | 
						|
		ScanTrainPositionForLookAheadStation(t, TileVirtXY(t->x_pos, t->y_pos));
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
int AdvanceTrainReservationLookaheadEnd(const Train *v, int lookahead_end_position)
 | 
						|
{
 | 
						|
	if (_settings_game.vehicle.realistic_braking_aspect_limited != TRBALM_ON || _extra_aspects == 0) {
 | 
						|
		return v->lookahead->reservation_end_position + 1;
 | 
						|
	}
 | 
						|
 | 
						|
	if (lookahead_end_position > v->lookahead->reservation_end_position) return lookahead_end_position;
 | 
						|
 | 
						|
	int32_t threshold = v->lookahead->current_position + 24;
 | 
						|
	uint8_t known_signals_ahead = 1;
 | 
						|
	bool allow_skip_no_aspect_inc = false;
 | 
						|
	if (v->IsInDepot()) {
 | 
						|
		known_signals_ahead = _extra_aspects + 1;
 | 
						|
		allow_skip_no_aspect_inc = true;
 | 
						|
	}
 | 
						|
	for (const TrainReservationLookAheadItem &item : v->lookahead->items) {
 | 
						|
		if (item.end >= v->lookahead->reservation_end_position) break;
 | 
						|
		if (item.type == TRLIT_SIGNAL) {
 | 
						|
			if (HasBit(item.data_aux, TRSLAI_COMBINED_SHUNT)) {
 | 
						|
				/* Combined normal/shunt in shunt mode */
 | 
						|
				allow_skip_no_aspect_inc = false;
 | 
						|
				if (item.start <= threshold) {
 | 
						|
					known_signals_ahead = 1;
 | 
						|
					continue;
 | 
						|
				} else {
 | 
						|
					if (item.start > lookahead_end_position) lookahead_end_position = item.start;
 | 
						|
					return lookahead_end_position;
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			if (item.start <= threshold) {
 | 
						|
				/* Signal is within visual range */
 | 
						|
				uint8_t style = item.data_aux >> 8;
 | 
						|
				uint8_t max_aspect = (style == 0) ? _extra_aspects : _new_signal_styles[style - 1].lookahead_extra_aspects;
 | 
						|
				if (max_aspect == 0xFF) {
 | 
						|
					/* This signal has unlimited lookahead */
 | 
						|
					return v->lookahead->reservation_end_position + 1;
 | 
						|
				}
 | 
						|
				if (!HasBit(item.data_aux, TRSLAI_NEXT_ONLY)) allow_skip_no_aspect_inc = true;
 | 
						|
				max_aspect += ((HasBit(item.data_aux, TRSLAI_NO_ASPECT_INC) && allow_skip_no_aspect_inc) ? 1 : 2);
 | 
						|
				if (max_aspect > known_signals_ahead) known_signals_ahead = max_aspect;
 | 
						|
			}
 | 
						|
			if (!HasBit(item.data_aux, TRSLAI_NO_ASPECT_INC) || !allow_skip_no_aspect_inc) {
 | 
						|
				known_signals_ahead--;
 | 
						|
				if (known_signals_ahead == 0) {
 | 
						|
					if (item.start > lookahead_end_position) lookahead_end_position = item.start;
 | 
						|
					return lookahead_end_position;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/* Didn't need to stop at a signal along the reservation */
 | 
						|
	if (v->lookahead->reservation_end_position >= lookahead_end_position) {
 | 
						|
		lookahead_end_position = v->lookahead->reservation_end_position;
 | 
						|
		if (known_signals_ahead > 1) lookahead_end_position++;
 | 
						|
	}
 | 
						|
	return lookahead_end_position;
 | 
						|
}
 | 
						|
 | 
						|
void SetTrainReservationLookaheadEnd(Train *v)
 | 
						|
{
 | 
						|
	v->lookahead->lookahead_end_position = AdvanceTrainReservationLookaheadEnd(v, v->lookahead->lookahead_end_position);
 | 
						|
}
 | 
						|
 | 
						|
void FillTrainReservationLookAhead(Train *v)
 | 
						|
{
 | 
						|
	TileIndex tile;
 | 
						|
	Trackdir  trackdir;
 | 
						|
 | 
						|
	if (v->lookahead == nullptr && (v->track & TRACK_BIT_WORMHOLE)) {
 | 
						|
		TryCreateLookAheadForTrainInTunnelBridge(v);
 | 
						|
		if (v->lookahead == nullptr) return;
 | 
						|
	}
 | 
						|
 | 
						|
	int32_t old_reservation_end_position = 0;
 | 
						|
 | 
						|
	if (v->lookahead == nullptr) {
 | 
						|
		v->lookahead.reset(new TrainReservationLookAhead());
 | 
						|
		v->lookahead->current_position = 0;
 | 
						|
		v->lookahead->next_extend_position = 0;
 | 
						|
 | 
						|
		/* Special case, if called from TrainController,
 | 
						|
		 * v->tile, v->track and v->direction can be updated to the new tile,
 | 
						|
		 * but v->x_pos and v->y_pos can still use the cordinates on the old tile,
 | 
						|
		 * GetTileMarginInFrontOfTrain could erroneously return -5 if the old and
 | 
						|
		 * new directions don't match. */
 | 
						|
		v->lookahead->reservation_end_position = std::max(GetTileMarginInFrontOfTrain(v), -4);
 | 
						|
 | 
						|
		v->lookahead->tunnel_bridge_reserved_tiles = 0;
 | 
						|
		v->lookahead->flags = 0;
 | 
						|
		v->lookahead->speed_restriction = v->speed_restriction;
 | 
						|
		v->lookahead->cached_zpos = v->CalculateOverallZPos();
 | 
						|
		v->lookahead->zpos_refresh_remaining = v->GetZPosCacheUpdateInterval();
 | 
						|
		FillLookAheadCurveDataFromTrainPosition(v);
 | 
						|
		tile = v->tile;
 | 
						|
		trackdir = v->GetVehicleTrackdir();
 | 
						|
		TileIndex virt_tile = TileVirtXY(v->x_pos, v->y_pos);
 | 
						|
		if (tile != virt_tile) {
 | 
						|
			v->lookahead->reservation_end_position += (IsDiagonalDirection(v->direction) ? 16 : 8);
 | 
						|
		}
 | 
						|
		int station_offset_tiles = ScanTrainPositionForLookAheadStation(v, tile);
 | 
						|
		if (station_offset_tiles > 0) {
 | 
						|
			TileIndexDiff diff = TileOffsByDiagDir(TrackdirToExitdir(trackdir));
 | 
						|
			tile += station_offset_tiles * diff;
 | 
						|
			v->lookahead->reservation_end_position += station_offset_tiles * TILE_SIZE;
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		old_reservation_end_position = v->lookahead->reservation_end_position;
 | 
						|
		tile = v->lookahead->reservation_end_tile;
 | 
						|
		trackdir = v->lookahead->reservation_end_trackdir;
 | 
						|
		if (IsTunnelBridgeSignalSimulationEntranceTile(tile) && TrackdirEntersTunnelBridge(tile, trackdir)) {
 | 
						|
			TileIndex end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
			int raw_free_tiles;
 | 
						|
			if (HasBit(v->lookahead->flags, TRLF_TB_EXIT_FREE)) {
 | 
						|
				raw_free_tiles = INT_MAX;
 | 
						|
			} else {
 | 
						|
				raw_free_tiles = GetAvailableFreeTilesInSignalledTunnelBridgeWithStartOffset(tile, end, v->lookahead->tunnel_bridge_reserved_tiles + 1);
 | 
						|
				ApplyAvailableFreeTunnelBridgeTiles(v->lookahead.get(), raw_free_tiles, tile, end);
 | 
						|
			}
 | 
						|
			if (!(HasAcrossTunnelBridgeReservation(end) && GetTunnelBridgeExitSignalState(end) == SIGNAL_STATE_GREEN && raw_free_tiles == INT_MAX)) {
 | 
						|
				/* do not attempt to follow through a signalled tunnel/bridge if it is not empty or the far end is not reserved */
 | 
						|
				FlushDeferredDetermineCombineNormalShuntMode(v);
 | 
						|
				SetTrainReservationLookaheadEnd(v);
 | 
						|
				return;
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (IsRailDepotTile(tile) && !GetDepotReservationTrackBits(tile)) {
 | 
						|
		FlushDeferredDetermineCombineNormalShuntMode(v);
 | 
						|
		SetTrainReservationLookaheadEnd(v);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	FollowReservationFlags flags = FRF_NONE;
 | 
						|
	if (HasBit(v->lookahead->flags, TRLF_TB_EXIT_FREE)) flags |= FRF_TB_EXIT_FREE;
 | 
						|
	PBSTileInfo res = FollowReservation(v->owner, GetRailTypeInfo(v->railtype)->all_compatible_railtypes, tile, trackdir, flags, v, v->lookahead.get());
 | 
						|
 | 
						|
	if (IsTunnelBridgeWithSignalSimulation(res.tile) && TrackdirEntersTunnelBridge(res.tile, res.trackdir)) {
 | 
						|
		SB(v->lookahead->flags, TRLF_CHUNNEL, 1, (IsTunnel(res.tile) && Tunnel::GetByTile(res.tile)->is_chunnel) ? 1 : 0);
 | 
						|
		if (v->lookahead->current_position < v->lookahead->reservation_end_position - ((int)TILE_SIZE * (1 + v->lookahead->tunnel_bridge_reserved_tiles))) {
 | 
						|
			/* Vehicle is not itself in this tunnel/bridge, scan how much is available */
 | 
						|
			TileIndex end = INVALID_TILE;
 | 
						|
			int free_tiles;
 | 
						|
			if (GetTunnelBridgeEntranceSignalState(res.tile) == SIGNAL_STATE_GREEN) {
 | 
						|
				end = GetOtherTunnelBridgeEnd(res.tile);
 | 
						|
				free_tiles = GetAvailableFreeTilesInSignalledTunnelBridge(res.tile, end, res.tile);
 | 
						|
			} else {
 | 
						|
				free_tiles = -1;
 | 
						|
			}
 | 
						|
			ApplyAvailableFreeTunnelBridgeTiles(v->lookahead.get(), free_tiles, res.tile, end);
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		ClrBit(v->lookahead->flags, TRLF_TB_EXIT_FREE);
 | 
						|
		ClrBit(v->lookahead->flags, TRLF_CHUNNEL);
 | 
						|
		if (v->lookahead->tunnel_bridge_reserved_tiles != 0) {
 | 
						|
			v->lookahead->reservation_end_position -= (v->lookahead->tunnel_bridge_reserved_tiles * (int)TILE_SIZE);
 | 
						|
			v->lookahead->tunnel_bridge_reserved_tiles = 0;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	v->lookahead->reservation_end_tile = res.tile;
 | 
						|
	v->lookahead->reservation_end_trackdir = res.trackdir;
 | 
						|
 | 
						|
	FlushDeferredDetermineCombineNormalShuntMode(v);
 | 
						|
	SetTrainReservationLookaheadEnd(v);
 | 
						|
 | 
						|
	if (_settings_game.vehicle.train_speed_adaptation && v->signal_speed_restriction > 0 && v->lookahead->reservation_end_position > old_reservation_end_position) {
 | 
						|
		for (const TrainReservationLookAheadItem &item : v->lookahead->items) {
 | 
						|
			if (item.type == TRLIT_SPEED_ADAPTATION && item.end + 1 >= old_reservation_end_position && item.end + 1 < v->lookahead->reservation_end_position) {
 | 
						|
				uint16_t signal_speed = GetLowestSpeedTrainAdaptationSpeedAtSignal(item.data_id, item.data_aux);
 | 
						|
 | 
						|
				if (signal_speed == 0) {
 | 
						|
					/* unrestricted signal ahead, remove current speed adaptation */
 | 
						|
					v->signal_speed_restriction = 0;
 | 
						|
					break;
 | 
						|
				}
 | 
						|
				if (signal_speed > v->signal_speed_restriction) {
 | 
						|
					/* signal ahead with higher speed, increase current speed adaptation */
 | 
						|
					v->signal_speed_restriction = signal_speed;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Find the train which has reserved a specific path.
 | 
						|
 *
 | 
						|
 * @param tile A tile on the path.
 | 
						|
 * @param track A reserved track on the tile.
 | 
						|
 * @return The vehicle holding the reservation or nullptr if the path is stray.
 | 
						|
 */
 | 
						|
Train *GetTrainForReservation(TileIndex tile, Track track)
 | 
						|
{
 | 
						|
	assert_msg_tile(HasReservedTracks(tile, TrackToTrackBits(track)), tile, "track: %u", track);
 | 
						|
	Trackdir  trackdir = TrackToTrackdir(track);
 | 
						|
 | 
						|
	RailTypes rts = GetRailTypeInfo(GetTileRailTypeByTrack(tile, track))->all_compatible_railtypes;
 | 
						|
 | 
						|
	/* Follow the path from tile to both ends, one of the end tiles should
 | 
						|
	 * have a train on it. We need FollowReservation to ignore one-way signals
 | 
						|
	 * here, as one of the two search directions will be the "wrong" way. */
 | 
						|
	for (int i = 0; i < 2; ++i, trackdir = ReverseTrackdir(trackdir)) {
 | 
						|
		/* If the tile has a one-way block signal in the current trackdir, skip the
 | 
						|
		 * search in this direction as the reservation can't come from this side.*/
 | 
						|
		if (HasOnewaySignalBlockingTrackdir(tile, ReverseTrackdir(trackdir)) && !HasPbsSignalOnTrackdir(tile, trackdir)) continue;
 | 
						|
 | 
						|
		FindTrainOnTrackInfo ftoti;
 | 
						|
		ftoti.res = FollowReservation(GetTileOwner(tile), rts, tile, trackdir, FRF_IGNORE_ONEWAY, nullptr, nullptr);
 | 
						|
 | 
						|
		FindVehicleOnPos(ftoti.res.tile, VEH_TRAIN, &ftoti, FindTrainOnTrackEnum);
 | 
						|
		if (ftoti.best != nullptr) return ftoti.best;
 | 
						|
 | 
						|
		/* Special case for stations: check the whole platform for a vehicle. */
 | 
						|
		if (IsRailStationTile(ftoti.res.tile)) {
 | 
						|
			TileIndexDiff diff = TileOffsByDiagDir(TrackdirToExitdir(ReverseTrackdir(ftoti.res.trackdir)));
 | 
						|
			for (TileIndex st_tile = ftoti.res.tile + diff; IsCompatibleTrainStationTile(st_tile, ftoti.res.tile); st_tile += diff) {
 | 
						|
				FindVehicleOnPos(st_tile, VEH_TRAIN, &ftoti, FindTrainOnTrackEnum);
 | 
						|
				if (ftoti.best != nullptr) return ftoti.best;
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		if (IsTileType(ftoti.res.tile, MP_TUNNELBRIDGE) && IsTrackAcrossTunnelBridge(ftoti.res.tile, TrackdirToTrack(ftoti.res.trackdir))) {
 | 
						|
			if (IsTunnelBridgeWithSignalSimulation(ftoti.res.tile)) {
 | 
						|
				/* Special case for signalled bridges/tunnels: find best train on bridge/tunnel if exit reserved. */
 | 
						|
				if (IsTunnelBridgeSignalSimulationExit(ftoti.res.tile) && !(IsTunnelBridgeEffectivelyPBS(ftoti.res.tile) && GetTunnelBridgeExitSignalState(ftoti.res.tile) == SIGNAL_STATE_RED)) {
 | 
						|
					ftoti.best = GetTrainClosestToTunnelBridgeEnd(ftoti.res.tile, GetOtherTunnelBridgeEnd(ftoti.res.tile));
 | 
						|
				}
 | 
						|
			} else {
 | 
						|
				/* Special case for bridges/tunnels: check the other end as well. */
 | 
						|
				FindVehicleOnPos(GetOtherTunnelBridgeEnd(ftoti.res.tile), VEH_TRAIN, &ftoti, FindTrainOnTrackEnum);
 | 
						|
			}
 | 
						|
			if (ftoti.best != nullptr) return ftoti.best;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return nullptr;
 | 
						|
}
 | 
						|
 | 
						|
CommandCost CheckTrainReservationPreventsTrackModification(TileIndex tile, Track track)
 | 
						|
{
 | 
						|
	if (_settings_game.vehicle.train_braking_model == TBM_REALISTIC) {
 | 
						|
		return CheckTrainReservationPreventsTrackModification(GetTrainForReservation(tile, track));
 | 
						|
	}
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
CommandCost CheckTrainReservationPreventsTrackModification(const Train *v)
 | 
						|
{
 | 
						|
	if (_settings_game.vehicle.train_braking_model == TBM_REALISTIC && v != nullptr && v->UsingRealisticBraking() && (v->cur_speed > 0 || !(v->vehstatus & (VS_STOPPED | VS_CRASHED)))) {
 | 
						|
		return_cmd_error(STR_ERROR_CANNOT_MODIFY_TRACK_TRAIN_APPROACHING);
 | 
						|
	}
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
static Vehicle *TrainInTunnelBridgePreventsTrackModificationEnum(Vehicle *v, void *)
 | 
						|
{
 | 
						|
	if (CheckTrainReservationPreventsTrackModification(Train::From(v)->First()).Failed()) return v;
 | 
						|
 | 
						|
	return nullptr;
 | 
						|
}
 | 
						|
 | 
						|
CommandCost CheckTrainInTunnelBridgePreventsTrackModification(TileIndex start, TileIndex end)
 | 
						|
{
 | 
						|
	if (_settings_game.vehicle.train_braking_model != TBM_REALISTIC) return CommandCost();
 | 
						|
 | 
						|
	if (HasVehicleOnPos(start, VEH_TRAIN, nullptr, &TrainInTunnelBridgePreventsTrackModificationEnum) ||
 | 
						|
			HasVehicleOnPos(end, VEH_TRAIN, nullptr, &TrainInTunnelBridgePreventsTrackModificationEnum)) {
 | 
						|
		return_cmd_error(STR_ERROR_CANNOT_MODIFY_TRACK_TRAIN_APPROACHING);
 | 
						|
	}
 | 
						|
	return CommandCost();
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * This is called to retrieve the previous signal, as required
 | 
						|
 * This is not run all the time as it is somewhat expensive and most restrictions will not test for the previous signal
 | 
						|
 */
 | 
						|
TileIndex VehiclePosTraceRestrictPreviousSignalCallback(const Train *v, const void *, TraceRestrictPBSEntrySignalAuxField mode)
 | 
						|
{
 | 
						|
	if (mode == TRPESAF_RES_END_TILE) return INVALID_TILE;
 | 
						|
 | 
						|
	TileIndex tile;
 | 
						|
	Trackdir  trackdir;
 | 
						|
 | 
						|
	if (mode == TRPESAF_RES_END && v->lookahead != nullptr) {
 | 
						|
		tile = v->lookahead->reservation_end_tile;
 | 
						|
		trackdir = v->lookahead->reservation_end_trackdir;
 | 
						|
	} else {
 | 
						|
		if (IsRailDepotTile(v->tile)) {
 | 
						|
			return v->tile;
 | 
						|
		}
 | 
						|
		if (v->track & TRACK_BIT_WORMHOLE && IsTileType(v->tile, MP_TUNNELBRIDGE) && IsTunnelBridgeSignalSimulationExit(v->tile) && IsTunnelBridgeEffectivelyPBS(v->tile)) {
 | 
						|
			return v->tile;
 | 
						|
		}
 | 
						|
		tile = v->tile;
 | 
						|
		trackdir = v->GetVehicleTrackdir();
 | 
						|
	}
 | 
						|
 | 
						|
	// scan forwards from vehicle position, for the case that train is waiting at/approaching PBS signal
 | 
						|
 | 
						|
	CFollowTrackRail ft(v);
 | 
						|
 | 
						|
	for (;;) {
 | 
						|
		if (IsTileType(tile, MP_RAILWAY) && HasSignalOnTrackdir(tile, trackdir)) {
 | 
						|
			if (HasPbsSignalOnTrackdir(tile, trackdir)) {
 | 
						|
				// found PBS signal
 | 
						|
				return tile;
 | 
						|
			} else {
 | 
						|
				// wrong type of signal
 | 
						|
				return INVALID_TILE;
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		if (IsTileType(tile, MP_TUNNELBRIDGE) && IsTunnelBridgeSignalSimulationExit(tile) && IsTunnelBridgeEffectivelyPBS(tile) && TrackdirExitsTunnelBridge(tile, trackdir)) {
 | 
						|
			return tile;
 | 
						|
		}
 | 
						|
 | 
						|
		// advance to next tile
 | 
						|
		if (!ft.Follow(tile, trackdir)) {
 | 
						|
			// ran out of track
 | 
						|
			return INVALID_TILE;
 | 
						|
		}
 | 
						|
 | 
						|
		if (KillFirstBit(ft.m_new_td_bits) != TRACKDIR_BIT_NONE) {
 | 
						|
			// reached a junction tile
 | 
						|
			return INVALID_TILE;
 | 
						|
		}
 | 
						|
 | 
						|
		tile = ft.m_new_tile;
 | 
						|
		trackdir = FindFirstTrackdir(ft.m_new_td_bits);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Test whether a train's reservation passes through a given tile.
 | 
						|
 */
 | 
						|
bool TrainReservationPassesThroughTile(const Train *v, TileIndex search_tile)
 | 
						|
{
 | 
						|
	bool found = false;
 | 
						|
	FollowReservationEnumerate(v->owner, GetRailTypeInfo(v->railtype)->all_compatible_railtypes, v->tile, v->GetVehicleTrackdir(), FRF_NONE, [&](TileIndex tile, Trackdir trackdir) -> bool {
 | 
						|
		if (tile == search_tile) {
 | 
						|
			found = true;
 | 
						|
			return true;
 | 
						|
		}
 | 
						|
		return false;
 | 
						|
	});
 | 
						|
	return found;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Determine whether a certain track on a tile is a safe position to end a path.
 | 
						|
 *
 | 
						|
 * @param v the vehicle to test for
 | 
						|
 * @param tile The tile
 | 
						|
 * @param trackdir The trackdir to test
 | 
						|
 * @param include_line_end Should end-of-line tiles be considered safe?
 | 
						|
 * @param forbid_90deg Don't allow trains to make 90 degree turns
 | 
						|
 * @return True if it is a safe position
 | 
						|
 */
 | 
						|
bool IsSafeWaitingPosition(const Train *v, TileIndex tile, Trackdir trackdir, bool include_line_end, bool forbid_90deg)
 | 
						|
{
 | 
						|
	if (IsRailDepotTile(tile)) return true;
 | 
						|
 | 
						|
	if (IsTileType(tile, MP_RAILWAY)) {
 | 
						|
		/* For non-pbs signals, stop on the signal tile. */
 | 
						|
		if (HasSignalOnTrackdir(tile, trackdir) && !IsPbsSignal(GetSignalType(tile, TrackdirToTrack(trackdir)))) return true;
 | 
						|
	}
 | 
						|
 | 
						|
	if (IsTunnelBridgeSignalSimulationEntranceTile(tile) && IsTrackAcrossTunnelBridge(tile, TrackdirToTrack(trackdir))) {
 | 
						|
		return true;
 | 
						|
	}
 | 
						|
 | 
						|
	/* Check next tile. For performance reasons, we check for 90 degree turns ourself. */
 | 
						|
	CFollowTrackRail ft(v, GetRailTypeInfo(v->railtype)->all_compatible_railtypes);
 | 
						|
 | 
						|
	/* End of track? */
 | 
						|
	if (!ft.Follow(tile, trackdir)) {
 | 
						|
		/* Last tile of a terminus station is a safe position. */
 | 
						|
		if (include_line_end) return true;
 | 
						|
	}
 | 
						|
 | 
						|
	/* Check for reachable tracks. */
 | 
						|
	ft.m_new_td_bits &= DiagdirReachesTrackdirs(ft.m_exitdir);
 | 
						|
	if (ft.m_tiles_skipped == 0 && Rail90DegTurnDisallowedTilesFromTrackdir(ft.m_old_tile, ft.m_new_tile, ft.m_old_td, forbid_90deg)) ft.m_new_td_bits &= ~TrackdirCrossesTrackdirs(trackdir);
 | 
						|
	if (ft.m_new_td_bits == TRACKDIR_BIT_NONE) return include_line_end;
 | 
						|
 | 
						|
	if (ft.m_new_td_bits != TRACKDIR_BIT_NONE && KillFirstBit(ft.m_new_td_bits) == TRACKDIR_BIT_NONE) {
 | 
						|
		Trackdir td = FindFirstTrackdir(ft.m_new_td_bits);
 | 
						|
		/* PBS signal on next trackdir? Conditionally safe position. */
 | 
						|
		if (HasPbsSignalOnTrackdir(ft.m_new_tile, td)) {
 | 
						|
			if (GetSignalType(ft.m_new_tile, TrackdirToTrack(td)) == SIGTYPE_NO_ENTRY) return include_line_end;
 | 
						|
			if (GetSignalAlwaysReserveThrough(ft.m_new_tile, TrackdirToTrack(td))) return false;
 | 
						|
			if (IsRestrictedSignal(ft.m_new_tile)) {
 | 
						|
				const TraceRestrictProgram *prog = GetExistingTraceRestrictProgram(ft.m_new_tile, TrackdirToTrack(td));
 | 
						|
				if (prog && prog->actions_used_flags & TRPAUF_RESERVE_THROUGH) {
 | 
						|
					TraceRestrictProgramResult out;
 | 
						|
					prog->Execute(v, TraceRestrictProgramInput(ft.m_new_tile, td, &VehiclePosTraceRestrictPreviousSignalCallback, nullptr), out);
 | 
						|
					if (out.flags & TRPRF_RESERVE_THROUGH) {
 | 
						|
						return false;
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
			return true;
 | 
						|
		}
 | 
						|
		/* One-way PBS signal against us? Safe if end-of-line is allowed. */
 | 
						|
		if (IsTileType(ft.m_new_tile, MP_RAILWAY) && HasSignalOnTrackdir(ft.m_new_tile, ReverseTrackdir(td)) &&
 | 
						|
				GetSignalType(ft.m_new_tile, TrackdirToTrack(td)) == SIGTYPE_PBS_ONEWAY) {
 | 
						|
			return include_line_end;
 | 
						|
		}
 | 
						|
		if (IsRailTunnelBridgeTile(ft.m_new_tile) &&
 | 
						|
				IsTrackAcrossTunnelBridge(ft.m_new_tile, TrackdirToTrack(td)) &&
 | 
						|
				IsTunnelBridgeSignalSimulationExitOnly(ft.m_new_tile) && IsTunnelBridgeEffectivelyPBS(ft.m_new_tile)) {
 | 
						|
			return include_line_end;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return false;
 | 
						|
}
 | 
						|
 | 
						|
void PBSWaitingPositionRestrictedSignalState::TraceRestrictExecuteResEndSlotIntl(const Train *v)
 | 
						|
{
 | 
						|
	TraceRestrictProgramActionsUsedFlags actions_used_flags = TRPAUF_PBS_RES_END_SLOT;
 | 
						|
	const bool tb_entrance_slots = _settings_game.vehicle.train_braking_model == TBM_REALISTIC && IsTunnelBridgeSignalSimulationEntranceTile(this->tile);
 | 
						|
	if (tb_entrance_slots) actions_used_flags |= TRPAUF_SLOT_ACQUIRE;
 | 
						|
 | 
						|
	if (prog->actions_used_flags & actions_used_flags) {
 | 
						|
		TraceRestrictProgramResult out;
 | 
						|
		TraceRestrictProgramInput input(this->tile, this->trackdir, &VehiclePosTraceRestrictPreviousSignalCallback, nullptr);
 | 
						|
		input.permitted_slot_operations = TRPISP_PBS_RES_END_ACQUIRE;
 | 
						|
		if (tb_entrance_slots) input.permitted_slot_operations |= TRPISP_ACQUIRE;
 | 
						|
		prog->Execute(v, input, out);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
bool IsWaitingPositionFreeTraceRestrictExecute(const TraceRestrictProgram *prog, const Train *v, TileIndex tile, Trackdir trackdir)
 | 
						|
{
 | 
						|
	if (prog != nullptr && prog->actions_used_flags & TRPAUF_PBS_RES_END_WAIT) {
 | 
						|
		TraceRestrictProgramInput input(tile, trackdir, &VehiclePosTraceRestrictPreviousSignalCallback, nullptr);
 | 
						|
		input.permitted_slot_operations = TRPISP_PBS_RES_END_ACQ_DRY;
 | 
						|
		TraceRestrictProgramResult out;
 | 
						|
		prog->Execute(v, input, out);
 | 
						|
		if (out.flags & TRPRF_PBS_RES_END_WAIT) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Check if a safe position is free.
 | 
						|
 *
 | 
						|
 * @param v the vehicle to test for
 | 
						|
 * @param tile The tile
 | 
						|
 * @param trackdir The trackdir to test
 | 
						|
 * @param forbid_90deg Don't allow trains to make 90 degree turns
 | 
						|
 * @param restricted_signal_state Restricted signal state in/out
 | 
						|
 * @return True if the position is free
 | 
						|
 */
 | 
						|
bool IsWaitingPositionFree(const Train *v, TileIndex tile, Trackdir trackdir, bool forbid_90deg, PBSWaitingPositionRestrictedSignalState *restricted_signal_state)
 | 
						|
{
 | 
						|
	Track     track = TrackdirToTrack(trackdir);
 | 
						|
	TrackBits reserved = GetReservedTrackbits(tile);
 | 
						|
 | 
						|
	/* Tile reserved? Can never be a free waiting position. */
 | 
						|
	if (TrackOverlapsTracks(reserved, track)) return false;
 | 
						|
 | 
						|
	/* Not reserved and depot or not a pbs signal -> free. */
 | 
						|
	if (IsRailDepotTile(tile)) return true;
 | 
						|
 | 
						|
	auto pbs_res_end_wait_test = [v, restricted_signal_state](TileIndex t, Trackdir td, bool tunnel_bridge) -> bool {
 | 
						|
		if (tunnel_bridge ? IsTunnelBridgeRestrictedSignal(t) : IsRestrictedSignal(t)) {
 | 
						|
			const TraceRestrictProgram *prog = GetExistingTraceRestrictProgram(t, TrackdirToTrack(td));
 | 
						|
			if (restricted_signal_state != nullptr && prog != nullptr) {
 | 
						|
				restricted_signal_state->prog = prog;
 | 
						|
				restricted_signal_state->tile = t;
 | 
						|
				restricted_signal_state->trackdir = td;
 | 
						|
				if (restricted_signal_state->defer_test_if_slot_conditional && (prog->actions_used_flags & TRPAUF_SLOT_CONDITIONALS) && (prog->actions_used_flags & TRPAUF_PBS_RES_END_WAIT)) {
 | 
						|
					restricted_signal_state->deferred_test = true;
 | 
						|
					return true;
 | 
						|
				}
 | 
						|
			}
 | 
						|
			return IsWaitingPositionFreeTraceRestrictExecute(prog, v, t, td);
 | 
						|
		}
 | 
						|
		return true;
 | 
						|
	};
 | 
						|
 | 
						|
	if (IsTileType(tile, MP_RAILWAY) && HasSignalOnTrackdir(tile, trackdir) && !IsPbsSignal(GetSignalType(tile, track))) {
 | 
						|
		return pbs_res_end_wait_test(tile, trackdir, false);
 | 
						|
	}
 | 
						|
 | 
						|
	if (IsTunnelBridgeSignalSimulationEntranceTile(tile) && IsTrackAcrossTunnelBridge(tile, TrackdirToTrack(trackdir))) {
 | 
						|
		bool free = pbs_res_end_wait_test(tile, trackdir, true);
 | 
						|
		if (free && IsTunnelBridgeSignalSimulationBidirectional(tile)) {
 | 
						|
			TileIndex other_end = GetOtherTunnelBridgeEnd(tile);
 | 
						|
			if (HasAcrossTunnelBridgeReservation(other_end) && GetTunnelBridgeExitSignalState(other_end) == SIGNAL_STATE_RED) return false;
 | 
						|
			Direction dir = DiagDirToDir(GetTunnelBridgeDirection(other_end));
 | 
						|
			if (HasVehicleOnPos(other_end, VEH_TRAIN, &dir, [](Vehicle *v, void *data) -> Vehicle * {
 | 
						|
				DirDiff diff = DirDifference(v->direction, *((Direction *) data));
 | 
						|
				if (diff == DIRDIFF_SAME) return v;
 | 
						|
				if (diff == DIRDIFF_45RIGHT || diff == DIRDIFF_45LEFT) {
 | 
						|
					if (GetAcrossTunnelBridgeTrackBits(v->tile) & Train::From(v)->track) return v;
 | 
						|
				}
 | 
						|
				return nullptr;
 | 
						|
			})) return false;
 | 
						|
		}
 | 
						|
		return free;
 | 
						|
	}
 | 
						|
 | 
						|
	/* Check the next tile, if it's a PBS signal, it has to be free as well. */
 | 
						|
	CFollowTrackRail ft(v, GetRailTypeInfo(v->railtype)->all_compatible_railtypes);
 | 
						|
 | 
						|
	if (!ft.Follow(tile, trackdir)) return true;
 | 
						|
 | 
						|
	/* Check for reachable tracks. */
 | 
						|
	ft.m_new_td_bits &= DiagdirReachesTrackdirs(ft.m_exitdir);
 | 
						|
	if (Rail90DegTurnDisallowedTilesFromTrackdir(ft.m_old_tile, ft.m_new_tile, ft.m_old_td, forbid_90deg)) ft.m_new_td_bits &= ~TrackdirCrossesTrackdirs(trackdir);
 | 
						|
 | 
						|
	if (HasReservedTracks(ft.m_new_tile, TrackdirBitsToTrackBits(ft.m_new_td_bits))) return false;
 | 
						|
 | 
						|
	if (ft.m_new_td_bits != TRACKDIR_BIT_NONE && KillFirstBit(ft.m_new_td_bits) == TRACKDIR_BIT_NONE) {
 | 
						|
		Trackdir td = FindFirstTrackdir(ft.m_new_td_bits);
 | 
						|
		/* PBS signal on next trackdir? */
 | 
						|
		if (HasPbsSignalOnTrackdir(ft.m_new_tile, td)) {
 | 
						|
			return pbs_res_end_wait_test(ft.m_new_tile, td, false);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return true;
 | 
						|
}
 |