1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "table/strings.h"

     4

#include "vehicle.h"

     5

#include "window.h"

     6

#include "engine.h"

     7

#include "gui.h"

     8

#include "command.h"

     9

#include "gfx.h"

    10

    11

VehicleSortListingTypeFunctions * const _vehicle_sorter[] = {

    12

	&VehicleUnsortedSorter,

    13

	&VehicleNumberSorter,

    14

	&VehicleNameSorter,

    15

	&VehicleAgeSorter,

    16

	&VehicleProfitThisYearSorter,

    17

	&VehicleProfitLastYearSorter,

    18

	&VehicleCargoSorter,

    19

	&VehicleReliabilitySorter,

    20

	&VehicleMaxSpeedSorter

    21

};

    22

    23

const StringID _vehicle_sort_listing[] = {

    24

	STR_SORT_BY_UNSORTED,

    25

	STR_SORT_BY_NUMBER,

    26

	STR_SORT_BY_DROPDOWN_NAME,

    27

	STR_SORT_BY_AGE,

    28

	STR_SORT_BY_PROFIT_THIS_YEAR,

    29

	STR_SORT_BY_PROFIT_LAST_YEAR,

    30

	STR_SORT_BY_TOTAL_CAPACITY_PER_CARGOTYPE,

    31

	STR_SORT_BY_RELIABILITY,

    32

	STR_SORT_BY_MAX_SPEED,

    33

	INVALID_STRING_ID

    34

};

    35

    36

const StringID _rail_types_list[] = {

    37

	STR_RAIL_VEHICLES,

    38

	STR_MONORAIL_VEHICLES,

    39

	STR_MAGLEV_VEHICLES,

    40

	INVALID_STRING_ID

    41

};

    42

    43

void RebuildVehicleLists(void)

    44

{

    45

	Window *w;

    46

    47

	for (w = _windows; w != _last_window; ++w)

    48

		switch (w->window_class) {

    49

		case WC_TRAINS_LIST: case WC_ROADVEH_LIST:

    50

		case WC_SHIPS_LIST:  case WC_AIRCRAFT_LIST:

    51

			WP(w, vehiclelist_d).flags |= VL_REBUILD;

    52

			SetWindowDirty(w);

    53

			break;

    54

		default: break;

    55

		}

    56

}

    57

    58

void ResortVehicleLists(void)

    59

{

    60

	Window *w;

    61

    62

	for (w = _windows; w != _last_window; ++w)

    63

		switch (w->window_class) {

    64

		case WC_TRAINS_LIST: case WC_ROADVEH_LIST:

    65

		case WC_SHIPS_LIST:  case WC_AIRCRAFT_LIST:

    66

			WP(w, vehiclelist_d).flags |= VL_RESORT;

    67

			SetWindowDirty(w);

    68

			break;

    69

		default: break;

    70

		}

    71

}

    72

    73

void BuildVehicleList(vehiclelist_d *vl, int type, int owner, int station)

    74

{

    75

	int subtype = (type != VEH_Aircraft) ? TS_Front_Engine : 2;

    76

	int n = 0;

    77

	int i;

    78

    79

	if (!(vl->flags & VL_REBUILD)) return;

    80

    81

	/* Create array for sorting */

    82

	_vehicle_sort = realloc(_vehicle_sort, _vehicles_size * sizeof(_vehicle_sort[0]));

    83

	if (_vehicle_sort == NULL)

    84

		error("Could not allocate memory for the vehicle-sorting-list");

    85

    86

	DEBUG(misc, 1) ("Building vehicle list for player %d station %d...",

    87

		owner, station);

    88

    89

	if (station != -1) {

    90

		const Vehicle *v;

    91

		FOR_ALL_VEHICLES(v) {

    92

			if (v->type == type && v->subtype <= subtype) {

    93

				const Order *order;

    94

    95

				FOR_VEHICLE_ORDERS(v, order) {

    96

					if (order->type == OT_GOTO_STATION && order->station == station) {

    97

						_vehicle_sort[n].index = v->index;

    98

						_vehicle_sort[n].owner = v->owner;

    99

						++n;

   100

						break;

   101

					}

   102

				}

   103

			}

   104

		}

   105

	} else {

   106

		const Vehicle *v;

   107

		FOR_ALL_VEHICLES(v) {

   108

			if (v->type == type && v->subtype <= subtype && v->owner == owner) {

   109

				_vehicle_sort[n].index = v->index;

   110

				_vehicle_sort[n].owner = v->owner;

   111

				++n;

   112

			}

   113

		}

   114

	}

   115

   116

	vl->sort_list = realloc(vl->sort_list, n * sizeof(vl->sort_list[0]));

   117

	if (n!=0 && vl->sort_list == NULL)

   118

		error("Could not allocate memory for the vehicle-sorting-list");

   119

	vl->list_length = n;

   120

   121

	for (i = 0; i < n; ++i)

   122

		vl->sort_list[i] = _vehicle_sort[i];

   123

   124

	vl->flags &= ~VL_REBUILD;

   125

	vl->flags |= VL_RESORT;

   126

}

   127

   128

void SortVehicleList(vehiclelist_d *vl)

   129

{

   130

	if (!(vl->flags & VL_RESORT)) return;

   131

   132

	_internal_sort_order = vl->flags & VL_DESC;

   133

	_internal_name_sorter_id = STR_SV_TRAIN_NAME;

   134

	_last_vehicle_idx = 0; // used for "cache" in namesorting

   135

	qsort(vl->sort_list, vl->list_length, sizeof(vl->sort_list[0]),

   136

		_vehicle_sorter[vl->sort_type]);

   137

   138

	vl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;

   139

	vl->flags &= ~VL_RESORT;

   140

}

   141

   142

   143

/* General Vehicle GUI based procedures that are independent of vehicle types */

   144

void InitializeVehiclesGuiList(void)

   145

{

   146

}

   147

   148

// draw the vehicle profit button in the vehicle list window.

   149

void DrawVehicleProfitButton(Vehicle *v, int x, int y)

   150

{

   151

	uint32 ormod;

   152

   153

	// draw profit-based colored icons

   154

	if(v->age <= 365 * 2)

   155

		ormod = 0x3158000; // grey

   156

	else if(v->profit_last_year < 0)

   157

		ormod = 0x30b8000; //red

   158

	else if(v->profit_last_year < 10000)

   159

		ormod = 0x30a8000; // yellow

   160

	else

   161

		ormod = 0x30d8000; // green

   162

	DrawSprite((SPR_BLOT) | ormod, x, y);

   163

}

   164

   165

/************ Sorter functions *****************/

   166

int CDECL GeneralOwnerSorter(const void *a, const void *b)

   167

{

   168

	return (*(const SortStruct*)a).owner - (*(const SortStruct*)b).owner;

   169

}

   170

   171

/* Variables you need to set before calling this function!

   172

* 1. (byte)_internal_sort_type:					sorting criteria to sort on

   173

* 2. (bool)_internal_sort_order:				sorting order, descending/ascending

   174

* 3. (uint32)_internal_name_sorter_id:	default StringID of the vehicle when no name is set. eg

   175

*    STR_SV_TRAIN_NAME for trains or STR_SV_AIRCRAFT_NAME for aircraft

   176

*/

   177

int CDECL VehicleUnsortedSorter(const void *a, const void *b)

   178

{

   179

	return GetVehicle((*(const SortStruct*)a).index)->index - GetVehicle((*(const SortStruct*)b).index)->index;

   180

}

   181

   182

// if the sorting criteria had the same value, sort vehicle by unitnumber

   183

#define VEHICLEUNITNUMBERSORTER(r, a, b) {if (r == 0) {r = a->unitnumber - b->unitnumber;}}

   184

   185

int CDECL VehicleNumberSorter(const void *a, const void *b)

   186

{

   187

	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);

   188

	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);

   189

	int r = va->unitnumber - vb->unitnumber;

   190

   191

	return (_internal_sort_order & 1) ? -r : r;

   192

}

   193

   194

static char _bufcache[64];	// used together with _last_vehicle_idx to hopefully speed up stringsorting

   195

int CDECL VehicleNameSorter(const void *a, const void *b)

   196

{

   197

	const SortStruct *cmp1 = (const SortStruct*)a;

   198

	const SortStruct *cmp2 = (const SortStruct*)b;

   199

	const Vehicle *va = GetVehicle(cmp1->index);

   200

	const Vehicle *vb = GetVehicle(cmp2->index);

   201

	char buf1[64] = "\0";

   202

	int r;

   203

   204

	if (va->string_id != _internal_name_sorter_id) {

   205

		SetDParam(0, va->string_id);

   206

		GetString(buf1, STR_JUST_STRING);

   207

	}

   208

   209

	if ( cmp2->index != _last_vehicle_idx) {

   210

		_last_vehicle_idx = cmp2->index;

   211

		_bufcache[0] = '\0';

   212

		if (vb->string_id != _internal_name_sorter_id) {

   213

			SetDParam(0, vb->string_id);

   214

			GetString(_bufcache, STR_JUST_STRING);

   215

		}

   216

	}

   217

   218

	r =  strcmp(buf1, _bufcache);	// sort by name

   219

   220

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   221

   222

	return (_internal_sort_order & 1) ? -r : r;

   223

}

   224

   225

int CDECL VehicleAgeSorter(const void *a, const void *b)

   226

{

   227

	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);

   228

	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);

   229

	int r = va->age - vb->age;

   230

   231

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   232

   233

	return (_internal_sort_order & 1) ? -r : r;

   234

}

   235

   236

int CDECL VehicleProfitThisYearSorter(const void *a, const void *b)

   237

{

   238

	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);

   239

	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);

   240

	int r = va->profit_this_year - vb->profit_this_year;

   241

   242

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   243

   244

	return (_internal_sort_order & 1) ? -r : r;

   245

}

   246

   247

int CDECL VehicleProfitLastYearSorter(const void *a, const void *b)

   248

{

   249

	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);

   250

	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);

   251

	int r = va->profit_last_year - vb->profit_last_year;

   252

   253

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   254

   255

	return (_internal_sort_order & 1) ? -r : r;

   256

}

   257

   258

int CDECL VehicleCargoSorter(const void *a, const void *b)

   259

{

   260

	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);

   261

	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);

   262

	const Vehicle *v;

   263

	int r = 0;

   264

	int i;

   265

	uint _cargo_counta[NUM_CARGO];

   266

	uint _cargo_countb[NUM_CARGO];

   267

	memset(_cargo_counta, 0, sizeof(_cargo_counta));

   268

	memset(_cargo_countb, 0, sizeof(_cargo_countb));

   269

   270

	for (v = va; v != NULL; v = v->next)

   271

		_cargo_counta[v->cargo_type] += v->cargo_cap;

   272

   273

	for (v = vb; v != NULL; v = v->next)

   274

		_cargo_countb[v->cargo_type] += v->cargo_cap;

   275

   276

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

   277

		r = _cargo_counta[i] - _cargo_countb[i];

   278

		if (r != 0)

   279

			break;

   280

	}

   281

   282

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   283

   284

	return (_internal_sort_order & 1) ? -r : r;

   285

}

   286

   287

int CDECL VehicleReliabilitySorter(const void *a, const void *b)

   288

{

   289

	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);

   290

	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);

   291

	int r = va->reliability - vb->reliability;

   292

   293

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   294

   295

	return (_internal_sort_order & 1) ? -r : r;

   296

}

   297

   298

int CDECL VehicleMaxSpeedSorter(const void *a, const void *b)

   299

{

   300

	const Vehicle *va = GetVehicle((*(const SortStruct*)a).index);

   301

	const Vehicle *vb = GetVehicle((*(const SortStruct*)b).index);

   302

	int max_speed_a = 0xFFFF, max_speed_b = 0xFFFF;

   303

	int r;

   304

	const Vehicle *ua = va, *ub = vb;

   305

	do {

   306

		if (RailVehInfo(ua->engine_type)->max_speed != 0)

   307

			max_speed_a = min(max_speed_a, RailVehInfo(ua->engine_type)->max_speed);

   308

	} while ((ua = ua->next) != NULL);

   309

   310

	do {

   311

		if (RailVehInfo(ub->engine_type)->max_speed != 0)

   312

			max_speed_b = min(max_speed_b, RailVehInfo(ub->engine_type)->max_speed);

   313

	} while ((ub = ub->next) != NULL);

   314

   315

	r = max_speed_a - max_speed_b;

   316

   317

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   318

   319

	return (_internal_sort_order & 1) ? -r : r;

   320

}

   321

   322

// this define is to match engine.c, but engine.c keeps it to itself

   323

// ENGINE_AVAILABLE is used in ReplaceVehicleWndProc

   324

#define ENGINE_AVAILABLE ((e->flags & 1 && HASBIT(info->railtype_climates, _opt.landscape)) || HASBIT(e->player_avail, _local_player))

   325

   326

/*  if show_outdated is selected, it do not sort psudo engines properly but it draws all engines

   327

 *	if used compined with show_cars set to false, it will work as intended. Replace window do it like that

   328

 *  this was a big hack even before show_outdated was added. Stupid newgrf :p										*/

   329

static void train_engine_drawing_loop(int *x, int *y, int *pos, int *sel, int *selected_id, byte railtype,

   330

	uint8 lines_drawn, bool is_engine, bool show_cars, bool show_outdated)

   331

{

   332

	int i;

   333

	byte colour;

   334

   335

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

   336

		const Engine *e = DEREF_ENGINE(i);

   337

		const RailVehicleInfo *rvi = RailVehInfo(i);

   338

		const EngineInfo *info = &_engine_info[i];

   339

   340

		if ( _player_num_engines[i] == 0 && show_outdated ) continue;

   341

   342

		if ( rvi->power == 0 && !(show_cars) )   // disables display of cars (works since they do not have power)

   343

			continue;

   344

   345

		if (*sel == 0) *selected_id = i;

   346

   347

   348

		colour = *sel == 0 ? 0xC : 0x10;

   349

		if (!(ENGINE_AVAILABLE && show_outdated && RailVehInfo(i)->power && e->railtype == railtype)) {

   350

			if (e->railtype != railtype || !(rvi->flags & RVI_WAGON) != is_engine ||

   351

				!HASBIT(e->player_avail, _local_player))

   352

				continue;

   353

		} /*else {

   354

		// TODO find a nice red colour for vehicles being replaced

   355

			if ( _autoreplace_array[i] != i )

   356

				colour = *sel == 0 ? 0x44 : 0x45;

   357

		} */

   358

   359

		if (IS_INT_INSIDE(--*pos, -lines_drawn, 0)) {

   360

			DrawString(*x + 59, *y + 2, GetCustomEngineName(i),

   361

				colour);

   362

			DrawTrainEngine(*x + 29, *y + 6, i,

   363

				SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));

   364

			if ( show_outdated ) {

   365

				SetDParam(0, _player_num_engines[i]);

   366

				DrawStringRightAligned(213, *y+5, STR_TINY_BLACK, 0);

   367

			}

   368

			*y += 14;

   369

		}

   370

		--*sel;

   371

	}

   372

}

   373

   374

   375

static void SetupScrollStuffForReplaceWindow(Window *w)

   376

{

   377

	byte railtype;

   378

	int selected_id[2] = {-1,-1};

   379

	int sel[2];

   380

	int count = 0;

   381

	int count2 = 0;

   382

	int engine_id;

   383

   384

	sel[0] = WP(w,replaceveh_d).sel_index[0];

   385

	sel[1] = WP(w,replaceveh_d).sel_index[1];

   386

   387

	switch (WP(w,replaceveh_d).vehicletype) {

   388

		case VEH_Train: {

   389

			railtype = _railtype_selected_in_replace_gui;

   390

			w->widget[13].color = _player_colors[_local_player];	// sets the colour of that art thing

   391

			w->widget[16].color = _player_colors[_local_player];	// sets the colour of that art thing

   392

			for (engine_id = 0; engine_id < NUM_TRAIN_ENGINES; engine_id++) {

   393

				const Engine *e = DEREF_ENGINE(engine_id);

   394

				const EngineInfo *info = &_engine_info[engine_id];

   395

   396

				if (ENGINE_AVAILABLE && RailVehInfo(engine_id)->power && e->railtype == railtype ) {

   397

					if ( _player_num_engines[engine_id] ) {

   398

						if (sel[0]==0)  selected_id[0] = engine_id;

   399

						count++;

   400

						sel[0]--;

   401

					}

   402

					if (HASBIT(e->player_avail, _local_player)) {

   403

						if (sel[1]==0)  selected_id[1] = engine_id;

   404

						count2++;

   405

						sel[1]--;

   406

					}

   407

				}

   408

			}

   409

			break;

   410

		}

   411

		case VEH_Road: {

   412

			int num = NUM_ROAD_ENGINES;

   413

			Engine *e = DEREF_ENGINE(ROAD_ENGINES_INDEX);

   414

			byte cargo;

   415

			EngineInfo *info;

   416

			engine_id = ROAD_ENGINES_INDEX;

   417

   418

			do {

   419

				info = &_engine_info[engine_id];

   420

				if (_player_num_engines[engine_id] ) {

   421

					if (sel[0]==0)  selected_id[0] = engine_id;

   422

					count++;

   423

					sel[0]--;

   424

				}

   425

			} while (++engine_id,++e,--num);

   426

   427

			if ( selected_id[0] != -1 ) {   // only draw right array if we have anything in the left one

   428

				num = NUM_ROAD_ENGINES;

   429

				engine_id = ROAD_ENGINES_INDEX;

   430

				e = DEREF_ENGINE(ROAD_ENGINES_INDEX);

   431

				cargo = RoadVehInfo(selected_id[0])->cargo_type;

   432

   433

				do {

   434

					if ( cargo == RoadVehInfo(engine_id)->cargo_type && HASBIT(e->player_avail, _local_player)) {

   435

						count2++;

   436

						if (sel[1]==0)  selected_id[1] = engine_id;

   437

						sel[1]--;

   438

					}

   439

				} while (++engine_id,++e,--num);

   440

			}

   441

			break;

   442

		}

   443

   444

		case VEH_Ship: {

   445

			int num = NUM_SHIP_ENGINES;

   446

			Engine *e = DEREF_ENGINE(SHIP_ENGINES_INDEX);

   447

			byte cargo, refittable;

   448

			EngineInfo *info;

   449

			engine_id = SHIP_ENGINES_INDEX;

   450

   451

			do {

   452

				info = &_engine_info[engine_id];

   453

				if (_player_num_engines[engine_id] ) {

   454

					if ( sel[0] == 0 )  selected_id[0] = engine_id;

   455

					count++;

   456

					sel[0]--;

   457

				}

   458

			} while (++engine_id,++e,--num);

   459

   460

			if ( selected_id[0] != -1 ) {

   461

				num = NUM_SHIP_ENGINES;

   462

				e = DEREF_ENGINE(SHIP_ENGINES_INDEX);

   463

				engine_id = SHIP_ENGINES_INDEX;

   464

				cargo = ShipVehInfo(selected_id[0])->cargo_type;

   465

				refittable = ShipVehInfo(selected_id[0])->refittable;

   466

   467

				do {

   468

					if (HASBIT(e->player_avail, _local_player)

   469

					&& ( cargo == ShipVehInfo(engine_id)->cargo_type || refittable & ShipVehInfo(engine_id)->refittable)) {

   470

   471

						if ( sel[1]==0)  selected_id[1] = engine_id;

   472

						sel[1]--;

   473

						count2++;

   474

					}

   475

				} while (++engine_id,++e,--num);

   476

			}

   477

			break;

   478

		}   //end of ship

   479

   480

		case VEH_Aircraft:{

   481

			int num = NUM_AIRCRAFT_ENGINES;

   482

			byte subtype;

   483

			Engine *e = DEREF_ENGINE(AIRCRAFT_ENGINES_INDEX);

   484

			EngineInfo *info;

   485

			engine_id = AIRCRAFT_ENGINES_INDEX;

   486

   487

			do {

   488

				info = &_engine_info[engine_id];

   489