1

/* $Id$ */

     2

     3

/** @file yapf_costrail.hpp */

     4

     5

#ifndef  YAPF_COSTRAIL_HPP

     6

#define  YAPF_COSTRAIL_HPP

     7

     8

     9

template <class Types>

    10

class CYapfCostRailT

    11

	: public CYapfCostBase

    12

	, public CostRailSettings

    13

{

    14

public:

    15

	typedef typename Types::Tpf Tpf;              ///< the pathfinder class (derived from THIS class)

    16

	typedef typename Types::TrackFollower TrackFollower;

    17

	typedef typename Types::NodeList::Titem Node; ///< this will be our node type

    18

	typedef typename Node::Key Key;               ///< key to hash tables

    19

	typedef typename Node::CachedData CachedData;

    20

    21

protected:

    22

	int           m_max_cost;

    23

	CBlobT<int>   m_sig_look_ahead_costs;

    24

public:

    25

	bool          m_stopped_on_first_two_way_signal;

    26

protected:

    27

    28

	static const int s_max_segment_cost = 10000;

    29

    30

	CYapfCostRailT()

    31

		: m_max_cost(0)

    32

		, m_stopped_on_first_two_way_signal(false)

    33

	{

    34

		// pre-compute look-ahead penalties into array

    35

		int p0 = Yapf().PfGetSettings().rail_look_ahead_signal_p0;

    36

		int p1 = Yapf().PfGetSettings().rail_look_ahead_signal_p1;

    37

		int p2 = Yapf().PfGetSettings().rail_look_ahead_signal_p2;

    38

		int *pen = m_sig_look_ahead_costs.GrowSizeNC(Yapf().PfGetSettings().rail_look_ahead_max_signals);

    39

		for (uint i = 0; i < Yapf().PfGetSettings().rail_look_ahead_max_signals; i++)

    40

			pen[i] = p0 + i * (p1 + i * p2);

    41

	}

    42

    43

	/// to access inherited path finder

    44

	Tpf& Yapf() {return *static_cast<Tpf*>(this);}

    45

    46

public:

    47

	FORCEINLINE int SlopeCost(TileIndex tile, Trackdir td)

    48

	{

    49

		CPerfStart perf_cost(Yapf().m_perf_slope_cost);

    50

		if (!stSlopeCost(tile, td)) return 0;

    51

		return Yapf().PfGetSettings().rail_slope_penalty;

    52

	}

    53

    54

	FORCEINLINE int CurveCost(Trackdir td1, Trackdir td2)

    55

	{

    56

		int cost = 0;

    57

		if (TrackFollower::Allow90degTurns()

    58

				&& ((TrackdirToTrackdirBits(td2) & (TrackdirBits)TrackdirCrossesTrackdirs(td1)) != 0)) {

    59

			// 90-deg curve penalty

    60

			cost += Yapf().PfGetSettings().rail_curve90_penalty;

    61

		} else if (td2 != NextTrackdir(td1)) {

    62

			// 45-deg curve penalty

    63

			cost += Yapf().PfGetSettings().rail_curve45_penalty;

    64

		}

    65

		return cost;

    66

	}

    67

    68

	/** return one tile cost. If tile is a tunnel entry, it is moved to the end of tunnel */

    69

	FORCEINLINE int OneTileCost(TileIndex prev_tile, TileIndex& tile, Trackdir trackdir)

    70

	{

    71

		int cost = 0;

    72

		// set base cost

    73

		if (IsDiagonalTrackdir(trackdir)) {

    74

			cost += YAPF_TILE_LENGTH;

    75

			switch (GetTileType(tile)) {

    76

				case MP_STREET:

    77

					/* Increase the cost for level crossings */

    78

					if (IsLevelCrossing(tile))

    79

						cost += Yapf().PfGetSettings().rail_crossing_penalty;

    80

					break;

    81

    82

				default:

    83

					break;

    84

			}

    85

		} else {

    86

			// non-diagonal trackdir

    87

			cost = YAPF_TILE_CORNER_LENGTH;

    88

		}

    89

		return cost;

    90

	}

    91

    92

	int SignalCost(Node& n, TileIndex tile, Trackdir trackdir)

    93

	{

    94

		int cost = 0;

    95

		// if there is one-way signal in the opposite direction, then it is not our way

    96

		CPerfStart perf_cost(Yapf().m_perf_other_cost);

    97

		if (IsTileType(tile, MP_RAILWAY)) {

    98

			bool has_signal_against = HasSignalOnTrackdir(tile, ReverseTrackdir(trackdir));

    99

			bool has_signal_along = HasSignalOnTrackdir(tile, trackdir);

   100

			if (has_signal_against && !has_signal_along) {

   101

				// one-way signal in opposite direction

   102

				n.m_segment->flags_u.flags_s.m_end_of_line = true;

   103

			} else if (has_signal_along) {

   104

				SignalState sig_state = GetSignalStateByTrackdir(tile, trackdir);

   105

				// cache the look-ahead polynomial constant only if we didn't pass more signals than the look-ahead limit is

   106

				int look_ahead_cost = (n.m_num_signals_passed < m_sig_look_ahead_costs.Size()) ? m_sig_look_ahead_costs.Data()[n.m_num_signals_passed] : 0;

   107

				if (sig_state != SIGNAL_STATE_RED) {

   108

					// green signal

   109

					n.flags_u.flags_s.m_last_signal_was_red = false;

   110

					// negative look-ahead red-signal penalties would cause problems later, so use them as positive penalties for green signal

   111

					if (look_ahead_cost < 0) {

   112

						// add its negation to the cost

   113

						cost -= look_ahead_cost;

   114

					}

   115

				} else {

   116

					// we have a red signal in our direction

   117

					// was it first signal which is two-way?

   118

					if (Yapf().TreatFirstRedTwoWaySignalAsEOL() && n.flags_u.flags_s.m_choice_seen && has_signal_against && n.m_num_signals_passed == 0) {

   119

						// yes, the first signal is two-way red signal => DEAD END

   120

						n.m_segment->flags_u.flags_s.m_end_of_line = true;

   121

						Yapf().m_stopped_on_first_two_way_signal = true;

   122

						return -1;

   123

					}

   124

					SignalType sig_type = GetSignalType(tile, TrackdirToTrack(trackdir));

   125

					n.m_last_red_signal_type = sig_type;

   126

					n.flags_u.flags_s.m_last_signal_was_red = true;

   127

   128

					// look-ahead signal penalty

   129

					if (look_ahead_cost > 0) {

   130

						// add the look ahead penalty only if it is positive

   131

						cost += look_ahead_cost;

   132

					}

   133

   134

					// special signal penalties

   135

					if (n.m_num_signals_passed == 0) {

   136

						switch (sig_type) {

   137

							case SIGTYPE_COMBO:

   138

							case SIGTYPE_EXIT:   cost += Yapf().PfGetSettings().rail_firstred_exit_penalty; break; // first signal is red pre-signal-exit

   139

							case SIGTYPE_NORMAL:

   140

							case SIGTYPE_ENTRY:  cost += Yapf().PfGetSettings().rail_firstred_penalty; break;

   141

						};

   142

					}

   143

				}

   144

				n.m_num_signals_passed++;

   145

				n.m_segment->m_last_signal_tile = tile;

   146

				n.m_segment->m_last_signal_td = trackdir;

   147

			}

   148

		}

   149

		return cost;

   150

	}

   151

   152

	FORCEINLINE int PlatformLengthPenalty(int platform_length)

   153

	{

   154

		int cost = 0;

   155

		const Vehicle* v = Yapf().GetVehicle();

   156

		assert(v != NULL);

   157

		assert(v->type == VEH_TRAIN);

   158

		assert(v->u.rail.cached_total_length != 0);

   159

		int needed_platform_length = (v->u.rail.cached_total_length + TILE_SIZE - 1) / TILE_SIZE;

   160

		if (platform_length > needed_platform_length) {

   161

			// apply penalty for longer platform than needed

   162

			cost += Yapf().PfGetSettings().rail_longer_platform_penalty;

   163

		} else if (needed_platform_length > platform_length) {

   164

			// apply penalty for shorter platform than needed

   165

			cost += Yapf().PfGetSettings().rail_shorter_platform_penalty;

   166

		}

   167

		return cost;

   168

	}

   169

   170

public:

   171

	FORCEINLINE void SetMaxCost(int max_cost) {m_max_cost = max_cost;}

   172

   173

	/** Called by YAPF to calculate the cost from the origin to the given node.

   174

	 *  Calculates only the cost of given node, adds it to the parent node cost

   175

	 *  and stores the result into Node::m_cost member */

   176

	FORCEINLINE bool PfCalcCost(Node &n, const TrackFollower &tf)

   177

	{

   178

		assert(!n.flags_u.flags_s.m_targed_seen);

   179

		CPerfStart perf_cost(Yapf().m_perf_cost);

   180

		int parent_cost = (n.m_parent != NULL) ? n.m_parent->m_cost : 0;

   181

		int first_tile_cost = 0;

   182

		int segment_cost = 0;

   183

		int extra_cost = 0;

   184

		const Vehicle* v = Yapf().GetVehicle();

   185

   186

		// start at n.m_key.m_tile / n.m_key.m_td and walk to the end of segment

   187

		TileIndex prev_tile      = (n.m_parent != NULL) ? n.m_parent->GetLastTile() : INVALID_TILE;

   188

		Trackdir  prev_trackdir  = (n.m_parent != NULL) ? n.m_parent->GetLastTrackdir() : INVALID_TRACKDIR;

   189

		TileType  prev_tile_type = (n.m_parent != NULL) ? GetTileType(n.m_parent->GetLastTile()) : MP_VOID;

   190

   191

		TileIndex tile = n.m_key.m_tile;

   192

		Trackdir trackdir = n.m_key.m_td;

   193

		TileType tile_type = GetTileType(tile);

   194

   195

		RailType rail_type = GetTileRailType(tile);

   196

   197

		bool target_seen = Yapf().PfDetectDestination(tile, trackdir);

   199

   200

		if (tf.m_is_station) {

   201

			// station tiles have an extra penalty

   202

			segment_cost += Yapf().PfGetSettings().rail_station_penalty * (tf.m_tiles_skipped + 1);

   203

		}

   204

   205

		while (true) {

   206

			segment_cost += Yapf().OneTileCost(prev_tile, tile, trackdir);

   207

			segment_cost += Yapf().CurveCost(prev_trackdir, trackdir);

   208

			segment_cost += Yapf().SlopeCost(tile, trackdir);

   209

			segment_cost += Yapf().SignalCost(n, tile, trackdir);

   210

			if (n.m_segment->flags_u.flags_s.m_end_of_line) {

   211

				break;

   212

			}

   213

   214

			// finish if we have reached the destination

   215

			if (target_seen) {

   217

				break;

   218

			}

   219

   220

			// finish on first station tile - segment should end here to avoid target skipping

   221

			// when cached segments are used

   222

			if (tile_type == MP_STATION && prev_tile_type != MP_STATION) {

   223

				break;

   224

			}

   225

   226

			// finish also on waypoint - same workaround as for first station tile

   227

			if (tile_type == MP_RAILWAY && IsRailWaypoint(tile)) {

   228

				break;

   229

			}

   230

   231

			// if there are no reachable trackdirs on the next tile, we have end of road

   232

			TrackFollower F(v, &Yapf().m_perf_ts_cost);

   233

			if (!F.Follow(tile, trackdir)) {

   234

				// we can't continue?

   235

				// n.m_segment->flags_u.flags_s.m_end_of_line = true;

   236

				break;

   237

			}

   238

   239

			// if there are more trackdirs available & reachable, we are at the end of segment

   240

			if (KillFirstBit2x64(F.m_new_td_bits) != 0) {

   241

				break;

   242

			}

   243

   244

			Trackdir new_td = (Trackdir)FindFirstBit2x64(F.m_new_td_bits);

   245

   246

			{

   247

				// end segment if train is about to enter simple loop with no junctions

   248

				// so next time it should stop on the next if

   249

				if (segment_cost > s_max_segment_cost && IsTileType(F.m_new_tile, MP_RAILWAY))

   250

					break;

   251

   252

				// stop if train is on simple loop with no junctions

   253

				if (F.m_new_tile == n.m_key.m_tile && new_td == n.m_key.m_td)

   254

					return false;

   255

			}

   256

   257

			// if tail type changes, finish segment (cached segment can't contain more rail types)

   258

			{

   259

				RailType new_rail_type = GetTileRailType(F.m_new_tile);

   260

				if (new_rail_type != rail_type) {

   261

					break;

   262

				}

   263

				rail_type = new_rail_type;

   264

			}

   265

   266

			// move to the next tile

   267

			prev_tile = tile;

   268

			prev_trackdir = trackdir;

   269

			prev_tile_type = tile_type;

   270

   271

			tile = F.m_new_tile;

   272

			trackdir = new_td;

   273

			tile_type = GetTileType(tile);

   274

   275

			target_seen = Yapf().PfDetectDestination(tile, trackdir);

   276

   277

			// reversing in depot penalty

   278

			if (tile == prev_tile) {

   279

				segment_cost += Yapf().PfGetSettings().rail_depot_reverse_penalty;

   280

				break;

   281

			}

   282

   283

			// if we skipped some tunnel tiles, add their cost

   284

			segment_cost += YAPF_TILE_LENGTH * F.m_tiles_skipped;

   285

   286

			// add penalty for skipped station tiles

   287

			if (F.m_is_station)

   288

			{

   289

				uint platform_length = F.m_tiles_skipped + 1;

   290

				if (target_seen) {

   291

					// it is our destination station

   292

					segment_cost += PlatformLengthPenalty(platform_length);

   293

				} else {

   294

					// station is not our destination station, apply penalty for skipped platform tiles

   295

					segment_cost += Yapf().PfGetSettings().rail_station_penalty * platform_length;

   296

				}

   297

			}

   298

   299

			// add min/max speed penalties

   300

			int min_speed = 0;

   301

			int max_speed = F.GetSpeedLimit(&min_speed);

   302

			if (max_speed < v->max_speed)

   303

				segment_cost += YAPF_TILE_LENGTH * (v->max_speed - max_speed) / v->max_speed;

   304

			if (min_speed > v->max_speed)

   305

				segment_cost += YAPF_TILE_LENGTH * (min_speed - v->max_speed);

   306

   307

			// finish if we already exceeded the maximum cost

   308

			if (m_max_cost > 0 && (parent_cost + first_tile_cost + segment_cost) > m_max_cost) {

   309

				return false;

   310

			}

   311

   312

			if (first_tile_cost == 0) {

   313

				// we just have done first tile

   314

				first_tile_cost = segment_cost;

   315

				segment_cost = 0;

   316

   317

				// look if we can reuse existing (cached) segment cost

   318

				if (n.m_segment->m_cost >= 0) {

   319

					// reuse the cached segment cost

   320

					break;

   321

				}

   322

			}

   323

			// segment cost was not filled yes, we have not cached it yet

   324

			n.SetLastTileTrackdir(tile, trackdir);

   325

   326

		} // while (true)

   327

   328

		if (first_tile_cost == 0) {

   329

			// we have just finished first tile

   330

			first_tile_cost = segment_cost;

   331

			segment_cost = 0;

   332

		}

   333

   334

		// do we have cached segment cost?

   335

		if (n.m_segment->m_cost >= 0) {

   336

			// reuse the cached segment cost

   337

			segment_cost = n.m_segment->m_cost;

   338

		} else {

   339

			// save segment cost

   340

			n.m_segment->m_cost = segment_cost;

   341

   342

			// save end of segment back to the node

   343

			n.SetLastTileTrackdir(tile, trackdir);

   344

		}

   345

   346

		// special costs for the case we have reached our target

   348

			n.flags_u.flags_s.m_targed_seen = true;

   349

			if (n.flags_u.flags_s.m_last_signal_was_red) {

   350

				if (n.m_last_red_signal_type == SIGTYPE_EXIT) {

   351

					// last signal was red pre-signal-exit

   352

					extra_cost += Yapf().PfGetSettings().rail_lastred_exit_penalty;

   353

				} else {

   354

					// last signal was red, but not exit

   355

					extra_cost += Yapf().PfGetSettings().rail_lastred_penalty;

   356

				}

   357

			}

   358

		}

   359

   360

		// total node cost

   361

		n.m_cost = parent_cost + first_tile_cost + segment_cost + extra_cost;

   362

   364

	}

   365

   366

	FORCEINLINE bool CanUseGlobalCache(Node& n) const

   367

	{

   368

		return (n.m_parent != NULL)

   369

			&& (n.m_parent->m_num_signals_passed >= m_sig_look_ahead_costs.Size());

   370

	}

   371

   372

	FORCEINLINE void ConnectNodeToCachedData(Node& n, CachedData& ci)

   373

	{

   374

		n.m_segment = &ci;

   375

		if (n.m_segment->m_cost < 0) {

   376

			n.m_segment->m_last_tile = n.m_key.m_tile;

   377

			n.m_segment->m_last_td = n.m_key.m_td;

   378

		}

   379

	}

   380

   381

};

   382

   383

   384

   385

#endif /* YAPF_COSTRAIL_HPP */