1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

     6

#include "command.h"

     7

#include "functions.h"

     8

#include "gfx.h"

     9

#include "map.h"

    10

#include "order.h"

    11

#include "rail_map.h"

    12

#include "saveload.h"

    13

#include "station.h"

    14

#include "tile.h"

    15

#include "town.h"

    16

#include "waypoint.h"

    17

#include "variables.h"

    18

#include "table/strings.h"

    19

#include "vehicle.h"

    20

#include "yapf/yapf.h"

    21

#include "date.h"

    22

    23

enum {

    24

	/* Max waypoints: 64000 (8 * 8000) */

    25

	WAYPOINT_POOL_BLOCK_SIZE_BITS = 3,       /* In bits, so (1 << 3) == 8 */

    26

	WAYPOINT_POOL_MAX_BLOCKS      = 8000,

    27

    28

	MAX_WAYPOINTS_PER_TOWN        = 64,

    29

};

    30

    31

/**

    32

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

    33

 */

    34

static void WaypointPoolNewBlock(uint start_item)

    35

{

    36

	Waypoint *wp;

    37

    38

	FOR_ALL_WAYPOINTS_FROM(wp, start_item) wp->index = start_item++;

    39

}

    40

    41

/* Initialize the town-pool */

    42

MemoryPool _waypoint_pool = { "Waypoints", WAYPOINT_POOL_MAX_BLOCKS, WAYPOINT_POOL_BLOCK_SIZE_BITS, sizeof(Waypoint), &WaypointPoolNewBlock, NULL, 0, 0, NULL };

    43

    44

/* Create a new waypoint */

    45

static Waypoint* AllocateWaypoint(void)

    46

{

    47

	Waypoint *wp;

    48

    49

	FOR_ALL_WAYPOINTS(wp) {

    50

		if (wp->xy == 0) {

    51

			uint index = wp->index;

    52

    53

			memset(wp, 0, sizeof(*wp));

    54

			wp->index = index;

    55

    56

			return wp;

    57

		}

    58

	}

    59

    60

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

    61

	if (AddBlockToPool(&_waypoint_pool)) return AllocateWaypoint();

    62

    63

	return NULL;

    64

}

    65

    66

/* Update the sign for the waypoint */

    67

static void UpdateWaypointSign(Waypoint* wp)

    68

{

    69

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

    70

	SetDParam(0, wp->index);

    71

	UpdateViewportSignPos(&wp->sign, pt.x, pt.y - 0x20, STR_WAYPOINT_VIEWPORT);

    72

}

    73

    74

/* Redraw the sign of a waypoint */

    75

static void RedrawWaypointSign(const Waypoint* wp)

    76

{

    77

	MarkAllViewportsDirty(

    78

		wp->sign.left - 6,

    79

		wp->sign.top,

    80

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

    81

		wp->sign.top + 48);

    82

}

    83

    84

/* Update all signs */

    85

void UpdateAllWaypointSigns(void)

    86

{

    87

	Waypoint *wp;

    88

    89

	FOR_ALL_WAYPOINTS(wp) {

    90

		if (wp->xy != 0) UpdateWaypointSign(wp);

    91

	}

    92

}

    93

    94

/* Set the default name for a waypoint */

    95

static void MakeDefaultWaypointName(Waypoint* wp)

    96

{

    97

	Waypoint *local_wp;

    98

	bool used_waypoint[MAX_WAYPOINTS_PER_TOWN];

    99

	int i;

   100

   101

	wp->town_index = ClosestTownFromTile(wp->xy, (uint)-1)->index;

   102

   103

	memset(used_waypoint, 0, sizeof(used_waypoint));

   104

   105

	/* Find an unused waypoint number belonging to this town */

   106

	FOR_ALL_WAYPOINTS(local_wp) {

   107

		if (wp == local_wp) continue;

   108

   109

		if (local_wp->xy && local_wp->string == STR_NULL && local_wp->town_index == wp->town_index)

   110

			used_waypoint[local_wp->town_cn] = true;

   111

	}

   112

   113

	/* Find an empty spot */

   114

	for (i = 0; used_waypoint[i] && i < MAX_WAYPOINTS_PER_TOWN; i++) {}

   115

   116

	wp->string = STR_NULL;

   117

	wp->town_cn = i;

   118

}

   119

   120

/* Find a deleted waypoint close to a tile. */

   121

static Waypoint *FindDeletedWaypointCloseTo(TileIndex tile)

   122

{

   123

	Waypoint *wp, *best = NULL;

   124

	uint thres = 8;

   125

   126

	FOR_ALL_WAYPOINTS(wp) {

   127

		if (wp->deleted && wp->xy != 0) {

   128

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

   129

   130

			if (cur_dist < thres) {

   131

				thres = cur_dist;

   132

				best = wp;

   133

			}

   134

		}

   135

	}

   136

   137

	return best;

   138

}

   139

   140

/**

   141

 * Update waypoint graphics id against saved GRFID/localidx.

   142

 * This is to ensure the chosen graphics are correct if GRF files are changed.

   143

 */

   144

void UpdateAllWaypointCustomGraphics(void)

   145

{

   146

	Waypoint *wp;

   147

   148

	FOR_ALL_WAYPOINTS(wp) {

   149

		uint i;

   150

   151

		if (wp->grfid == 0) continue;

   152

   153

		for (i = 0; i < GetNumCustomStations(STAT_CLASS_WAYP); i++) {

   154

			const StationSpec *statspec = GetCustomStationSpec(STAT_CLASS_WAYP, i);

   155

			if (statspec != NULL && statspec->grfid == wp->grfid && statspec->localidx == wp->localidx) {

   156

				wp->stat_id = i;

   157

				break;

   158

			}

   159

		}

   160

	}

   161

}

   162

   163

/** Convert existing rail to waypoint. Eg build a waypoint station over

   164

 * piece of rail

   165

 * @param tile tile where waypoint will be built

   166

 * @param p1 graphics for waypoint type, 0 indicates standard graphics

   167

 * @param p2 unused

   168

 *

   169

 * @todo When checking for the tile slope,

   170

 * distingush between "Flat land required" and "land sloped in wrong direction"

   171

 */

   172

int32 CmdBuildTrainWaypoint(TileIndex tile, uint32 flags, uint32 p1, uint32 p2)

   173

{

   174

	Waypoint *wp;

   175

	Slope tileh;

   176

	Axis axis;

   177

   178

	SET_EXPENSES_TYPE(EXPENSES_CONSTRUCTION);

   179

   180

	/* if custom gfx are used, make sure it is within bounds */

   181

	if (p1 >= GetNumCustomStations(STAT_CLASS_WAYP)) return CMD_ERROR;

   182

   183

	if (!IsTileType(tile, MP_RAILWAY) ||

   184

			GetRailTileType(tile) != RAIL_TILE_NORMAL || (

   185

				(axis = AXIS_X, GetTrackBits(tile) != TRACK_BIT_X) &&

   186

				(axis = AXIS_Y, GetTrackBits(tile) != TRACK_BIT_Y)

   187

			)) {

   188

		return_cmd_error(STR_1005_NO_SUITABLE_RAILROAD_TRACK);

   189

	}

   190

   191

	if (!CheckTileOwnership(tile)) return CMD_ERROR;

   192

	if (!EnsureNoVehicle(tile)) return CMD_ERROR;

   193

   194

	tileh = GetTileSlope(tile, NULL);

   195

	if (tileh != SLOPE_FLAT &&

   196

			(!_patches.build_on_slopes || IsSteepSlope(tileh) || !(tileh & (0x3 << axis)) || !(tileh & ~(0x3 << axis)))) {

   197

		return_cmd_error(STR_0007_FLAT_LAND_REQUIRED);

   198

	}

   199

   200

	/* Check if there is an already existing, deleted, waypoint close to us that we can reuse. */

   201

	wp = FindDeletedWaypointCloseTo(tile);

   202

	if (wp == NULL) {

   203

		wp = AllocateWaypoint();

   204

		if (wp == NULL) return CMD_ERROR;

   205

   206

		wp->town_index = 0;

   207

		wp->string = STR_NULL;

   208

		wp->town_cn = 0;

   209

	}

   210

   211

	if (flags & DC_EXEC) {

   212

		const StationSpec* statspec;

   213

   214

		MakeRailWaypoint(tile, GetTileOwner(tile), axis, GetRailType(tile), wp->index);

   215

		MarkTileDirtyByTile(tile);

   216

   217

		statspec = GetCustomStationSpec(STAT_CLASS_WAYP, p1);

   218

   219

		if (statspec != NULL) {

   220

			wp->stat_id = p1;

   221

			wp->grfid = statspec->grfid;

   222

			wp->localidx = statspec->localidx;

   223

		} else {

   224

			// Specified custom graphics do not exist, so use default.

   225

			wp->stat_id = 0;

   226

			wp->grfid = 0;

   227

			wp->localidx = 0;

   228

		}

   229

   230

		wp->deleted = 0;

   231

		wp->xy = tile;

   232

		wp->build_date = _date;

   233

   234

		if (wp->town_index == 0) MakeDefaultWaypointName(wp);

   235

   236

		UpdateWaypointSign(wp);

   237

		RedrawWaypointSign(wp);

   238

	}

   239

   240

	return _price.build_train_depot;

   241

}

   242

   243

/* Internal handler to delete a waypoint */

   244

static void DoDeleteWaypoint(Waypoint *wp)

   245

{

   246

	Order order;

   247

   248

	wp->xy = 0;

   249

   250

	order.type = OT_GOTO_WAYPOINT;

   251

	order.station = wp->index;

   252

	DeleteDestinationFromVehicleOrder(order);

   253

   254

	if (wp->string != STR_NULL) DeleteName(wp->string);

   255

   256

	RedrawWaypointSign(wp);

   257

}

   258

   259

/* Daily loop for waypoints */

   260

void WaypointsDailyLoop(void)

   261

{

   262

	Waypoint *wp;

   263

   264

	/* Check if we need to delete a waypoint */

   265

	FOR_ALL_WAYPOINTS(wp) {

   266

		if (wp->deleted != 0 && --wp->deleted == 0) DoDeleteWaypoint(wp);

   267

	}

   268

}

   269

   270

/* Remove a waypoint */

   271

int32 RemoveTrainWaypoint(TileIndex tile, uint32 flags, bool justremove)

   272

{

   273

	Waypoint *wp;

   274

   275

	/* Make sure it's a waypoint */

   276

	if (!IsTileType(tile, MP_RAILWAY) ||

   277

			!IsRailWaypoint(tile) ||

   278

			(!CheckTileOwnership(tile) && _current_player != OWNER_WATER) ||

   279

			!EnsureNoVehicle(tile)) {

   280

		return CMD_ERROR;

   281

	}

   282

   283

	if (flags & DC_EXEC) {

   284

		wp = GetWaypointByTile(tile);

   285

   286

		wp->deleted = 30; // let it live for this many days before we do the actual deletion.

   287

		RedrawWaypointSign(wp);

   288

   289

		if (justremove) {

   290

			MakeRailNormal(tile, GetTileOwner(tile), GetRailWaypointBits(tile), GetRailType(tile));

   291

			MarkTileDirtyByTile(tile);

   292

		} else {

   293

			DoClearSquare(tile);

   294

			SetSignalsOnBothDir(tile, GetRailWaypointTrack(tile));

   295

			YapfNotifyTrackLayoutChange(tile, GetRailWaypointTrack(tile));

   296

		}

   297

	}

   298

   299

	return _price.remove_train_depot;

   300

}

   301

   302

/** Delete a waypoint

   303

 * @param tile tile where waypoint is to be deleted

   304

 * @param p1 unused

   305

 * @param p2 unused

   306

 */

   307

int32 CmdRemoveTrainWaypoint(TileIndex tile, uint32 flags, uint32 p1, uint32 p2)

   308

{

   309

	SET_EXPENSES_TYPE(EXPENSES_CONSTRUCTION);

   310

	return RemoveTrainWaypoint(tile, flags, true);

   311

}

   312

   313

/** Rename a waypoint.

   314

 * @param tile unused

   315

 * @param p1 id of waypoint

   316

 * @param p2 unused

   317

 */

   318

int32 CmdRenameWaypoint(TileIndex tile, uint32 flags, uint32 p1, uint32 p2)

   319

{

   320

	Waypoint *wp;

   321

   322

	if (!IsWaypointIndex(p1)) return CMD_ERROR;

   323

   324

	if (_cmd_text[0] != '\0') {

   325

		StringID str = AllocateNameUnique(_cmd_text, 0);

   326

   327

		if (str == 0) return CMD_ERROR;

   328

   329

		if (flags & DC_EXEC) {

   330

			wp = GetWaypoint(p1);

   331

			if (wp->string != STR_NULL) DeleteName(wp->string);

   332

   333

			wp->string = str;

   334

			wp->town_cn = 0;

   335

   336

			UpdateWaypointSign(wp);

   337

			MarkWholeScreenDirty();

   338

		} else {

   339

			DeleteName(str);

   340

		}

   341

	} else {

   342

		if (flags & DC_EXEC) {

   343

			wp = GetWaypoint(p1);

   344

			if (wp->string != STR_NULL) DeleteName(wp->string);

   345

   346

			MakeDefaultWaypointName(wp);

   347

			UpdateWaypointSign(wp);

   348

			MarkWholeScreenDirty();

   349

		}

   350

	}

   351

	return 0;

   352

}

   353

   354

/* This hacks together some dummy one-shot Station structure for a waypoint. */

   355

Station *ComposeWaypointStation(TileIndex tile)

   356

{

   357

	Waypoint *wp = GetWaypointByTile(tile);

   358

	static Station stat;

   359

   360

	stat.train_tile = stat.xy = wp->xy;

   361

	stat.town = GetTown(wp->town_index);

   362

	stat.string_id = wp->string == STR_NULL ? /* FIXME? */ 0 : wp->string;

   363

	stat.build_date = wp->build_date;

   364

   365

	return &stat;

   366

}

   367

   368

/* Draw a waypoint */

   369

void DrawWaypointSprite(int x, int y, int stat_id, RailType railtype)

   370

{

   371

	x += 33;

   372

	y += 17;

   373

   374

	if (!DrawStationTile(x, y, railtype, AXIS_X, STAT_CLASS_WAYP, stat_id)) {

   375

		DrawDefaultWaypointSprite(x, y, railtype);

   376

	}

   377

}

   378

   379

/* Fix savegames which stored waypoints in their old format */

   380

void FixOldWaypoints(void)

   381

{

   382

	Waypoint *wp;

   383

   384

	/* Convert the old 'town_or_string', to 'string' / 'town' / 'town_cn' */

   385

	FOR_ALL_WAYPOINTS(wp) {

   386

		if (wp->xy == 0) continue;

   387

   388

		wp->town_index = ClosestTownFromTile(wp->xy, (uint)-1)->index;

   389

		wp->town_cn = 0;

   390

		if (wp->string & 0xC000) {

   391

			wp->town_cn = wp->string & 0x3F;

   392

			wp->string = STR_NULL;

   393

		}

   394

	}

   395

}

   396

   397

void InitializeWaypoints(void)

   398

{

   399

	CleanPool(&_waypoint_pool);

   400

	AddBlockToPool(&_waypoint_pool);

   401

}

   402

   403

static const SaveLoad _waypoint_desc[] = {

   410

   415

   416

	SLE_END()

   417

};

   418

   419

static void Save_WAYP(void)

   420

{

   421

	Waypoint *wp;

   422

   423

	FOR_ALL_WAYPOINTS(wp) {

   424

		if (wp->xy != 0) {

   425

			SlSetArrayIndex(wp->index);

   426

			SlObject(wp, _waypoint_desc);

   427

		}

   428

	}

   429

}

   430

   431

static void Load_WAYP(void)

   432

{

   433

	int index;

   434

   435

	while ((index = SlIterateArray()) != -1) {

   436

		Waypoint *wp;

   437

   438

		if (!AddBlockIfNeeded(&_waypoint_pool, index))

   439

			error("Waypoints: failed loading savegame: too many waypoints");

   440

   441

		wp = GetWaypoint(index);

   442

		SlObject(wp, _waypoint_desc);

   443

	}

   444

}

   445

   446

const ChunkHandler _waypoint_chunk_handlers[] = {

   447

	{ 'CHKP', Save_WAYP, Load_WAYP, CH_ARRAY | CH_LAST},

   448

};