1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "aircraft.h"

     6

#include "debug.h"

     7

#include "functions.h"

     8

#include "table/sprites.h"

     9

#include "table/strings.h"

    10

#include "window.h"

    11

#include "gui.h"

    12

#include "vehicle.h"

    13

#include "gfx.h"

    14

#include "station.h"

    15

#include "command.h"

    16

#include "engine.h"

    17

#include "player.h"

    18

#include "depot.h"

    19

#include "airport.h"

    20

#include "vehicle_gui.h"

    21

#include "newgrf_engine.h"

    22

#include "date.h"

    23

#include "strings.h"

    24

    25

    26

typedef enum BuildVehicleWidgets {

    27

	BUILD_VEHICLE_WIDGET_CLOSEBOX = 0,

    28

	BUILD_VEHICLE_WIDGET_CAPTION,

    29

	BUILD_VEHICLE_WIDGET_SORT_ASSENDING_DESCENDING,

    30

	BUILD_VEHICLE_WIDGET_SORT_TEXT,

    31

	BUILD_VEHICLE_WIDGET_SORT_DROPDOWN,

    32

	BUILD_VEHICLE_WIDGET_LIST,

    33

	BUILD_VEHICLE_WIDGET_SCROLLBAR,

    34

	BUILD_VEHICLE_WIDGET_PANEL,

    35

	BUILD_VEHICLE_WIDGET_BUILD,

    36

	BUILD_VEHICLE_WIDGET_RENAME,

    37

	BUILD_VEHICLE_WIDGET_RESIZE,

    38

} BuildVehicleWidget;

    39

    40

static const Widget _build_vehicle_widgets[] = {

    41

	{   WWT_CLOSEBOX,   RESIZE_NONE,    14,     0,    10,     0,    13, STR_00C5,                STR_018B_CLOSE_WINDOW },

    42

	{    WWT_CAPTION,   RESIZE_NONE,    14,    11,   239,     0,    13, STR_A005_NEW_AIRCRAFT,   STR_018C_WINDOW_TITLE_DRAG_THIS },

    43

	{ WWT_PUSHTXTBTN,   RESIZE_NONE,    14,     0,    80,    14,    25, STR_SORT_BY,             STR_SORT_ORDER_TIP},

    44

	{      WWT_PANEL,   RESIZE_NONE,    14,    81,   227,    14,    25, 0x0,                     STR_SORT_CRITERIA_TIP},

    45

	{    WWT_TEXTBTN,   RESIZE_NONE,    14,   228,   239,    14,    25, STR_0225,                STR_SORT_CRITERIA_TIP},

    46

	{     WWT_MATRIX, RESIZE_BOTTOM,    14,     0,   227,    26,   121, 0x401,                   STR_A025_AIRCRAFT_SELECTION_LIST },

    47

	{  WWT_SCROLLBAR, RESIZE_BOTTOM,    14,   228,   239,    26,   121, 0x0,                     STR_0190_SCROLL_BAR_SCROLLS_LIST },

    48

	{      WWT_PANEL,     RESIZE_TB,    14,     0,   239,   122,   213, 0x0,                     STR_NULL },

    49

    50

	{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,     0,   114,   214,   225, STR_A006_BUILD_AIRCRAFT, STR_A026_BUILD_THE_HIGHLIGHTED_AIRCRAFT },

    51

	{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,   115,   227,   214,   225, STR_A037_RENAME,         STR_A038_RENAME_AIRCRAFT_TYPE },

    52

	{  WWT_RESIZEBOX,     RESIZE_TB,    14,   228,   239,   214,   225, 0x0,                     STR_RESIZE_BUTTON },

    53

	{   WIDGETS_END},

    54

};

    55

    56

static bool _internal_sort_order; // descending/ascending

    57

static byte _last_sort_criteria = 0;

    58

static bool _last_sort_order = false;

    59

    60

static int CDECL EngineNumberSorter(const void *a, const void *b)

    61

{

    62

	const EngineID va = *(const EngineID*)a;

    63

	const EngineID vb = *(const EngineID*)b;

    64

	int r = va - vb;

    65

    66

	return _internal_sort_order ? -r : r;

    67

}

    68

    69

static int CDECL EngineIntroDateSorter(const void *a, const void *b)

    70

{

    71

	const int va = GetEngine(*(const EngineID*)a)->intro_date;

    72

	const int vb = GetEngine(*(const EngineID*)b)->intro_date;

    73

	const int r = va - vb;

    74

    75

	if (r == 0) {

    76

		/* Use EngineID to sort instead since we want consistent sorting */

    77

		return EngineNumberSorter(a, b);

    78

	}

    79

	return _internal_sort_order ? -r : r;

    80

}

    81

    82

static int CDECL EngineNameSorter(const void *a, const void *b)

    83

{

    84

	static EngineID last_engine[2] = { INVALID_ENGINE, INVALID_ENGINE };

    85

	static char     last_name[2][64] = { "\0", "\0" };

    86

    87

	const EngineID va = *(const EngineID*)a;

    88

	const EngineID vb = *(const EngineID*)b;

    89

	int r;

    90

    91

	if (va != last_engine[0]) {

    92

		last_engine[0] = va;

    93

		GetString(last_name[0], GetCustomEngineName(va), lastof(last_name[0]));

    94

	}

    95

    96

	if (vb != last_engine[1]) {

    97

		last_engine[1] = vb;

    98

		GetString(last_name[1], GetCustomEngineName(vb), lastof(last_name[1]));

    99

	}

   100

   101

	r = strcmp(last_name[0], last_name[1]); // sort by name

   102

   103

	if (r == 0) {

   104

		/* Use EngineID to sort instead since we want consistent sorting */

   105

		return EngineNumberSorter(a, b);

   106

	}

   107

	return _internal_sort_order ? -r : r;

   108

}

   109

   110

static int CDECL EngineReliabilitySorter(const void *a, const void *b)

   111

{

   112

	const int va = GetEngine(*(const EngineID*)a)->reliability;

   113

	const int vb = GetEngine(*(const EngineID*)b)->reliability;

   114

	const int r = va - vb;

   115

   116

	if (r == 0) {

   117

		/* Use EngineID to sort instead since we want consistent sorting */

   118

		return EngineNumberSorter(a, b);

   119

	}

   120

	return _internal_sort_order ? -r : r;

   121

}

   122

   123

/* Aircraft sorting functions */

   124

   125

static int CDECL AircraftEngineCostSorter(const void *a, const void *b)

   126

{

   127

	const int va = AircraftVehInfo(*(const EngineID*)a)->base_cost;

   128

	const int vb = AircraftVehInfo(*(const EngineID*)b)->base_cost;

   129

	int r = va - vb;

   130

   131

	return _internal_sort_order ? -r : r;

   132

}

   133

   134

static int CDECL AircraftEngineSpeedSorter(const void *a, const void *b)

   135

{

   136

	const int va = AircraftVehInfo(*(const EngineID*)a)->max_speed;

   137

	const int vb = AircraftVehInfo(*(const EngineID*)b)->max_speed;

   138

	const int r = va - vb;

   139

   140

	if (r == 0) {

   141

		/* Use EngineID to sort instead since we want consistent sorting */

   142

		return EngineNumberSorter(a, b);

   143

	}

   144

	return _internal_sort_order ? -r : r;

   145

}

   146

   147

static int CDECL AircraftEngineRunningCostSorter(const void *a, const void *b)

   148

{

   149

	const int va = AircraftVehInfo(*(const EngineID*)a)->running_cost;

   150

	const int vb = AircraftVehInfo(*(const EngineID*)b)->running_cost;

   151

	const int r = va - vb;

   152

   153

	if (r == 0) {

   154

		/* Use EngineID to sort instead since we want consistent sorting */

   155

		return EngineNumberSorter(a, b);

   156

	}

   157

	return _internal_sort_order ? -r : r;

   158

}

   159

   160

static int CDECL AircraftEngineCargoSorter(const void *a, const void *b)

   161

{

   162

	const int va = AircraftVehInfo(*(const EngineID*)a)->passenger_capacity;

   163

	const int vb = AircraftVehInfo(*(const EngineID*)b)->passenger_capacity;

   164

	const int r = va - vb;

   165

   166

	if (r == 0) {

   167

		/* Use EngineID to sort instead since we want consistent sorting */

   168

		return EngineNumberSorter(a, b);

   169

	}

   170

	return _internal_sort_order ? -r : r;

   171

}

   172

   173

static EngList_SortTypeFunction * const _aircraft_sorter[] = {

   174

	&EngineNumberSorter,

   175

	&AircraftEngineCostSorter,

   176

	&AircraftEngineSpeedSorter,

   177

	&EngineIntroDateSorter,

   178

	&EngineNameSorter,

   179

	&AircraftEngineRunningCostSorter,

   180

	&EngineReliabilitySorter,

   181

	&AircraftEngineCargoSorter,

   182

};

   183

   184

static const StringID _aircraft_sort_listing[] = {

   185

	STR_ENGINE_SORT_ENGINE_ID,

   186

	STR_ENGINE_SORT_COST,

   187

	STR_SORT_BY_MAX_SPEED,

   188

	STR_ENGINE_SORT_INTRO_DATE,

   189

	STR_SORT_BY_DROPDOWN_NAME,

   190

	STR_ENGINE_SORT_RUNNING_COST,

   191

	STR_SORT_BY_RELIABILITY,

   192

	STR_ENGINE_SORT_CARGO_CAPACITY,

   193

	INVALID_STRING_ID

   194

};

   195

   196

   197

/**

   198

* Draw the purchase info details of an aircraft at a given location.

   199

 * @param x,y location where to draw the info

   200

 * @param engine_number the engine of which to draw the info of

   201

 */

   202

void DrawAircraftPurchaseInfo(int x, int y, uint w, EngineID engine_number)

   203

{

   204

	const AircraftVehicleInfo *avi = AircraftVehInfo(engine_number);

   205

	const Engine *e = GetEngine(engine_number);

   206

	CargoID cargo;

   207

	YearMonthDay ymd;

   208

	ConvertDateToYMD(e->intro_date, &ymd);

   209

   210

	/* Purchase cost - Max speed */

   211

	SetDParam(0, avi->base_cost * (_price.aircraft_base>>3)>>5);

   212

	SetDParam(1, avi->max_speed * 128 / 10);

   213

	DrawString(x, y, STR_PURCHASE_INFO_COST_SPEED, 0);

   214

	y += 10;

   215

   216

	/* Cargo capacity */

   217

	cargo = FindFirstRefittableCargo(engine_number);

   218

	if (cargo == CT_INVALID || cargo == CT_PASSENGERS) {

   219

		SetDParam(0, avi->passenger_capacity);

   220

		SetDParam(1, avi->mail_capacity);

   221

		DrawString(x, y, STR_PURCHASE_INFO_AIRCRAFT_CAPACITY, 0);

   222

	} else {

   223

		/* Note, if the default capacity is selected by the refit capacity

   224

		* callback, then the capacity shown is likely to be incorrect. */

   225

		SetDParam(0, cargo);

   226

		SetDParam(1, AircraftDefaultCargoCapacity(cargo, engine_number));

   227

		SetDParam(2, STR_9842_REFITTABLE);

   228

		DrawString(x, y, STR_PURCHASE_INFO_CAPACITY, 0);

   229

	}

   230

	y += 10;

   231

   232

	/* Running cost */

   233

	SetDParam(0, avi->running_cost * _price.aircraft_running >> 8);

   234

	DrawString(x, y, STR_PURCHASE_INFO_RUNNINGCOST, 0);

   235

	y += 10;

   236

   237

	/* Design date - Life length */

   238

	SetDParam(0, ymd.year);

   239

	SetDParam(1, e->lifelength);

   240

	DrawString(x, y, STR_PURCHASE_INFO_DESIGNED_LIFE, 0);

   241

	y += 10;

   242

   243

	/* Reliability */

   244

	SetDParam(0, e->reliability * 100 >> 16);

   245

	DrawString(x, y, STR_PURCHASE_INFO_RELIABILITY, 0);

   246

	y += 10;

   247

   248

	/* Additional text from NewGRF */

   249

	y += ShowAdditionalText(x, y, w, engine_number);

   250

	y += ShowRefitOptionsList(x, y, w, engine_number);

   251

}

   252

   253

void DrawAircraftImage(const Vehicle *v, int x, int y, VehicleID selection)

   254

{

   255

	PalSpriteID pal = (v->vehstatus & VS_CRASHED) ? PALETTE_CRASH : GetVehiclePalette(v);

   256

	DrawSprite(GetAircraftImage(v, DIR_W) | pal, x + 25, y + 10);

   257

	if (v->subtype == 0) {

   258

		SpriteID rotor_sprite = GetCustomRotorSprite(v, true);

   259

		if (rotor_sprite == 0) rotor_sprite = SPR_ROTOR_STOPPED;

   260

		DrawSprite(rotor_sprite, x + 25, y + 5);

   261

	}

   262

	if (v->index == selection) {

   263

		DrawFrameRect(x - 1, y - 1, x + 58, y + 21, 0xF, FR_BORDERONLY);

   264

	}

   265

}

   266

   267

void CcBuildAircraft(bool success, TileIndex tile, uint32 p1, uint32 p2)

   268

{

   269

	if (success) {

   270

		const Vehicle *v = GetVehicle(_new_vehicle_id);

   271

   272

		if (v->tile == _backup_orders_tile) {

   273

			_backup_orders_tile = 0;

   274

			RestoreVehicleOrders(v, _backup_orders_data);

   275

		}

   276

		ShowAircraftViewWindow(v);

   277

	}

   278

}

   279

   280

static void GenerateBuildAircraftList(Window *w)

   281

{

   282

	EngineID eid;

   283

	buildvehicle_d *bv = &WP(w, buildvehicle_d);

   284

   285

	EngList_RemoveAll(&bv->eng_list);

   286

   287

	for (eid = AIRCRAFT_ENGINES_INDEX; eid < AIRCRAFT_ENGINES_INDEX + NUM_AIRCRAFT_ENGINES; eid++) {

   288

		if (IsEngineBuildable(eid, VEH_Aircraft)) {

   289

			const AircraftVehicleInfo *avi = AircraftVehInfo(eid);

   290

			switch (bv->filter.acc_planes) {

   291

				case HELICOPTERS_ONLY:

   292

					if (avi->subtype != 0) continue; // if not helicopter

   293

					break;

   294

   295

				case AIRCRAFT_ONLY:

   296

					if (avi->subtype == 0) continue; // if helicopter

   297

					break;

   298

   299

				case ALL:

   300

					break;

   301

			}

   302

			EngList_Add(&bv->eng_list, eid);

   303

		}

   304

	}

   305

}

   306

   307

static void GenerateBuildList(Window *w)

   308

{

   309

	buildvehicle_d *bv = &WP(w, buildvehicle_d);

   310

   311

	switch (bv->vehicle_type) {

   312

		case VEH_Aircraft:

   313

			GenerateBuildAircraftList(w);

   314

			_internal_sort_order = WP(w,buildvehicle_d).descending_sort_order;

   315

			EngList_Sort(&WP(w, buildvehicle_d).eng_list, _aircraft_sorter[WP(w,buildvehicle_d).sort_criteria]);

   316

			break;

   317

   318

		default: NOT_REACHED();

   319

	}

   320

}

   321

   323

{

   325

}

   326

   328

{

   330

}

   331

   332

static void DrawBuildAircraftWindow(Window *w)

   333

{

   334

	buildvehicle_d *bv = &WP(w, buildvehicle_d);

   335

   336

	SetWindowWidgetDisabledState(w, BUILD_VEHICLE_WIDGET_BUILD, w->window_number == 0);

   337

   338

	GenerateBuildList(w);

   339

   340

	if (bv->sel_engine != INVALID_ENGINE) {

   341

		int i;

   342

		bool found = false;

   343

		int num_planes = GetEngineArrayLength(w);

   344

		for (i = 0; i < num_planes; i++) {

   345

			if (bv->sel_engine != GetEngineArray(w)[i]) continue;

   346

			found = true;

   347

			break;

   348

		}

   349

		if (!found) bv->sel_engine = INVALID_ENGINE;

   350

	}

   351

   352

	SetVScrollCount(w, GetEngineArrayLength(w));

   353

	DrawWindowWidgets(w);

   354

   355

	{

   356

		int x = 2;

   357

		int y = 27;

   358

		EngineID selected_id = bv->sel_engine;

   359

		EngineID eid = w->vscroll.pos;

   360

		const EngineID *list = GetEngineArray(w);

   361

		uint16 list_length = GetEngineArrayLength(w);

   362

		uint16 max = min(w->vscroll.pos + w->vscroll.cap, list_length);

   363

   364

		for(; eid < max; eid++) {

   365

			const EngineID engine = list[eid];

   366

   367

			DrawString(x + 62, y + 7, GetCustomEngineName(engine), engine == selected_id ? 0xC : 0x10);

   368

			DrawAircraftEngine(x + 29, y + 10, engine, GetEnginePalette(engine, _local_player));

   369

			y += 24;

   370

		}

   371

   372

		if (selected_id != INVALID_ENGINE) {

   373

			const Widget *wi = &w->widget[BUILD_VEHICLE_WIDGET_PANEL];

   374

			DrawAircraftPurchaseInfo(x, wi->top + 1, wi->right - wi->left - 2, selected_id);

   375

		}

   376

	}

   377

	DrawString(85, 15, _aircraft_sort_listing[bv->sort_criteria], 0x10);

   378

	DoDrawString(bv->descending_sort_order ? DOWNARROW : UPARROW, 69, 15, 0x10);

   379

}

   380

   381

static void BuildAircraftClickEvent(Window *w, WindowEvent *e)

   382

{

   383

	buildvehicle_d *bv = &WP(w, buildvehicle_d);

   384

   385

	switch (e->we.click.widget) {

   386

		case BUILD_VEHICLE_WIDGET_SORT_ASSENDING_DESCENDING:

   387

			bv->descending_sort_order = !bv->descending_sort_order;

   388

			_last_sort_order = bv->descending_sort_order;

   389

			GenerateBuildList(w);

   390

			SetWindowDirty(w);

   391

			break;

   392

   393

		case BUILD_VEHICLE_WIDGET_LIST: {

   394

			uint i = (e->we.click.pt.y - 26) / 24;

   395

			if (i < w->vscroll.cap) {

   396

				i += w->vscroll.pos;

   397

				bv->sel_engine = (i < GetEngineArrayLength(w)) ? GetEngineArray(w)[i] : INVALID_ENGINE;

   398

				SetWindowDirty(w);

   399

			}

   400

			break;

   401

		}

   402

   403

		case BUILD_VEHICLE_WIDGET_SORT_TEXT: case BUILD_VEHICLE_WIDGET_SORT_DROPDOWN:/* Select sorting criteria dropdown menu */

   404

			ShowDropDownMenu(w, _aircraft_sort_listing, bv->sort_criteria, BUILD_VEHICLE_WIDGET_SORT_DROPDOWN, 0, 0);

   405

			return;

   406

   407

		case BUILD_VEHICLE_WIDGET_BUILD: {

   408

			EngineID sel_eng = bv->sel_engine;

   409

			if (sel_eng != INVALID_ENGINE) {

   410

				DoCommandP(w->window_number, sel_eng, 0, CcBuildAircraft, CMD_BUILD_AIRCRAFT | CMD_MSG(STR_A008_CAN_T_BUILD_AIRCRAFT));

   411

			}

   412

			break;

   413

		}

   414

   415

		case BUILD_VEHICLE_WIDGET_RENAME: {

   416

			EngineID sel_eng = bv->sel_engine;

   417

			if (sel_eng != INVALID_ENGINE) {

   418

				bv->rename_engine = sel_eng;

   419

				ShowQueryString(GetCustomEngineName(sel_eng), STR_A039_RENAME_AIRCRAFT_TYPE, 31, 160, w->window_class, w->window_number, CS_ALPHANUMERAL);

   420

			}

   421

			break;

   422

		}

   423

	}

   424

}

   425

   426

static void NewAircraftWndProc(Window *w, WindowEvent *e)

   427

{

   428

	buildvehicle_d *bv = &WP(w, buildvehicle_d);

   429

   430

	switch (e->event) {

   431

		case WE_INVALIDATE_DATA:

   432

			SetWindowDirty(w);

   433

			break;

   434

   435

		case WE_DESTROY:

   436

			EngList_Destroy(&bv->eng_list);

   437

			break;

   438

   439

		case WE_PAINT:

   440

			DrawBuildAircraftWindow(w);

   441

			break;

   442

   443

		case WE_CLICK:

   444

			BuildAircraftClickEvent(w, e);

   445

			break;

   446

   447

		case WE_ON_EDIT_TEXT: {

   448

			if (e->we.edittext.str[0] != '\0') {

   449

				_cmd_text = e->we.edittext.str;

   450

				DoCommandP(0, bv->rename_engine, 0, NULL, CMD_RENAME_ENGINE | CMD_MSG(STR_A03A_CAN_T_RENAME_AIRCRAFT_TYPE));

   451

			}

   452

			break;

   453

		}

   454

   455

		case WE_DROPDOWN_SELECT: /* we have selected a dropdown item in the list */

   456

			if (bv->sort_criteria != e->we.dropdown.index) {

   457

				bv->sort_criteria = e->we.dropdown.index;

   458

				_last_sort_criteria = e->we.dropdown.index;

   459

				GenerateBuildList(w);

   460

			}

   461

			SetWindowDirty(w);

   462

			break;

   463

   464

		case WE_RESIZE:

   465

			w->vscroll.cap += e->we.sizing.diff.y / 24;

   466

			w->widget[BUILD_VEHICLE_WIDGET_LIST].data = (w->vscroll.cap << 8) + 1;

   467

			break;

   468

	}

   469

}

   470

   471

static const WindowDesc _build_vehicle_desc = {

   472

	WDP_AUTO, WDP_AUTO, 240, 226,

   473

	WC_BUILD_VEHICLE,0,

   474

	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS | WDF_RESIZABLE,

   475

	_build_vehicle_widgets,

   476

	NewAircraftWndProc

   477

};

   478

   479

void ShowBuildVehicleWindow(TileIndex tile, byte type)

   480

{

   481

	buildvehicle_d *bv;

   482

	Window *w;

   483

   484

	DeleteWindowById(WC_BUILD_VEHICLE, tile);

   485

	w = AllocateWindowDescFront(&_build_vehicle_desc, tile);

   486

	if (w == NULL) return;

   487

   488

	if (tile != 0) {

   489

		w->caption_color = GetTileOwner(tile);

   490

	} else {

   491

		w->caption_color = _local_player;

   492

	}

   493

   494

	w->resize.step_height = GetVehicleListHeight(type);

   495

	w->vscroll.cap = 4;

   496

	w->widget[BUILD_VEHICLE_WIDGET_LIST].data = (w->vscroll.cap << 8) + 1;

   497

   498

	bv = &WP(w, buildvehicle_d);

   499

	EngList_Create(&bv->eng_list);

   500

	bv->sel_engine           = INVALID_ENGINE;

   501

	bv->sort_criteria        = _last_sort_criteria;

   502

	bv->descending_sort_order = _last_sort_order;

   503

   504

	bv->vehicle_type = type;

   505

   506

	switch (type) {

   507

		case VEH_Aircraft: {

   508

			byte acc_planes = GetAirport(GetStationByTile(tile)->airport_type)->acc_planes;

   509

			bv->filter.acc_planes = acc_planes;

   510

			break;

   511

		}

   512

		default: NOT_REACHED();

   513

	}

   514