tron@2186: /* $Id$ */
tron@2186: 
darkvater@164: #include "stdafx.h"
Darkvater@1891: #include "openttd.h"
tron@1299: #include "debug.h"
tron@2163: #include "functions.h"
tron@2154: #include "player.h"
tron@2466: #include "station.h"
tron@1309: #include "strings.h"
tron@1363: #include "table/sprites.h"
tron@507: #include "table/strings.h"
darkvater@164: #include "vehicle.h"
tron@588: #include "window.h"
bjarni@842: #include "engine.h"
bjarni@842: #include "gui.h"
bjarni@842: #include "command.h"
bjarni@842: #include "gfx.h"
tron@2159: #include "variables.h"
tron@2159: #include "vehicle_gui.h"
bjarni@2552: #include "viewport.h"
tron@2159: 
tron@2159: Sorting _sorting;
tron@2159: 
tron@2159: static uint32 _internal_name_sorter_id; // internal StringID for default vehicle-names
tron@2159: static uint32 _last_vehicle_idx;        // cached index to hopefully speed up name-sorting
tron@2159: static bool   _internal_sort_order;     // descending/ascending
darkvater@164: 
tron@2288: static uint16 _player_num_engines[256];
tron@2520: static RailType _railtype_selected_in_replace_gui;
tron@2288: 
tron@2462: 
tron@2462: typedef int CDECL VehicleSortListingTypeFunction(const void*, const void*);
tron@2462: 
tron@2462: static VehicleSortListingTypeFunction VehicleUnsortedSorter;
tron@2462: static VehicleSortListingTypeFunction VehicleNumberSorter;
tron@2462: static VehicleSortListingTypeFunction VehicleNameSorter;
tron@2462: static VehicleSortListingTypeFunction VehicleAgeSorter;
tron@2462: static VehicleSortListingTypeFunction VehicleProfitThisYearSorter;
tron@2462: static VehicleSortListingTypeFunction VehicleProfitLastYearSorter;
tron@2462: static VehicleSortListingTypeFunction VehicleCargoSorter;
tron@2462: static VehicleSortListingTypeFunction VehicleReliabilitySorter;
tron@2462: static VehicleSortListingTypeFunction VehicleMaxSpeedSorter;
tron@2462: 
tron@2462: static VehicleSortListingTypeFunction* const _vehicle_sorter[] = {
tron@505: 	&VehicleUnsortedSorter,
tron@505: 	&VehicleNumberSorter,
tron@505: 	&VehicleNameSorter,
tron@505: 	&VehicleAgeSorter,
tron@505: 	&VehicleProfitThisYearSorter,
tron@505: 	&VehicleProfitLastYearSorter,
tron@505: 	&VehicleCargoSorter,
tron@505: 	&VehicleReliabilitySorter,
tron@505: 	&VehicleMaxSpeedSorter
tron@505: };
tron@505: 
tron@505: const StringID _vehicle_sort_listing[] = {
tron@505: 	STR_SORT_BY_UNSORTED,
tron@505: 	STR_SORT_BY_NUMBER,
tron@505: 	STR_SORT_BY_DROPDOWN_NAME,
tron@505: 	STR_SORT_BY_AGE,
tron@505: 	STR_SORT_BY_PROFIT_THIS_YEAR,
tron@505: 	STR_SORT_BY_PROFIT_LAST_YEAR,
tron@505: 	STR_SORT_BY_TOTAL_CAPACITY_PER_CARGOTYPE,
tron@505: 	STR_SORT_BY_RELIABILITY,
tron@505: 	STR_SORT_BY_MAX_SPEED,
tron@505: 	INVALID_STRING_ID
tron@505: };
tron@505: 
tron@2462: static const StringID _rail_types_list[] = {
bjarni@842: 	STR_RAIL_VEHICLES,
bjarni@842: 	STR_MONORAIL_VEHICLES,
bjarni@842: 	STR_MAGLEV_VEHICLES,
bjarni@842: 	INVALID_STRING_ID
bjarni@842: };
bjarni@842: 
tron@588: void RebuildVehicleLists(void)
tron@588: {
tron@588: 	Window *w;
tron@588: 
tron@588: 	for (w = _windows; w != _last_window; ++w)
tron@588: 		switch (w->window_class) {
darkvater@967: 		case WC_TRAINS_LIST: case WC_ROADVEH_LIST:
darkvater@967: 		case WC_SHIPS_LIST:  case WC_AIRCRAFT_LIST:
darkvater@967: 			WP(w, vehiclelist_d).flags |= VL_REBUILD;
darkvater@967: 			SetWindowDirty(w);
darkvater@967: 			break;
darkvater@967: 		default: break;
tron@588: 		}
tron@588: }
tron@588: 
tron@588: void ResortVehicleLists(void)
tron@588: {
tron@588: 	Window *w;
tron@588: 
tron@588: 	for (w = _windows; w != _last_window; ++w)
tron@588: 		switch (w->window_class) {
darkvater@967: 		case WC_TRAINS_LIST: case WC_ROADVEH_LIST:
darkvater@967: 		case WC_SHIPS_LIST:  case WC_AIRCRAFT_LIST:
darkvater@967: 			WP(w, vehiclelist_d).flags |= VL_RESORT;
darkvater@967: 			SetWindowDirty(w);
darkvater@967: 			break;
darkvater@967: 		default: break;
tron@588: 		}
tron@588: }
tron@588: 
tron@2466: void BuildVehicleList(vehiclelist_d* vl, int type, PlayerID owner, StationID station)
tron@588: {
bjarni@1067: 	int subtype = (type != VEH_Aircraft) ? TS_Front_Engine : 2;
tron@588: 	int n = 0;
tron@588: 	int i;
tron@588: 
tron@588: 	if (!(vl->flags & VL_REBUILD)) return;
tron@588: 
truelight@919: 	/* Create array for sorting */
truelight@1279: 	_vehicle_sort = realloc(_vehicle_sort, GetVehiclePoolSize() * sizeof(_vehicle_sort[0]));
truelight@919: 	if (_vehicle_sort == NULL)
truelight@919: 		error("Could not allocate memory for the vehicle-sorting-list");
truelight@919: 
tron@588: 	DEBUG(misc, 1) ("Building vehicle list for player %d station %d...",
truelight@867: 		owner, station);
tron@588: 
tron@2466: 	if (station != INVALID_STATION) {
tron@588: 		const Vehicle *v;
tron@588: 		FOR_ALL_VEHICLES(v) {
tron@588: 			if (v->type == type && v->subtype <= subtype) {
truelight@1024: 				const Order *order;
truelight@1024: 
truelight@1024: 				FOR_VEHICLE_ORDERS(v, order) {
truelight@1024: 					if (order->type == OT_GOTO_STATION && order->station == station) {
truelight@919: 						_vehicle_sort[n].index = v->index;
truelight@919: 						_vehicle_sort[n].owner = v->owner;
tron@588: 						++n;
tron@588: 						break;
tron@588: 					}
truelight@1024: 				}
tron@588: 			}
tron@588: 		}
tron@588: 	} else {
tron@588: 		const Vehicle *v;
tron@588: 		FOR_ALL_VEHICLES(v) {
tron@588: 			if (v->type == type && v->subtype <= subtype && v->owner == owner) {
truelight@919: 				_vehicle_sort[n].index = v->index;
truelight@919: 				_vehicle_sort[n].owner = v->owner;
tron@588: 				++n;
tron@588: 			}
tron@588: 		}
tron@588: 	}
tron@588: 
tron@2549: 	free(vl->sort_list);
tron@2549: 	vl->sort_list = malloc(n * sizeof(vl->sort_list[0]));
tron@2549: 	if (n != 0 && vl->sort_list == NULL)
truelight@919: 		error("Could not allocate memory for the vehicle-sorting-list");
tron@588: 	vl->list_length = n;
tron@588: 
tron@2549: 	for (i = 0; i < n; ++i) vl->sort_list[i] = _vehicle_sort[i];
tron@588: 
tron@588: 	vl->flags &= ~VL_REBUILD;
tron@588: 	vl->flags |= VL_RESORT;
tron@588: }
tron@588: 
tron@588: void SortVehicleList(vehiclelist_d *vl)
tron@588: {
tron@588: 	if (!(vl->flags & VL_RESORT)) return;
tron@588: 
tron@588: 	_internal_sort_order = vl->flags & VL_DESC;
tron@588: 	_internal_name_sorter_id = STR_SV_TRAIN_NAME;
tron@588: 	_last_vehicle_idx = 0; // used for "cache" in namesorting
tron@588: 	qsort(vl->sort_list, vl->list_length, sizeof(vl->sort_list[0]),
tron@588: 		_vehicle_sorter[vl->sort_type]);
tron@588: 
tron@588: 	vl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;
tron@588: 	vl->flags &= ~VL_RESORT;
tron@588: }
tron@588: 
tron@588: 
darkvater@164: /* General Vehicle GUI based procedures that are independent of vehicle types */
tron@1093: void InitializeVehiclesGuiList(void)
darkvater@164: {
tron@2519: 	_railtype_selected_in_replace_gui = RAILTYPE_RAIL;
darkvater@164: }
darkvater@164: 
darkvater@164: // draw the vehicle profit button in the vehicle list window.
Darkvater@2436: void DrawVehicleProfitButton(const Vehicle *v, int x, int y)
darkvater@164: {
darkvater@164: 	uint32 ormod;
darkvater@164: 
darkvater@164: 	// draw profit-based colored icons
darkvater@164: 	if(v->age <= 365 * 2)
Darkvater@1688: 		ormod = PALETTE_TO_GREY;
darkvater@164: 	else if(v->profit_last_year < 0)
Darkvater@1688: 		ormod = PALETTE_TO_RED;
darkvater@164: 	else if(v->profit_last_year < 10000)
Darkvater@1688: 		ormod = PALETTE_TO_YELLOW;
darkvater@164: 	else
Darkvater@1688: 		ormod = PALETTE_TO_GREEN;
Darkvater@1688: 	DrawSprite(SPR_BLOT | ormod, x, y);
darkvater@164: }
darkvater@164: 
hackykid@1859: /** Draw the list of available refit options for a consist.
Darkvater@1802:  * Draw the list and highlight the selected refit option (if any)
hackykid@1859:  * @param *v first vehicle in consist to get the refit-options of
Darkvater@1802:  * @param sel selected refit cargo-type in the window
Darkvater@1802:  * @return the cargo type that is hightlighted, CT_INVALID if none
Darkvater@1802:  */
Darkvater@1802: CargoID DrawVehicleRefitWindow(const Vehicle *v, int sel)
Darkvater@1802: {
Darkvater@1802: 	uint32 cmask;
Darkvater@1802: 	CargoID cid, cargo = CT_INVALID;
Darkvater@1802: 	int y = 25;
hackykid@1859: 	const Vehicle* u;
Darkvater@1802: #define show_cargo(ctype) { \
Darkvater@1802: 		byte colour = 16; \
Darkvater@1802: 		if (sel == 0) { \
Darkvater@1802: 			cargo = ctype; \
Darkvater@1802: 			colour = 12; \
Darkvater@1802: } \
Darkvater@1802: 		sel--; \
Darkvater@1802: 		DrawString(6, y, _cargoc.names_s[ctype], colour); \
Darkvater@1802: 		y += 10; \
Darkvater@1802: }
Darkvater@1802: 
hackykid@1859: 		/* Check if vehicle has custom refit or normal ones, and get its bitmasked value.
hackykid@1859: 		 * If its a train, 'or' this with the refit masks of the wagons. Now just 'and'
hackykid@1859: 		 * it with the bitmask of available cargo on the current landscape, and
hackykid@1859: 		 * where the bits are set: those are available */
hackykid@1859: 		cmask = 0;
hackykid@1859: 		u = v;
hackykid@1859: 		do {
hackykid@1859: 			if (_engine_refit_masks[u->engine_type] != 0) { // newgrf custom refit mask
hackykid@1859: 				cmask |= _engine_refit_masks[u->engine_type];
hackykid@1859: 			} else if (u->cargo_cap != 0) {
hackykid@1859: 				// rail wagons cant be refitted by default
hackykid@1859: 				if (v->type != VEH_Train || !(RailVehInfo(u->engine_type)->flags & RVI_WAGON))
hackykid@1859: 					cmask |= _default_refitmasks[v->type - VEH_Train];
hackykid@1859: 			}
hackykid@1859: 			u = u->next;
hackykid@1859: 		} while (v->type == VEH_Train && u != NULL);
hackykid@1859: 
Darkvater@1802: 		cmask &= _landscape_global_cargo_mask[_opt_ptr->landscape];
Darkvater@1802: 
Darkvater@1802: 		/* Check which cargo has been selected from the refit window and draw list */
Darkvater@1802: 		for (cid = 0; cmask != 0; cmask >>= 1, cid++) {
Darkvater@1802: 			if (HASBIT(cmask, 0)) // vehicle is refittable to this cargo
Darkvater@1802: 				show_cargo(_local_cargo_id_ctype[cid]);
Darkvater@1802: 		}
Darkvater@1802: 		return cargo;
Darkvater@1802: }
Darkvater@1802: 
darkvater@164: /************ Sorter functions *****************/
darkvater@164: int CDECL GeneralOwnerSorter(const void *a, const void *b)
darkvater@164: {
darkvater@168: 	return (*(const SortStruct*)a).owner - (*(const SortStruct*)b).owner;
darkvater@168: }
darkvater@168: 
darkvater@168: /* Variables you need to set before calling this function!
darkvater@168: * 1. (byte)_internal_sort_type:					sorting criteria to sort on
darkvater@168: * 2. (bool)_internal_sort_order:				sorting order, descending/ascending
darkvater@168: * 3. (uint32)_internal_name_sorter_id:	default StringID of the vehicle when no name is set. eg
darkvater@168: *    STR_SV_TRAIN_NAME for trains or STR_SV_AIRCRAFT_NAME for aircraft
darkvater@168: */
tron@2462: static int CDECL VehicleUnsortedSorter(const void *a, const void *b)
darkvater@168: {
tron@1294: 	return ((const SortStruct*)a)->index - ((const SortStruct*)b)->index;
darkvater@168: }
darkvater@168: 
darkvater@243: // if the sorting criteria had the same value, sort vehicle by unitnumber
darkvater@243: #define VEHICLEUNITNUMBERSORTER(r, a, b) {if (r == 0) {r = a->unitnumber - b->unitnumber;}}
darkvater@243: 
tron@2462: static int CDECL VehicleNumberSorter(const void *a, const void *b)
darkvater@168: {
truelight@919: 	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);
truelight@919: 	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);
darkvater@168: 	int r = va->unitnumber - vb->unitnumber;
darkvater@168: 
darkvater@168: 	return (_internal_sort_order & 1) ? -r : r;
darkvater@164: }
darkvater@164: 
darkvater@164: static char _bufcache[64];	// used together with _last_vehicle_idx to hopefully speed up stringsorting
tron@2462: static int CDECL VehicleNameSorter(const void *a, const void *b)
darkvater@164: {
darkvater@171: 	const SortStruct *cmp1 = (const SortStruct*)a;
darkvater@171: 	const SortStruct *cmp2 = (const SortStruct*)b;
truelight@919: 	const Vehicle *va = GetVehicle(cmp1->index);
truelight@919: 	const Vehicle *vb = GetVehicle(cmp2->index);
darkvater@164: 	char buf1[64] = "\0";
darkvater@164: 	int r;
darkvater@164: 
darkvater@168: 	if (va->string_id != _internal_name_sorter_id) {
tron@534: 		SetDParam(0, va->string_id);
darkvater@1017: 		GetString(buf1, STR_JUST_STRING);
darkvater@164: 	}
darkvater@164: 
darkvater@164: 	if ( cmp2->index != _last_vehicle_idx) {
darkvater@164: 		_last_vehicle_idx = cmp2->index;
darkvater@164: 		_bufcache[0] = '\0';
darkvater@168: 		if (vb->string_id != _internal_name_sorter_id) {
tron@534: 			SetDParam(0, vb->string_id);
darkvater@1017: 			GetString(_bufcache, STR_JUST_STRING);
darkvater@164: 		}
darkvater@164: 	}
darkvater@164: 
darkvater@164: 	r =  strcmp(buf1, _bufcache);	// sort by name
truelight@193: 
darkvater@243: 	VEHICLEUNITNUMBERSORTER(r, va, vb);
darkvater@168: 
darkvater@168: 	return (_internal_sort_order & 1) ? -r : r;
darkvater@164: }
darkvater@164: 
tron@2462: static int CDECL VehicleAgeSorter(const void *a, const void *b)
darkvater@164: {
truelight@919: 	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);
truelight@919: 	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);
darkvater@168: 	int r = va->age - vb->age;
darkvater@164: 
darkvater@243: 	VEHICLEUNITNUMBERSORTER(r, va, vb);
darkvater@168: 
darkvater@168: 	return (_internal_sort_order & 1) ? -r : r;
darkvater@168: }
darkvater@168: 
tron@2462: static int CDECL VehicleProfitThisYearSorter(const void *a, const void *b)
darkvater@168: {
truelight@919: 	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);
truelight@919: 	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);
darkvater@168: 	int r = va->profit_this_year - vb->profit_this_year;
truelight@193: 
darkvater@243: 	VEHICLEUNITNUMBERSORTER(r, va, vb);
darkvater@168: 
darkvater@168: 	return (_internal_sort_order & 1) ? -r : r;
darkvater@168: }
darkvater@168: 
tron@2462: static int CDECL VehicleProfitLastYearSorter(const void *a, const void *b)
darkvater@168: {
truelight@919: 	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);
truelight@919: 	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);
darkvater@168: 	int r = va->profit_last_year - vb->profit_last_year;
darkvater@168: 
darkvater@243: 	VEHICLEUNITNUMBERSORTER(r, va, vb);
darkvater@168: 
darkvater@168: 	return (_internal_sort_order & 1) ? -r : r;
darkvater@168: }
darkvater@168: 
tron@2462: static int CDECL VehicleCargoSorter(const void *a, const void *b)
darkvater@168: {
tron@2549: 	const Vehicle* va = GetVehicle(((const SortStruct*)a)->index);
tron@2549: 	const Vehicle* vb = GetVehicle(((const SortStruct*)b)->index);
tron@2549: 	const Vehicle* v;
tron@2549: 	AcceptedCargo cargoa;
tron@2549: 	AcceptedCargo cargob;
darkvater@168: 	int r = 0;
darkvater@168: 	int i;
darkvater@168: 
tron@2549: 	memset(cargoa, 0, sizeof(cargoa));
tron@2549: 	memset(cargob, 0, sizeof(cargob));
tron@2549: 	for (v = va; v != NULL; v = v->next) cargoa[v->cargo_type] += v->cargo_cap;
tron@2549: 	for (v = vb; v != NULL; v = v->next) cargob[v->cargo_type] += v->cargo_cap;
darkvater@168: 
darkvater@168: 	for (i = 0; i < NUM_CARGO; i++) {
tron@2549: 		r = cargoa[i] - cargob[i];
tron@2549: 		if (r != 0) break;
darkvater@168: 	}
darkvater@164: 
tron@594: 	VEHICLEUNITNUMBERSORTER(r, va, vb);
tron@594: 
darkvater@168: 	return (_internal_sort_order & 1) ? -r : r;
darkvater@164: }
darkvater@168: 
tron@2462: static int CDECL VehicleReliabilitySorter(const void *a, const void *b)
darkvater@168: {
truelight@919: 	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);
truelight@919: 	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);
darkvater@168: 	int r = va->reliability - vb->reliability;
darkvater@168: 
darkvater@243: 	VEHICLEUNITNUMBERSORTER(r, va, vb);
darkvater@168: 
darkvater@168: 	return (_internal_sort_order & 1) ? -r : r;
darkvater@168: }
darkvater@168: 
tron@2462: static int CDECL VehicleMaxSpeedSorter(const void *a, const void *b)
darkvater@168: {
truelight@919: 	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);
truelight@919: 	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);
celestar@1179: 	int max_speed_a = 0xFFFF, max_speed_b = 0xFFFF;
celestar@1179: 	int r;
celestar@1179: 	const Vehicle *ua = va, *ub = vb;
celestar@1179: 
Celestar@1191: 	if (va->type == VEH_Train && vb->type == VEH_Train) {
Celestar@1191: 		do {
Celestar@1191: 			if (RailVehInfo(ua->engine_type)->max_speed != 0)
Celestar@1191: 				max_speed_a = min(max_speed_a, RailVehInfo(ua->engine_type)->max_speed);
Celestar@1191: 		} while ((ua = ua->next) != NULL);
celestar@1179: 
Celestar@1191: 		do {
Celestar@1191: 			if (RailVehInfo(ub->engine_type)->max_speed != 0)
Celestar@1191: 				max_speed_b = min(max_speed_b, RailVehInfo(ub->engine_type)->max_speed);
Celestar@1191: 		} while ((ub = ub->next) != NULL);
Celestar@1191: 
Celestar@1191: 		r = max_speed_a - max_speed_b;
Celestar@1191: 	} else {
Celestar@1191: 		r = va->max_speed - vb->max_speed;
Celestar@1191: 	}
darkvater@168: 
darkvater@243: 	VEHICLEUNITNUMBERSORTER(r, va, vb);
darkvater@168: 
darkvater@168: 	return (_internal_sort_order & 1) ? -r : r;
truelight@193: }
bjarni@842: 
bjarni@842: // this define is to match engine.c, but engine.c keeps it to itself
bjarni@842: // ENGINE_AVAILABLE is used in ReplaceVehicleWndProc
tron@2530: #define ENGINE_AVAILABLE ((e->flags & 1 && HASBIT(info->climates, _opt.landscape)) || HASBIT(e->player_avail, _local_player))
bjarni@842: 
bjarni@842: /*  if show_outdated is selected, it do not sort psudo engines properly but it draws all engines
truelight@867:  *	if used compined with show_cars set to false, it will work as intended. Replace window do it like that
bjarni@842:  *  this was a big hack even before show_outdated was added. Stupid newgrf :p										*/
tron@2520: static void train_engine_drawing_loop(int *x, int *y, int *pos, int *sel, int *selected_id, RailType railtype,
bjarni@842: 	uint8 lines_drawn, bool is_engine, bool show_cars, bool show_outdated)
bjarni@842: {
tron@2477: 	EngineID i;
bjarni@842: 	byte colour;
truelight@867: 
bjarni@842: 	for (i = 0; i < NUM_TRAIN_ENGINES; i++) {
tron@1926: 		const Engine *e = GetEngine(i);
bjarni@842: 		const RailVehicleInfo *rvi = RailVehInfo(i);
bjarni@842: 		const EngineInfo *info = &_engine_info[i];
bjarni@842: 
bjarni@1128: 		if ( _player_num_engines[i] == 0 && show_outdated ) continue;
bjarni@1128: 
bjarni@842: 		if ( rvi->power == 0 && !(show_cars) )   // disables display of cars (works since they do not have power)
bjarni@842: 			continue;
truelight@867: 
bjarni@842: 		if (*sel == 0) *selected_id = i;
bjarni@842: 
bjarni@842: 
bjarni@842: 		colour = *sel == 0 ? 0xC : 0x10;
bjarni@842: 		if (!(ENGINE_AVAILABLE && show_outdated && RailVehInfo(i)->power && e->railtype == railtype)) {
bjarni@842: 			if (e->railtype != railtype || !(rvi->flags & RVI_WAGON) != is_engine ||
bjarni@842: 				!HASBIT(e->player_avail, _local_player))
bjarni@842: 				continue;
truelight@867: 		} /*else {
bjarni@842: 		// TODO find a nice red colour for vehicles being replaced
bjarni@842: 			if ( _autoreplace_array[i] != i )
bjarni@842: 				colour = *sel == 0 ? 0x44 : 0x45;
bjarni@842: 		} */
truelight@867: 
bjarni@842: 		if (IS_INT_INSIDE(--*pos, -lines_drawn, 0)) {
bjarni@842: 			DrawString(*x + 59, *y + 2, GetCustomEngineName(i),
bjarni@842: 				colour);
bjarni@842: 			DrawTrainEngine(*x + 29, *y + 6, i,
bjarni@842: 				SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));
bjarni@1128: 			if ( show_outdated ) {
bjarni@1128: 				SetDParam(0, _player_num_engines[i]);
bjarni@1128: 				DrawStringRightAligned(213, *y+5, STR_TINY_BLACK, 0);
bjarni@1128: 			}
bjarni@842: 			*y += 14;
bjarni@842: 		}
bjarni@842: 		--*sel;
bjarni@842: 	}
bjarni@842: }
bjarni@842: 
bjarni@842: 
bjarni@842: static void SetupScrollStuffForReplaceWindow(Window *w)
bjarni@842: {
tron@2520: 	RailType railtype;
bjarni@842: 	int selected_id[2] = {-1,-1};
truelight@912: 	int sel[2];
bjarni@842: 	int count = 0;
bjarni@842: 	int count2 = 0;
tron@2477: 	EngineID engine_id;
truelight@867: 
truelight@912: 	sel[0] = WP(w,replaceveh_d).sel_index[0];
truelight@912: 	sel[1] = WP(w,replaceveh_d).sel_index[1];
tron@915: 
bjarni@842: 	switch (WP(w,replaceveh_d).vehicletype) {
bjarni@842: 		case VEH_Train: {
bjarni@1178: 			railtype = _railtype_selected_in_replace_gui;
bjarni@870: 			w->widget[13].color = _player_colors[_local_player];	// sets the colour of that art thing
bjarni@870: 			w->widget[16].color = _player_colors[_local_player];	// sets the colour of that art thing
bjarni@842: 			for (engine_id = 0; engine_id < NUM_TRAIN_ENGINES; engine_id++) {
tron@1926: 				const Engine *e = GetEngine(engine_id);
bjarni@842: 				const EngineInfo *info = &_engine_info[engine_id];
truelight@867: 
bjarni@1128: 				if (ENGINE_AVAILABLE && RailVehInfo(engine_id)->power && e->railtype == railtype ) {
bjarni@1128: 					if ( _player_num_engines[engine_id] ) {
bjarni@1128: 						if (sel[0]==0)  selected_id[0] = engine_id;
bjarni@1134: 						count++;
bjarni@1128: 						sel[0]--;
bjarni@1128: 					}
bjarni@842: 					if (HASBIT(e->player_avail, _local_player)) {
bjarni@842: 						if (sel[1]==0)  selected_id[1] = engine_id;
bjarni@1128: 						count2++;
bjarni@1128: 						sel[1]--;
bjarni@842: 					}
bjarni@842: 				}
bjarni@1128: 			}
bjarni@842: 			break;
bjarni@1128: 		}
bjarni@842: 		case VEH_Road: {
bjarni@842: 			int num = NUM_ROAD_ENGINES;
tron@1926: 			Engine *e = GetEngine(ROAD_ENGINES_INDEX);
bjarni@842: 			byte cargo;
bjarni@842: 			EngineInfo *info;
bjarni@842: 			engine_id = ROAD_ENGINES_INDEX;
bjarni@842: 
bjarni@842: 			do {
bjarni@842: 				info = &_engine_info[engine_id];
bjarni@1128: 				if (_player_num_engines[engine_id] ) {
bjarni@842: 					if (sel[0]==0)  selected_id[0] = engine_id;
bjarni@842: 					count++;
truelight@867: 					sel[0]--;
bjarni@842: 				}
bjarni@842: 			} while (++engine_id,++e,--num);
truelight@867: 
bjarni@842: 			if ( selected_id[0] != -1 ) {   // only draw right array if we have anything in the left one
bjarni@842: 				num = NUM_ROAD_ENGINES;
bjarni@842: 				engine_id = ROAD_ENGINES_INDEX;
tron@1926: 				e = GetEngine(ROAD_ENGINES_INDEX);
bjarni@842: 				cargo = RoadVehInfo(selected_id[0])->cargo_type;
truelight@867: 
bjarni@842: 				do {
bjarni@842: 					if ( cargo == RoadVehInfo(engine_id)->cargo_type && HASBIT(e->player_avail, _local_player)) {
bjarni@842: 						count2++;
bjarni@842: 						if (sel[1]==0)  selected_id[1] = engine_id;
bjarni@842: 						sel[1]--;
bjarni@842: 					}
bjarni@842: 				} while (++engine_id,++e,--num);
truelight@867: 			}
bjarni@842: 			break;
bjarni@842: 		}
truelight@867: 
bjarni@842: 		case VEH_Ship: {
bjarni@842: 			int num = NUM_SHIP_ENGINES;
tron@1926: 			Engine *e = GetEngine(SHIP_ENGINES_INDEX);
truelight@867: 			byte cargo, refittable;
bjarni@842: 			EngineInfo *info;
bjarni@842: 			engine_id = SHIP_ENGINES_INDEX;
bjarni@842: 
bjarni@842: 			do {
bjarni@842: 				info = &_engine_info[engine_id];
bjarni@1128: 				if (_player_num_engines[engine_id] ) {
bjarni@842: 					if ( sel[0] == 0 )  selected_id[0] = engine_id;
bjarni@842: 					count++;
truelight@867: 					sel[0]--;
bjarni@842: 				}
bjarni@842: 			} while (++engine_id,++e,--num);
truelight@867: 
bjarni@842: 			if ( selected_id[0] != -1 ) {
bjarni@842: 				num = NUM_SHIP_ENGINES;
tron@1926: 				e = GetEngine(SHIP_ENGINES_INDEX);
bjarni@842: 				engine_id = SHIP_ENGINES_INDEX;
bjarni@842: 				cargo = ShipVehInfo(selected_id[0])->cargo_type;
bjarni@842: 				refittable = ShipVehInfo(selected_id[0])->refittable;
truelight@867: 
truelight@867: 				do {
truelight@867: 					if (HASBIT(e->player_avail, _local_player)
bjarni@842: 					&& ( cargo == ShipVehInfo(engine_id)->cargo_type || refittable & ShipVehInfo(engine_id)->refittable)) {
truelight@867: 
bjarni@842: 						if ( sel[1]==0)  selected_id[1] = engine_id;
bjarni@842: 						sel[1]--;
bjarni@842: 						count2++;
bjarni@842: 					}
bjarni@842: 				} while (++engine_id,++e,--num);
bjarni@842: 			}
bjarni@842: 			break;
bjarni@842: 		}   //end of ship
truelight@867: 
bjarni@842: 		case VEH_Aircraft:{
bjarni@842: 			int num = NUM_AIRCRAFT_ENGINES;
bjarni@843: 			byte subtype;
tron@1926: 			Engine *e = GetEngine(AIRCRAFT_ENGINES_INDEX);
bjarni@842: 			EngineInfo *info;
bjarni@842: 			engine_id = AIRCRAFT_ENGINES_INDEX;
bjarni@842: 
bjarni@842: 			do {
bjarni@842: 				info = &_engine_info[engine_id];
bjarni@1128: 				if (_player_num_engines[engine_id]) {
bjarni@842: 					count++;
bjarni@842: 					if (sel[0]==0)  selected_id[0] = engine_id;
bjarni@842: 					sel[0]--;
bjarni@842: 				}
bjarni@842: 			} while (++engine_id,++e,--num);
truelight@867: 
bjarni@842: 			if ( selected_id[0] != -1 ) {
bjarni@842: 				num = NUM_AIRCRAFT_ENGINES;
tron@1926: 				e = GetEngine(AIRCRAFT_ENGINES_INDEX);
bjarni@842: 				subtype = AircraftVehInfo(selected_id[0])->subtype;
bjarni@842: 				engine_id = AIRCRAFT_ENGINES_INDEX;
bjarni@842: 				do {
bjarni@842: 					if (HASBIT(e->player_avail, _local_player)) {
bjarni@1182: 						if (HASBIT(subtype, 0) == HASBIT(AircraftVehInfo(engine_id)->subtype, 0)) {
bjarni@842: 							count2++;
bjarni@842: 							if (sel[1]==0)  selected_id[1] = engine_id;
bjarni@842: 							sel[1]--;
bjarni@842: 						}
bjarni@842: 					}
bjarni@842: 				} while (++engine_id,++e,--num);
bjarni@842: 			}
bjarni@842: 			break;
bjarni@842: 		}
bjarni@842: 	}
bjarni@842: 	// sets up the number of items in each list
bjarni@842: 	SetVScrollCount(w, count);
bjarni@842: 	SetVScroll2Count(w, count2);
bjarni@842: 	WP(w,replaceveh_d).sel_engine[0] = selected_id[0];
bjarni@842: 	WP(w,replaceveh_d).sel_engine[1] = selected_id[1];
truelight@867: 
bjarni@842: 	WP(w,replaceveh_d).count[0] = count;
bjarni@842: 	WP(w,replaceveh_d).count[1] = count2;
bjarni@842: 	return;
bjarni@842: }
bjarni@842: 
bjarni@842: 
bjarni@842: static void DrawEngineArrayInReplaceWindow(Window *w, int x, int y, int x2, int y2, int pos, int pos2,
truelight@867: 	int sel1, int sel2, int selected_id1, int selected_id2)
bjarni@842: {
truelight@912: 	int sel[2];
truelight@912: 	int selected_id[2];
truelight@912: 
truelight@912: 	sel[0] = sel1;
truelight@912: 	sel[1] = sel2;
tron@915: 
truelight@912: 	selected_id[0] = selected_id1;
truelight@912: 	selected_id[1] = selected_id2;
tron@915: 
bjarni@842: 	switch (WP(w,replaceveh_d).vehicletype) {
bjarni@842: 		case VEH_Train: {
tron@2520: 			RailType railtype = _railtype_selected_in_replace_gui;
hackykid@1923: 			DrawString(157, 99 + (14 * w->vscroll.cap), _rail_types_list[railtype], 0x10);
bjarni@842: 			/* draw sorting criteria string */
bjarni@842: 
bjarni@842: 			/* Ensure that custom engines which substituted wagons
bjarni@842: 			* are sorted correctly.
bjarni@842: 			* XXX - DO NOT EVER DO THIS EVER AGAIN! GRRR hacking in wagons as
bjarni@842: 			* engines to get more types.. Stays here until we have our own format
bjarni@842: 			* then it is exit!!! */
bjarni@842: 			train_engine_drawing_loop(&x, &y, &pos, &sel[0], &selected_id[0], railtype, w->vscroll.cap, true, false, true); // True engines
bjarni@842: 			train_engine_drawing_loop(&x2, &y2, &pos2, &sel[1], &selected_id[1], railtype, w->vscroll.cap, true, false, false); // True engines
bjarni@842: 			train_engine_drawing_loop(&x2, &y2, &pos2, &sel[1], &selected_id[1], railtype, w->vscroll.cap, false, false, false); // Feeble wagons
bjarni@842: 			break;
bjarni@842: 		}
truelight@867: 
bjarni@842: 		case VEH_Road: {
bjarni@842: 			int num = NUM_ROAD_ENGINES;
tron@1926: 			Engine *e = GetEngine(ROAD_ENGINES_INDEX);
tron@2477: 			EngineID engine_id = ROAD_ENGINES_INDEX;
bjarni@842: 			byte cargo;
bjarni@842: 			EngineInfo *info;
truelight@867: 
bjarni@845: 			if ( selected_id[0] >= ROAD_ENGINES_INDEX && selected_id[0] <= SHIP_ENGINES_INDEX ) {
truelight@867: 				cargo = RoadVehInfo(selected_id[0])->cargo_type;
bjarni@842: 
bjarni@845: 				do {
bjarni@845: 					info = &_engine_info[engine_id];
bjarni@1128: 					if (_player_num_engines[engine_id]) {
bjarni@845: 						if (IS_INT_INSIDE(--pos, -w->vscroll.cap, 0)) {
bjarni@845: 							DrawString(x+59, y+2, GetCustomEngineName(engine_id), sel[0]==0 ? 0xC : 0x10);
bjarni@845: 							DrawRoadVehEngine(x+29, y+6, engine_id, SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));
bjarni@1128: 							SetDParam(0, _player_num_engines[engine_id]);
bjarni@1128: 							DrawStringRightAligned(213, y+5, STR_TINY_BLACK, 0);
bjarni@845: 							y += 14;
bjarni@845: 						}
bjarni@1128: 					sel[0]--;
bjarni@1128: 					}
truelight@867: 
bjarni@1128: 					if ( RoadVehInfo(engine_id)->cargo_type == cargo && HASBIT(e->player_avail, _local_player) ) {
bjarni@1128: 						if (IS_INT_INSIDE(--pos2, -w->vscroll.cap, 0) && RoadVehInfo(engine_id)->cargo_type == cargo) {
bjarni@1128: 							DrawString(x2+59, y2+2, GetCustomEngineName(engine_id), sel[1]==0 ? 0xC : 0x10);
bjarni@1128: 							DrawRoadVehEngine(x2+29, y2+6, engine_id, SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));
bjarni@1128: 							y2 += 14;
bjarni@845: 						}
bjarni@1128: 						sel[1]--;
bjarni@842: 					}
bjarni@845: 				} while (++engine_id, ++e,--num);
bjarni@845: 			}
bjarni@842: 			break;
bjarni@842: 		}
truelight@867: 
bjarni@842: 		case VEH_Ship: {
bjarni@842: 			int num = NUM_SHIP_ENGINES;
tron@1926: 			Engine *e = GetEngine(SHIP_ENGINES_INDEX);
tron@2477: 			EngineID engine_id = SHIP_ENGINES_INDEX;
truelight@867: 			byte cargo, refittable;
bjarni@842: 			EngineInfo *info;
bjarni@842: 
bjarni@842: 			if ( selected_id[0] != -1 ) {
bjarni@842: 				cargo = ShipVehInfo(selected_id[0])->cargo_type;
bjarni@842: 				refittable = ShipVehInfo(selected_id[0])->refittable;
truelight@867: 
bjarni@845: 				do {
bjarni@845: 					info = &_engine_info[engine_id];
bjarni@1128: 					if (_player_num_engines[engine_id]) {
bjarni@845: 						if (IS_INT_INSIDE(--pos, -w->vscroll.cap, 0)) {
bjarni@845: 							DrawString(x+75, y+7, GetCustomEngineName(engine_id), sel[0]==0 ? 0xC : 0x10);
bjarni@845: 							DrawShipEngine(x+35, y+10, engine_id, SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));
bjarni@1128: 							SetDParam(0, _player_num_engines[engine_id]);
bjarni@1128: 							DrawStringRightAligned(213, y+15, STR_TINY_BLACK, 0);
bjarni@845: 							y += 24;
bjarni@845: 						}
bjarni@1128: 						sel[0]--;
bjarni@1128: 					}
bjarni@1128: 					if ( selected_id[0] != -1 ) {
bjarni@1128: 						if (HASBIT(e->player_avail, _local_player) && ( cargo == ShipVehInfo(engine_id)->cargo_type || refittable & ShipVehInfo(engine_id)->refittable)) {
bjarni@1128: 							if (IS_INT_INSIDE(--pos2, -w->vscroll.cap, 0)) {
bjarni@1128: 								DrawString(x2+75, y2+7, GetCustomEngineName(engine_id), sel[1]==0 ? 0xC : 0x10);
bjarni@1128: 								DrawShipEngine(x2+35, y2+10, engine_id, SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));
bjarni@1128: 								y2 += 24;
bjarni@845: 							}
bjarni@1128: 							sel[1]--;
bjarni@845: 						}
bjarni@842: 					}
bjarni@845: 				} while (++engine_id, ++e,--num);
bjarni@845: 			}
bjarni@845: 			break;
bjarni@842: 		}   //end of ship
truelight@867: 
bjarni@842: 		case VEH_Aircraft: {
bjarni@842: 			if ( selected_id[0] != -1 ) {
bjarni@842: 				int num = NUM_AIRCRAFT_ENGINES;
tron@1926: 				Engine *e = GetEngine(AIRCRAFT_ENGINES_INDEX);
tron@2477: 				EngineID engine_id = AIRCRAFT_ENGINES_INDEX;
bjarni@842: 				byte subtype = AircraftVehInfo(selected_id[0])->subtype;
bjarni@842: 				EngineInfo *info;
bjarni@842: 
bjarni@842: 				do {
bjarni@842: 					info = &_engine_info[engine_id];
bjarni@1128: 					if (_player_num_engines[engine_id]) {
bjarni@842: 						if (sel[0]==0) selected_id[0] = engine_id;
bjarni@842: 						if (IS_INT_INSIDE(--pos, -w->vscroll.cap, 0)) {
bjarni@842: 							DrawString(x+62, y+7, GetCustomEngineName(engine_id), sel[0]==0 ? 0xC : 0x10);
bjarni@842: 							DrawAircraftEngine(x+29, y+10, engine_id, SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));
bjarni@1128: 							SetDParam(0, _player_num_engines[engine_id]);
bjarni@1128: 							DrawStringRightAligned(213, y+15, STR_TINY_BLACK, 0);
bjarni@842: 							y += 24;
bjarni@842: 						}
bjarni@1128: 						sel[0]--;
bjarni@1128: 					}
bjarni@1182: 					if ( (HASBIT(subtype, 0) == HASBIT(AircraftVehInfo(engine_id)->subtype, 0))
bjarni@1128: 						&& HASBIT(e->player_avail, _local_player) ) {
bjarni@1128: 						if (sel[1]==0) selected_id[1] = engine_id;
bjarni@1128: 						if (IS_INT_INSIDE(--pos2, -w->vscroll.cap, 0)) {
bjarni@1128: 							DrawString(x2+62, y2+7, GetCustomEngineName(engine_id), sel[1]==0 ? 0xC : 0x10);
bjarni@1128: 							DrawAircraftEngine(x2+29, y2+10, engine_id, SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));
bjarni@1128: 							y2 += 24;
bjarni@1128: 						}
bjarni@842: 						sel[1]--;
bjarni@842: 					}
bjarni@842: 				} while (++engine_id, ++e,--num);
bjarni@842: 			}
bjarni@842: 			break;
bjarni@842: 		}   // end of aircraft
bjarni@842: 	}
bjarni@842: 
bjarni@842: }
bjarni@842: static void ReplaceVehicleWndProc(Window *w, WindowEvent *e)
bjarni@842: {
Darkvater@2375: 	static const StringID _vehicle_type_names[4] = {STR_019F_TRAIN, STR_019C_ROAD_VEHICLE, STR_019E_SHIP,STR_019D_AIRCRAFT};
bjarni@842: 
Darkvater@2375: 	switch (e->event) {
Darkvater@2375: 		case WE_PAINT: {
Darkvater@2375: 				const Player *p = GetPlayer(_local_player);
bjarni@842: 				int pos = w->vscroll.pos;
bjarni@842: 				int selected_id[2] = {-1,-1};
bjarni@842: 				int x = 1;
bjarni@842: 				int y = 15;
bjarni@842: 				int pos2 = w->vscroll2.pos;
bjarni@842: 				int x2 = 1 + 228;
bjarni@842: 				int y2 = 15;
truelight@912: 				int sel[2];
truelight@912: 				sel[0] = WP(w,replaceveh_d).sel_index[0];
truelight@912: 				sel[1] = WP(w,replaceveh_d).sel_index[1];
truelight@867: 
bjarni@1128: 				{
bjarni@1128: 					uint i;
bjarni@1128: 					const Vehicle *vehicle;
bjarni@1128: 
bjarni@1128: 					for (i = 0; i < lengthof(_player_num_engines); i++) {
bjarni@1128: 						_player_num_engines[i] = 0;
bjarni@1128: 					}
bjarni@1128: 					FOR_ALL_VEHICLES(vehicle) {
bjarni@1128: 						if ( vehicle->owner == _local_player ) {
bjarni@1128: 							if (vehicle->type == VEH_Aircraft && vehicle->subtype > 2) continue;
bjarni@1128: 
bjarni@1128: 							// do not count the vehicles, that contains only 0 in all var
bjarni@1128: 							if (vehicle->engine_type == 0 && vehicle->spritenum == 0 ) continue;
bjarni@1128: 
tron@1926: 							if (vehicle->type != GetEngine(vehicle->engine_type)->type) continue;
bjarni@1128: 
bjarni@1128: 							_player_num_engines[vehicle->engine_type]++;
bjarni@1128: 						}
bjarni@1128: 					}
bjarni@1128: 				}
bjarni@1128: 
bjarni@842: 				SetupScrollStuffForReplaceWindow(w);
truelight@867: 
bjarni@842: 				selected_id[0] = WP(w,replaceveh_d).sel_engine[0];
bjarni@842: 				selected_id[1] = WP(w,replaceveh_d).sel_engine[1];
truelight@867: 
bjarni@842: 			// sets the selected left item to the top one if it's greater than the number of vehicles in the left side
bjarni@842: 
bjarni@842: 				if ( WP(w,replaceveh_d).count[0] <= sel[0] ) {
bjarni@842: 					if (WP(w,replaceveh_d).count[0]) {
bjarni@842: 						sel[0] = 0;
bjarni@842: 						WP(w,replaceveh_d).sel_index[0] = 0;
bjarni@842: 						w->vscroll.pos = 0;
bjarni@842: 						// now we go back to set selected_id[1] properly
bjarni@842: 						SetWindowDirty(w);
bjarni@842: 						return;
bjarni@842: 					} else { //there are no vehicles in the left window
bjarni@842: 						selected_id[1] = -1;
bjarni@842: 					}
bjarni@842: 				}
bjarni@842: 
bjarni@842: 				if ( WP(w,replaceveh_d).count[1] <= sel[1] ) {
bjarni@842: 					if (WP(w,replaceveh_d).count[1]) {
bjarni@842: 						sel[1] = 0;
bjarni@842: 						WP(w,replaceveh_d).sel_index[1] = 0;
bjarni@842: 						w->vscroll2.pos = 0;
bjarni@842: 						// now we go back to set selected_id[1] properly
bjarni@842: 						SetWindowDirty(w);
bjarni@842: 						return;
bjarni@842: 					} else { //there are no vehicles in the right window
bjarni@842: 						selected_id[1] = -1;
bjarni@842: 					}
bjarni@842: 				}
truelight@867: 
bjarni@2293: 				if ( selected_id[0] == selected_id[1] || p->engine_replacement[selected_id[0]] == selected_id[1]
bjarni@842: 					|| selected_id[0] == -1 || selected_id[1] == -1 )
bjarni@842: 					SETBIT(w->disabled_state, 4);
bjarni@842: 				else
bjarni@842: 					CLRBIT(w->disabled_state, 4);
truelight@867: 
bjarni@2293: 				if (p->engine_replacement[selected_id[0]] == INVALID_ENGINE || selected_id[0] == -1)
bjarni@842: 					SETBIT(w->disabled_state, 6);
bjarni@842: 				else
bjarni@842: 					CLRBIT(w->disabled_state, 6);
truelight@867: 
bjarni@842: 				// now the actual drawing of the window itself takes place
Darkvater@2375: 				SetDParam(0, _vehicle_type_names[WP(w, replaceveh_d).vehicletype - VEH_Train]);
bjarni@842: 				DrawWindowWidgets(w);
truelight@867: 
bjarni@842: 				// sets up the string for the vehicle that is being replaced to
bjarni@842: 				if ( selected_id[0] != -1 ) {
bjarni@2293: 					if (p->engine_replacement[selected_id[0]] == INVALID_ENGINE)
bjarni@842: 						SetDParam(0, STR_NOT_REPLACING);
bjarni@842: 					else
bjarni@2293: 						SetDParam(0, GetCustomEngineName(p->engine_replacement[selected_id[0]]));
bjarni@842: 				} else {
bjarni@842: 					SetDParam(0, STR_NOT_REPLACING_VEHICLE_SELECTED);
bjarni@842: 				}
truelight@867: 
truelight@867: 
Darkvater@2375: 				DrawString(145, (w->resize.step_height == 24 ? 67 : 87 ) + ( w->resize.step_height * w->vscroll.cap), STR_02BD, 0x10);
bjarni@842: 
truelight@867: 
bjarni@842: 				/*	now we draw the two arrays according to what we just counted */
bjarni@842: 				DrawEngineArrayInReplaceWindow(w, x, y, x2, y2, pos, pos2, sel[0], sel[1], selected_id[0], selected_id[1]);
truelight@867: 
bjarni@842: 				WP(w,replaceveh_d).sel_engine[0] = selected_id[0];
bjarni@842: 				WP(w,replaceveh_d).sel_engine[1] = selected_id[1];
bjarni@842: 				/* now we draw the info about the vehicles we selected */
bjarni@842: 				switch (WP(w,replaceveh_d).vehicletype) {
bjarni@842: 					case VEH_Train: {
bjarni@842: 						byte i = 0;
bjarni@842: 						int offset = 0;
truelight@867: 
bjarni@842: 						for ( i = 0 ; i < 2 ; i++) {
bjarni@842: 							if ( i )
bjarni@842: 							offset = 228;
bjarni@842: 							if (selected_id[i] != -1) {
bjarni@842: 								if (!(RailVehInfo(selected_id[i])->flags & RVI_WAGON)) {
bjarni@842: 									/* it's an engine */
hackykid@1907: 									DrawTrainEnginePurchaseInfo(2 + offset, 15 + (14 * w->vscroll.cap), selected_id[i]);
bjarni@842: 								} else {
bjarni@842: 									/* it's a wagon. Train cars are not replaced with the current GUI, but this code is ready for newgrf if anybody adds that*/
hackykid@1907: 									DrawTrainWagonPurchaseInfo(2 + offset, 15 + (14 * w->vscroll.cap), selected_id[i]);
bjarni@842: 								}
bjarni@842: 							}
bjarni@842: 						}
bjarni@842: 						break;
bjarni@842: 					}   //end if case  VEH_Train
truelight@867: 
bjarni@842: 					case VEH_Road: {
bjarni@842: 						if (selected_id[0] != -1) {
hackykid@1907: 							DrawRoadVehPurchaseInfo(2, 15 + (14 * w->vscroll.cap), selected_id[0]);
bjarni@842: 							if (selected_id[1] != -1) {
hackykid@1907: 								DrawRoadVehPurchaseInfo(2 + 228, 15 + (14 * w->vscroll.cap), selected_id[1]);
bjarni@842: 							}
bjarni@842: 						}
bjarni@842: 						break;
bjarni@842: 					}   // end of VEH_Road
truelight@867: 
bjarni@842: 					case VEH_Ship: {
bjarni@842: 						if (selected_id[0] != -1) {
hackykid@1907: 							DrawShipPurchaseInfo(2, 15 + (24 * w->vscroll.cap), selected_id[0]);
bjarni@842: 							if (selected_id[1] != -1) {
hackykid@1907: 								DrawShipPurchaseInfo(2 + 228, 15 + (24 * w->vscroll.cap), selected_id[1]);
bjarni@842: 							}
bjarni@842: 						}
bjarni@842: 						break;
truelight@867: 					}   // end of VEH_Ship
truelight@867: 
bjarni@842: 					case VEH_Aircraft: {
bjarni@842: 						if (selected_id[0] != -1) {
hackykid@1907: 							DrawAircraftPurchaseInfo(2, 15 + (24 * w->vscroll.cap), selected_id[0]);
bjarni@842: 							if (selected_id[1] != -1) {
hackykid@1907: 								DrawAircraftPurchaseInfo(2 + 228, 15 + (24 * w->vscroll.cap), selected_id[1]);
bjarni@842: 							}
bjarni@842: 						}
bjarni@842: 						break;
bjarni@842: 					}   // end of VEH_Aircraft
bjarni@842: 				}
Darkvater@1462: 			} break;   // end of paint
bjarni@842: 
bjarni@842: 		case WE_CLICK: {
Darkvater@2375: 			// these 3 variables is used if any of the lists is clicked
Darkvater@2375: 			uint16 click_scroll_pos = w->vscroll2.pos;
Darkvater@2375: 			uint16 click_scroll_cap = w->vscroll2.cap;
Darkvater@2375: 			byte click_side = 1;
Darkvater@2375: 
Darkvater@2375: 			switch (e->click.widget) {
bjarni@2206: 				case 14: case 15: { /* Select sorting criteria dropdown menu */
peter1138@2448: 					ShowDropDownMenu(w, _rail_types_list, _railtype_selected_in_replace_gui, 15, 0, ~GetPlayer(_local_player)->avail_railtypes);
bjarni@869: 					break;
bjarni@842: 				}
bjarni@2293: 				case 4: { /* Start replacing */
bjarni@2293: 					EngineID veh_from = WP(w, replaceveh_d).sel_engine[0];
bjarni@2293: 					EngineID veh_to = WP(w, replaceveh_d).sel_engine[1];
bjarni@2293: 					DoCommandP(0, 3, veh_from + (veh_to << 16), NULL, CMD_REPLACE_VEHICLE);
bjarni@842: 					SetWindowDirty(w);
bjarni@842: 					break;
bjarni@842: 				}
truelight@867: 
bjarni@2293: 				case 6: { /* Stop replacing */
bjarni@2293: 					EngineID veh_from = WP(w, replaceveh_d).sel_engine[0];
bjarni@2293: 					DoCommandP(0, 3, veh_from + (INVALID_ENGINE << 16), NULL, CMD_REPLACE_VEHICLE);
bjarni@842: 					SetWindowDirty(w);
bjarni@842: 					break;
bjarni@842: 				}
truelight@867: 
truelight@867: 				case 7:
bjarni@842: 					// sets up that the left one was clicked. The default values are for the right one (9)
bjarni@842: 					// this way, the code for 9 handles both sides
bjarni@842: 					click_scroll_pos = w->vscroll.pos;
bjarni@842: 					click_scroll_cap = w->vscroll.cap;
bjarni@842: 					click_side = 0;
bjarni@842: 				case 9: {
bjarni@869: 					uint i = (e->click.pt.y - 14) / w->resize.step_height;
bjarni@842: 					if (i < click_scroll_cap) {
bjarni@842: 						WP(w,replaceveh_d).sel_index[click_side] = i + click_scroll_pos;
bjarni@842: 						SetWindowDirty(w);
bjarni@842: 					}
bjarni@842: 				} break;
bjarni@842: 			}
truelight@867: 
bjarni@842: 		} break;
bjarni@842: 
bjarni@842: 		case WE_DROPDOWN_SELECT: { /* we have selected a dropdown item in the list */
bjarni@1178: 			_railtype_selected_in_replace_gui = e->dropdown.index;
bjarni@842: 			SetWindowDirty(w);
truelight@867: 		} break;
truelight@867: 
truelight@867: 		case WE_RESIZE: {
truelight@867: 			w->vscroll.cap  += e->sizing.diff.y / (int)w->resize.step_height;
truelight@867: 			w->vscroll2.cap += e->sizing.diff.y / (int)w->resize.step_height;
truelight@867: 
truelight@867: 			w->widget[7].unkA = (w->vscroll.cap  << 8) + 1;
truelight@867: 			w->widget[9].unkA = (w->vscroll2.cap << 8) + 1;
truelight@867: 		} break;
bjarni@842: 	}
bjarni@842: }
bjarni@842: 
bjarni@842: static const Widget _replace_rail_vehicle_widgets[] = {
Darkvater@1465: {   WWT_CLOSEBOX,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5,       STR_018B_CLOSE_WINDOW},
bjarni@869: {    WWT_CAPTION,   RESIZE_NONE,    14,    11,   443,     0,    13, STR_REPLACE_VEHICLES_WHITE, STR_018C_WINDOW_TITLE_DRAG_THIS},
Darkvater@2375: {  WWT_STICKYBOX,   RESIZE_NONE,    14,   444,   455,     0,    13, STR_NULL,       STR_STICKY_BUTTON},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,     0,   227,   126,   197, STR_NULL,       STR_NULL},
hackykid@1923: { WWT_PUSHTXTBTN,     RESIZE_TB,    14,     0,   138,   210,   221, STR_REPLACE_VEHICLES_START, STR_REPLACE_HELP_START_BUTTON},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   139,   316,   198,   209, STR_NULL,       STR_REPLACE_HELP_REPLACE_INFO_TAB},
hackykid@1923: { WWT_PUSHTXTBTN,     RESIZE_TB,    14,   306,   443,   210,   221, STR_REPLACE_VEHICLES_STOP,  STR_REPLACE_HELP_STOP_BUTTON},
Darkvater@1465: {     WWT_MATRIX, RESIZE_BOTTOM,    14,     0,   215,    14,   125, 0x801,          STR_REPLACE_HELP_LEFT_ARRAY},
Darkvater@2375: {  WWT_SCROLLBAR, RESIZE_BOTTOM,    14,   216,   227,    14,   125, STR_NULL,       STR_0190_SCROLL_BAR_SCROLLS_LIST},
Darkvater@1465: {     WWT_MATRIX, RESIZE_BOTTOM,    14,   228,   443,    14,   125, 0x801,          STR_REPLACE_HELP_RIGHT_ARRAY},
Darkvater@2375: { WWT_SCROLL2BAR, RESIZE_BOTTOM,    14,   444,   455,    14,   125, STR_NULL,       STR_0190_SCROLL_BAR_SCROLLS_LIST},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   228,   455,   126,   197, STR_NULL,       STR_NULL},
bjarni@869: // train specific stuff
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,     0,   138,   198,   209, STR_NULL,       STR_NULL},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   139,   153,   210,   221, STR_NULL,       STR_NULL},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   154,   277,   210,   221, STR_NULL,       STR_REPLACE_HELP_RAILTYPE},
hackykid@1923: {   WWT_CLOSEBOX,     RESIZE_TB,    14,   278,   289,   210,   221, STR_0225,       STR_REPLACE_HELP_RAILTYPE},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   290,   305,   210,   221, STR_NULL,       STR_NULL},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   317,   455,   198,   209, STR_NULL,       STR_NULL},
bjarni@869: // end of train specific stuff
Darkvater@2375: {  WWT_RESIZEBOX,     RESIZE_TB,    14,   444,   455,   210,   221, STR_NULL,       STR_RESIZE_BUTTON},
bjarni@842: {   WIDGETS_END},
bjarni@842: };
bjarni@842: 
bjarni@842: static const Widget _replace_road_vehicle_widgets[] = {
Darkvater@1465: {   WWT_CLOSEBOX,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5,        STR_018B_CLOSE_WINDOW},
Darkvater@1465: {    WWT_CAPTION,   RESIZE_NONE,    14,    11,   443,     0,    13, STR_REPLACE_VEHICLES_WHITE,  STR_018C_WINDOW_TITLE_DRAG_THIS},
Darkvater@2375: {  WWT_STICKYBOX,   RESIZE_NONE,    14,   444,   455,     0,    13, STR_NULL,       STR_STICKY_BUTTON},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,     0,   227,   126,   197, STR_NULL,       STR_NULL},
Darkvater@2375: { WWT_PUSHTXTBTN,     RESIZE_TB,    14,     0,   138,   198,   209, STR_REPLACE_VEHICLES_START,  STR_REPLACE_HELP_START_BUTTON},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   139,   305,   198,   209, STR_NULL,       STR_REPLACE_HELP_REPLACE_INFO_TAB},
Darkvater@2375: { WWT_PUSHTXTBTN,     RESIZE_TB,    14,   306,   443,   198,   209, STR_REPLACE_VEHICLES_STOP,   STR_REPLACE_HELP_STOP_BUTTON},
Darkvater@1465: {     WWT_MATRIX, RESIZE_BOTTOM,    14,     0,   215,    14,   125, 0x801,          STR_REPLACE_HELP_LEFT_ARRAY},
Darkvater@2375: {  WWT_SCROLLBAR, RESIZE_BOTTOM,    14,   216,   227,    14,   125, STR_NULL,       STR_0190_SCROLL_BAR_SCROLLS_LIST},
Darkvater@1465: {     WWT_MATRIX, RESIZE_BOTTOM,    14,   228,   443,    14,   125, 0x801,          STR_REPLACE_HELP_RIGHT_ARRAY},
Darkvater@2375: { WWT_SCROLL2BAR, RESIZE_BOTTOM,    14,   444,   455,    14,   125, STR_NULL,       STR_0190_SCROLL_BAR_SCROLLS_LIST},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   228,   455,   126,   197, STR_NULL,       STR_NULL},
Darkvater@2375: {  WWT_RESIZEBOX,     RESIZE_TB,    14,   444,   455,   198,   209, STR_NULL,       STR_RESIZE_BUTTON},
bjarni@842: {   WIDGETS_END},
bjarni@842: };
bjarni@842: 
bjarni@842: static const Widget _replace_ship_aircraft_vehicle_widgets[] = {
Darkvater@1465: {   WWT_CLOSEBOX,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5,       STR_018B_CLOSE_WINDOW},
Darkvater@1465: {    WWT_CAPTION,   RESIZE_NONE,    14,    11,   443,     0,    13, STR_REPLACE_VEHICLES_WHITE,  STR_018C_WINDOW_TITLE_DRAG_THIS},
Darkvater@2375: {  WWT_STICKYBOX,   RESIZE_NONE,    14,   444,   455,     0,    13, STR_NULL,       STR_STICKY_BUTTON},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,     0,   227,   110,   161, STR_NULL,       STR_NULL},
Darkvater@1465: { WWT_PUSHTXTBTN,     RESIZE_TB,    14,     0,   138,   162,   173, STR_REPLACE_VEHICLES_START,  STR_REPLACE_HELP_START_BUTTON},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   139,   305,   162,   173, STR_NULL,       STR_REPLACE_HELP_REPLACE_INFO_TAB},
Darkvater@1465: { WWT_PUSHTXTBTN,     RESIZE_TB,    14,   306,   443,   162,   173, STR_REPLACE_VEHICLES_STOP,   STR_REPLACE_HELP_STOP_BUTTON},
Darkvater@1465: {     WWT_MATRIX, RESIZE_BOTTOM,    14,     0,   215,    14,   109, 0x401,          STR_REPLACE_HELP_LEFT_ARRAY},
Darkvater@2375: {  WWT_SCROLLBAR, RESIZE_BOTTOM,    14,   216,   227,    14,   109, STR_NULL,       STR_0190_SCROLL_BAR_SCROLLS_LIST},
Darkvater@1465: {     WWT_MATRIX, RESIZE_BOTTOM,    14,   228,   443,    14,   109, 0x401,          STR_REPLACE_HELP_RIGHT_ARRAY},
Darkvater@2375: { WWT_SCROLL2BAR, RESIZE_BOTTOM,    14,   444,   455,    14,   109, STR_NULL,       STR_0190_SCROLL_BAR_SCROLLS_LIST},
Darkvater@2375: {      WWT_PANEL,     RESIZE_TB,    14,   228,   455,   110,   161, STR_NULL,       STR_NULL},
Darkvater@2375: {  WWT_RESIZEBOX,     RESIZE_TB,    14,   444,   455,   162,   173, STR_NULL,       STR_RESIZE_BUTTON},
bjarni@842: {   WIDGETS_END},
bjarni@842: };
bjarni@842: 
bjarni@842: static const WindowDesc _replace_rail_vehicle_desc = {
hackykid@1923: 	-1, -1, 456, 222,
bjarni@842: 	WC_REPLACE_VEHICLE,0,
truelight@867: 	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS | WDF_STICKY_BUTTON | WDF_RESIZABLE,
bjarni@842: 	_replace_rail_vehicle_widgets,
bjarni@842: 	ReplaceVehicleWndProc
bjarni@842: };
bjarni@842: 
bjarni@842: static const WindowDesc _replace_road_vehicle_desc = {
Darkvater@2375: 	-1, -1, 456, 210,
bjarni@842: 	WC_REPLACE_VEHICLE,0,
truelight@867: 	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS | WDF_STICKY_BUTTON | WDF_RESIZABLE,
bjarni@842: 	_replace_road_vehicle_widgets,
bjarni@842: 	ReplaceVehicleWndProc
bjarni@842: };
bjarni@842: 
bjarni@842: static const WindowDesc _replace_ship_aircraft_vehicle_desc = {
bjarni@842: 	-1, -1, 456, 174,
bjarni@842: 	WC_REPLACE_VEHICLE,0,
truelight@867: 	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS | WDF_STICKY_BUTTON | WDF_RESIZABLE,
bjarni@842: 	_replace_ship_aircraft_vehicle_widgets,
bjarni@842: 	ReplaceVehicleWndProc
bjarni@842: };
bjarni@842: 
bjarni@842: 
bjarni@842: void ShowReplaceVehicleWindow(byte vehicletype)
bjarni@842: {
bjarni@842: 	Window *w;
truelight@867: 
tron@2549: 	DeleteWindowById(WC_REPLACE_VEHICLE, vehicletype);
truelight@867: 
bjarni@842: 	switch (vehicletype) {
bjarni@842: 		case VEH_Train:
bjarni@842: 			w = AllocateWindowDescFront(&_replace_rail_vehicle_desc, vehicletype);
bjarni@842: 			w->vscroll.cap  = 8;
truelight@867: 			w->resize.step_height = 14;
bjarni@842: 			break;
bjarni@842: 		case VEH_Road:
bjarni@842: 			w = AllocateWindowDescFront(&_replace_road_vehicle_desc, vehicletype);
bjarni@842: 			w->vscroll.cap  = 8;
truelight@867: 			w->resize.step_height = 14;
bjarni@842: 			break;
tron@2549: 		case VEH_Ship:
tron@2549: 		case VEH_Aircraft:
bjarni@842: 			w = AllocateWindowDescFront(&_replace_ship_aircraft_vehicle_desc, vehicletype);
bjarni@842: 			w->vscroll.cap  = 4;
truelight@867: 			w->resize.step_height = 24;
bjarni@842: 			break;
bjarni@845: 		default: return;
bjarni@842: 	}
bjarni@869: 	w->caption_color = _local_player;
bjarni@842: 	WP(w,replaceveh_d).vehicletype = vehicletype;
bjarni@842: 	w->vscroll2.cap = w->vscroll.cap;   // these two are always the same
bjarni@842: }
darkvater@1246: 
darkvater@1246: void InitializeGUI(void)
darkvater@1246: {
darkvater@1246: 	memset(&_sorting, 0, sizeof(_sorting));
darkvater@1246: }
bjarni@2552: 
bjarni@2552: /** Assigns an already open vehicle window to a new vehicle.
bjarni@2552: * Assigns an already open vehicle window to a new vehicle. If the vehicle got any sub window open (orders and so on) it will change owner too
bjarni@2552: * @param *from_v the current owner of the window
bjarni@2552: * @param *to_v the new owner of the window
bjarni@2552: */
bjarni@2552: void ChangeVehicleViewWindow(const Vehicle *from_v, const Vehicle *to_v)
bjarni@2552: {
bjarni@2552: 	Window *w;
bjarni@2552: 
bjarni@2552: 	w = FindWindowById(WC_VEHICLE_VIEW, from_v->index);
bjarni@2552: 
bjarni@2552: 	if (w != NULL) {
bjarni@2552: 		w->window_number = to_v->index;
bjarni@2553: 		WP(w, vp_d).follow_vehicle = to_v->index;	// tell the viewport to follow the new vehicle
bjarni@2552: 		SetWindowDirty(w);
bjarni@2552: 
bjarni@2552: 		w = FindWindowById(WC_VEHICLE_ORDERS, from_v->index);
bjarni@2552: 
bjarni@2552: 		if (w != NULL) {
bjarni@2552: 			w->window_number = to_v->index;
bjarni@2552: 			SetWindowDirty(w);
bjarni@2552: 		}
bjarni@2552: 
bjarni@2552: 		w = FindWindowById(WC_VEHICLE_REFIT, from_v->index);
bjarni@2552: 
bjarni@2552: 		if (w != NULL) {
bjarni@2552: 			w->window_number = to_v->index;
bjarni@2552: 			SetWindowDirty(w);
bjarni@2552: 		}
bjarni@2552: 
bjarni@2552: 		w = FindWindowById(WC_VEHICLE_DETAILS, from_v->index);
bjarni@2552: 
bjarni@2552: 		if (w != NULL) {
bjarni@2552: 			w->window_number = to_v->index;
bjarni@2552: 			SetWindowDirty(w);
bjarni@2552: 		}
bjarni@2552: 	}
bjarni@2552: }