1

#include "stdafx.h"

     2

#include "ttd.h"

     3

#include "table/strings.h"

     4

#include "map.h"

     5

#include "window.h"

     6

#include "gui.h"

     7

#include "viewport.h"

     8

#include "gfx.h"

     9

#include "sound.h"

    10

#include "command.h"

    11

#include "vehicle.h"

    12

#include "station.h"

    13

    14

static uint _cur_railtype;

    15

static bool _remove_button_clicked;

    16

static byte _build_depot_direction;

    17

static byte _waypoint_count=1;

    18

static byte _cur_waypoint_type;

    19

    20

struct {

    21

	byte orientation;

    22

	byte numtracks;

    23

	byte platlength;

    24

	bool dragdrop;

    25

} _railstation;

    26

    27

    28

static void HandleStationPlacement(uint start, uint end);

    29

static void ShowBuildTrainDepotPicker();

    30

static void ShowBuildWaypointPicker();

    31

static void ShowStationBuilder();

    32

    33

typedef void OnButtonClick(Window *w);

    34

    35

void CcPlaySound1E(bool success, uint tile, uint32 p1, uint32 p2)

    36

{

    37

	if (success) SndPlayTileFx(SND_20_SPLAT_2, tile);

    38

}

    39

    40

static void GenericPlaceRail(uint tile, int cmd)

    41

{

    42

	DoCommandP(tile, _cur_railtype, cmd, CcPlaySound1E,

    43

		_remove_button_clicked ?

    44

		CMD_REMOVE_SINGLE_RAIL | CMD_MSG(STR_1012_CAN_T_REMOVE_RAILROAD_TRACK) | CMD_AUTO | CMD_NO_WATER :

    45

		CMD_BUILD_SINGLE_RAIL | CMD_MSG(STR_1011_CAN_T_BUILD_RAILROAD_TRACK) | CMD_AUTO | CMD_NO_WATER

    46

		);

    47

}

    48

    49

static void PlaceRail_N(uint tile)

    50

{

    51

	int cmd = _tile_fract_coords.x > _tile_fract_coords.y ? 4 : 5;

    52

	GenericPlaceRail(tile, cmd);

    53

}

    54

    55

static void PlaceRail_NE(uint tile)

    56

{

    57

	VpStartPlaceSizing(tile, VPM_FIX_Y);

    58

}

    59

    60

static void PlaceRail_E(uint tile)

    61

{

    62

	int cmd = _tile_fract_coords.x + _tile_fract_coords.y <= 15 ? 2 : 3;

    63

	GenericPlaceRail(tile, cmd);

    64

}

    65

    66

static void PlaceRail_NW(uint tile)

    67

{

    68

	VpStartPlaceSizing(tile, VPM_FIX_X);

    69

}

    70

    71

static void PlaceRail_AutoRail(uint tile)

    72

{

    73

	VpStartPlaceSizing(tile, VPM_RAILDIRS);

    74

}

    75

    76

static void PlaceExtraDepotRail(uint tile, uint16 extra)

    77

{

    78

	byte b = _map5[tile];

    79

    80

	if (b & 0xC0 || !(b & (extra >> 8)))

    81

		return;

    82

    83

	DoCommandP(tile, _cur_railtype, extra & 0xFF, NULL, CMD_BUILD_SINGLE_RAIL | CMD_AUTO | CMD_NO_WATER);

    84

}

    85

    86

static const uint16 _place_depot_extra[12] = {

    87

	0x604,		0x2102,		0x1202,		0x505,

    88

	0x2400,		0x2801,		0x1800,		0x1401,

    89

	0x2203,		0x904,		0x0A05,		0x1103,

    90

};

    91

    92

    93

void CcRailDepot(bool success, uint tile, uint32 p1, uint32 p2)

    94

{

    95

	if (success) {

    96

		int dir = p2;

    97

    98

		SndPlayTileFx(SND_20_SPLAT_2, tile);

    99

		ResetObjectToPlace();

   100

   102

   103

		if (IS_TILETYPE(tile, MP_RAILWAY)) {

   104

			PlaceExtraDepotRail(tile, _place_depot_extra[dir]);

   105

			PlaceExtraDepotRail(tile, _place_depot_extra[dir + 4]);

   106

			PlaceExtraDepotRail(tile, _place_depot_extra[dir + 8]);

   107

		}

   108

	}

   109

}

   110

   111

static void PlaceRail_Depot(uint tile)

   112

{

   113

	DoCommandP(tile, _cur_railtype, _build_depot_direction, CcRailDepot,

   114

		CMD_BUILD_TRAIN_DEPOT | CMD_AUTO | CMD_NO_WATER | CMD_MSG(STR_100E_CAN_T_BUILD_TRAIN_DEPOT));

   115

}

   116

   117

static void PlaceRail_Waypoint(uint tile)

   118

{

   119

	if (!_remove_button_clicked) {

   120

		DoCommandP(tile, _waypoint_count > 0 ? (0x100 + _cur_waypoint_type) : 0, 0, CcPlaySound1E, CMD_BUILD_TRAIN_WAYPOINT | CMD_MSG(STR_CANT_BUILD_TRAIN_WAYPOINT));

   121

	} else {

   122

		DoCommandP(tile, 0, 0, CcPlaySound1E, CMD_REMOVE_TRAIN_WAYPOINT | CMD_MSG(STR_CANT_REMOVE_TRAIN_WAYPOINT));

   123

	}

   124

}

   125

   126

void CcStation(bool success, uint tile, uint32 p1, uint32 p2)

   127

{

   128

	if (success) {

   129

		SndPlayTileFx(SND_20_SPLAT_2, tile);

   130

		ResetObjectToPlace();

   131

	}

   132

}

   133

   134

static void PlaceRail_Station(uint tile)

   135

{

   136

	if(_remove_button_clicked)

   137

		DoCommandP(tile, 0, 0, CcPlaySound1E, CMD_REMOVE_FROM_RAILROAD_STATION | CMD_MSG(STR_CANT_REMOVE_PART_OF_STATION));

   138

	else if (_railstation.dragdrop) {

   139

		VpStartPlaceSizing(tile, VPM_X_AND_Y_LIMITED);

   140

		VpSetPlaceSizingLimit(_patches.station_spread);

   141

	} else {

   142

		// TODO: Custom station selector GUI. Now we just try using first custom station

   143

		// (and fall back to normal stations if it isn't available).

   144

		DoCommandP(tile, _railstation.orientation | (_railstation.numtracks<<8) | (_railstation.platlength<<16),_cur_railtype|1<<4, CcStation,

   145

				CMD_BUILD_RAILROAD_STATION | CMD_NO_WATER | CMD_AUTO | CMD_MSG(STR_100F_CAN_T_BUILD_RAILROAD_STATION));

   146

	}

   147

}

   148

   149

static void GenericPlaceSignals(uint tile)

   150

{

   151

	uint trackstat;

   152

	int i;

   153

   154

	trackstat = (byte)GetTileTrackStatus(tile, TRANSPORT_RAIL);

   155

   156

	if ((trackstat & 0x30) == 0x30) {

   157

		trackstat = (_tile_fract_coords.x <= _tile_fract_coords.y) ? 0x20 : 0x10;

   158

	}

   159

   160

	if ((trackstat & 0x0C) == 0x0C) {

   161

		trackstat = (_tile_fract_coords.x + _tile_fract_coords.y <= 15) ? 4 : 8;

   162

	}

   163

   164

	// Lookup the bit index

   165

	i = 0;

   166

	if (trackstat != 0) {	while (!(trackstat & 1)) { i++; trackstat >>= 1; }}

   167

   168

	if (!_remove_button_clicked) {

   169

		DoCommandP(tile, i + (_ctrl_pressed ? 8 : 0), 0, CcPlaySound1E,

   170

			CMD_BUILD_SIGNALS | CMD_AUTO | CMD_MSG(STR_1010_CAN_T_BUILD_SIGNALS_HERE));

   171

	} else {

   172

		DoCommandP(tile, i, 0, CcPlaySound1E,

   173

			CMD_REMOVE_SIGNALS | CMD_AUTO | CMD_MSG(STR_1013_CAN_T_REMOVE_SIGNALS_FROM));

   174

	}

   175

}

   176

   177

static void PlaceRail_Bridge(uint tile)

   178

{

   179

	VpStartPlaceSizing(tile, VPM_X_OR_Y);

   180

}

   181

   182

void CcBuildRailTunnel(bool success, uint tile, uint32 p1, uint32 p2)

   183

{

   184

	if (success) {

   185

		SndPlayTileFx(SND_20_SPLAT_2, tile);

   186

		ResetObjectToPlace();

   187

	} else {

   188

		SetRedErrorSquare(_build_tunnel_endtile);

   189

	}

   190

}

   191

   192

static void PlaceRail_Tunnel(uint tile)

   193

{

   194

	DoCommandP(tile, _cur_railtype, 0, CcBuildRailTunnel,

   195

		CMD_BUILD_TUNNEL | CMD_AUTO | CMD_MSG(STR_5016_CAN_T_BUILD_TUNNEL_HERE));

   196

}

   197

   198

void PlaceProc_BuyLand(uint tile)

   199

{

   200

	DoCommandP(tile, 0, 0, CcPlaySound1E, CMD_PURCHASE_LAND_AREA | CMD_AUTO | CMD_NO_WATER | CMD_MSG(STR_5806_CAN_T_PURCHASE_THIS_LAND));

   201

}

   202

   203

static void PlaceRail_ConvertRail(uint tile)

   204

{

   205

	VpStartPlaceSizing(tile, VPM_X_AND_Y | (1<<4));

   206

}

   207

   208

static void PlaceRail_AutoSignals(uint tile)

   209

{

   210

	VpStartPlaceSizing(tile, VPM_SIGNALDIRS);

   211

}

   212

   213

static void BuildRailClick_N(Window *w)

   214

{

   215

	HandlePlacePushButton(w, 4, _cur_railtype*4 + 0x4EF, 1, PlaceRail_N);

   216

}

   217

   218

static void BuildRailClick_NE(Window *w)

   219

{

   220

	HandlePlacePushButton(w, 5, _cur_railtype*4 + 0x4F0, 1, PlaceRail_NE);

   221

}

   222

   223

static void BuildRailClick_E(Window *w)

   224

{

   225

	HandlePlacePushButton(w, 6, _cur_railtype*4 + 0x4F1, 1, PlaceRail_E);

   226

}

   227

   228

static void BuildRailClick_NW(Window *w)

   229

{

   230

	HandlePlacePushButton(w, 7, _cur_railtype*4 + 0x4F2, 1, PlaceRail_NW);

   231

}

   232

   233

static void BuildRailClick_AutoRail(Window *w)

   234

{

   235

	HandlePlacePushButton(w, 8, _cur_railtype + SPR_OPENTTD_BASE + 4, 1, PlaceRail_AutoRail);

   236

}

   237

   238

static void BuildRailClick_Demolish(Window *w)

   239

{

   240

	HandlePlacePushButton(w, 9, ANIMCURSOR_DEMOLISH, 1, PlaceProc_DemolishArea);

   241

}

   242

   243

static const SpriteID _depot_cursors[] = {

   244

	0x510,

   245

	SPR_OPENTTD_BASE + 14,

   246

	SPR_OPENTTD_BASE + 15,

   247

};

   248

   249

static void BuildRailClick_Depot(Window *w)

   250

{

   251

	if (HandlePlacePushButton(w, 10, _depot_cursors[_cur_railtype], 1, PlaceRail_Depot)) ShowBuildTrainDepotPicker();

   252

}

   253

   254

static void BuildRailClick_Waypoint(Window *w)

   255

{

   256

	_waypoint_count = GetCustomStationsCount(STAT_CLASS_WAYP);

   257

	if (HandlePlacePushButton(w, 11, SPR_OPENTTD_BASE + 7, 1, PlaceRail_Waypoint)

   258

	    && _waypoint_count > 1)

   259

		ShowBuildWaypointPicker();

   260

}

   261

   262

static void BuildRailClick_Station(Window *w)

   263

{

   264

	if (HandlePlacePushButton(w, 12, 0x514, 1, PlaceRail_Station)) ShowStationBuilder();

   265

}

   266

   267

static void BuildRailClick_AutoSignals(Window *w)

   268

{

   269

	HandlePlacePushButton(w, 13, ANIMCURSOR_BUILDSIGNALS , 1, PlaceRail_AutoSignals);

   270

}

   271

   272

static void BuildRailClick_Bridge(Window *w)

   273

{

   274

	HandlePlacePushButton(w, 14, 0xA21, 1, PlaceRail_Bridge);

   275

}

   276

   277

static void BuildRailClick_Tunnel(Window *w)

   278

{

   279

	HandlePlacePushButton(w, 15, 0x982 + _cur_railtype, 3, PlaceRail_Tunnel);

   280

}

   281

   282

static void BuildRailClick_Remove(Window *w)

   283

{

   284

	if (w->disabled_state & (1<<16))

   285

		return;

   286

	SetWindowDirty(w);

   287

	SndPlayFx(SND_15_BEEP);

   288

   289

	_thd.make_square_red = !!((w->click_state ^= (1 << 16)) & (1<<16));

   290

	MarkTileDirty(_thd.pos.x, _thd.pos.y);

   291

	_remove_button_clicked = (w->click_state & (1 << 16)) != 0;

   292

   293

	// handle station builder

   294

	if( w->click_state & (1 << 16) )

   295

	{

   296

		if(_remove_button_clicked)

   297

			SetTileSelectSize(1, 1);

   298

		else

   299

			BringWindowToFrontById(WC_BUILD_STATION, 0);

   300

	}

   301

}

   302

   303

static void BuildRailClick_Convert(Window *w)

   304

{

   305

	HandlePlacePushButton(w, 17, (SPR_OPENTTD_BASE + 26) + _cur_railtype * 2, 1, PlaceRail_ConvertRail);

   306

}

   307

   308

static void BuildRailClick_Landscaping(Window *w)

   309

{

   310

	ShowTerraformToolbar();

   311

}

   312

   313

   314

static void DoRailroadTrack(int mode)

   315

{

   316

	DoCommandP(TILE_FROM_XY(_thd.selstart.x, _thd.selstart.y), PACK_POINT(_thd.selend.x, _thd.selend.y), (mode << 4) | _cur_railtype, NULL,

   317

		_remove_button_clicked ?

   318

		CMD_REMOVE_RAILROAD_TRACK | CMD_AUTO | CMD_NO_WATER | CMD_MSG(STR_1012_CAN_T_REMOVE_RAILROAD_TRACK) :

   319

		CMD_BUILD_RAILROAD_TRACK | CMD_AUTO | CMD_NO_WATER | CMD_MSG(STR_1011_CAN_T_BUILD_RAILROAD_TRACK)

   320

	);

   321

}

   322

   323

typedef struct {

   324

	byte bit, a,b, mouse;

   325

} BestFitStruct;

   326

   327

#define M(d,m) (d << 6) | (m)

   328

static const BestFitStruct _bestfit[] = {

   329

	// both edges have rail

   330

	{2, M(0, 1+8), M(3, 2+8), 0}, // upper track

   331

	{3, M(2, 1+4), M(1, 2+4), 0}, // lower track

   332

   333

	{4, M(2, 1+32), M(3, 2+32), 1<<2}, // left track

   334

	{5, M(0, 1+16), M(1, 2+16), 1<<3}, // right track

   335

   336

	{0, M(0,1+8+16), M(2,1+4+32), 0}, // diag1 track

   337

	{1, M(3,2+8+32), M(1,2+4+16), 0}, // diag2 track

   338

   339

	// one edge with rail

   340

	{0, M(0,1), 0, 0}, // diag1 track

   341

	{0, M(2,1), 0, 0}, // diag1 track

   342

   343

	{1, M(1,2), 0, 0}, // diag2 track

   344

	{1, M(3,2), 0, 0}, // diag2 track

   345

   346

	{2, M(0,8), 0, 1<<0}, // upper track

   347

	{2, M(3,8), 0, 1<<0}, // upper track

   348

   349

	{3, M(1,4), 0, 1<<1}, // lower track

   350

	{3, M(2,4), 0, 1<<1}, // lower track

   351

   352

	{4, M(2,32), 0, 1<<2}, // left track

   353

	{4, M(3,32), 0, 1<<2}, // left track

   354

   355

	{5, M(0,16), 0, 1<<3}, // right track

   356

	{5, M(1,16), 0, 1<<3}, // right track

   357

   358

	{0xff, 0, 0, 0},

   359

};

   360

#undef M

   361

   362

static int GetBestArea(int x, int y)

   363

{

   364

	int r = 0;

   365

	if (x + y > 0x10) r += 2;

   366

	else if (x + y == 0x10) return -1;

   367

	if (y - x > 0) r += 1;

   368

	else if (y - x == 0) return -1;

   369

	return r;

   370

}

   371

   372

static int GetBestFit1x1(int x, int y)

   373

{

   374

	byte m[5];

   375

	const BestFitStruct *bfs;

   376

	byte mouse;

   377

	uint tile;

   378

	int best;

   379

	int i;

   380

   381

	// determine the mouse regions

   382

	mouse = ((x & 0xF) + (y & 0xF) < 0x10 ? 1 << 0 : 1 << 1) +

   383

					((x & 0xF) > (y & 0xF) ? 1 << 2 : 1 << 3);

   384

   385

	// get the rail in each direction

   386

	tile = TILE_FROM_XY(x,y);

   387

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

   388

		static TileIndexDiff _tile_inc[5] = {

   389

			TILE_XY(-1, 0),

   390

			TILE_XY(0, 1) - TILE_XY(-1, 0),

   391

			TILE_XY(1, 0) - TILE_XY(0, 1),

   392

			TILE_XY(0, -1) - TILE_XY(1, 0),

   393

			TILE_XY(0, 0) - TILE_XY(0, -1),

   394

		};

   395

		tile += _tile_inc[i];

   396

		m[i] = 0;

   397

		if (IS_TILETYPE(tile, MP_RAILWAY) && _map5[tile] < 0x80)

   398

			m[i] = _map5[tile]&0x3F;

   399

   400

		// handle tracks under bridge

   401

		if(IS_TILETYPE(tile, MP_TUNNELBRIDGE) && (_map5[tile]&0xF8)==0xE0)

   402

			m[i] = (byte) !(_map5[tile]&0x01) + 1;

   403

   404

		if (_remove_button_clicked) m[i] ^= 0x3F;

   405

	}

   406

   407

	// check "mouse gesture"?

   408

	{

   409

		int a1,a2;

   410

		if ((a1 = GetBestArea(x & 0xF, y & 0xF)) != -1 && (a2 = GetBestArea(_tile_fract_coords.x, _tile_fract_coords.y)) != -1 && a1 != a2) {

   411

			static const byte _get_dir_by_areas[4][4] = {

   412

				{0,2,4,1},

   413

				{2,0,0,5},

   414

				{4,0,0,3},

   415

				{1,5,3,0},

   416

			};

   417

			i = _get_dir_by_areas[a2][a1];

   418

			if (!HASBIT(m[4], i))

   419

				return i;

   420

		}

   421

	}

   422

	// check each bestfit struct

   423

	for(bfs = _bestfit, best=-1; bfs->bit != 0xFF; bfs++) {

   424

		if ((bfs->a & m[bfs->a >> 6]) && (!bfs->b || bfs->b & m[bfs->b >> 6]) && !HASBIT(m[4], bfs->bit)) {

   425

			if ( (byte)(~mouse & bfs->mouse) == 0)

   426

				return bfs->bit;

   427

			if (best != -1)

   428

				return best;

   429

			best = bfs->bit;

   430

		}

   431

	}

   432

   433

	return best;

   434

}

   435

   436

static void SwapSelection()

   437

{

   438

	TileHighlightData *thd = &_thd;

   439

	Point pt = thd->selstart;

   440

	thd->selstart.x = thd->selend.x & ~0xF;

   441

	thd->selstart.y = thd->selend.y & ~0xF;

   442

	thd->selend = pt;

   443

}

   444

   445

static bool Check2x1AutoRail(int mode)

   446

{

   447

	TileHighlightData *thd = &_thd;

   448

	int fxpy = _tile_fract_coords.x + _tile_fract_coords.y;

   449

	int sxpy = (thd->selend.x & 0xF) + (thd->selend.y & 0xF);

   450

	int fxmy = _tile_fract_coords.x - _tile_fract_coords.y;

   451

	int sxmy = (thd->selend.x & 0xF) - (thd->selend.y & 0xF);

   452

   453

	switch(mode) {

   454

	case 0:

   455

		if (fxpy >= 20 && sxpy <= 12) { SwapSelection(); DoRailroadTrack(0); return true; }

   456

		if (fxmy < -3 && sxmy > 3) { DoRailroadTrack(0); return true; }

   457

		break;

   458

   459

	case 1:

   460

		if (fxmy > 3 && sxmy < -3) { SwapSelection(); DoRailroadTrack(0); return true; }

   461

		if (fxpy <= 12 && sxpy >= 20) { DoRailroadTrack(0); return true; }

   462

		break;

   463

   464

	case 2:

   465

		if (fxmy > 3 && sxmy < -3) { DoRailroadTrack(3); return true; }

   466

		if (fxpy >= 20 && sxpy <= 12) { SwapSelection(); DoRailroadTrack(0); return true; }

   467

		break;

   468

   469

	case 3:

   470

		if (fxmy < -3 && sxmy > 3) { SwapSelection(); DoRailroadTrack(3); return true; }

   471

		if (fxpy <= 12 && sxpy >= 20) { DoRailroadTrack(0); return true; }

   472

		break;

   473

	}

   474

   475

	return false;

   476

}

   477

   478

   479

static void HandleAutodirPlacement()

   480

{

   481

	TileHighlightData *thd = &_thd;

   482

	int bit;

   483

   484

	if (thd->drawstyle == HT_RECT) {

   485

		int dx = thd->selstart.x - (thd->selend.x&~0xF);

   486

		int dy = thd->selstart.y - (thd->selend.y&~0xF);

   487

   488

		if (dx == 0 && dy == 0 ) {

   489

			// 1x1 tile

   490

			bit = GetBestFit1x1(thd->selend.x, thd->selend.y);

   491

			if (bit == -1) return;

   492

			GenericPlaceRail(TILE_FROM_XY(thd->selend.x, thd->selend.y), bit);

   493

		} else if (dx == 0) {

   494

			if (dy == -16) {

   495

				if (Check2x1AutoRail(0)) return;

   496

			} else if (dy == 16) {

   497

				if (Check2x1AutoRail(1)) return;

   498

			}

   499

			// same x coordinate

   500

			DoRailroadTrack(VPM_FIX_X);

   501

		} else {

   502

			// same y coordinate

   503

			// check it's it -16 or 16, then we must check if it should be normal tiles or special tiles.

   504

			if (dx == -16) {

   505

				if (Check2x1AutoRail(2)) return;

   506

			} else if (dx == 16) {

   507

				if (Check2x1AutoRail(3)) return;

   508

			}

   509

			DoRailroadTrack(VPM_FIX_Y);

   510

		}

   511

	} else {

   512

		DoRailroadTrack(thd->drawstyle & 1 ? 0 : 3);

   513

	}

   514

}

   515

   516

static void HandleAutoSignalPlacement()