1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "station.h"

     4

#include "gfx.h"

     5

#include "window.h"

     6

#include "viewport.h"

     7

#include "command.h"

     8

#include "town.h"

     9

#include "vehicle.h"

    10

#include "news.h"

    11

#include "saveload.h"

    12

#include "economy.h"

    13

#include "player.h"

    14

#include "airport.h"

    15

#include "sprite.h"

    16

#include "table/directions.h"

    17

    18

// FIXME -- need to be embedded into Airport variable. Is dynamically

    19

// deducteable from graphics-tile array, so will not be needed

    20

const byte _airport_size_x[5] = {4, 6, 1, 6, 7 };

    21

const byte _airport_size_y[5] = {3, 6, 1, 6, 7 };

    22

    23

void ShowAircraftDepotWindow(uint tile);

    24

extern void UpdateAirplanesOnNewStation(Station *st);

    25

    26

static void MarkStationDirty(Station *st)

    27

{

    28

	if (st->sign.width_1 != 0) {

    29

		InvalidateWindowWidget(WC_STATION_VIEW, st->index, 1);

    30

    31

		MarkAllViewportsDirty(

    32

			st->sign.left - 6,

    33

			st->sign.top,

    34

			st->sign.left + (st->sign.width_1 << 2) + 12,

    35

			st->sign.top + 48);

    36

	}

    37

}

    38

    39

#define CHECK_STATIONS_ERR ((Station*)-1)

    40

    41

static Station *GetStationAround(uint tile, int w, int h, int closest_station)

    42

{

    43

	// check around to see if there's any stations there

    44

	BEGIN_TILE_LOOP(tile_cur, w + 2, h + 2, tile - TILE_XY(1,1))

    45

		if (IS_TILETYPE(tile_cur, MP_STATION)) {

    46

			int t;

    47

			t = _map2[tile_cur];

    48

			{

    49

				Station *st = DEREF_STATION(t);

    50

				// you cannot take control of an oilrig!!

    51

				if (st->airport_type == AT_OILRIG && st->facilities == (FACIL_AIRPORT|FACIL_DOCK))

    52

					continue;

    53

			}

    54

    55

			if (closest_station == -1) {

    56

				closest_station = t;

    57

			} else if (closest_station != t) {

    58

				_error_message = STR_3006_ADJOINS_MORE_THAN_ONE_EXISTING;

    59

				return CHECK_STATIONS_ERR;

    60

			}

    61

		}

    62

	END_TILE_LOOP(tile_cur, w + 2, h + 2, tile - TILE_XY(1,1))

    63

	return (closest_station == -1) ? NULL : DEREF_STATION(closest_station);

    64

}

    65

    66

TileIndex GetStationTileForVehicle(Vehicle *v, Station *st)

    67

{

    68

	switch (v->type) {

    69

		case VEH_Train: 		return st->train_tile;

    70

		case VEH_Aircraft:	return st->airport_tile;

    71

		case VEH_Ship:			return st->dock_tile;

    72

		case VEH_Road:			return (v->cargo_type == CT_PASSENGERS) ? st->bus_tile : st->lorry_tile;

    73

		default:

    74

			assert(false);

    75

			return 0;

    76

	}

    77

}

    78

    79

static bool CheckStationSpreadOut(Station *st, uint tile, int w, int h)

    80

{

    81

	byte station_index = st->index;

    82

	uint i;

    83

	uint x1 = GET_TILE_X(tile);

    84

	uint y1 = GET_TILE_Y(tile);

    85

	uint x2 = x1 + w - 1;

    86

	uint y2 = y1 + h - 1;

    87

	uint t;

    88

    89

	for(i=0; i!=TILES_X*TILES_Y; i++) {

    90

		if (IS_TILETYPE(i, MP_STATION) && _map2[i] == station_index) {

    91

			t = GET_TILE_X(i);

    92

			if (t < x1) x1 = t;

    93

			if (t > x2) x2 = t;

    94

    95

			t = GET_TILE_Y(i);

    96

			if (t < y1) y1 = t;

    97

			if (t > y2) y2 = t;

    98

		}

    99

	}

   100

   101

	if (y2-y1 >= _patches.station_spread || x2-x1 >= _patches.station_spread) {

   102

		_error_message = STR_306C_STATION_TOO_SPREAD_OUT;

   103

		return false;

   104

	}

   105

   106

	return true;

   107

}

   108

   109

static Station *AllocateStation()

   110

{

   111

	Station *st, *a_free = NULL;

   112

	int count;

   113

	int num_free = 0;

   114

	int i;

   115

   116

	for(st = _stations, count = lengthof(_stations); count != 0; count--, st++) {

   117

		if (st->xy == 0) {

   118

			num_free++;

   119

			if (a_free==NULL)

   120

				a_free = st;

   121

		}

   122

	}

   123

   124

	if (a_free == NULL ||

   125

			(num_free < 30 && IS_HUMAN_PLAYER(_current_player))) {

   126

		_error_message = STR_3008_TOO_MANY_STATIONS_LOADING;

   127

		return NULL;

   128

	}

   129

   130

	i = a_free->index;

   131

	memset(a_free, 0, sizeof(Station));

   132

	a_free->index = i;

   133

	return a_free;

   134

}

   135

   136

   137

static int CountMapSquareAround(uint tile, byte type, byte min, byte max) {

   138

	static const TileIndexDiff _count_square_table[7*7+1] = {

   139

		TILE_XY(-3,-3), 1, 1, 1, 1, 1, 1,

   140

		TILE_XY(-6,1),  1, 1, 1, 1, 1, 1,

   141

		TILE_XY(-6,1),  1, 1, 1, 1, 1, 1,

   142

		TILE_XY(-6,1),  1, 1, 1, 1, 1, 1,

   143

		TILE_XY(-6,1),  1, 1, 1, 1, 1, 1,

   144

		TILE_XY(-6,1),  1, 1, 1, 1, 1, 1,

   145

		TILE_XY(-6,1),  1, 1, 1, 1, 1, 1,

   146

		0,

   147

	};

   148

	int j;

   149

	const TileIndexDiff *p = _count_square_table;

   150

	int num = 0;

   151

   152

	while ( (j=*p++) != 0 ) {

   153

		tile = TILE_MASK(tile + j);

   154

   155

		if (IS_TILETYPE(tile, type) && _map5[tile] >= min && _map5[tile] <= max)

   156

			num++;

   157

	}

   158

   159

	return num;

   160

}

   161

   162

#define M(x) ((x) - STR_SV_STNAME)

   163

   164

static bool GenerateStationName(Station *st, uint tile, int flag)

   165

{

   166

	static const uint32 _gen_station_name_bits[] = {

   167

		0,																/* 0 */

   168

		1 << M(STR_SV_STNAME_AIRPORT),		/* 1 */

   169

		1 << M(STR_SV_STNAME_OILFIELD),		/* 2 */

   170

		1 << M(STR_SV_STNAME_DOCKS),			/* 3 */

   171

		0x1FF << M(STR_SV_STNAME_BUOY_1),	/* 4 */

   172

		1 << M(STR_SV_STNAME_HELIPORT),		/* 5 */

   173

	};

   174

   175

	Town *t = st->town;

   176

	uint32 free_names = (uint32)-1;

   177

	int found;

   178

	uint z,z2;

   179

	unsigned long tmp;

   180

   181

	{

   182

		Station *s;

   183

   184

		FOR_ALL_STATIONS(s) {

   185

			if (s != st && s->xy != 0 && s->town==t) {

   186

				uint str = M(s->string_id);

   187

				if (str <= 0x20) {

   188

					if (str == M(STR_SV_STNAME_FOREST))

   189

						str = M(STR_SV_STNAME_WOODS);

   190

					CLRBIT(free_names, str);

   191

				}

   192

			}

   193

		}

   194

	}

   195

   196

	/* check default names */

   197

	tmp = free_names & _gen_station_name_bits[flag];

   198

	if (tmp != 0) {

   199

		found = FindFirstBit(tmp);

   200

		goto done;

   201

	}

   202

   203

	/* check mine? */

   204

	if (HASBIT(free_names, M(STR_SV_STNAME_MINES))) {

   205

		if (CountMapSquareAround(tile, MP_INDUSTRY, 0, 6) >= 2 ||

   206

		    CountMapSquareAround(tile, MP_INDUSTRY, 0x64, 0x73) >= 2 ||

   207

				CountMapSquareAround(tile, MP_INDUSTRY, 0x2F, 0x33) >= 2 ||

   208

				CountMapSquareAround(tile, MP_INDUSTRY, 0x48, 0x58) >= 2 ||

   209

				CountMapSquareAround(tile, MP_INDUSTRY, 0x5B, 0x63) >= 2) {

   210

					found = M(STR_SV_STNAME_MINES);

   211

					goto done;

   212

				}

   213

	}

   214

   215

	/* check close enough to town to get central as name? */

   216

	if (GetTileDist1D(tile,t->xy) < 8) {

   217

		found = M(STR_SV_STNAME);

   218

		if (HASBIT(free_names, M(STR_SV_STNAME))) goto done;

   219

   220

		found = M(STR_SV_STNAME_CENTRAL);

   221

		if (HASBIT(free_names, M(STR_SV_STNAME_CENTRAL))) goto done;

   222

	}

   223

   224

	/* Check lakeside */

   225

	if (HASBIT(free_names, M(STR_SV_STNAME_LAKESIDE)) &&

   226

			!CheckDistanceFromEdge(tile, 20) &&

   227

			CountMapSquareAround(tile, MP_WATER, 0, 0) >= 5) {

   228

				found = M(STR_SV_STNAME_LAKESIDE);

   229

				goto done;

   230

			}

   231

   232

	/* Check woods */

   233

	if (HASBIT(free_names, M(STR_SV_STNAME_WOODS)) && (

   234

			CountMapSquareAround(tile, MP_TREES, 0, 255) >= 8 ||

   235

			CountMapSquareAround(tile, MP_INDUSTRY, 0x10, 0x11) >= 2 )) {

   236

				found = (_opt.landscape==LT_DESERT) ? M(STR_SV_STNAME_FOREST) : M(STR_SV_STNAME_WOODS);

   237

				goto done;

   238

	}

   239

   240

	/* check elevation compared to town */

   241

	z = GetTileZ(tile);

   242

	z2 = GetTileZ(t->xy);

   243

	if (z < z2) {

   244

		found = M(STR_SV_STNAME_VALLEY);

   245

		if (HASBIT(free_names, M(STR_SV_STNAME_VALLEY))) goto done;

   246

	} else if (z > z2) {

   247

		found = M(STR_SV_STNAME_HEIGHTS);

   248

		if (HASBIT(free_names, M(STR_SV_STNAME_HEIGHTS))) goto done;

   249

	}

   250

   251

	/* check direction compared to town */

   252

	{

   253

		static const int8 _direction_and_table[] = {

   254

			~( (1<<M(STR_SV_STNAME_WEST)) | (1<<M(STR_SV_STNAME_EAST)) | (1<<M(STR_SV_STNAME_NORTH)) ),

   255

			~( (1<<M(STR_SV_STNAME_SOUTH)) | (1<<M(STR_SV_STNAME_WEST)) | (1<<M(STR_SV_STNAME_NORTH)) ),

   256

			~( (1<<M(STR_SV_STNAME_SOUTH)) | (1<<M(STR_SV_STNAME_EAST)) | (1<<M(STR_SV_STNAME_NORTH)) ),

   257

			~( (1<<M(STR_SV_STNAME_SOUTH)) | (1<<M(STR_SV_STNAME_WEST)) | (1<<M(STR_SV_STNAME_EAST)) ),

   258

		};

   259

   260

		free_names &= _direction_and_table[

   261

			(GET_TILE_X(tile) < GET_TILE_X(t->xy)) +

   262

			(GET_TILE_Y(tile) < GET_TILE_Y(t->xy))*2];

   263

	}

   264

   265

	tmp = free_names & ((1<<1)|(1<<2)|(1<<3)|(1<<4)|(1<<6)|(1<<7)|(1<<12)|(1<<26)|(1<<27)|(1<<28)|(1<<29)|(1<<30));

   266

	if (tmp == 0) {

   267

		_error_message = STR_3007_TOO_MANY_STATIONS_LOADING;

   268

		return false;

   269

	}

   270

	found = FindFirstBit(tmp);

   271

   272

done:

   273

	st->string_id = found + STR_SV_STNAME;

   274

	return true;

   275

}

   276

#undef M

   277

   278

static Station *GetClosestStationFromTile(uint tile, uint threshold, byte owner)

   279

{

   280

	Station *st, *best_station = NULL;

   281

	uint cur_dist;

   282

   283

	FOR_ALL_STATIONS(st) {

   284

		cur_dist = GetTileDist(tile, st->xy);

   285

		if (cur_dist < threshold && (owner == 0xFF || st->owner == owner)) {

   286

			threshold = cur_dist;

   287

			best_station = st;

   288

		}

   289

	}

   290

   291

	return best_station;

   292

}

   293

   294

static void StationInitialize(Station *st, TileIndex tile)

   295

{

   296

	int i;

   297

	GoodsEntry *ge;

   298

   299

	st->xy = tile;

   300

	st->bus_tile = st->lorry_tile = st->airport_tile = st->dock_tile = st->train_tile = 0;

   301

	st->had_vehicle_of_type = 0;

   302

	st->time_since_load = 255;

   303

	st->time_since_unload = 255;

   304

	st->delete_ctr = 0;

   305

	st->facilities = 0;

   306

   307

	st->last_vehicle = INVALID_VEHICLE;

   308

   309

	for(i=0,ge=st->goods; i!=NUM_CARGO; i++, ge++) {

   310

		ge->waiting_acceptance = 0;

   311

		ge->days_since_pickup = 0;

   312

		ge->enroute_from = 0xFF;

   313

		ge->rating = 175;

   314

		ge->last_speed = 0;

   315

		ge->last_age = 0xFF;

   316

	}

   317

   318

	_global_station_sort_dirty = true; // build a new station

   319

}

   320

   321

// Update the virtual coords needed to draw the station sign.

   322

// st = Station to update for.

   323

static void UpdateStationVirtCoord(Station *st)

   324

{

   325

	Point pt = RemapCoords2(GET_TILE_X(st->xy) * 16, GET_TILE_Y(st->xy) * 16);

   326

	pt.y -= 32;

   327

	if (st->facilities&FACIL_AIRPORT && st->airport_type==AT_OILRIG) pt.y -= 16;

   328

   329

	SET_DPARAM16(0, st->index);

   330

	SET_DPARAM8(1, st->facilities);

   331

	UpdateViewportSignPos(&st->sign, pt.x, pt.y, STR_305C_0);

   332

}

   333

   334

// Update the virtual coords needed to draw the station sign for all stations.

   335

void UpdateAllStationVirtCoord()

   336

{

   337

	Station *st;

   338

	FOR_ALL_STATIONS(st) {

   339

		if (st->xy != 0)

   340

			UpdateStationVirtCoord(st);

   341

	}

   342

}

   343

   344

// Update the station virt coords while making the modified parts dirty.

   345

static void UpdateStationVirtCoordDirty(Station *st)

   346

{

   347

	MarkStationDirty(st);

   348

	UpdateStationVirtCoord(st);

   349

	MarkStationDirty(st);

   350

}

   351

   352

// Get a mask of the cargo types that the station accepts.

   353

static uint GetAcceptanceMask(Station *st)

   354

{

   355

	uint mask = 0;

   356

	uint cur_mask = 1;

   357

	int i;

   358

	for(i=0; i!=NUM_CARGO; i++,cur_mask*=2) {

   359

		if (st->goods[i].waiting_acceptance & 0x8000)

   360

			mask |= cur_mask;

   361

	}

   362

	return mask;

   363

}

   364

   365

// Items contains the two cargo names that are to be accepted or rejected.

   366

// msg is the string id of the message to display.

   367

static void ShowRejectOrAcceptNews(Station *st, uint32 items, StringID msg)

   368

{

   369

	if (items) {

   370

		SET_DPARAM32(2, items >> 16);

   371

		SET_DPARAM32(1, items & 0xFFFF);

   372

		SET_DPARAM16(0, st->index);

   373

		AddNewsItem(msg + ((items >> 16)?1:0), NEWS_FLAGS(NM_SMALL, NF_VIEWPORT|NF_TILE, NT_ACCEPTANCE, 0), st->xy, 0);

   374

	}

   375

}

   376

   377

// Get a list of the cargo types being produced around the tile.

   378

void GetProductionAroundTiles(uint *produced, uint tile, int w, int h)

   379

{

   380

	int x,y;

   381

	int x1,y1,x2,y2;

   382

	int xc,yc;

   383

	byte cargos[2];

   384

   385

	memset(produced, 0, NUM_CARGO * sizeof(uint));

   386

   387

	x = GET_TILE_X(tile);

   388

	y = GET_TILE_Y(tile);

   389

   390

	// expand the region by 4 tiles on each side

   391

	// while making sure that we remain inside the board.

   392

	x2 = min(x + w+4, TILE_X_MAX+1);

   393

	x1 = max(x-4, 0);

   394

   395

	y2 = min(y + h+4, TILE_Y_MAX+1);

   396

	y1 = max(y-4, 0);

   397

   398

	assert(x1 < x2);

   399

	assert(y1 < y2);

   400

	assert(w > 0);

   401

	assert(h > 0);

   402

   403

	yc = y1;

   404

	do {

   405

		xc = x1;

   406

		do {

   407

			if (!(IS_INSIDE_1D(xc, x, w) && IS_INSIDE_1D(yc, y, h))) {

   408

				GetProducedCargoProc *gpc;

   409

				uint tile = TILE_XY(xc, yc);

   410

				gpc = _tile_type_procs[GET_TILETYPE(tile)]->get_produced_cargo_proc;

   411

				if (gpc != NULL) {

   412

					cargos[0] = cargos[1] = 0xFF;

   413

					gpc(tile, cargos);

   414

					if (cargos[0] != 0xFF) {

   415

						produced[cargos[0]]++;

   416

						if (cargos[1] != 0xFF) {

   417

							produced[cargos[1]]++;

   418

						}

   419

					}

   420

				}

   421

			}

   422

		} while (++xc != x2);

   423

	} while (++yc != y2);

   424

}

   425

   426

// Get a list of the cargo types that are accepted around the tile.

   427

void GetAcceptanceAroundTiles(uint *accepts, uint tile, int w, int h)

   428

{

   429

	int x,y;

   430

	int x1,y1,x2,y2;

   431

	int xc,yc;

   432

	AcceptedCargo ac;

   433

   434

	memset(accepts, 0, NUM_CARGO * sizeof(uint));

   435

   436

	x = GET_TILE_X(tile);

   437

	y = GET_TILE_Y(tile);

   438

   439

	// expand the region by 4 tiles on each side

   440

	// while making sure that we remain inside the board.

   441

	x2 = min(x + w + 4, TILE_X_MAX+1);

   442

	y2 = min(y + h + 4, TILE_Y_MAX+1);

   443

	x1 = max(x-4, 0);

   444

	y1 = max(y-4, 0);

   445

   446

	assert(x1 < x2);

   447

	assert(y1 < y2);

   448

	assert(w > 0);

   449

	assert(h > 0);

   450

   451

	yc = y1;

   452

	do {

   453

		xc = x1;

   454

		do {

   455

			uint tile = TILE_XY(xc, yc);

   456

			if (!IS_TILETYPE(tile, MP_STATION)) {

   457

				GetAcceptedCargo(tile, &ac);

   458

				accepts[ac.type_1] += ac.amount_1;

   459

				accepts[ac.type_2] += ac.amount_2;

   460

				accepts[ac.type_3] += ac.amount_3;

   461

			}

   462

		} while (++xc != x2);

   463

	} while (++yc != y2);

   464

}

   465

   466

// Update the acceptance for a station.

   467

// show_msg controls whether to display a message that acceptance was changed.

   468

static void UpdateStationAcceptance(Station *st, bool show_msg)

   469

{

   470

	uint old_acc, new_acc;

   471

	TileIndex span[1+1+2+2+1];

   472

	int i;

   473

	int min_x, min_y, max_x, max_y;

   474

	uint accepts[NUM_CARGO];

   475

   476

	// Don't update acceptance for a buoy

   477

	if (st->had_vehicle_of_type & HVOT_BUOY)

   478

		return;

   479

   480

	/* old accepted goods types */

   481

	old_acc = GetAcceptanceMask(st);

   482

   483

	// Put all the tiles that span an area in the table.

   484

	span[3] = span[5] = 0;

   485

	span[0] = st->bus_tile;

   486

	span[1] = st->lorry_tile;

   487

	span[2] = st->train_tile;

   488

	if (st->train_tile != 0) {

   489

		span[3] = st->train_tile + TILE_XY(st->trainst_w-1, st->trainst_h-1);

   490

	}

   491

	span[4] = st->airport_tile;

   492

	if (st->airport_tile != 0) {

   493

		span[5] = st->airport_tile + TILE_XY(_airport_size_x[st->airport_type]-1, _airport_size_y[st->airport_type]-1);

   494

	}

   495

	span[6] = st->dock_tile;

   496

   497

	// Construct a rectangle from those points

   498

	min_x = min_y = 0x7FFFFFFF;

   499

	max_x = max_y = 0;

   500

   501

	for(i=0; i!=7; i++) {

   502

		uint tile = span[i];

   503

		if (tile) {

   504

			min_x = min(min_x,GET_TILE_X(tile));

   505

			max_x = max(max_x,GET_TILE_X(tile));

   506

			min_y = min(min_y,GET_TILE_Y(tile));

   507

			max_y = max(max_y,GET_TILE_Y(tile));

   508

		}

   509

	}

   510

   511

	// And retrieve the acceptance.

   512

	if (max_x != 0) {

   513

		GetAcceptanceAroundTiles(accepts, TILE_XY(min_x, min_y), max_x - min_x + 1, max_y-min_y+1);

   514

	} else {

   515

		memset(accepts, 0, sizeof(accepts));

   516

	}

   517

   518

	// Adjust in case our station only accepts fewer kinds of goods

   519

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

   520

		uint amt = min(accepts[i], 15);

   521

   522

		// Make sure the station can accept the goods type.

   523

		if ((i != CT_PASSENGERS && !(st->facilities & (byte)~FACIL_BUS_STOP)) ||

   524

				(i == CT_PASSENGERS && !(st->facilities & (byte)~FACIL_TRUCK_STOP)))

   525

			amt = 0;

   526