1

#include "stdafx.h"

     2

#include "ttd.h"

     3

     4

#include "window.h"

     5

#include "gui.h"

     6

#include "gfx.h"

     7

#include "vehicle.h"

     8

#include "viewport.h"

     9

#include "station.h"

    10

#include "command.h"

    11

#include "player.h"

    12

#include "engine.h"

    13

    14

    15

int _traininfo_vehicle_pitch = 0;

    16

    17

static Engine * const _rail_engines[3] = {

    18

	&_engines[0],

    19

	&_engines[NUM_NORMAL_RAIL_ENGINES],

    20

	&_engines[NUM_MONORAIL_ENGINES + NUM_NORMAL_RAIL_ENGINES],

    21

};

    22

    23

const byte _rail_engines_count[3] = {

    24

	NUM_NORMAL_RAIL_ENGINES,

    25

	NUM_MONORAIL_ENGINES,

    26

	NUM_MAGLEV_ENGINES,

    27

};

    28

    29

const byte _rail_engines_start[3] = {

    30

	0,

    31

	NUM_NORMAL_RAIL_ENGINES,

    32

	NUM_MONORAIL_ENGINES + NUM_NORMAL_RAIL_ENGINES,

    33

};

    34

    35

static void CcBuildWagon(bool success, uint tile, uint32 p1, uint32 p2)

    36

{

    37

	Vehicle *v,*found;

    38

    39

	if (!success)

    40

		return;

    41

    42

	// find a locomotive in the depot.

    43

	found = NULL;

    44

	FOR_ALL_VEHICLES(v) {

    45

		if (v->type == VEH_Train && v->subtype==0 &&

    46

				v->tile == tile &&

    47

				v->u.rail.track == 0x80) {

    48

			if (found != NULL) // must be exactly one.

    49

				return;

    50

			found = v;

    51

		}

    52

	}

    53

    54

	// if we found a loco,

    55

	if (found != NULL) {

    56

		found = GetLastVehicleInChain(found);

    57

		// put the new wagon at the end of the loco.

    58

		DoCommandP(0, _new_wagon_id | (found->index<<16), 0, NULL, CMD_MOVE_RAIL_VEHICLE);

    59

		_vehicle_sort_dirty[VEHTRAIN] = true;

    60

	}

    61

}

    62

    63

static void CcBuildLoco(bool success, uint tile, uint32 p1, uint32 p2)

    64

{

    65

	Vehicle *v;

    66

    67

	if (!success)

    68

		return;

    69

    70

	v = &_vehicles[_new_train_id];

    71

	if (tile == _backup_orders_tile) {

    72

		_backup_orders_tile = 0;

    73

		RestoreVehicleOrders(v, _backup_orders_data);

    74

	}

    75

	ShowTrainViewWindow(v);

    76

}

    77

    78

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

    79

{

    80

	switch(e->event) {

    81

	case WE_PAINT:

    82

    83

		if (w->window_number == 0)

    84

			SETBIT(w->disabled_state, 5);

    85

    86

		{

    87

			int count = 0;

    88

			int num = _rail_engines_count[WP(w,buildtrain_d).railtype];

    89

			Engine *e = _rail_engines[WP(w,buildtrain_d).railtype];

    90

			do {

    91

				if (HASBIT(e->player_avail, _local_player))

    92

					count++;

    93

			} while (++e,--num);

    94

			SetVScrollCount(w, count);

    95

		}

    96

    97

		SET_DPARAM16(0, WP(w,buildtrain_d).railtype + STR_881C_NEW_RAIL_VEHICLES);

    98

		DrawWindowWidgets(w);

    99

   100

		{

   101

			int num = _rail_engines_count[WP(w,buildtrain_d).railtype];

   102

			Engine *e = _rail_engines[WP(w,buildtrain_d).railtype];

   103

			int sel = WP(w,buildtrain_d).sel_index;

   104

			int pos = w->vscroll.pos;

   105

			int x = 1;

   106

			int y = 15;

   107

			int engine_id = _rail_engines_start[WP(w,buildtrain_d).railtype];

   108

			int selected_id = -1;

   109

   110

			do {

   111

				if (HASBIT(e->player_avail, _local_player)) {

   112

					if (sel==0) selected_id = engine_id;

   113

					if (IS_INT_INSIDE(--pos, -8, 0)) {

   114

						DrawString(x+59, y+2, GetCustomEngineName(engine_id), sel==0 ? 0xC : 0x10);

   115

						DrawTrainEngine(x+29, y+6+_traininfo_vehicle_pitch, engine_id, SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));

   116

						y += 14;

   117

					}

   118

					sel--;

   119

				}

   120

			} while (++engine_id, ++e,--num);

   121

   122

			WP(w,buildtrain_d).sel_engine = selected_id;

   123

   124

			if (selected_id != -1) {

   125

				const RailVehicleInfo *rvi = &_rail_vehicle_info[selected_id];

   126

				Engine *e;

   127

				YearMonthDay ymd;

   128

   129

				if (!(rvi->flags & RVI_WAGON)) {

   130

					/* it's an engine */

   131

					int multihead = (rvi->flags&RVI_MULTIHEAD?1:0);

   132

   133

					SET_DPARAM32(0, rvi->base_cost * (_price.build_railvehicle >> 3) >> 5);

   134

					SET_DPARAM16(2, rvi->max_speed * 10 >> 4);

   135

					SET_DPARAM16(3, rvi->power << multihead);

   136

					SET_DPARAM16(1, rvi->weight << multihead);

   137

					SET_DPARAM32(4, (rvi->running_cost_base * _price.running_rail[rvi->engclass] >> 8) << multihead);

   138

   139

					SET_DPARAM16(5, STR_8838_N_A);

   140

					if (rvi->capacity != 0) {

   141

						SET_DPARAM16(6, rvi->capacity << multihead);

   142

						SET_DPARAM16(5, _cargoc.names_long_p[rvi->cargo_type]);

   143

					}

   144

   145

					e = &_engines[selected_id];

   146

   147

					SET_DPARAM16(8, e->lifelength);

   148

					SET_DPARAM8(9, e->reliability * 100 >> 16);

   149

					ConvertDayToYMD(&ymd, e->intro_date);

   150

					SET_DPARAM16(7, ymd.year + 1920);

   151

   152

					DrawString(2, 0x7F, STR_8817_COST_WEIGHT_T_SPEED_POWER, 0);

   153

				} else {

   154

					/* it's a wagon */

   155

   156

					SET_DPARAM32(0,

   157

						DoCommandByTile(w->window_number, selected_id, 0, DC_QUERY_COST, CMD_BUILD_RAIL_VEHICLE)

   158

					);

   159

					SET_DPARAM16(4, rvi->capacity);

   160

					SET_DPARAM16(1, rvi->weight);

   161

					SET_DPARAM16(3, _cargoc.names_long_p[rvi->cargo_type]);

   162

					SET_DPARAM16(2, (_cargoc.weights[rvi->cargo_type] * rvi->capacity >> 4) + rvi->weight);

   163

					DrawString(2, 0x7F, STR_8821_COST_WEIGHT_T_T_CAPACITY, 0);

   164

				}

   165

			}

   166

		}

   167

	break;

   168

   169

	case WE_CLICK: {

   170

		switch(e->click.widget) {

   171

		case 2: {

   172

			uint i = (e->click.pt.y - 14) / 14;

   173

			if (i < 8) {

   174

				WP(w,buildtrain_d).sel_index = i + w->vscroll.pos;

   175

				SetWindowDirty(w);

   176

			}

   177

		} break;

   178

		case 5: {

   179

			int sel_eng;

   180

			sel_eng = WP(w,buildtrain_d).sel_engine;

   181

			if (sel_eng != -1)

   182

				DoCommandP(w->window_number, sel_eng, 0, (_rail_vehicle_info[sel_eng].flags & RVI_WAGON) ? CcBuildWagon : CcBuildLoco, CMD_BUILD_RAIL_VEHICLE | CMD_MSG(STR_882B_CAN_T_BUILD_RAILROAD_VEHICLE));

   183

		}	break;

   184

		case 6:

   185

			WP(w,buildtrain_d).rename_engine = WP(w,buildtrain_d).sel_engine;

   186

			ShowQueryString(

   187

				GetCustomEngineName(WP(w,buildtrain_d).sel_engine),

   188

				STR_886A_RENAME_TRAIN_VEHICLE_TYPE,

   189

				31,

   190

				160,

   191

				w->window_class,

   192

				w->window_number);

   193

			break;

   194

		}

   195

	} break;

   196

   197

	case WE_4:

   198

		if (w->window_number != 0 && !FindWindowById(WC_VEHICLE_DEPOT, w->window_number)) {

   199

			DeleteWindow(w);

   200

		}

   201

		break;

   202

   203

	case WE_ON_EDIT_TEXT: {

   204

		byte *b = e->edittext.str;

   205

		if (*b == 0)

   206

			return;

   207

   208

		memcpy(_decode_parameters, b, 32);

   209

		DoCommandP(0, WP(w,buildtrain_d).rename_engine, 0, NULL, CMD_RENAME_ENGINE | CMD_MSG(STR_886B_CAN_T_RENAME_TRAIN_VEHICLE));

   210

	} break;

   211

	}

   212

}

   213

   214

static const Widget _new_rail_vehicle_widgets[] = {

   215

{   WWT_CLOSEBOX,    14,     0,    10,     0,    13, STR_00C5,	STR_018B_CLOSE_WINDOW},

   216

{    WWT_CAPTION,    14,    11,   227,     0,    13, STR_0315,	STR_018C_WINDOW_TITLE_DRAG_THIS},

   217

{     WWT_MATRIX,    14,     0,   216,    14,   125, 0x801,			STR_8843_TRAIN_VEHICLE_SELECTION},

   218

{  WWT_SCROLLBAR,    14,   217,   227,    14,   125, 0x0,				STR_0190_SCROLL_BAR_SCROLLS_LIST},

   219

{      WWT_PANEL,    14,     0,   227,   126,   187, 0x0,				STR_NULL},

   220

{ WWT_PUSHTXTBTN,    14,     0,   113,   188,   199, STR_881F_BUILD_VEHICLE,	STR_8844_BUILD_THE_HIGHLIGHTED_TRAIN},

   221

{ WWT_PUSHTXTBTN,    14,   114,   227,   188,   199, STR_8820_RENAME,					STR_8845_RENAME_TRAIN_VEHICLE_TYPE},

   222

{   WIDGETS_END},

   223

};

   224

   225

static const WindowDesc _new_rail_vehicle_desc = {

   226

	-1, -1, 228, 200,

   227

	WC_BUILD_VEHICLE,0,

   228

	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS,

   229

	_new_rail_vehicle_widgets,

   230

	NewRailVehicleWndProc

   231

};

   232

   233

static void ShowBuildTrainWindow(uint tile)

   234

{

   235

	Window *w;

   236

   237

	DeleteWindowById(WC_BUILD_VEHICLE, tile);

   238

   239

	w = AllocateWindowDesc(&_new_rail_vehicle_desc);

   240

	w->window_number = tile;

   241

	w->vscroll.cap = 8;

   242

   243

	if (tile != 0) {

   244

		w->caption_color = _map_owner[tile];

   245

		WP(w,buildtrain_d).railtype = _map3_lo[tile] & 0xF;

   246

	} else {

   247

		w->caption_color = _local_player;

   248

		WP(w,buildtrain_d).railtype = DEREF_PLAYER(_local_player)->max_railtype - 1;

   249

	}

   250

}

   251

   252

static void DrawTrainImage(Vehicle *v, int x, int y, int count, int skip, VehicleID selection)

   253

{

   254

	do {

   255

		if (--skip < 0) {

   256

			int image = GetTrainImage(v, 6);

   257

			uint32 ormod = SPRITE_PALETTE(PLAYER_SPRITE_COLOR(v->owner));

   258

			if (v->vehstatus & VS_CRASHED) ormod = 0x3248000;

   259

			DrawSprite(image | ormod, x+14, y+6+_traininfo_vehicle_pitch);

   260

			if (v->index == selection) DrawFrameRect(x-1, y-1, x+28, y+12, 15, 0x10);

   261

			x += 29;

   262

			count--;

   263

		}

   264

   265

		if (!(v = v->next))

   266

			break;

   267

	} while (count);

   268

}

   269

   270

static void DrawTrainDepotWindow(Window *w)

   271

{

   272

	uint tile;

   273

	Vehicle *v, *u;

   274

	int num,x,y,i, hnum;

   275

	Depot *d;

   276

   277

	tile = w->window_number;

   278

   279

	/* setup disabled buttons */

   280

	w->disabled_state = (_map_owner[tile]==_local_player) ? 0 : ((1<<3)|(1<<4)|(1<<6));

   281

   282

	/* determine amount of items for scroller */

   283

	num = 1;

   284

	hnum = 0;

   285

	FOR_ALL_VEHICLES(v) {

   286

		if (v->type == VEH_Train &&

   287

				  (v->subtype == 0 || v->subtype == 4) &&

   288

				v->tile == (TileIndex)tile &&

   289

				v->u.rail.track == 0x80) {

   290

					num++;

   291

					// determine number of items in the X direction.

   292

					if (v->subtype == 0) {

   293

						i = -1;

   294

						u = v;

   295

						do i++; while ( (u=u->next) != NULL);

   296

						if (i > hnum) hnum = i;

   297

					}

   298

		}

   299

	}

   300

	SetVScrollCount(w, num);

   301

	SetHScrollCount(w, hnum);

   302

   303

	/* locate the depot struct */

   304

	for(d=_depots; d->xy != (TileIndex)tile; d++) {}

   305

   306

	SET_DPARAM16(0,d->town_index);

   307

	DrawWindowWidgets(w);

   308

   309

	x = 2;

   310

	y = 15;

   311

	num = w->vscroll.pos;

   312

   313

	// draw all trains

   314

	FOR_ALL_VEHICLES(v) {

   315

		if (v->type == VEH_Train &&

   316

				v->subtype == 0 &&

   317

				v->tile == (TileIndex)tile &&

   318

				v->u.rail.track == 0x80 &&

   319

				--num < 0 && num >= -6) {

   320

   321

			DrawTrainImage(v, x+21, y, 10, w->hscroll.pos, WP(w,traindepot_d).sel);

   322

			/* Draw the train number */

   323

			SET_DPARAM16(0, v->unitnumber);

   324

			DrawString(x, y, (v->max_age - 366 < v->age) ? STR_00E3 : STR_00E2, 0);

   325

			/* Draw the pretty flag */

   326

			DrawSprite(v->vehstatus&VS_STOPPED ? 0xC12 : 0xC13, x+15, y);

   327

   328

			y += 14;

   329

		}

   330

	}

   331

   332

	// draw all remaining vehicles

   333

	FOR_ALL_VEHICLES(v) {

   334

		if (v->type == VEH_Train &&

   335

				v->subtype == 4 &&

   336

				v->tile == (TileIndex)tile &&

   337

				v->u.rail.track == 0x80 &&

   338

				--num < 0 && num >= -6) {

   339

   340

			DrawTrainImage(v, x+50, y, 9, 0, WP(w,traindepot_d).sel);

   341

			DrawString(x, y+2, STR_8816, 0);

   342

			y += 14;

   343

		}

   344

	}

   345

}

   346

   347

typedef struct GetDepotVehiclePtData {

   348

	Vehicle *head;

   349

	Vehicle *wagon;

   350

} GetDepotVehiclePtData;

   351

   352

static int GetVehicleFromTrainDepotWndPt(Window *w, int x, int y, GetDepotVehiclePtData *d)

   353

{

   354

	int area_x;

   355

	int row;

   356

	Vehicle *v;

   357

   358

	x = x - 23;

   359

	if (x < 0) {

   360

		area_x = (x >= -10) ? -2 : -1;

   361

	} else {

   362

		area_x = x / 29;

   363

	}

   364

   365

	row = (y - 14) / 14;

   366

	if ( (uint) row >= 6)

   367

		return 1; /* means err */

   368

   369

	row += w->vscroll.pos;

   370

   371

	/* go through all the locomotives */

   372

	FOR_ALL_VEHICLES(v) {

   373

		if (v->type == VEH_Train &&

   374

				v->subtype == 0 &&

   375

				v->tile == w->window_number &&

   376

				v->u.rail.track == 0x80 &&

   377

				--row < 0) {

   378

					area_x += w->hscroll.pos;

   379

					goto found_it;

   380

		}

   381

	}

   382

   383

	area_x--; /* free wagons don't have an initial loco. */

   384

   385

	/* and then the list of free wagons */

   386

	FOR_ALL_VEHICLES(v) {

   387

		if (v->type == VEH_Train &&

   388

				v->subtype == 4 &&

   389

				v->tile == w->window_number &&

   390

				v->u.rail.track == 0x80 &&

   391

				--row < 0)

   392

					goto found_it;

   393

	}

   394

   395

	d->head = NULL;

   396

	d->wagon = NULL;

   397

   398

	/* didn't find anything, get out */

   399

	return 0;

   400

   401

found_it:

   402

	d->head = d->wagon = v;

   403

   404

	/* either pressed the flag or the number, but only when it's a loco */

   405

	if (area_x < 0 && v->subtype==0)

   406

		return area_x;

   407

   408

	/* find the vehicle in this row that was clicked */

   409

	while (--area_x >= 0) {

   410

		v = v->next;

   411

		if (v == NULL) break;

   412

	}

   413

   414

	d->wagon = v;

   415

   416

	return 0;

   417

}

   418

   419

static void TrainDepotMoveVehicle(Vehicle *wagon, int sel, Vehicle *head)

   420

{

   421

	Vehicle *v;

   422

   423

	v = &_vehicles[sel];

   424

   425

	if (/*v->subtype == 0 ||*/ v == wagon)

   426

		return;

   427

   428

	if (wagon == NULL) {

   429

		if (head != NULL)

   430

			wagon = GetLastVehicleInChain(head);

   431

	} else  {

   432

		wagon = GetPrevVehicleInChain(wagon);

   433

		if (wagon == NULL)

   434

			return;

   435

	}

   436

   437

	if (wagon == v)

   438

		return;

   439

   440

	DoCommandP(v->tile, v->index + ((wagon==NULL ? (uint32)-1 : wagon->index) << 16), _ctrl_pressed ? 1 : 0, NULL, CMD_MOVE_RAIL_VEHICLE | CMD_MSG(STR_8837_CAN_T_MOVE_VEHICLE));

   441

}

   442

   443

static void TrainDepotClickTrain(Window *w, int x, int y)

   444

{

   445

	GetDepotVehiclePtData gdvp;

   446

	int mode, sel;

   447

	Vehicle *v;

   448

   449

	mode = GetVehicleFromTrainDepotWndPt(w, x, y, &gdvp);

   450

   451

	// share / copy orders

   452

	if (_thd.place_mode && mode <= 0) { _place_clicked_vehicle = gdvp.head; return; }

   453

   454

	v = gdvp.wagon;

   455

   456

	switch(mode) {

   457

	case 0: // start dragging of vehicle

   458

		sel = (int16)WP(w,traindepot_d).sel;

   459

		if (sel != -1) {

   460

			WP(w,traindepot_d).sel = INVALID_VEHICLE;

   461

			TrainDepotMoveVehicle(v, sel, gdvp.head);

   462

		} else if (v != NULL) {

   463

			WP(w,traindepot_d).sel = v->index;

   464

			SetObjectToPlaceWnd( SPRITE_PALETTE(PLAYER_SPRITE_COLOR(v->owner)) + GetTrainImage(v, 6), 4, w);

   465

			SetWindowDirty(w);

   466

		}

   467

		break;

   468

   469

	case -1: // show info window

   470

		ShowTrainViewWindow(v);

   471

		break;

   472

   473

	case -2: // click start/stop flag

   474

		DoCommandP(v->tile, v->index, 0, NULL, CMD_START_STOP_TRAIN | CMD_MSG(STR_883B_CAN_T_STOP_START_TRAIN));

   475

		break;

   476

	}

   477

}

   478

   479

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

   480

{

   481

	switch(e->event) {

   482

	case WE_PAINT:

   483

		DrawTrainDepotWindow(w);

   484

		break;

   485

   486

	case WE_CLICK: {

   487

		switch(e->click.widget) {

   488

		case 6:

   489

			ShowBuildTrainWindow(w->window_number);

   490

			break;

   491

		case 7:

   492

			ScrollMainWindowToTile(w->window_number);

   493

			break;

   494

		case 2:

   495

			TrainDepotClickTrain(w, e->click.pt.x, e->click.pt.y);

   496

			break;

   497

		}

   498

	} break;

   499

   500

	case WE_DESTROY:

   501

		DeleteWindowById(WC_BUILD_VEHICLE, w->window_number);

   502

		break;

   503

   504

	case WE_DRAGDROP: {

   505

		switch(e->click.widget) {

   506

		case 4:

   507

		case 3: {

   508

			Vehicle *v;

   509

			int sell_cmd;

   510

   511

			/* sell vehicle */

   512

			if (w->disabled_state & (1 << e->click.widget))

   513

				return;

   514

   515

			if (WP(w,traindepot_d).sel == INVALID_VEHICLE)

   516

				return;

   517

   518

			v = &_vehicles[WP(w,traindepot_d).sel];

   519

   520

			WP(w,traindepot_d).sel = INVALID_VEHICLE;

   521

			SetWindowDirty(w);

   522

   523

			HandleButtonClick(w, e->click.widget);