1

/* $Id$ */

     2

/** @file elrail.cpp

     3

 * This file deals with displaying wires and pylons for electric railways.

     4

 * <h2>Basics</h2>

     5

 *

     6

 * <h3>Tile Types</h3>

     7

 *

     8

 * We have two different types of tiles in the drawing code:

     9

 * Normal Railway Tiles (NRTs) which can have more than one track on it, and

    10

 * Special Railways tiles (SRTs) which have only one track (like crossings, depots

    11

 * stations, etc).

    12

 *

    13

 * <h3>Location Categories</h3>

    14

 *

    15

 * All tiles are categorized into three location groups (TLG):

    16

 * Group 0: Tiles with both an even X coordinate and an even Y coordinate

    17

 * Group 1: Tiles with an even X and an odd Y coordinate

    18

 * Group 2: Tiles with an odd X and an even Y coordinate

    19

 * Group 3: Tiles with both an odd X and Y coordnate.

    20

 *

    21

 * <h3>Pylon Points</h3>

    22

 * <h4>Control Points</h4>

    23

 * A Pylon Control Point (PCP) is a position where a wire (or rather two)

    24

 * is mounted onto a pylon.

    25

 * Each NRT does contain 4 PCPs which are bitmapped to a byte

    26

 * variable and are represented by the DiagDirection enum

    27

 *

    28

 * Each track ends on two PCPs and thus requires one pylon on each end. However,

    29

 * there is one exception: Straight-and-level tracks only have one pylon every

    30

 * other tile.

    31

 *

    32

 * Now on each edge there are two PCPs: One from each adjacent tile. Both PCPs

    33

 * are merged using an OR operation (i. e. if one tile needs a PCP at the postion

    34

 * in question, both tiles get it).

    35

 *

    36

 * <h4>Position Points</h4>

    37

 * A Pylon Position Point (PPP) is a position where a pylon is located on the

    38

 * ground.  Each PCP owns 8 in (45 degree steps) PPPs that are located around

    39

 * it. PPPs are represented using the Direction enum. Each track bit has PPPs

    40

 * that are impossible (because the pylon would be situated on the track) and

    41

 * some that are preferred (because the pylon would be rectangular to the track).

    42

 *

    43

 * <img src="../../elrail_tile.png">

    44

 * <img src="../../elrail_track.png">

    45

 *

    46

 */

    47

    48

#include "stdafx.h"

    49

#include "openttd.h"

    50

#include "station_map.h"

    51

#include "tile.h"

    52

#include "viewport.h"

    53

#include "functions.h" /* We should REALLY get rid of this goddamn file, as it is butt-ugly */

    54

#include "variables.h" /* ... same here */

    55

#include "landscape.h"

    56

#include "rail.h"

    57

#include "debug.h"

    58

#include "tunnel_map.h"

    59

#include "road_map.h"

    60

#include "bridge_map.h"

    61

#include "bridge.h"

    62

#include "rail_map.h"

    63

#include "table/sprites.h"

    64

#include "table/elrail_data.h"

    65

#include "vehicle.h"

    66

#include "train.h"

    67

#include "gui.h"

    68

    69

static inline TLG GetTLG(TileIndex t)

    70

{

    71

	return (TLG)((HASBIT(TileX(t), 0) << 1) + HASBIT(TileY(t), 0));

    72

}

    73

    74

/** Finds which Rail Bits are present on a given tile. For bridge tiles,

    75

 * returns track bits under the bridge

    76

 */

    77

static TrackBits GetRailTrackBitsUniversal(TileIndex t, byte *override)

    78

{

    79

	switch (GetTileType(t)) {

    80

		case MP_RAILWAY:

    81

			if (GetRailType(t) != RAILTYPE_ELECTRIC) return TRACK_BIT_NONE;

    82

			switch (GetRailTileType(t)) {

    83

				case RAIL_TILE_NORMAL: case RAIL_TILE_SIGNALS:

    84

					return GetTrackBits(t);

    85

				case RAIL_TILE_WAYPOINT:

    86

					return GetRailWaypointBits(t);

    87

				default:

    88

					return TRACK_BIT_NONE;

    89

			}

    90

			break;

    91

    92

		case MP_TUNNELBRIDGE:

    93

			if (IsTunnel(t)) {

    94

				if (GetRailType(t) != RAILTYPE_ELECTRIC) return TRACK_BIT_NONE;

    95

				if (override != NULL) *override = 1 << GetTunnelDirection(t);

    96

				return AxisToTrackBits(DiagDirToAxis(GetTunnelDirection(t)));

    97

			} else {

    98

				if (GetRailType(t) != RAILTYPE_ELECTRIC) return TRACK_BIT_NONE;

    99

				if (override != NULL && DistanceMax(t, GetOtherBridgeEnd(t)) > 1) {

   100

					*override = 1 << GetBridgeRampDirection(t);

   101

				}

   102

				return AxisToTrackBits(DiagDirToAxis(GetBridgeRampDirection(t)));

   103

			}

   104

   105

		case MP_STREET:

   106

			if (GetRoadTileType(t) != ROAD_TILE_CROSSING) return TRACK_BIT_NONE;

   107

			if (GetRailType(t) != RAILTYPE_ELECTRIC) return TRACK_BIT_NONE;

   108

			return GetCrossingRailBits(t);

   109

   110

		case MP_STATION:

   111

			if (!IsRailwayStation(t)) return TRACK_BIT_NONE;

   112

			if (GetRailType(t) != RAILTYPE_ELECTRIC) return TRACK_BIT_NONE;

   113

			if (!IsStationTileElectrifiable(t)) return TRACK_BIT_NONE;

   114

			return TrackToTrackBits(GetRailStationTrack(t));

   115

   116

		default:

   117

			return TRACK_BIT_NONE;

   118

	}

   119

}

   120

   121

/** Corrects the tileh for certain tile types. Returns an effective tileh for the track on the tile.

   122

 * @param tile The tile to analyse

   123

 * @param *tileh the tileh

   124

 */

   125

static void AdjustTileh(TileIndex tile, Slope *tileh)

   126

{

   127

	if (IsTileType(tile, MP_TUNNELBRIDGE)) {

   128

		if (IsTunnel(tile)) {

   129

			*tileh = SLOPE_STEEP; // XXX - Hack to make tunnel entrances to always have a pylon

   130

		} else if (*tileh != SLOPE_FLAT) {

   131

			*tileh = SLOPE_FLAT;

   132

		} else {

   133

			switch (GetBridgeRampDirection(tile)) {

   134

				case DIAGDIR_NE: *tileh = SLOPE_NE; break;

   135

				case DIAGDIR_SE: *tileh = SLOPE_SE; break;

   136

				case DIAGDIR_SW: *tileh = SLOPE_SW; break;

   137

				case DIAGDIR_NW: *tileh = SLOPE_NW; break;

   138

				default: NOT_REACHED();

   139

			}

   140

		}

   141

	}

   142

}

   143

   144

/** Draws wires and, if required, pylons on a given tile

   145

 * @param ti The Tileinfo to draw the tile for

   146

 */

   147

static void DrawCatenaryRailway(const TileInfo *ti)

   148

{

   149

	/* Pylons are placed on a tile edge, so we need to take into account

   150

	 * the track configuration of 2 adjacent tiles. trackconfig[0] stores the

   151

	 * current tile (home tile) while [1] holds the neighbour */

   152

	TrackBits trackconfig[TS_END];

   153

	bool isflat[TS_END];

   154

	/* Note that ti->tileh has already been adjusted for Foundations */

   155

	Slope tileh[TS_END] = { ti->tileh, SLOPE_FLAT };

   156

   157

	TLG tlg = GetTLG(ti->tile);

   158

	byte PCPstatus = 0;

   159

	byte OverridePCP = 0;

   160

	byte PPPpreferred[DIAGDIR_END];

   161

	byte PPPallowed[DIAGDIR_END];

   162

	DiagDirection i;

   163

	Track t;

   164

   165

	/* Find which rail bits are present, and select the override points.

   166

	 * We don't draw a pylon:

   167

	 * 1) INSIDE a tunnel (we wouldn't see it anyway)

   168

	 * 2) on the "far" end of a bridge head (the one that connects to bridge middle),

   169

	 *    because that one is drawn on the bridge. Exception is for length 0 bridges

   170

	 *    which have no middle tiles */

   171

	trackconfig[TS_HOME] = GetRailTrackBitsUniversal(ti->tile, &OverridePCP);

   172

	/* If a track bit is present that is not in the main direction, the track is level */

   173

	isflat[TS_HOME] = ((trackconfig[TS_HOME] & (TRACK_BIT_HORZ | TRACK_BIT_VERT)) != 0);

   174

   175

	AdjustTileh(ti->tile, &tileh[TS_HOME]);

   176

   177

	for (i = DIAGDIR_NE; i < DIAGDIR_END; i++) {

   178

		TileIndex neighbour = ti->tile + TileOffsByDiagDir(i);

   179

		uint foundation = 0;

   180

		int k;

   181

   182

		/* Here's one of the main headaches. GetTileSlope does not correct for possibly

   183

		 * existing foundataions, so we do have to do that manually later on.*/

   184

		tileh[TS_NEIGHBOUR] = GetTileSlope(neighbour, NULL);

   185

		trackconfig[TS_NEIGHBOUR] = GetRailTrackBitsUniversal(neighbour, NULL);

   186

		if (IsTunnelTile(neighbour) && i != GetTunnelDirection(neighbour)) trackconfig[TS_NEIGHBOUR] = TRACK_BIT_NONE;

   187

		isflat[TS_NEIGHBOUR] = ((trackconfig[TS_NEIGHBOUR] & (TRACK_BIT_HORZ | TRACK_BIT_VERT)) != 0);

   188

   189

		PPPpreferred[i] = 0xFF; // We start with preferring everything (end-of-line in any direction)

   190

		PPPallowed[i] = AllowedPPPonPCP[i];

   191

   192

		/* We cycle through all the existing tracks at a PCP and see what

   193

		 * PPPs we want to have, or may not have at all */

   194

		for (k = 0; k < NUM_TRACKS_AT_PCP; k++) {

   195

			/* Next to us, we have a bridge head, don't worry about that one, if it shows away from us */

   196

			if (TrackSourceTile[i][k] == TS_NEIGHBOUR &&

   197

			    IsBridgeTile(neighbour) &&

   198

			    GetBridgeRampDirection(neighbour) == ReverseDiagDir(i)) {

   199

				continue;

   200

			}

   201

   202

			/* We check whether the track in question (k) is present in the tile

   203

			 * (TrackSourceTile) */

   204

			if (HASBIT(trackconfig[TrackSourceTile[i][k]], TracksAtPCP[i][k])) {

   205

				/* track found, if track is in the neighbour tile, adjust the number

   206

				 * of the PCP for preferred/allowed determination*/

   207

				DiagDirection PCPpos = (TrackSourceTile[i][k] == TS_HOME) ? i : ReverseDiagDir(i);

   208

				SETBIT(PCPstatus, i); // This PCP is in use

   209

   210

				PPPpreferred[i] &= PreferredPPPofTrackAtPCP[TracksAtPCP[i][k]][PCPpos];

   211

				PPPallowed[i] &= ~DisallowedPPPofTrackAtPCP[TracksAtPCP[i][k]][PCPpos];

   212

			}

   213

		}

   214

   215

		/* Deactivate all PPPs if PCP is not used */

   216

		PPPpreferred[i] *= HASBIT(PCPstatus, i);

   217

		PPPallowed[i] *= HASBIT(PCPstatus, i);

   218

   219

		/* A station is always "flat", so adjust the tileh accordingly */

   220

		if (IsTileType(neighbour, MP_STATION)) tileh[TS_NEIGHBOUR] = SLOPE_FLAT;

   221

   222

		/* Read the foundataions if they are present, and adjust the tileh */

   223

		if (IsTileType(neighbour, MP_RAILWAY) && GetRailType(neighbour) == RAILTYPE_ELECTRIC) foundation = GetRailFoundation(tileh[TS_NEIGHBOUR], trackconfig[TS_NEIGHBOUR]);

   224

		if (IsBridgeTile(neighbour)) {

   225

			foundation = GetBridgeFoundation(tileh[TS_NEIGHBOUR], DiagDirToAxis(GetBridgeRampDirection(neighbour)));

   226

		}

   227

   228

		if (foundation != 0) {

   229

			tileh[TS_NEIGHBOUR] = foundation < 15 ? SLOPE_FLAT : _inclined_tileh[foundation - 15];

   230

		}

   231

   232

		AdjustTileh(neighbour, &tileh[TS_NEIGHBOUR]);

   233

   234

		/* If we have a straight (and level) track, we want a pylon only every 2 tiles

   235

		 * Delete the PCP if this is the case. */

   236

		/* Level means that the slope is the same, or the track is flat */

   237

		if (tileh[TS_HOME] == tileh[TS_NEIGHBOUR] || (isflat[TS_HOME] && isflat[TS_NEIGHBOUR])) {

   238

			for (k = 0; k < NUM_IGNORE_GROUPS; k++)

   239

				if (PPPpreferred[i] == IgnoredPCP[k][tlg][i]) CLRBIT(PCPstatus, i);

   240

		}

   241

   242

		/* Now decide where we draw our pylons. First try the preferred PPPs, but they may not exist.

   243

		 * In that case, we try the any of the allowed ones. if they don't exist either, don't draw

   244

		 * anything. Note that the preferred PPPs still contain the end-of-line markers.

   245

		 * Remove those (simply by ANDing with allowed, since these markers are never allowed) */

   246

		if ((PPPallowed[i] & PPPpreferred[i]) != 0) PPPallowed[i] &= PPPpreferred[i];

   247

   248

		if (MayHaveBridgeAbove(ti->tile) && IsBridgeAbove(ti->tile)) {

   249

			Track bridgetrack = GetBridgeAxis(ti->tile) == AXIS_X ? TRACK_X : TRACK_Y;

   250

			uint height = GetBridgeHeight(GetNorthernBridgeEnd(ti->tile));

   251

   252

			if ((height <= GetTileMaxZ(ti->tile) + TILE_HEIGHT) &&

   253

					(i == PCPpositions[bridgetrack][0] || i == PCPpositions[bridgetrack][1])) {

   254

				SETBIT(OverridePCP, i);

   255

			}

   256

		}

   257

   258

		if (PPPallowed[i] != 0 && HASBIT(PCPstatus, i) && !HASBIT(OverridePCP, i)) {

   259

			for (k = 0; k < DIR_END; k++) {

   260

				byte temp = PPPorder[i][GetTLG(ti->tile)][k];

   261

   262

				if (HASBIT(PPPallowed[i], temp)) {

   263

					uint x  = ti->x + x_pcp_offsets[i] + x_ppp_offsets[temp];

   264

					uint y  = ti->y + y_pcp_offsets[i] + y_ppp_offsets[temp];

   265

   266

					/* Don't build the pylon if it would be outside the tile */

   267

					if (!HASBIT(OwnedPPPonPCP[i], temp)) {

   268

						/* We have a neighour that will draw it, bail out */

   269

						if (trackconfig[TS_NEIGHBOUR] != 0) break;

   270

						continue; /* No neighbour, go looking for a better position */

   271

					}

   272

   273

					SpriteID img = pylons_normal[temp];

   274

					SpriteID pal = PAL_NONE;

   275

					if (HASBIT(_transparent_opt, TO_BUILDINGS)) {

   276

						SETBIT(img, PALETTE_MODIFIER_TRANSPARENT);

   277

						pal = PALETTE_TO_TRANSPARENT;

   278

					}

   279

   280

					AddSortableSpriteToDraw(img, pal, x, y, 1, 1, 10,

   281

							GetSlopeZ(ti->x + x_pcp_offsets[i], ti->y + y_pcp_offsets[i]));

   282

					break; /* We already have drawn a pylon, bail out */

   283

				}

   284

			}

   285

		}

   286

	}

   287

   288

	/* Don't draw a wire under a low bridge */

   289

	if (MayHaveBridgeAbove(ti->tile) && IsBridgeAbove(ti->tile) && !HASBIT(_transparent_opt, TO_BUILDINGS)) {

   290

		uint height = GetBridgeHeight(GetNorthernBridgeEnd(ti->tile));

   291

   292

		if (height <= GetTileMaxZ(ti->tile) + TILE_HEIGHT) return;

   293

	}

   294

   295

	/* Drawing of pylons is finished, now draw the wires */

   296

	for (t = TRACK_BEGIN; t < TRACK_END; t++) {

   297

		if (HASBIT(trackconfig[TS_HOME], t)) {

   298

			if (IsTunnelTile(ti->tile)) {

   299

				const SortableSpriteStruct *sss = &CatenarySpriteData_Tunnel[GetTunnelDirection(ti->tile)];

   300

   301

				AddSortableSpriteToDraw(

   302

					sss->image, PAL_NONE, ti->x + sss->x_offset, ti->y + sss->y_offset,

   303

					sss->x_size, sss->y_size, sss->z_size,

   304

					GetTileZ(ti->tile) + sss->z_offset

   305

				);

   306

				break;

   307

			}

   308

			byte PCPconfig = HASBIT(PCPstatus, PCPpositions[t][0]) +

   309

				(HASBIT(PCPstatus, PCPpositions[t][1]) << 1);

   310

   311

			const SortableSpriteStruct *sss;

   312

			int tileh_selector = !(tileh[TS_HOME] % 3) * tileh[TS_HOME] / 3; /* tileh for the slopes, 0 otherwise */

   313

   314

			assert(PCPconfig != 0); /* We have a pylon on neither end of the wire, that doesn't work (since we have no sprites for that) */

   315

			assert(!IsSteepSlope(tileh[TS_HOME]));

   316

			sss = &CatenarySpriteData[Wires[tileh_selector][t][PCPconfig]];

   317

   318

			SpriteID img = sss->image;

   319

			SpriteID pal = PAL_NONE;

   320

			if (HASBIT(_transparent_opt, TO_BUILDINGS)) {

   321

				SETBIT(img, PALETTE_MODIFIER_TRANSPARENT);

   322

				pal = PALETTE_TO_TRANSPARENT;

   323

			}

   324

   325

			AddSortableSpriteToDraw(img, pal, ti->x + sss->x_offset, ti->y + sss->y_offset,

   326

				sss->x_size, sss->y_size, sss->z_size, GetSlopeZ(ti->x + min(sss->x_offset, TILE_SIZE - 1), ti->y + min(sss->y_offset, TILE_SIZE - 1)) + sss->z_offset);

   327

		}

   328

	}

   329

}

   330

   331

static void DrawCatenaryOnBridge(const TileInfo *ti)

   332

{

   333

	TileIndex end = GetSouthernBridgeEnd(ti->tile);

   334

	TileIndex start = GetOtherBridgeEnd(end);

   335

   336

	uint length = GetBridgeLength(start, end);

   337

	uint num = DistanceMax(ti->tile, start);

   338

	uint height;

   339

   340

	const SortableSpriteStruct *sss;

   341

	Axis axis = GetBridgeAxis(ti->tile);

   342

	TLG tlg = GetTLG(ti->tile);

   343

   344

	CatenarySprite offset = (CatenarySprite)(axis == AXIS_X ? 0 : WIRE_Y_FLAT_BOTH - WIRE_X_FLAT_BOTH);

   345

   346

	if ((length % 2) && num == length) {

   347

		/* Draw the "short" wire on the southern end of the bridge

   348

		 * only needed if the length of the bridge is odd */

   349

		sss = &CatenarySpriteData[WIRE_X_FLAT_BOTH + offset];

   350

	} else {

   351

		/* Draw "long" wires on all other tiles of the bridge (one pylon every two tiles) */

   352

		sss = &CatenarySpriteData[WIRE_X_FLAT_SW + (num % 2) + offset];

   353

	}

   354

   355

	height = GetBridgeHeight(end);

   356

   357

	SpriteID img = sss->image;

   358

	SpriteID pal = PAL_NONE;

   359

	if (HASBIT(_transparent_opt, TO_BUILDINGS)) {

   360

		SETBIT(img, PALETTE_MODIFIER_TRANSPARENT);

   361

		pal = PALETTE_TO_TRANSPARENT;

   362

	}

   363

   364

	AddSortableSpriteToDraw(img, pal, ti->x + sss->x_offset, ti->y + sss->y_offset,

   365

		sss->x_size, sss->y_size, sss->z_size, height + sss->z_offset

   366

	);

   367

   368

	/* Finished with wires, draw pylons */

   369

	/* every other tile needs a pylon on the northern end */

   370

	if (num % 2) {

   371

		if (axis == AXIS_X) {

   372

			img = pylons_bridge[0 + HASBIT(tlg, 0)];

   373

			if (HASBIT(_transparent_opt, TO_BUILDINGS)) SETBIT(img, PALETTE_MODIFIER_TRANSPARENT);

   374

			AddSortableSpriteToDraw(img, pal, ti->x, ti->y + 4 + 8 * HASBIT(tlg, 0), 1, 1, 10, height);

   375

		} else {

   376

			img = pylons_bridge[2 + HASBIT(tlg, 1)];

   377

			if (HASBIT(_transparent_opt, TO_BUILDINGS)) SETBIT(img, PALETTE_MODIFIER_TRANSPARENT);

   378

			AddSortableSpriteToDraw(img, pal, ti->x + 4 + 8 * HASBIT(tlg, 1), ti->y, 1, 1, 10, height);

   379

		}

   380

	}

   381

   382

	/* need a pylon on the southern end of the bridge */

   383

	if (DistanceMax(ti->tile, start) == length) {

   384

		if (axis == AXIS_X) {

   385

			img = pylons_bridge[0 + HASBIT(tlg, 0)];

   386

			if (HASBIT(_transparent_opt, TO_BUILDINGS)) SETBIT(img, PALETTE_MODIFIER_TRANSPARENT);

   387

			AddSortableSpriteToDraw(img, pal, ti->x + 16, ti->y + 4 + 8 * HASBIT(tlg, 0), 1, 1, 10, height);

   388

		} else {

   389

			img = pylons_bridge[2 + HASBIT(tlg, 1)];

   390

			if (HASBIT(_transparent_opt, TO_BUILDINGS)) SETBIT(img, PALETTE_MODIFIER_TRANSPARENT);

   391

			AddSortableSpriteToDraw(img, pal, ti->x + 4 + 8 * HASBIT(tlg, 1), ti->y + 16, 1, 1, 10, height);

   392

		}

   393

	}

   394

}

   395

   396

void DrawCatenary(const TileInfo *ti)

   397

{

   398

	if (MayHaveBridgeAbove(ti->tile) && IsBridgeAbove(ti->tile)) {

   399

		TileIndex head = GetNorthernBridgeEnd(ti->tile);

   400

   401

		if (GetBridgeTransportType(head) == TRANSPORT_RAIL && GetRailType(head) == RAILTYPE_ELECTRIC) {

   402

			DrawCatenaryOnBridge(ti);

   403

		}

   404

	}

   405

	if (_patches.disable_elrails) return;

   406

   407

	switch (GetTileType(ti->tile)) {

   408

		case MP_RAILWAY:

   409

			if (IsRailDepot(ti->tile)) {

   410

				const SortableSpriteStruct *sss = &CatenarySpriteData_Depot[GetRailDepotDirection(ti->tile)];

   411

   412

				SpriteID img = sss->image;

   413

				SpriteID pal = PAL_NONE;

   414

				if (HASBIT(_transparent_opt, TO_BUILDINGS)) {

   415

					SETBIT(img, PALETTE_MODIFIER_TRANSPARENT);

   416

					pal = PALETTE_TO_TRANSPARENT;

   417

				}

   418

   419

				AddSortableSpriteToDraw(

   420

					img, pal, ti->x + sss->x_offset, ti->y + sss->y_offset,

   421

					sss->x_size, sss->y_size, sss->z_size,

   422

					GetTileMaxZ(ti->tile) + sss->z_offset

   423

				);

   424

				return;

   425

			}

   426

			break;

   427

   428

		case MP_TUNNELBRIDGE:

   429

		case MP_STREET:

   430

		case MP_STATION:

   431

			break;

   432

   433

		default: return;

   434

	}

   435

	DrawCatenaryRailway(ti);

   436

}

   437

   438

int32 SettingsDisableElrail(int32 p1)

   439

{

   440

	EngineID e_id;

   441

	Vehicle *v;

   442

	Player *p;

   443

	bool disable = (p1 != 0);

   444

   445

	/* we will now walk through all electric train engines and change their railtypes if it is the wrong one*/

   446

	const RailType old_railtype = disable ? RAILTYPE_ELECTRIC : RAILTYPE_RAIL;

   447

	const RailType new_railtype = disable ? RAILTYPE_RAIL : RAILTYPE_ELECTRIC;

   448

   449

	/* walk through all train engines */

   450

	for (e_id = 0; e_id < NUM_TRAIN_ENGINES; e_id++) {

   451

		RailVehicleInfo *rv_info = &_rail_vehicle_info[e_id];

   452

		/* if it is an electric rail engine and its railtype is the wrong one */

   453

		if (rv_info->engclass == 2 && rv_info->railtype == old_railtype) {

   454

			/* change it to the proper one */

   455

			rv_info->railtype = new_railtype;

   456

		}

   457

	}

   458

   459

	/* when disabling elrails, make sure that all existing trains can run on

   460

	*  normal rail too */

   461

	if (disable) {

   462

		FOR_ALL_VEHICLES(v) {

   463

			if (v->type == VEH_TRAIN && v->u.rail.railtype == RAILTYPE_ELECTRIC) {

   464

				/* this railroad vehicle is now compatible only with elrail,

   465

				*  so add there also normal rail compatibility */

   466

				v->u.rail.compatible_railtypes |= (1 << RAILTYPE_RAIL);

   467

				v->u.rail.railtype = RAILTYPE_RAIL;

   468

				SETBIT(v->u.rail.flags, VRF_EL_ENGINE_ALLOWED_NORMAL_RAIL);

   469

			}

   470

		}

   471

	}

   472

   473

	/* setup total power for trains */

   474

	FOR_ALL_VEHICLES(v) {

   475

		/* power is cached only for front engines */

   476

		if (v->type == VEH_TRAIN && IsFrontEngine(v)) TrainPowerChanged(v);

   477

	}

   478

   479

	FOR_ALL_PLAYERS(p) p->avail_railtypes = GetPlayerRailtypes(p->index);

   480

   481

	/* This resets the _last_built_railtype, which will be invalid for electric

   482

	* rails. It may have unintended consequences if that function is ever

   483

	* extended, though. */

   484

	ReinitGuiAfterToggleElrail(disable);

   485

	return 0;

   486

}