1

#include "stdafx.h"

     2

#include "openttd.h"

     3

#include "table/sprites.h"

     4

#include "table/strings.h"

     6

#include "map.h"

     7

#include "tile.h"

     8

#include "window.h"

     9

#include "gui.h"

    10

#include "viewport.h"

    11

#include "gfx.h"

    12

#include "sound.h"

    13

#include "command.h"

    14

#include "vehicle.h"

    15

#include "station.h"

    16

#include "waypoint.h"

    17

#include "debug.h"

    18

#include "variables.h"

    19

    20

static uint _cur_railtype;

    21

static bool _remove_button_clicked;

    22

static byte _build_depot_direction;

    23

static byte _waypoint_count=1;

    24

static byte _cur_waypoint_type;

    25

    26

struct {

    27

	byte orientation;

    28

	byte numtracks;

    29

	byte platlength;

    30

	bool dragdrop;

    31

} _railstation;

    32

    33

    34

static void HandleStationPlacement(TileIndex start, TileIndex end);

    35

static void ShowBuildTrainDepotPicker(void);

    36

static void ShowBuildWaypointPicker(void);

    37

static void ShowStationBuilder(void);

    38

    39

typedef void OnButtonClick(Window *w);

    40

    41

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

    42

{

    43

	if (success) SndPlayTileFx(SND_20_SPLAT_2, tile);

    44

}

    45

    46

static void GenericPlaceRail(TileIndex tile, int cmd)

    47

{

    48

	DoCommandP(tile, _cur_railtype, cmd, CcPlaySound1E,

    49

		_remove_button_clicked ?

    50

		CMD_REMOVE_SINGLE_RAIL | CMD_MSG(STR_1012_CAN_T_REMOVE_RAILROAD_TRACK) | CMD_AUTO | CMD_NO_WATER :

    51

		CMD_BUILD_SINGLE_RAIL | CMD_MSG(STR_1011_CAN_T_BUILD_RAILROAD_TRACK) | CMD_AUTO | CMD_NO_WATER

    52

		);

    53

}

    54

    55

static void PlaceRail_N(TileIndex tile)

    56

{

    57

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

    58

	GenericPlaceRail(tile, cmd);

    59

}

    60

    61

static void PlaceRail_NE(TileIndex tile)

    62

{

    63

	VpStartPlaceSizing(tile, VPM_FIX_Y);

    64

}

    65

    66

static void PlaceRail_E(TileIndex tile)

    67

{

    68

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

    69

	GenericPlaceRail(tile, cmd);

    70

}

    71

    72

static void PlaceRail_NW(TileIndex tile)

    73

{

    74

	VpStartPlaceSizing(tile, VPM_FIX_X);

    75

}

    76

    77

static void PlaceRail_AutoRail(TileIndex tile)

    78

{

    79

	VpStartPlaceSizing(tile, VPM_RAILDIRS);

    80

}

    81

    82

static void PlaceExtraDepotRail(TileIndex tile, uint16 extra)

    83

{

    84

	byte b = _m[tile].m5;

    85

    86

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

    87

		return;

    88

    89

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

    90

}

    91

    92

static const uint16 _place_depot_extra[12] = {

    93

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

    94

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

    95

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

    96

};

    97

    98

    99

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

   100

{

   101

	if (success) {

   102

		int dir = p2;

   103

   104

		SndPlayTileFx(SND_20_SPLAT_2, tile);

   105

		ResetObjectToPlace();

   106

   107

		tile += TileOffsByDir(dir);

   108

   109

		if (IsTileType(tile, MP_RAILWAY)) {

   110

			PlaceExtraDepotRail(tile, _place_depot_extra[dir]);

   111

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

   112

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

   113

		}

   114

	}

   115

}

   116

   117

static void PlaceRail_Depot(TileIndex tile)

   118

{

   119

	DoCommandP(tile, _cur_railtype, _build_depot_direction, CcRailDepot,

   120

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

   121

}

   122

   123

static void PlaceRail_Waypoint(TileIndex tile)

   124

{

   125

	if (!_remove_button_clicked) {

   126

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

   127

	} else {

   128

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

   129

	}

   130

}

   131

   132

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

   133

{

   134

	if (success) {

   135

		SndPlayTileFx(SND_20_SPLAT_2, tile);

   136

		ResetObjectToPlace();

   137

	}

   138

}

   139

   140

static void PlaceRail_Station(TileIndex tile)

   141

{

   142

	if(_remove_button_clicked)

   143

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

   144

	else if (_railstation.dragdrop) {

   145

		VpStartPlaceSizing(tile, VPM_X_AND_Y_LIMITED);

   146

		VpSetPlaceSizingLimit(_patches.station_spread);

   147

	} else {

   148

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

   149

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

   150

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

   151

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

   152

	}

   153

}

   154

   155

static void GenericPlaceSignals(TileIndex tile)

   156

{

   157

	uint trackstat;

   158

	int i;

   159

   160

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

   161

   162

	if ((trackstat & 0x30)) // N-S direction

   163

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

   164

   165

	if ((trackstat & 0x0C)) // E-W direction

   166

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

   167

   168

	// Lookup the bit index

   169

	i = 0;

   170

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

   171

   172

	if (!_remove_button_clicked) {

   173

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

   174

			CMD_BUILD_SIGNALS | CMD_AUTO | CMD_MSG(STR_1010_CAN_T_BUILD_SIGNALS_HERE));

   175

	} else {

   176

		DoCommandP(tile, i, 0, CcPlaySound1E,

   177

			CMD_REMOVE_SIGNALS | CMD_AUTO | CMD_MSG(STR_1013_CAN_T_REMOVE_SIGNALS_FROM));

   178

	}

   179

}

   180

   181

static void PlaceRail_Bridge(TileIndex tile)

   182

{

   183

	VpStartPlaceSizing(tile, VPM_X_OR_Y);

   184

}

   185

   186

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

   187

{

   188

	if (success) {

   189

		SndPlayTileFx(SND_20_SPLAT_2, tile);

   190

		ResetObjectToPlace();

   191

	} else {

   192

		SetRedErrorSquare(_build_tunnel_endtile);

   193

	}

   194

}

   195

   196

static void PlaceRail_Tunnel(TileIndex tile)

   197

{

   198

	DoCommandP(tile, _cur_railtype, 0, CcBuildRailTunnel,

   199

		CMD_BUILD_TUNNEL | CMD_AUTO | CMD_MSG(STR_5016_CAN_T_BUILD_TUNNEL_HERE));

   200

}

   201

   202

void PlaceProc_BuyLand(TileIndex tile)

   203

{

   204

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

   205

}

   206

   207

static void PlaceRail_ConvertRail(TileIndex tile)

   208

{

   209

	VpStartPlaceSizing(tile, VPM_X_AND_Y | GUI_PlaceProc_ConvertRailArea);

   210

}

   211

   212

static void PlaceRail_AutoSignals(TileIndex tile)

   213

{

   214

	VpStartPlaceSizing(tile, VPM_SIGNALDIRS);

   215

}

   216

   217

static void BuildRailClick_N(Window *w)

   218

{

   219

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

   220

}

   221

   222

static void BuildRailClick_NE(Window *w)

   223

{

   224

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

   225

}

   226

   227

static void BuildRailClick_E(Window *w)

   228

{

   229

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

   230

}

   231

   232

static void BuildRailClick_NW(Window *w)

   233

{

   234

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

   235

}

   236

   237

static void BuildRailClick_AutoRail(Window *w)

   238

{

   239

	HandlePlacePushButton(w, 8, SPR_CURSOR_AUTORAIL + _cur_railtype, VHM_RAIL, PlaceRail_AutoRail);

   240

}

   241

   242

static void BuildRailClick_Demolish(Window *w)

   243

{

   244

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

   245

}

   246

   247

static const CursorID _depot_cursors[] = {

   248

	SPR_CURSOR_RAIL_DEPOT,

   249

	SPR_CURSOR_MONO_DEPOT,

   250

	SPR_CURSOR_MAGLEV_DEPOT,

   251

};

   252

   253

static void BuildRailClick_Depot(Window *w)

   254

{

   255

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

   256

}

   257

   258

static void BuildRailClick_Waypoint(Window *w)

   259

{

   260

	_waypoint_count = GetCustomStationsCount(STAT_CLASS_WAYP);

   261

	if (HandlePlacePushButton(w, 11, SPR_CURSOR_WAYPOINT, 1, PlaceRail_Waypoint)

   262

	    && _waypoint_count > 0)

   263

		ShowBuildWaypointPicker();

   264

}

   265

   266

static void BuildRailClick_Station(Window *w)

   267

{

   268

	if (HandlePlacePushButton(w, 12, SPR_CURSOR_RAIL_STATION, 1, PlaceRail_Station)) ShowStationBuilder();

   269

}

   270

   271

static void BuildRailClick_AutoSignals(Window *w)

   272

{

   273

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

   274

}

   275

   276

static void BuildRailClick_Bridge(Window *w)

   277

{

   278

	HandlePlacePushButton(w, 14, SPR_CURSOR_BRIDGE, 1, PlaceRail_Bridge);

   279

}

   280

   281

static void BuildRailClick_Tunnel(Window *w)

   282

{

   283

	HandlePlacePushButton(w, 15, SPR_CURSOR_TUNNEL_RAIL + _cur_railtype, 3, PlaceRail_Tunnel);

   284

}

   285

   286

static void BuildRailClick_Remove(Window *w)

   287

{

   288

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

   289

		return;

   290

	SetWindowDirty(w);

   291

	SndPlayFx(SND_15_BEEP);

   292

   293

	w->click_state ^= (1 << 16);

   294

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

   295

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

   296

   297

	// handle station builder

   298

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

   299

	{

   300

		if(_remove_button_clicked)

   301

			SetTileSelectSize(1, 1);

   302

		else

   303

			BringWindowToFrontById(WC_BUILD_STATION, 0);

   304

	}

   305

}

   306

   307

static void BuildRailClick_Convert(Window *w)

   308

{

   309

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

   310

}

   311

   312

static void BuildRailClick_Landscaping(Window *w)

   313

{

   314

	ShowTerraformToolbar();

   315

}

   316

   317

static void DoRailroadTrack(int mode)

   318

{

   319

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

   320

		_remove_button_clicked ?

   321

		CMD_REMOVE_RAILROAD_TRACK | CMD_AUTO | CMD_NO_WATER | CMD_MSG(STR_1012_CAN_T_REMOVE_RAILROAD_TRACK) :

   322

		CMD_BUILD_RAILROAD_TRACK  | CMD_AUTO | CMD_NO_WATER | CMD_MSG(STR_1011_CAN_T_BUILD_RAILROAD_TRACK)

   323

	);

   324

}

   325

   326

static void HandleAutodirPlacement(void)

   327

{

   328

	TileHighlightData *thd = &_thd;

   329

	int trackstat = thd->drawstyle & 0xF; // 0..5

   330

   331

	if (thd->drawstyle & HT_RAIL) { // one tile case

   332

		GenericPlaceRail(TileVirtXY(thd->selend.x, thd->selend.y), trackstat);

   333

		return;

   334

	}

   335

   336

	DoRailroadTrack(trackstat);

   337

}

   338

   339

static void HandleAutoSignalPlacement(void)

   340

{

   341

	TileHighlightData *thd = &_thd;

   342

	byte trackstat = thd->drawstyle & 0xF; // 0..5

   343

   344

	if (thd->drawstyle == HT_RECT) { // one tile case

   345

		GenericPlaceSignals(TileVirtXY(thd->selend.x, thd->selend.y));

   346

		return;

   347

	}

   348

   349

	// _patches.drag_signals_density is given as a parameter such that each user in a network

   350

	// game can specify his/her own signal density

   351

	DoCommandP(TileVirtXY(thd->selstart.x, thd->selstart.y), TileVirtXY(thd->selend.x, thd->selend.y),

   352

	(_ctrl_pressed ? 1 << 3 : 0) | (trackstat << 4) | (_patches.drag_signals_density << 24),

   353

	CcPlaySound1E,

   354

	(_remove_button_clicked ?	CMD_REMOVE_SIGNAL_TRACK | CMD_AUTO | CMD_NO_WATER | CMD_MSG(STR_1013_CAN_T_REMOVE_SIGNALS_FROM) :

   355

	                          CMD_BUILD_SIGNAL_TRACK  | CMD_AUTO | CMD_NO_WATER | CMD_MSG(STR_1010_CAN_T_BUILD_SIGNALS_HERE) ) );

   356

}

   357

   358

static OnButtonClick * const _build_railroad_button_proc[] = {

   359

	BuildRailClick_N,

   360

	BuildRailClick_NE,

   361

	BuildRailClick_E,

   362

	BuildRailClick_NW,

   363

	BuildRailClick_AutoRail,

   364

	BuildRailClick_Demolish,

   365

	BuildRailClick_Depot,

   366

	BuildRailClick_Waypoint,

   367

	BuildRailClick_Station,

   368

	BuildRailClick_AutoSignals,

   369

	BuildRailClick_Bridge,

   370

	BuildRailClick_Tunnel,

   371

	BuildRailClick_Remove,

   372

	BuildRailClick_Convert,

   373

	BuildRailClick_Landscaping,

   374

};

   375

   376

static const uint16 _rail_keycodes[] = {

   377

	'1',

   378

	'2',

   379

	'3',

   380

	'4',

   381

	'5',

   382

	'6',

   383

	'7', // depot

   384

	'8', // waypoint

   385

	'9', // station

   386

	'S', // signals

   387

	'B', // bridge

   388

	'T', // tunnel

   389

	'R', // remove

   390

	'C', // convert rail

   391

	'L', // landscaping

   392

};

   393

   394

   395

   396

   397

static void BuildRailToolbWndProc(Window *w, WindowEvent *e)

   398

{

   399

	switch(e->event) {

   400

	case WE_PAINT:

   401

		w->disabled_state &= ~(1 << 16);

   402

		if (!(w->click_state & ((1<<4)|(1<<5)|(1<<6)|(1<<7)|(1<<8)|(1<<11)|(1<<12)|(1<<13)))) {

   403

			w->disabled_state |= (1 << 16);

   404

			w->click_state &= ~(1<<16);

   405

		}

   406

		DrawWindowWidgets(w);

   407

		break;

   408

   409

	case WE_CLICK:

   410

		if (e->click.widget >= 4) {

   411

			_remove_button_clicked = false;

   412

			_build_railroad_button_proc[e->click.widget - 4](w);

   413

		}

   414

	break;

   415

   416

	case WE_KEYPRESS: {

   417

		int i;

   418

   419

		for(i=0; i!=lengthof(_rail_keycodes); i++) {

   420

			if (e->keypress.keycode == _rail_keycodes[i]) {

   421

				e->keypress.cont = false;

   422

				_remove_button_clicked = false;

   423

				_build_railroad_button_proc[i](w);

   424

				break;

   425

			}

   426

		}

   427

		MarkTileDirty(_thd.pos.x, _thd.pos.y); // redraw tile selection

   428

		break;

   429

	}

   430

   431

	case WE_PLACE_OBJ:

   432

		_place_proc(e->place.tile);

   433

		return;

   434

   435

	case WE_PLACE_DRAG: {

   436

		VpSelectTilesWithMethod(e->place.pt.x, e->place.pt.y, e->place.userdata & 0xF);

   437

		return;

   438

	}

   439

   440

	case WE_PLACE_MOUSEUP:

   441

		if (e->click.pt.x != -1) {

   442

			TileIndex start_tile = e->place.starttile;

   443

			TileIndex end_tile = e->place.tile;

   444

   445

			if (e->place.userdata == VPM_X_OR_Y) {

   446

				ResetObjectToPlace();

   447

				ShowBuildBridgeWindow(start_tile, end_tile, _cur_railtype);

   448

			} else if (e->place.userdata == VPM_RAILDIRS) {

   449

				bool old = _remove_button_clicked;

   450

				if (_ctrl_pressed) _remove_button_clicked = true;

   451

				HandleAutodirPlacement();

   452

				_remove_button_clicked = old;

   453

			} else if (e->place.userdata == VPM_SIGNALDIRS) {

   454

				HandleAutoSignalPlacement();

   455

			} else if ((e->place.userdata & 0xF) == VPM_X_AND_Y) {

   456

				if (GUIPlaceProcDragXY(e)) break;

   457

   458

				if ((e->place.userdata >> 4) == GUI_PlaceProc_ConvertRailArea >> 4)

   459

					DoCommandP(end_tile, start_tile, _cur_railtype, CcPlaySound10, CMD_CONVERT_RAIL | CMD_MSG(STR_CANT_CONVERT_RAIL));

   460

			} else if (e->place.userdata == VPM_X_AND_Y_LIMITED) {

   461

				HandleStationPlacement(start_tile, end_tile);

   462

			} else

   463

				DoRailroadTrack(e->place.userdata & 1);

   464

		}

   465

		break;

   466

   467

	case WE_ABORT_PLACE_OBJ:

   468

		UnclickWindowButtons(w);

   469

		SetWindowDirty(w);

   470

   471

		w = FindWindowById(WC_BUILD_STATION, 0);

   472

		if (w != NULL) WP(w,def_d).close=true;

   473

		w = FindWindowById(WC_BUILD_DEPOT, 0);

   474

		if (w != NULL) WP(w,def_d).close=true;

   475

		break;

   476

   477

	case WE_PLACE_PRESIZE: {

   478

		TileIndex tile = e->place.tile;

   479

   480

		DoCommandByTile(tile, 0, 0, DC_AUTO, CMD_BUILD_TUNNEL);

   481

		VpSetPresizeRange(tile, _build_tunnel_endtile==0?tile:_build_tunnel_endtile);

   482

	} break;

   483

	}

   484

}

   485

   486

   487

static const Widget _build_railroad_widgets[] = {

   488

{   WWT_CLOSEBOX,   RESIZE_NONE,     7,     0,    10,     0,    13, STR_00C5,STR_018B_CLOSE_WINDOW},

   489

{    WWT_CAPTION,   RESIZE_NONE,     7,    11,   359,     0,    13, STR_100A_RAILROAD_CONSTRUCTION, STR_018C_WINDOW_TITLE_DRAG_THIS},

   490

{  WWT_STICKYBOX,   RESIZE_NONE,     7,   360,   371,     0,    13, 0x0,     STR_STICKY_BUTTON},

   491

   492

{      WWT_PANEL,   RESIZE_NONE,     7,   110,   113,    14,    35, 0x0,			STR_NULL},

   493

   494

{      WWT_PANEL,   RESIZE_NONE,     7,    0,     21,    14,    35, 0x4E3,		STR_1018_BUILD_RAILROAD_TRACK},

   495

{      WWT_PANEL,   RESIZE_NONE,     7,    22,    43,    14,    35, 0x4E4,		STR_1018_BUILD_RAILROAD_TRACK},

   496

{      WWT_PANEL,   RESIZE_NONE,     7,    44,    65,    14,    35, 0x4E5,		STR_1018_BUILD_RAILROAD_TRACK},

   497

{      WWT_PANEL,   RESIZE_NONE,     7,    66,    87,    14,    35, 0x4E6,		STR_1018_BUILD_RAILROAD_TRACK},

   498

{      WWT_PANEL,   RESIZE_NONE,     7,    88,   109,    14,    35, SPR_OPENTTD_BASE + 0, STR_BUILD_AUTORAIL_TIP},

   499

   500

{      WWT_PANEL,   RESIZE_NONE,     7,   114,   135,    14,    35, 0x2BF,		STR_018D_DEMOLISH_BUILDINGS_ETC},

   501

{      WWT_PANEL,   RESIZE_NONE,     7,   136,   157,    14,    35, 0x50E,		STR_1019_BUILD_TRAIN_DEPOT_FOR_BUILDING},

   502

{      WWT_PANEL,   RESIZE_NONE,     7,   158,   179,    14,    35, SPR_OPENTTD_BASE + 3, STR_CONVERT_RAIL_TO_WAYPOINT_TIP},