1

/* $Id$ */

     2

     3

/** @file station_cmd.c */

     4

     5

#include "stdafx.h"

     6

#include "openttd.h"

     7

#include "bridge_map.h"

     8

#include "debug.h"

     9

#include "functions.h"

    10

#include "station_map.h"

    11

#include "table/sprites.h"

    12

#include "table/strings.h"

    13

#include "map.h"

    14

#include "tile.h"

    15

#include "station.h"

    16

#include "gfx.h"

    17

#include "window.h"

    18

#include "viewport.h"

    19

#include "command.h"

    20

#include "town.h"

    21

#include "vehicle.h"

    22

#include "news.h"

    23

#include "saveload.h"

    24

#include "economy.h"

    25

#include "player.h"

    26

#include "airport.h"

    27

#include "sprite.h"

    28

#include "depot.h"

    29

#include "train.h"

    30

#include "water_map.h"

    31

#include "industry_map.h"

    32

#include "newgrf_callbacks.h"

    33

#include "newgrf_station.h"

    34

#include "yapf/yapf.h"

    35

#include "date.h"

    36

#include "helpers.hpp"

    37

#include "misc/autoptr.hpp"

    38

    39

/**

    40

 * Called if a new block is added to the station-pool

    41

 */

    42

static void StationPoolNewBlock(uint start_item)

    43

{

    44

	Station *st;

    45

    46

	/* We don't use FOR_ALL here, because FOR_ALL skips invalid items.

    47

	 *  TODO - This is just a temporary stage, this will be removed. */

    48

	for (st = GetStation(start_item); st != NULL; st = (st->index + 1U < GetStationPoolSize()) ? GetStation(st->index + 1U) : NULL) st->index = start_item++;

    49

}

    50

    51

static void StationPoolCleanBlock(uint start_item, uint end_item)

    52

{

    53

	uint i;

    54

    55

	for (i = start_item; i <= end_item; i++) {

    56

		Station *st = GetStation(i);

    57

		if (IsValidStation(st)) st->~Station();

    58

	}

    59

}

    60

    61

/**

    62

 * Called if a new block is added to the roadstop-pool

    63

 */

    64

static void RoadStopPoolNewBlock(uint start_item)

    65

{

    66

	RoadStop *rs;

    67

    68

	/* We don't use FOR_ALL here, because FOR_ALL skips invalid items.

    69

	 * TODO - This is just a temporary stage, this will be removed. */

    70

	for (rs = GetRoadStop(start_item); rs != NULL; rs = (rs->index + 1U < GetRoadStopPoolSize()) ? GetRoadStop(rs->index + 1U) : NULL) {

    71

		rs->xy    = INVALID_TILE;

    72

		rs->index = start_item++;

    73

	}

    74

}

    75

    76

DEFINE_OLD_POOL(Station, Station, StationPoolNewBlock, StationPoolCleanBlock)

    77

DEFINE_OLD_POOL(RoadStop, RoadStop, RoadStopPoolNewBlock, NULL)

    78

    79

    80

extern void UpdateAirplanesOnNewStation(Station *st);

    81

    82

    83

RoadStop* GetRoadStopByTile(TileIndex tile, RoadStop::Type type)

    84

{

    85

	const Station* st = GetStationByTile(tile);

    86

	RoadStop* rs;

    87

    88

	for (rs = st->GetPrimaryRoadStop(type); rs->xy != tile; rs = rs->next) {

    89

		assert(rs->next != NULL);

    90

	}

    91

    92

	return rs;

    93

}

    94

    95

    96

static uint GetNumRoadStopsInStation(const Station* st, RoadStop::Type type)

    97

{

    98

	uint num = 0;

    99

	const RoadStop *rs;

   100

   101

	assert(st != NULL);

   102

	for (rs = st->GetPrimaryRoadStop(type); rs != NULL; rs = rs->next) num++;

   103

   104

	return num;

   105

}

   106

   107

   108

/* Calculate the radius of the station. Basicly it is the biggest

   109

 *  radius that is available within the station */

   110

static uint FindCatchmentRadius(const Station* st)

   111

{

   112

	CatchmentAera ret = CA_NONE;

   113

   114

	if (st->bus_stops != NULL)   ret = max(ret, CA_BUS);

   115

	if (st->truck_stops != NULL) ret = max(ret, CA_TRUCK);

   116

	if (st->train_tile) ret = max(ret, CA_TRAIN);

   117

	if (st->dock_tile)  ret = max(ret, CA_DOCK);

   118

   119

	if (st->airport_tile) {

   120

		switch (st->airport_type) {

   121

			case AT_OILRIG:        ret = max(ret, CA_AIR_OILPAD);   break;

   122

			case AT_SMALL:         ret = max(ret, CA_AIR_SMALL);    break;

   123

			case AT_HELIPORT:      ret = max(ret, CA_AIR_HELIPORT); break;

   124

			case AT_LARGE:         ret = max(ret, CA_AIR_LARGE);    break;

   125

			case AT_METROPOLITAN:  ret = max(ret, CA_AIR_METRO);    break;

   126

			case AT_INTERNATIONAL: ret = max(ret, CA_AIR_INTER);    break;

   127

			case AT_COMMUTER:      ret = max(ret, CA_AIR_COMMUTER); break;

   128

			case AT_HELIDEPOT:     ret = max(ret, CA_AIR_HELIDEPOT); break;

   129

			case AT_INTERCON:      ret = max(ret, CA_AIR_INTERCON); break;

   130

			case AT_HELISTATION:   ret = max(ret, CA_AIR_HELISTATION); break;

   131

		}

   132

	}

   133

   134

	return ret;

   135

}

   136

   137

#define CHECK_STATIONS_ERR ((Station*)-1)

   138

   139

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

   140

{

   141

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

   142

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

   143

		if (IsTileType(tile_cur, MP_STATION)) {

   144

			StationID t = GetStationIndex(tile_cur);

   145

			{

   146

				Station *st = GetStation(t);

   147

				// you cannot take control of an oilrig!!

   148

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

   149

					continue;

   150

			}

   151

   152

			if (closest_station == INVALID_STATION) {

   153

				closest_station = t;

   154

			} else if (closest_station != t) {

   155

				_error_message = STR_3006_ADJOINS_MORE_THAN_ONE_EXISTING;

   156

				return CHECK_STATIONS_ERR;

   157

			}

   158

		}

   159

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

   160

	return (closest_station == INVALID_STATION) ? NULL : GetStation(closest_station);

   161

}

   162

   163

static Station *AllocateStation(void)

   164

{

   165

	Station *st = NULL;

   166

   167

	/* We don't use FOR_ALL here, because FOR_ALL skips invalid items.

   168

	 * TODO - This is just a temporary stage, this will be removed. */

   169

	for (st = GetStation(0); st != NULL; st = (st->index + 1U < GetStationPoolSize()) ? GetStation(st->index + 1U) : NULL) {

   170

		if (!IsValidStation(st)) {

   171

			StationID index = st->index;

   172

   173

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

   174

			st->index = index;

   175

			return st;

   176

		}

   177

	}

   178

   179

	/* Check if we can add a block to the pool */

   180

	if (AddBlockToPool(&_Station_pool)) return AllocateStation();

   181

   182

	_error_message = STR_3008_TOO_MANY_STATIONS_LOADING;

   183

	return NULL;

   184

}

   185

   186

   187

/**

   188

 * Counts the numbers of tiles matching a specific type in the area around

   189

 * @param tile the center tile of the 'count area'

   190

 * @param type the type of tile searched for

   191

 * @param industry when type == MP_INDUSTRY, the type of the industry,

   192

 *                 in all other cases this parameter is ignored

   193

 * @result the noumber of matching tiles around

   194

 */

   195

static int CountMapSquareAround(TileIndex tile, TileType type, IndustryType industry)

   196

{

   197

	TileIndex cur_tile;

   198

	int dx, dy;

   199

	int num = 0;

   200

   201

	for (dx = -3; dx <= 3; dx++) {

   202

		for (dy = -3; dy <= 3; dy++) {

   203

			cur_tile = TILE_MASK(tile + TileDiffXY(dx, dy));

   204

   205

			if (IsTileType(cur_tile, type)) {

   206

				switch (type) {

   207

					case MP_INDUSTRY:

   208

						if (GetIndustryType(cur_tile) == industry)

   209

							num++;

   210

						break;

   211

   212

					case MP_WATER:

   213

						if (!IsWater(cur_tile))

   214

							break;

   215

						/* FALL THROUGH WHEN WATER TILE */

   216

					case MP_TREES:

   217

						num++;

   218

						break;

   219

   220

					default:

   221

						break;

   222

				}

   223

			}

   224

		}

   225

	}

   226

   227

	return num;

   228

}

   229

   230

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

   231

   232

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

   233

{

   234

	static const uint32 _gen_station_name_bits[] = {

   235

		0,                                      /* 0 */

   236

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

   237

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

   238

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

   239

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

   240

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

   241

	};

   242

   243

	Town *t = st->town;

   244

	uint32 free_names = (uint32)-1;

   245

	int found;

   246

	uint z,z2;

   247

	unsigned long tmp;

   248

   249

	{

   250

		Station *s;

   251

   252

		FOR_ALL_STATIONS(s) {

   253

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

   254

				uint str = M(s->string_id);

   255

				if (str <= 0x20) {

   256

					if (str == M(STR_SV_STNAME_FOREST))

   257

						str = M(STR_SV_STNAME_WOODS);

   258

					CLRBIT(free_names, str);

   259

				}

   260

			}

   261

		}

   262

	}

   263

   264

	/* check default names */

   265

	tmp = free_names & _gen_station_name_bits[flag];

   266

	if (tmp != 0) {

   267

		found = FindFirstBit(tmp);

   268

		goto done;

   269

	}

   270

   271

	/* check mine? */

   272

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

   273

		if (CountMapSquareAround(tile, MP_INDUSTRY, IT_COAL_MINE) >= 2 ||

   274

				CountMapSquareAround(tile, MP_INDUSTRY, IT_IRON_MINE) >= 2 ||

   275

				CountMapSquareAround(tile, MP_INDUSTRY, IT_COPPER_MINE) >= 2 ||

   276

				CountMapSquareAround(tile, MP_INDUSTRY, IT_GOLD_MINE) >= 2 ||

   277

				CountMapSquareAround(tile, MP_INDUSTRY, IT_DIAMOND_MINE) >= 2) {

   278

			found = M(STR_SV_STNAME_MINES);

   279

			goto done;

   280

		}

   281

	}

   282

   283

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

   284

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

   285

		found = M(STR_SV_STNAME);

   286

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

   287

   288

		found = M(STR_SV_STNAME_CENTRAL);

   289

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

   290

	}

   291

   292

	/* Check lakeside */

   293

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

   294

			DistanceFromEdge(tile) < 20 &&

   295

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

   296

		found = M(STR_SV_STNAME_LAKESIDE);

   297

		goto done;

   298

	}

   299

   300

	/* Check woods */

   301

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

   302

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

   303

				CountMapSquareAround(tile, MP_INDUSTRY, IT_FOREST) >= 2)

   304

			) {

   305

		found = _opt.landscape == LT_DESERT ?

   306

			M(STR_SV_STNAME_FOREST) : M(STR_SV_STNAME_WOODS);

   307

		goto done;

   308

	}

   309

   310

	/* check elevation compared to town */

   311

	z = GetTileZ(tile);

   312

	z2 = GetTileZ(t->xy);

   313

	if (z < z2) {

   314

		found = M(STR_SV_STNAME_VALLEY);

   315

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

   316

	} else if (z > z2) {

   317

		found = M(STR_SV_STNAME_HEIGHTS);

   318

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

   319

	}

   320

   321

	/* check direction compared to town */

   322

	{

   323

		static const int8 _direction_and_table[] = {

   324

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

   325

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

   326

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

   327

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

   328

		};

   329

   330

		free_names &= _direction_and_table[

   331

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

   332

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

   333

	}

   334

   335

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

   336

	if (tmp == 0) {

   337

		_error_message = STR_3007_TOO_MANY_STATIONS_LOADING;

   338

		return false;

   339

	}

   340

	found = FindFirstBit(tmp);

   341

   342

done:

   343

	st->string_id = found + STR_SV_STNAME;

   344

	return true;

   345

}

   346

#undef M

   347

   348

static Station* GetClosestStationFromTile(TileIndex tile, uint threshold, PlayerID owner)

   349

{

   350

	Station* best_station = NULL;

   351

	Station* st;

   352

   353

	FOR_ALL_STATIONS(st) {

   354

		if ((owner == PLAYER_SPECTATOR || st->owner == owner)) {

   355

			uint cur_dist = DistanceManhattan(tile, st->xy);

   356

   357

			if (cur_dist < threshold) {

   358

				threshold = cur_dist;

   359

				best_station = st;

   360

			}

   361

		}

   362

	}

   363

   364

	return best_station;

   365

}

   366

   367

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

   368

// st = Station to update for.

   369

static void UpdateStationVirtCoord(Station *st)

   370

{

   371

	Point pt = RemapCoords2(TileX(st->xy) * TILE_SIZE, TileY(st->xy) * TILE_SIZE);

   372

   373

	pt.y -= 32;

   374

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

   375

   376

	SetDParam(0, st->index);

   377

	SetDParam(1, st->facilities);

   378

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

   379

}

   380

   381

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

   382

void UpdateAllStationVirtCoord(void)

   383

{

   384

	Station* st;

   385

   386

	FOR_ALL_STATIONS(st) {

   387

		UpdateStationVirtCoord(st);

   388

	}

   389

}

   390

   391

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

   392

static void UpdateStationVirtCoordDirty(Station *st)

   393

{

   394

	st->MarkDirty();

   395

	UpdateStationVirtCoord(st);

   396

	st->MarkDirty();

   397

}

   398

   399

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

   400

static uint GetAcceptanceMask(const Station *st)

   401

{

   402

	uint mask = 0;

   403

	uint i;

   404

   405

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

   406

		if (st->goods[i].waiting_acceptance & 0x8000) mask |= 1 << i;

   407

	}

   408

	return mask;

   409

}

   410

   411

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

   412

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

   413

static void ShowRejectOrAcceptNews(const Station *st, uint num_items, CargoID *cargo, StringID msg)

   414

{

   415

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

   416

		SetDParam(i + 1, _cargoc.names_s[cargo[i]]);

   417

	}

   418

   419

	SetDParam(0, st->index);

   420

	AddNewsItem(msg, NEWS_FLAGS(NM_SMALL, NF_VIEWPORT|NF_TILE, NT_ACCEPTANCE, 0), st->xy, 0);

   421

}

   422

   423

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

   424

void GetProductionAroundTiles(AcceptedCargo produced, TileIndex tile,

   425

	int w, int h, int rad)

   426

{

   427

	int x,y;

   428

	int x1,y1,x2,y2;

   429

	int xc,yc;

   430

   431

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

   432

   433

	x = TileX(tile);

   434

	y = TileY(tile);

   435

   436

	// expand the region by rad tiles on each side

   437

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

   438

	x2 = min(x + w + rad, MapSizeX());

   439

	x1 = max(x - rad, 0);

   440

   441

	y2 = min(y + h + rad, MapSizeY());

   442

	y1 = max(y - rad, 0);

   443

   444

	assert(x1 < x2);

   445

	assert(y1 < y2);

   446

	assert(w > 0);

   447

	assert(h > 0);

   448

   449

	for (yc = y1; yc != y2; yc++) {

   450

		for (xc = x1; xc != x2; xc++) {

   451

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

   452

				GetProducedCargoProc *gpc;

   453

				TileIndex tile = TileXY(xc, yc);

   454

   455

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

   456

				if (gpc != NULL) {

   457

					CargoID cargos[2] = { CT_INVALID, CT_INVALID };

   458

   459

					gpc(tile, cargos);

   460

					if (cargos[0] != CT_INVALID) {

   461

						produced[cargos[0]]++;

   462

						if (cargos[1] != CT_INVALID) {

   463

							produced[cargos[1]]++;

   464

						}

   465

					}

   466

				}

   467

			}

   468

		}

   469

	}

   470

}

   471

   472

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

   473

void GetAcceptanceAroundTiles(AcceptedCargo accepts, TileIndex tile,

   474

	int w, int h, int rad)

   475

{

   476

	int x,y;

   477

	int x1,y1,x2,y2;

   478

	int xc,yc;

   479

   480

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

   481

   482

	x = TileX(tile);

   483

	y = TileY(tile);

   484

   485

	// expand the region by rad tiles on each side

   486

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

   487

	x2 = min(x + w + rad, MapSizeX());

   488

	y2 = min(y + h + rad, MapSizeY());

   489

	x1 = max(x - rad, 0);

   490

	y1 = max(y - rad, 0);

   491

   492

	assert(x1 < x2);

   493

	assert(y1 < y2);

   494

	assert(w > 0);

   495

	assert(h > 0);

   496

   497

	for (yc = y1; yc != y2; yc++) {