1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "table/strings.h"

     4

#include "map.h"

     5

#include "station.h"

     6

#include "gfx.h"

     7

#include "window.h"

     8

#include "viewport.h"

     9

#include "command.h"

    10

#include "town.h"

    11

#include "vehicle.h"

    12

#include "news.h"

    13

#include "saveload.h"

    14

#include "economy.h"

    15

#include "player.h"

    16

#include "airport.h"

    17

#include "sprite.h"

    18

#include "table/directions.h"

    19

    20

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

    21

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

    22

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

    23

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

    24

    25

void ShowAircraftDepotWindow(uint tile);

    26

extern void UpdateAirplanesOnNewStation(Station *st);

    27

    28

static void MarkStationDirty(Station *st)

    29

{

    30

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

    31

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

    32

    33

		MarkAllViewportsDirty(

    34

			st->sign.left - 6,

    35

			st->sign.top,

    36

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

    37

			st->sign.top + 48);

    38

	}

    39

}

    40

    41

#define CHECK_STATIONS_ERR ((Station*)-1)

    42

    43

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

    44

{

    45

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

    46

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

    47

		if (IS_TILETYPE(tile_cur, MP_STATION)) {

    48

			int t;

    49

			t = _map2[tile_cur];

    50

			{

    51

				Station *st = DEREF_STATION(t);

    52

				// you cannot take control of an oilrig!!

    53

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

    54

					continue;

    55

			}

    56

    57

			if (closest_station == -1) {

    58

				closest_station = t;

    59

			} else if (closest_station != t) {

    60

				_error_message = STR_3006_ADJOINS_MORE_THAN_ONE_EXISTING;

    61

				return CHECK_STATIONS_ERR;

    62

			}

    63

		}

    64

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

    65

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

    66

}

    67

    68

TileIndex GetStationTileForVehicle(Vehicle *v, Station *st)

    69

{

    70

	switch (v->type) {

    71

		case VEH_Train: 		return st->train_tile;

    72

		case VEH_Aircraft:	return st->airport_tile;

    73

		case VEH_Ship:			return st->dock_tile;

    74

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

    75

		default:

    76

			assert(false);

    77

			return 0;

    78

	}

    79

}

    80

    81

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

    82

{

    83

	byte station_index = st->index;

    84

	uint i;

    85

	uint x1 = GET_TILE_X(tile);

    86

	uint y1 = GET_TILE_Y(tile);

    87

	uint x2 = x1 + w - 1;

    88

	uint y2 = y1 + h - 1;

    89

	uint t;

    90

    91

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

    92

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

    93

			t = GET_TILE_X(i);

    94

			if (t < x1) x1 = t;

    95

			if (t > x2) x2 = t;

    96

    97

			t = GET_TILE_Y(i);

    98

			if (t < y1) y1 = t;

    99

			if (t > y2) y2 = t;

   100

		}

   101

	}

   102

   103

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

   104

		_error_message = STR_306C_STATION_TOO_SPREAD_OUT;

   105

		return false;

   106

	}

   107

   108

	return true;

   109

}

   110

   111

static Station *AllocateStation()

   112

{

   113

	Station *st, *a_free = NULL;

   114

	int count;

   115

	int num_free = 0;

   116

	int i;

   117

   118

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

   119

		if (st->xy == 0) {

   120

			num_free++;

   121

			if (a_free==NULL)

   122

				a_free = st;

   123

		}

   124

	}

   125

   126

	if (a_free == NULL ||

   127

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

   128

		_error_message = STR_3008_TOO_MANY_STATIONS_LOADING;

   129

		return NULL;

   130

	}

   131

   132

	i = a_free->index;

   133

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

   134

	a_free->index = i;

   135

	return a_free;

   136

}

   137

   138

   139

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

   140

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

   141

		TILE_XY(-3,-3), 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

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

   147

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

   148

		0,

   149

	};

   150

	int j;

   151

	const TileIndexDiff *p = _count_square_table;

   152

	int num = 0;

   153

   154

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

   155

		tile = TILE_MASK(tile + j);

   156

   157

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

   158

			num++;

   159

	}

   160

   161

	return num;

   162

}

   163

   164

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

   165

   166

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

   167

{

   168

	static const uint32 _gen_station_name_bits[] = {

   169

		0,																/* 0 */

   170

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

   171

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

   172

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

   173

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

   174

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

   175

	};

   176

   177

	Town *t = st->town;

   178

	uint32 free_names = (uint32)-1;

   179

	int found;

   180

	uint z,z2;

   181

	unsigned long tmp;

   182

   183

	{

   184

		Station *s;

   185

   186

		FOR_ALL_STATIONS(s) {

   187

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

   188

				uint str = M(s->string_id);

   189

				if (str <= 0x20) {

   190

					if (str == M(STR_SV_STNAME_FOREST))

   191

						str = M(STR_SV_STNAME_WOODS);

   192

					CLRBIT(free_names, str);

   193

				}

   194

			}

   195

		}

   196

	}

   197

   198

	/* check default names */

   199

	tmp = free_names & _gen_station_name_bits[flag];

   200

	if (tmp != 0) {

   201

		found = FindFirstBit(tmp);

   202

		goto done;

   203

	}

   204

   205

	/* check mine? */

   206

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

   207

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

   208

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

   209

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

   210

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

   211

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

   212

					found = M(STR_SV_STNAME_MINES);

   213

					goto done;

   214

				}

   215

	}

   216

   217

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

   218

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

   219

		found = M(STR_SV_STNAME);

   220

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

   221

   222

		found = M(STR_SV_STNAME_CENTRAL);

   223

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

   224

	}

   225

   226

	/* Check lakeside */

   227

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

   228

			!CheckDistanceFromEdge(tile, 20) &&

   229

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

   230

				found = M(STR_SV_STNAME_LAKESIDE);

   231

				goto done;

   232

			}

   233

   234

	/* Check woods */

   235

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

   236

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

   237

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

   238

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

   239

				goto done;

   240

	}

   241

   242

	/* check elevation compared to town */

   243

	z = GetTileZ(tile);

   244

	z2 = GetTileZ(t->xy);

   245

	if (z < z2) {

   246

		found = M(STR_SV_STNAME_VALLEY);

   247

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

   248

	} else if (z > z2) {

   249

		found = M(STR_SV_STNAME_HEIGHTS);

   250

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

   251

	}

   252

   253

	/* check direction compared to town */

   254

	{

   255

		static const int8 _direction_and_table[] = {

   256

			~( (1<<M(STR_SV_STNAME_WEST)) | (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_NORTH)) ),

   258

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

   259

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

   260

		};

   261

   262

		free_names &= _direction_and_table[

   263

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

   264

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

   265

	}

   266

   267

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

   268

	if (tmp == 0) {

   269

		_error_message = STR_3007_TOO_MANY_STATIONS_LOADING;

   270

		return false;

   271

	}

   272

	found = FindFirstBit(tmp);

   273

   274

done:

   275

	st->string_id = found + STR_SV_STNAME;

   276

	return true;

   277

}

   278

#undef M

   279

   280

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

   281

{

   282

	Station *st, *best_station = NULL;

   283

	uint cur_dist;

   284

   285

	FOR_ALL_STATIONS(st) {

   286

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

   287

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

   288

			threshold = cur_dist;

   289

			best_station = st;

   290

		}

   291

	}

   292

   293

	return best_station;

   294

}

   295

   296

static void StationInitialize(Station *st, TileIndex tile)

   297

{

   298

	int i;

   299

	GoodsEntry *ge;

   300

   301

	st->xy = tile;

   302

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

   303

	st->had_vehicle_of_type = 0;

   304

	st->time_since_load = 255;

   305

	st->time_since_unload = 255;

   306

	st->delete_ctr = 0;

   307

	st->facilities = 0;

   308

   309

	st->last_vehicle = INVALID_VEHICLE;

   310

   311

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

   312

		ge->waiting_acceptance = 0;

   313

		ge->days_since_pickup = 0;

   314

		ge->enroute_from = 0xFF;

   315

		ge->rating = 175;

   316

		ge->last_speed = 0;

   317

		ge->last_age = 0xFF;

   318

	}

   319

   320

	_global_station_sort_dirty = true; // build a new station

   321

}

   322

   323

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

   324

// st = Station to update for.

   325

static void UpdateStationVirtCoord(Station *st)

   326

{

   327

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

   328

	pt.y -= 32;

   329

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

   330

   331

	SetDParam(0, st->index);

   332

	SetDParam(1, st->facilities);

   333

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

   334

}

   335

   336

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

   337

void UpdateAllStationVirtCoord()

   338

{

   339

	Station *st;

   340

	FOR_ALL_STATIONS(st) {

   341

		if (st->xy != 0)

   342

			UpdateStationVirtCoord(st);

   343

	}

   344

}

   345

   346

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

   347

static void UpdateStationVirtCoordDirty(Station *st)

   348

{

   349

	MarkStationDirty(st);

   350

	UpdateStationVirtCoord(st);

   351

	MarkStationDirty(st);

   352

}

   353

   354

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

   355

static uint GetAcceptanceMask(Station *st)

   356

{

   357

	uint mask = 0;

   358

	uint cur_mask = 1;

   359

	int i;

   360

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

   361

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

   362

			mask |= cur_mask;

   363

	}

   364

	return mask;

   365

}

   366

   367

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

   368

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

   369

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

   370

{

   371

	if (items) {

   372

		SetDParam(2, items >> 16);

   373

		SetDParam(1, items & 0xFFFF);

   374

		SetDParam(0, st->index);

   375

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

   376

	}

   377

}

   378

   379

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

   380

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

   381

{

   382

	int x,y;

   383

	int x1,y1,x2,y2;

   384

	int xc,yc;

   385

	byte cargos[2];

   386

   387

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

   388

   389

	x = GET_TILE_X(tile);

   390

	y = GET_TILE_Y(tile);

   391

   392

	// expand the region by 4 tiles on each side

   393

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

   394

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

   395

	x1 = max(x-rad, 0);

   396

   397

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

   398

	y1 = max(y-rad, 0);

   399

   400

	assert(x1 < x2);

   401

	assert(y1 < y2);

   402

	assert(w > 0);

   403

	assert(h > 0);

   404

   405

	yc = y1;

   406

	do {

   407

		xc = x1;

   408

		do {

   409

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

   410

				GetProducedCargoProc *gpc;

   411

				uint tile = TILE_XY(xc, yc);

   412

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

   413

				if (gpc != NULL) {

   414

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

   415

					gpc(tile, cargos);

   416

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

   417

						produced[cargos[0]]++;

   418

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

   419

							produced[cargos[1]]++;

   420

						}

   421

					}

   422

				}

   423

			}

   424

		} while (++xc != x2);

   425

	} while (++yc != y2);

   426

}

   427

   428

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

   429

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

   430

{

   431

	int x,y;

   432

	int x1,y1,x2,y2;

   433

	int xc,yc;

   434

   435

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

   436

   437

	x = GET_TILE_X(tile);

   438

	y = GET_TILE_Y(tile);

   439

   440

	// expand the region by 4 tiles on each side

   441

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

   442

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

   443

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

   444

	x1 = max(x-rad, 0);

   445

	y1 = max(y-rad, 0);

   446

   447

	assert(x1 < x2);

   448

	assert(y1 < y2);

   449

	assert(w > 0);

   450

	assert(h > 0);

   451

   452

	yc = y1;

   453

	do {

   454

		xc = x1;

   455

		do {

   456

			uint tile = TILE_XY(xc, yc);

   457

			if (!IS_TILETYPE(tile, MP_STATION)) {

   458

				AcceptedCargo ac;

   459

				int i;

   460

   461

				GetAcceptedCargo(tile, ac);

   462

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

   463

					accepts[i] += ac[i];

   464

			}

   465

		} while (++xc != x2);

   466

	} while (++yc != y2);

   467

}

   468

   469

// Update the acceptance for a station.

   470

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

   471

static void UpdateStationAcceptance(Station *st, bool show_msg)

   472

{

   473

	uint old_acc, new_acc;

   474

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

   475

	int i;

   476

	int min_x, min_y, max_x, max_y;

   477

	int rad = 4;	//Put this to surpress a compiler warning

   478

	uint accepts[NUM_CARGO];

   479

   480

	// Don't update acceptance for a buoy

   481

	if (st->had_vehicle_of_type & HVOT_BUOY)

   482

		return;

   483

   484

	/* old accepted goods types */

   485

	old_acc = GetAcceptanceMask(st);

   486

   487

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

   488

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

   489

	span[0] = st->bus_tile;

   490

	span[1] = st->lorry_tile;

   491

	span[2] = st->train_tile;

   492

	if (st->train_tile != 0) {

   493

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

   494

	}

   495

	span[4] = st->airport_tile;

   496

	if (st->airport_tile != 0) {

   497

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

   498

	}

   499

	span[6] = st->dock_tile;

   500

   501

	// Construct a rectangle from those points

   502

	min_x = min_y = 0x7FFFFFFF;

   503

	max_x = max_y = 0;

   504

   505

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

   506

		uint tile = span[i];

   507

		if (tile) {

   508

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

   509

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

   510

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

   511

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

   512

		}

   513

	}

   514

	if (_patches.modified_catchment) {

   515

		FIND_CATCHMENT_RADIUS(st,rad)

   516

	} else {

   517

		rad = 4;

   518

	}

   519

   520

	// And retrieve the acceptance.

   521

	if (max_x != 0) {

   522

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