1

/* $Id$ */

     2

     3

/** @file station_gui.cpp */

     4

     5

#include "stdafx.h"

     6

#include "openttd.h"

     7

#include "debug.h"

     8

#include "functions.h"

     9

#include "strings.h"

    10

#include "table/strings.h"

    11

#include "window.h"

    12

#include "gui.h"

    13

#include "station.h"

    14

#include "gfx.h"

    15

#include "player.h"

    16

#include "economy.h"

    17

#include "town.h"

    18

#include "command.h"

    19

#include "variables.h"

    20

#include "vehicle_gui.h"

    21

#include "date.h"

    22

#include "vehicle.h"

    23

#include "table/sprites.h"

    24

#include "helpers.hpp"

    25

#include "cargotype.h"

    26

    27

enum StationListWidgets {

    28

	STATIONLIST_WIDGET_CLOSEBOX = 0,

    29

	STATIONLIST_WIDGET_LIST = 3,

    30

	STATIONLIST_WIDGET_TRAIN =6,

    31

	STATIONLIST_WIDGET_TRUCK,

    32

	STATIONLIST_WIDGET_BUS,

    33

	STATIONLIST_WIDGET_AIRPLANE,

    34

	STATIONLIST_WIDGET_SHIP,

    35

	STATIONLIST_WIDGET_CARGOSTART = 20,

    36

	STATIONLIST_WIDGET_NOCARGOWAITING = 12,

    37

	STATIONLIST_WIDGET_FACILALL = 14,

    38

	STATIONLIST_WIDGET_CARGOALL,

    39

	STATIONLIST_WIDGET_SORTBY,

    40

	STATIONLIST_WIDGET_SORTCRITERIA,

    41

	STATIONLIST_WIDGET_SORTDROPBTN,

    42

};

    43

    44

typedef int CDECL StationSortListingTypeFunction(const void*, const void*);

    45

    46

static StationSortListingTypeFunction StationNameSorter;

    47

static StationSortListingTypeFunction StationTypeSorter;

    48

static StationSortListingTypeFunction StationWaitingSorter;

    49

static StationSortListingTypeFunction StationRatingMaxSorter;

    50

    51

/** Draw small boxes of cargo amount and ratings data at the given

    52

 * coordinates. If amount exceeds 576 units, it is shown 'full', same

    53

 * goes for the rating: at above 90% orso (224) it is also 'full'

    54

 * Each cargo-bar is 16 pixels wide and 6 pixels high

    55

 * Each rating 14 pixels wide and 1 pixel high and is 1 pixel below the cargo-bar

    56

 * @param x coordinate to draw the box at

    57

 * @param y coordinate to draw the box at

    58

 * @param type Cargo type

    59

 * @param amount Cargo amount

    60

 * @param rating ratings data for that particular cargo */

    61

static void StationsWndShowStationRating(int x, int y, CargoID type, uint amount, byte rating)

    62

{

    63

	const CargoSpec *cs = GetCargo(type);

    64

	if (!cs->IsValid()) return;

    65

    66

	int colour = cs->rating_colour;

    67

	uint w = (minu(amount, 576) + 5) / 36;

    68

    69

	/* Draw total cargo (limited) on station (fits into 16 pixels) */

    70

	if (w != 0) GfxFillRect(x, y, x + w - 1, y + 6, colour);

    71

    72

	/* Draw a one pixel-wide bar of additional cargo meter, useful

    73

	 * for stations with only a small amount (<=30) */

    74

	if (w == 0) {

    75

		uint rest = amount / 5;

    76

		if (rest != 0) {

    77

			w += x;

    78

			GfxFillRect(w, y + 6 - rest, w, y + 6, colour);

    79

		}

    80

	}

    81

    82

	DrawString(x + 1, y, cs->abbrev, 0x10);

    83

    84

	/* Draw green/red ratings bar (fits into 14 pixels) */

    85

	y += 8;

    86

	GfxFillRect(x + 1, y, x + 14, y, 0xB8);

    87

	rating = minu(rating,  224) / 16;

    88

	if (rating != 0) GfxFillRect(x + 1, y, x + rating, y, 0xD0);

    89

}

    90

    91

const StringID _station_sort_listing[] = {

    92

	STR_SORT_BY_DROPDOWN_NAME,

    93

	STR_SORT_BY_FACILITY,

    94

	STR_SORT_BY_WAITING,

    95

	STR_SORT_BY_RATING_MAX,

    96

	INVALID_STRING_ID

    97

};

    98

    99

static char _bufcache[64];

   100

static const Station* _last_station;

   101

static int _internal_sort_order;

   102

   103

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

   104

{

   105

	const Station* st1 = *(const Station**)a;

   106

	const Station* st2 = *(const Station**)b;

   107

	char buf1[64];

   108

	int r;

   109

   110

	SetDParam(0, st1->index);

   111

	GetString(buf1, STR_STATION, lastof(buf1));

   112

   113

	if (st2 != _last_station) {

   114

		_last_station = st2;

   115

		SetDParam(0, st2->index);

   116

		GetString(_bufcache, STR_STATION, lastof(_bufcache));

   117

	}

   118

   119

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

   120

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

   121

}

   122

   123

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

   124

{

   125

	const Station* st1 = *(const Station**)a;

   126

	const Station* st2 = *(const Station**)b;

   127

	return (_internal_sort_order & 1) ? st2->facilities - st1->facilities : st1->facilities - st2->facilities;

   128

}

   129

   130

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

   131

{

   132

	const Station* st1 = *(const Station**)a;

   133

	const Station* st2 = *(const Station**)b;

   134

	int sum1 = 0, sum2 = 0;

   135

   136

	for (CargoID j = 0; j < NUM_CARGO; j++) {

   137

		if (st1->goods[j].waiting_acceptance & 0xfff) sum1 += GetTransportedGoodsIncome(st1->goods[j].waiting_acceptance & 0xfff, 20, 50, j);

   138

		if (st2->goods[j].waiting_acceptance & 0xfff) sum2 += GetTransportedGoodsIncome(st2->goods[j].waiting_acceptance & 0xfff, 20, 50, j);

   139

	}

   140

   141

	return (_internal_sort_order & 1) ? sum2 - sum1 : sum1 - sum2;

   142

}

   143

   144

/**

   145

 * qsort-compatible version of sorting two stations by maximum rating

   146

 * @param a   First object to be sorted, must be of type (const Station *)

   147

 * @param b   Second object to be sorted, must be of type (const Station *)

   148

 * @return    The sort order

   149

 * @retval >0 a should come before b in the list

   150

 * @retval <0 b should come before a in the list

   151

 */

   152

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

   153

{

   154

	const Station* st1 = *(const Station**)a;

   155

	const Station* st2 = *(const Station**)b;

   156

	byte maxr1 = 0;

   157

	byte maxr2 = 0;

   158

   159

	for (CargoID j = 0; j < NUM_CARGO; j++) {

   160

		if (st1->goods[j].enroute_from != INVALID_STATION) maxr1 = max(maxr1, st1->goods[j].rating);

   161

		if (st2->goods[j].enroute_from != INVALID_STATION) maxr2 = max(maxr2, st2->goods[j].rating);

   162

	}

   163

   164

	return (_internal_sort_order & 1) ? maxr2 - maxr1 : maxr1 - maxr2;

   165

}

   166

   167

enum StationListFlags {

   168

	SL_ORDER   = 0x01,

   169

	SL_RESORT  = 0x02,

   170

	SL_REBUILD = 0x04,

   171

};

   172

   173

DECLARE_ENUM_AS_BIT_SET(StationListFlags);

   174

   175

struct plstations_d {

   176

	const Station** sort_list;

   177

	uint16 list_length;

   178

	byte sort_type;

   179

	StationListFlags flags;

   180

	uint16 resort_timer;  //was byte refresh_counter;

   181

};

   182

assert_compile(WINDOW_CUSTOM_SIZE >= sizeof(plstations_d));

   183

   184

void RebuildStationLists()

   185

{

   186

	Window* const *wz;

   187

   188

	FOR_ALL_WINDOWS(wz) {

   189

		Window *w = *wz;

   190

		if (w->window_class == WC_STATION_LIST) {

   191

			WP(w, plstations_d).flags |= SL_REBUILD;

   192

			SetWindowDirty(w);

   193

		}

   194

	}

   195

}

   196

   197

void ResortStationLists()

   198

{

   199

	Window* const *wz;

   200

   201

	FOR_ALL_WINDOWS(wz) {

   202

		Window *w = *wz;

   203

		if (w->window_class == WC_STATION_LIST) {

   204

			WP(w, plstations_d).flags |= SL_RESORT;

   205

			SetWindowDirty(w);

   206

		}

   207

	}

   208

}

   209

   210

static void BuildStationsList(plstations_d* sl, PlayerID owner, byte facilities, uint32 cargo_filter, bool include_empty)

   211

{

   212

	uint n = 0;

   213

	const Station *st;

   214

   215

	if (!(sl->flags & SL_REBUILD)) return;

   216

   217

	/* Create array for sorting */

   218

	const Station** station_sort = MallocT<const Station*>(GetMaxStationIndex() + 1);

   219

	if (station_sort == NULL) error("Could not allocate memory for the station-sorting-list");

   220

   221

	DEBUG(misc, 3, "Building station list for player %d", owner);

   222

   223

	FOR_ALL_STATIONS(st) {

   224

		if (st->owner == owner) {

   225

			if (facilities & st->facilities) { //only stations with selected facilities

   226

				int num_waiting_cargo = 0;

   227

				for (CargoID j = 0; j < NUM_CARGO; j++) {

   228

					if (st->goods[j].waiting_acceptance & 0xFFF) {

   229

						num_waiting_cargo++; //count number of waiting cargo

   230

						if (HASBIT(cargo_filter, j)) {

   231

							station_sort[n++] = st;

   232

							break;

   233

						}

   234

					}

   235

				}

   236

				/* stations without waiting cargo */

   237

				if (num_waiting_cargo == 0 && include_empty) {

   238

					station_sort[n++] = st;

   239

				}

   240

			}

   241

		}

   242

	}

   243

   244

	free((void*)sl->sort_list);

   245

	sl->sort_list = MallocT<const Station*>(n);

   246

	if (n != 0 && sl->sort_list == NULL) error("Could not allocate memory for the station-sorting-list");

   247

	sl->list_length = n;

   248

   249

	for (uint i = 0; i < n; ++i) sl->sort_list[i] = station_sort[i];

   250

   251

	sl->flags &= ~SL_REBUILD;

   252

	sl->flags |= SL_RESORT;

   253

	free((void*)station_sort);

   254

}

   255

   256

static void SortStationsList(plstations_d *sl)

   257

{

   258

	static StationSortListingTypeFunction* const _station_sorter[] = {

   259

		&StationNameSorter,

   260

		&StationTypeSorter,

   261

		&StationWaitingSorter,

   262

		&StationRatingMaxSorter

   263

	};

   264

   265

	if (!(sl->flags & SL_RESORT)) return;

   266

   267

	_internal_sort_order = sl->flags & SL_ORDER;

   268

	_last_station = NULL; // used for "cache" in namesorting

   269

	qsort((void*)sl->sort_list, sl->list_length, sizeof(sl->sort_list[0]), _station_sorter[sl->sort_type]);

   270

   271

	sl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;

   272

	sl->flags &= ~SL_RESORT;

   273

}

   274

   275

static const uint32 _cargo_filter_max = ~0;

   276

static uint32 _cargo_filter = _cargo_filter_max;

   277

   278

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

   279

{

   280

	const PlayerID owner = (PlayerID)w->window_number;

   281

	static byte facilities = FACIL_TRAIN | FACIL_TRUCK_STOP | FACIL_BUS_STOP | FACIL_AIRPORT | FACIL_DOCK;

   282

	static Listing station_sort = {0, 0};

   283

	static bool include_empty = true;

   284

   285

	plstations_d *sl = &WP(w, plstations_d);

   286

   287

	switch (e->event) {

   288

		case WE_CREATE: /* set up resort timer */

   289

			if (_cargo_filter == _cargo_filter_max) _cargo_filter = _cargo_mask;

   290

   291

			for (uint i = 0; i < 5; i++) {

   292

				if (HASBIT(facilities, i)) LowerWindowWidget(w, i + STATIONLIST_WIDGET_TRAIN);

   293

			}

   294

			SetWindowWidgetLoweredState(w, STATIONLIST_WIDGET_FACILALL, facilities == (FACIL_TRAIN | FACIL_TRUCK_STOP | FACIL_BUS_STOP | FACIL_AIRPORT | FACIL_DOCK));

   295

			SetWindowWidgetLoweredState(w, STATIONLIST_WIDGET_CARGOALL, _cargo_filter == _cargo_mask && include_empty);

   296

			SetWindowWidgetLoweredState(w, STATIONLIST_WIDGET_NOCARGOWAITING, include_empty);

   297

   298

			sl->sort_list = NULL;

   299

			sl->flags = SL_REBUILD;

   300

			sl->sort_type = station_sort.criteria;

   301

			if (station_sort.order) sl->flags |= SL_ORDER;

   302

			sl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;

   303

			break;

   304

   305

		case WE_PAINT: {

   306

			BuildStationsList(sl, owner, facilities, _cargo_filter, include_empty);

   307

			SortStationsList(sl);

   308

   309

			SetVScrollCount(w, sl->list_length);

   310

   311

			/* draw widgets, with player's name in the caption */

   312

			const Player* p = GetPlayer(owner);

   313

			SetDParam(0, p->name_1);

   314

			SetDParam(1, p->name_2);

   315

			SetDParam(2, w->vscroll.count);

   316

			DrawWindowWidgets(w);

   317

   318

			/* draw sorting criteria string */

   319

			DrawString(85, 26, _station_sort_listing[sl->sort_type], 0x10);

   320

			/* draw arrow pointing up/down for ascending/descending sorting */

   321

			DoDrawString(sl->flags & SL_ORDER ? DOWNARROW : UPARROW, 69, 26, 0x10);

   322

   323

			int cg_ofst;

   324

			int x = 89;

   325

			int y = 14;

   326

			int xb = 2; ///< offset from left of widget

   327

   328

			uint i = 0;

   329

			for (CargoID c = 0; c < NUM_CARGO; c++) {

   330

				const CargoSpec *cs = GetCargo(c);

   331

				if (!cs->IsValid()) continue;

   332

   333

				cg_ofst = HASBIT(_cargo_filter, c) ? 2 : 1;

   334

				GfxFillRect(x + cg_ofst, y + cg_ofst, x + cg_ofst + 10 , y + cg_ofst + 7, cs->rating_colour);

   335

				DrawStringCentered(x + 6 + cg_ofst, y + cg_ofst, cs->abbrev, 0x10);

   336

				x += 14;

   337

				i++;

   338

			}

   339

   340

			x += 6;

   341

			cg_ofst = IsWindowWidgetLowered(w, STATIONLIST_WIDGET_NOCARGOWAITING) ? 2 : 1;

   342

			DrawStringCentered(x + cg_ofst, y + cg_ofst, STR_ABBREV_NONE, 16);

   343

			x += 14;

   344

			cg_ofst = IsWindowWidgetLowered(w, STATIONLIST_WIDGET_CARGOALL) ? 2 : 1;

   345

			DrawStringCentered(x + cg_ofst, y + cg_ofst, STR_ABBREV_ALL, 16);

   346

   347

			cg_ofst = IsWindowWidgetLowered(w, STATIONLIST_WIDGET_FACILALL) ? 2 : 1;

   348

			DrawString(71 + cg_ofst, y + cg_ofst, STR_ABBREV_ALL, 16);

   349

   350

			if (w->vscroll.count == 0) { // player has no stations

   351

				DrawString(xb, 40, STR_304A_NONE, 0);

   352

				return;

   353

			}

   354

   355

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

   356

			y = 40; // start of the list-widget

   357

   358

			for (int i = w->vscroll.pos; i < max; ++i) { // do until max number of stations of owner

   359

				const Station *st = sl->sort_list[i];

   360

				int x;

   361

   362

				assert(st->xy != 0);

   363

				assert(st->owner == owner);

   364

   365

				SetDParam(0, st->index);

   366

				SetDParam(1, st->facilities);

   367

				x = DrawString(xb, y, STR_3049_0, 0) + 5;

   368

   369

				/* show cargo waiting and station ratings */

   370

				for (CargoID j = 0; j < NUM_CARGO; j++) {

   371

					uint amount = GB(st->goods[j].waiting_acceptance, 0, 12);

   372

					if (amount != 0) {

   373

						StationsWndShowStationRating(x, y, j, amount, st->goods[j].rating);

   374

						x += 20;

   375

					}

   376

				}

   377

				y += 10;

   378

			}

   379

			break;

   380

		}

   381

   382

		case WE_CLICK:

   383

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

   384

				case STATIONLIST_WIDGET_LIST: {

   385

					uint32 id_v = (e->we.click.pt.y - 41) / 10;

   386

   387

					if (id_v >= w->vscroll.cap) return; // click out of bounds

   388

   389

					id_v += w->vscroll.pos;

   390

   391

					if (id_v >= sl->list_length) return; // click out of list bound

   392

   393

					const Station *st = sl->sort_list[id_v];

   394

					assert(st->owner == owner);

   395

					ScrollMainWindowToTile(st->xy);

   396

					break;

   397

				}

   398

   399

				case STATIONLIST_WIDGET_TRAIN:

   400

				case STATIONLIST_WIDGET_TRUCK:

   401

				case STATIONLIST_WIDGET_BUS:

   402

				case STATIONLIST_WIDGET_AIRPLANE:

   403

				case STATIONLIST_WIDGET_SHIP:

   404

					if (_ctrl_pressed) {

   405

						TOGGLEBIT(facilities, e->we.click.widget - STATIONLIST_WIDGET_TRAIN);

   406

						ToggleWidgetLoweredState(w, e->we.click.widget);

   407

					} else {

   408

						for (uint i = 0; facilities != 0; i++, facilities >>= 1) {

   409

							if (HASBIT(facilities, 0)) RaiseWindowWidget(w, i + STATIONLIST_WIDGET_TRAIN);

   410

						}

   411

						SETBIT(facilities, e->we.click.widget - STATIONLIST_WIDGET_TRAIN);

   412

						LowerWindowWidget(w, e->we.click.widget);

   413

					}

   414

					SetWindowWidgetLoweredState(w, STATIONLIST_WIDGET_FACILALL, facilities == (FACIL_TRAIN | FACIL_TRUCK_STOP | FACIL_BUS_STOP | FACIL_AIRPORT | FACIL_DOCK));

   415

					sl->flags |= SL_REBUILD;

   416

					SetWindowDirty(w);

   417

					break;

   418

   419

				case STATIONLIST_WIDGET_FACILALL:

   420

					for (uint i = 0; i < 5; i++) {

   421

						LowerWindowWidget(w, i + STATIONLIST_WIDGET_TRAIN);

   422

					}

   423

					LowerWindowWidget(w, STATIONLIST_WIDGET_FACILALL);

   424

   425

					facilities = FACIL_TRAIN | FACIL_TRUCK_STOP | FACIL_BUS_STOP | FACIL_AIRPORT | FACIL_DOCK;

   426

					sl->flags |= SL_REBUILD;

   427

					SetWindowDirty(w);

   428

					break;

   429

   430

				case STATIONLIST_WIDGET_CARGOALL: {

   431

					uint i = 0;

   432

					for (CargoID c = 0; c < NUM_CARGO; c++) {

   433

						if (!GetCargo(c)->IsValid()) continue;

   434

						LowerWindowWidget(w, i + STATIONLIST_WIDGET_CARGOSTART);

   435

						i++;

   436

					}

   437

					LowerWindowWidget(w, STATIONLIST_WIDGET_NOCARGOWAITING);

   438

					LowerWindowWidget(w, STATIONLIST_WIDGET_CARGOALL);

   439

   440

					_cargo_filter = _cargo_mask;

   441

					include_empty = true;

   442

					sl->flags |= SL_REBUILD;

   443

					SetWindowDirty(w);

   444

					break;

   445

				}

   446

   447

				case STATIONLIST_WIDGET_SORTBY: /*flip sorting method asc/desc*/

   448

					sl->flags ^= SL_ORDER; //DESC-flag

   449

					station_sort.order = GB(sl->flags, 0, 1);

   450

					sl->flags |= SL_RESORT;

   451

					w->flags4 |= 5 << WF_TIMEOUT_SHL;

   452

					LowerWindowWidget(w, STATIONLIST_WIDGET_SORTBY);

   453

					SetWindowDirty(w);

   454

					break;

   455

   456

				case STATIONLIST_WIDGET_SORTCRITERIA:

   457

				case STATIONLIST_WIDGET_SORTDROPBTN: /* select sorting criteria dropdown menu */

   458

					ShowDropDownMenu(w, _station_sort_listing, sl->sort_type, STATIONLIST_WIDGET_SORTDROPBTN, 0, 0);

   459

					break;

   460

   461

				case STATIONLIST_WIDGET_NOCARGOWAITING:

   462

					if (_ctrl_pressed) {

   463

						include_empty = !include_empty;

   464

						ToggleWidgetLoweredState(w, STATIONLIST_WIDGET_NOCARGOWAITING);

   465

					} else {

   466

						for (uint i = STATIONLIST_WIDGET_CARGOSTART; i < w->widget_count; i++) {

   467

							RaiseWindowWidget(w, i);

   468

						}

   469

   470

						_cargo_filter = 0;

   471

						include_empty = true;

   472

   473

						LowerWindowWidget(w, STATIONLIST_WIDGET_NOCARGOWAITING);

   474

					}

   475

					sl->flags |= SL_REBUILD;

   476

					SetWindowWidgetLoweredState(w, STATIONLIST_WIDGET_CARGOALL, _cargo_filter == _cargo_mask && include_empty);

   477

					SetWindowDirty(w);

   478

					break;

   479

   480

				default:

   481

					if (e->we.click.widget >= STATIONLIST_WIDGET_CARGOSTART) { //change cargo_filter

   482

						/* Determine the selected cargo type */

   483

						CargoID c;

   484

						int i = 0;

   485

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

   486

							if (!GetCargo(c)->IsValid()) continue;