1

/* $Id$ */

     2

     3

#ifndef  YAPF_COSTRAIL_HPP

     4

#define  YAPF_COSTRAIL_HPP

     5

     6

     7

template <class Types>

     8

class CYapfCostRailT

     9

	: public CYapfCostBase

    10

	, public CostRailSettings

    11

{

    12

public:

    13

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

    14

	typedef typename Types::TrackFollower TrackFollower;

    15

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

    16

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

    17

	typedef typename Node::CachedData CachedData;

    18

    19

protected:

    20

	int           m_max_cost;

    21

	CBlobT<int>   m_sig_look_ahead_costs;

    22

    23

	static const int s_max_segment_cost = 10000;

    24

    25

	CYapfCostRailT() : m_max_cost(0)

    26

	{

    27

		// pre-compute look-ahead penalties into array

    28

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

    29

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

    30

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

    31

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

    32

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

    33

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

    34

	}

    35

    36

	/// to access inherited path finder

    37

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

    38

    39

public:

    40

	FORCEINLINE int SlopeCost(TileIndex tile, Trackdir td)

    41

	{

    42

		CPerfStart perf_cost(Yapf().m_perf_slope_cost);

    43

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

    44

		return Yapf().PfGetSettings().rail_slope_penalty;

    45

	}

    46

    47

	FORCEINLINE int CurveCost(Trackdir td1, Trackdir td2)

    48

	{

    49

		int cost = 0;

    50

		if (TrackFollower::Allow90degTurns()

    51

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

    52

			// 90-deg curve penalty

    53

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

    54

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

    55

			// 45-deg curve penalty

    56

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

    57

		}

    58

		return cost;

    59

	}

    60

    61

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

    62

	FORCEINLINE int OneTileCost(TileIndex& tile, Trackdir trackdir)

    63

	{

    64

		int cost = 0;

    65

		// set base cost

    66

		if (IsDiagonalTrackdir(trackdir)) {

    67

			cost += YAPF_TILE_LENGTH;

    68

			switch (GetTileType(tile)) {

    69

				case MP_STREET:

    70

					/* Increase the cost for level crossings */

    71

					if (IsLevelCrossing(tile))

    72

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

    73

					break;

    74

    75

				case MP_STATION:

    76

					// penalty for passing station tiles

    77

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

    78

					break;

    79

    80

				default:

    81

					break;

    82

			}

    83

		} else {

    84

			// non-diagonal trackdir

    85

			cost = YAPF_TILE_CORNER_LENGTH;

    86

		}

    87

		return cost;

    88

	}

    89

    90

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

    91

	{

    92

		int cost = 0;

    93

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

    94

		CPerfStart perf_cost(Yapf().m_perf_other_cost);

    95

		if (IsTileType(tile, MP_RAILWAY)) {

    96

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

    97

			bool has_signal_along = HasSignalOnTrackdir(tile, trackdir);

    98

			if (has_signal_against && !has_signal_along) {

    99

				// one-way signal in opposite direction

   100

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

   101

			} else if (has_signal_along) {

   102

				SignalState sig_state = GetSignalStateByTrackdir(tile, trackdir);

   103

				if (sig_state != SIGNAL_STATE_RED) {

   104

					// green signal

   105

					n.flags_u.flags_s.m_last_signal_was_red = false;

   106

				} else {

   107

					// we have a red signal in our direction

   108

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

   109

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

   110

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

   111

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

   112

						return -1;

   113

					}

   114

					SignalType sig_type = GetSignalType(tile);

   115

					n.m_last_red_signal_type = sig_type;

   116

					n.flags_u.flags_s.m_last_signal_was_red = true;

   117

   118

					// look-ahead signal penalty

   119

					if (n.m_num_signals_passed < m_sig_look_ahead_costs.Size()) {

   120

						cost += m_sig_look_ahead_costs.Data()[n.m_num_signals_passed];

   121

					}

   122

   123

					// special signal penalties

   124

					if (n.m_num_signals_passed == 0) {

   125

						switch (sig_type) {

   126

							case SIGTYPE_COMBO:

   127

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

   128

							case SIGTYPE_NORMAL:

   129

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

   130

						};

   131

					}

   132

				}

   133

				n.m_num_signals_passed++;

   134

				n.m_segment->m_last_signal_tile = tile;

   135

				n.m_segment->m_last_signal_td = trackdir;

   136

			}

   137

		}

   138

		return cost;

   139

	}

   140

   141

	FORCEINLINE int PlatformLengthPenalty(int platform_length)

   142

	{

   143

		int cost = 0;

   144

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

   145

		assert(v != NULL);

   146

		assert(v->type == VEH_Train);

   147

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

   148

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

   149

		if (platform_length > needed_platform_length) {

   150

			// apply penalty for longer platform than needed

   151

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

   152

		} else if (needed_platform_length > platform_length) {

   153

			// apply penalty for shorter platform than needed

   154

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

   155

		}

   156

		return cost;

   157

	}

   158

   159

public:

   160

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

   161

   162

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

   163

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

   164

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

   165

	FORCEINLINE bool PfCalcCost(Node& n)

   166

	{

   167

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

   168

		CPerfStart perf_cost(Yapf().m_perf_cost);

   169

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

   170

		int first_tile_cost = 0;

   171

		int segment_cost = 0;

   172

		int extra_cost = 0;

   173

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

   174

   175

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

   176

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

   177

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

   178

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

   179

   180

		TileIndex tile = n.m_key.m_tile;

   181

		Trackdir trackdir = n.m_key.m_td;

   182

		TileType tile_type = GetTileType(tile);

   183

   184

		RailType rail_type = GetTileRailType(tile, trackdir);

   185

   186

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

   187

   188

		while (true) {

   189

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

   190

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

   191

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

   192

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

   193

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

   194

				break;

   195

			}

   196

   197

			// finish if we have reached the destination

   198

			if (target_seen) {

   199

				break;

   200

			}

   201

   202

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

   203

			// when cached segments are used

   204

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

   205

				break;

   206

			}

   207

   208

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

   209

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

   210

				break;

   211

			}

   212

   213

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

   214

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

   215

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

   216

				// we can't continue?

   217

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

   218

				break;

   219

			}

   220

   221

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

   222

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

   223

				break;

   224

			}

   225

   226

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

   227

   228

			{

   229

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

   230

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

   231

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

   232

					break;

   233

   234

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

   235

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

   236

					return false;

   237

			}

   238

   239

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

   240

			{

   241

				RailType new_rail_type = GetTileRailType(F.m_new_tile, (Trackdir)FindFirstBit2x64(F.m_new_td_bits));

   242

				if (new_rail_type != rail_type) {

   243

					break;

   244

				}

   245

				rail_type = new_rail_type;

   246

			}

   247

   248

			// move to the next tile

   249

			prev_tile = tile;

   250

			prev_trackdir = trackdir;

   251

			prev_tile_type = tile_type;

   252

   253

			tile = F.m_new_tile;

   254

			trackdir = new_td;

   255

			tile_type = GetTileType(tile);

   256

   257

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

   258

   259

			// reversing in depot penalty

   260

			if (tile == prev_tile) {

   261

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

   262

				break;

   263

			}

   264

   265

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

   266

			segment_cost += YAPF_TILE_LENGTH * F.m_tiles_skipped;

   267

   268

			// add penalty for skipped station tiles

   269

			if (F.m_is_station)

   270

			{

   271

				if (target_seen) {

   272

					// it is our destination station

   273

					uint platform_length = F.m_tiles_skipped + 1;

   274

					segment_cost += PlatformLengthPenalty(platform_length);

   275

				} else {

   276

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

   277

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

   278

				}

   279

			}

   280

   281

			// add min/max speed penalties

   282

			int min_speed = 0;

   283

			int max_speed = F.GetSpeedLimit(&min_speed);

   284

			if (max_speed < v->max_speed)

   285

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

   286

			if (min_speed > v->max_speed)

   287

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

   288

   289

			// finish if we already exceeded the maximum cost

   290

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

   291

				return false;

   292

			}

   293

   294

			if (first_tile_cost == 0) {

   295

				// we just have done first tile

   296

				first_tile_cost = segment_cost;

   297

				segment_cost = 0;

   298

   299

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

   300

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

   301

					// reuse the cached segment cost

   302

					break;

   303

				}

   304

			}

   305

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

   306

			n.SetLastTileTrackdir(tile, trackdir);

   307

   308

		} // while (true)

   309

   310

		if (first_tile_cost == 0) {

   311

			// we have just finished first tile

   312

			first_tile_cost = segment_cost;

   313

			segment_cost = 0;

   314

		}

   315

   316

		// do we have cached segment cost?

   317

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

   318

			// reuse the cached segment cost

   319

			segment_cost = n.m_segment->m_cost;

   320

		} else {

   321

			// save segment cost

   322

			n.m_segment->m_cost = segment_cost;

   323

   324

			// save end of segment back to the node

   325

			n.SetLastTileTrackdir(tile, trackdir);

   326

		}

   327

   328

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

   329

		if (target_seen) {

   330

			n.flags_u.flags_s.m_targed_seen = true;

   331

			if (n.flags_u.flags_s.m_last_signal_was_red) {

   332

				if (n.m_last_red_signal_type == SIGTYPE_EXIT) {

   333

					// last signal was red pre-signal-exit

   334

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

   335

				} else {

   336

					// last signal was red, but not exit

   337

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

   338

				}

   339

			}

   340

		}

   341

   342

		// total node cost

   343

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

   344

   345

		return !n.m_segment->flags_u.flags_s.m_end_of_line;

   346

	}

   347

   348

	FORCEINLINE bool CanUseGlobalCache(Node& n) const

   349

	{

   350

		return (n.m_parent != NULL)

   351

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

   352

	}

   353

   354

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

   355

	{

   356

		n.m_segment = &ci;

   357

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

   358

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

   359

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

   360

		}

   361

	}

   362

   363

};

   364

   365

   366

   367

#endif /* YAPF_COSTRAIL_HPP */