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:

    50

			case WC_ROADVEH_LIST:

    51

			case WC_SHIPS_LIST:

    52

			case WC_AIRCRAFT_LIST:

    53

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

    54

				SetWindowDirty(w);

    55

				break;

    56

    57

			default:

    58

				break;

    59

		}

    60

}

    61

    62

void ResortVehicleLists(void)

    63

{

    64

	Window *w;

    65

    66

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

    67

		switch (w->window_class) {

    68

			case WC_TRAINS_LIST:

    69

			case WC_ROADVEH_LIST:

    70

			case WC_SHIPS_LIST:

    71

			case WC_AIRCRAFT_LIST:

    72

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

    73

				SetWindowDirty(w);

    74

				break;

    75

    76

			default:

    77

				break;

    78

		}

    79

}

    80

    81

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

    82

{

    83

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

    84

	int n = 0;

    85

	int i;

    86

    87

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

    88

    89

	/* Create array for sorting */

    90

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

    91

	if (_vehicle_sort == NULL)

    92

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

    93

    94

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

    95

		owner, station);

    96

    97

	if (station != -1) {

    98

		const Vehicle *v;

    99

		FOR_ALL_VEHICLES(v) {

   100

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

   101

				const Order *ord;

   102

				for (ord = v->schedule_ptr; ord->type != OT_NOTHING; ++ord)

   103

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

   104

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

   105

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

   106

						++n;

   107

						break;

   108

					}

   109

			}

   110

		}

   111

	} else {

   112

		const Vehicle *v;

   113

		FOR_ALL_VEHICLES(v) {

   114

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

   115

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

   116

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

   117

				++n;

   118

			}

   119

		}

   120

	}

   121

   122

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

   123

	if (vl->sort_list == NULL)

   124

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

   125

	vl->list_length = n;

   126

   127

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

   128

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

   129

   130

	vl->flags &= ~VL_REBUILD;

   131

	vl->flags |= VL_RESORT;

   132

}

   133

   134

void SortVehicleList(vehiclelist_d *vl)

   135

{

   136

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

   137

   138

	_internal_sort_order = vl->flags & VL_DESC;

   139

	_internal_name_sorter_id = STR_SV_TRAIN_NAME;

   140

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

   141

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

   142

		_vehicle_sorter[vl->sort_type]);

   143

   144

	vl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;

   145

	vl->flags &= ~VL_RESORT;

   146

}

   147

   148

   149

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

   150

void InitializeVehiclesGuiList()

   151

{

   152

}

   153

   154

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

   155

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

   156

{

   157

	uint32 ormod;

   158

   159

	// draw profit-based colored icons

   160

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

   161

		ormod = 0x3158000; // grey

   162

	else if(v->profit_last_year < 0)

   163

		ormod = 0x30b8000; //red

   164

	else if(v->profit_last_year < 10000)

   165

		ormod = 0x30a8000; // yellow

   166

	else

   167

		ormod = 0x30d8000; // green

   168

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

   169

}

   170

   171

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

   172

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

   173

{

   174

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

   175

}

   176

   177

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

   178

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

   179

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

   180

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

   181

*    STR_SV_TRAIN_NAME for trains or STR_SV_AIRCRAFT_NAME for aircraft

   182

*/

   183

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

   184

{

   185

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

   186

}

   187

   188

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

   189

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

   190

   191

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

   192

{

   193

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

   194

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

   195

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

   196

   197

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

   198

}

   199

   200

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

   201

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

   202

{

   203

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

   204

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

   205

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

   206

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

   207

	char buf1[64] = "\0";

   208

	int r;

   209

   210

	if (va->string_id != _internal_name_sorter_id) {

   211

		SetDParam(0, va->string_id);

   212

		GetString(buf1, STR_0315);

   213

	}

   214

   215

	if ( cmp2->index != _last_vehicle_idx) {

   216

		_last_vehicle_idx = cmp2->index;

   217

		_bufcache[0] = '\0';

   218

		if (vb->string_id != _internal_name_sorter_id) {

   219

			SetDParam(0, vb->string_id);

   220

			GetString(_bufcache, STR_0315);

   221

		}

   222

	}

   223

   224

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

   225

   226

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   227

   228

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

   229

}

   230

   231

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

   232

{

   233

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

   234

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

   235

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

   236

   237

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   238

   239

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

   240

}

   241

   242

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

   243

{

   244

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

   245

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

   246

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

   247

   248

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   249

   250

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

   251

}

   252

   253

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

   254

{

   255

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

   256

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

   257

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

   258

   259

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   260

   261

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

   262

}

   263

   264

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

   265

{

   266

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

   267

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

   268

	const Vehicle *v;

   269

	int r = 0;

   270

	int i;

   271

	uint _cargo_counta[NUM_CARGO];

   272

	uint _cargo_countb[NUM_CARGO];

   273

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

   274

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

   275

   276

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

   277

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

   278

   279

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

   280

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

   281

   282

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

   283

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

   284

		if (r != 0)

   285

			break;

   286

	}

   287

   288

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   289

   290

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

   291

}

   292

   293

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

   294

{

   295

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

   296

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

   297

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

   298

   299

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   300

   301

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

   302

}

   303

   304

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

   305

{

   306

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

   307

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

   308

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

   309

   310

	VEHICLEUNITNUMBERSORTER(r, va, vb);

   311

   312

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

   313

}

   314

   315

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

   316

// ENGINE_AVAILABLE is used in ReplaceVehicleWndProc

   317

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

   318

   319

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

   320

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

   321

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

   322

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

   323

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

   324

{

   325

	int i;

   326

	byte colour;

   327

   328

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

   329

		const Engine *e = DEREF_ENGINE(i);

   330

		const RailVehicleInfo *rvi = RailVehInfo(i);

   331

		const EngineInfo *info = &_engine_info[i];

   332

   333

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

   334

			continue;

   335

   336

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

   337

   338

   339

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

   340

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

   341

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

   342

				!HASBIT(e->player_avail, _local_player))

   343

				continue;

   344

		} /*else {

   345

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

   346

			if ( _autoreplace_array[i] != i )

   347

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

   348

		} */

   349

   350

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

   351

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

   352

				colour);

   353

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

   354

				SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));

   355

			*y += 14;

   356

		}

   357

		--*sel;

   358

	}

   359

}

   360

   361

   362

static void SetupScrollStuffForReplaceWindow(Window *w)

   363

{

   364

	byte railtype;

   365

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

   366

	int sel[2];

   367

	int count = 0;

   368

	int count2 = 0;

   369

	int engine_id;

   370

   371

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

   372

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

   373

   374

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

   375

		case VEH_Train: {

   376

			railtype = WP(w,replaceveh_d).railtype;

   377

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

   378

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

   379

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

   380

				const Engine *e = DEREF_ENGINE(engine_id);

   381

				const EngineInfo *info = &_engine_info[engine_id];

   382

   383

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

   384

					count++;

   385

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

   386

					sel[0]--;

   387

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

   388

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

   389

							count2++;

   390

							sel[1]--;

   391

						}

   392

					}

   393

				}

   394

			break;

   395

			}

   396

		case VEH_Road: {

   397

			int num = NUM_ROAD_ENGINES;

   398

			Engine *e = DEREF_ENGINE(ROAD_ENGINES_INDEX);

   399

			byte cargo;

   400

			EngineInfo *info;

   401

			engine_id = ROAD_ENGINES_INDEX;

   402

   403

			do {

   404

				info = &_engine_info[engine_id];

   405

				if (ENGINE_AVAILABLE) {

   406

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

   407

					count++;

   408

					sel[0]--;

   409

				}

   410

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

   411

   412

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

   413

				num = NUM_ROAD_ENGINES;

   414

				engine_id = ROAD_ENGINES_INDEX;

   415

				e = DEREF_ENGINE(ROAD_ENGINES_INDEX);

   416

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

   417

   418

				do {

   419

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

   420

						count2++;

   421

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

   422

						sel[1]--;

   423

					}

   424

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

   425

			}

   426

			break;

   427

		}

   428

   429

		case VEH_Ship: {

   430

			int num = NUM_SHIP_ENGINES;

   431

			Engine *e = DEREF_ENGINE(SHIP_ENGINES_INDEX);

   432

			byte cargo, refittable;

   433

			EngineInfo *info;

   434

			engine_id = SHIP_ENGINES_INDEX;

   435

   436

			do {

   437

				info = &_engine_info[engine_id];

   438

				if (ENGINE_AVAILABLE) {

   439

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

   440

					count++;

   441

					sel[0]--;

   442

				}

   443

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

   444

   445

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

   446

				num = NUM_SHIP_ENGINES;

   447

				e = DEREF_ENGINE(SHIP_ENGINES_INDEX);

   448

				engine_id = SHIP_ENGINES_INDEX;

   449

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

   450

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

   451

   452

				do {

   453

					if (HASBIT(e->player_avail, _local_player)

   454

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

   455

   456

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

   457

						sel[1]--;

   458

						count2++;

   459

					}

   460

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

   461

			}

   462

			break;

   463

		}   //end of ship

   464

   465

		case VEH_Aircraft:{

   466

			int num = NUM_AIRCRAFT_ENGINES;

   467

			byte subtype;

   468

			Engine *e = DEREF_ENGINE(AIRCRAFT_ENGINES_INDEX);

   469

			EngineInfo *info;

   470

			engine_id = AIRCRAFT_ENGINES_INDEX;

   471

   472

			do {

   473

				info = &_engine_info[engine_id];

   474

				if (ENGINE_AVAILABLE) {

   475

					count++;

   476

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

   477

					sel[0]--;

   478

				}

   479

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

   480

   481

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

   482

				num = NUM_AIRCRAFT_ENGINES;

   483

				e = DEREF_ENGINE(AIRCRAFT_ENGINES_INDEX);

   484

				subtype = AircraftVehInfo(selected_id[0])->subtype;

   485

				engine_id = AIRCRAFT_ENGINES_INDEX;

   486

				do {

   487

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

   488

						if ( (subtype && AircraftVehInfo(engine_id)->subtype) || (!(subtype) && !AircraftVehInfo(engine_id)->subtype) ) {

   489

							count2++;

   490

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

   491

							sel[1]--;