1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "debug.h"

     6

#include "functions.h"

     7

#include "strings.h"

     8

#include "table/strings.h"

     9

#include "window.h"

    10

#include "gui.h"

    11

#include "station.h"

    12

#include "gfx.h"

    13

#include "player.h"

    14

#include "economy.h"

    15

#include "town.h"

    16

#include "command.h"

    17

#include "variables.h"

    18

#include "vehicle_gui.h"

    19

#include "date.h"

    20

#include "vehicle.h"

    21

#include "helpers.hpp"

    22

    23

enum StationListWidgets {

    24

	STATIONLIST_WIDGET_CLOSEBOX = 0,

    25

	STATIONLIST_WIDGET_LIST = 3,

    26

	STATIONLIST_WIDGET_TRAIN =6,

    27

	STATIONLIST_WIDGET_TRUCK,

    28

	STATIONLIST_WIDGET_BUS,

    29

	STATIONLIST_WIDGET_AIRPLANE,

    30

	STATIONLIST_WIDGET_SHIP,

    31

	STATIONLIST_WIDGET_CARGOSTART = 12,

    32

	STATIONLIST_WIDGET_NOCARGOWAITING = 24,

    33

	STATIONLIST_WIDGET_FACILALL = 26,

    34

	STATIONLIST_WIDGET_CARGOALL,

    35

	STATIONLIST_WIDGET_SORTBY,

    36

	STATIONLIST_WIDGET_SORTCRITERIA,

    37

	STATIONLIST_WIDGET_SORTDROPBTN,

    38

	CARGO_ALL_SELECTED = 0x1FFF,

    39

};

    40

    41

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

    42

    43

static StationSortListingTypeFunction StationNameSorter;

    44

static StationSortListingTypeFunction StationTypeSorter;

    45

static StationSortListingTypeFunction StationWaitingSorter;

    46

static StationSortListingTypeFunction StationRatingMaxSorter;

    47

    48

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

    49

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

    50

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

    51

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

    52

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

    53

 * @param x,y X/Y coordinate to draw the box at

    54

 * @param type Cargo type

    55

 * @param amount Cargo amount

    56

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

    57

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

    58

{

    59

	int colour = _cargo_colours[type];

    60

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

    61

    62

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

    63

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

    64

    65

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

    66

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

    67

	if (w == 0) {

    68

		uint rest = amount / 5;

    69

		if (rest != 0) {

    70

			w += x;

    71

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

    72

		}

    73

	}

    74

    75

	DrawString(x + 1, y, _cargoc.names_short[type], 0x10);

    76

    77

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

    78

	y += 8;

    79

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

    80

	rating = minu(rating,  224) / 16;

    81

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

    82

}

    83

    84

const StringID _station_sort_listing[] = {

    85

	STR_SORT_BY_DROPDOWN_NAME,

    86

	STR_SORT_BY_FACILITY,

    87

	STR_SORT_BY_WAITING,

    88

	STR_SORT_BY_RATING_MAX,

    89

	INVALID_STRING_ID

    90

};

    91

    92

static char _bufcache[64];

    93

static const Station* _last_station;

    94

static int _internal_sort_order;

    95

    96

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

    97

{

    98

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

    99

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

   100

	char buf1[64];

   101

	int r;

   102

   103

	SetDParam(0, st1->index);

   104

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

   105

   106

	if (st2 != _last_station) {

   107

		_last_station = st2;

   108

		SetDParam(0, st2->index);

   109

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

   110

	}

   111

   112

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

   113

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

   114

}

   115

   116

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

   117

{

   118

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

   119

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

   120

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

   121

}

   122

   123

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

   124

{

   125

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

   126

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

   127

	int sum1 = 0, sum2 = 0;

   128

	int j;

   129

   130

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

   131

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

   132

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

   133

	}

   134

   135

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

   136

}

   137

   138

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

   139

{

   140

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

   141

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

   142

	byte maxr1 = 0;

   143

	byte maxr2 = 0;

   144

	int j;

   145

   146

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

   147

		if (st1->goods[j].waiting_acceptance & 0xfff) maxr1 = max(maxr1, st1->goods[j].rating);

   148

		if (st2->goods[j].waiting_acceptance & 0xfff) maxr2 = max(maxr2, st2->goods[j].rating);

   149

	}

   150

   151

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

   152

}

   153

   154

typedef enum StationListFlags {

   155

	SL_ORDER   = 0x01,

   156

	SL_RESORT  = 0x02,

   157

	SL_REBUILD = 0x04,

   158

} StationListFlags;

   159

   160

DECLARE_ENUM_AS_BIT_SET(StationListFlags);

   161

   162

typedef struct plstations_d {

   163

	const Station** sort_list;

   164

	uint16 list_length;

   165

	byte sort_type;

   166

	StationListFlags flags;

   167

	uint16 resort_timer;  //was byte refresh_counter;

   168

} plstations_d;

   169

assert_compile(WINDOW_CUSTOM_SIZE >= sizeof(plstations_d));

   170

   171

void RebuildStationLists(void)

   172

{

   173

	Window* const *wz;

   174

   175

	FOR_ALL_WINDOWS(wz) {

   176

		Window *w = *wz;

   177

		if (w->window_class == WC_STATION_LIST) {

   178

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

   179

			SetWindowDirty(w);

   180

		}

   181

	}

   182

}

   183

   184

void ResortStationLists(void)

   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_RESORT;

   192

			SetWindowDirty(w);

   193

		}

   194

	}

   195

}

   196

   197

static void BuildStationsList(plstations_d* sl, PlayerID owner, byte facilities, uint16 cargo_filter)

   198

{

   199

	uint n = 0;

   200

	uint i, j;

   201

	const Station** station_sort;

   202

	const Station *st;

   203

   204

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

   205

   206

	/* Create array for sorting */

   207

	MallocT(&station_sort, GetMaxStationIndex() + 1);

   208

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

   209

   210

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

   211

   212

	FOR_ALL_STATIONS(st) {

   213

		if (st->owner == owner) {

   214

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

   215

				int num_waiting_cargo = 0;

   216

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

   217

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

   218

						num_waiting_cargo++; //count number of waiting cargo

   219

						if (HASBIT(cargo_filter, j)) {

   220

							station_sort[n++] = st;

   221

							break;

   222

						}

   223

					}

   224

				}

   225

				//stations without waiting cargo

   226

				if (num_waiting_cargo == 0 && HASBIT(cargo_filter, NUM_CARGO)) {

   227

					station_sort[n++] = st;

   228

				}

   229

			}

   230

		}

   231

	}

   232

   233

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

   234

	MallocT(&sl->sort_list, n);

   235

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

   236

	sl->list_length = n;

   237

   238

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

   239

   240

	sl->flags &= ~SL_REBUILD;

   241

	sl->flags |= SL_RESORT;

   242

	free((void*)station_sort);

   243

}

   244

   245

static void SortStationsList(plstations_d *sl)

   246

{

   247

	static StationSortListingTypeFunction* const _station_sorter[] = {

   248

		&StationNameSorter,

   249

		&StationTypeSorter,

   250

		&StationWaitingSorter,

   251

		&StationRatingMaxSorter

   252

	};

   253

   254

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

   255

   256

	_internal_sort_order = sl->flags & SL_ORDER;

   257

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

   258

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

   259

   260

	sl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;

   261

	sl->flags &= ~SL_RESORT;

   262

}

   263

   264

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

   265

{

   266

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

   267

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

   268

	static uint16 cargo_filter = CARGO_ALL_SELECTED;

   269

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

   270

   271

	switch (e->event) {

   272

	case WE_CREATE: { /* set up resort timer */

   273

		int i;

   274

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

   275

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

   276

		}

   277

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

   278

			if (HASBIT(cargo_filter, i)) LowerWindowWidget(w, i + STATIONLIST_WIDGET_CARGOSTART);

   279

		}

   280

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

   281

		SetWindowWidgetLoweredState(w, STATIONLIST_WIDGET_CARGOALL, cargo_filter == CARGO_ALL_SELECTED);

   282

		SetWindowWidgetLoweredState(w, STATIONLIST_WIDGET_NOCARGOWAITING, HASBIT(cargo_filter, STATIONLIST_WIDGET_NOCARGOWAITING - NUM_CARGO));

   283

   284

		sl->sort_list = NULL;

   285

		sl->flags = SL_REBUILD;

   286

		sl->sort_type = 0;

   287

		sl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;

   288

		break;

   289

	}

   290

   291

	case WE_PAINT: {

   292

		BuildStationsList(sl, owner, facilities, cargo_filter);

   293

		SortStationsList(sl);

   294

   295

		SetVScrollCount(w, sl->list_length);

   296

   297

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

   298

		{

   299

			const Player* p = GetPlayer(owner);

   300

			SetDParam(0, p->name_1);

   301

			SetDParam(1, p->name_2);

   302

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

   303

			DrawWindowWidgets(w);

   304

		}

   305

   306

		{

   307

			int max;

   308

			int i, cg_ofst;

   309

			int x = 0, y = 0, xb = 2; // offset from top of widget

   310

   311

			/* draw sorting criteria string */

   312

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

   313

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

   314

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

   315

   316

   317

			x = 89;

   318

			y = 14;

   319

   320

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

   321

				cg_ofst = IsWindowWidgetLowered(w, i + STATIONLIST_WIDGET_CARGOSTART) ? 2 : 1;

   322

   323

				GfxFillRect(x + cg_ofst, y + cg_ofst, x + cg_ofst + 10 , y + cg_ofst + 7, _cargo_colours[i]);

   324

				DrawStringCentered(x + 6 + cg_ofst, y + cg_ofst, _cargoc.names_short[i], 0x10);

   325

				x += 14;

   326

			}

   327

   328

			x += 6;

   329

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

   330

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

   331

			x += 14;

   332

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

   333

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

   334

   335

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

   336

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

   337

   338

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

   339

				DrawString(xb, 40, STR_304A_NONE, 0);

   340

				return;

   341

			}

   342

   343

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

   344

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

   345

   346

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

   347

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

   348

				CargoID j;

   349

				int x;

   350

   351

				assert(st->xy != 0);

   352

				assert(st->owner == owner);

   353

   354

				SetDParam(0, st->index);

   355

				SetDParam(1, st->facilities);

   356

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

   357

   358

				// show cargo waiting and station ratings

   359

				for (j = 0; j != NUM_CARGO; j++) {

   360

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

   361

   362

					if (amount != 0) {

   363

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

   364

						x += 20;

   365

					}

   366

				}

   367

				y += 10;

   368

			}

   369

		}

   370

	} break;

   371

   372

	case WE_CLICK: {

   373

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

   374

		case STATIONLIST_WIDGET_LIST: {

   375

			const Station* st;

   376

   377

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

   378

   379

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

   380

   381

			id_v += w->vscroll.pos;

   382

   383

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

   384

   385

			st = sl->sort_list[id_v];

   386

			assert(st->owner == owner);

   387

			ScrollMainWindowToTile(st->xy);

   388

			break;

   389

		}

   390

   391

		case STATIONLIST_WIDGET_TRAIN:

   392

		case STATIONLIST_WIDGET_TRUCK:

   393

		case STATIONLIST_WIDGET_BUS:

   394

		case STATIONLIST_WIDGET_AIRPLANE:

   395

		case STATIONLIST_WIDGET_SHIP:

   396

			if (_ctrl_pressed) {

   397

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

   398

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

   399

			} else {

   400

				int i;

   401

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

   402

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

   403

				}

   404

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

   405

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

   406

			}

   407

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

   408

			sl->flags |= SL_REBUILD;

   409

			SetWindowDirty(w);

   410

		break;

   411

   412

		case STATIONLIST_WIDGET_FACILALL: {

   413

			int i;

   414

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

   415

				LowerWindowWidget(w, i + STATIONLIST_WIDGET_TRAIN);

   416

			}

   417

			LowerWindowWidget(w, STATIONLIST_WIDGET_FACILALL);

   418

   419

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

   420

			sl->flags |= SL_REBUILD;

   421

			SetWindowDirty(w);

   422

			break;

   423

		}

   424

		case STATIONLIST_WIDGET_CARGOALL: {

   425

			int i;

   426

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

   427

				LowerWindowWidget(w, i + STATIONLIST_WIDGET_CARGOSTART);

   428

			}

   429

			LowerWindowWidget(w, STATIONLIST_WIDGET_NOCARGOWAITING);

   430

			LowerWindowWidget(w, STATIONLIST_WIDGET_CARGOALL);

   431

   432

			cargo_filter = CARGO_ALL_SELECTED;

   433

			sl->flags |= SL_REBUILD;

   434

			SetWindowDirty(w);

   435

			break;

   436

		}

   437

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

   438

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

   439

			sl->flags |= SL_RESORT;

   440

			w->flags4 |= 5 << WF_TIMEOUT_SHL;

   441

			LowerWindowWidget(w, STATIONLIST_WIDGET_SORTBY);

   442

			SetWindowDirty(w);

   443

		break;

   444

   445

		case STATIONLIST_WIDGET_SORTCRITERIA:

   446

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

   447

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

   448

		break;

   449

   450

		default:

   451

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

   452

				if (_ctrl_pressed) {

   453

					TOGGLEBIT(cargo_filter, e->we.click.widget - STATIONLIST_WIDGET_CARGOSTART);

   454

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

   455

				} else {

   456

					int i;

   457

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

   458

						if (HASBIT(cargo_filter, 0)) RaiseWindowWidget(w, i + STATIONLIST_WIDGET_CARGOSTART);

   459

					}

   460

					SETBIT(cargo_filter, e->we.click.widget - STATIONLIST_WIDGET_CARGOSTART);

   461

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

   462

				}

   463

				sl->flags |= SL_REBUILD;

   464

				SetWindowWidgetLoweredState(w, STATIONLIST_WIDGET_CARGOALL, cargo_filter == CARGO_ALL_SELECTED);

   465

				SetWindowDirty(w);

   466

			}

   467

		}

   468

	} break;

   469

   470

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

   471

		if (sl->sort_type != e->we.dropdown.index) {

   472

			// value has changed -> resort

   473

			sl->sort_type = e->we.dropdown.index;

   474

			sl->flags |= SL_RESORT;

   475

		}

   476

		SetWindowDirty(w);

   477

		break;

   478

   479

	case WE_TICK:

   480

		if (--sl->resort_timer == 0) {

   481

			DEBUG(misc, 3, "Periodic rebuild station list player %d", owner);

   482

			sl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;

   483

			sl->flags |= SL_REBUILD;

   484

			SetWindowDirty(w);

   485

		}

   486