1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "debug.h"

     4

#include "table/sprites.h"

     5

#include "table/strings.h"

     6

#include "map.h"

     7

#include "tile.h"

     8

#include "station.h"

     9

#include "gfx.h"

    10

#include "window.h"

    11

#include "viewport.h"

    12

#include "command.h"

    13

#include "town.h"

    14

#include "vehicle.h"

    15

#include "news.h"

    16

#include "saveload.h"

    17

#include "economy.h"

    18

#include "player.h"

    19

#include "airport.h"

    20

#include "sprite.h"

    21

#include "npf.h"

    22

#include "depot.h"

    23

    24

enum {

    25

	/* Max stations: 64000 (64 * 1000) */

    26

	STATION_POOL_BLOCK_SIZE_BITS = 6,       /* In bits, so (1 << 6) == 64 */

    27

	STATION_POOL_MAX_BLOCKS      = 1000,

    28

    29

	/* Max roadstops: 64000 (32 * 2000) */

    30

	ROADSTOP_POOL_BLOCK_SIZE_BITS = 5,       /* In bits, so (1 << 5) == 32 */

    31

	ROADSTOP_POOL_MAX_BLOCKS      = 2000,

    32

};

    33

    34

/**

    35

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

    36

 */

    37

static void StationPoolNewBlock(uint start_item)

    38

{

    39

	Station *st;

    40

    41

	FOR_ALL_STATIONS_FROM(st, start_item)

    42

		st->index = start_item++;

    43

}

    44

    45

/**

    46

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

    47

 */

    48

static void RoadStopPoolNewBlock(uint start_item)

    49

{

    50

	RoadStop *rs;

    51

    52

	FOR_ALL_ROADSTOPS_FROM(rs, start_item)

    53

		rs->index = start_item++;

    54

}

    55

    56

/* Initialize the station-pool and roadstop-pool */

    57

MemoryPool _station_pool = { "Stations", STATION_POOL_MAX_BLOCKS, STATION_POOL_BLOCK_SIZE_BITS, sizeof(Station), &StationPoolNewBlock, 0, 0, NULL };

    58

MemoryPool _roadstop_pool = { "RoadStop", ROADSTOP_POOL_MAX_BLOCKS, ROADSTOP_POOL_BLOCK_SIZE_BITS, sizeof(RoadStop), &RoadStopPoolNewBlock, 0, 0, NULL };

    59

    60

    61

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

    62

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

    63

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

    64

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

    65

    66

void ShowAircraftDepotWindow(uint tile);

    67

extern void UpdateAirplanesOnNewStation(Station *st);

    68

    69

static void MarkStationDirty(Station *st)

    70

{

    71

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

    72

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

    73

    74

		MarkAllViewportsDirty(

    75

			st->sign.left - 6,

    76

			st->sign.top,

    77

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

    78

			st->sign.top + 48);

    79

	}

    80

}

    81

    82

static void InitializeRoadStop(RoadStop *road_stop, RoadStop *previous, TileIndex tile, uint index)

    83

{

    84

	road_stop->xy = tile;

    85

	road_stop->used = true;

    86

	road_stop->status = 3; //stop is free

    87

	road_stop->slot[0] = road_stop->slot[1] = INVALID_SLOT;

    88

	road_stop->next = NULL;

    89

	road_stop->prev = previous;

    90

	road_stop->station = index;

    91

}

    92

    93

RoadStop * GetPrimaryRoadStop(const Station *st, RoadStopType type)

    94

{

    95

	switch (type) {

    96

		case RS_BUS: return st->bus_stops;

    97

		case RS_TRUCK: return st->truck_stops;

    98

		default:

    99

			NOT_REACHED();

   100

	}

   101

   102

	return NULL;

   103

}

   104

   105

RoadStop * GetRoadStopByTile(TileIndex tile, RoadStopType type)

   106

{

   107

	const Station *st = GetStation(_map2[tile]);

   108

	RoadStop *rs;

   109

   110

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

   111

		assert(rs->next != NULL);

   112

   113

	return rs;

   114

}

   115

   116

uint GetNumRoadStops(const Station *st, RoadStopType type)

   117

{

   118

	int num = 0;

   119

	const RoadStop *rs;

   120

   121

	assert(st != NULL);

   122

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

   123

   124

	return num;

   125

}

   126

   127

RoadStop *AllocateRoadStop( void )

   128

{

   129

	RoadStop *rs;

   130

   131

	FOR_ALL_ROADSTOPS(rs) {

   132

		if (!rs->used) {

   133

			uint index = rs->index;

   134

   135

			memset(rs, 0, sizeof(RoadStop));

   136

			rs->index = index;

   137

   138

			return rs;

   139

		}

   140

	}

   141

   142

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

   143

	if (AddBlockToPool(&_roadstop_pool))

   144

		return AllocateRoadStop();

   145

   146

	return NULL;

   147

}

   148

   149

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

   150

    radius that is available within the station */

   151

static byte FindCatchmentRadius(Station *st)

   152

{

   153

	byte ret = 0;

   154

   155

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

   156

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

   157

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

   158

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

   159

   160

	if (st->airport_tile) {

   161

		switch (st->airport_type) {

   162

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

   163

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

   164

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

   165

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

   166

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

   167

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

   168

		}

   169

	}

   170

   171

	return ret;

   172

}

   173

   174

#define CHECK_STATIONS_ERR ((Station*)-1)

   175

   176

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

   177

{

   178

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

   179

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

   180

		if (IsTileType(tile_cur, MP_STATION)) {

   181

			int t;

   182

			t = _map2[tile_cur];

   183

			{

   184

				Station *st = GetStation(t);

   185

				// you cannot take control of an oilrig!!

   186

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

   187

					continue;

   188

			}

   189

   190

			if (closest_station == -1) {

   191

				closest_station = t;

   192

			} else if (closest_station != t) {

   193

				_error_message = STR_3006_ADJOINS_MORE_THAN_ONE_EXISTING;

   194

				return CHECK_STATIONS_ERR;

   195

			}

   196

		}

   197

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

   198

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

   199

}

   200

   201

TileIndex GetStationTileForVehicle(const Vehicle *v, const Station *st)

   202

{

   203

	switch (v->type) {

   204

		case VEH_Train: 		return st->train_tile;

   205

		case VEH_Aircraft:	return st->airport_tile;

   206

		case VEH_Ship:			return st->dock_tile;

   207

		case VEH_Road:

   208

			if (v->cargo_type == CT_PASSENGERS) {

   209

				if (st->bus_stops != NULL)

   210

					return st->bus_stops->xy;

   211

				else

   212

					return 0;

   213

			} else {

   214

				if (st->truck_stops != NULL)

   215

					return st->truck_stops->xy;

   216

				else

   217

					return 0;

   218

			}

   219

		default:

   220

			assert(false);

   221

			return 0;

   222

	}

   223

}

   224

   225

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

   226

{

   227

	uint16 station_index = st->index;

   228

	uint i;

   229

	uint x1 = TileX(tile);

   230

	uint y1 = TileY(tile);

   231

	uint x2 = x1 + w - 1;

   232

	uint y2 = y1 + h - 1;

   233

	uint t;

   234

   235

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

   236

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

   237

			t = TileX(i);

   238

			if (t < x1) x1 = t;

   239

			if (t > x2) x2 = t;

   240

   241

			t = TileY(i);

   242

			if (t < y1) y1 = t;

   243

			if (t > y2) y2 = t;

   244

		}

   245

	}

   246

   247

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

   248

		_error_message = STR_306C_STATION_TOO_SPREAD_OUT;

   249

		return false;

   250

	}

   251

   252

	return true;

   253

}

   254

   255

static Station *AllocateStation(void)

   256

{

   257

	Station *st = NULL;

   258

   259

	FOR_ALL_STATIONS(st) {

   260

		if (st->xy == 0) {

   261

			uint index = st->index;

   262

   263

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

   264

			st->index = index;

   265

   266

			return st;

   267

		}

   268

	}

   269

   270

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

   271

	if (AddBlockToPool(&_station_pool))

   272

		return AllocateStation();

   273

   274

	_error_message = STR_3008_TOO_MANY_STATIONS_LOADING;

   275

	return NULL;

   276

}

   277

   278

   279

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

   280

	static const TileIndexDiffC _count_square_table[] = {

   281

		{-3, -3}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0},

   282

		{-6,  1}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0},

   283

		{-6,  1}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0},

   284

		{-6,  1}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0},

   285

		{-6,  1}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0},

   286

		{-6,  1}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0},

   287

		{-6,  1}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0}, {1, 0}

   288

	};

   289

	const TileIndexDiffC *p;

   290

	int num = 0;

   291

   292

	for (p = _count_square_table; p != endof(_count_square_table); ++p) {

   293

		tile = TILE_MASK(tile + ToTileIndexDiff(*p));

   294

   295

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

   296

			num++;

   297

	}

   298

   299

	return num;

   300

}

   301

   302

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

   303

   304

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

   305

{

   306

	static const uint32 _gen_station_name_bits[] = {

   307

		0,																/* 0 */

   308

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

   309

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

   310

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

   311

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

   312

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

   313

	};

   314

   315

	Town *t = st->town;

   316

	uint32 free_names = (uint32)-1;

   317

	int found;

   318

	uint z,z2;

   319

	unsigned long tmp;

   320

   321

	{

   322

		Station *s;

   323

   324

		FOR_ALL_STATIONS(s) {

   325

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

   326

				uint str = M(s->string_id);

   327

				if (str <= 0x20) {

   328

					if (str == M(STR_SV_STNAME_FOREST))

   329

						str = M(STR_SV_STNAME_WOODS);

   330

					CLRBIT(free_names, str);

   331

				}

   332

			}

   333

		}

   334

	}

   335

   336

	/* check default names */

   337

	tmp = free_names & _gen_station_name_bits[flag];

   338

	if (tmp != 0) {

   339

		found = FindFirstBit(tmp);

   340

		goto done;

   341

	}

   342

   343

	/* check mine? */

   344

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

   345

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

   346

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

   347

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

   348

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

   349

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

   350

					found = M(STR_SV_STNAME_MINES);

   351

					goto done;

   352

				}

   353

	}

   354

   355

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

   356

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

   357

		found = M(STR_SV_STNAME);

   358

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

   359

   360

		found = M(STR_SV_STNAME_CENTRAL);

   361

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

   362

	}

   363

   364

	/* Check lakeside */

   365

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

   366

			DistanceFromEdge(tile) < 20 &&

   367

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

   368

				found = M(STR_SV_STNAME_LAKESIDE);

   369

				goto done;

   370

			}

   371

   372

	/* Check woods */

   373

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

   374

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

   375

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

   376

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

   377

				goto done;

   378

	}

   379

   380

	/* check elevation compared to town */

   381

	z = GetTileZ(tile);

   382

	z2 = GetTileZ(t->xy);

   383

	if (z < z2) {

   384

		found = M(STR_SV_STNAME_VALLEY);

   385

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

   386

	} else if (z > z2) {

   387

		found = M(STR_SV_STNAME_HEIGHTS);

   388

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

   389

	}

   390

   391

	/* check direction compared to town */

   392

	{

   393

		static const int8 _direction_and_table[] = {

   394

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

   395

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

   396

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

   397

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

   398

		};

   399

   400

		free_names &= _direction_and_table[

   401

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

   402

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

   403

	}

   404

   405

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

   406

	if (tmp == 0) {

   407

		_error_message = STR_3007_TOO_MANY_STATIONS_LOADING;

   408

		return false;

   409

	}

   410

	found = FindFirstBit(tmp);

   411

   412

done:

   413

	st->string_id = found + STR_SV_STNAME;

   414

	return true;

   415

}

   416

#undef M

   417

   418

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

   419

{

   420

	Station *st, *best_station = NULL;

   421

	uint cur_dist;

   422

   423

	FOR_ALL_STATIONS(st) {

   424

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

   425

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

   426

			threshold = cur_dist;

   427

			best_station = st;

   428

		}

   429

	}

   430

   431

	return best_station;

   432

}

   433

   434

static void StationInitialize(Station *st, TileIndex tile)

   435

{

   436

	int i;

   437

	GoodsEntry *ge;

   438

   439

	st->xy = tile;

   440

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

   441

	st->bus_stops = st->truck_stops = NULL;

   442

	st->had_vehicle_of_type = 0;

   443

	st->time_since_load = 255;

   444

	st->time_since_unload = 255;

   445

	st->delete_ctr = 0;

   446

	st->facilities = 0;

   447

   448

	st->last_vehicle = INVALID_VEHICLE;

   449

   450

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

   451

		ge->waiting_acceptance = 0;

   452

		ge->days_since_pickup = 0;

   453

		ge->enroute_from = INVALID_STATION;

   454

		ge->rating = 175;

   455

		ge->last_speed = 0;

   456

		ge->last_age = 0xFF;

   457

	}

   458

   459

	_global_station_sort_dirty = true; // build a new station

   460

}

   461

   462

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

   463

// st = Station to update for.

   464

static void UpdateStationVirtCoord(Station *st)

   465

{

   466

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

   467

   468

	pt.y -= 32;

   469

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

   470

   471

	SetDParam(0, st->index);

   472

	SetDParam(1, st->facilities);

   473

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

   474

}

   475

   476

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

   477

void UpdateAllStationVirtCoord(void)

   478

{

   479

	Station *st;

   480

	FOR_ALL_STATIONS(st) {

   481

		if (st->xy != 0)

   482

			UpdateStationVirtCoord(st);

   483

	}

   484

}

   485

   486

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

   487

static void UpdateStationVirtCoordDirty(Station *st)

   488

{

   489

	MarkStationDirty(st);

   490

	UpdateStationVirtCoord(st);

   491

	MarkStationDirty(st);

   492

}

   493

   494

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

   495

static uint GetAcceptanceMask(Station *st)

   496

{

   497

	uint mask = 0;

   498

	uint cur_mask = 1;

   499

	int i;

   500

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

   501

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

   502

			mask |= cur_mask;

   503

	}

   504

	return mask;

   505

}

   506

   507

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