1

/* $Id$ */

     2

     3

/** @file rail_gui.cpp File for dealing with rail construction user interface */

     4

     5

#include "stdafx.h"

     6

#include "openttd.h"

     7

#include "tile_cmd.h"

     8

#include "landscape.h"

     9

#include "gui.h"

    10

#include "window_gui.h"

    11

#include "station_gui.h"

    12

#include "terraform_gui.h"

    13

#include "viewport_func.h"

    14

#include "gfx_func.h"

    15

#include "command_func.h"

    16

#include "station.h"

    17

#include "waypoint.h"

    18

#include "debug.h"

    19

#include "variables.h"

    20

#include "newgrf_callbacks.h"

    21

#include "newgrf_station.h"

    22

#include "train.h"

    23

#include "strings_func.h"

    24

#include "functions.h"

    25

#include "window_func.h"

    26

#include "date_func.h"

    27

#include "sound_func.h"

    28

#include "player_func.h"

    29

#include "settings_type.h"

    30

#include "widgets/dropdown_type.h"

    31

#include "widgets/dropdown_func.h"

    32

    33

#include "bridge_map.h"

    34

#include "rail_map.h"

    35

#include "road_map.h"

    36

#include "station_map.h"

    37

#include "tunnel_map.h"

    38

#include "tunnelbridge_map.h"

    39

    40

#include "table/sprites.h"

    41

#include "table/strings.h"

    42

    43

static RailType _cur_railtype;

    44

static bool _remove_button_clicked;

    45

static DiagDirection _build_depot_direction;

    46

static byte _waypoint_count = 1;

    47

static byte _cur_waypoint_type;

    48

static bool _convert_signal_button; // convert signal button in the signal GUI pressed

    49

static SignalVariant _cur_signal_variant; // set the signal variant (for signal GUI)

    50

static SignalType _cur_signal_type; // set the signal type (for signal GUI)

    51

    52

static struct {

    53

	byte orientation;

    54

	byte numtracks;

    55

	byte platlength;

    56

	bool dragdrop;

    57

    58

	bool newstations;

    59

	StationClassIDByte station_class;

    60

	byte station_type;

    61

	byte station_count;

    62

} _railstation;

    63

    64

    65

static void HandleStationPlacement(TileIndex start, TileIndex end);

    66

static void ShowBuildTrainDepotPicker();

    67

static void ShowBuildWaypointPicker();

    68

static void ShowStationBuilder();

    69

static void ShowSignalBuilder();

    70

    71

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

    72

{

    73

	if (success) SndPlayTileFx(SND_20_SPLAT_2, tile);

    74

}

    75

    76

static void GenericPlaceRail(TileIndex tile, int cmd)

    77

{

    78

	DoCommandP(tile, _cur_railtype, cmd, CcPlaySound1E,

    80

		CMD_REMOVE_SINGLE_RAIL | CMD_MSG(STR_1012_CAN_T_REMOVE_RAILROAD_TRACK) | CMD_NO_WATER :

    81

		CMD_BUILD_SINGLE_RAIL | CMD_MSG(STR_1011_CAN_T_BUILD_RAILROAD_TRACK) | CMD_NO_WATER

    82

	);

    83

}

    84

    85

static void PlaceRail_N(TileIndex tile)

    86

{

    87

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

    88

	GenericPlaceRail(tile, cmd);

    89

}

    90

    91

static void PlaceRail_NE(TileIndex tile)

    92

{

    93

	VpStartPlaceSizing(tile, VPM_FIX_Y, DDSP_PLACE_RAIL_NE);

    94

}

    95

    96

static void PlaceRail_E(TileIndex tile)

    97

{

    98

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

    99

	GenericPlaceRail(tile, cmd);

   100

}

   101

   102

static void PlaceRail_NW(TileIndex tile)

   103

{

   104

	VpStartPlaceSizing(tile, VPM_FIX_X, DDSP_PLACE_RAIL_NW);

   105

}

   106

   107

static void PlaceRail_AutoRail(TileIndex tile)

   108

{

   109

	VpStartPlaceSizing(tile, VPM_RAILDIRS, DDSP_PLACE_AUTORAIL);

   110

}

   111

   112

static void PlaceExtraDepotRail(TileIndex tile, uint16 extra)

   113

{

   114

	if (GetRailTileType(tile) != RAIL_TILE_NORMAL) return;

   115

	if ((GetTrackBits(tile) & GB(extra, 8, 8)) == 0) return;

   116

   117

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

   118

}

   119

   120

static const uint16 _place_depot_extra[12] = {

   121

	0x0604, 0x2102, 0x1202, 0x0505,

   122

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

   123

	0x2203, 0x0904, 0x0A05, 0x1103,

   124

};

   125

   126

   127

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

   128

{

   129

	if (success) {

   130

		DiagDirection dir = (DiagDirection)p2;

   131

   132

		SndPlayTileFx(SND_20_SPLAT_2, tile);

   133

		ResetObjectToPlace();

   134

   135

		tile += TileOffsByDiagDir(dir);

   136

   137

		if (IsTileType(tile, MP_RAILWAY)) {

   138

			PlaceExtraDepotRail(tile, _place_depot_extra[dir]);

   139

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

   140

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

   141

		}

   142

	}

   143

}

   144

   145

static void PlaceRail_Depot(TileIndex tile)

   146

{

   147

	DoCommandP(tile, _cur_railtype, _build_depot_direction, CcRailDepot,

   148

		CMD_BUILD_TRAIN_DEPOT | CMD_NO_WATER | CMD_MSG(STR_100E_CAN_T_BUILD_TRAIN_DEPOT));

   149

}

   150

   151

static void PlaceRail_Waypoint(TileIndex tile)

   152

{

   153

	if (_remove_button_clicked || _ctrl_pressed) {

   154

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

   155

	} else {

   156

		DoCommandP(tile, _cur_waypoint_type, 0, CcPlaySound1E, CMD_BUILD_TRAIN_WAYPOINT | CMD_MSG(STR_CANT_BUILD_TRAIN_WAYPOINT));

   157

	}

   158

}

   159

   160

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

   161

{

   162

	if (success) {

   163

		SndPlayTileFx(SND_20_SPLAT_2, tile);

   164

		/* Only close the station builder window if the default station is chosen. */

   165

		if (_railstation.station_class == STAT_CLASS_DFLT && _railstation.station_type == 0) ResetObjectToPlace();

   166

	}

   167

}

   168

   169

static void PlaceRail_Station(TileIndex tile)

   170

{

   171

	if (_remove_button_clicked) {

   172

		VpStartPlaceSizing(tile, VPM_X_AND_Y, DDSP_REMOVE_STATION);

   173

	} else if (_railstation.dragdrop) {

   174

		VpStartPlaceSizing(tile, VPM_X_AND_Y_LIMITED, DDSP_BUILD_STATION);

   175

		VpSetPlaceSizingLimit(_patches.station_spread);

   176

	} else {

   177

		DoCommandP(tile,

   178

				_railstation.orientation | (_railstation.numtracks << 8) | (_railstation.platlength << 16) | (_ctrl_pressed << 24),

   179

				_cur_railtype | (_railstation.station_class << 8) | (_railstation.station_type << 16), CcStation,

   180

				CMD_BUILD_RAILROAD_STATION | CMD_NO_WATER | CMD_MSG(STR_100F_CAN_T_BUILD_RAILROAD_STATION));

   181

	}

   182

}

   183

   184

/**

   185

 * Build a new signal or edit/remove a present signal, use CmdBuildSingleSignal() or CmdRemoveSingleSignal() in rail_cmd.cpp

   186

 *

   187

 * @param tile The tile where the signal will build or edit

   188

 */

   189

static void GenericPlaceSignals(TileIndex tile)

   190

{

   191

	TrackBits trackbits = (TrackBits)GB(GetTileTrackStatus(tile, TRANSPORT_RAIL, 0), 0, 6);

   192

   193

	if (trackbits & TRACK_BIT_VERT) { // N-S direction

   194

		trackbits = (_tile_fract_coords.x <= _tile_fract_coords.y) ? TRACK_BIT_RIGHT : TRACK_BIT_LEFT;

   195

	}

   196

   197

	if (trackbits & TRACK_BIT_HORZ) { // E-W direction

   198

		trackbits = (_tile_fract_coords.x + _tile_fract_coords.y <= 15) ? TRACK_BIT_UPPER : TRACK_BIT_LOWER;

   199

	}

   200

   201

	Track track = FindFirstTrack(trackbits);

   202

   203

	if (_remove_button_clicked) {

   204

		DoCommandP(tile, track, 0, CcPlaySound1E,

   205

			CMD_REMOVE_SIGNALS | CMD_MSG(STR_1013_CAN_T_REMOVE_SIGNALS_FROM));

   206

	} else {

   207

		if (!_patches.enable_signal_gui) _cur_signal_variant = _cur_year < _patches.semaphore_build_before ? SIG_SEMAPHORE : SIG_ELECTRIC;

   208

   209

		/* various bitstuffed elements for CmdBuildSingleSignal() */

   210

		uint32 p1 = track;

   211

		SB(p1, 3, 1, _ctrl_pressed);

   212

		SB(p1, 4, 1, _cur_signal_variant);

   213

		SB(p1, 5, 2, _patches.enable_signal_gui ? _cur_signal_type : SIGTYPE_NORMAL);

   214

		SB(p1, 7, 1, _convert_signal_button);

   215

   216

		DoCommandP(tile, p1, 0, CcPlaySound1E, CMD_BUILD_SIGNALS |

   217

			CMD_MSG(_convert_signal_button ? STR_SIGNAL_CAN_T_CONVERT_SIGNALS_HERE : STR_1010_CAN_T_BUILD_SIGNALS_HERE));

   218

	}

   219

}

   220

   221

static void PlaceRail_Bridge(TileIndex tile)

   222

{

   223

	VpStartPlaceSizing(tile, VPM_X_OR_Y, DDSP_BUILD_BRIDGE);

   224

}

   225

   226

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

   227

{

   228

	if (success) {

   229

		SndPlayTileFx(SND_20_SPLAT_2, tile);

   230

		ResetObjectToPlace();

   231

	} else {

   232

		SetRedErrorSquare(_build_tunnel_endtile);

   233

	}

   234

}

   235

   236

static void PlaceRail_Tunnel(TileIndex tile)

   237

{

   238

	DoCommandP(tile, _cur_railtype, 0, CcBuildRailTunnel,

   239

		CMD_BUILD_TUNNEL | CMD_MSG(STR_5016_CAN_T_BUILD_TUNNEL_HERE));

   240

}

   241

   242

void PlaceProc_BuyLand(TileIndex tile)

   243

{

   244

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

   245

}

   246

   247

static void PlaceRail_ConvertRail(TileIndex tile)

   248

{

   249

	VpStartPlaceSizing(tile, VPM_X_AND_Y, DDSP_CONVERT_RAIL);

   250

}

   251

   252

static void PlaceRail_AutoSignals(TileIndex tile)

   253

{

   254

	VpStartPlaceSizing(tile, VPM_SIGNALDIRS, DDSP_BUILD_SIGNALS);

   255

}

   256

   257

   258

/** Enum referring to the widgets of the build rail toolbar */

   259

enum RailToolbarWidgets {

   260

	RTW_CLOSEBOX = 0,

   261

	RTW_CAPTION,

   262

	RTW_STICKY,

   263

	RTW_SPACER,

   264

	RTW_BUILD_NS,

   265

	RTW_BUILD_X,

   266

	RTW_BUILD_EW,

   267

	RTW_BUILD_Y,

   268

	RTW_AUTORAIL,

   269

	RTW_DEMOLISH,

   270

	RTW_BUILD_DEPOT,

   271

	RTW_BUILD_WAYPOINT,

   272

	RTW_BUILD_STATION,

   273

	RTW_BUILD_SIGNALS,

   274

	RTW_BUILD_BRIDGE,

   275

	RTW_BUILD_TUNNEL,

   276

	RTW_REMOVE,

   277

	RTW_CONVERT_RAIL,

   278

};

   279

   317

static void BuildRailClick_N(Window *w)

   318

{

   319

	HandlePlacePushButton(w, RTW_BUILD_NS, GetRailTypeInfo(_cur_railtype)->cursor.rail_ns, VHM_RECT, PlaceRail_N);

   320

}

   321

   322

static void BuildRailClick_NE(Window *w)

   323

{

   324

	HandlePlacePushButton(w, RTW_BUILD_X, GetRailTypeInfo(_cur_railtype)->cursor.rail_swne, VHM_RECT, PlaceRail_NE);

   325

}

   326

   327

static void BuildRailClick_E(Window *w)

   328

{

   329

	HandlePlacePushButton(w, RTW_BUILD_EW, GetRailTypeInfo(_cur_railtype)->cursor.rail_ew, VHM_RECT, PlaceRail_E);

   330

}

   331

   332

static void BuildRailClick_NW(Window *w)

   333

{

   334

	HandlePlacePushButton(w, RTW_BUILD_Y, GetRailTypeInfo(_cur_railtype)->cursor.rail_nwse, VHM_RECT, PlaceRail_NW);

   335

}

   336

   337

static void BuildRailClick_AutoRail(Window *w)

   338

{

   339

	HandlePlacePushButton(w, RTW_AUTORAIL, GetRailTypeInfo(_cur_railtype)->cursor.autorail, VHM_RAIL, PlaceRail_AutoRail);

   340

}

   341

   342

static void BuildRailClick_Demolish(Window *w)

   343

{

   344

	HandlePlacePushButton(w, RTW_DEMOLISH, ANIMCURSOR_DEMOLISH, VHM_RECT, PlaceProc_DemolishArea);

   345

}

   346

   347

static void BuildRailClick_Depot(Window *w)

   348

{

   349

	if (HandlePlacePushButton(w, RTW_BUILD_DEPOT, GetRailTypeInfo(_cur_railtype)->cursor.depot, VHM_RECT, PlaceRail_Depot)) {

   350

		ShowBuildTrainDepotPicker();

   351

	}

   352

}

   353

   354

static void BuildRailClick_Waypoint(Window *w)

   355

{

   356

	_waypoint_count = GetNumCustomStations(STAT_CLASS_WAYP);

   357

	if (HandlePlacePushButton(w, RTW_BUILD_WAYPOINT, SPR_CURSOR_WAYPOINT, VHM_RECT, PlaceRail_Waypoint) &&

   358

			_waypoint_count > 1) {

   359

		ShowBuildWaypointPicker();

   360

	}

   361

}

   362

   363

static void BuildRailClick_Station(Window *w)

   364

{

   365

	if (HandlePlacePushButton(w, RTW_BUILD_STATION, SPR_CURSOR_RAIL_STATION, VHM_RECT, PlaceRail_Station)) ShowStationBuilder();

   366

}

   367

   368

/** The "build signal"-button proc from BuildRailToolbWndProc() (start ShowSignalBuilder() and/or HandleAutoSignalPlacement()) */

   369

static void BuildRailClick_AutoSignals(Window *w)

   370

{

   371

	if (_patches.enable_signal_gui) {

   372

		if (HandlePlacePushButton(w, RTW_BUILD_SIGNALS, ANIMCURSOR_BUILDSIGNALS, VHM_RECT, PlaceRail_AutoSignals)) ShowSignalBuilder();

   373

	} else {

   374

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

   375

	}

   376

}

   377

   378

static void BuildRailClick_Bridge(Window *w)

   379

{

   380

	HandlePlacePushButton(w, RTW_BUILD_BRIDGE, SPR_CURSOR_BRIDGE, VHM_RECT, PlaceRail_Bridge);

   381

}

   382

   383

static void BuildRailClick_Tunnel(Window *w)

   384

{

   385

	HandlePlacePushButton(w, RTW_BUILD_TUNNEL, GetRailTypeInfo(_cur_railtype)->cursor.tunnel, VHM_SPECIAL, PlaceRail_Tunnel);

   386

}

   387

   388

static void BuildRailClick_Remove(Window *w)

   389

{

   390

	if (w->IsWidgetDisabled(RTW_REMOVE)) return;

   392

	SndPlayFx(SND_15_BEEP);

   393

}

   394

   395

static void BuildRailClick_Convert(Window *w)

   396

{

   397

	HandlePlacePushButton(w, RTW_CONVERT_RAIL, GetRailTypeInfo(_cur_railtype)->cursor.convert, VHM_RECT, PlaceRail_ConvertRail);

   398

}

   399

   400

   401

static void DoRailroadTrack(int mode)

   402

{

   403

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

   405

		CMD_REMOVE_RAILROAD_TRACK | CMD_NO_WATER | CMD_MSG(STR_1012_CAN_T_REMOVE_RAILROAD_TRACK) :

   406

		CMD_BUILD_RAILROAD_TRACK  | CMD_NO_WATER | CMD_MSG(STR_1011_CAN_T_BUILD_RAILROAD_TRACK)

   407

	);

   408

}

   409

   410

static void HandleAutodirPlacement()

   411

{

   412

	TileHighlightData *thd = &_thd;

   413

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

   414

   415

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

   416

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

   417

		return;

   418

	}

   419

   420

	DoRailroadTrack(trackstat);

   421

}

   422

   423

/**

   424

 * Build new signals or remove signals or (if only one tile marked) edit a signal.

   425

 *

   426

 * If one tile marked abort and use GenericPlaceSignals()

   427

 * else use CmdBuildSingleSignal() or CmdRemoveSingleSignal() in rail_cmd.cpp to build many signals

   428

 */

   429

static void HandleAutoSignalPlacement()

   430

{

   431

	TileHighlightData *thd = &_thd;

   432

	uint32 p2 = GB(thd->drawstyle, 0, 3); // 0..5

   433

   434

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

   435

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

   436

		return;

   437

	}

   438

   439

	/* XXX Steal ctrl for autosignal function, until we get some GUI */

   440

	SB(p2,  3, 1, 0);

   441

	SB(p2,  4, 1, _cur_year < _patches.semaphore_build_before);

   442

	SB(p2,  6, 1, _ctrl_pressed);

   443

	SB(p2, 24, 8, _patches.drag_signals_density);

   444

   445

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

   446

	 * in a network game can specify his/her own signal density */

   447

	DoCommandP(

   448

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

   449

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

   450

		p2,

   451

		CcPlaySound1E,

   452

		_remove_button_clicked ?

   453

			CMD_REMOVE_SIGNAL_TRACK | CMD_NO_WATER | CMD_MSG(STR_1013_CAN_T_REMOVE_SIGNALS_FROM) :

   454

			CMD_BUILD_SIGNAL_TRACK  | CMD_NO_WATER | CMD_MSG(STR_1010_CAN_T_BUILD_SIGNALS_HERE)

   455

	);

   456

}

   457

   458

   459

typedef void OnButtonClick(Window *w);

   460

   461

static OnButtonClick * const _build_railroad_button_proc[] = {

   462

	BuildRailClick_N,

   463

	BuildRailClick_NE,

   464

	BuildRailClick_E,

   465

	BuildRailClick_NW,

   466

	BuildRailClick_AutoRail,

   467

	BuildRailClick_Demolish,

   468

	BuildRailClick_Depot,

   469

	BuildRailClick_Waypoint,

   470

	BuildRailClick_Station,

   471

	BuildRailClick_AutoSignals,

   472

	BuildRailClick_Bridge,

   473

	BuildRailClick_Tunnel,

   474

	BuildRailClick_Remove,

   475

	BuildRailClick_Convert

   476

};

   477

   478

static const uint16 _rail_keycodes[] = {

   479

	'1',

   480

	'2',

   481

	'3',

   482

	'4',

   483

	'5',

   484

	'6',

   485

	'7', // depot

   486

	'8', // waypoint

   487

	'9', // station

   488

	'S', // signals

   489

	'B', // bridge

   490

	'T', // tunnel

   491

	'R', // remove