bjarni@4800: /* $Id$ */
bjarni@4800: 
bjarni@4800: #include "stdafx.h"
bjarni@4800: #include "openttd.h"
bjarni@4800: #include "aircraft.h"
bjarni@4800: #include "debug.h"
bjarni@4800: #include "functions.h"
bjarni@4800: #include "table/sprites.h"
bjarni@4800: #include "table/strings.h"
bjarni@4800: #include "window.h"
bjarni@4800: #include "gui.h"
bjarni@4800: #include "vehicle.h"
bjarni@4800: #include "gfx.h"
bjarni@4800: #include "station.h"
bjarni@4800: #include "command.h"
bjarni@4800: #include "engine.h"
bjarni@4800: #include "player.h"
bjarni@4800: #include "depot.h"
bjarni@4800: #include "airport.h"
bjarni@4800: #include "vehicle_gui.h"
bjarni@4800: #include "newgrf_engine.h"
bjarni@4800: #include "date.h"
bjarni@4800: #include "strings.h"
bjarni@4800: 
bjarni@4800: 
bjarni@4800: typedef enum BuildVehicleWidgets {
bjarni@4800: 	BUILD_VEHICLE_WIDGET_CLOSEBOX = 0,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_CAPTION,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_SORT_ASSENDING_DESCENDING,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_SORT_TEXT,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_SORT_DROPDOWN,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_LIST,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_SCROLLBAR,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_PANEL,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_PLANES,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_JETS,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_HELICOPTERS,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_BUILD,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_RENAME,
bjarni@4800: 	BUILD_VEHICLE_WIDGET_RESIZE,
bjarni@4800: } BuildVehicleWidget;
bjarni@4800: 
bjarni@4800: static const Widget _build_vehicle_widgets[] = {
bjarni@4800: 	{   WWT_CLOSEBOX,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5,                STR_018B_CLOSE_WINDOW },
bjarni@4800: 	{    WWT_CAPTION,   RESIZE_NONE,    14,    11,   239,     0,    13, STR_A005_NEW_AIRCRAFT,   STR_018C_WINDOW_TITLE_DRAG_THIS },
bjarni@4800: 	{ WWT_PUSHTXTBTN,   RESIZE_NONE,    14,     0,    80,    14,    25, STR_SORT_BY,             STR_SORT_ORDER_TIP},
bjarni@4800: 	{      WWT_PANEL,   RESIZE_NONE,    14,    81,   227,    14,    25, 0x0,                     STR_SORT_CRITERIA_TIP},
bjarni@4800: 	{    WWT_TEXTBTN,   RESIZE_NONE,    14,   228,   239,    14,    25, STR_0225,                STR_SORT_CRITERIA_TIP},
bjarni@4800: 	{     WWT_MATRIX, RESIZE_BOTTOM,    14,     0,   227,    26,   121, 0x401,                   STR_A025_AIRCRAFT_SELECTION_LIST },
bjarni@4800: 	{  WWT_SCROLLBAR, RESIZE_BOTTOM,    14,   228,   239,    26,   121, 0x0,                     STR_0190_SCROLL_BAR_SCROLLS_LIST },
peter1138@4931: 	{      WWT_PANEL,     RESIZE_TB,    14,     0,   239,   122,   213, 0x0,                     STR_NULL },
bjarni@4800: 
peter1138@4931: 	{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,     0,    79,   214,   225, STR_BLACK_PLANES,        STR_BUILD_PLANES_TIP },
peter1138@4931: 	{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,    80,   159,   214,   225, STR_BLACK_JETS,          STR_BUILD_JETS_TIP },
peter1138@4931: 	{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,   160,   239,   214,   225, STR_BLACK_HELICOPTERS,   STR_BUILD_HELICOPTERS_TIP },
bjarni@4800: 
peter1138@4931: 	{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,     0,   114,   226,   237, STR_A006_BUILD_AIRCRAFT, STR_A026_BUILD_THE_HIGHLIGHTED_AIRCRAFT },
peter1138@4931: 	{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,   115,   227,   226,   237, STR_A037_RENAME,         STR_A038_RENAME_AIRCRAFT_TYPE },
peter1138@4931: 	{  WWT_RESIZEBOX,     RESIZE_TB,    14,   228,   239,   226,   237, 0x0,                     STR_RESIZE_BUTTON },
bjarni@4800: 	{   WIDGETS_END},
bjarni@4800: };
bjarni@4800: 
bjarni@4800: static bool _internal_sort_order; // descending/ascending
bjarni@4800: static byte _last_sort_criteria = 0;
bjarni@4800: static bool _last_sort_order = false;
bjarni@4800: 
bjarni@4800: typedef int CDECL VehicleSortListingTypeFunction(const void*, const void*);
bjarni@4800: 
bjarni@4800: static int CDECL EngineNumberSorter(const void *a, const void *b)
bjarni@4800: {
bjarni@4800: 	const EngineID va = *(const EngineID*)a;
bjarni@4800: 	const EngineID vb = *(const EngineID*)b;
bjarni@4800: 	int r = va - vb;
bjarni@4800: 
bjarni@4800: 	return _internal_sort_order ? -r : r;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static int CDECL EngineIntroDateSorter(const void *a, const void *b)
bjarni@4800: {
bjarni@4800: 	const int va = GetEngine(*(const EngineID*)a)->intro_date;
bjarni@4800: 	const int vb = GetEngine(*(const EngineID*)b)->intro_date;
bjarni@4800: 	const int r = va - vb;
bjarni@4800: 
bjarni@4800: 	if (r == 0) {
bjarni@4800: 		/* Use EngineID to sort instead since we want consistent sorting */
bjarni@4800: 		return EngineNumberSorter(a, b);
bjarni@4800: 	}
bjarni@4800: 	return _internal_sort_order ? -r : r;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static int CDECL EngineNameSorter(const void *a, const void *b)
bjarni@4800: {
bjarni@4804: 	static EngineID last_engine[2] = { INVALID_ENGINE, INVALID_ENGINE };
bjarni@4804: 	static char     last_name[2][64] = { "\0", "\0" };
bjarni@4804: 
bjarni@4800: 	const EngineID va = *(const EngineID*)a;
bjarni@4800: 	const EngineID vb = *(const EngineID*)b;
bjarni@4800: 	int r;
bjarni@4800: 
bjarni@4804: 	if (va != last_engine[0]) {
bjarni@4804: 		last_engine[0] = va;
Darkvater@4912: 		GetString(last_name[0], GetCustomEngineName(va), lastof(last_name[0]));
bjarni@4800: 	}
bjarni@4800: 
bjarni@4804: 	if (vb != last_engine[1]) {
bjarni@4804: 		last_engine[1] = vb;
Darkvater@4912: 		GetString(last_name[1], GetCustomEngineName(vb), lastof(last_name[1]));
bjarni@4804: 	}
bjarni@4804: 
bjarni@4804: 	r = strcmp(last_name[0], last_name[1]); // sort by name
bjarni@4800: 
bjarni@4800: 	if (r == 0) {
bjarni@4800: 		/* Use EngineID to sort instead since we want consistent sorting */
bjarni@4800: 		return EngineNumberSorter(a, b);
bjarni@4800: 	}
bjarni@4804: 	return _internal_sort_order ? -r : r;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static int CDECL EngineReliabilitySorter(const void *a, const void *b)
bjarni@4800: {
bjarni@4800: 	const int va = GetEngine(*(const EngineID*)a)->reliability;
bjarni@4800: 	const int vb = GetEngine(*(const EngineID*)b)->reliability;
bjarni@4800: 	const int r = va - vb;
bjarni@4800: 
bjarni@4800: 	if (r == 0) {
bjarni@4800: 		/* Use EngineID to sort instead since we want consistent sorting */
bjarni@4800: 		return EngineNumberSorter(a, b);
bjarni@4800: 	}
bjarni@4800: 	return _internal_sort_order ? -r : r;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: /* Aircraft sorting functions */
bjarni@4800: 
bjarni@4800: static int CDECL AircraftEngineCostSorter(const void *a, const void *b)
bjarni@4800: {
bjarni@4800: 	const int va = AircraftVehInfo(*(const EngineID*)a)->base_cost;
bjarni@4800: 	const int vb = AircraftVehInfo(*(const EngineID*)b)->base_cost;
bjarni@4800: 	int r = va - vb;
bjarni@4800: 
bjarni@4800: 	return _internal_sort_order ? -r : r;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static int CDECL AircraftEngineSpeedSorter(const void *a, const void *b)
bjarni@4800: {
bjarni@4800: 	const int va = AircraftVehInfo(*(const EngineID*)a)->max_speed;
bjarni@4800: 	const int vb = AircraftVehInfo(*(const EngineID*)b)->max_speed;
bjarni@4800: 	const int r = va - vb;
bjarni@4800: 
bjarni@4800: 	if (r == 0) {
bjarni@4800: 		/* Use EngineID to sort instead since we want consistent sorting */
bjarni@4800: 		return EngineNumberSorter(a, b);
bjarni@4800: 	}
bjarni@4800: 	return _internal_sort_order ? -r : r;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static int CDECL AircraftEngineRunningCostSorter(const void *a, const void *b)
bjarni@4800: {
bjarni@4800: 	const int va = AircraftVehInfo(*(const EngineID*)a)->running_cost;
bjarni@4800: 	const int vb = AircraftVehInfo(*(const EngineID*)b)->running_cost;
bjarni@4800: 	const int r = va - vb;
bjarni@4800: 
bjarni@4800: 	if (r == 0) {
bjarni@4800: 		/* Use EngineID to sort instead since we want consistent sorting */
bjarni@4800: 		return EngineNumberSorter(a, b);
bjarni@4800: 	}
bjarni@4800: 	return _internal_sort_order ? -r : r;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static int CDECL AircraftEngineCargoSorter(const void *a, const void *b)
bjarni@4800: {
bjarni@4800: 	const int va = AircraftVehInfo(*(const EngineID*)a)->passenger_capacity;
bjarni@4800: 	const int vb = AircraftVehInfo(*(const EngineID*)b)->passenger_capacity;
bjarni@4800: 	const int r = va - vb;
bjarni@4800: 
bjarni@4800: 	if (r == 0) {
bjarni@4800: 		/* Use EngineID to sort instead since we want consistent sorting */
bjarni@4800: 		return EngineNumberSorter(a, b);
bjarni@4800: 	}
bjarni@4800: 	return _internal_sort_order ? -r : r;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static VehicleSortListingTypeFunction* const _aircraft_sorter[] = {
bjarni@4800: 	&EngineNumberSorter,
bjarni@4800: 	&AircraftEngineCostSorter,
bjarni@4800: 	&AircraftEngineSpeedSorter,
bjarni@4800: 	&EngineIntroDateSorter,
bjarni@4800: 	&EngineNameSorter,
bjarni@4800: 	&AircraftEngineRunningCostSorter,
bjarni@4800: 	&EngineReliabilitySorter,
bjarni@4800: 	&AircraftEngineCargoSorter,
bjarni@4800: };
bjarni@4800: 
bjarni@4800: static const StringID _aircraft_sort_listing[] = {
bjarni@4800: 	STR_ENGINE_SORT_ENGINE_ID,
bjarni@4800: 	STR_ENGINE_SORT_COST,
bjarni@4800: 	STR_SORT_BY_MAX_SPEED,
bjarni@4800: 	STR_ENGINE_SORT_INTRO_DATE,
bjarni@4800: 	STR_SORT_BY_DROPDOWN_NAME,
bjarni@4800: 	STR_ENGINE_SORT_RUNNING_COST,
bjarni@4800: 	STR_SORT_BY_RELIABILITY,
bjarni@4800: 	STR_ENGINE_SORT_CARGO_CAPACITY,
bjarni@4800: 	INVALID_STRING_ID
bjarni@4800: };
bjarni@4800: 
bjarni@4800: 
bjarni@4800: /**
bjarni@4800: * Draw the purchase info details of an aircraft at a given location.
bjarni@4800:  * @param x,y location where to draw the info
bjarni@4800:  * @param engine_number the engine of which to draw the info of
bjarni@4800:  */
peter1138@4930: void DrawAircraftPurchaseInfo(int x, int y, uint w, EngineID engine_number)
bjarni@4800: {
bjarni@4800: 	const AircraftVehicleInfo *avi = AircraftVehInfo(engine_number);
bjarni@4800: 	const Engine *e = GetEngine(engine_number);
bjarni@4800: 	CargoID cargo;
bjarni@4800: 	YearMonthDay ymd;
bjarni@4800: 	ConvertDateToYMD(e->intro_date, &ymd);
bjarni@4800: 
bjarni@4800: 	/* Purchase cost - Max speed */
bjarni@4800: 	SetDParam(0, avi->base_cost * (_price.aircraft_base>>3)>>5);
bjarni@4800: 	SetDParam(1, avi->max_speed * 128 / 10);
bjarni@4800: 	DrawString(x, y, STR_PURCHASE_INFO_COST_SPEED, 0);
bjarni@4800: 	y += 10;
bjarni@4800: 
bjarni@4800: 	/* Cargo capacity */
bjarni@4800: 	cargo = FindFirstRefittableCargo(engine_number);
bjarni@4800: 	if (cargo == CT_INVALID || cargo == CT_PASSENGERS) {
bjarni@4800: 		SetDParam(0, avi->passenger_capacity);
bjarni@4800: 		SetDParam(1, avi->mail_capacity);
bjarni@4800: 		DrawString(x, y, STR_PURCHASE_INFO_AIRCRAFT_CAPACITY, 0);
bjarni@4800: 	} else {
bjarni@4800: 		/* Note, if the default capacity is selected by the refit capacity
bjarni@4800: 		* callback, then the capacity shown is likely to be incorrect. */
peter1138@4896: 		SetDParam(0, cargo);
bjarni@4800: 		SetDParam(1, AircraftDefaultCargoCapacity(cargo, engine_number));
bjarni@4800: 		SetDParam(2, STR_9842_REFITTABLE);
bjarni@4800: 		DrawString(x, y, STR_PURCHASE_INFO_CAPACITY, 0);
bjarni@4800: 	}
bjarni@4800: 	y += 10;
bjarni@4800: 
bjarni@4800: 	/* Running cost */
bjarni@4800: 	SetDParam(0, avi->running_cost * _price.aircraft_running >> 8);
bjarni@4800: 	DrawString(x, y, STR_PURCHASE_INFO_RUNNINGCOST, 0);
bjarni@4800: 	y += 10;
bjarni@4800: 
bjarni@4800: 	/* Design date - Life length */
bjarni@4800: 	SetDParam(0, ymd.year);
bjarni@4800: 	SetDParam(1, e->lifelength);
bjarni@4800: 	DrawString(x, y, STR_PURCHASE_INFO_DESIGNED_LIFE, 0);
bjarni@4800: 	y += 10;
bjarni@4800: 
bjarni@4800: 	/* Reliability */
bjarni@4800: 	SetDParam(0, e->reliability * 100 >> 16);
bjarni@4800: 	DrawString(x, y, STR_PURCHASE_INFO_RELIABILITY, 0);
bjarni@4800: 	y += 10;
bjarni@4800: 
bjarni@4800: 	/* Additional text from NewGRF */
peter1138@4930: 	y += ShowAdditionalText(x, y, w, engine_number);
peter1138@4932: 	y += ShowRefitOptionsList(x, y, w, engine_number);
bjarni@4800: }
bjarni@4800: 
bjarni@4800: void DrawAircraftImage(const Vehicle *v, int x, int y, VehicleID selection)
bjarni@4800: {
bjarni@4800: 	PalSpriteID pal = (v->vehstatus & VS_CRASHED) ? PALETTE_CRASH : GetVehiclePalette(v);
bjarni@4800: 	DrawSprite(GetAircraftImage(v, DIR_W) | pal, x + 25, y + 10);
bjarni@4800: 	if (v->subtype == 0) {
bjarni@4800: 		SpriteID rotor_sprite = GetCustomRotorSprite(v, true);
bjarni@4800: 		if (rotor_sprite == 0) rotor_sprite = SPR_ROTOR_STOPPED;
bjarni@4800: 		DrawSprite(rotor_sprite, x + 25, y + 5);
bjarni@4800: 	}
bjarni@4800: 	if (v->index == selection) {
bjarni@4800: 		DrawFrameRect(x - 1, y - 1, x + 58, y + 21, 0xF, FR_BORDERONLY);
bjarni@4800: 	}
bjarni@4800: }
bjarni@4800: 
bjarni@4800: void CcBuildAircraft(bool success, TileIndex tile, uint32 p1, uint32 p2)
bjarni@4800: {
bjarni@4800: 	if (success) {
bjarni@4800: 		const Vehicle *v = GetVehicle(_new_vehicle_id);
bjarni@4800: 
bjarni@4800: 		if (v->tile == _backup_orders_tile) {
bjarni@4800: 			_backup_orders_tile = 0;
bjarni@4800: 			RestoreVehicleOrders(v, _backup_orders_data);
bjarni@4800: 		}
bjarni@4800: 		ShowAircraftViewWindow(v);
bjarni@4800: 	}
bjarni@4800: }
bjarni@4800: 
bjarni@4807: static inline void ExtendEngineListSize(EngineID **engine_list, uint16 *engine_list_length, uint16 step_size, uint16 max)
bjarni@4800: {
bjarni@4800: 	*engine_list_length = min(*engine_list_length + step_size, max);
bjarni@4800: 	*engine_list = realloc((void*)*engine_list, (*engine_list_length) * sizeof((*engine_list)[0]));
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static void GenerateBuildAircraftList(EngineID **planes, uint16 *num_planes, EngineID **jets, uint16 *num_jets, EngineID **helicopters, uint16 *num_helicopters)
bjarni@4800: {
bjarni@4800: 	uint16 plane_length      = *num_planes;
bjarni@4800: 	uint16 jet_length        = *num_jets;
bjarni@4800: 	uint16 helicopter_length = *num_helicopters;
bjarni@4800: 	EngineID eid;
bjarni@4800: 
bjarni@4800: 	(*num_planes)      = 0;
bjarni@4800: 	(*num_jets)        = 0;
bjarni@4800: 	(*num_helicopters) = 0;
bjarni@4800: 
bjarni@4800: 	for (eid = AIRCRAFT_ENGINES_INDEX; eid < AIRCRAFT_ENGINES_INDEX + NUM_AIRCRAFT_ENGINES; eid++) {
bjarni@4800: 
bjarni@4800: 		if (IsEngineBuildable(eid, VEH_Aircraft)) {
bjarni@4800: 			const AircraftVehicleInfo *avi = AircraftVehInfo(eid);
bjarni@4800: 
bjarni@4800: 			switch (avi->subtype) {
bjarni@4800: 				case AIR_CTOL: // Propeller planes
bjarni@4807: 					if (*num_planes == plane_length) ExtendEngineListSize(planes, &plane_length, 5, NUM_AIRCRAFT_ENGINES);
bjarni@4800: 					(*planes)[(*num_planes)++] = eid;
bjarni@4800: 					break;
bjarni@4800: 
bjarni@4800: 				case (AIR_CTOL | AIR_FAST): // Jet planes
bjarni@4807: 					if (*num_jets == jet_length) ExtendEngineListSize(jets, &jet_length, 5, NUM_AIRCRAFT_ENGINES);
bjarni@4800: 					(*jets)[(*num_jets)++] = eid;
bjarni@4800: 					break;
bjarni@4800: 
bjarni@4800: 				case 0: // Helicopters
bjarni@4807: 					if (*num_helicopters == helicopter_length) ExtendEngineListSize(helicopters, &helicopter_length, 5, NUM_AIRCRAFT_ENGINES);
bjarni@4800: 					(*helicopters)[(*num_helicopters)++] = eid;
bjarni@4800: 					break;
bjarni@4800: 			}
bjarni@4800: 		}
bjarni@4800: 	}
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static void GenerateBuildList(Window *w)
bjarni@4800: {
bjarni@4800: 	switch (WP(w, buildvehicle_d).vehicle_type) {
bjarni@4800: 		case VEH_Aircraft:
bjarni@4800: 			GenerateBuildAircraftList(&WP(w, buildvehicle_d).list_a, &WP(w, buildvehicle_d).list_a_length,
bjarni@4800: 									  &WP(w, buildvehicle_d).list_b, &WP(w, buildvehicle_d).list_b_length,
bjarni@4800: 									  &WP(w, buildvehicle_d).list_c, &WP(w, buildvehicle_d).list_c_length);
bjarni@4800: 			break;
bjarni@4800: 
bjarni@4800: 		default: NOT_REACHED();
bjarni@4800: 	}
bjarni@4800: 
bjarni@4800: 	/* Ensure that we do not have trailing unused blocks in the arrays. We will never be able to access them anyway since we are unaware that they are there */
bjarni@4800: 	WP(w, buildvehicle_d).list_a = realloc((void*)WP(w, buildvehicle_d).list_a, WP(w, buildvehicle_d).list_a_length * sizeof(WP(w, buildvehicle_d).list_a[0]));
bjarni@4800: 	WP(w, buildvehicle_d).list_b = realloc((void*)WP(w, buildvehicle_d).list_b, WP(w, buildvehicle_d).list_b_length * sizeof(WP(w, buildvehicle_d).list_b[0]));
bjarni@4800: 	WP(w, buildvehicle_d).list_c = realloc((void*)WP(w, buildvehicle_d).list_c, WP(w, buildvehicle_d).list_c_length * sizeof(WP(w, buildvehicle_d).list_c[0]));
bjarni@4800: 
bjarni@4800: 	WP(w, buildvehicle_d).data_invalidated = false; // No need to regenerate the list anymore. We just did it
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static inline EngineID *GetEngineArray(Window *w)
bjarni@4800: {
bjarni@4800: 	switch (WP(w,buildvehicle_d).show_engine_button) {
bjarni@4800: 		case 1: return WP(w, buildvehicle_d).list_a;
bjarni@4800: 		case 2: return WP(w, buildvehicle_d).list_b;
bjarni@4800: 		case 3: return WP(w, buildvehicle_d).list_c;
bjarni@4800: 		default: NOT_REACHED();
bjarni@4800: 	}
bjarni@4800: 	return NULL;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static inline uint16 GetEngineArrayLength(Window *w)
bjarni@4800: {
bjarni@4800: 	switch (WP(w,buildvehicle_d).show_engine_button) {
bjarni@4800: 		case 1: return WP(w, buildvehicle_d).list_a_length;
bjarni@4800: 		case 2: return WP(w, buildvehicle_d).list_b_length;
bjarni@4800: 		case 3: return WP(w, buildvehicle_d).list_c_length;
bjarni@4800: 		default: NOT_REACHED();
bjarni@4800: 	}
bjarni@4800: 	return 0;
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static void SortAircraftBuildList(Window *w)
bjarni@4800: {
bjarni@4800: 	_internal_sort_order = WP(w,buildvehicle_d).decenting_sort_order;
bjarni@4800: 	qsort((void*)GetEngineArray(w), GetEngineArrayLength(w), sizeof(GetEngineArray(w)[0]),
bjarni@4800: 		  _aircraft_sorter[WP(w,buildvehicle_d).sort_criteria]);
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static void DrawBuildAircraftWindow(Window *w)
bjarni@4800: {
bjarni@4800: 	SetWindowWidgetLoweredState(w, BUILD_VEHICLE_WIDGET_PLANES,      WP(w,buildvehicle_d).show_engine_button == 1);
bjarni@4800: 	SetWindowWidgetLoweredState(w, BUILD_VEHICLE_WIDGET_JETS,        WP(w,buildvehicle_d).show_engine_button == 2);
bjarni@4800: 	SetWindowWidgetLoweredState(w, BUILD_VEHICLE_WIDGET_HELICOPTERS, WP(w,buildvehicle_d).show_engine_button == 3);
bjarni@4800: 
bjarni@4800: 	SetWindowWidgetDisabledState(w, BUILD_VEHICLE_WIDGET_BUILD, w->window_number == 0);
bjarni@4800: 
bjarni@4800: 	if (WP(w, buildvehicle_d).data_invalidated) {
bjarni@4800: 		GenerateBuildList(w);
bjarni@4800: 
bjarni@4800: 		if (WP(w,buildvehicle_d).sel_engine != INVALID_ENGINE) {
bjarni@4800: 			int i;
bjarni@4800: 			bool found = false;
bjarni@4800: 			if (HASBIT(WP(w,buildvehicle_d).show_engine_button, 0)) {
bjarni@4800: 				for (i = 0; i < GetEngineArrayLength(w); i++) {
bjarni@4800: 					if (WP(w,buildvehicle_d).sel_engine != GetEngineArray(w)[i]) continue;
bjarni@4800: 					found = true;
bjarni@4800: 					break;
bjarni@4800: 				}
bjarni@4800: 			}
bjarni@4800: 			if (!found) WP(w,buildvehicle_d).sel_engine = INVALID_ENGINE;
bjarni@4800: 		}
bjarni@4800: 	}
bjarni@4800: 
bjarni@4800: 	SetVScrollCount(w, GetEngineArrayLength(w));
bjarni@4800: 	DrawWindowWidgets(w);
bjarni@4800: 
bjarni@4800: 	if (WP(w,buildvehicle_d).sel_engine == INVALID_ENGINE && GetEngineArrayLength(w) != 0) {
bjarni@4800: 		WP(w,buildvehicle_d).sel_engine = GetEngineArray(w)[0];
bjarni@4800: 	}
bjarni@4800: 
bjarni@4800: 	{
bjarni@4800: 		int x = 2;
bjarni@4800: 		int y = 27;
bjarni@4800: 		EngineID selected_id = WP(w,buildvehicle_d).sel_engine;
bjarni@4800: 		EngineID eid = w->vscroll.pos;
bjarni@4800: 		EngineID *list = GetEngineArray(w);
bjarni@4800: 		uint16 list_length = GetEngineArrayLength(w);
bjarni@4800: 		uint16 max = min(w->vscroll.pos + w->vscroll.cap, list_length);
bjarni@4800: 
bjarni@4800: 		for(; eid < max; eid++) {
bjarni@4800: 			const EngineID engine = list[eid];
bjarni@4800: 
bjarni@4800: 			DrawString(x + 62, y + 7, GetCustomEngineName(engine), engine == selected_id ? 0xC : 0x10);
bjarni@4800: 			DrawAircraftEngine(x + 29, y + 10, engine, GetEnginePalette(engine, _local_player));
bjarni@4800: 			y += 24;
bjarni@4800: 		}
bjarni@4800: 
bjarni@4800: 		if (selected_id != INVALID_ENGINE) {
peter1138@4930: 			const Widget *wi = &w->widget[BUILD_VEHICLE_WIDGET_PANEL];
peter1138@4930: 			DrawAircraftPurchaseInfo(x, wi->top + 1, wi->right - wi->left - 2, selected_id);
bjarni@4800: 		}
bjarni@4800: 	}
bjarni@4800: 	DrawString(85, 15, _aircraft_sort_listing[WP(w,buildvehicle_d).sort_criteria], 0x10);
bjarni@4800: 	DoDrawString(WP(w,buildvehicle_d).decenting_sort_order ? DOWNARROW : UPARROW, 69, 15, 0x10);
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static void BuildAircraftClickEvent(Window *w, WindowEvent *e)
bjarni@4800: {
bjarni@4800: 	byte click_state = 0;
bjarni@4800: 
bjarni@4800: 	switch (e->we.click.widget) {
bjarni@4800: 		case BUILD_VEHICLE_WIDGET_SORT_ASSENDING_DESCENDING:
bjarni@4800: 			WP(w,buildvehicle_d).decenting_sort_order = !WP(w,buildvehicle_d).decenting_sort_order;
bjarni@4800: 			_last_sort_order = WP(w,buildvehicle_d).decenting_sort_order;
bjarni@4800: 			SortAircraftBuildList(w);
bjarni@4800: 			SetWindowDirty(w);
bjarni@4800: 			break;
bjarni@4800: 
bjarni@4800: 		case BUILD_VEHICLE_WIDGET_LIST: {
bjarni@4800: 			uint i = (e->we.click.pt.y - 26) / 24;
bjarni@4800: 			if (i < w->vscroll.cap) {
bjarni@4800: 				i += w->vscroll.pos;
bjarni@4800: 
bjarni@4800: 				if (i < GetEngineArrayLength(w)) {
bjarni@4800: 					WP(w,buildvehicle_d).sel_engine = GetEngineArray(w)[i];
bjarni@4800: 					SetWindowDirty(w);
bjarni@4800: 				}
bjarni@4800: 			}
bjarni@4800: 		} break;
bjarni@4800: 
bjarni@4800: 		case BUILD_VEHICLE_WIDGET_SORT_TEXT: case BUILD_VEHICLE_WIDGET_SORT_DROPDOWN:/* Select sorting criteria dropdown menu */
bjarni@4800: 			ShowDropDownMenu(w, _aircraft_sort_listing, WP(w,buildvehicle_d).sort_criteria, BUILD_VEHICLE_WIDGET_SORT_DROPDOWN, 0, 0);
bjarni@4800: 			return;
bjarni@4800: 
bjarni@4800: 		case BUILD_VEHICLE_WIDGET_HELICOPTERS: click_state++;
bjarni@4800: 		case BUILD_VEHICLE_WIDGET_JETS:        click_state++;
bjarni@4800: 		case BUILD_VEHICLE_WIDGET_PLANES:      click_state++;
bjarni@4800: 
bjarni@4800: 			if (WP(w,buildvehicle_d).show_engine_button == click_state) break; // We clicked the pressed button
bjarni@4800: 
bjarni@4800: 				WP(w,buildvehicle_d).sel_engine = INVALID_ENGINE;
bjarni@4800: 			WP(w,buildvehicle_d).show_engine_button = click_state;
bjarni@4800: 			w->vscroll.pos = 0;
bjarni@4800: 			SortAircraftBuildList(w);
bjarni@4800: 			SetWindowDirty(w);
bjarni@4800: 			break;
bjarni@4800: 
bjarni@4800: 		case BUILD_VEHICLE_WIDGET_BUILD: {
bjarni@4800: 			EngineID sel_eng = WP(w,buildvehicle_d).sel_engine;
bjarni@4800: 			if (sel_eng != INVALID_ENGINE)
bjarni@4800: 				DoCommandP(w->window_number, sel_eng, 0, CcBuildAircraft, CMD_BUILD_AIRCRAFT | CMD_MSG(STR_A008_CAN_T_BUILD_AIRCRAFT));
bjarni@4800: 		} break;
bjarni@4800: 
bjarni@4800: 		case BUILD_VEHICLE_WIDGET_RENAME: {
bjarni@4800: 			EngineID sel_eng = WP(w,buildvehicle_d).sel_engine;
bjarni@4800: 			if (sel_eng != INVALID_ENGINE) {
bjarni@4800: 				WP(w,buildvehicle_d).rename_engine = sel_eng;
bjarni@4800: 				ShowQueryString(GetCustomEngineName(sel_eng),
bjarni@4800: 								STR_A039_RENAME_AIRCRAFT_TYPE, 31, 160, w->window_class, w->window_number, CS_ALPHANUMERAL);
bjarni@4800: 			}
bjarni@4800: 		} break;
bjarni@4800: 	}
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static void NewAircraftWndProc(Window *w, WindowEvent *e)
bjarni@4800: {
bjarni@4800: 	switch (e->event) {
bjarni@4800: 		case WE_INVALIDATE_DATA:
bjarni@4800: 			WP(w,buildvehicle_d).data_invalidated = true;
bjarni@4800: 			SetWindowDirty(w);
bjarni@4800: 			break;
bjarni@4800: 
bjarni@4800: 		case WE_DESTROY:
bjarni@4800: 			free((void*)WP(w, buildvehicle_d).list_a);
bjarni@4800: 			free((void*)WP(w, buildvehicle_d).list_b);
bjarni@4800: 			free((void*)WP(w, buildvehicle_d).list_c);
bjarni@4800: 			break;
bjarni@4800: 
bjarni@4800: 		case WE_PAINT:
bjarni@4800: 			DrawBuildAircraftWindow(w);
bjarni@4800: 			break;
bjarni@4800: 
bjarni@4800: 		case WE_CLICK:
bjarni@4800: 			BuildAircraftClickEvent(w, e);
bjarni@4800: 			break;
bjarni@4800: 
bjarni@4800: 		case WE_ON_EDIT_TEXT: {
bjarni@4800: 			if (e->we.edittext.str[0] != '\0') {
bjarni@4800: 				_cmd_text = e->we.edittext.str;
bjarni@4800: 				DoCommandP(0, WP(w, buildvehicle_d).rename_engine, 0, NULL,
bjarni@4800: 						   CMD_RENAME_ENGINE | CMD_MSG(STR_A03A_CAN_T_RENAME_AIRCRAFT_TYPE));
bjarni@4800: 			}
bjarni@4800: 		} break;
bjarni@4800: 
bjarni@4800: 		case WE_DROPDOWN_SELECT: /* we have selected a dropdown item in the list */
bjarni@4800: 			if (WP(w,buildvehicle_d).sort_criteria != e->we.dropdown.index) {
bjarni@4800: 				WP(w,buildvehicle_d).sort_criteria = e->we.dropdown.index;
bjarni@4800: 				_last_sort_criteria = e->we.dropdown.index;
bjarni@4800: 				SortAircraftBuildList(w);
bjarni@4800: 			}
bjarni@4800: 			SetWindowDirty(w);
bjarni@4800: 				break;
bjarni@4800: 
bjarni@4800: 			case WE_RESIZE:
bjarni@4800: 				w->vscroll.cap += e->we.sizing.diff.y / 24;
bjarni@4800: 				w->widget[BUILD_VEHICLE_WIDGET_LIST].data = (w->vscroll.cap << 8) + 1;
bjarni@4800: 				break;
bjarni@4800: 	}
bjarni@4800: }
bjarni@4800: 
bjarni@4800: static const WindowDesc _build_vehicle_desc = {
Darkvater@5070: 	WDP_AUTO, WDP_AUTO, 240, 238,
bjarni@4800: 	WC_BUILD_VEHICLE,0,
bjarni@4800: 	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS | WDF_RESIZABLE,
bjarni@4800: 	_build_vehicle_widgets,
bjarni@4800: 	NewAircraftWndProc
bjarni@4800: };
bjarni@4800: 
bjarni@4800: /* Disable the aircraft subtype buttons for the types, that can't be build at the current airport */
bjarni@4800: static void CreateAircraftWindow(Window *w)
bjarni@4800: {
bjarni@4800: 	TileIndex tile = w->window_number;
bjarni@4800: 
bjarni@4800: 	if (tile == 0) {
bjarni@4800: 		WP(w, buildvehicle_d).show_engine_button = 1;
bjarni@4800: 	} else {
bjarni@4800: 		byte acc_planes = GetAirport(GetStationByTile(tile)->airport_type)->acc_planes;
bjarni@4800: 
bjarni@4800: 		WP(w, buildvehicle_d).show_engine_button = 0;
bjarni@4800: 		if (acc_planes == HELICOPTERS_ONLY || acc_planes == ALL) {
bjarni@4800: 			WP(w, buildvehicle_d).show_engine_button = 3;
bjarni@4800: 		} else {
bjarni@4800: 			DisableWindowWidget(w, BUILD_VEHICLE_WIDGET_HELICOPTERS);
bjarni@4800: 		}
bjarni@4800: 
bjarni@4800: 		if (acc_planes == AIRCRAFT_ONLY || acc_planes == ALL) {
bjarni@4800: 			/* Set the start clicked button to jets if the list isn't empty. If not, then show propeller planes */
bjarni@4800: 			WP(w, buildvehicle_d).show_engine_button = WP(w, buildvehicle_d).list_b_length == 0 ? 1 : 2;
bjarni@4800: 		} else {
bjarni@4800: 			DisableWindowWidget(w, BUILD_VEHICLE_WIDGET_JETS);
bjarni@4800: 			DisableWindowWidget(w, BUILD_VEHICLE_WIDGET_PLANES);
bjarni@4800: 		}
bjarni@4800: 
bjarni@4800: 		if (WP(w, buildvehicle_d).show_engine_button == 0) {
bjarni@4800: 			/* No plane type are buildable here */
bjarni@4800: 			NOT_REACHED();
bjarni@4800: 			WP(w, buildvehicle_d).show_engine_button = 1;
bjarni@4800: 		}
bjarni@4800: 	}
bjarni@4800: }
bjarni@4800: 
bjarni@4800: void ShowBuildVehicleWindow(TileIndex tile, byte type)
bjarni@4800: {
bjarni@4800: 	Window *w;
bjarni@4800: 
bjarni@4800: 	DeleteWindowById(WC_BUILD_VEHICLE, tile);
bjarni@4800: 
bjarni@4800: 	w = AllocateWindowDescFront(&_build_vehicle_desc, tile);
bjarni@4800: 
bjarni@4800: 	if (w == NULL) return;
bjarni@4800: 
bjarni@4800: 	WP(w, buildvehicle_d).vehicle_type = type;
bjarni@4800: 
bjarni@4800: 	w->resize.step_height = GetVehicleListHeight(type);
bjarni@4800: 	w->vscroll.cap = 4;
bjarni@4800: 	w->widget[BUILD_VEHICLE_WIDGET_LIST].data = (w->vscroll.cap << 8) + 1;
bjarni@4800: 
bjarni@4800: 	if (tile != 0) {
bjarni@4800: 		w->caption_color = GetTileOwner(tile);
bjarni@4800: 	} else {
bjarni@4800: 		w->caption_color = _local_player;
bjarni@4800: 	}
bjarni@4800: 
bjarni@4800: 	WP(w, buildvehicle_d).list_a_length = 0;
bjarni@4800: 	WP(w, buildvehicle_d).list_b_length = 0;
bjarni@4800: 	WP(w, buildvehicle_d).list_c_length = 0;
bjarni@4800: 	WP(w, buildvehicle_d).list_a        = NULL;
bjarni@4800: 	WP(w, buildvehicle_d).list_b        = NULL;
bjarni@4800: 	WP(w, buildvehicle_d).list_c        = NULL;
bjarni@4800: 	WP(w, buildvehicle_d).sel_engine           = INVALID_ENGINE;
bjarni@4800: 	WP(w, buildvehicle_d).sort_criteria        = _last_sort_criteria;
bjarni@4800: 	WP(w, buildvehicle_d).decenting_sort_order = _last_sort_order;
bjarni@4800: 
bjarni@4800: 	GenerateBuildList(w);
bjarni@4800: 	switch (type) {
bjarni@4800: 		case VEH_Aircraft: CreateAircraftWindow(w); break;
bjarni@4800: 		default: NOT_REACHED();
bjarni@4800: 	}
bjarni@4800: 
bjarni@4800: 	SortAircraftBuildList(w);
bjarni@4800: }