1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "roadveh.h"

     6

#include "ship.h"

     7

#include "news_func.h"

     8

#include "player_func.h"

     9

#include "debug.h"

    10

#include "vehicle_gui.h"

    11

#include "depot.h"

    12

#include "train.h"

    13

#include "aircraft.h"

    14

#include "cargotype.h"

    15

#include "group.h"

    16

#include "strings_func.h"

    17

#include "command_func.h"

    18

#include "vehicle_func.h"

    19

#include "functions.h"

    20

#include "variables.h"

    21

#include "autoreplace_func.h"

    22

#include "articulated_vehicles.h"

    23

    24

#include "table/strings.h"

    25

    26

    27

/*

    28

 * move the cargo from one engine to another if possible

    29

 */

    30

static void MoveVehicleCargo(Vehicle *dest, Vehicle *source)

    31

{

    32

	Vehicle *v = dest;

    33

    34

	do {

    35

		do {

    36

			if (source->cargo_type != dest->cargo_type)

    37

				continue; // cargo not compatible

    38

    39

			if (dest->cargo.Count() == dest->cargo_cap)

    40

				continue; // the destination vehicle is already full

    41

    42

			uint units_moved = min(source->cargo.Count(), dest->cargo_cap - dest->cargo.Count());

    43

			source->cargo.MoveTo(&dest->cargo, units_moved);

    44

    45

			// copy the age of the cargo

    46

			dest->day_counter  = source->day_counter;

    47

			dest->tick_counter = source->tick_counter;

    48

    49

		} while (source->cargo.Count() > 0 && (dest = dest->Next()) != NULL);

    50

		dest = v;

    51

	} while ((source = source->Next()) != NULL);

    52

    53

	/*

    54

	 * The of the train will be incorrect at this moment. This is due

    55

	 * to the fact that removing the old wagon updates the weight of

    56

	 * the complete train, which is without the weight of cargo we just

    57

	 * moved back into some (of the) new wagon(s).

    58

	 */

    59

	if (dest->type == VEH_TRAIN) TrainConsistChanged(dest->First());

    60

}

    61

    62

static bool VerifyAutoreplaceRefitForOrders(const Vehicle *v, const EngineID engine_type)

    63

{

    64

	const Order *o;

    65

	const Vehicle *u;

    66

    67

	if (v->type == VEH_TRAIN) {

    68

		u = v->First();

    69

	} else {

    70

		u = v;

    71

	}

    72

    73

	FOR_VEHICLE_ORDERS(u, o) {

    74

		if (!(o->refit_cargo < NUM_CARGO)) continue;

    75

		if (!CanRefitTo(v->engine_type, o->refit_cargo)) continue;

    76

		if (!CanRefitTo(engine_type, o->refit_cargo)) return false;

    77

	}

    78

    79

	return true;

    80

}

    81

    82

/**

    83

 * Function to find what type of cargo to refit to when autoreplacing

    84

 * @param *v Original vehicle, that is being replaced

    85

 * @param engine_type The EngineID of the vehicle that is being replaced to

    86

 * @return The cargo type to replace to

    87

 *    CT_NO_REFIT is returned if no refit is needed

    88

 *    CT_INVALID is returned when both old and new vehicle got cargo capacity and refitting the new one to the old one's cargo type isn't possible

    89

 */

    90

static CargoID GetNewCargoTypeForReplace(Vehicle *v, EngineID engine_type)

    91

{

    92

	CargoID new_cargo_type = GetEngineCargoType(engine_type);

    93

    94

	if (new_cargo_type == CT_INVALID) return CT_NO_REFIT; // Don't try to refit an engine with no cargo capacity

    95

    96

	if (v->cargo_cap != 0 && (v->cargo_type == new_cargo_type || CanRefitTo(engine_type, v->cargo_type))) {

    97

		if (VerifyAutoreplaceRefitForOrders(v, engine_type)) {

    98

			return v->cargo_type == new_cargo_type ? (CargoID)CT_NO_REFIT : v->cargo_type;

    99

		} else {

   100

			return CT_INVALID;

   101

		}

   102

	}

   103

	if (v->type != VEH_TRAIN) return CT_INVALID; // We can't refit the vehicle to carry the cargo we want

   104

   105

	/* Below this line it's safe to assume that the vehicle in question is a train */

   106

   107

	if (v->cargo_cap != 0) return CT_INVALID; // trying to replace a vehicle with cargo capacity into another one with incompatible cargo type

   108

   109

	/* the old engine didn't have cargo capacity, but the new one does

   110

	 * now we will figure out what cargo the train is carrying and refit to fit this */

   111

	v = v->First();

   112

	do {

   113

		if (v->cargo_cap == 0) continue;

   114

		/* Now we found a cargo type being carried on the train and we will see if it is possible to carry to this one */

   115

		if (v->cargo_type == new_cargo_type) return CT_NO_REFIT;

   116

		if (CanRefitTo(engine_type, v->cargo_type)) return v->cargo_type;

   117

	} while ((v = v->Next()) != NULL);

   118

	return CT_NO_REFIT; // We failed to find a cargo type on the old vehicle and we will not refit the new one

   119

}

   120

   121

/* Replaces a vehicle (used to be called autorenew)

   122

 * This function is only called from MaybeReplaceVehicle()

   123

 * Must be called with _current_player set to the owner of the vehicle

   124

 * @param w Vehicle to replace

   125

 * @param flags is the flags to use when calling DoCommand(). Mainly DC_EXEC counts

   126

 * @return value is cost of the replacement or CMD_ERROR

   127

 */

   128

static CommandCost ReplaceVehicle(Vehicle **w, byte flags, Money total_cost)

   129

{

   130

	CommandCost cost;

   131

	CommandCost sell_value;

   132

	Vehicle *old_v = *w;

   133

	const Player *p = GetPlayer(old_v->owner);

   134

	EngineID new_engine_type;

   135

	const UnitID cached_unitnumber = old_v->unitnumber;

   136

	bool new_front = false;

   137

	Vehicle *new_v = NULL;

   138

	char *vehicle_name = NULL;

   139

	CargoID replacement_cargo_type;

   140

   141

	/* Check if there is a autoreplacement set for the vehicle */

   142

	new_engine_type = EngineReplacementForPlayer(p, old_v->engine_type, old_v->group_id);

   143

	/* if not, just renew to the same type */

   144

	if (new_engine_type == INVALID_ENGINE) new_engine_type = old_v->engine_type;

   145

   146

	replacement_cargo_type = GetNewCargoTypeForReplace(old_v, new_engine_type);

   147

   148

	/* check if we can't refit to the needed type, so no replace takes place to prevent the vehicle from altering cargo type */

   149

	if (replacement_cargo_type == CT_INVALID) return CommandCost();

   150

   151

	sell_value = DoCommand(0, old_v->index, 0, DC_QUERY_COST, GetCmdSellVeh(old_v));

   152

   153

	/* We give the player a loan of the same amount as the sell value.

   154

	 * This is needed in case he needs the income from the sale to build the new vehicle.

   155

	 * We take it back if building fails or when we really sell the old engine */

   156

	SubtractMoneyFromPlayer(sell_value);

   157

   158

	cost = DoCommand(old_v->tile, new_engine_type, 3, flags, GetCmdBuildVeh(old_v));

   159

	if (CmdFailed(cost)) {

   160

		/* Take back the money we just gave the player */

   161

		sell_value.MultiplyCost(-1);

   162

		SubtractMoneyFromPlayer(sell_value);

   163

		return cost;

   164

	}

   165

   166

	if (replacement_cargo_type != CT_NO_REFIT) {

   167

		/* add refit cost */

   168

		CommandCost refit_cost = GetRefitCost(new_engine_type);

   169

		if (old_v->type == VEH_TRAIN && RailVehInfo(new_engine_type)->railveh_type == RAILVEH_MULTIHEAD) {

   170

			/* Since it's a dualheaded engine we have to pay once more because the rear end is being refitted too. */

   171

			refit_cost.AddCost(refit_cost);

   172

		}

   173

		cost.AddCost(refit_cost);

   174

	}

   175

   176

	if (flags & DC_EXEC) {

   177

		new_v = GetVehicle(_new_vehicle_id);

   178

		*w = new_v; //we changed the vehicle, so MaybeReplaceVehicle needs to work on the new one. Now we tell it what the new one is

   179

   180

		/* refit if needed */

   181

		if (replacement_cargo_type != CT_NO_REFIT) {

   182

			if (CmdFailed(DoCommand(0, new_v->index, replacement_cargo_type, DC_EXEC, GetCmdRefitVeh(new_v)))) {

   183

				/* Being here shows a failure, which most likely is in GetNewCargoTypeForReplace() or incorrect estimation costs */

   184

				error("Autoreplace failed to refit. Replace engine %d to %d and refit to cargo %d", old_v->engine_type, new_v->engine_type, replacement_cargo_type);

   185

			}

   186

		}

   187

   188

		if (new_v->type == VEH_TRAIN && HasBit(old_v->u.rail.flags, VRF_REVERSE_DIRECTION) && !IsMultiheaded(new_v) && !(new_v->Next() != NULL && IsArticulatedPart(new_v->Next()))) {

   189

			// we are autorenewing to a single engine, so we will turn it as the old one was turned as well

   190

			SetBit(new_v->u.rail.flags, VRF_REVERSE_DIRECTION);

   191

		}

   192

   193

		if (old_v->type == VEH_TRAIN && !IsFrontEngine(old_v)) {

   194

			/* this is a railcar. We need to move the car into the train

   195

			 * We add the new engine after the old one instead of replacing it. It will give the same result anyway when we

   196

			 * sell the old engine in a moment

   197

			 */

   198

			/* Get the vehicle in front of the one we move out */

   199

			Vehicle *front = old_v->Previous();

   200

			if (front == NULL) {

   201

				/* It would appear that we have the front wagon of a row of wagons without engines */

   202

				Vehicle *next = old_v->Next();

   203

				if (next != NULL) {

   204

					/* Move the chain to the new front wagon */

   205

					DoCommand(0, (new_v->index << 16) | next->index, 1, DC_EXEC, CMD_MOVE_RAIL_VEHICLE);

   206

				}

   207

			} else {

   208

				/* If the vehicle in front is the rear end of a dualheaded engine, then we need to use the one in front of that one */

   209

				if (IsRearDualheaded(front)) front = front->Previous();

   210

				/* Now we move the old one out of the train */

   211

				DoCommand(0, (INVALID_VEHICLE << 16) | old_v->index, 0, DC_EXEC, CMD_MOVE_RAIL_VEHICLE);

   212

				/* Add the new vehicle */

   213

				DoCommand(0, (front->index << 16) | new_v->index, 1, DC_EXEC, CMD_MOVE_RAIL_VEHICLE);

   214

			}

   215

		} else {

   216

			// copy/clone the orders

   217

			DoCommand(0, (old_v->index << 16) | new_v->index, old_v->IsOrderListShared() ? CO_SHARE : CO_COPY, DC_EXEC, CMD_CLONE_ORDER);

   218

			new_v->cur_order_index = old_v->cur_order_index;

   219

			ChangeVehicleViewWindow(old_v, new_v);

   220

			new_v->profit_this_year = old_v->profit_this_year;

   221

			new_v->profit_last_year = old_v->profit_last_year;

   222

			new_v->service_interval = old_v->service_interval;

   223

			DoCommand(0, old_v->group_id, new_v->index, flags, CMD_ADD_VEHICLE_GROUP);

   224

			new_front = true;

   225

			new_v->unitnumber = old_v->unitnumber; // use the same unit number

   226

			new_v->dest_tile  = old_v->dest_tile;

   227

   228

			new_v->current_order = old_v->current_order;

   229

			if (old_v->type == VEH_TRAIN && GetNextVehicle(old_v) != NULL){

   230

				Vehicle *temp_v = GetNextVehicle(old_v);

   231

   232

				// move the entire train to the new engine, excluding the old engine

   233

				if (IsMultiheaded(old_v) && temp_v == old_v->u.rail.other_multiheaded_part) {

   234

					// we got front and rear of a multiheaded engine right after each other. We should work with the next in line instead

   235

					temp_v = GetNextVehicle(temp_v);

   236

				}

   237

   238

				if (temp_v != NULL) {

   239

					DoCommand(0, (new_v->index << 16) | temp_v->index, 1, DC_EXEC, CMD_MOVE_RAIL_VEHICLE);

   240

				}

   241

			}

   242

		}

   243

		/* We are done setting up the new vehicle. Now we move the cargo from the old one to the new one */

   244

		MoveVehicleCargo(new_v->type == VEH_TRAIN ? new_v->First() : new_v, old_v);

   245

   246

		// Get the name of the old vehicle if it has a custom name.

   247

		if (old_v->name != NULL) vehicle_name = strdup(old_v->name);

   248

	} else { // flags & DC_EXEC not set

   249

		CommandCost tmp_move;

   250

   251

		if (old_v->type == VEH_TRAIN && IsFrontEngine(old_v)) {

   252

			Vehicle *next_veh = GetNextUnit(old_v); // don't try to move the rear multiheaded engine or articulated parts

   253

			if (next_veh != NULL) {

   254

				/* Verify that the wagons can be placed on the engine in question.

   255

				 * This is done by building an engine, test if the wagons can be added and then sell the test engine. */

   256

				DoCommand(old_v->tile, new_engine_type, 3, DC_EXEC, GetCmdBuildVeh(old_v));

   257

				Vehicle *temp = GetVehicle(_new_vehicle_id);

   258

				tmp_move = DoCommand(0, (temp->index << 16) | next_veh->index, 1, 0, CMD_MOVE_RAIL_VEHICLE);

   259

				DoCommand(0, temp->index, 0, DC_EXEC, GetCmdSellVeh(old_v));

   260

			}

   261

		}

   262

   263

		/* Ensure that the player will not end up having negative money while autoreplacing

   264

		 * This is needed because the only other check is done after the income from selling the old vehicle is substracted from the cost */

   265

		if (CmdFailed(tmp_move) || p->player_money < (cost.GetCost() + total_cost)) {

   266

			/* Pay back the loan */

   267

			sell_value.MultiplyCost(-1);

   268

			SubtractMoneyFromPlayer(sell_value);

   269

			return CMD_ERROR;

   270

		}

   271

	}

   272

   273

	/* Take back the money we just gave the player just before building the vehicle

   274

	 * The player will get the same amount now that the sale actually takes place */

   275

	sell_value.MultiplyCost(-1);

   276

	SubtractMoneyFromPlayer(sell_value);

   277

   278

	/* sell the engine/ find out how much you get for the old engine (income is returned as negative cost) */

   279

	cost.AddCost(DoCommand(0, old_v->index, 0, flags, GetCmdSellVeh(old_v)));

   280

   281

	if (new_front) {

   282

		/* now we assign the old unitnumber to the new vehicle */

   283

		new_v->unitnumber = cached_unitnumber;

   284

	}

   285

   286

	/* Transfer the name of the old vehicle */

   287

	if ((flags & DC_EXEC) && vehicle_name != NULL) {

   288

		_cmd_text = vehicle_name;

   289

		DoCommand(0, new_v->index, 0, DC_EXEC, CMD_NAME_VEHICLE);

   290

		free(vehicle_name);

   291

	}

   292

   293

	return cost;

   294

}

   295

   296

/** replaces a vehicle if it's set for autoreplace or is too old

   297

 * (used to be called autorenew)

   298

 * @param v The vehicle to replace

   299

 * if the vehicle is a train, v needs to be the front engine

   300

 * @param check Checks if the replace is valid. No action is done at all

   301

 * @param display_costs If set, a cost animation is shown (only if check is false)

   302

 * @return CMD_ERROR if something went wrong. Otherwise the price of the replace

   303

 */

   304

CommandCost MaybeReplaceVehicle(Vehicle *v, bool check, bool display_costs)

   305

{

   306

	Vehicle *w;

   307

	const Player *p = GetPlayer(v->owner);

   308

	byte flags = 0;

   309

	CommandCost cost, temp_cost;

   310

	bool stopped;

   311

   312

	/* Remember the length in case we need to trim train later on

   313

	 * If it's not a train, the value is unused

   314

	 * round up to the length of the tiles used for the train instead of the train length instead

   315

	 * Useful when newGRF uses custom length */

   316

	uint16 old_total_length = (v->type == VEH_TRAIN ?

   317

		(v->u.rail.cached_total_length + TILE_SIZE - 1) / TILE_SIZE * TILE_SIZE :

   318

		-1

   319

	);

   320

   321

   322

	_current_player = v->owner;

   323

   324

	assert(IsPlayerBuildableVehicleType(v));

   325

   326

	assert(v->vehstatus & VS_STOPPED); // the vehicle should have been stopped in VehicleEnteredDepotThisTick() if needed

   327

   328

	/* Remember the flag v->leave_depot_instantly because if we replace the vehicle, the vehicle holding this flag will be sold

   329

	 * If it is set, then we only stopped the vehicle to replace it (if needed) and we will need to start it again.

   330

	 * We also need to reset the flag since it should remain false except from when the vehicle enters a depot until autoreplace is handled in the same tick */

   331

	stopped = v->leave_depot_instantly;

   332

	v->leave_depot_instantly = false;

   333

   334

	for (;;) {

   335

		cost = CommandCost(EXPENSES_NEW_VEHICLES);

   336

		w = v;

   337

		do {

   338

			if (w->type == VEH_TRAIN && IsRearDualheaded(w)) {

   339

				/* we build the rear ends of multiheaded trains with the front ones */

   340

				continue;

   341

			}

   342

   343

			// check if the vehicle should be replaced

   344

			if (!w->NeedsAutorenewing(p) || // replace if engine is too old

   345

					w->max_age == 0) { // rail cars got a max age of 0

   346

				if (!EngineHasReplacementForPlayer(p, w->engine_type, w->group_id)) continue;

   347

			}

   348

   349

			/* Now replace the vehicle */

   350

			temp_cost = ReplaceVehicle(&w, flags, cost.GetCost());

   351

   352

			if (CmdFailed(temp_cost)) break; // replace failed for some reason. Leave the vehicle alone

   353

   354

			if (flags & DC_EXEC &&

   355

					(w->type != VEH_TRAIN || w->u.rail.first_engine == INVALID_ENGINE)) {

   356

				/* now we bought a new engine and sold the old one. We need to fix the

   357

				 * pointers in order to avoid pointing to the old one for trains: these

   358

				 * pointers should point to the front engine and not the cars

   359

				 */

   360

				v = w;

   361

			}

   362

			cost.AddCost(temp_cost);

   363

		} while (w->type == VEH_TRAIN && (w = GetNextVehicle(w)) != NULL);

   364

   365

		if (!(flags & DC_EXEC) && (p->player_money < (cost.GetCost() + p->engine_renew_money) || cost.GetCost() == 0)) {

   366

			if (!check && p->player_money < (cost.GetCost() + p->engine_renew_money) && ( _local_player == v->owner ) && cost.GetCost() != 0) {

   367

				StringID message;

   368

				SetDParam(0, v->unitnumber);

   369

				switch (v->type) {

   370

					case VEH_TRAIN:    message = STR_TRAIN_AUTORENEW_FAILED;       break;

   371

					case VEH_ROAD:     message = STR_ROADVEHICLE_AUTORENEW_FAILED; break;

   372

					case VEH_SHIP:     message = STR_SHIP_AUTORENEW_FAILED;        break;

   373

					case VEH_AIRCRAFT: message = STR_AIRCRAFT_AUTORENEW_FAILED;    break;

   374

						// This should never happen

   375

					default: NOT_REACHED(); message = 0; break;

   376

				}

   377

   378

				AddNewsItem(message, NM_SMALL, NF_VIEWPORT|NF_VEHICLE, NT_ADVICE, DNC_NONE, v->index, 0);

   379

			}

   380

			if (stopped) v->vehstatus &= ~VS_STOPPED;

   381

			if (display_costs) _current_player = OWNER_NONE;

   382

			return CMD_ERROR;

   383

		}

   384

   385

		if (flags & DC_EXEC) {

   386

			break; // we are done replacing since the loop ran once with DC_EXEC

   387

		} else if (check) {

   388

			/* It's a test only and we know that we can do this

   389

			 * NOTE: payment for wagon removal is NOT included in this price */

   390

			return cost;

   391

		}

   392

		// now we redo the loop, but this time we actually do stuff since we know that we can do it

   393

		flags |= DC_EXEC;

   394

	}

   395

   396

	/* If setting is on to try not to exceed the old length of the train with the replacement */

   397

	if (v->type == VEH_TRAIN && p->renew_keep_length) {

   398

		Vehicle *temp;

   399

		w = v;

   400

   401

		while (v->u.rail.cached_total_length > old_total_length) {

   402

			// the train is too long. We will remove cars one by one from the start of the train until it's short enough

   403

			while (w != NULL && RailVehInfo(w->engine_type)->railveh_type != RAILVEH_WAGON) {

   404

				w = GetNextVehicle(w);

   405

			}

   406

			if (w == NULL) {

   407

				// we failed to make the train short enough

   408

				SetDParam(0, v->unitnumber);

   409

				AddNewsItem(STR_TRAIN_TOO_LONG_AFTER_REPLACEMENT, NM_SMALL, NF_VIEWPORT | NF_VEHICLE, NT_ADVICE, DNC_NONE, v->index, 0);

   410

				break;

   411

			}

   412

			temp = w;

   413

			w = GetNextVehicle(w);

   414

			DoCommand(0, (INVALID_VEHICLE << 16) | temp->index, 0, DC_EXEC, CMD_MOVE_RAIL_VEHICLE);

   415

			MoveVehicleCargo(v, temp);

   416

			cost.AddCost(DoCommand(0, temp->index, 0, DC_EXEC, CMD_SELL_RAIL_WAGON));

   417

		}

   418

	}

   419

   420

	if (stopped) v->vehstatus &= ~VS_STOPPED;

   421

	if (display_costs) {

   422

		if (IsLocalPlayer()) ShowCostOrIncomeAnimation(v->x_pos, v->y_pos, v->z_pos, cost.GetCost());

   423

		_current_player = OWNER_NONE;

   424

	}

   425

	return cost;

   426

}