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/sprites.h"

    19

#include "table/strings.h"

    20

    21

enum {

    22

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

    23

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

    24

	WAYPOINT_POOL_MAX_BLOCKS      = 8000,

    25

    26

	MAX_WAYPOINTS_PER_TOWN        = 64,

    27

};

    28

    29

/**

    30

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

    31

 */

    32

static void WaypointPoolNewBlock(uint start_item)

    33

{

    34

	Waypoint *wp;

    35

    36

	FOR_ALL_WAYPOINTS_FROM(wp, start_item)

    37

		wp->index = start_item++;

    38

}

    39

    40

/* Initialize the town-pool */

    41

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

    42

    43

/* Create a new waypoint */

    44

static Waypoint* AllocateWaypoint(void)

    45

{

    46

	Waypoint *wp;

    47

    48

	FOR_ALL_WAYPOINTS(wp) {

    49

		if (wp->xy == 0) {

    50

			uint index = wp->index;

    51

    52

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

    53

			wp->index = index;

    54

    55

			return wp;

    56

		}

    57

	}

    58

    59

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

    60

	if (AddBlockToPool(&_waypoint_pool))

    61

		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) * 16, TileY(wp->xy) * 16);

    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)

    91

			UpdateWaypointSign(wp);

    92

	}

    93

}

    94

    95

/* Set the default name for a waypoint */

    96

static void MakeDefaultWaypointName(Waypoint* wp)

    97

{

    98

	Waypoint *local_wp;

    99

	bool used_waypoint[MAX_WAYPOINTS_PER_TOWN];

   100

	int i;

   101

   102

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

   103

   104

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

   105

   106

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

   107

	FOR_ALL_WAYPOINTS(local_wp) {

   108

		if (wp == local_wp)

   109

			continue;

   110

   111

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

   112

			used_waypoint[local_wp->town_cn] = true;

   113

	}

   114

   115

	/* Find an empty spot */

   116

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

   117

   118

	wp->string = STR_NULL;

   119

	wp->town_cn = i;

   120

}

   121

   122

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

   123

static Waypoint *FindDeletedWaypointCloseTo(TileIndex tile)

   124

{

   125

	Waypoint *wp, *best = NULL;

   126

	uint thres = 8, cur_dist;

   127

   128

	FOR_ALL_WAYPOINTS(wp) {

   129

		if (wp->deleted && wp->xy) {

   130

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

   131

			if (cur_dist < thres) {

   132

				thres = cur_dist;

   133

				best = wp;

   134

			}

   135

		}

   136

	}

   137

   138

	return best;

   139

}

   140

   141

/**

   142

 * Update waypoint graphics id against saved GRFID/localidx.

   143

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

   144

 */

   145

void UpdateAllWaypointCustomGraphics(void)

   146

{

   147

	Waypoint *wp;

   148

   149

	FOR_ALL_WAYPOINTS(wp) {

   150

		uint i;

   151

   152

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

   153

   154

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

   155

			const StationSpec *spec = GetCustomStation(STAT_CLASS_WAYP, i);

   156

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

   157

				wp->stat_id = i;

   158

				break;

   159

			}

   160

		}

   161

	}

   162

}

   163

   164

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

   165

 * piece of rail

   166

 * @param x,y coordinates where waypoint will be built

   167

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

   168

 * @param p2 unused

   169

 *

   170

 * @todo When checking for the tile slope,

   171

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

   172

 */

   173

int32 CmdBuildTrainWaypoint(int x, int y, uint32 flags, uint32 p1, uint32 p2)

   174

{

   175

	TileIndex tile = TileVirtXY(x, y);

   176

	Waypoint *wp;

   177

	uint tileh;

   178

	Axis axis;

   179

   180

	SET_EXPENSES_TYPE(EXPENSES_CONSTRUCTION);

   181

   182

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

   183

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

   184

   185

	if (!IsTileType(tile, MP_RAILWAY) ||

   186

			GetRailTileType(tile) != RAIL_TYPE_NORMAL || (

   187

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

   188

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

   189

			)) {

   190

		return_cmd_error(STR_1005_NO_SUITABLE_RAILROAD_TRACK);

   191

	}

   192

   193

	if (!CheckTileOwnership(tile))

   194

		return CMD_ERROR;

   195

   196

	if (!EnsureNoVehicle(tile)) return CMD_ERROR;

   197

   198

	tileh = GetTileSlope(tile, NULL);

   199

	if (tileh != 0) {

   200

		if (!_patches.build_on_slopes || IsSteepTileh(tileh) || !(tileh & (0x3 << axis)) || !(tileh & ~(0x3 << axis)))

   201

			return_cmd_error(STR_0007_FLAT_LAND_REQUIRED);

   202

	}

   203

   204

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

   205

	wp = FindDeletedWaypointCloseTo(tile);

   206

	if (wp == NULL) {

   207

		wp = AllocateWaypoint();

   208

		if (wp == NULL) return CMD_ERROR;

   209

   210

		wp->town_index = 0;

   211

		wp->string = STR_NULL;

   212

		wp->town_cn = 0;

   213

	}

   214

   215

	if (flags & DC_EXEC) {

   216

		const StationSpec *spec = NULL;

   217

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

   218

		MarkTileDirtyByTile(tile);

   219

   220

		if (GB(p1, 0, 8) < GetNumCustomStations(STAT_CLASS_WAYP))

   221

			spec = GetCustomStation(STAT_CLASS_WAYP, GB(p1, 0, 8));

   222

   223

		if (spec != NULL) {

   224

			SETBIT(_m[tile].m3, 4);

   225

			wp->stat_id = GB(p1, 0, 8);

   226

			wp->grfid = spec->grfid;

   227

			wp->localidx = spec->localidx;

   228

		} else {

   229

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

   230

			CLRBIT(_m[tile].m3, 4);

   231

			wp->stat_id = 0;

   232

			wp->grfid = 0;

   233

			wp->localidx = 0;

   234

		}

   235

   236

		wp->deleted = 0;

   237

		wp->xy = tile;

   238

		wp->build_date = _date;

   239

   240

		if (wp->town_index == STR_NULL)

   241

			MakeDefaultWaypointName(wp);

   242

   243

		UpdateWaypointSign(wp);

   244

		RedrawWaypointSign(wp);

   245

	}

   246

   247

	return _price.build_train_depot;

   248

}

   249

   250

/* Internal handler to delete a waypoint */

   251

static void DoDeleteWaypoint(Waypoint *wp)

   252

{

   253

	Order order;

   254

   255

	wp->xy = 0;

   256

   257

	order.type = OT_GOTO_WAYPOINT;

   258

	order.station = wp->index;

   259

	DeleteDestinationFromVehicleOrder(order);

   260

   261

	if (wp->string != STR_NULL)

   262

		DeleteName(wp->string);

   263

   264

	RedrawWaypointSign(wp);

   265

}

   266

   267

/* Daily loop for waypoints */

   268

void WaypointsDailyLoop(void)

   269

{

   270

	Waypoint *wp;

   271

   272

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

   273

	FOR_ALL_WAYPOINTS(wp) {

   274

		if (wp->deleted && !--wp->deleted) {

   275

			DoDeleteWaypoint(wp);

   276

		}

   277

	}

   278

}

   279

   280

/* Remove a waypoint */

   281

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

   282

{

   283

	Waypoint *wp;

   284

   285

	/* Make sure it's a waypoint */

   286

	if (!IsTileType(tile, MP_RAILWAY) || !IsRailWaypoint(tile))

   287

		return CMD_ERROR;

   288

   289

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

   290

		return CMD_ERROR;

   291

   292

	if (!EnsureNoVehicle(tile))

   293

		return CMD_ERROR;

   294

   295

	if (flags & DC_EXEC) {

   296

		Axis direction = _m[tile].m5 & RAIL_WAYPOINT_TRACK_MASK;

   297

   298

		wp = GetWaypointByTile(tile);

   299

   300

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

   301

		RedrawWaypointSign(wp);

   302

   303

		if (justremove) {

   304

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

   305

			MarkTileDirtyByTile(tile);

   306

		} else {

   307

			DoClearSquare(tile);

   308

			SetSignalsOnBothDir(tile, direction == AXIS_X ? TRACK_X : TRACK_Y);

   309

		}

   310

	}

   311

   312

	return _price.remove_train_depot;

   313

}

   314

   315

/** Delete a waypoint

   316

 * @param x,y coordinates where waypoint is to be deleted

   317

 * @param p1 unused

   318

 * @param p2 unused

   319

 */

   320

int32 CmdRemoveTrainWaypoint(int x, int y, uint32 flags, uint32 p1, uint32 p2)

   321

{

   322

	TileIndex tile = TileVirtXY(x, y);

   323

	SET_EXPENSES_TYPE(EXPENSES_CONSTRUCTION);

   324

	return RemoveTrainWaypoint(tile, flags, true);

   325

}

   326

   327

/** Rename a waypoint.

   328

 * @param x,y unused

   329

 * @param p1 id of waypoint

   330

 * @param p2 unused

   331

 */

   332

int32 CmdRenameWaypoint(int x, int y, uint32 flags, uint32 p1, uint32 p2)

   333

{

   334

	Waypoint *wp;

   335

	StringID str;

   336

   337

	if (!IsWaypointIndex(p1)) return CMD_ERROR;

   338

   339

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

   340

		str = AllocateNameUnique(_cmd_text, 0);

   341

		if (str == 0)

   342

			return CMD_ERROR;

   343

   344

		if (flags & DC_EXEC) {

   345

			wp = GetWaypoint(p1);

   346

			if (wp->string != STR_NULL)

   347

				DeleteName(wp->string);

   348

   349

			wp->string = str;

   350

			wp->town_cn = 0;

   351

   352

			UpdateWaypointSign(wp);

   353

			MarkWholeScreenDirty();

   354

		} else {

   355

			DeleteName(str);

   356

		}

   357

	} else {

   358

		if (flags & DC_EXEC) {

   359

			wp = GetWaypoint(p1);

   360

			if (wp->string != STR_NULL)

   361

				DeleteName(wp->string);

   362

   363

			MakeDefaultWaypointName(wp);

   364

			UpdateWaypointSign(wp);

   365

			MarkWholeScreenDirty();

   366

		}

   367

	}

   368

	return 0;

   369

}

   370

   371

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

   372

Station *ComposeWaypointStation(TileIndex tile)

   373

{

   374

	Waypoint *wp = GetWaypointByTile(tile);

   375

	static Station stat;

   376

   377

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

   378

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

   379

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

   380

	stat.build_date = wp->build_date;

   381

	stat.class_id = 6;

   382

	stat.stat_id = wp->stat_id;

   383

   384

	return &stat;

   385

}

   386

   387

extern uint16 _custom_sprites_base;

   388

   389

   390

/* Draw a waypoint */

   391

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

   392

{

   393

	const StationSpec *stat;

   394

	uint32 relocation;

   395

	const DrawTileSprites *cust;

   396

	DrawTileSeqStruct const *seq;

   397

	const RailtypeInfo *rti = GetRailTypeInfo(railtype);

   398

	uint32 ormod, img;

   399

   400

	ormod = SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player));

   401

   402

	x += 33;

   403

	y += 17;

   404

   405

	stat = GetCustomStation(STAT_CLASS_WAYP, stat_id);

   406

	if (stat == NULL) {

   407

		// stat is NULL for default waypoints and when waypoint graphics are

   408

		// not loaded.

   409

		DrawDefaultWaypointSprite(x, y, railtype);

   410

		return;

   411

	}

   412

   413

	relocation = GetCustomStationRelocation(stat, NULL, 1);

   414

	// emulate station tile - open with building

   415

	// add 1 to get the other direction

   416

	cust = &stat->renderdata[2];

   417

   418

	img = cust->ground_sprite;

   419

	img += (img < _custom_sprites_base) ? rti->total_offset : railtype;

   420

   421

	if (img & PALETTE_MODIFIER_COLOR) img = (img & SPRITE_MASK);

   422

	DrawSprite(img, x, y);

   423

   424

	foreach_draw_tile_seq(seq, cust->seq) {

   425

		Point pt = RemapCoords(seq->delta_x, seq->delta_y, seq->delta_z);

   426

		uint32 image = seq->image + relocation;

   427

		DrawSprite((image & SPRITE_MASK) | ormod, x + pt.x, y + pt.y);

   428

	}

   429

}

   430

   431

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

   432

void FixOldWaypoints(void)

   433

{

   434

	Waypoint *wp;

   435

   436

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

   437

	FOR_ALL_WAYPOINTS(wp) {

   438

		if (wp->xy == 0)

   439

			continue;

   440

   441

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

   442

		wp->town_cn = 0;

   443

		if (wp->string & 0xC000) {

   444

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

   445

			wp->string = STR_NULL;

   446

		}

   447

	}

   448

}

   449

   450

void InitializeWaypoints(void)

   451

{

   452

	CleanPool(&_waypoint_pool);

   453

	AddBlockToPool(&_waypoint_pool);

   454

}

   455

   456

static const SaveLoad _waypoint_desc[] = {

   457

	SLE_CONDVAR(Waypoint, xy, SLE_FILE_U16 | SLE_VAR_U32, 0, 5),

   458

	SLE_CONDVAR(Waypoint, xy, SLE_UINT32, 6, SL_MAX_VERSION),

   459

	SLE_CONDVAR(Waypoint, town_index, SLE_UINT16, 12, SL_MAX_VERSION),

   460

	SLE_CONDVAR(Waypoint, town_cn, SLE_UINT8, 12, SL_MAX_VERSION),

   461

	SLE_VAR(Waypoint, string, SLE_UINT16),

   462

	SLE_VAR(Waypoint, deleted, SLE_UINT8),

   463

   464

	SLE_CONDVAR(Waypoint, build_date, SLE_UINT16,  3, SL_MAX_VERSION),

   465

	SLE_CONDVAR(Waypoint, localidx,   SLE_UINT8,   3, SL_MAX_VERSION),

   466

	SLE_CONDVAR(Waypoint, grfid,      SLE_UINT32, 17, SL_MAX_VERSION),

   467

   468

	SLE_END()

   469

};

   470

   471

static void Save_WAYP(void)

   472

{

   473

	Waypoint *wp;

   474

   475

	FOR_ALL_WAYPOINTS(wp) {

   476

		if (wp->xy != 0) {

   477

			SlSetArrayIndex(wp->index);

   478

			SlObject(wp, _waypoint_desc);

   479

		}

   480

	}

   481

}

   482

   483

static void Load_WAYP(void)

   484

{

   485

	int index;

   486

   487

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

   488

		Waypoint *wp;

   489

   490

		if (!AddBlockIfNeeded(&_waypoint_pool, index))

   491

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

   492

   493

		wp = GetWaypoint(index);

   494

		SlObject(wp, _waypoint_desc);

   495

	}

   496

}

   497

   498

const ChunkHandler _waypoint_chunk_handlers[] = {

   499

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

   500

};