31

typedef GUIList<const Group*> GUIGroupList;

    33

static void BuildGroupList(GUIGroupList *gl, PlayerID owner, VehicleType vehicle_type)

    37

	if (!(gl->flags & VL_REBUILD)) return;

    39

	const Group **list = MallocT<const Group*>(GetGroupArraySize());

    41

	const Group *g;

    46

	gl->sort_list = ReallocT(gl->sort_list, n);

    47

	gl->list_length = n;

    50

	free(list);

    52

	gl->flags &= ~VL_REBUILD;

    53

	gl->flags |= VL_RESORT;

    86

static void SortGroupList(GUIGroupList *gl)

    88

	if (!(gl->flags & VL_RESORT)) return;

    90

	qsort((void*)gl->sort_list, gl->list_length, sizeof(gl->sort_list[0]), GroupNameSorter);

    92

	gl->resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS;

    93

	gl->flags &= ~VL_RESORT;

   125

enum GroupActionListFunction {

   126

	GALF_REPLACE,

   127

	GALF_SERVICE,

   128

	GALF_DEPOT,

   129

	GALF_ADD_SHARED,

   130

	GALF_REMOVE_ALL,

   131

};

   132

   133

/**

   134

 * Update/redraw the group action dropdown

   135

 * @param w   the window the dropdown belongs to

   136

 * @param gid the currently selected group in the window

   137

 */

   138

static void ShowGroupActionDropdown(Window *w, GroupID gid)

   139

{

   140

	DropDownList *list = new DropDownList();

   141

   142

	list->push_back(new DropDownListStringItem(STR_REPLACE_VEHICLES,    GALF_REPLACE, false));

   143

	list->push_back(new DropDownListStringItem(STR_SEND_FOR_SERVICING,  GALF_SERVICE, false));

   144

	list->push_back(new DropDownListStringItem(STR_SEND_TRAIN_TO_DEPOT, GALF_DEPOT,   false));

   145

   146

	if (IsValidGroupID(gid)) {

   147

		list->push_back(new DropDownListStringItem(STR_GROUP_ADD_SHARED_VEHICLE,  GALF_ADD_SHARED, false));

   148

		list->push_back(new DropDownListStringItem(STR_GROUP_REMOVE_ALL_VEHICLES, GALF_REMOVE_ALL, false));

   149

	}

   150

   151

	ShowDropDownList(w, list, 0, GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN);

   152

}

   153

   185

struct VehicleGroupWindow : public Window, public VehicleListBase {

   186

	GroupID group_sel;

   187

	VehicleID vehicle_sel;

   188

	GUIGroupList groups;

   190

	VehicleGroupWindow(const WindowDesc *desc, WindowNumber window_number) : Window(desc, NULL, window_number)

   191

	{

   192

		const PlayerID owner = (PlayerID)GB(this->window_number, 0, 8);

   194

		this->caption_color = owner;

   195

		this->hscroll.cap = 224;

   196

		this->resize.step_width = 1;

   198

		switch (this->vehicle_type) {

   199

			default: NOT_REACHED();

   200

			case VEH_TRAIN:

   201

			case VEH_ROAD:

   202

				this->vscroll.cap = 9;

   203

				this->vscroll2.cap = 6;

   204

				this->resize.step_height = PLY_WND_PRC__SIZE_OF_ROW_SMALL;

   205

				break;

   206

			case VEH_SHIP:

   207

			case VEH_AIRCRAFT:

   208

				this->vscroll.cap = 9;

   209

				this->vscroll2.cap = 4;

   210

				this->resize.step_height = PLY_WND_PRC__SIZE_OF_ROW_BIG2;

   211

				break;

   212

		}

   214

		this->widget[GRP_WIDGET_LIST_GROUP].data = (this->vscroll.cap << 8) + 1;

   215

		this->widget[GRP_WIDGET_LIST_VEHICLE].data = (this->vscroll2.cap << 8) + 1;

   217

		switch (this->vehicle_type) {

   218

			default: NOT_REACHED();

   219

			case VEH_TRAIN:    this->sorting = &_sorting.train;    break;

   220

			case VEH_ROAD:     this->sorting = &_sorting.roadveh;  break;

   221

			case VEH_SHIP:     this->sorting = &_sorting.ship;     break;

   222

			case VEH_AIRCRAFT: this->sorting = &_sorting.aircraft; break;

   223

		}

   225

		this->vehicle_type = (VehicleType)GB(this->window_number, 11, 5);

   226

   227

		this->vehicles.sort_list = NULL;

   228

		this->vehicles.sort_type = this->sorting->criteria;

   229

		this->vehicles.flags = VL_REBUILD | (this->sorting->order ? VL_DESC : VL_NONE);

   230

		this->vehicles.resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS; // Set up resort timer

   231

   232

		this->groups.sort_list = NULL;

   233

		this->groups.flags = VL_REBUILD | VL_NONE;

   234

		this->groups.resort_timer = DAY_TICKS * PERIODIC_RESORT_DAYS; // Set up resort timer

   235

   236

		this->group_sel = ALL_GROUP;

   237

		this->vehicle_sel = INVALID_VEHICLE;

   238

   239

		switch (this->vehicle_type) {

   240

			case VEH_TRAIN:

   241

				this->widget[GRP_WIDGET_LIST_VEHICLE].tooltips = STR_883D_TRAINS_CLICK_ON_TRAIN_FOR;

   242

				this->widget[GRP_WIDGET_AVAILABLE_VEHICLES].data = STR_AVAILABLE_TRAINS;

   243

   244

				this->widget[GRP_WIDGET_CREATE_GROUP].data = SPR_GROUP_CREATE_TRAIN;

   245

				this->widget[GRP_WIDGET_RENAME_GROUP].data = SPR_GROUP_RENAME_TRAIN;

   246

				this->widget[GRP_WIDGET_DELETE_GROUP].data = SPR_GROUP_DELETE_TRAIN;

   247

				break;

   248

   249

			case VEH_ROAD:

   250

				this->widget[GRP_WIDGET_LIST_VEHICLE].tooltips = STR_901A_ROAD_VEHICLES_CLICK_ON;

   251

				this->widget[GRP_WIDGET_AVAILABLE_VEHICLES].data = STR_AVAILABLE_ROAD_VEHICLES;

   252

   253

				this->widget[GRP_WIDGET_CREATE_GROUP].data = SPR_GROUP_CREATE_ROADVEH;

   254

				this->widget[GRP_WIDGET_RENAME_GROUP].data = SPR_GROUP_RENAME_ROADVEH;

   255

				this->widget[GRP_WIDGET_DELETE_GROUP].data = SPR_GROUP_DELETE_ROADVEH;

   256

				break;

   257

   258

			case VEH_SHIP:

   259

				this->widget[GRP_WIDGET_LIST_VEHICLE].tooltips = STR_9823_SHIPS_CLICK_ON_SHIP_FOR;

   260

				this->widget[GRP_WIDGET_AVAILABLE_VEHICLES].data = STR_AVAILABLE_SHIPS;

   261

   262

				this->widget[GRP_WIDGET_CREATE_GROUP].data = SPR_GROUP_CREATE_SHIP;

   263

				this->widget[GRP_WIDGET_RENAME_GROUP].data = SPR_GROUP_RENAME_SHIP;

   264

				this->widget[GRP_WIDGET_DELETE_GROUP].data = SPR_GROUP_DELETE_SHIP;

   265

				break;

   266

   267

			case VEH_AIRCRAFT:

   268

				this->widget[GRP_WIDGET_LIST_VEHICLE].tooltips = STR_A01F_AIRCRAFT_CLICK_ON_AIRCRAFT;

   269

				this->widget[GRP_WIDGET_AVAILABLE_VEHICLES].data = STR_AVAILABLE_AIRCRAFT;

   270

   271

				this->widget[GRP_WIDGET_CREATE_GROUP].data = SPR_GROUP_CREATE_AIRCRAFT;

   272

				this->widget[GRP_WIDGET_RENAME_GROUP].data = SPR_GROUP_RENAME_AIRCRAFT;

   273

				this->widget[GRP_WIDGET_DELETE_GROUP].data = SPR_GROUP_DELETE_AIRCRAFT;

   274

				break;

   275

   276

			default: NOT_REACHED();

   277

		}

   278

   279

		this->FindWindowPlacementAndResize(desc);

   282

	~VehicleGroupWindow()

   283

	{

   284

		free((void*)this->vehicles.sort_list);

   285

		free((void*)this->groups.sort_list);

   286

	}

   288

	virtual void OnInvalidateData(int data)

   289

	{

   290

		this->vehicles.flags |= VL_REBUILD;

   291

		this->groups.flags |= VL_REBUILD;

   292

		if (!(IsAllGroupID(this->group_sel) || IsDefaultGroupID(this->group_sel) || IsValidGroupID(this->group_sel))) {

   293

			this->group_sel = ALL_GROUP;

   294

			HideDropDownMenu(this);

   295

		}

   296

		this->SetDirty();

   297

	}

   299

	virtual void OnPaint()

   300

	{

   301

		const PlayerID owner = (PlayerID)GB(this->window_number, 0, 8);

   302

		int x = 203;

   303

		int y2 = PLY_WND_PRC__OFFSET_TOP_WIDGET;

   304

		int y1 = PLY_WND_PRC__OFFSET_TOP_WIDGET + 2;

   305

		int max;

   306

		int i;

   308

		/* If we select the all vehicles, this->list will contain all vehicles of the player

   309

			* else this->list will contain all vehicles which belong to the selected group */

   310

		BuildVehicleList(this, owner, this->group_sel, IsAllGroupID(this->group_sel) ? VLW_STANDARD : VLW_GROUP_LIST);

   311

		SortVehicleList(this);

   314

		BuildGroupList(&this->groups, owner, this->vehicle_type);

   315

		SortGroupList(&this->groups);

   317

		SetVScrollCount(this, this->groups.list_length);

   318

		SetVScroll2Count(this, this->vehicles.list_length);

   320

		/* The drop down menu is out, *but* it may not be used, retract it. */

   321

		if (this->vehicles.list_length == 0 && this->IsWidgetLowered(GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN)) {

   322

			this->RaiseWidget(GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN);

   323

			HideDropDownMenu(this);

   324

		}

   326

		/* Disable all lists management button when the list is empty */

   327

		this->SetWidgetsDisabledState(this->vehicles.list_length == 0 || _local_player != owner,

   328

				GRP_WIDGET_STOP_ALL,

   329

				GRP_WIDGET_START_ALL,

   330

				GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN,

   331

				WIDGET_LIST_END);

   332

   333

		/* Disable the group specific function when we select the default group or all vehicles */

   334

		this->SetWidgetsDisabledState(IsDefaultGroupID(this->group_sel) || IsAllGroupID(this->group_sel) || _local_player != owner,

   335

				GRP_WIDGET_DELETE_GROUP,

   336

				GRP_WIDGET_RENAME_GROUP,

   337

				GRP_WIDGET_REPLACE_PROTECTION,

   338

				WIDGET_LIST_END);

   339

   340

		/* Disable remaining buttons for non-local player

   341

			* Needed while changing _local_player, eg. by cheats

   342

			* All procedures (eg. move vehicle to another group)

   343

			*  verify, whether you are the owner of the vehicle,

   344

			*  so it doesn't have to be disabled

   345

			*/

   346

		this->SetWidgetsDisabledState(_local_player != owner,

   347

				GRP_WIDGET_CREATE_GROUP,

   348

				GRP_WIDGET_AVAILABLE_VEHICLES,

   349

				WIDGET_LIST_END);

   352

		/* If selected_group == DEFAULT_GROUP || ALL_GROUP, draw the standard caption

   353

				We list all vehicles or ungrouped vehicles */

   354

		if (IsDefaultGroupID(this->group_sel) || IsAllGroupID(this->group_sel)) {

   355

			SetDParam(0, owner);

   356

			SetDParam(1, this->vehicles.list_length);

   358

			switch (this->vehicle_type) {

   360

					this->widget[GRP_WIDGET_CAPTION].data = STR_881B_TRAINS;

   361

					this->widget[GRP_WIDGET_REPLACE_PROTECTION].data = SPR_GROUP_REPLACE_OFF_TRAIN;

   364

					this->widget[GRP_WIDGET_CAPTION].data = STR_9001_ROAD_VEHICLES;

   365

					this->widget[GRP_WIDGET_REPLACE_PROTECTION].data = SPR_GROUP_REPLACE_OFF_ROADVEH;

   368

					this->widget[GRP_WIDGET_CAPTION].data = STR_9805_SHIPS;

   369

					this->widget[GRP_WIDGET_REPLACE_PROTECTION].data = SPR_GROUP_REPLACE_OFF_SHIP;

   372

					this->widget[GRP_WIDGET_CAPTION].data =  STR_A009_AIRCRAFT;

   373

					this->widget[GRP_WIDGET_REPLACE_PROTECTION].data = SPR_GROUP_REPLACE_OFF_AIRCRAFT;

   377

		} else {

   378

			const Group *g = GetGroup(this->group_sel);

   380

			SetDParam(0, g->index);

   381

			SetDParam(1, g->num_vehicle);

   383

			switch (this->vehicle_type) {

   384

				case VEH_TRAIN:

   385

					this->widget[GRP_WIDGET_CAPTION].data = STR_GROUP_TRAINS_CAPTION;

   386

					this->widget[GRP_WIDGET_REPLACE_PROTECTION].data = (g->replace_protection) ? SPR_GROUP_REPLACE_ON_TRAIN : SPR_GROUP_REPLACE_OFF_TRAIN;

   387

					break;

   388

				case VEH_ROAD:

   389

					this->widget[GRP_WIDGET_CAPTION].data = STR_GROUP_ROADVEH_CAPTION;

   390

					this->widget[GRP_WIDGET_REPLACE_PROTECTION].data = (g->replace_protection) ? SPR_GROUP_REPLACE_ON_ROADVEH : SPR_GROUP_REPLACE_OFF_ROADVEH;

   391

					break;

   392

				case VEH_SHIP:

   393

					this->widget[GRP_WIDGET_CAPTION].data = STR_GROUP_SHIPS_CAPTION;

   394

					this->widget[GRP_WIDGET_REPLACE_PROTECTION].data = (g->replace_protection) ? SPR_GROUP_REPLACE_ON_SHIP : SPR_GROUP_REPLACE_OFF_SHIP;

   395

					break;

   396

				case VEH_AIRCRAFT:

   397

					this->widget[GRP_WIDGET_CAPTION].data = STR_GROUP_AIRCRAFTS_CAPTION;

   398

					this->widget[GRP_WIDGET_REPLACE_PROTECTION].data = (g->replace_protection) ? SPR_GROUP_REPLACE_ON_AIRCRAFT : SPR_GROUP_REPLACE_OFF_AIRCRAFT;

   399

					break;

   400

				default: NOT_REACHED();

   404

		/* Set text of sort by dropdown */

   405

		this->widget[GRP_WIDGET_SORT_BY_DROPDOWN].data = _vehicle_sort_listing[this->vehicles.sort_type];

   407

		DrawWindowWidgets(this);

   409

		/* Draw Matrix Group

   410

			* The selected group is drawn in white */

   411

		StringID str_all_veh, str_no_group_veh;

   413

		switch (this->vehicle_type) {

   414

			case VEH_TRAIN:

   415

				str_all_veh = STR_GROUP_ALL_TRAINS;

   416

				str_no_group_veh = STR_GROUP_DEFAULT_TRAINS;

   417

				break;

   418

			case VEH_ROAD:

   419

				str_all_veh = STR_GROUP_ALL_ROADS;

   420

				str_no_group_veh = STR_GROUP_DEFAULT_ROADS;

   421

				break;

   422

			case VEH_SHIP:

   423

				str_all_veh = STR_GROUP_ALL_SHIPS;

   424

				str_no_group_veh = STR_GROUP_DEFAULT_SHIPS;

   425

				break;

   426

			case VEH_AIRCRAFT:

   427

				str_all_veh = STR_GROUP_ALL_AIRCRAFTS;

   428

				str_no_group_veh = STR_GROUP_DEFAULT_AIRCRAFTS;

   429

				break;

   430

			default: NOT_REACHED();

   431

		}

   432

		DrawString(10, y1, str_all_veh, IsAllGroupID(this->group_sel) ? TC_WHITE : TC_BLACK);

   434

		y1 += 13;

   436

		DrawString(10, y1, str_no_group_veh, IsDefaultGroupID(this->group_sel) ? TC_WHITE : TC_BLACK);

   438

		max = min(this->vscroll.pos + this->vscroll.cap, this->groups.list_length);

   439

		for (i = this->vscroll.pos ; i < max ; ++i) {

   440

			const Group *g = this->groups.sort_list[i];

   441

   442

			assert(g->owner == owner);

   443

   444

			y1 += PLY_WND_PRC__SIZE_OF_ROW_TINY;

   445

   446

			/* draw the selected group in white, else we draw it in black */

   447

			SetDParam(0, g->index);

   448

			DrawString(10, y1, STR_GROUP_NAME, (this->group_sel == g->index) ? TC_WHITE : TC_BLACK);

   449

   450

			/* draw the number of vehicles of the group */

   451

			SetDParam(0, g->num_vehicle);

   452

			DrawStringRightAligned(187, y1 + 1, STR_GROUP_TINY_NUM, (this->group_sel == g->index) ? TC_WHITE : TC_BLACK);

   453

		}

   454

   455

		DrawSortButtonState(this, GRP_WIDGET_SORT_BY_ORDER, this->vehicles.flags & VL_DESC ? SBS_DOWN : SBS_UP);

   456

   457

		/* Draw Matrix Vehicle according to the vehicle list built before */

   458

		max = min(this->vscroll2.pos + this->vscroll2.cap, this->vehicles.list_length);

   459

		for (i = this->vscroll2.pos ; i < max ; ++i) {

   460

			const Vehicle* v = this->vehicles.sort_list[i];

   461

   462

			assert(v->type == this->vehicle_type && v->owner == owner);

   463

   464

			DrawVehicleImage(v, x + 19, y2 + 6, this->vehicle_sel, this->hscroll.cap, 0);

   465

			DrawVehicleProfitButton(v, x, y2 + 13);

   466

   467

			SetDParam(0, v->unitnumber);

   468

			DrawString(x, y2 + 2, v->IsInDepot() ? STR_021F : (v->age > v->max_age - 366 ? STR_00E3 : STR_00E2), TC_FROMSTRING);

   469

   470

			if (this->resize.step_height == PLY_WND_PRC__SIZE_OF_ROW_BIG2) DrawSmallOrderList(v, x + 138, y2);

   471

   472

			SetDParam(0, v->GetDisplayProfitThisYear());

   473

			SetDParam(1, v->GetDisplayProfitLastYear());

   474

			DrawString(x + 19, y2 + this->resize.step_height - 8, STR_0198_PROFIT_THIS_YEAR_LAST_YEAR, TC_FROMSTRING);

   475

   476

			if (IsValidGroupID(v->group_id)) {

   477

				SetDParam(0, v->group_id);

   478

				DrawString(x + 19, y2, STR_GROUP_TINY_NAME, TC_BLACK);

   479

			}

   480

   481

			y2 += this->resize.step_height;

   482

		}

   483

	}

   484

   485

	virtual void OnClick(Point pt, int widget)

   486

	{

   487

		if (widget != GRP_WIDGET_SORT_BY_DROPDOWN && widget != GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN) HideDropDownMenu(this);

   488

   489

		switch(widget) {

   490

			case GRP_WIDGET_SORT_BY_ORDER: // Flip sorting method ascending/descending

   491

				this->vehicles.flags ^= VL_DESC;

   492

				this->vehicles.flags |= VL_RESORT;

   493

   494

				this->sorting->order = !!(this->vehicles.flags & VL_DESC);

   495

				this->SetDirty();

   496

				break;