1

/* $Id$ */

     2

     3

#include "stdafx.h"

     4

#include "openttd.h"

     5

#include "debug.h"

     6

#include "functions.h"

     7

#include "table/sprites.h"

     8

#include "table/strings.h"

     9

#include "map.h"

    10

#include "window.h"

    11

#include "gui.h"

    12

#include "gfx.h"

    13

#include "vehicle.h"

    14

#include "viewport.h"

    15

#include "station.h"

    16

#include "command.h"

    17

#include "player.h"

    18

#include "engine.h"

    19

#include "vehicle_gui.h"

    20

#include "depot.h"

    21

#include "train.h"

    22

    23

int _traininfo_vehicle_pitch = 0;

    24

    25

/**

    26

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

    27

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

    28

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

    29

 */

    30

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

    31

{

    32

	const RailVehicleInfo *rvi = RailVehInfo(engine_number);

    33

	const Engine* e = GetEngine(engine_number);

    34

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

    35

	YearMonthDay ymd;

    36

	ConvertDayToYMD(&ymd, e->intro_date);

    37

    38

	/* Purchase Cost - Engine weight */

    39

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

    40

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

    41

	DrawString(x,y, STR_PURCHASE_INFO_COST_WEIGHT, 0);

    42

	y += 10;

    43

    44

	/* Max speed - Engine power */

    45

	SetDParam(0, rvi->max_speed * 10 >> 4);

    46

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

    47

	DrawString(x,y, STR_PURCHASE_INFO_SPEED_POWER, 0);

    48

	y += 10;

    49

    50

	/* Running cost */

    51

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

    52

	DrawString(x,y, STR_PURCHASE_INFO_RUNNINGCOST, 0);

    53

	y += 10;

    54

    55

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

    56

	if (rvi->pow_wag_power != 0) {

    57

		SetDParam(0, rvi->pow_wag_power);

    58

		SetDParam(1, rvi->pow_wag_weight);

    59

		DrawString(x,y, STR_PURCHASE_INFO_PWAGPOWER_PWAGWEIGHT, 0);

    60

		y += 10;

    61

	};

    62

    63

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

    64

	SetDParam(0, STR_8838_N_A);

    65

	SetDParam(2, STR_EMPTY);

    66

	if (rvi->capacity != 0) {

    67

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

    68

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

    69

		SetDParam(2, STR_9842_REFITTABLE);

    70

	}

    71

	DrawString(x,y, STR_PURCHASE_INFO_CAPACITY, 0);

    72

	y += 10;

    73

    74

	/* Design date - Life length */

    75

	SetDParam(0, ymd.year + 1920);

    76

	SetDParam(1, e->lifelength);

    77

	DrawString(x,y, STR_PURCHASE_INFO_DESIGNED_LIFE, 0);

    78

	y += 10;

    79

    80

	/* Reliability */

    81

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

    82

	DrawString(x,y, STR_PURCHASE_INFO_RELIABILITY, 0);

    83

	y += 10;

    84

}

    85

    86

/**

    87

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

    88

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

    89

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

    90

 */

    91

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

    92

{

    93

	const RailVehicleInfo *rvi = RailVehInfo(engine_number);

    94

	bool refittable = (_engine_info[engine_number].refit_mask != 0);

    95

    96

	/* Purchase cost */

    97

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

    98

	DrawString(x, y, STR_PURCHASE_INFO_COST, 0);

    99

	y += 10;

   100

   101

	/* Wagon weight - (including cargo) */

   102

	SetDParam(0, rvi->weight);

   103

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

   104

	DrawString(x, y, STR_PURCHASE_INFO_WEIGHT_CWEIGHT, 0);

   105

	y += 10;

   106

   107

	/* Cargo type + capacity */

   108

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

   109

	SetDParam(1, rvi->capacity);

   110

	SetDParam(2, refittable ? STR_9842_REFITTABLE : STR_EMPTY);

   111

	DrawString(x, y, STR_PURCHASE_INFO_CAPACITY, 0);

   112

	y += 10;

   113

   114

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

   115

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

   116

		SetDParam(0, rvi->max_speed * 10 >> 4);

   117

		DrawString(x,y, STR_PURCHASE_INFO_SPEED, 0);

   118

		y += 10;

   119

	}

   120

}

   121

   122

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

   123

{

   124

	Vehicle *v,*found;

   125

   126

	if (!success)

   127

		return;

   128

   129

	// find a locomotive in the depot.

   130

	found = NULL;

   131

	FOR_ALL_VEHICLES(v) {

   132

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

   133

				v->tile == tile &&

   134

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

   135

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

   136

				return;

   137

			found = v;

   138

		}

   139

	}

   140

   141

	// if we found a loco,

   142

	if (found != NULL) {

   143

		found = GetLastVehicleInChain(found);

   144

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

   145

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

   146

		RebuildVehicleLists();

   147

	}

   148

}

   149

   150

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

   151

{

   152

	const Vehicle* v;

   153

   154

	if (!success) return;

   155

   156

	v = GetVehicle(_new_train_id);

   157

	if (tile == _backup_orders_tile) {

   158

		_backup_orders_tile = 0;

   159

		RestoreVehicleOrders(v, _backup_orders_data);

   160

	}

   161

	ShowTrainViewWindow(v);

   162

}

   163

   164

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

   165

{

   166

	if (success) ShowTrainViewWindow(GetVehicle(_new_train_id));

   167

}

   168

   169

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

   170

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

   171

{

   172

	EngineID i;

   173

   174

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

   175

		const Engine *e = GetEngine(i);

   176

		const RailVehicleInfo *rvi = RailVehInfo(i);

   177

   178

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

   179

				!HASBIT(e->player_avail, _local_player))

   180

			continue;

   181

   182

		if (*sel == 0)

   183

			*selected_id = i;

   184

   185

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

   186

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

   187

			DrawTrainEngine(*x + 29, *y + 6, i,

   188

				SPRITE_PALETTE(PLAYER_SPRITE_COLOR(_local_player)));

   189

			*y += 14;

   190

		}

   191

		--*sel;

   192

	}

   193

}

   194

   195

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

   196

{

   198

	case WE_PAINT:

   199

   200

		if (w->window_number == 0)

   201

			SETBIT(w->disabled_state, 5);

   202

   203

		{

   204

			int count = 0;

   205

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

   206

			EngineID i;

   207

   208

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

   209

				const Engine *e = GetEngine(i);

   210

				if (IsCompatibleRail(e->railtype, railtype)

   211

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

   212

					count++;

   213

			}

   214

			SetVScrollCount(w, count);

   215

		}

   216

   217

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

   218

		DrawWindowWidgets(w);

   219

   220

		{

   221

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

   222

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

   223

			int pos = w->vscroll.pos;

   224

			int x = 1;

   225

			int y = 15;

   226

			EngineID selected_id = INVALID_ENGINE;

   227

   228

			/* Ensure that custom engines which substituted wagons

   229

			 * are sorted correctly.

   230

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

   231

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

   232

			 * then it is exit!!! */

   233

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

   234

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

   235

   236

			WP(w,buildtrain_d).sel_engine = selected_id;

   237

   238

			if (selected_id != INVALID_ENGINE) {

   239

				const RailVehicleInfo *rvi = RailVehInfo(selected_id);

   240

   241

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

   242

					/* it's an engine */

   243

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

   244

				} else {

   245

					/* it's a wagon */

   246

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

   247

				}

   248

			}

   249

		}

   250

	break;

   251

   252

	case WE_CLICK: {

   254

		case 2: {

   255

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

   256

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

   257

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

   258

				SetWindowDirty(w);

   259

			}

   260

		} break;

   261

		case 5: {

   262

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

   263

			if (sel_eng != INVALID_ENGINE)

   264

				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));

   265

		}	break;

   266

		case 6: { /* rename */

   267

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

   268

			if (sel_eng != INVALID_ENGINE) {

   269

				WP(w,buildtrain_d).rename_engine = sel_eng;

   270

				ShowQueryString(GetCustomEngineName(sel_eng),

   271

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

   272

			}

   273

		} break;

   274

		}

   275

	} break;

   276

   277

	case WE_4:

   278

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

   279

			DeleteWindow(w);

   280

		}

   281

		break;

   282

   283

	case WE_ON_EDIT_TEXT: {

   284

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

   285

			_cmd_text = e->edittext.str;

   286

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

   287

				CMD_RENAME_ENGINE | CMD_MSG(STR_886B_CAN_T_RENAME_TRAIN_VEHICLE));

   288

		}

   289

	} break;

   290

   291

	case WE_RESIZE: {

   292

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

   293

			break;

   294

   295

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

   296

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

   297

	} break;

   298

	}

   299

}

   300

   301

static const Widget _new_rail_vehicle_widgets[] = {

   302

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

   303

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

   304

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

   305

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

   306

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

   307

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

   308

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

   309

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

   310

{   WIDGETS_END},

   311

};

   312

   313

static const WindowDesc _new_rail_vehicle_desc = {

   314

	-1, -1, 228, 210,

   315

	WC_BUILD_VEHICLE,0,

   316

	WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_UNCLICK_BUTTONS | WDF_RESIZABLE,

   317

	_new_rail_vehicle_widgets,

   318

	NewRailVehicleWndProc

   319

};

   320

   321

static void ShowBuildTrainWindow(TileIndex tile)

   322

{

   323

	Window *w;

   324

   325

	DeleteWindowById(WC_BUILD_VEHICLE, tile);

   326

   327

	w = AllocateWindowDesc(&_new_rail_vehicle_desc);

   328

	w->window_number = tile;

   329

	w->vscroll.cap = 8;

   330

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

   331

   332

	w->resize.step_height = 14;

   333

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

   334

   335

	if (tile != 0) {

   336

		w->caption_color = GetTileOwner(tile);

   337

		WP(w,buildtrain_d).railtype = GB(_m[tile].m3, 0, 4);

   338

	} else {

   339

		w->caption_color = _local_player;

   340

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

   341

	}

   342

}

   343

   344

/**

   345

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

   346

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

   347

 * @return Number of pixels across.

   348

 */

   349

static int WagonLengthToPixels(int len) {

   350

	return (len * 29) / 8;

   351

}

   352

   353

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

   354

{

   355

	int dx = 0;

   356

	count *= 8;

   357

   358

	do {

   359

		if (--skip < 0) {

   360

			int image = GetTrainImage(v, 6);

   361

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

   362

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

   363

   364

			if (dx + width <= count) {

   365

				if (v->vehstatus & VS_CRASHED)

   366

					ormod = PALETTE_CRASH;

   367

				DrawSprite(image | ormod, x + 14 + WagonLengthToPixels(dx), y + 6 + (is_custom_sprite(RailVehInfo(v->engine_type)->image_index) ? _traininfo_vehicle_pitch : 0));

   368

				if (v->index == selection)

   369

					DrawFrameRect(x - 1 + WagonLengthToPixels(dx), y - 1, x + WagonLengthToPixels(dx + width) - 1, y + 12, 15, FR_BORDERONLY);

   370

			}

   371

			dx += width;

   372

		}

   373

   374

		v = v->next;

   375

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

   376

}

   377

   378

static void DrawTrainDepotWindow(Window *w)

   379

{

   380

	TileIndex tile;

   381

	Vehicle *v, *u;

   382

	int num,x,y,i, hnum;

   383

	Depot *depot;

   384

   385

	tile = w->window_number;

   386

   387

	/* setup disabled buttons */

   388

	w->disabled_state =

   389

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

   390

   391

	/* determine amount of items for scroller */

   392

	num = 0;

   393

	hnum = 8;

   394

	FOR_ALL_VEHICLES(v) {

   395

		if (v->type == VEH_Train &&

   396

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

   397

				v->tile == tile &&

   398

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

   399

			num++;

   400

			// determine number of items in the X direction.

   401

			if (IsFrontEngine(v)) {

   402

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

   403

			}

   404

		}

   405

	}

   406

   407

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

   408

	num++;

   409

   410

	SetVScrollCount(w, num);

   411

	SetHScrollCount(w, (hnum + 7) / 8);

   412

   413

	/* locate the depot struct */

   414

	depot = GetDepotByTile(tile);

   415

	assert(depot != NULL);

   416

   417

	SetDParam(0, depot->town_index);

   418

	DrawWindowWidgets(w);

   419

   420

	x = 2;

   421

	y = 15;

   422

	num = w->vscroll.pos;

   423

   424

	// draw all trains

   425

	FOR_ALL_VEHICLES(v) {

   426

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

   427

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

   428

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

   429

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

   430

			/* Draw the train number */

   431

			SetDParam(0, v->unitnumber);

   432

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

   433

   434

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

   435

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

   436

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

   437

   438

			/* Draw the pretty flag */

   439

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

   440

   441

			y += 14;

   442

		}

   443

	}

   444

   445

	// draw all remaining vehicles

   446

	FOR_ALL_VEHICLES(v) {

   447

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

   448

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

   449

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

   450

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

   451

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

   452

   453

			/*Draw the train counter */

   454

			i = 0;

   455

			u = v;

   456

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

   457

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

   458

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

   459

			y += 14;

   460

		}

   461

	}

   462

}

   463

   464

typedef struct GetDepotVehiclePtData {

   465

	Vehicle *head;

   466

	Vehicle *wagon;

   467

} GetDepotVehiclePtData;

   468

   469

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

   470

{

   471

	int row;

   472

	int skip = 0;

   473

	Vehicle *v;

   474

   475

	x = x - 23;

   476

   477

	row = (y - 14) / 14;

   478

	if ((uint)row >= w->vscroll.cap) return 1; /* means err */

   479

   480

	row += w->vscroll.pos;

   481

   482

	/* go through all the locomotives */

   483

	FOR_ALL_VEHICLES(v) {

   484

		if (v->type == VEH_Train &&

   485

				IsFrontEngine(v) &&

   486

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

   487

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

   488

				--row < 0) {

   489

			skip = w->hscroll.pos;

   490

			goto found_it;

   491

		}

   492

	}

   493

   494

	x -= 29; /* free wagons don't have an initial loco. */