1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "debug.h"

     6

#include "functions.h"

     7

#include "rail_map.h"

     8

#include "table/sprites.h"

     9

#include "table/strings.h"

    10

#include "map.h"

    11

#include "window.h"

    12

#include "gui.h"

    13

#include "gfx.h"

    14

#include "vehicle.h"

    15

#include "viewport.h"

    16

#include "station.h"

    17

#include "command.h"

    18

#include "player.h"

    19

#include "engine.h"

    20

#include "vehicle_gui.h"

    21

#include "depot.h"

    22

#include "train.h"

    23

#include "newgrf_engine.h"

    24

#include "date.h"

    25

    26

/**

    27

 * Draw the purchase info details of train engine at a given location.

    28

 * @param x,y location where to draw the info

    29

 * @param engine_number the engine of which to draw the info of

    30

 */

    31

void DrawTrainEnginePurchaseInfo(int x, int y, EngineID engine_number)

    32

{

    33

	const RailVehicleInfo *rvi = RailVehInfo(engine_number);

    34

	const Engine *e = GetEngine(engine_number);

    35

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

    36

	YearMonthDay ymd;

    37

	ConvertDateToYMD(e->intro_date, &ymd);

    38

    39

	/* Purchase Cost - Engine weight */

    40

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

    41

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

    42

	DrawString(x,y, STR_PURCHASE_INFO_COST_WEIGHT, 0);

    43

	y += 10;

    44

    45

	/* Max speed - Engine power */

    46

	SetDParam(0, rvi->max_speed);

    47

	SetDParam(1, rvi->power << multihead);

    48

	DrawString(x,y, STR_PURCHASE_INFO_SPEED_POWER, 0);

    49

	y += 10;

    50

    51

	/* Running cost */

    52

	SetDParam(0, (rvi->running_cost_base * _price.running_rail[rvi->running_cost_class] >> 8) << multihead);

    53

	DrawString(x,y, STR_PURCHASE_INFO_RUNNINGCOST, 0);

    54

	y += 10;

    55

    56

	/* Powered wagons power - Powered wagons extra weight */

    57

	if (rvi->pow_wag_power != 0) {

    58

		SetDParam(0, rvi->pow_wag_power);

    59

		SetDParam(1, rvi->pow_wag_weight);

    60

		DrawString(x,y, STR_PURCHASE_INFO_PWAGPOWER_PWAGWEIGHT, 0);

    61

		y += 10;

    62

	};

    63

    64

	/* Cargo type + capacity, or N/A */

    65

	SetDParam(0, STR_8838_N_A);

    66

	SetDParam(2, STR_EMPTY);

    67

	if (rvi->capacity != 0) {

    68

		SetDParam(0, _cargoc.names_long[rvi->cargo_type]);

    69

		SetDParam(1, rvi->capacity << multihead);

    70

		SetDParam(2, STR_9842_REFITTABLE);

    71

	}

    72

	DrawString(x,y, STR_PURCHASE_INFO_CAPACITY, 0);

    73

	y += 10;

    74

    75

	/* Design date - Life length */

    76

	SetDParam(0, ymd.year);

    77

	SetDParam(1, e->lifelength);

    78

	DrawString(x,y, STR_PURCHASE_INFO_DESIGNED_LIFE, 0);

    79

	y += 10;

    80

    81

	/* Reliability */

    82

	SetDParam(0, e->reliability * 100 >> 16);

    83

	DrawString(x,y, STR_PURCHASE_INFO_RELIABILITY, 0);

    84

	y += 10;

    85

    86

	/* Additional text from NewGRF */

    87

	// XXX 227 will become a calculated width...

    88

	y += ShowAdditionalText(x, y, 227, engine_number);

    89

}

    90

    91

/**

    92

 * Draw the purchase info details of a train wagon at a given location.

    93

 * @param x,y location where to draw the info

    94

 * @param engine_number the engine of which to draw the info of

    95

 */

    96

void DrawTrainWagonPurchaseInfo(int x, int y, EngineID engine_number)

    97

{

    98

	const RailVehicleInfo *rvi = RailVehInfo(engine_number);

    99

	bool refittable = (EngInfo(engine_number)->refit_mask != 0);

   100

   101

	/* Purchase cost */

   102

	SetDParam(0, (rvi->base_cost * _price.build_railwagon) >> 8);

   103

	DrawString(x, y, STR_PURCHASE_INFO_COST, 0);

   104

	y += 10;

   105

   106

	/* Wagon weight - (including cargo) */

   107

	SetDParam(0, rvi->weight);

   108

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

   109

	DrawString(x, y, STR_PURCHASE_INFO_WEIGHT_CWEIGHT, 0);

   110

	y += 10;

   111

   112

	/* Cargo type + capacity, or N/A */

   113

	SetDParam(0, STR_8838_N_A);

   114

	SetDParam(2, STR_EMPTY);

   115

	if (rvi->capacity != 0) {

   116

		SetDParam(0, _cargoc.names_long[rvi->cargo_type]);

   117

		SetDParam(1, rvi->capacity);

   118

		SetDParam(2, refittable ? STR_9842_REFITTABLE : STR_EMPTY);

   119

	}

   120

	DrawString(x, y, STR_PURCHASE_INFO_CAPACITY, 0);

   121

	y += 10;

   122

   123

	/* Wagon speed limit, displayed if above zero */

   124

	if (rvi->max_speed > 0 && _patches.wagon_speed_limits) {

   125

		SetDParam(0, rvi->max_speed);

   126

		DrawString(x,y, STR_PURCHASE_INFO_SPEED, 0);

   127

		y += 10;

   128

	}

   129

   130

	/* Additional text from NewGRF */

   131

	y += ShowAdditionalText(x, y, 227, engine_number);

   132

}

   133

   134

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

   135

{

   136

	Vehicle *v, *found;

   137

   138

	if (!success) return;

   139

   140

	// find a locomotive in the depot.

   141

	found = NULL;

   142

	FOR_ALL_VEHICLES(v) {

   143

		if (v->type == VEH_Train && IsFrontEngine(v) &&

   144

				v->tile == tile &&

   145

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

   146

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

   147

			found = v;

   148

		}

   149

	}

   150

   151

	// if we found a loco,

   152

	if (found != NULL) {

   153

		found = GetLastVehicleInChain(found);

   154

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

   155

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

   156

		RebuildVehicleLists();

   157

	}

   158

}

   159

   160

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

   161

{

   162

	const Vehicle *v;

   163

   164

	if (!success) return;

   165

   166

	v = GetVehicle(_new_vehicle_id);

   167

	if (tile == _backup_orders_tile) {

   168

		_backup_orders_tile = 0;

   169

		RestoreVehicleOrders(v, _backup_orders_data);

   170

	}

   171

	ShowTrainViewWindow(v);

   172

}

   173

   174

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

   175

{

   176

	if (success) ShowTrainViewWindow(GetVehicle(_new_vehicle_id));

   177

}

   178

   179

static void engine_drawing_loop(int *x, int *y, int *pos, int *sel,

   180

	EngineID *selected_id, RailType railtype, byte show_max, bool is_engine)

   181

{

   182

	EngineID j;

   183

   184

	for (j = 0; j < NUM_TRAIN_ENGINES; j++) {

   185

		EngineID i = GetRailVehAtPosition(j);

   186

		const Engine *e = GetEngine(i);

   187

		const RailVehicleInfo *rvi = RailVehInfo(i);

   188

   189

		if (!HasPowerOnRail(e->railtype, railtype) || !(rvi->flags & RVI_WAGON) != is_engine ||

   190

				!HASBIT(e->player_avail, _local_player))

   191

			continue;

   192

   193

		if (*sel == 0) *selected_id = i;

   194

   195

		if (IS_INT_INSIDE(--*pos, -show_max, 0)) {

   196

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

   197

			DrawTrainEngine(*x + 29, *y + 6, i, GetEnginePalette(i, _local_player));

   198

			*y += 14;

   199

		}

   200

		--*sel;

   201

	}

   202

}

   203

   204

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

   205

{

   206

	switch (e->event) {

   207

	case WE_PAINT:

   208

   209

		if (w->window_number == 0) SETBIT(w->disabled_state, 5);

   210

   211

		{

   212

			int count = 0;

   213

			RailType railtype = WP(w,buildtrain_d).railtype;

   214

			EngineID i;

   215

   216

			for (i = 0; i < NUM_TRAIN_ENGINES; i++) {

   217

				const Engine *e = GetEngine(i);

   218

				if (HasPowerOnRail(e->railtype, railtype) &&

   219

						HASBIT(e->player_avail, _local_player)) {

   220

					count++;

   221

				}

   222

			}

   223

			SetVScrollCount(w, count);

   224

		}

   225

   226

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

   227

		DrawWindowWidgets(w);

   228

   229

		{

   230

			RailType railtype = WP(w,buildtrain_d).railtype;

   231

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

   232

			int pos = w->vscroll.pos;

   233

			int x = 1;

   234

			int y = 15;

   235

			EngineID selected_id = INVALID_ENGINE;

   236

   237

			/* Ensure that custom engines which substituted wagons

   238

			 * are sorted correctly.

   239

			 * XXX - DO NOT EVER DO THIS EVER AGAIN! GRRR hacking in wagons as

   240

			 * engines to get more types.. Stays here until we have our own format

   241

			 * then it is exit!!! */

   242

			engine_drawing_loop(&x, &y, &pos, &sel, &selected_id, railtype, w->vscroll.cap, true); // True engines

   243

			engine_drawing_loop(&x, &y, &pos, &sel, &selected_id, railtype, w->vscroll.cap, false); // Feeble wagons

   244

   245

			WP(w,buildtrain_d).sel_engine = selected_id;

   246

   247

			if (selected_id != INVALID_ENGINE) {

   248

				const RailVehicleInfo *rvi = RailVehInfo(selected_id);

   249

   250

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

   251

					/* it's an engine */

   252

					DrawTrainEnginePurchaseInfo(2, w->widget[4].top + 1,selected_id);

   253

				} else {

   254

					/* it's a wagon */

   255

					DrawTrainWagonPurchaseInfo(2, w->widget[4].top + 1, selected_id);

   256

				}

   257

			}

   258

		}

   259

	break;

   260

   261

	case WE_CLICK: {

   262

		switch (e->click.widget) {

   263

		case 2: {

   264

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

   265

			if (i < w->vscroll.cap) {

   266

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

   267

				SetWindowDirty(w);

   268

			}

   269

		} break;

   270

		case 5: {

   271

			EngineID sel_eng = WP(w,buildtrain_d).sel_engine;

   272

			if (sel_eng != INVALID_ENGINE)

   273

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

   274

		}	break;

   275

		case 6: { /* rename */

   276

			EngineID sel_eng = WP(w,buildtrain_d).sel_engine;

   277

			if (sel_eng != INVALID_ENGINE) {

   278

				WP(w,buildtrain_d).rename_engine = sel_eng;

   279

				ShowQueryString(GetCustomEngineName(sel_eng),

   280

					STR_886A_RENAME_TRAIN_VEHICLE_TYPE, 31, 160, w->window_class, w->window_number, CS_ALPHANUMERAL);

   281

			}

   282

		} break;

   283

		}

   284

	} break;

   285

   286

	case WE_ON_EDIT_TEXT: {

   287

		if (e->edittext.str[0] != '\0') {

   288

			_cmd_text = e->edittext.str;

   289

			DoCommandP(0, WP(w,buildtrain_d).rename_engine, 0, NULL,

   290

				CMD_RENAME_ENGINE | CMD_MSG(STR_886B_CAN_T_RENAME_TRAIN_VEHICLE));

   291

		}

   292

	} break;

   293

   294

	case WE_RESIZE: {

   295

		if (e->sizing.diff.y == 0)

   296

			break;

   297

   298

		w->vscroll.cap += e->sizing.diff.y / 14;

   299

		w->widget[2].unkA = (w->vscroll.cap << 8) + 1;

   300

	} break;

   301

	}

   302

}

   303

   304

static const Widget _new_rail_vehicle_widgets[] = {

   305

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

   306

{    WWT_CAPTION,   RESIZE_NONE,    14,    11,   227,     0,    13, STR_JUST_STRING,        STR_018C_WINDOW_TITLE_DRAG_THIS},

   307

{     WWT_MATRIX, RESIZE_BOTTOM,    14,     0,   215,    14,   125, 0x801,                  STR_8843_TRAIN_VEHICLE_SELECTION},

   308

{  WWT_SCROLLBAR, RESIZE_BOTTOM,    14,   216,   227,    14,   125, 0x0,                    STR_0190_SCROLL_BAR_SCROLLS_LIST},

   309

{      WWT_PANEL,     RESIZE_TB,    14,     0,   227,   126,   197, 0x0,                    STR_NULL},

   310

{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,     0,   107,   198,   209, STR_881F_BUILD_VEHICLE, STR_8844_BUILD_THE_HIGHLIGHTED_TRAIN},

   311

{ WWT_PUSHTXTBTN,     RESIZE_TB,    14,   108,   215,   198,   209, STR_8820_RENAME,        STR_8845_RENAME_TRAIN_VEHICLE_TYPE},

   312

{  WWT_RESIZEBOX,     RESIZE_TB,    14,   216,   227,   198,   209, 0x0,                    STR_RESIZE_BUTTON},

   313

{   WIDGETS_END},

   314

};

   315

   316

static const WindowDesc _new_rail_vehicle_desc = {

   317

	-1, -1, 228, 210,

   318

	WC_BUILD_VEHICLE,0,

   319

	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS | WDF_RESIZABLE,

   320

	_new_rail_vehicle_widgets,

   321

	NewRailVehicleWndProc

   322

};

   323

   324

void ShowBuildTrainWindow(TileIndex tile)

   325

{

   326

	Window *w;

   327

   328

	DeleteWindowById(WC_BUILD_VEHICLE, tile);

   329

   330

	w = AllocateWindowDesc(&_new_rail_vehicle_desc);

   331

	w->window_number = tile;

   332

	w->vscroll.cap = 8;

   333

	w->widget[2].unkA = (w->vscroll.cap << 8) + 1;

   334

   335

	w->resize.step_height = 14;

   336

	w->resize.height = w->height - 14 * 4; // Minimum of 4 vehicles in the display

   337

   338

	if (tile != 0) {

   339

		w->caption_color = GetTileOwner(tile);

   340

		WP(w,buildtrain_d).railtype = GetRailType(tile);

   341

	} else {

   342

		w->caption_color = _local_player;

   343

		WP(w,buildtrain_d).railtype = GetBestRailtype(GetPlayer(_local_player));

   344

	}

   345

}

   346

   347

/**

   348

 * Get the number of pixels for the given wagon length.

   349

 * @param len Length measured in 1/8ths of a standard wagon.

   350

 * @return Number of pixels across.

   351

 */

   352

static int WagonLengthToPixels(int len) {

   353

	return (len * _traininfo_vehicle_width) / 8;

   354

}

   355

   356

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

   357

{

   358

	DrawPixelInfo tmp_dpi, *old_dpi;

   359

	int dx = -(skip * 8) / _traininfo_vehicle_width;

   360

	/* Position of highlight box */

   361

	int highlight_l = 0;

   362

	int highlight_r = 0;

   363

   364

	if (!FillDrawPixelInfo(&tmp_dpi, x - 2, y - 1, count + 1, 14)) return;

   365

   366

	count = (count * 8) / _traininfo_vehicle_width;

   367

   368

	old_dpi = _cur_dpi;

   369

	_cur_dpi = &tmp_dpi;

   370

   371

	do {

   372

		int width = v->u.rail.cached_veh_length;

   373

   374

		if (dx + width > 0) {

   375

			if (dx <= count) {

   376

				PalSpriteID pal = (v->vehstatus & VS_CRASHED) ? PALETTE_CRASH : GetVehiclePalette(v);

   377

				DrawSprite(GetTrainImage(v, DIR_W) | pal, 16 + WagonLengthToPixels(dx), 7 + (is_custom_sprite(RailVehInfo(v->engine_type)->image_index) ? _traininfo_vehicle_pitch : 0));

   378

				if (v->index == selection) {

   379

					/* Set the highlight position */

   380

					highlight_l = WagonLengthToPixels(dx) + 1;

   381

					highlight_r = WagonLengthToPixels(dx + width) + 1;

   382

				}

   383

			}

   384

		}

   385

		dx += width;

   386

   387

		v = v->next;

   388

	} while (dx < count && v != NULL);

   389

   390

	if (highlight_l != highlight_r) {

   391

		/* Draw the highlight. Now done after drawing all the engines, as

   392

		 * the next engine after the highlight could overlap it. */

   393

		DrawFrameRect(highlight_l, 0, highlight_r, 13, 15, FR_BORDERONLY);

   394

	}

   395

   396

	_cur_dpi = old_dpi;

   397

}

   398

   399

static void DrawTrainDepotWindow(Window *w)

   400

{

   401

	TileIndex tile;

   402

	Vehicle *v, *u;

   403

	int num,x,y,i, hnum;

   404

	Depot *depot;

   405

   406

	tile = w->window_number;

   407

   408

	/* setup disabled buttons */

   409

	w->disabled_state =

   410

		IsTileOwner(tile, _local_player) ? 0 : ((1 << 4) | (1 << 5) | (1 << 8) | (1<<9));

   411

   412

	/* determine amount of items for scroller */

   413

	num = 0;

   414

	hnum = 8;

   415

	FOR_ALL_VEHICLES(v) {

   416

		if (v->type == VEH_Train &&

   417

				(IsFrontEngine(v) || IsFreeWagon(v)) &&

   418

				v->tile == tile &&

   419

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

   420

			num++;

   421

			// determine number of items in the X direction.

   422

			if (IsFrontEngine(v)) {

   423

				hnum = max(hnum, v->u.rail.cached_total_length);

   424

			}

   425

		}

   426

	}

   427

   428

	/* Always have 1 empty row, so people can change the setting of the train */

   429

	num++;

   430

   431

	SetVScrollCount(w, num);

   432

	SetHScrollCount(w, WagonLengthToPixels(hnum));

   433

   434

	/* locate the depot struct */

   435

	depot = GetDepotByTile(tile);

   436

	assert(depot != NULL);

   437

   438

	SetDParam(0, depot->town_index);

   439

	DrawWindowWidgets(w);

   440

   441

	x = 2;

   442

	y = 15;

   443

	num = w->vscroll.pos;

   444

   445

	// draw all trains

   446

	FOR_ALL_VEHICLES(v) {

   447

		if (v->type == VEH_Train && IsFrontEngine(v) &&

   448

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

   449

				--num < 0 && num >= -w->vscroll.cap) {

   450

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

   451

			/* Draw the train number */

   452

			SetDParam(0, v->unitnumber);

   453

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

   454

   455

			// Number of wagons relative to a standard length wagon (rounded up)

   456

			SetDParam(0, (v->u.rail.cached_total_length + 7) / 8);

   457

			DrawStringRightAligned(w->widget[6].right - 1, y + 4, STR_TINY_BLACK, 0); // Draw the counter

   458

   459

			/* Draw the pretty flag */

   460

			DrawSprite(v->vehstatus & VS_STOPPED ? SPR_FLAG_VEH_STOPPED : SPR_FLAG_VEH_RUNNING, x + 15, y);

   461

   462

			y += 14;

   463

		}

   464

	}

   465

   466

	// draw all remaining vehicles

   467

	FOR_ALL_VEHICLES(v) {

   468

		if (v->type == VEH_Train && IsFreeWagon(v) &&

   469

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

   470

				--num < 0 && num >= -w->vscroll.cap) {

   471

			DrawTrainImage(v, x+50, y, w->hscroll.cap - 29, 0, WP(w,traindepot_d).sel);

   472

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

   473

   474

			/*Draw the train counter */

   475

			i = 0;

   476

			u = v;

   477

			do i++; while ( (u=u->next) != NULL); // Determine length of train

   478

			SetDParam(0, i);                      // Set the counter

   479

			DrawStringRightAligned(w->widget[6].right - 1, y + 4, STR_TINY_BLACK, 0); // Draw the counter

   480

			y += 14;

   481

		}

   482

	}

   483

}

   484

   485

typedef struct GetDepotVehiclePtData {

   486

	Vehicle *head;

   487

	Vehicle *wagon;

   488

} GetDepotVehiclePtData;

   489

   490

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

   491

{