4 * This file is part of OpenTTD.
5 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
6 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
7 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
10 /** @file group_gui.cpp GUI for the group window. */
13 #include "textbuf_gui.h"
14 #include "command_func.h"
15 #include "vehicle_gui.h"
16 #include "vehicle_base.h"
17 #include "string_func.h"
18 #include "strings_func.h"
19 #include "window_func.h"
20 #include "vehicle_func.h"
21 #include "autoreplace_gui.h"
22 #include "company_func.h"
23 #include "widgets/dropdown_func.h"
24 #include "tilehighlight_func.h"
25 #include "vehicle_gui_base.h"
26 #include "core/geometry_func.hpp"
27 #include "company_base.h"
29 #include "widgets/group_widget.h"
31 #include "table/sprites.h"
33 #include "safeguards.h"
35 static const int LEVEL_WIDTH
= 10; ///< Indenting width of a sub-group in pixels
37 typedef GUIList
<const Group
*> GUIGroupList
;
39 static const NWidgetPart _nested_group_widgets
[] = {
40 NWidget(NWID_HORIZONTAL
), // Window header
41 NWidget(WWT_CLOSEBOX
, COLOUR_GREY
),
42 NWidget(WWT_CAPTION
, COLOUR_GREY
, WID_GL_CAPTION
),
43 NWidget(WWT_SHADEBOX
, COLOUR_GREY
),
44 NWidget(WWT_DEFSIZEBOX
, COLOUR_GREY
),
45 NWidget(WWT_STICKYBOX
, COLOUR_GREY
),
47 NWidget(NWID_HORIZONTAL
),
49 NWidget(NWID_VERTICAL
),
50 NWidget(WWT_PANEL
, COLOUR_GREY
), SetMinimalTextLines(1, WD_DROPDOWNTEXT_TOP
+ WD_DROPDOWNTEXT_BOTTOM
), SetFill(1, 0), EndContainer(),
51 NWidget(WWT_PANEL
, COLOUR_GREY
, WID_GL_ALL_VEHICLES
), SetFill(1, 0), EndContainer(),
52 NWidget(WWT_PANEL
, COLOUR_GREY
, WID_GL_DEFAULT_VEHICLES
), SetFill(1, 0), EndContainer(),
53 NWidget(NWID_HORIZONTAL
),
54 NWidget(WWT_MATRIX
, COLOUR_GREY
, WID_GL_LIST_GROUP
), SetMatrixDataTip(1, 0, STR_GROUPS_CLICK_ON_GROUP_FOR_TOOLTIP
),
55 SetFill(1, 0), SetResize(0, 1), SetScrollbar(WID_GL_LIST_GROUP_SCROLLBAR
),
56 NWidget(NWID_VSCROLLBAR
, COLOUR_GREY
, WID_GL_LIST_GROUP_SCROLLBAR
),
58 NWidget(WWT_PANEL
, COLOUR_GREY
, WID_GL_INFO
), SetFill(1, 0), EndContainer(),
59 NWidget(NWID_HORIZONTAL
),
60 NWidget(WWT_PUSHIMGBTN
, COLOUR_GREY
, WID_GL_CREATE_GROUP
), SetFill(0, 1),
61 SetDataTip(SPR_GROUP_CREATE_TRAIN
, STR_GROUP_CREATE_TOOLTIP
),
62 NWidget(WWT_PUSHIMGBTN
, COLOUR_GREY
, WID_GL_DELETE_GROUP
), SetFill(0, 1),
63 SetDataTip(SPR_GROUP_DELETE_TRAIN
, STR_GROUP_DELETE_TOOLTIP
),
64 NWidget(WWT_PUSHIMGBTN
, COLOUR_GREY
, WID_GL_RENAME_GROUP
), SetFill(0, 1),
65 SetDataTip(SPR_GROUP_RENAME_TRAIN
, STR_GROUP_RENAME_TOOLTIP
),
66 NWidget(WWT_PANEL
, COLOUR_GREY
), SetFill(1, 1), EndContainer(),
67 NWidget(WWT_PUSHIMGBTN
, COLOUR_GREY
, WID_GL_REPLACE_PROTECTION
), SetFill(0, 1),
68 SetDataTip(SPR_GROUP_REPLACE_OFF_TRAIN
, STR_GROUP_REPLACE_PROTECTION_TOOLTIP
),
72 NWidget(NWID_VERTICAL
),
73 NWidget(NWID_HORIZONTAL
),
74 NWidget(WWT_PUSHTXTBTN
, COLOUR_GREY
, WID_GL_SORT_BY_ORDER
), SetMinimalSize(81, 12), SetDataTip(STR_BUTTON_SORT_BY
, STR_TOOLTIP_SORT_ORDER
),
75 NWidget(WWT_DROPDOWN
, COLOUR_GREY
, WID_GL_SORT_BY_DROPDOWN
), SetMinimalSize(167, 12), SetDataTip(0x0, STR_TOOLTIP_SORT_CRITERIA
),
76 NWidget(WWT_PANEL
, COLOUR_GREY
), SetMinimalSize(12, 12), SetResize(1, 0), EndContainer(),
78 NWidget(NWID_HORIZONTAL
),
79 NWidget(WWT_MATRIX
, COLOUR_GREY
, WID_GL_LIST_VEHICLE
), SetMinimalSize(248, 0), SetMatrixDataTip(1, 0, STR_NULL
), SetResize(1, 1), SetFill(1, 0), SetScrollbar(WID_GL_LIST_VEHICLE_SCROLLBAR
),
80 NWidget(NWID_VSCROLLBAR
, COLOUR_GREY
, WID_GL_LIST_VEHICLE_SCROLLBAR
),
82 NWidget(WWT_PANEL
, COLOUR_GREY
), SetMinimalSize(1, 0), SetFill(1, 1), SetResize(1, 0), EndContainer(),
83 NWidget(NWID_HORIZONTAL
),
84 NWidget(WWT_PUSHTXTBTN
, COLOUR_GREY
, WID_GL_AVAILABLE_VEHICLES
), SetMinimalSize(106, 12), SetFill(0, 1),
85 SetDataTip(STR_BLACK_STRING
, STR_VEHICLE_LIST_AVAILABLE_ENGINES_TOOLTIP
),
86 NWidget(WWT_PANEL
, COLOUR_GREY
), SetMinimalSize(0, 12), SetFill(1, 1), SetResize(1, 0), EndContainer(),
87 NWidget(WWT_DROPDOWN
, COLOUR_GREY
, WID_GL_MANAGE_VEHICLES_DROPDOWN
), SetMinimalSize(118, 12), SetFill(0, 1),
88 SetDataTip(STR_VEHICLE_LIST_MANAGE_LIST
, STR_VEHICLE_LIST_MANAGE_LIST_TOOLTIP
),
89 NWidget(WWT_PUSHIMGBTN
, COLOUR_GREY
, WID_GL_STOP_ALL
), SetMinimalSize(12, 12), SetFill(0, 1),
90 SetDataTip(SPR_FLAG_VEH_STOPPED
, STR_VEHICLE_LIST_MASS_STOP_LIST_TOOLTIP
),
91 NWidget(WWT_PUSHIMGBTN
, COLOUR_GREY
, WID_GL_START_ALL
), SetMinimalSize(12, 12), SetFill(0, 1),
92 SetDataTip(SPR_FLAG_VEH_RUNNING
, STR_VEHICLE_LIST_MASS_START_LIST_TOOLTIP
),
93 NWidget(WWT_RESIZEBOX
, COLOUR_GREY
),
99 class VehicleGroupWindow
: public BaseVehicleListWindow
{
101 /* Columns in the group list */
103 VGC_NAME
, ///< Group name.
104 VGC_PROTECT
, ///< Autoreplace protect icon.
105 VGC_AUTOREPLACE
, ///< Autoreplace active icon.
106 VGC_PROFIT
, ///< Profit icon.
107 VGC_NUMBER
, ///< Number of vehicles in the group.
112 VehicleID vehicle_sel
; ///< Selected vehicle
113 GroupID group_sel
; ///< Selected group (for drag/drop)
114 GroupID group_rename
; ///< Group being renamed, INVALID_GROUP if none
115 GroupID group_over
; ///< Group over which a vehicle is dragged, INVALID_GROUP if none
116 GroupID group_confirm
; ///< Group awaiting delete confirmation
117 GUIGroupList groups
; ///< List of groups
118 uint tiny_step_height
; ///< Step height for the group list
121 SmallVector
<int, 16> indents
; ///< Indentation levels
123 Dimension column_size
[VGC_END
]; ///< Size of the columns in the group list.
125 void AddParents(GUIGroupList
*source
, GroupID parent
, int indent
)
127 for (const Group
**g
= source
->Begin(); g
!= source
->End(); g
++) {
128 if ((*g
)->parent
== parent
) {
129 *this->groups
.Append() = *g
;
130 *this->indents
.Append() = indent
;
131 AddParents(source
, (*g
)->index
, indent
+ 1);
136 /** Sort the groups by their name */
137 static int CDECL
GroupNameSorter(const Group
* const *a
, const Group
* const *b
)
139 static const Group
*last_group
[2] = { NULL
, NULL
};
140 static char last_name
[2][64] = { "", "" };
142 if (*a
!= last_group
[0]) {
144 SetDParam(0, (*a
)->index
);
145 GetString(last_name
[0], STR_GROUP_NAME
, lastof(last_name
[0]));
148 if (*b
!= last_group
[1]) {
150 SetDParam(0, (*b
)->index
);
151 GetString(last_name
[1], STR_GROUP_NAME
, lastof(last_name
[1]));
154 int r
= strnatcmp(last_name
[0], last_name
[1]); // Sort by name (natural sorting).
155 if (r
== 0) return (*a
)->index
- (*b
)->index
;
160 * (Re)Build the group list.
162 * @param owner The owner of the window
164 void BuildGroupList(Owner owner
)
166 if (!this->groups
.NeedRebuild()) return;
168 this->groups
.Clear();
169 this->indents
.Clear();
175 if (g
->owner
== owner
&& g
->vehicle_type
== this->vli
.vtype
) {
181 list
.Sort(&GroupNameSorter
);
183 AddParents(&list
, INVALID_GROUP
, 0);
185 this->groups
.Compact();
186 this->groups
.RebuildDone();
190 * Compute tiny_step_height and column_size
191 * @return Total width required for the group list.
193 uint
ComputeGroupInfoSize()
195 this->column_size
[VGC_NAME
] = maxdim(GetStringBoundingBox(STR_GROUP_DEFAULT_TRAINS
+ this->vli
.vtype
), GetStringBoundingBox(STR_GROUP_ALL_TRAINS
+ this->vli
.vtype
));
196 this->column_size
[VGC_NAME
].width
= max(170u, this->column_size
[VGC_NAME
].width
);
197 this->tiny_step_height
= this->column_size
[VGC_NAME
].height
;
199 this->column_size
[VGC_PROTECT
] = GetSpriteSize(SPR_GROUP_REPLACE_PROTECT
);
200 this->tiny_step_height
= max(this->tiny_step_height
, this->column_size
[VGC_PROTECT
].height
);
202 this->column_size
[VGC_AUTOREPLACE
] = GetSpriteSize(SPR_GROUP_REPLACE_ACTIVE
);
203 this->tiny_step_height
= max(this->tiny_step_height
, this->column_size
[VGC_AUTOREPLACE
].height
);
205 this->column_size
[VGC_PROFIT
].width
= 0;
206 this->column_size
[VGC_PROFIT
].height
= 0;
207 static const SpriteID profit_sprites
[] = {SPR_PROFIT_NA
, SPR_PROFIT_NEGATIVE
, SPR_PROFIT_SOME
, SPR_PROFIT_LOT
};
208 for (uint i
= 0; i
< lengthof(profit_sprites
); i
++) {
209 Dimension d
= GetSpriteSize(profit_sprites
[i
]);
210 this->column_size
[VGC_PROFIT
] = maxdim(this->column_size
[VGC_PROFIT
], d
);
212 this->tiny_step_height
= max(this->tiny_step_height
, this->column_size
[VGC_PROFIT
].height
);
214 SetDParamMaxValue(0, GroupStatistics::Get(this->vli
.company
, ALL_GROUP
, this->vli
.vtype
).num_vehicle
, 3, FS_SMALL
);
215 this->column_size
[VGC_NUMBER
] = GetStringBoundingBox(STR_TINY_COMMA
);
216 this->tiny_step_height
= max(this->tiny_step_height
, this->column_size
[VGC_NUMBER
].height
);
218 this->tiny_step_height
+= WD_MATRIX_TOP
;
220 return WD_FRAMERECT_LEFT
+ 8 +
221 this->column_size
[VGC_NAME
].width
+ 8 +
222 this->column_size
[VGC_PROTECT
].width
+ 2 +
223 this->column_size
[VGC_AUTOREPLACE
].width
+ 2 +
224 this->column_size
[VGC_PROFIT
].width
+ 2 +
225 this->column_size
[VGC_NUMBER
].width
+ 2 +
230 * Draw a row in the group list.
231 * @param y Top of the row.
232 * @param left Left of the row.
233 * @param right Right of the row.
234 * @param g_id Group to list.
235 * @param indent Indentation level.
236 * @param protection Whether autoreplace protection is set.
238 void DrawGroupInfo(int y
, int left
, int right
, GroupID g_id
, int indent
= 0, bool protection
= false) const
240 /* Highlight the group if a vehicle is dragged over it */
241 if (g_id
== this->group_over
) {
242 GfxFillRect(left
+ WD_FRAMERECT_LEFT
, y
+ WD_FRAMERECT_TOP
, right
- WD_FRAMERECT_RIGHT
, y
+ this->tiny_step_height
- WD_FRAMERECT_BOTTOM
- WD_MATRIX_TOP
, _colour_gradient
[COLOUR_GREY
][7]);
245 if (g_id
== NEW_GROUP
) return;
247 /* draw the selected group in white, else we draw it in black */
248 TextColour colour
= g_id
== this->vli
.index
? TC_WHITE
: TC_BLACK
;
249 const GroupStatistics
&stats
= GroupStatistics::Get(this->vli
.company
, g_id
, this->vli
.vtype
);
250 bool rtl
= _current_text_dir
== TD_RTL
;
252 /* draw group name */
254 if (IsAllGroupID(g_id
)) {
255 str
= STR_GROUP_ALL_TRAINS
+ this->vli
.vtype
;
256 } else if (IsDefaultGroupID(g_id
)) {
257 str
= STR_GROUP_DEFAULT_TRAINS
+ this->vli
.vtype
;
260 str
= STR_GROUP_NAME
;
262 int x
= rtl
? right
- WD_FRAMERECT_RIGHT
- 8 - this->column_size
[VGC_NAME
].width
+ 1 : left
+ WD_FRAMERECT_LEFT
+ 8;
263 DrawString(x
+ indent
* LEVEL_WIDTH
, x
+ this->column_size
[VGC_NAME
].width
- 1, y
+ (this->tiny_step_height
- this->column_size
[VGC_NAME
].height
) / 2, str
, colour
);
265 /* draw autoreplace protection */
266 x
= rtl
? x
- 8 - this->column_size
[VGC_PROTECT
].width
: x
+ 8 + this->column_size
[VGC_NAME
].width
;
267 if (protection
) DrawSprite(SPR_GROUP_REPLACE_PROTECT
, PAL_NONE
, x
, y
+ (this->tiny_step_height
- this->column_size
[VGC_PROTECT
].height
) / 2);
269 /* draw autoreplace status */
270 x
= rtl
? x
- 2 - this->column_size
[VGC_AUTOREPLACE
].width
: x
+ 2 + this->column_size
[VGC_PROTECT
].width
;
271 if (stats
.autoreplace_defined
) DrawSprite(SPR_GROUP_REPLACE_ACTIVE
, stats
.autoreplace_finished
? PALETTE_CRASH
: PAL_NONE
, x
, y
+ (this->tiny_step_height
- this->column_size
[VGC_AUTOREPLACE
].height
) / 2);
273 /* draw the profit icon */
274 x
= rtl
? x
- 2 - this->column_size
[VGC_PROFIT
].width
: x
+ 2 + this->column_size
[VGC_AUTOREPLACE
].width
;
276 if (stats
.num_profit_vehicle
== 0) {
278 } else if (stats
.profit_last_year
< 0) {
279 spr
= SPR_PROFIT_NEGATIVE
;
280 } else if (stats
.profit_last_year
< 10000 * stats
.num_profit_vehicle
) { // TODO magic number
281 spr
= SPR_PROFIT_SOME
;
283 spr
= SPR_PROFIT_LOT
;
285 DrawSprite(spr
, PAL_NONE
, x
, y
+ (this->tiny_step_height
- this->column_size
[VGC_PROFIT
].height
) / 2);
287 /* draw the number of vehicles of the group */
288 x
= rtl
? x
- 2 - this->column_size
[VGC_NUMBER
].width
: x
+ 2 + this->column_size
[VGC_PROFIT
].width
;
289 SetDParam(0, stats
.num_vehicle
);
290 DrawString(x
, x
+ this->column_size
[VGC_NUMBER
].width
- 1, y
+ (this->tiny_step_height
- this->column_size
[VGC_NUMBER
].height
) / 2, STR_TINY_COMMA
, colour
, SA_RIGHT
| SA_FORCE
);
294 * Mark the widget containing the currently highlighted group as dirty.
296 void DirtyHighlightedGroupWidget()
298 if (this->group_over
== INVALID_GROUP
) return;
300 if (IsAllGroupID(this->group_over
)) {
301 this->SetWidgetDirty(WID_GL_ALL_VEHICLES
);
302 } else if (IsDefaultGroupID(this->group_over
)) {
303 this->SetWidgetDirty(WID_GL_DEFAULT_VEHICLES
);
305 this->SetWidgetDirty(WID_GL_LIST_GROUP
);
310 VehicleGroupWindow(WindowDesc
*desc
, WindowNumber window_number
) : BaseVehicleListWindow(desc
, window_number
)
312 this->CreateNestedTree();
314 this->vscroll
= this->GetScrollbar(WID_GL_LIST_VEHICLE_SCROLLBAR
);
315 this->group_sb
= this->GetScrollbar(WID_GL_LIST_GROUP_SCROLLBAR
);
317 switch (this->vli
.vtype
) {
318 default: NOT_REACHED();
319 case VEH_TRAIN
: this->sorting
= &_sorting
.train
; break;
320 case VEH_ROAD
: this->sorting
= &_sorting
.roadveh
; break;
321 case VEH_SHIP
: this->sorting
= &_sorting
.ship
; break;
322 case VEH_AIRCRAFT
: this->sorting
= &_sorting
.aircraft
; break;
325 this->vli
.index
= ALL_GROUP
;
326 this->vehicle_sel
= INVALID_VEHICLE
;
327 this->group_sel
= INVALID_GROUP
;
328 this->group_rename
= INVALID_GROUP
;
329 this->group_over
= INVALID_GROUP
;
331 this->vehicles
.SetListing(*this->sorting
);
332 this->vehicles
.ForceRebuild();
333 this->vehicles
.NeedResort();
335 this->BuildVehicleList();
336 this->SortVehicleList();
338 this->groups
.ForceRebuild();
339 this->groups
.NeedResort();
340 this->BuildGroupList(vli
.company
);
342 this->GetWidget
<NWidgetCore
>(WID_GL_CAPTION
)->widget_data
= STR_VEHICLE_LIST_TRAIN_CAPTION
+ this->vli
.vtype
;
343 this->GetWidget
<NWidgetCore
>(WID_GL_LIST_VEHICLE
)->tool_tip
= STR_VEHICLE_LIST_TRAIN_LIST_TOOLTIP
+ this->vli
.vtype
;
345 this->GetWidget
<NWidgetCore
>(WID_GL_CREATE_GROUP
)->widget_data
+= this->vli
.vtype
;
346 this->GetWidget
<NWidgetCore
>(WID_GL_RENAME_GROUP
)->widget_data
+= this->vli
.vtype
;
347 this->GetWidget
<NWidgetCore
>(WID_GL_DELETE_GROUP
)->widget_data
+= this->vli
.vtype
;
348 this->GetWidget
<NWidgetCore
>(WID_GL_REPLACE_PROTECTION
)->widget_data
+= this->vli
.vtype
;
350 this->FinishInitNested(window_number
);
351 this->owner
= vli
.company
;
354 ~VehicleGroupWindow()
356 *this->sorting
= this->vehicles
.GetListing();
359 virtual void UpdateWidgetSize(int widget
, Dimension
*size
, const Dimension
&padding
, Dimension
*fill
, Dimension
*resize
)
362 case WID_GL_LIST_GROUP
: {
363 size
->width
= this->ComputeGroupInfoSize();
364 resize
->height
= this->tiny_step_height
;
366 /* Minimum height is the height of the list widget minus all and default vehicles... */
367 size
->height
= 4 * GetVehicleListHeight(this->vli
.vtype
, this->tiny_step_height
) - 2 * this->tiny_step_height
;
369 /* ... minus the buttons at the bottom ... */
370 uint max_icon_height
= GetSpriteSize(this->GetWidget
<NWidgetCore
>(WID_GL_CREATE_GROUP
)->widget_data
).height
;
371 max_icon_height
= max(max_icon_height
, GetSpriteSize(this->GetWidget
<NWidgetCore
>(WID_GL_RENAME_GROUP
)->widget_data
).height
);
372 max_icon_height
= max(max_icon_height
, GetSpriteSize(this->GetWidget
<NWidgetCore
>(WID_GL_DELETE_GROUP
)->widget_data
).height
);
373 max_icon_height
= max(max_icon_height
, GetSpriteSize(this->GetWidget
<NWidgetCore
>(WID_GL_REPLACE_PROTECTION
)->widget_data
).height
);
375 /* ... minus the height of the group info ... */
376 max_icon_height
+= (FONT_HEIGHT_NORMAL
* 3) + WD_FRAMERECT_TOP
+ WD_FRAMERECT_BOTTOM
;
378 /* Get a multiple of tiny_step_height of that amount */
379 size
->height
= Ceil(size
->height
- max_icon_height
, tiny_step_height
);
383 case WID_GL_ALL_VEHICLES
:
384 case WID_GL_DEFAULT_VEHICLES
:
385 size
->width
= this->ComputeGroupInfoSize();
386 size
->height
= this->tiny_step_height
;
389 case WID_GL_SORT_BY_ORDER
: {
390 Dimension d
= GetStringBoundingBox(this->GetWidget
<NWidgetCore
>(widget
)->widget_data
);
391 d
.width
+= padding
.width
+ Window::SortButtonWidth() * 2; // Doubled since the string is centred and it also looks better.
392 d
.height
+= padding
.height
;
393 *size
= maxdim(*size
, d
);
397 case WID_GL_LIST_VEHICLE
:
398 this->ComputeGroupInfoSize();
399 resize
->height
= GetVehicleListHeight(this->vli
.vtype
, this->tiny_step_height
);
400 size
->height
= 4 * resize
->height
;
403 case WID_GL_MANAGE_VEHICLES_DROPDOWN
: {
404 Dimension d
= this->GetActionDropdownSize(true, true);
405 d
.height
+= padding
.height
;
406 d
.width
+= padding
.width
;
407 *size
= maxdim(*size
, d
);
412 size
->height
= (FONT_HEIGHT_NORMAL
* 3) + WD_FRAMERECT_TOP
+ WD_FRAMERECT_BOTTOM
;
419 * Some data on this window has become invalid.
420 * @param data Information about the changed data.
421 * @param gui_scope Whether the call is done from GUI scope. You may not do everything when not in GUI scope. See #InvalidateWindowData() for details.
423 virtual void OnInvalidateData(int data
= 0, bool gui_scope
= true)
426 /* This needs to be done in command-scope to enforce rebuilding before resorting invalid data */
427 this->vehicles
.ForceRebuild();
428 this->groups
.ForceRebuild();
430 this->vehicles
.ForceResort();
431 this->groups
.ForceResort();
434 /* Process ID-invalidation in command-scope as well */
435 if (this->group_rename
!= INVALID_GROUP
&& !Group::IsValidID(this->group_rename
)) {
436 DeleteWindowByClass(WC_QUERY_STRING
);
437 this->group_rename
= INVALID_GROUP
;
440 if (!(IsAllGroupID(this->vli
.index
) || IsDefaultGroupID(this->vli
.index
) || Group::IsValidID(this->vli
.index
))) {
441 this->vli
.index
= ALL_GROUP
;
442 HideDropDownMenu(this);
447 virtual void SetStringParameters(int widget
) const
450 case WID_GL_AVAILABLE_VEHICLES
:
451 SetDParam(0, STR_VEHICLE_LIST_AVAILABLE_TRAINS
+ this->vli
.vtype
);
455 /* If selected_group == DEFAULT_GROUP || ALL_GROUP, draw the standard caption
456 * We list all vehicles or ungrouped vehicles */
457 if (IsDefaultGroupID(this->vli
.index
) || IsAllGroupID(this->vli
.index
)) {
458 SetDParam(0, STR_COMPANY_NAME
);
459 SetDParam(1, this->vli
.company
);
460 SetDParam(2, this->vehicles
.Length());
461 SetDParam(3, this->vehicles
.Length());
463 const Group
*g
= Group::Get(this->vli
.index
);
465 SetDParam(0, STR_GROUP_NAME
);
466 SetDParam(1, g
->index
);
467 SetDParam(2, g
->statistics
.num_vehicle
);
468 SetDParam(3, g
->statistics
.num_vehicle
);
474 virtual void OnPaint()
476 /* If we select the all vehicles, this->list will contain all vehicles of the owner
477 * else this->list will contain all vehicles which belong to the selected group */
478 this->BuildVehicleList();
479 this->SortVehicleList();
481 this->BuildGroupList(this->owner
);
483 this->group_sb
->SetCount(this->groups
.Length());
484 this->vscroll
->SetCount(this->vehicles
.Length());
486 /* The drop down menu is out, *but* it may not be used, retract it. */
487 if (this->vehicles
.Length() == 0 && this->IsWidgetLowered(WID_GL_MANAGE_VEHICLES_DROPDOWN
)) {
488 this->RaiseWidget(WID_GL_MANAGE_VEHICLES_DROPDOWN
);
489 HideDropDownMenu(this);
492 /* Disable all lists management button when the list is empty */
493 this->SetWidgetsDisabledState(this->vehicles
.Length() == 0 || _local_company
!= this->vli
.company
,
496 WID_GL_MANAGE_VEHICLES_DROPDOWN
,
499 /* Disable the group specific function when we select the default group or all vehicles */
500 this->SetWidgetsDisabledState(IsDefaultGroupID(this->vli
.index
) || IsAllGroupID(this->vli
.index
) || _local_company
!= this->vli
.company
,
503 WID_GL_REPLACE_PROTECTION
,
506 /* Disable remaining buttons for non-local companies
507 * Needed while changing _local_company, eg. by cheats
508 * All procedures (eg. move vehicle to another group)
509 * verify, whether you are the owner of the vehicle,
510 * so it doesn't have to be disabled
512 this->SetWidgetsDisabledState(_local_company
!= this->vli
.company
,
514 WID_GL_AVAILABLE_VEHICLES
,
517 /* If not a default group and the group has replace protection, show an enabled replace sprite. */
518 uint16 protect_sprite
= SPR_GROUP_REPLACE_OFF_TRAIN
;
519 if (!IsDefaultGroupID(this->vli
.index
) && !IsAllGroupID(this->vli
.index
) && Group::Get(this->vli
.index
)->replace_protection
) protect_sprite
= SPR_GROUP_REPLACE_ON_TRAIN
;
520 this->GetWidget
<NWidgetCore
>(WID_GL_REPLACE_PROTECTION
)->widget_data
= protect_sprite
+ this->vli
.vtype
;
522 /* Set text of sort by dropdown */
523 this->GetWidget
<NWidgetCore
>(WID_GL_SORT_BY_DROPDOWN
)->widget_data
= this->vehicle_sorter_names
[this->vehicles
.SortType()];
528 virtual void DrawWidget(const Rect
&r
, int widget
) const
531 case WID_GL_ALL_VEHICLES
:
532 DrawGroupInfo(r
.top
+ WD_FRAMERECT_TOP
, r
.left
, r
.right
, ALL_GROUP
);
535 case WID_GL_DEFAULT_VEHICLES
:
536 DrawGroupInfo(r
.top
+ WD_FRAMERECT_TOP
, r
.left
, r
.right
, DEFAULT_GROUP
);
542 uint32 occupancy
= 0;
543 uint32 vehicle_count
= this->vehicles
.Length();
545 for (uint i
= 0; i
< vehicle_count
; i
++) {
546 const Vehicle
*v
= this->vehicles
[i
];
547 assert(v
->owner
== this->owner
);
549 this_year
+= v
->GetDisplayProfitThisYear();
550 last_year
+= v
->GetDisplayProfitLastYear();
551 occupancy
+= v
->trip_occupancy
;
554 const int left
= r
.left
+ WD_FRAMERECT_LEFT
+ 8;
555 const int right
= r
.right
- WD_FRAMERECT_RIGHT
- 8;
557 int y
= r
.top
+ WD_FRAMERECT_TOP
;
558 DrawString(left
, right
, y
, STR_GROUP_PROFIT_THIS_YEAR
, TC_BLACK
);
559 SetDParam(0, this_year
);
560 DrawString(left
, right
, y
, STR_JUST_CURRENCY_LONG
, TC_BLACK
, SA_RIGHT
);
562 y
+= FONT_HEIGHT_NORMAL
;
563 DrawString(left
, right
, y
, STR_GROUP_PROFIT_LAST_YEAR
, TC_BLACK
);
564 SetDParam(0, last_year
);
565 DrawString(left
, right
, y
, STR_JUST_CURRENCY_LONG
, TC_BLACK
, SA_RIGHT
);
567 y
+= FONT_HEIGHT_NORMAL
;
568 DrawString(left
, right
, y
, STR_GROUP_OCCUPANCY
, TC_BLACK
);
569 if (vehicle_count
> 0) {
570 SetDParam(0, occupancy
/ vehicle_count
);
571 DrawString(left
, right
, y
, STR_GROUP_OCCUPANCY_VALUE
, TC_BLACK
, SA_RIGHT
);
577 case WID_GL_LIST_GROUP
: {
578 int y1
= r
.top
+ WD_FRAMERECT_TOP
;
579 int max
= min(this->group_sb
->GetPosition() + this->group_sb
->GetCapacity(), this->groups
.Length());
580 for (int i
= this->group_sb
->GetPosition(); i
< max
; ++i
) {
581 const Group
*g
= this->groups
[i
];
583 assert(g
->owner
== this->owner
);
585 DrawGroupInfo(y1
, r
.left
, r
.right
, g
->index
, this->indents
[i
], g
->replace_protection
);
587 y1
+= this->tiny_step_height
;
589 if ((uint
)this->group_sb
->GetPosition() + this->group_sb
->GetCapacity() > this->groups
.Length()) {
590 DrawGroupInfo(y1
, r
.left
, r
.right
, NEW_GROUP
);
595 case WID_GL_SORT_BY_ORDER
:
596 this->DrawSortButtonState(WID_GL_SORT_BY_ORDER
, this->vehicles
.IsDescSortOrder() ? SBS_DOWN
: SBS_UP
);
599 case WID_GL_LIST_VEHICLE
:
600 if (this->vli
.index
!= ALL_GROUP
) {
601 /* Mark vehicles which are in sub-groups */
603 uint max
= min(this->vscroll
->GetPosition() + this->vscroll
->GetCapacity(), this->vehicles
.Length());
604 for (uint i
= this->vscroll
->GetPosition(); i
< max
; ++i
) {
605 const Vehicle
*v
= this->vehicles
[i
];
606 if (v
->group_id
!= this->vli
.index
) {
607 GfxFillRect(r
.left
+ 1, y
+ 1, r
.right
- 1, y
+ this->resize
.step_height
- 2, _colour_gradient
[COLOUR_GREY
][3], FILLRECT_CHECKER
);
609 y
+= this->resize
.step_height
;
613 this->DrawVehicleListItems(this->vehicle_sel
, this->resize
.step_height
, r
);
618 static void DeleteGroupCallback(Window
*win
, bool confirmed
)
621 VehicleGroupWindow
*w
= (VehicleGroupWindow
*)win
;
622 w
->vli
.index
= ALL_GROUP
;
623 DoCommandP(0, w
->group_confirm
, 0, CMD_DELETE_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_DELETE
));
627 virtual void OnClick(Point pt
, int widget
, int click_count
)
630 case WID_GL_SORT_BY_ORDER
: // Flip sorting method ascending/descending
631 this->vehicles
.ToggleSortOrder();
635 case WID_GL_SORT_BY_DROPDOWN
: // Select sorting criteria dropdown menu
636 ShowDropDownMenu(this, this->vehicle_sorter_names
, this->vehicles
.SortType(), WID_GL_SORT_BY_DROPDOWN
, 0, (this->vli
.vtype
== VEH_TRAIN
|| this->vli
.vtype
== VEH_ROAD
) ? 0 : (1 << 10));
639 case WID_GL_ALL_VEHICLES
: // All vehicles button
640 if (!IsAllGroupID(this->vli
.index
)) {
641 this->vli
.index
= ALL_GROUP
;
642 this->vehicles
.ForceRebuild();
647 case WID_GL_DEFAULT_VEHICLES
: // Ungrouped vehicles button
648 if (!IsDefaultGroupID(this->vli
.index
)) {
649 this->vli
.index
= DEFAULT_GROUP
;
650 this->vehicles
.ForceRebuild();
655 case WID_GL_LIST_GROUP
: { // Matrix Group
656 uint id_g
= this->group_sb
->GetScrolledRowFromWidget(pt
.y
, this, WID_GL_LIST_GROUP
, 0, this->tiny_step_height
);
657 if (id_g
>= this->groups
.Length()) return;
659 this->group_sel
= this->vli
.index
= this->groups
[id_g
]->index
;
661 SetObjectToPlaceWnd(SPR_CURSOR_MOUSE
, PAL_NONE
, HT_DRAG
, this);
663 this->vehicles
.ForceRebuild();
668 case WID_GL_LIST_VEHICLE
: { // Matrix Vehicle
669 uint id_v
= this->vscroll
->GetScrolledRowFromWidget(pt
.y
, this, WID_GL_LIST_VEHICLE
);
670 if (id_v
>= this->vehicles
.Length()) return; // click out of list bound
672 const Vehicle
*v
= this->vehicles
[id_v
];
673 if (VehicleClicked(v
)) break;
675 this->vehicle_sel
= v
->index
;
677 SetObjectToPlaceWnd(SPR_CURSOR_MOUSE
, PAL_NONE
, HT_DRAG
, this);
678 SetMouseCursorVehicle(v
, EIT_IN_LIST
);
679 _cursor
.vehchain
= true;
685 case WID_GL_CREATE_GROUP
: { // Create a new group
686 DoCommandP(0, this->vli
.vtype
, 0, CMD_CREATE_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_CREATE
), CcCreateGroup
);
690 case WID_GL_DELETE_GROUP
: { // Delete the selected group
691 this->group_confirm
= this->vli
.index
;
692 ShowQuery(STR_QUERY_GROUP_DELETE_CAPTION
, STR_GROUP_DELETE_QUERY_TEXT
, this, DeleteGroupCallback
);
696 case WID_GL_RENAME_GROUP
: // Rename the selected roup
697 this->ShowRenameGroupWindow(this->vli
.index
, false);
700 case WID_GL_AVAILABLE_VEHICLES
:
701 ShowBuildVehicleWindow(INVALID_TILE
, this->vli
.vtype
);
704 case WID_GL_MANAGE_VEHICLES_DROPDOWN
: {
705 DropDownList
*list
= this->BuildActionDropdownList(true, Group::IsValidID(this->vli
.index
));
706 ShowDropDownList(this, list
, 0, WID_GL_MANAGE_VEHICLES_DROPDOWN
);
710 case WID_GL_START_ALL
:
711 case WID_GL_STOP_ALL
: { // Start/stop all vehicles of the list
712 DoCommandP(0, (1 << 1) | (widget
== WID_GL_START_ALL
? (1 << 0) : 0), this->vli
.Pack(), CMD_MASS_START_STOP
);
716 case WID_GL_REPLACE_PROTECTION
: {
717 const Group
*g
= Group::GetIfValid(this->vli
.index
);
719 DoCommandP(0, this->vli
.index
, (g
->replace_protection
? 0 : 1) | (_ctrl_pressed
<< 1), CMD_SET_GROUP_REPLACE_PROTECTION
);
726 void OnDragDrop_Group(Point pt
, int widget
)
728 const Group
*g
= Group::Get(this->group_sel
);
731 case WID_GL_ALL_VEHICLES
: // All vehicles
732 case WID_GL_DEFAULT_VEHICLES
: // Ungroupd vehicles
733 if (g
->parent
!= INVALID_GROUP
) {
734 DoCommandP(0, this->group_sel
| (1 << 16), INVALID_GROUP
, CMD_ALTER_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_SET_PARENT
));
737 this->group_sel
= INVALID_GROUP
;
738 this->group_over
= INVALID_GROUP
;
742 case WID_GL_LIST_GROUP
: { // Matrix group
743 uint id_g
= this->group_sb
->GetScrolledRowFromWidget(pt
.y
, this, WID_GL_LIST_GROUP
, 0, this->tiny_step_height
);
744 GroupID new_g
= id_g
>= this->groups
.Length() ? INVALID_GROUP
: this->groups
[id_g
]->index
;
746 if (this->group_sel
!= new_g
&& g
->parent
!= new_g
) {
747 DoCommandP(0, this->group_sel
| (1 << 16), new_g
, CMD_ALTER_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_SET_PARENT
));
750 this->group_sel
= INVALID_GROUP
;
751 this->group_over
= INVALID_GROUP
;
758 void OnDragDrop_Vehicle(Point pt
, int widget
)
761 case WID_GL_DEFAULT_VEHICLES
: // Ungrouped vehicles
762 DoCommandP(0, DEFAULT_GROUP
, this->vehicle_sel
| (_ctrl_pressed
? 1 << 31 : 0), CMD_ADD_VEHICLE_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_ADD_VEHICLE
));
764 this->vehicle_sel
= INVALID_VEHICLE
;
765 this->group_over
= INVALID_GROUP
;
770 case WID_GL_LIST_GROUP
: { // Matrix group
771 const VehicleID vindex
= this->vehicle_sel
;
772 this->vehicle_sel
= INVALID_VEHICLE
;
773 this->group_over
= INVALID_GROUP
;
776 uint id_g
= this->group_sb
->GetScrolledRowFromWidget(pt
.y
, this, WID_GL_LIST_GROUP
, 0, this->tiny_step_height
);
777 GroupID new_g
= id_g
>= this->groups
.Length() ? NEW_GROUP
: this->groups
[id_g
]->index
;
779 DoCommandP(0, new_g
, vindex
| (_ctrl_pressed
? 1 << 31 : 0), CMD_ADD_VEHICLE_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_ADD_VEHICLE
), new_g
== NEW_GROUP
? CcAddVehicleNewGroup
: NULL
);
783 case WID_GL_LIST_VEHICLE
: { // Matrix vehicle
784 const VehicleID vindex
= this->vehicle_sel
;
785 this->vehicle_sel
= INVALID_VEHICLE
;
786 this->group_over
= INVALID_GROUP
;
789 uint id_v
= this->vscroll
->GetScrolledRowFromWidget(pt
.y
, this, WID_GL_LIST_VEHICLE
);
790 if (id_v
>= this->vehicles
.Length()) return; // click out of list bound
792 const Vehicle
*v
= this->vehicles
[id_v
];
793 if (!VehicleClicked(v
) && vindex
== v
->index
) {
794 ShowVehicleViewWindow(v
);
801 virtual void OnDragDrop(Point pt
, int widget
)
803 if (this->vehicle_sel
!= INVALID_VEHICLE
) OnDragDrop_Vehicle(pt
, widget
);
804 if (this->group_sel
!= INVALID_GROUP
) OnDragDrop_Group(pt
, widget
);
806 _cursor
.vehchain
= false;
809 virtual void OnQueryTextFinished(char *str
)
811 if (str
!= NULL
) DoCommandP(0, this->group_rename
, 0, CMD_ALTER_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_RENAME
), NULL
, str
);
812 this->group_rename
= INVALID_GROUP
;
815 virtual void OnResize()
817 this->group_sb
->SetCapacityFromWidget(this, WID_GL_LIST_GROUP
);
818 this->vscroll
->SetCapacityFromWidget(this, WID_GL_LIST_VEHICLE
);
821 virtual void OnDropdownSelect(int widget
, int index
)
824 case WID_GL_SORT_BY_DROPDOWN
:
825 this->vehicles
.SetSortType(index
);
828 case WID_GL_MANAGE_VEHICLES_DROPDOWN
:
829 assert(this->vehicles
.Length() != 0);
832 case ADI_REPLACE
: // Replace window
833 ShowReplaceGroupVehicleWindow(this->vli
.index
, this->vli
.vtype
);
835 case ADI_SERVICE
: // Send for servicing
836 case ADI_DEPOT
: { // Send to Depots
837 DoCommandP(0, DEPOT_MASS_SEND
| (index
== ADI_SERVICE
? DEPOT_SERVICE
: 0U), this->vli
.Pack(), GetCmdSendToDepot(this->vli
.vtype
));
841 case ADI_ADD_SHARED
: // Add shared Vehicles
842 assert(Group::IsValidID(this->vli
.index
));
844 DoCommandP(0, this->vli
.index
, this->vli
.vtype
, CMD_ADD_SHARED_VEHICLE_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_ADD_SHARED_VEHICLE
));
846 case ADI_REMOVE_ALL
: // Remove all Vehicles from the selected group
847 assert(Group::IsValidID(this->vli
.index
));
849 DoCommandP(0, this->vli
.index
, 0, CMD_REMOVE_ALL_VEHICLES_GROUP
| CMD_MSG(STR_ERROR_GROUP_CAN_T_REMOVE_ALL_VEHICLES
));
851 default: NOT_REACHED();
855 default: NOT_REACHED();
861 virtual void OnTick()
863 if (_pause_mode
!= PM_UNPAUSED
) return;
864 if (this->groups
.NeedResort() || this->vehicles
.NeedResort()) {
869 virtual void OnPlaceObjectAbort()
871 /* abort drag & drop */
872 this->vehicle_sel
= INVALID_VEHICLE
;
873 this->DirtyHighlightedGroupWidget();
874 this->group_over
= INVALID_GROUP
;
875 this->SetWidgetDirty(WID_GL_LIST_VEHICLE
);
878 virtual void OnMouseDrag(Point pt
, int widget
)
880 if (this->vehicle_sel
== INVALID_VEHICLE
&& this->group_sel
== INVALID_GROUP
) return;
882 /* A vehicle is dragged over... */
883 GroupID new_group_over
= INVALID_GROUP
;
885 case WID_GL_DEFAULT_VEHICLES
: // ... the 'default' group.
886 new_group_over
= DEFAULT_GROUP
;
889 case WID_GL_LIST_GROUP
: { // ... the list of custom groups.
890 uint id_g
= this->group_sb
->GetScrolledRowFromWidget(pt
.y
, this, WID_GL_LIST_GROUP
, 0, this->tiny_step_height
);
891 new_group_over
= id_g
>= this->groups
.Length() ? NEW_GROUP
: this->groups
[id_g
]->index
;
896 /* Do not highlight when dragging over the current group */
897 if (this->vehicle_sel
!= INVALID_VEHICLE
) {
898 if (Vehicle::Get(vehicle_sel
)->group_id
== new_group_over
) new_group_over
= INVALID_GROUP
;
899 } else if (this->group_sel
!= INVALID_GROUP
) {
900 if (this->group_sel
== new_group_over
|| Group::Get(this->group_sel
)->parent
== new_group_over
) new_group_over
= INVALID_GROUP
;
903 /* Mark widgets as dirty if the group changed. */
904 if (new_group_over
!= this->group_over
) {
905 this->DirtyHighlightedGroupWidget();
906 this->group_over
= new_group_over
;
907 this->DirtyHighlightedGroupWidget();
911 void ShowRenameGroupWindow(GroupID group
, bool empty
)
913 assert(Group::IsValidID(group
));
914 this->group_rename
= group
;
915 /* Show empty query for new groups */
916 StringID str
= STR_EMPTY
;
919 str
= STR_GROUP_NAME
;
921 ShowQueryString(str
, STR_GROUP_RENAME_CAPTION
, MAX_LENGTH_GROUP_NAME_CHARS
, this, CS_ALPHANUMERAL
, QSF_ENABLE_DEFAULT
| QSF_LEN_IN_CHARS
);
925 * Tests whether a given vehicle is selected in the window, and unselects it if necessary.
926 * Called when the vehicle is deleted.
927 * @param vehicle Vehicle that is going to be deleted
929 void UnselectVehicle(VehicleID vehicle
)
931 if (this->vehicle_sel
== vehicle
) ResetObjectToPlace();
936 static WindowDesc
_other_group_desc(
937 WDP_AUTO
, "list_groups", 460, 246,
940 _nested_group_widgets
, lengthof(_nested_group_widgets
)
943 static WindowDesc
_train_group_desc(
944 WDP_AUTO
, "list_groups_train", 525, 246,
945 WC_TRAINS_LIST
, WC_NONE
,
947 _nested_group_widgets
, lengthof(_nested_group_widgets
)
951 * Show the group window for the given company and vehicle type.
952 * @param company The company to show the window for.
953 * @param vehicle_type The type of vehicle to show it for.
955 void ShowCompanyGroup(CompanyID company
, VehicleType vehicle_type
)
957 if (!Company::IsValidID(company
)) return;
959 WindowNumber num
= VehicleListIdentifier(VL_GROUP_LIST
, vehicle_type
, company
).Pack();
960 if (vehicle_type
== VEH_TRAIN
) {
961 AllocateWindowDescFront
<VehicleGroupWindow
>(&_train_group_desc
, num
);
963 _other_group_desc
.cls
= GetWindowClassForVehicleType(vehicle_type
);
964 AllocateWindowDescFront
<VehicleGroupWindow
>(&_other_group_desc
, num
);
969 * Finds a group list window determined by vehicle type and owner
970 * @param vt vehicle type
971 * @param owner owner of groups
972 * @return pointer to VehicleGroupWindow, NULL if not found
974 static inline VehicleGroupWindow
*FindVehicleGroupWindow(VehicleType vt
, Owner owner
)
976 return (VehicleGroupWindow
*)FindWindowById(GetWindowClassForVehicleType(vt
), VehicleListIdentifier(VL_GROUP_LIST
, vt
, owner
).Pack());
980 * Opens a 'Rename group' window for newly created group
981 * @param success did command succeed?
983 * @param p1 vehicle type
985 * @see CmdCreateGroup
987 void CcCreateGroup(const CommandCost
&result
, TileIndex tile
, uint32 p1
, uint32 p2
)
989 if (result
.Failed()) return;
990 assert(p1
<= VEH_AIRCRAFT
);
992 VehicleGroupWindow
*w
= FindVehicleGroupWindow((VehicleType
)p1
, _current_company
);
993 if (w
!= NULL
) w
->ShowRenameGroupWindow(_new_group_id
, true);
997 * Open rename window after adding a vehicle to a new group via drag and drop.
998 * @param success Did command succeed?
999 * @param tile Unused.
1001 * @param p2 Bit 0-19: Vehicle ID.
1003 void CcAddVehicleNewGroup(const CommandCost
&result
, TileIndex tile
, uint32 p1
, uint32 p2
)
1005 if (result
.Failed()) return;
1006 assert(Vehicle::IsValidID(GB(p2
, 0, 20)));
1008 CcCreateGroup(result
, 0, Vehicle::Get(GB(p2
, 0, 20))->type
, 0);
1012 * Removes the highlight of a vehicle in a group window
1013 * @param *v Vehicle to remove all highlights from
1015 void DeleteGroupHighlightOfVehicle(const Vehicle
*v
)
1017 /* If we haven't got any vehicles on the mouse pointer, we haven't got any highlighted in any group windows either
1018 * If that is the case, we can skip looping though the windows and save time
1020 if (_special_mouse_mode
!= WSM_DRAGDROP
) return;
1022 VehicleGroupWindow
*w
= FindVehicleGroupWindow(v
->type
, v
->owner
);
1023 if (w
!= NULL
) w
->UnselectVehicle(v
->index
);