tron@2186: /* $Id$ */
tron@2186: 
rubidium@9111: /** @file engine.cpp Base for all engine handling. */
belugas@6125: 
truelight@0: #include "stdafx.h"
Darkvater@1891: #include "openttd.h"
tron@1299: #include "debug.h"
rubidium@10208: #include "company_base.h"
rubidium@10208: #include "company_func.h"
rubidium@8116: #include "command_func.h"
rubidium@8763: #include "news_func.h"
truelight@0: #include "saveload.h"
tron@2159: #include "variables.h"
peter1138@2708: #include "train.h"
Darkvater@5854: #include "aircraft.h"
belugas@3866: #include "newgrf_cargo.h"
peter1138@9070: #include "newgrf_engine.h"
rubidium@6643: #include "group.h"
rubidium@8114: #include "strings_func.h"
rubidium@8225: #include "gfx_func.h"
rubidium@8131: #include "functions.h"
rubidium@8131: #include "window_func.h"
rubidium@8140: #include "date_func.h"
rubidium@8212: #include "autoreplace_gui.h"
rubidium@8214: #include "string_func.h"
rubidium@8270: #include "settings_type.h"
smatz@8847: #include "oldpool_func.h"
peter1138@9070: #include "core/alloc_func.hpp"
smatz@9211: #include "vehicle_func.h"
peter1138@9417: #include <map>
truelight@0: 
rubidium@8264: #include "table/strings.h"
rubidium@8264: #include "table/engines.h"
rubidium@8264: 
peter1138@9070: DEFINE_OLD_POOL_GENERIC(Engine, Engine)
tron@2763: 
truelight@0: enum {
truelight@4300: 	YEAR_ENGINE_AGING_STOPS = 2050,
truelight@4300: };
truelight@4300: 
truelight@0: 
peter1138@9070: /** Number of engines of each vehicle type in original engine data */
peter1138@9070: const uint8 _engine_counts[4] = {
peter1138@9070: 	lengthof(_orig_rail_vehicle_info),
peter1138@9070: 	lengthof(_orig_road_vehicle_info),
peter1138@9070: 	lengthof(_orig_ship_vehicle_info),
peter1138@9070: 	lengthof(_orig_aircraft_vehicle_info),
peter1138@9070: };
peter1138@9070: 
peter1138@9070: /** Offset of the first engine of each vehicle type in original engine data */
peter1138@9070: const uint8 _engine_offsets[4] = {
peter1138@9070: 	0,
peter1138@9070: 	lengthof(_orig_rail_vehicle_info),
peter1138@9070: 	lengthof(_orig_rail_vehicle_info) + lengthof(_orig_road_vehicle_info),
peter1138@9070: 	lengthof(_orig_rail_vehicle_info) + lengthof(_orig_road_vehicle_info) + lengthof(_orig_ship_vehicle_info),
peter1138@9070: };
peter1138@9070: 
peter1138@9070: Engine::Engine() :
peter1138@9070: 	name(NULL),
peter1138@9070: 	overrides_count(0),
peter1138@9070: 	overrides(NULL)
peter1138@9070: {
peter1138@9070: }
peter1138@9070: 
peter1138@9070: Engine::Engine(VehicleType type, EngineID base)
peter1138@9070: {
peter1138@9070: 	this->type = type;
peter1138@9070: 	this->internal_id = base;
peter1138@9070: 	this->list_position = base;
peter1138@9070: 
peter1138@9070: 	/* Check if this base engine is within the original engine data range */
peter1138@9070: 	if (base >= _engine_counts[type]) {
peter1138@9070: 		/* Mark engine as valid anyway */
peter1138@9070: 		this->info.climates = 0x80;
peter1138@9070: 		return;
peter1138@9070: 	}
peter1138@9070: 
peter1138@9070: 	/* Copy the original engine info for this slot */
peter1138@9070: 	this->info = _orig_engine_info[_engine_offsets[type] + base];
peter1138@9070: 
peter1138@9070: 	/* Copy the original engine data for this slot */
peter1138@9070: 	switch (type) {
peter1138@9070: 		default: NOT_REACHED();
peter1138@9070: 
peter1138@9070: 		case VEH_TRAIN:
peter1138@9070: 			this->u.rail = _orig_rail_vehicle_info[base];
peter1138@9070: 			this->image_index = this->u.rail.image_index;
peter1138@9070: 			this->info.string_id = STR_8000_KIRBY_PAUL_TANK_STEAM + base;
peter1138@9070: 			break;
peter1138@9070: 
peter1138@9070: 		case VEH_ROAD:
peter1138@9070: 			this->u.road = _orig_road_vehicle_info[base];
peter1138@9070: 			this->image_index = this->u.road.image_index;
peter1138@9070: 			this->info.string_id = STR_8074_MPS_REGAL_BUS + base;
peter1138@9070: 			break;
peter1138@9070: 
peter1138@9070: 		case VEH_SHIP:
peter1138@9070: 			this->u.ship = _orig_ship_vehicle_info[base];
peter1138@9070: 			this->image_index = this->u.ship.image_index;
peter1138@9070: 			this->info.string_id = STR_80CC_MPS_OIL_TANKER + base;
peter1138@9070: 			break;
peter1138@9070: 
peter1138@9070: 		case VEH_AIRCRAFT:
peter1138@9070: 			this->u.air = _orig_aircraft_vehicle_info[base];
peter1138@9070: 			this->image_index = this->u.air.image_index;
peter1138@9070: 			this->info.string_id = STR_80D7_SAMPSON_U52 + base;
peter1138@9070: 			break;
peter1138@9070: 	}
peter1138@9070: }
peter1138@9070: 
peter1138@9070: Engine::~Engine()
peter1138@9070: {
peter1138@9070: 	UnloadWagonOverrides(this);
peter1138@9070: 	free(this->name);
peter1138@9070: }
peter1138@9070: 
rubidium@10207: /** Sets cached values in Company::num_vehicles and Group::num_vehicles
smatz@9220:  */
smatz@9211: void SetCachedEngineCounts()
smatz@9211: {
smatz@9211: 	uint engines = GetEnginePoolSize();
smatz@9211: 
rubidium@10207: 	/* Set up the engine count for all companies */
rubidium@10207: 	Company *c;
rubidium@10207: 	FOR_ALL_COMPANIES(c) {
rubidium@10207: 		free(c->num_engines);
rubidium@10207: 		c->num_engines = CallocT<EngineID>(engines);
smatz@9211: 	}
smatz@9211: 
smatz@9211: 	/* Recalculate */
smatz@9211: 	Group *g;
smatz@9211: 	FOR_ALL_GROUPS(g) {
smatz@9211: 		free(g->num_engines);
smatz@9211: 		g->num_engines = CallocT<EngineID>(engines);
smatz@9211: 	}
smatz@9211: 
smatz@9211: 	const Vehicle *v;
smatz@9211: 	FOR_ALL_VEHICLES(v) {
smatz@9211: 		if (!IsEngineCountable(v)) continue;
smatz@9211: 
smatz@9211: 		assert(v->engine_type < engines);
smatz@9211: 
rubidium@10207: 		GetCompany(v->owner)->num_engines[v->engine_type]++;
smatz@9211: 
smatz@9211: 		if (v->group_id == DEFAULT_GROUP) continue;
smatz@9211: 
smatz@9211: 		g = GetGroup(v->group_id);
smatz@9211: 		assert(v->type == g->vehicle_type);
smatz@9211: 		assert(v->owner == g->owner);
smatz@9211: 
smatz@9211: 		g->num_engines[v->engine_type]++;
smatz@9211: 	}
smatz@9211: }
smatz@9211: 
peter1138@8221: void SetupEngines()
peter1138@8221: {
peter1138@9070: 	_Engine_pool.CleanPool();
peter1138@9070: 	_Engine_pool.AddBlockToPool();
peter1138@8221: 
peter1138@9070: 	for (uint i = 0; i < lengthof(_orig_rail_vehicle_info); i++) new Engine(VEH_TRAIN, i);
peter1138@9070: 	for (uint i = 0; i < lengthof(_orig_road_vehicle_info); i++) new Engine(VEH_ROAD, i);
peter1138@9070: 	for (uint i = 0; i < lengthof(_orig_ship_vehicle_info); i++) new Engine(VEH_SHIP, i);
peter1138@9070: 	for (uint i = 0; i < lengthof(_orig_aircraft_vehicle_info); i++) new Engine(VEH_AIRCRAFT, i);
peter1138@8221: }
peter1138@8221: 
peter1138@8221: 
tron@2477: void ShowEnginePreviewWindow(EngineID engine);
truelight@0: 
rubidium@6247: void DeleteCustomEngineNames()
truelight@0: {
peter1138@8258: 	Engine *e;
peter1138@8258: 	FOR_ALL_ENGINES(e) {
peter1138@8258: 		free(e->name);
peter1138@8258: 		e->name = NULL;
truelight@0: 	}
truelight@0: 
truelight@0: 	_vehicle_design_names &= ~1;
truelight@0: }
truelight@0: 
rubidium@6247: void LoadCustomEngineNames()
truelight@0: {
rubidium@4432: 	/* XXX: not done */
Darkvater@5380: 	DEBUG(misc, 1, "LoadCustomEngineNames: not done");
truelight@0: }
truelight@0: 
peter1138@9468: /* Determine if an engine type is a wagon (and not a loco) */
peter1138@9468: static bool IsWagon(EngineID index)
peter1138@9468: {
peter1138@9468: 	const Engine *e = GetEngine(index);
peter1138@9468: 	return e->type == VEH_TRAIN && e->u.rail.railveh_type == RAILVEH_WAGON;
peter1138@9468: }
peter1138@9468: 
truelight@0: static void CalcEngineReliability(Engine *e)
truelight@0: {
truelight@0: 	uint age = e->age;
truelight@0: 
peter1138@6606: 	/* Check for early retirement */
rubidium@10207: 	if (e->company_avail != 0 && !_settings_game.vehicle.never_expire_vehicles) {
peter1138@9070: 		int retire_early = e->info.retire_early;
frosch@8509: 		uint retire_early_max_age = max(0, e->duration_phase_1 + e->duration_phase_2 - retire_early * 12);
frosch@8509: 		if (retire_early != 0 && age >= retire_early_max_age) {
peter1138@6606: 			/* Early retirement is enabled and we're past the date... */
rubidium@10207: 			e->company_avail = 0;
peter1138@6606: 			AddRemoveEngineFromAutoreplaceAndBuildWindows(e->type);
peter1138@6606: 		}
peter1138@6606: 	}
peter1138@6606: 
truelight@0: 	if (age < e->duration_phase_1) {
truelight@0: 		uint start = e->reliability_start;
truelight@0: 		e->reliability = age * (e->reliability_max - start) / e->duration_phase_1 + start;
rubidium@9413: 	} else if ((age -= e->duration_phase_1) < e->duration_phase_2 || _settings_game.vehicle.never_expire_vehicles) {
bjarni@4767: 		/* We are at the peak of this engines life. It will have max reliability.
bjarni@4767: 		 * This is also true if the engines never expire. They will not go bad over time */
truelight@0: 		e->reliability = e->reliability_max;
truelight@0: 	} else if ((age -= e->duration_phase_2) < e->duration_phase_3) {
truelight@0: 		uint max = e->reliability_max;
truelight@0: 		e->reliability = (int)age * (int)(e->reliability_final - max) / e->duration_phase_3 + max;
truelight@0: 	} else {
bjarni@4767: 		/* time's up for this engine.
bjarni@4767: 		 * We will now completely retire this design */
rubidium@10207: 		e->company_avail = 0;
truelight@0: 		e->reliability = e->reliability_final;
bjarni@5944: 		/* Kick this engine out of the lists */
bjarni@5944: 		AddRemoveEngineFromAutoreplaceAndBuildWindows(e->type);
truelight@0: 	}
bjarni@4767: 	InvalidateWindowClasses(WC_BUILD_VEHICLE); // Update to show the new reliability
bjarni@5944: 	InvalidateWindowClasses(WC_REPLACE_VEHICLE);
truelight@0: }
truelight@0: 
rubidium@6247: void StartupEngines()
truelight@0: {
truelight@0: 	Engine *e;
truelight@4300: 	/* Aging of vehicles stops, so account for that when starting late */
rubidium@4322: 	const Date aging_date = min(_date, ConvertYMDToDate(YEAR_ENGINE_AGING_STOPS, 0, 1));
truelight@0: 
peter1138@9070: 	FOR_ALL_ENGINES(e) {
peter1138@9070: 		const EngineInfo *ei = &e->info;
tron@2242: 		uint32 r;
truelight@0: 
truelight@0: 		e->age = 0;
truelight@0: 		e->flags = 0;
rubidium@10207: 		e->company_avail = 0;
truelight@193: 
belugas@6125: 		/* The magic value of 729 days below comes from the NewGRF spec. If the
belugas@6125: 		 * base intro date is before 1922 then the random number of days is not
belugas@6125: 		 * added. */
truelight@0: 		r = Random();
rubidium@4297: 		e->intro_date = ei->base_intro <= ConvertYMDToDate(1922, 0, 1) ? ei->base_intro : (Date)GB(r, 0, 9) + ei->base_intro;
truelight@0: 		if (e->intro_date <= _date) {
truelight@4300: 			e->age = (aging_date - e->intro_date) >> 5;
rubidium@10207: 			e->company_avail = (CompanyMask)-1;
truelight@0: 			e->flags |= ENGINE_AVAILABLE;
truelight@0: 		}
truelight@0: 
tron@2140: 		e->reliability_start = GB(r, 16, 14) + 0x7AE0;
truelight@0: 		r = Random();
tron@2140: 		e->reliability_max   = GB(r,  0, 14) + 0xBFFF;
tron@2140: 		e->reliability_final = GB(r, 16, 14) + 0x3FFF;
truelight@0: 
truelight@0: 		r = Random();
tron@2140: 		e->duration_phase_1 = GB(r, 0, 5) + 7;
tron@2140: 		e->duration_phase_2 = GB(r, 5, 4) + ei->base_life * 12 - 96;
tron@2140: 		e->duration_phase_3 = GB(r, 9, 7) + 120;
truelight@0: 
peter1138@9468: 		e->reliability_spd_dec = ei->decay_speed << 2;
truelight@0: 
peter1138@9468: 		if (IsWagon(e->index)) {
truelight@0: 			e->age = 0xFFFF;
truelight@0: 		} else {
truelight@0: 			CalcEngineReliability(e);
truelight@0: 		}
truelight@0: 
rubidium@9413: 		e->lifelength = ei->lifelength + _settings_game.vehicle.extend_vehicle_life;
truelight@0: 
belugas@6125: 		/* prevent certain engines from ever appearing. */
rubidium@9413: 		if (!HasBit(ei->climates, _settings_game.game_creation.landscape)) {
truelight@0: 			e->flags |= ENGINE_AVAILABLE;
rubidium@10207: 			e->company_avail = 0;
truelight@0: 		}
truelight@0: 	}
peter1138@10023: 
peter1138@10023: 	/* Update the bitmasks for the vehicle lists */
rubidium@10207: 	Company *c;
rubidium@10207: 	FOR_ALL_COMPANIES(c) {
rubidium@10207: 		c->avail_railtypes = GetCompanyRailtypes(c->index);
rubidium@10207: 		c->avail_roadtypes = GetCompanyRoadtypes(c->index);
peter1138@10023: 	}
truelight@0: }
truelight@0: 
rubidium@10207: static void AcceptEnginePreview(EngineID eid, CompanyID company)
truelight@0: {
tron@5823: 	Engine *e = GetEngine(eid);
rubidium@10207: 	Company *c = GetCompany(company);
truelight@0: 
rubidium@10207: 	SetBit(e->company_avail, company);
rubidium@6259: 	if (e->type == VEH_TRAIN) {
tron@5823: 		const RailVehicleInfo *rvi = RailVehInfo(eid);
tron@5823: 
tron@5823: 		assert(rvi->railtype < RAILTYPE_END);
rubidium@10207: 		SetBit(c->avail_railtypes, rvi->railtype);
rubidium@6685: 	} else if (e->type == VEH_ROAD) {
rubidium@10207: 		SetBit(c->avail_roadtypes, HasBit(EngInfo(eid)->misc_flags, EF_ROAD_TRAM) ? ROADTYPE_TRAM : ROADTYPE_ROAD);
tron@5823: 	}
truelight@0: 
rubidium@10207: 	e->preview_company_rank = 0xFF;
rubidium@10207: 	if (company == _local_company) {
bjarni@5944: 		AddRemoveEngineFromAutoreplaceAndBuildWindows(e->type);
bjarni@4766: 	}
truelight@0: }
truelight@0: 
rubidium@10207: static CompanyID GetBestCompany(uint8 pp)
Darkvater@1786: {
rubidium@10207: 	const Company *c;
Darkvater@1786: 	int32 best_hist;
rubidium@10207: 	CompanyID best_company;
Darkvater@1786: 	uint mask = 0;
Darkvater@1786: 
smatz@8572: 	do {
Darkvater@1786: 		best_hist = -1;
rubidium@10207: 		best_company = INVALID_COMPANY;
rubidium@10207: 		FOR_ALL_COMPANIES(c) {
rubidium@10207: 			if (c->block_preview == 0 && !HasBit(mask, c->index) &&
rubidium@10207: 					c->old_economy[0].performance_history > best_hist) {
rubidium@10207: 				best_hist = c->old_economy[0].performance_history;
rubidium@10207: 				best_company = c->index;
Darkvater@1786: 			}
Darkvater@1786: 		}
Darkvater@1786: 
rubidium@10207: 		if (best_company == INVALID_COMPANY) return INVALID_COMPANY;
Darkvater@1786: 
rubidium@10207: 		SetBit(mask, best_company);
smatz@8633: 	} while (--pp != 0);
Darkvater@1786: 
rubidium@10207: 	return best_company;
Darkvater@1786: }
Darkvater@1786: 
rubidium@6247: void EnginesDailyLoop()
truelight@0: {
truelight@4300: 	if (_cur_year >= YEAR_ENGINE_AGING_STOPS) return;
truelight@0: 
peter1138@9070: 	Engine *e;
peter1138@9070: 	FOR_ALL_ENGINES(e) {
peter1138@9070: 		EngineID i = e->index;
maedhros@6174: 		if (e->flags & ENGINE_EXCLUSIVE_PREVIEW) {
maedhros@6174: 			if (e->flags & ENGINE_OFFER_WINDOW_OPEN) {
rubidium@10207: 				if (e->preview_company_rank != 0xFF && !--e->preview_wait) {
maedhros@6174: 					e->flags &= ~ENGINE_OFFER_WINDOW_OPEN;
truelight@0: 					DeleteWindowById(WC_ENGINE_PREVIEW, i);
rubidium@10207: 					e->preview_company_rank++;
truelight@193: 				}
rubidium@10207: 			} else if (e->preview_company_rank != 0xFF) {
rubidium@10207: 				CompanyID best_company = GetBestCompany(e->preview_company_rank);
Darkvater@1786: 
rubidium@10207: 				if (best_company == INVALID_COMPANY) {
rubidium@10207: 					e->preview_company_rank = 0xFF;
Darkvater@1786: 					continue;
Darkvater@1786: 				}
truelight@193: 
rubidium@10207: 				if (!IsHumanCompany(best_company)) {
Darkvater@1786: 					/* XXX - TTDBUG: TTD has a bug here ???? */
rubidium@10207: 					AcceptEnginePreview(i, best_company);
truelight@0: 				} else {
maedhros@6174: 					e->flags |= ENGINE_OFFER_WINDOW_OPEN;
truelight@0: 					e->preview_wait = 20;
rubidium@10207: 					if (IsInteractiveCompany(best_company)) ShowEnginePreviewWindow(i);
truelight@0: 				}
truelight@0: 			}
truelight@0: 		}
truelight@0: 	}
truelight@0: }
truelight@0: 
rubidium@10207: /** Accept an engine prototype. XXX - it is possible that the top-company
Darkvater@1786:  * changes while you are waiting to accept the offer? Then it becomes invalid
tron@3491:  * @param tile unused
belugas@6481:  * @param flags operation to perfom
Darkvater@1786:  * @param p1 engine-prototype offered
Darkvater@1786:  * @param p2 unused
Darkvater@1786:  */
rubidium@6943: CommandCost CmdWantEnginePreview(TileIndex tile, uint32 flags, uint32 p1, uint32 p2)
truelight@0: {
Darkvater@1786: 	Engine *e;
tron@2951: 
Darkvater@1786: 	if (!IsEngineIndex(p1)) return CMD_ERROR;
tron@1926: 	e = GetEngine(p1);
rubidium@10207: 	if (GetBestCompany(e->preview_company_rank) != _current_company) return CMD_ERROR;
Darkvater@1786: 
rubidium@10207: 	if (flags & DC_EXEC) AcceptEnginePreview(p1, _current_company);
Darkvater@1786: 
rubidium@6950: 	return CommandCost();
truelight@0: }
truelight@0: 
rubidium@9199: StringID GetEngineCategoryName(EngineID engine);
rubidium@9199: 
tron@410: static void NewVehicleAvailable(Engine *e)
truelight@0: {
truelight@0: 	Vehicle *v;
rubidium@10207: 	Company *c;
peter1138@9070: 	EngineID index = e->index;
truelight@0: 
rubidium@10207: 	/* In case the company didn't build the vehicle during the intro period,
rubidium@10207: 	 * prevent that company from getting future intro periods for a while. */
maedhros@6174: 	if (e->flags & ENGINE_EXCLUSIVE_PREVIEW) {
rubidium@10207: 		FOR_ALL_COMPANIES(c) {
rubidium@10207: 			uint block_preview = c->block_preview;
truelight@919: 
rubidium@10207: 			if (!HasBit(e->company_avail, c->index)) continue;
truelight@193: 
truelight@919: 			/* We assume the user did NOT build it.. prove me wrong ;) */
rubidium@10207: 			c->block_preview = 20;
truelight@919: 
truelight@919: 			FOR_ALL_VEHICLES(v) {
rubidium@6259: 				if (v->type == VEH_TRAIN || v->type == VEH_ROAD || v->type == VEH_SHIP ||
rubidium@6259: 						(v->type == VEH_AIRCRAFT && IsNormalAircraft(v))) {
rubidium@10207: 					if (v->owner == c->index && v->engine_type == index) {
truelight@919: 						/* The user did prove me wrong, so restore old value */
rubidium@10207: 						c->block_preview = block_preview;
truelight@919: 						break;
truelight@919: 					}
truelight@0: 				}
truelight@0: 			}
truelight@0: 		}
truelight@0: 	}
truelight@0: 
maedhros@6174: 	e->flags = (e->flags & ~ENGINE_EXCLUSIVE_PREVIEW) | ENGINE_AVAILABLE;
bjarni@5944: 	AddRemoveEngineFromAutoreplaceAndBuildWindows(e->type);
dominik@114: 
rubidium@10207: 	/* Now available for all companies */
rubidium@10207: 	e->company_avail = (CompanyMask)-1;
dominik@114: 
belugas@6125: 	/* Do not introduce new rail wagons */
tron@2639: 	if (IsWagon(index)) return;
dominik@114: 
peter1138@8220: 	if (e->type == VEH_TRAIN) {
belugas@6125: 		/* maybe make another rail type available */
peter1138@9070: 		RailType railtype = e->u.rail.railtype;
tron@5823: 		assert(railtype < RAILTYPE_END);
rubidium@10207: 		FOR_ALL_COMPANIES(c) SetBit(c->avail_railtypes, railtype);
peter1138@8220: 	} else if (e->type == VEH_ROAD) {
rubidium@6686: 		/* maybe make another road type available */
rubidium@10207: 		FOR_ALL_COMPANIES(c) SetBit(c->avail_roadtypes, HasBit(e->info.misc_flags, EF_ROAD_TRAM) ? ROADTYPE_TRAM : ROADTYPE_ROAD);
rubidium@6686: 	}
rubidium@9199: 
rubidium@9199: 	SetDParam(0, GetEngineCategoryName(index));
rubidium@9199: 	SetDParam(1, index);
rubidium@9234: 	AddNewsItem(STR_NEW_VEHICLE_NOW_AVAILABLE_WITH_TYPE, NS_NEW_VEHICLES, index, 0);
truelight@0: }
truelight@0: 
rubidium@6247: void EnginesMonthlyLoop()
truelight@0: {
truelight@4300: 	if (_cur_year < YEAR_ENGINE_AGING_STOPS) {
peter1138@8401: 		Engine *e;
peter1138@8401: 		FOR_ALL_ENGINES(e) {
belugas@6125: 			/* Age the vehicle */
tron@2242: 			if (e->flags & ENGINE_AVAILABLE && e->age != 0xFFFF) {
truelight@0: 				e->age++;
truelight@0: 				CalcEngineReliability(e);
truelight@0: 			}
truelight@0: 
peter1138@2990: 			if (!(e->flags & ENGINE_AVAILABLE) && _date >= (e->intro_date + 365)) {
rubidium@10207: 				/* Introduce it to all companies */
truelight@0: 				NewVehicleAvailable(e);
maedhros@6174: 			} else if (!(e->flags & (ENGINE_AVAILABLE|ENGINE_EXCLUSIVE_PREVIEW)) && _date >= e->intro_date) {
rubidium@10207: 				/* Introduction date has passed.. show introducing dialog to one companies. */
maedhros@6174: 				e->flags |= ENGINE_EXCLUSIVE_PREVIEW;
dominik@257: 
belugas@6125: 				/* Do not introduce new rail wagons */
peter1138@9070: 				if (!IsWagon(e->index))
rubidium@10207: 					e->preview_company_rank = 1; // Give to the company with the highest rating.
truelight@0: 			}
truelight@0: 		}
truelight@0: 	}
truelight@0: }
truelight@0: 
peter1138@7097: static bool IsUniqueEngineName(const char *name)
peter1138@7097: {
peter1138@7097: 	char buf[512];
peter1138@7097: 
peter1138@9070: 	const Engine *e;
peter1138@9070: 	FOR_ALL_ENGINES(e) {
peter1138@9070: 		SetDParam(0, e->index);
peter1138@7097: 		GetString(buf, STR_ENGINE_NAME, lastof(buf));
peter1138@7097: 		if (strcmp(buf, name) == 0) return false;
peter1138@7097: 	}
peter1138@7097: 
peter1138@7097: 	return true;
peter1138@7097: }
peter1138@7097: 
Darkvater@1786: /** Rename an engine.
tron@3491:  * @param tile unused
belugas@6481:  * @param flags operation to perfom
Darkvater@1786:  * @param p1 engine ID to rename
Darkvater@1786:  * @param p2 unused
Darkvater@1786:  */
rubidium@6943: CommandCost CmdRenameEngine(TileIndex tile, uint32 flags, uint32 p1, uint32 p2)
truelight@0: {
rubidium@9914: 	if (!IsEngineIndex(p1)) return CMD_ERROR;
Darkvater@1786: 
smatz@10148: 	bool reset = StrEmpty(_cmd_text);
smatz@10148: 
smatz@10148: 	if (!reset) {
smatz@10148: 		if (strlen(_cmd_text) >= MAX_LENGTH_ENGINE_NAME_BYTES) return CMD_ERROR;
smatz@10148: 		if (!IsUniqueEngineName(_cmd_text)) return_cmd_error(STR_NAME_MUST_BE_UNIQUE);
smatz@10148: 	}
peter1138@7097: 
truelight@0: 	if (flags & DC_EXEC) {
peter1138@8258: 		Engine *e = GetEngine(p1);
peter1138@8258: 		free(e->name);
smatz@10148: 
smatz@10148: 		if (reset) {
smatz@10148: 			e->name = NULL;
smatz@10148: 			/* if we removed the last custom name, disable the 'Save custom names' button */
smatz@10148: 			_vehicle_design_names &= ~1;
smatz@10148: 			FOR_ALL_ENGINES(e) {
smatz@10148: 				if (e->name != NULL) {
smatz@10148: 					_vehicle_design_names |= 1;
smatz@10148: 					break;
smatz@10148: 				}
smatz@10148: 			}
smatz@10148: 		} else {
smatz@10148: 			e->name = strdup(_cmd_text);
smatz@10148: 			_vehicle_design_names |= 3;
smatz@10148: 		}
smatz@10148: 
truelight@0: 		MarkWholeScreenDirty();
truelight@0: 	}
truelight@0: 
rubidium@6950: 	return CommandCost();
truelight@0: }
truelight@0: 
truelight@0: 
peter1138@8479: /** Check if an engine is buildable.
rubidium@10207:  * @param engine  index of the engine to check.
rubidium@10207:  * @param type    the type the engine should be.
rubidium@10207:  * @param company index of the company.
peter1138@8479:  * @return True if an engine is valid, of the specified type, and buildable by
rubidium@10207:  *              the given company.
peter1138@2848:  */
rubidium@10207: bool IsEngineBuildable(EngineID engine, VehicleType type, CompanyID company)
peter1138@2848: {
belugas@6125: 	/* check if it's an engine that is in the engine array */
peter1138@2848: 	if (!IsEngineIndex(engine)) return false;
peter1138@2848: 
peter1138@8479: 	const Engine *e = GetEngine(engine);
peter1138@2848: 
belugas@6125: 	/* check if it's an engine of specified type */
peter1138@2848: 	if (e->type != type) return false;
peter1138@2848: 
belugas@6125: 	/* check if it's available */
rubidium@10207: 	if (!HasBit(e->company_avail, company)) return false;
peter1138@2848: 
peter1138@6530: 	if (type == VEH_TRAIN) {
rubidium@10207: 		/* Check if the rail type is available to this company */
rubidium@10207: 		const Company *c = GetCompany(company);
rubidium@10207: 		if (!HasBit(c->avail_railtypes, RailVehInfo(engine)->railtype)) return false;
peter1138@6530: 	}
peter1138@6530: 
peter1138@2848: 	return true;
peter1138@2848: }
peter1138@2848: 
bjarni@6505: /** Get the default cargo type for a certain engine type
bjarni@6505:  * @param engine The ID to get the cargo for
bjarni@6505:  * @return The cargo type. CT_INVALID means no cargo capacity
bjarni@6505:  */
bjarni@6505: CargoID GetEngineCargoType(EngineID engine)
bjarni@6505: {
bjarni@6505: 	assert(IsEngineIndex(engine));
bjarni@6505: 
bjarni@6505: 	switch (GetEngine(engine)->type) {
bjarni@6505: 		case VEH_TRAIN:
bjarni@6505: 			if (RailVehInfo(engine)->capacity == 0) return CT_INVALID;
bjarni@6505: 			return RailVehInfo(engine)->cargo_type;
bjarni@6505: 
bjarni@6505: 		case VEH_ROAD:
bjarni@6505: 			if (RoadVehInfo(engine)->capacity == 0) return CT_INVALID;
bjarni@6505: 			return RoadVehInfo(engine)->cargo_type;
bjarni@6505: 
bjarni@6505: 		case VEH_SHIP:
bjarni@6505: 			if (ShipVehInfo(engine)->capacity == 0) return CT_INVALID;
bjarni@6505: 			return ShipVehInfo(engine)->cargo_type;
bjarni@6505: 
bjarni@6505: 		case VEH_AIRCRAFT:
bjarni@6505: 			/* all aircraft starts as passenger planes with cargo capacity */
bjarni@6505: 			return CT_PASSENGERS;
bjarni@6505: 
bjarni@6505: 		default: NOT_REACHED(); return CT_INVALID;
bjarni@6505: 	}
bjarni@6505: }
bjarni@6505: 
Darkvater@1881: static const SaveLoad _engine_desc[] = {
rubidium@4344: 	SLE_CONDVAR(Engine, intro_date,          SLE_FILE_U16 | SLE_VAR_I32,  0,  30),
rubidium@4344: 	SLE_CONDVAR(Engine, intro_date,          SLE_INT32,                  31, SL_MAX_VERSION),
rubidium@4344: 	SLE_CONDVAR(Engine, age,                 SLE_FILE_U16 | SLE_VAR_I32,  0,  30),
rubidium@4344: 	SLE_CONDVAR(Engine, age,                 SLE_INT32,                  31, SL_MAX_VERSION),
rubidium@4344: 	    SLE_VAR(Engine, reliability,         SLE_UINT16),
rubidium@4344: 	    SLE_VAR(Engine, reliability_spd_dec, SLE_UINT16),
rubidium@4344: 	    SLE_VAR(Engine, reliability_start,   SLE_UINT16),
rubidium@4344: 	    SLE_VAR(Engine, reliability_max,     SLE_UINT16),
rubidium@4344: 	    SLE_VAR(Engine, reliability_final,   SLE_UINT16),
rubidium@4344: 	    SLE_VAR(Engine, duration_phase_1,    SLE_UINT16),
rubidium@4344: 	    SLE_VAR(Engine, duration_phase_2,    SLE_UINT16),
rubidium@4344: 	    SLE_VAR(Engine, duration_phase_3,    SLE_UINT16),
truelight@0: 
rubidium@4344: 	    SLE_VAR(Engine, lifelength,          SLE_UINT8),
rubidium@4344: 	    SLE_VAR(Engine, flags,               SLE_UINT8),
rubidium@10207: 	    SLE_VAR(Engine, preview_company_rank,SLE_UINT8),
rubidium@4344: 	    SLE_VAR(Engine, preview_wait,        SLE_UINT8),
tron@5823: 	SLE_CONDNULL(1, 0, 44),
rubidium@10207: 	    SLE_VAR(Engine, company_avail,       SLE_UINT8),
peter1138@8258: 	SLE_CONDSTR(Engine, name,                SLE_STR, 0,                 84, SL_MAX_VERSION),
truelight@0: 
belugas@6125: 	/* reserve extra space in savegame here. (currently 16 bytes) */
Darkvater@3222: 	SLE_CONDNULL(16, 2, SL_MAX_VERSION),
truelight@0: 
truelight@0: 	SLE_END()
truelight@0: };
truelight@0: 
peter1138@9070: static std::map<EngineID, Engine> _temp_engine;
peter1138@9070: 
peter1138@9070: Engine *GetTempDataEngine(EngineID index)
peter1138@9070: {
peter1138@9070: 	return &_temp_engine[index];
peter1138@9070: }
peter1138@9070: 
rubidium@6247: static void Save_ENGN()
truelight@0: {
peter1138@9070: 	Engine *e;
peter1138@9070: 	FOR_ALL_ENGINES(e) {
peter1138@9070: 		SlSetArrayIndex(e->index);
peter1138@9070: 		SlObject(e, _engine_desc);
truelight@0: 	}
truelight@0: }
truelight@0: 
rubidium@6247: static void Load_ENGN()
truelight@0: {
peter1138@9070: 	/* As engine data is loaded before engines are initialized we need to load
peter1138@9070: 	 * this information into a temporary array. This is then copied into the
peter1138@9070: 	 * engine pool after processing NewGRFs by CopyTempEngineData(). */
truelight@0: 	int index;
truelight@0: 	while ((index = SlIterateArray()) != -1) {
peter1138@9070: 		Engine *e = GetTempDataEngine(index);
peter1138@9070: 		SlObject(e, _engine_desc);
truelight@0: 	}
truelight@0: }
truelight@0: 
peter1138@9070: /**
peter1138@9070:  * Copy data from temporary engine array into the real engine pool.
peter1138@9070:  */
peter1138@9070: void CopyTempEngineData()
peter1138@9070: {
peter1138@9070: 	Engine *e;
peter1138@9070: 	FOR_ALL_ENGINES(e) {
peter1138@9070: 		if (e->index >= _temp_engine.size()) break;
peter1138@9070: 
peter1138@9070: 		const Engine *se = GetTempDataEngine(e->index);
peter1138@9070: 		e->intro_date          = se->intro_date;
peter1138@9070: 		e->age                 = se->age;
peter1138@9070: 		e->reliability         = se->reliability;
peter1138@9070: 		e->reliability_spd_dec = se->reliability_spd_dec;
peter1138@9070: 		e->reliability_start   = se->reliability_start;
peter1138@9070: 		e->reliability_max     = se->reliability_max;
peter1138@9070: 		e->reliability_final   = se->reliability_final;
peter1138@9070: 		e->duration_phase_1    = se->duration_phase_1;
peter1138@9070: 		e->duration_phase_2    = se->duration_phase_2;
peter1138@9070: 		e->duration_phase_3    = se->duration_phase_3;
peter1138@9070: 		e->lifelength          = se->lifelength;
peter1138@9070: 		e->flags               = se->flags;
rubidium@10207: 		e->preview_company_rank= se->preview_company_rank;
peter1138@9070: 		e->preview_wait        = se->preview_wait;
rubidium@10207: 		e->company_avail       = se->company_avail;
peter1138@9070: 		if (se->name != NULL) e->name = strdup(se->name);
peter1138@9070: 	}
peter1138@9070: 
peter1138@9070: 	/* Get rid of temporary data */
peter1138@9070: 	_temp_engine.clear();
peter1138@9070: }
peter1138@9070: 
peter1138@8258: static void Load_ENGS()
truelight@0: {
peter1138@9070: 	/* Load old separate String ID list into a temporary array. This
peter1138@9070: 	 * was always 256 entries. */
peter1138@9070: 	StringID names[256];
peter1138@8258: 
peter1138@8258: 	SlArray(names, lengthof(names), SLE_STRINGID);
peter1138@8258: 
peter1138@9070: 	/* Copy each string into the temporary engine array. */
peter1138@8258: 	for (EngineID engine = 0; engine < lengthof(names); engine++) {
peter1138@9070: 		Engine *e = GetTempDataEngine(engine);
peter1138@8258: 		e->name = CopyFromOldName(names[engine]);
peter1138@8258: 	}
truelight@0: }
truelight@0: 
rubidium@5587: extern const ChunkHandler _engine_chunk_handlers[] = {
peter1138@2848: 	{ 'ENGN', Save_ENGN,     Load_ENGN,     CH_ARRAY          },
frosch@10127: 	{ 'ENGS', NULL,          Load_ENGS,     CH_RIFF | CH_LAST },
truelight@0: };