?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Translations update
[openttd/fttd.git] / src / company_gui.cpp
blob66afab3cb37718136335b57ceef8c97dd4d382af
1 /* $Id$ */
2
3 /*
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/>.
8 */
9
10 /** @file company_gui.cpp %Company related GUIs. */
11
12 #include "stdafx.h"
13 #include "error.h"
14 #include "gui.h"
15 #include "window_gui.h"
16 #include "textbuf_gui.h"
17 #include "viewport_func.h"
18 #include "company_func.h"
19 #include "command_func.h"
20 #include "network/network.h"
21 #include "network/network_gui.h"
22 #include "network/network_func.h"
23 #include "newgrf.h"
24 #include "company_manager_face.h"
25 #include "strings_func.h"
26 #include "date_func.h"
27 #include "widgets/dropdown_type.h"
28 #include "tilehighlight_func.h"
29 #include "company_base.h"
30 #include "core/geometry_func.hpp"
31 #include "object_type.h"
32 #include "rail.h"
33 #include "engine_base.h"
34 #include "window_func.h"
35 #include "road_func.h"
36 #include "water.h"
37 #include "station_func.h"
38 #include "zoom_func.h"
39
40 #include "widgets/company_widget.h"
41
42
43 /** Company GUI constants. */
44 static const uint EXP_LINESPACE = 2; ///< Amount of vertical space for a horizontal (sub-)total line.
45 static const uint EXP_BLOCKSPACE = 10; ///< Amount of vertical space between two blocks of numbers.
46
47 static void DoSelectCompanyManagerFace(Window *parent);
48 static void ShowCompanyInfrastructure(CompanyID company);
49
50 /** Standard unsorted list of expenses. */
51 static const ExpensesType _expenses_list_1[] = {
52 EXPENSES_CONSTRUCTION,
53 EXPENSES_NEW_VEHICLES,
54 EXPENSES_TRAIN_RUN,
55 EXPENSES_ROADVEH_RUN,
56 EXPENSES_AIRCRAFT_RUN,
57 EXPENSES_SHIP_RUN,
58 EXPENSES_PROPERTY,
59 EXPENSES_TRAIN_INC,
60 EXPENSES_ROADVEH_INC,
61 EXPENSES_AIRCRAFT_INC,
62 EXPENSES_SHIP_INC,
63 EXPENSES_LOAN_INT,
64 EXPENSES_OTHER,
65 };
66
67 /** Grouped list of expenses. */
68 static const ExpensesType _expenses_list_2[] = {
69 EXPENSES_TRAIN_INC,
70 EXPENSES_ROADVEH_INC,
71 EXPENSES_AIRCRAFT_INC,
72 EXPENSES_SHIP_INC,
73 INVALID_EXPENSES,
74 EXPENSES_TRAIN_RUN,
75 EXPENSES_ROADVEH_RUN,
76 EXPENSES_AIRCRAFT_RUN,
77 EXPENSES_SHIP_RUN,
78 EXPENSES_PROPERTY,
79 EXPENSES_LOAN_INT,
80 INVALID_EXPENSES,
81 EXPENSES_CONSTRUCTION,
82 EXPENSES_NEW_VEHICLES,
83 EXPENSES_OTHER,
84 INVALID_EXPENSES,
85 };
86
87 /** Expense list container. */
88 struct ExpensesList {
89 const ExpensesType *const et; ///< Expenses items.
90 const uint length; ///< Number of items in list.
91 const uint num_subtotals; ///< Number of sub-totals in the list.
92
93 template <uint N>
94 ExpensesList (const ExpensesType (&et) [N], uint num_subtotals)
95 : et(et), length(N), num_subtotals(num_subtotals)
96 {
97 }
98
99 uint GetHeight() const
100 {
101 /* heading + line + texts of expenses + sub-totals + total line + total text */
102 return FONT_HEIGHT_NORMAL + EXP_LINESPACE + this->length * FONT_HEIGHT_NORMAL + num_subtotals * (EXP_BLOCKSPACE + EXP_LINESPACE) + EXP_LINESPACE + FONT_HEIGHT_NORMAL;
103 }
104
105 /** Compute width of the expenses categories in pixels. */
106 uint GetCategoriesWidth() const
107 {
108 uint width = 0;
109 bool invalid_expenses_measured = false; // Measure 'Total' width only once.
110 for (uint i = 0; i < this->length; i++) {
111 ExpensesType et = this->et[i];
112 if (et != INVALID_EXPENSES) {
113 width = max(width, GetStringBoundingBox(STR_FINANCES_SECTION_CONSTRUCTION + et).width);
114 } else if (!invalid_expenses_measured) {
115 invalid_expenses_measured = true;
116 width = max(width, GetStringBoundingBox(STR_FINANCES_TOTAL_CAPTION).width);
117 }
118 }
119 return width;
120 }
121 };
122
123 static const ExpensesList _expenses_list_types[] = {
124 ExpensesList (_expenses_list_1, 0),
125 ExpensesList (_expenses_list_2, 3),
126 };
127
128 /**
129 * Draw the expenses categories.
130 * @param r Available space for drawing.
131 * @note The environment must provide padding at the left and right of \a r.
132 */
133 static void DrawCategories(const Rect &r)
134 {
135 int y = r.top;
136
137 DrawString(r.left, r.right, y, STR_FINANCES_EXPENDITURE_INCOME_TITLE, TC_FROMSTRING, SA_HOR_CENTER, true);
138 y += FONT_HEIGHT_NORMAL + EXP_LINESPACE;
139
140 int type = _settings_client.gui.expenses_layout;
141 for (uint i = 0; i < _expenses_list_types[type].length; i++) {
142 const ExpensesType et = _expenses_list_types[type].et[i];
143 if (et == INVALID_EXPENSES) {
144 y += EXP_LINESPACE;
145 DrawString(r.left, r.right, y, STR_FINANCES_TOTAL_CAPTION, TC_FROMSTRING, SA_RIGHT);
146 y += FONT_HEIGHT_NORMAL + EXP_BLOCKSPACE;
147 } else {
148 DrawString(r.left, r.right, y, STR_FINANCES_SECTION_CONSTRUCTION + et);
149 y += FONT_HEIGHT_NORMAL;
150 }
151 }
152
153 DrawString(r.left, r.right, y + EXP_LINESPACE, STR_FINANCES_TOTAL_CAPTION, TC_FROMSTRING, SA_RIGHT);
154 }
155
156 /**
157 * Draw an amount of money.
158 * @param amount Amount of money to draw,
159 * @param left Left coordinate of the space to draw in.
160 * @param right Right coordinate of the space to draw in.
161 * @param top Top coordinate of the space to draw in.
162 */
163 static void DrawPrice(Money amount, int left, int right, int top)
164 {
165 StringID str = STR_FINANCES_NEGATIVE_INCOME;
166 if (amount < 0) {
167 amount = -amount;
168 str++;
169 }
170 SetDParam(0, amount);
171 DrawString(left, right, top, str, TC_FROMSTRING, SA_RIGHT);
172 }
173
174 /**
175 * Draw a column with prices.
176 * @param r Available space for drawing.
177 * @param year Year being drawn.
178 * @param tbl Pointer to table of amounts for \a year.
179 * @note The environment must provide padding at the left and right of \a r.
180 */
181 static void DrawYearColumn(const Rect &r, int year, const Money (*tbl)[EXPENSES_END])
182 {
183 int y = r.top;
184
185 SetDParam(0, year);
186 DrawString(r.left, r.right, y, STR_FINANCES_YEAR, TC_FROMSTRING, SA_RIGHT, true);
187 y += FONT_HEIGHT_NORMAL + EXP_LINESPACE;
188
189 Money sum = 0;
190 Money subtotal = 0;
191 int type = _settings_client.gui.expenses_layout;
192 for (uint i = 0; i < _expenses_list_types[type].length; i++) {
193 const ExpensesType et = _expenses_list_types[type].et[i];
194 if (et == INVALID_EXPENSES) {
195 Money cost = subtotal;
196 subtotal = 0;
197 GfxFillRect(r.left, y, r.right, y, PC_BLACK);
198 y += EXP_LINESPACE;
199 DrawPrice(cost, r.left, r.right, y);
200 y += FONT_HEIGHT_NORMAL + EXP_BLOCKSPACE;
201 } else {
202 Money cost = (*tbl)[et];
203 subtotal += cost;
204 sum += cost;
205 if (cost != 0) DrawPrice(cost, r.left, r.right, y);
206 y += FONT_HEIGHT_NORMAL;
207 }
208 }
209
210 GfxFillRect(r.left, y, r.right, y, PC_BLACK);
211 y += EXP_LINESPACE;
212 DrawPrice(sum, r.left, r.right, y);
213 }
214
215 static const NWidgetPart _nested_company_finances_widgets[] = {
216 NWidget(NWID_HORIZONTAL),
217 NWidget(WWT_CLOSEBOX, COLOUR_GREY),
218 NWidget(WWT_CAPTION, COLOUR_GREY, WID_CF_CAPTION), SetDataTip(STR_FINANCES_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
219 NWidget(WWT_IMGBTN, COLOUR_GREY, WID_CF_TOGGLE_SIZE), SetDataTip(SPR_LARGE_SMALL_WINDOW, STR_TOOLTIP_TOGGLE_LARGE_SMALL_WINDOW),
220 NWidget(WWT_SHADEBOX, COLOUR_GREY),
221 NWidget(WWT_STICKYBOX, COLOUR_GREY),
222 EndContainer(),
223 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_PANEL),
224 NWidget(WWT_PANEL, COLOUR_GREY),
225 NWidget(NWID_HORIZONTAL), SetPadding(WD_FRAMERECT_TOP, WD_FRAMERECT_RIGHT, WD_FRAMERECT_BOTTOM, WD_FRAMERECT_LEFT), SetPIP(0, 9, 0),
226 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_EXPS_CATEGORY), SetMinimalSize(120, 0), SetFill(0, 0),
227 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_EXPS_PRICE1), SetMinimalSize(86, 0), SetFill(0, 0),
228 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_EXPS_PRICE2), SetMinimalSize(86, 0), SetFill(0, 0),
229 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_EXPS_PRICE3), SetMinimalSize(86, 0), SetFill(0, 0),
230 EndContainer(),
231 EndContainer(),
232 EndContainer(),
233 NWidget(WWT_PANEL, COLOUR_GREY),
234 NWidget(NWID_HORIZONTAL), SetPadding(WD_FRAMERECT_TOP, WD_FRAMERECT_RIGHT, WD_FRAMERECT_BOTTOM, WD_FRAMERECT_LEFT),
235 NWidget(NWID_VERTICAL), // Vertical column with 'bank balance', 'loan'
236 NWidget(WWT_TEXT, COLOUR_GREY), SetDataTip(STR_FINANCES_BANK_BALANCE_TITLE, STR_NULL), SetFill(1, 0),
237 NWidget(WWT_TEXT, COLOUR_GREY), SetDataTip(STR_FINANCES_LOAN_TITLE, STR_NULL), SetFill(1, 0),
238 NWidget(NWID_SPACER), SetFill(0, 1),
239 EndContainer(),
240 NWidget(NWID_SPACER), SetFill(0, 0), SetMinimalSize(30, 0),
241 NWidget(NWID_VERTICAL), // Vertical column with bank balance amount, loan amount, and total.
242 NWidget(WWT_TEXT, COLOUR_GREY, WID_CF_BALANCE_VALUE), SetDataTip(STR_NULL, STR_NULL),
243 NWidget(WWT_TEXT, COLOUR_GREY, WID_CF_LOAN_VALUE), SetDataTip(STR_NULL, STR_NULL),
244 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_LOAN_LINE), SetMinimalSize(0, 2), SetFill(1, 0),
245 NWidget(WWT_TEXT, COLOUR_GREY, WID_CF_TOTAL_VALUE), SetDataTip(STR_NULL, STR_NULL),
246 EndContainer(),
247 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_MAXLOAN),
248 NWidget(NWID_HORIZONTAL),
249 NWidget(NWID_SPACER), SetFill(0, 1), SetMinimalSize(25, 0),
250 NWidget(NWID_VERTICAL), // Max loan information
251 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_MAXLOAN_GAP), SetFill(0, 0),
252 NWidget(WWT_TEXT, COLOUR_GREY, WID_CF_MAXLOAN_VALUE), SetDataTip(STR_FINANCES_MAX_LOAN, STR_NULL),
253 NWidget(NWID_SPACER), SetFill(0, 1),
254 EndContainer(),
255 EndContainer(),
256 EndContainer(),
257 NWidget(NWID_SPACER), SetFill(1, 1),
258 EndContainer(),
259 EndContainer(),
260 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_BUTTONS),
261 NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
262 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_INCREASE_LOAN), SetFill(1, 0), SetDataTip(STR_FINANCES_BORROW_BUTTON, STR_FINANCES_BORROW_TOOLTIP),
263 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_REPAY_LOAN), SetFill(1, 0), SetDataTip(STR_FINANCES_REPAY_BUTTON, STR_FINANCES_REPAY_TOOLTIP),
264 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_INFRASTRUCTURE), SetFill(1, 0), SetDataTip(STR_FINANCES_INFRASTRUCTURE_BUTTON, STR_COMPANY_VIEW_INFRASTRUCTURE_TOOLTIP),
265 EndContainer(),
266 EndContainer(),
267 };
268
269 /**
270 * Window class displaying the company finances.
271 * @todo #money_width should be calculated dynamically.
272 */
273 struct CompanyFinancesWindow : Window {
274 static Money max_money; ///< The maximum amount of money a company has had this 'run'
275 bool small; ///< Window is toggled to 'small'.
276
277 CompanyFinancesWindow(WindowDesc *desc, CompanyID company) : Window(desc)
278 {
279 this->small = false;
280 this->CreateNestedTree();
281 this->SetupWidgets();
282 this->FinishInitNested(company);
283
284 this->owner = (Owner)this->window_number;
285 }
286
287 virtual void SetStringParameters(int widget) const
288 {
289 switch (widget) {
290 case WID_CF_CAPTION:
291 SetDParam(0, (CompanyID)this->window_number);
292 SetDParam(1, (CompanyID)this->window_number);
293 break;
294
295 case WID_CF_MAXLOAN_VALUE:
296 SetDParam(0, _economy.max_loan);
297 break;
298
299 case WID_CF_INCREASE_LOAN:
300 case WID_CF_REPAY_LOAN:
301 SetDParam(0, LOAN_INTERVAL);
302 break;
303 }
304 }
305
306 virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
307 {
308 int type = _settings_client.gui.expenses_layout;
309 switch (widget) {
310 case WID_CF_EXPS_CATEGORY:
311 size->width = _expenses_list_types[type].GetCategoriesWidth();
312 size->height = _expenses_list_types[type].GetHeight();
313 break;
314
315 case WID_CF_EXPS_PRICE1:
316 case WID_CF_EXPS_PRICE2:
317 case WID_CF_EXPS_PRICE3:
318 size->height = _expenses_list_types[type].GetHeight();
319 /* FALL THROUGH */
320 case WID_CF_BALANCE_VALUE:
321 case WID_CF_LOAN_VALUE:
322 case WID_CF_TOTAL_VALUE:
323 SetDParamMaxValue(0, CompanyFinancesWindow::max_money);
324 size->width = max(GetStringBoundingBox(STR_FINANCES_NEGATIVE_INCOME).width, GetStringBoundingBox(STR_FINANCES_POSITIVE_INCOME).width) + padding.width;
325 break;
326
327 case WID_CF_MAXLOAN_GAP:
328 size->height = FONT_HEIGHT_NORMAL;
329 break;
330 }
331 }
332
333 virtual void DrawWidget(const Rect &r, int widget) const
334 {
335 switch (widget) {
336 case WID_CF_EXPS_CATEGORY:
337 DrawCategories(r);
338 break;
339
340 case WID_CF_EXPS_PRICE1:
341 case WID_CF_EXPS_PRICE2:
342 case WID_CF_EXPS_PRICE3: {
343 const Company *c = Company::Get((CompanyID)this->window_number);
344 int age = min(_cur_year - c->inaugurated_year, 2);
345 int wid_offset = widget - WID_CF_EXPS_PRICE1;
346 if (wid_offset <= age) {
347 DrawYearColumn(r, _cur_year - (age - wid_offset), c->yearly_expenses + (age - wid_offset));
348 }
349 break;
350 }
351
352 case WID_CF_BALANCE_VALUE: {
353 const Company *c = Company::Get((CompanyID)this->window_number);
354 SetDParam(0, c->money);
355 DrawString(r.left, r.right, r.top, STR_FINANCES_TOTAL_CURRENCY, TC_FROMSTRING, SA_RIGHT);
356 break;
357 }
358
359 case WID_CF_LOAN_VALUE: {
360 const Company *c = Company::Get((CompanyID)this->window_number);
361 SetDParam(0, c->current_loan);
362 DrawString(r.left, r.right, r.top, STR_FINANCES_TOTAL_CURRENCY, TC_FROMSTRING, SA_RIGHT);
363 break;
364 }
365
366 case WID_CF_TOTAL_VALUE: {
367 const Company *c = Company::Get((CompanyID)this->window_number);
368 SetDParam(0, c->money - c->current_loan);
369 DrawString(r.left, r.right, r.top, STR_FINANCES_TOTAL_CURRENCY, TC_FROMSTRING, SA_RIGHT);
370 break;
371 }
372
373 case WID_CF_LOAN_LINE:
374 GfxFillRect(r.left, r.top, r.right, r.top, PC_BLACK);
375 break;
376 }
377 }
378
379 /**
380 * Setup the widgets in the nested tree, such that the finances window is displayed properly.
381 * @note After setup, the window must be (re-)initialized.
382 */
383 void SetupWidgets()
384 {
385 int plane = this->small ? SZSP_NONE : 0;
386 this->GetWidget<NWidgetStacked>(WID_CF_SEL_PANEL)->SetDisplayedPlane(plane);
387 this->GetWidget<NWidgetStacked>(WID_CF_SEL_MAXLOAN)->SetDisplayedPlane(plane);
388
389 CompanyID company = (CompanyID)this->window_number;
390 plane = (company != _local_company) ? SZSP_NONE : 0;
391 this->GetWidget<NWidgetStacked>(WID_CF_SEL_BUTTONS)->SetDisplayedPlane(plane);
392 }
393
394 virtual void OnPaint()
395 {
396 if (!this->IsShaded()) {
397 if (!this->small) {
398 /* Check that the expenses panel height matches the height needed for the layout. */
399 int type = _settings_client.gui.expenses_layout;
400 if (_expenses_list_types[type].GetHeight() != this->GetWidget<NWidgetBase>(WID_CF_EXPS_CATEGORY)->current_y) {
401 this->SetupWidgets();
402 this->ReInit();
403 return;
404 }
405 }
406
407 /* Check that the loan buttons are shown only when the user owns the company. */
408 CompanyID company = (CompanyID)this->window_number;
409 int req_plane = (company != _local_company) ? SZSP_NONE : 0;
410 if (req_plane != this->GetWidget<NWidgetStacked>(WID_CF_SEL_BUTTONS)->shown_plane) {
411 this->SetupWidgets();
412 this->ReInit();
413 return;
414 }
415
416 const Company *c = Company::Get(company);
417 this->SetWidgetDisabledState(WID_CF_INCREASE_LOAN, c->current_loan == _economy.max_loan); // Borrow button only shows when there is any more money to loan.
418 this->SetWidgetDisabledState(WID_CF_REPAY_LOAN, company != _local_company || c->current_loan == 0); // Repay button only shows when there is any more money to repay.
419 }
420
421 this->DrawWidgets();
422 }
423
424 virtual void OnClick(Point pt, int widget, int click_count)
425 {
426 switch (widget) {
427 case WID_CF_TOGGLE_SIZE: // toggle size
428 this->small = !this->small;
429 this->SetupWidgets();
430 if (this->IsShaded()) {
431 /* Finances window is not resizable, so size hints given during unshading have no effect
432 * on the changed appearance of the window. */
433 this->SetShaded(false);
434 } else {
435 this->ReInit();
436 }
437 break;
438
439 case WID_CF_INCREASE_LOAN: // increase loan
440 DoCommandP(0, 0, _ctrl_pressed, CMD_INCREASE_LOAN);
441 break;
442
443 case WID_CF_REPAY_LOAN: // repay loan
444 DoCommandP(0, 0, _ctrl_pressed, CMD_DECREASE_LOAN);
445 break;
446
447 case WID_CF_INFRASTRUCTURE: // show infrastructure details
448 ShowCompanyInfrastructure((CompanyID)this->window_number);
449 break;
450 }
451 }
452
453 virtual void OnHundredthTick()
454 {
455 const Company *c = Company::Get((CompanyID)this->window_number);
456 if (c->money > CompanyFinancesWindow::max_money) {
457 CompanyFinancesWindow::max_money = max(c->money * 2, CompanyFinancesWindow::max_money * 4);
458 this->SetupWidgets();
459 this->ReInit();
460 }
461 }
462 };
463
464 /** First conservative estimate of the maximum amount of money */
465 Money CompanyFinancesWindow::max_money = INT32_MAX;
466
467 static WindowDesc _company_finances_desc(
468 WDP_AUTO, "company_finances", 0, 0,
469 WC_FINANCES, WC_NONE,
470 0,
471 _nested_company_finances_widgets, lengthof(_nested_company_finances_widgets)
472 );
473
474 /**
475 * Open the finances window of a company.
476 * @param company Company to show finances of.
477 * @pre is company a valid company.
478 */
479 void ShowCompanyFinances(CompanyID company)
480 {
481 if (!Company::IsValidID(company)) return;
482 if (BringWindowToFrontById(WC_FINANCES, company)) return;
483
484 new CompanyFinancesWindow(&_company_finances_desc, company);
485 }
486
487 /* List of colours for the livery window */
488 static const StringID _colour_dropdown[] = {
489 STR_COLOUR_DARK_BLUE,
490 STR_COLOUR_PALE_GREEN,
491 STR_COLOUR_PINK,
492 STR_COLOUR_YELLOW,
493 STR_COLOUR_RED,
494 STR_COLOUR_LIGHT_BLUE,
495 STR_COLOUR_GREEN,
496 STR_COLOUR_DARK_GREEN,
497 STR_COLOUR_BLUE,
498 STR_COLOUR_CREAM,
499 STR_COLOUR_MAUVE,
500 STR_COLOUR_PURPLE,
501 STR_COLOUR_ORANGE,
502 STR_COLOUR_BROWN,
503 STR_COLOUR_GREY,
504 STR_COLOUR_WHITE,
505 };
506
507 /* Association of liveries to livery classes */
508 static const LiveryClass _livery_class[LS_END] = {
509 LC_OTHER,
510 LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL, LC_RAIL,
511 LC_ROAD, LC_ROAD,
512 LC_SHIP, LC_SHIP,
513 LC_AIRCRAFT, LC_AIRCRAFT, LC_AIRCRAFT,
514 LC_ROAD, LC_ROAD,
515 };
516
517 class DropDownListColourItem : public DropDownListItem {
518 public:
519 DropDownListColourItem(int result, bool masked) : DropDownListItem(result, masked) {}
520
521 virtual ~DropDownListColourItem() {}
522
523 StringID String() const
524 {
525 return _colour_dropdown[this->result];
526 }
527
528 uint Height(uint width) const
529 {
530 return max(FONT_HEIGHT_NORMAL, ScaleGUITrad(12) + 2);
531 }
532
533 bool Selectable() const
534 {
535 return true;
536 }
537
538 void Draw(int left, int right, int top, int bottom, bool sel, int bg_colour) const
539 {
540 bool rtl = _current_text_dir == TD_RTL;
541 int height = bottom - top;
542 int icon_y_offset = height / 2;
543 int text_y_offset = (height - FONT_HEIGHT_NORMAL) / 2 + 1;
544 DrawSprite(SPR_VEH_BUS_SIDE_VIEW, PALETTE_RECOLOUR_START + this->result,
545 rtl ? right - 2 - ScaleGUITrad(14) : left + ScaleGUITrad(14) + 2,
546 top + icon_y_offset);
547 DrawString(rtl ? left + 2 : left + ScaleGUITrad(28) + 4,
548 rtl ? right - ScaleGUITrad(28) - 4 : right - 2,
549 top + text_y_offset, this->String(), sel ? TC_WHITE : TC_BLACK);
550 }
551 };
552
553 /** Company livery colour scheme window. */
554 struct SelectCompanyLiveryWindow : public Window {
555 private:
556 uint32 sel;
557 LiveryClass livery_class;
558 Dimension square;
559 Dimension box;
560 uint line_height;
561
562 /** Get the first selected livery if any, else LS_DEFAULT. */
563 LiveryScheme GetSelectionFirst (void) const
564 {
565 assert_compile (LS_BEGIN == (LiveryScheme)0);
566
567 if (this->sel == 0) return LS_DEFAULT;
568
569 LiveryScheme scheme = LS_BEGIN;
570 while (!HasBit (this->sel, scheme)) {
571 scheme++;
572 assert (scheme < LS_END);
573 }
574
575 return scheme;
576 }
577
578 void ShowColourDropDownMenu(uint32 widget)
579 {
580 uint32 used_colours = 0;
581
582 /* Disallow other company colours for the primary colour */
583 if (HasBit(this->sel, LS_DEFAULT) && widget == WID_SCL_PRI_COL_DROPDOWN) {
584 const Company *c;
585 FOR_ALL_COMPANIES(c) {
586 if (c->index != _local_company) SetBit(used_colours, c->colour);
587 }
588 }
589
590 /* Get the first selected livery to use as the default dropdown item */
591 LiveryScheme scheme = GetSelectionFirst();
592 const Livery *livery = &Company::Get((CompanyID)this->window_number)->livery[scheme];
593
594 DropDownList *list = new DropDownList();
595 for (uint i = 0; i < lengthof(_colour_dropdown); i++) {
596 *list->Append() = new DropDownListColourItem(i, HasBit(used_colours, i));
597 }
598
599 ShowDropDownList(this, list, widget == WID_SCL_PRI_COL_DROPDOWN ? livery->colour1 : livery->colour2, widget);
600 }
601
602 public:
603 SelectCompanyLiveryWindow(WindowDesc *desc, CompanyID company) : Window(desc)
604 {
605 this->livery_class = LC_OTHER;
606 this->sel = 1;
607
608 this->square = GetSpriteSize(SPR_SQUARE);
609 this->box = maxdim(GetSpriteSize(SPR_BOX_CHECKED), GetSpriteSize(SPR_BOX_EMPTY));
610 this->line_height = max(max(this->square.height, this->box.height), (uint)FONT_HEIGHT_NORMAL) + 4;
611
612 this->InitNested(company);
613 this->owner = company;
614 this->LowerWidget(WID_SCL_CLASS_GENERAL);
615 this->InvalidateData(1);
616 }
617
618 virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
619 {
620 switch (widget) {
621 case WID_SCL_SPACER_DROPDOWN: {
622 /* The matrix widget below needs enough room to print all the schemes. */
623 Dimension d = {0, 0};
624 for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
625 d = maxdim(d, GetStringBoundingBox(STR_LIVERY_DEFAULT + scheme));
626 }
627 size->width = max(size->width, 5 + this->box.width + d.width + WD_FRAMERECT_RIGHT);
628 break;
629 }
630
631 case WID_SCL_MATRIX: {
632 uint livery_height = 0;
633 for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
634 if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
635 livery_height++;
636 }
637 }
638 size->height = livery_height * this->line_height;
639 this->GetWidget<NWidgetCore>(WID_SCL_MATRIX)->widget_data = (livery_height << MAT_ROW_START) | (1 << MAT_COL_START);
640 break;
641 }
642
643 case WID_SCL_SEC_COL_DROPDOWN:
644 if (!_loaded_newgrf_features.has_2CC) {
645 size->width = 0;
646 break;
647 }
648 /* FALL THROUGH */
649 case WID_SCL_PRI_COL_DROPDOWN: {
650 int padding = this->square.width + NWidgetScrollbar::GetVerticalDimension().width + 10;
651 for (const StringID *id = _colour_dropdown; id != endof(_colour_dropdown); id++) {
652 size->width = max(size->width, GetStringBoundingBox(*id).width + padding);
653 }
654 break;
655 }
656 }
657 }
658
659 virtual void OnPaint()
660 {
661 /* Disable dropdown controls if no scheme is selected */
662 this->SetWidgetDisabledState(WID_SCL_PRI_COL_DROPDOWN, this->sel == 0);
663 this->SetWidgetDisabledState(WID_SCL_SEC_COL_DROPDOWN, this->sel == 0);
664
665 this->DrawWidgets();
666 }
667
668 virtual void SetStringParameters(int widget) const
669 {
670 switch (widget) {
671 case WID_SCL_PRI_COL_DROPDOWN:
672 case WID_SCL_SEC_COL_DROPDOWN: {
673 const Company *c = Company::Get((CompanyID)this->window_number);
674 LiveryScheme scheme = GetSelectionFirst();
675 SetDParam(0, STR_COLOUR_DARK_BLUE + ((widget == WID_SCL_PRI_COL_DROPDOWN) ? c->livery[scheme].colour1 : c->livery[scheme].colour2));
676 break;
677 }
678 }
679 }
680
681 virtual void DrawWidget(const Rect &r, int widget) const
682 {
683 if (widget != WID_SCL_MATRIX) return;
684
685 bool rtl = _current_text_dir == TD_RTL;
686
687 /* Horizontal coordinates of scheme name column. */
688 const NWidgetBase *nwi = this->GetWidget<NWidgetBase>(WID_SCL_SPACER_DROPDOWN);
689 int sch_left = nwi->pos_x;
690 int sch_right = sch_left + nwi->current_x - 1;
691 /* Horizontal coordinates of first dropdown. */
692 nwi = this->GetWidget<NWidgetBase>(WID_SCL_PRI_COL_DROPDOWN);
693 int pri_left = nwi->pos_x;
694 int pri_right = pri_left + nwi->current_x - 1;
695 /* Horizontal coordinates of second dropdown. */
696 nwi = this->GetWidget<NWidgetBase>(WID_SCL_SEC_COL_DROPDOWN);
697 int sec_left = nwi->pos_x;
698 int sec_right = sec_left + nwi->current_x - 1;
699
700 int text_left = (rtl ? (uint)WD_FRAMERECT_LEFT : (this->box.width + 5));
701 int text_right = (rtl ? (this->box.width + 5) : (uint)WD_FRAMERECT_RIGHT);
702
703 int box_offs = (this->line_height - this->box.height) / 2;
704 int square_offs = (this->line_height - this->square.height) / 2 + 1;
705 int text_offs = (this->line_height - FONT_HEIGHT_NORMAL) / 2 + 1;
706
707 int y = r.top;
708 const Company *c = Company::Get((CompanyID)this->window_number);
709 for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
710 if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
711 bool sel = HasBit(this->sel, scheme) != 0;
712
713 /* Optional check box + scheme name. */
714 if (scheme != LS_DEFAULT) {
715 DrawSprite(c->livery[scheme].in_use ? SPR_BOX_CHECKED : SPR_BOX_EMPTY, PAL_NONE, (rtl ? sch_right - (this->box.width + 5) + WD_FRAMERECT_RIGHT : sch_left) + WD_FRAMERECT_LEFT, y + box_offs);
716 }
717 DrawString(sch_left + text_left, sch_right - text_right, y + text_offs, STR_LIVERY_DEFAULT + scheme, sel ? TC_WHITE : TC_BLACK);
718
719 /* Text below the first dropdown. */
720 DrawSprite(SPR_SQUARE, GENERAL_SPRITE_COLOUR(c->livery[scheme].colour1), (rtl ? pri_right - (this->box.width + 5) + WD_FRAMERECT_RIGHT : pri_left) + WD_FRAMERECT_LEFT, y + square_offs);
721 DrawString(pri_left + text_left, pri_right - text_right, y + text_offs, STR_COLOUR_DARK_BLUE + c->livery[scheme].colour1, sel ? TC_WHITE : TC_GOLD);
722
723 /* Text below the second dropdown. */
724 if (sec_right > sec_left) { // Second dropdown has non-zero size.
725 DrawSprite(SPR_SQUARE, GENERAL_SPRITE_COLOUR(c->livery[scheme].colour2), (rtl ? sec_right - (this->box.width + 5) + WD_FRAMERECT_RIGHT : sec_left) + WD_FRAMERECT_LEFT, y + square_offs);
726 DrawString(sec_left + text_left, sec_right - text_right, y + text_offs, STR_COLOUR_DARK_BLUE + c->livery[scheme].colour2, sel ? TC_WHITE : TC_GOLD);
727 }
728
729 y += this->line_height;
730 }
731 }
732 }
733
734 virtual void OnClick(Point pt, int widget, int click_count)
735 {
736 switch (widget) {
737 /* Livery Class buttons */
738 case WID_SCL_CLASS_GENERAL:
739 case WID_SCL_CLASS_RAIL:
740 case WID_SCL_CLASS_ROAD:
741 case WID_SCL_CLASS_SHIP:
742 case WID_SCL_CLASS_AIRCRAFT:
743 this->RaiseWidget(this->livery_class + WID_SCL_CLASS_GENERAL);
744 this->livery_class = (LiveryClass)(widget - WID_SCL_CLASS_GENERAL);
745 this->LowerWidget(this->livery_class + WID_SCL_CLASS_GENERAL);
746
747 /* Select the first item in the list */
748 this->sel = 0;
749 for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
750 if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
751 this->sel = 1 << scheme;
752 break;
753 }
754 }
755
756 this->ReInit();
757 break;
758
759 case WID_SCL_PRI_COL_DROPDOWN: // First colour dropdown
760 ShowColourDropDownMenu(WID_SCL_PRI_COL_DROPDOWN);
761 break;
762
763 case WID_SCL_SEC_COL_DROPDOWN: // Second colour dropdown
764 ShowColourDropDownMenu(WID_SCL_SEC_COL_DROPDOWN);
765 break;
766
767 case WID_SCL_MATRIX: {
768 const NWidgetBase *wid = this->GetWidget<NWidgetBase>(WID_SCL_MATRIX);
769 LiveryScheme j = (LiveryScheme)((pt.y - wid->pos_y) / this->line_height);
770
771 for (LiveryScheme scheme = LS_BEGIN; scheme <= j; scheme++) {
772 if (_livery_class[scheme] != this->livery_class || !HasBit(_loaded_newgrf_features.used_liveries, scheme)) j++;
773 if (scheme >= LS_END) return;
774 }
775 if (j >= LS_END) return;
776
777 /* If clicking on the left edge, toggle using the livery */
778 if (_current_text_dir == TD_RTL ? pt.x - wid->pos_x > wid->current_x - (this->box.width + 5) : pt.x - wid->pos_x < (this->box.width + 5)) {
779 DoCommandP(0, j | (2 << 8), !Company::Get((CompanyID)this->window_number)->livery[j].in_use, CMD_SET_COMPANY_COLOUR);
780 }
781
782 if (_ctrl_pressed) {
783 ToggleBit(this->sel, j);
784 } else {
785 this->sel = 1 << j;
786 }
787 this->SetDirty();
788 break;
789 }
790 }
791 }
792
793 virtual void OnDropdownSelect(int widget, int index)
794 {
795 for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
796 /* Changed colour for the selected scheme, or all visible schemes if CTRL is pressed. */
797 if (HasBit(this->sel, scheme) || (_ctrl_pressed && _livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme))) {
798 DoCommandP(0, scheme | (widget == WID_SCL_PRI_COL_DROPDOWN ? 0 : 256), index, CMD_SET_COMPANY_COLOUR);
799 }
800 }
801 }
802
803 /**
804 * Some data on this window has become invalid.
805 * @param data Information about the changed data.
806 * @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.
807 */
808 virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
809 {
810 if (!gui_scope) return;
811 this->SetWidgetsDisabledState(true, WID_SCL_CLASS_RAIL, WID_SCL_CLASS_ROAD, WID_SCL_CLASS_SHIP, WID_SCL_CLASS_AIRCRAFT, WIDGET_LIST_END);
812
813 bool current_class_valid = this->livery_class == LC_OTHER;
814 if (_settings_client.gui.liveries == LIT_ALL || (_settings_client.gui.liveries == LIT_COMPANY && this->window_number == _local_company)) {
815 for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
816 if (HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
817 if (_livery_class[scheme] == this->livery_class) current_class_valid = true;
818 this->EnableWidget(WID_SCL_CLASS_GENERAL + _livery_class[scheme]);
819 } else {
820 ClrBit(this->sel, scheme);
821 }
822 }
823 }
824
825 if (!current_class_valid) {
826 Point pt = {0, 0};
827 this->OnClick(pt, WID_SCL_CLASS_GENERAL, 1);
828 } else if (data == 0) {
829 this->ReInit();
830 }
831 }
832 };
833
834 static const NWidgetPart _nested_select_company_livery_widgets [] = {
835 NWidget(NWID_HORIZONTAL),
836 NWidget(WWT_CLOSEBOX, COLOUR_GREY),
837 NWidget(WWT_CAPTION, COLOUR_GREY, WID_SCL_CAPTION), SetDataTip(STR_LIVERY_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
838 EndContainer(),
839 NWidget(NWID_HORIZONTAL),
840 NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_GENERAL), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_COMPANY_GENERAL, STR_LIVERY_GENERAL_TOOLTIP),
841 NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_RAIL), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_TRAINLIST, STR_LIVERY_TRAIN_TOOLTIP),
842 NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_ROAD), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_TRUCKLIST, STR_LIVERY_ROAD_VEHICLE_TOOLTIP),
843 NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_SHIP), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_SHIPLIST, STR_LIVERY_SHIP_TOOLTIP),
844 NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_AIRCRAFT), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_AIRPLANESLIST, STR_LIVERY_AIRCRAFT_TOOLTIP),
845 NWidget(WWT_PANEL, COLOUR_GREY), SetMinimalSize(90, 22), SetFill(1, 1), EndContainer(),
846 EndContainer(),
847 NWidget(NWID_HORIZONTAL),
848 NWidget(WWT_PANEL, COLOUR_GREY, WID_SCL_SPACER_DROPDOWN), SetMinimalSize(150, 12), SetFill(1, 1), EndContainer(),
849 NWidget(WWT_DROPDOWN, COLOUR_GREY, WID_SCL_PRI_COL_DROPDOWN), SetMinimalSize(125, 12), SetFill(0, 1), SetDataTip(STR_BLACK_STRING, STR_LIVERY_PRIMARY_TOOLTIP),
850 NWidget(WWT_DROPDOWN, COLOUR_GREY, WID_SCL_SEC_COL_DROPDOWN), SetMinimalSize(125, 12), SetFill(0, 1),
851 SetDataTip(STR_BLACK_STRING, STR_LIVERY_SECONDARY_TOOLTIP),
852 EndContainer(),
853 NWidget(WWT_MATRIX, COLOUR_GREY, WID_SCL_MATRIX), SetMinimalSize(275, 15), SetFill(1, 0), SetMatrixDataTip(1, 1, STR_LIVERY_PANEL_TOOLTIP),
854 };
855
856 static WindowDesc _select_company_livery_desc(
857 WDP_AUTO, "company_livery", 0, 0,
858 WC_COMPANY_COLOUR, WC_NONE,
859 0,
860 _nested_select_company_livery_widgets, lengthof(_nested_select_company_livery_widgets)
861 );
862
863 /**
864 * Draws the face of a company manager's face.
865 * @param cmf the company manager's face
866 * @param colour the (background) colour of the gradient
867 * @param x x-position to draw the face
868 * @param y y-position to draw the face
869 */
870 void DrawCompanyManagerFace(CompanyManagerFace cmf, int colour, int x, int y)
871 {
872 /* Draw the gradient (background) */
873 DrawSprite(SPR_GRADIENT, GENERAL_SPRITE_COLOUR(colour), x, y);
874
875 GenderEthnicity ge = (GenderEthnicity)GetCompanyManagerFaceBits(cmf, CMFV_GEN_ETHN, GE_WM);
876
877 bool has_moustache = !HasBit(ge, GENDER_FEMALE) && GetCompanyManagerFaceBits(cmf, CMFV_HAS_MOUSTACHE, ge) != 0;
878 bool has_tie_earring = !HasBit(ge, GENDER_FEMALE) || GetCompanyManagerFaceBits(cmf, CMFV_HAS_TIE_EARRING, ge) != 0;
879 bool has_glasses = GetCompanyManagerFaceBits(cmf, CMFV_HAS_GLASSES, ge) != 0;
880 PaletteID pal;
881
882 /* Modify eye colour palette only if 2 or more valid values exist */
883 if (_cmf_info[CMFV_EYE_COLOUR].valid_values[ge] < 2) {
884 pal = PAL_NONE;
885 } else {
886 switch (GetCompanyManagerFaceBits(cmf, CMFV_EYE_COLOUR, ge)) {
887 default: NOT_REACHED();
888 case 0: pal = PALETTE_TO_BROWN; break;
889 case 1: pal = PALETTE_TO_BLUE; break;
890 case 2: pal = PALETTE_TO_GREEN; break;
891 }
892 }
893
894 for (CompanyManagerFaceVariable cmfv = CMFV_CHEEKS; cmfv < CMFV_END; cmfv++) {
895 switch (cmfv) {
896 case CMFV_MOUSTACHE: if (!has_moustache) continue; break;
897 case CMFV_LIPS: // FALL THROUGH
898 case CMFV_NOSE: if (has_moustache) continue; break;
899 case CMFV_TIE_EARRING: if (!has_tie_earring) continue; break;
900 case CMFV_GLASSES: if (!has_glasses) continue; break;
901 default: break;
902 }
903 DrawSprite(GetCompanyManagerFaceSprite(cmf, cmfv, ge), (cmfv == CMFV_EYEBROWS) ? pal : PAL_NONE, x, y);
904 }
905 }
906
907 /** Nested widget description for the company manager face selection dialog */
908 static const NWidgetPart _nested_select_company_manager_face_widgets[] = {
909 NWidget(NWID_HORIZONTAL),
910 NWidget(WWT_CLOSEBOX, COLOUR_GREY),
911 NWidget(WWT_CAPTION, COLOUR_GREY, WID_SCMF_CAPTION), SetDataTip(STR_FACE_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
912 NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCMF_TOGGLE_LARGE_SMALL), SetDataTip(SPR_LARGE_SMALL_WINDOW, STR_FACE_ADVANCED_TOOLTIP),
913 EndContainer(),
914 NWidget(WWT_PANEL, COLOUR_GREY, WID_SCMF_SELECT_FACE),
915 NWidget(NWID_SPACER), SetMinimalSize(0, 2),
916 NWidget(NWID_HORIZONTAL), SetPIP(2, 2, 2),
917 NWidget(NWID_VERTICAL),
918 NWidget(NWID_HORIZONTAL),
919 NWidget(NWID_SPACER), SetFill(1, 0),
920 NWidget(WWT_EMPTY, COLOUR_GREY, WID_SCMF_FACE), SetMinimalSize(92, 119),
921 NWidget(NWID_SPACER), SetFill(1, 0),
922 EndContainer(),
923 NWidget(NWID_SPACER), SetMinimalSize(0, 2),
924 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_RANDOM_NEW_FACE), SetFill(1, 0), SetDataTip(STR_FACE_NEW_FACE_BUTTON, STR_FACE_NEW_FACE_TOOLTIP),
925 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_LOADSAVE), // Load/number/save buttons under the portrait in the advanced view.
926 NWidget(NWID_VERTICAL),
927 NWidget(NWID_SPACER), SetMinimalSize(0, 5), SetFill(0, 1),
928 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_LOAD), SetFill(1, 0), SetDataTip(STR_FACE_LOAD, STR_FACE_LOAD_TOOLTIP),
929 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_FACECODE), SetFill(1, 0), SetDataTip(STR_FACE_FACECODE, STR_FACE_FACECODE_TOOLTIP),
930 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_SAVE), SetFill(1, 0), SetDataTip(STR_FACE_SAVE, STR_FACE_SAVE_TOOLTIP),
931 NWidget(NWID_SPACER), SetMinimalSize(0, 5), SetFill(0, 1),
932 EndContainer(),
933 EndContainer(),
934 EndContainer(),
935 NWidget(NWID_VERTICAL),
936 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON), SetFill(1, 0), SetDataTip(STR_FACE_ADVANCED, STR_FACE_ADVANCED_TOOLTIP),
937 NWidget(NWID_SPACER), SetMinimalSize(0, 2),
938 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_MALEFEMALE), // Simple male/female face setting.
939 NWidget(NWID_VERTICAL),
940 NWidget(NWID_SPACER), SetFill(0, 1),
941 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_MALE), SetFill(1, 0), SetDataTip(STR_FACE_MALE_BUTTON, STR_FACE_MALE_TOOLTIP),
942 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_FEMALE), SetFill(1, 0), SetDataTip(STR_FACE_FEMALE_BUTTON, STR_FACE_FEMALE_TOOLTIP),
943 NWidget(NWID_SPACER), SetFill(0, 1),
944 EndContainer(),
945 EndContainer(),
946 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_PARTS), // Advanced face parts setting.
947 NWidget(NWID_VERTICAL),
948 NWidget(NWID_SPACER), SetMinimalSize(0, 2),
949 NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
950 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_MALE2), SetFill(1, 0), SetDataTip(STR_FACE_MALE_BUTTON, STR_FACE_MALE_TOOLTIP),
951 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_FEMALE2), SetFill(1, 0), SetDataTip(STR_FACE_FEMALE_BUTTON, STR_FACE_FEMALE_TOOLTIP),
952 EndContainer(),
953 NWidget(NWID_SPACER), SetMinimalSize(0, 2),
954 NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
955 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_ETHNICITY_EUR), SetFill(1, 0), SetDataTip(STR_FACE_EUROPEAN, STR_FACE_SELECT_EUROPEAN),
956 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_ETHNICITY_AFR), SetFill(1, 0), SetDataTip(STR_FACE_AFRICAN, STR_FACE_SELECT_AFRICAN),
957 EndContainer(),
958 NWidget(NWID_SPACER), SetMinimalSize(0, 4),
959 NWidget(NWID_HORIZONTAL),
960 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT), SetFill(1, 0),
961 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAS_MOUSTACHE_EARRING), SetDataTip(STR_EMPTY, STR_FACE_MOUSTACHE_EARRING_TOOLTIP),
962 EndContainer(),
963 NWidget(NWID_HORIZONTAL),
964 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_HAS_GLASSES_TEXT), SetFill(1, 0),
965 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAS_GLASSES), SetDataTip(STR_EMPTY, STR_FACE_GLASSES_TOOLTIP),
966 EndContainer(),
967 NWidget(NWID_SPACER), SetMinimalSize(0, 2), SetFill(1, 0),
968 NWidget(NWID_HORIZONTAL),
969 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_HAIR_TEXT), SetFill(1, 0),
970 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_HAIR_L), SetDataTip(AWV_DECREASE, STR_FACE_HAIR_TOOLTIP),
971 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAIR), SetDataTip(STR_EMPTY, STR_FACE_HAIR_TOOLTIP),
972 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_HAIR_R), SetDataTip(AWV_INCREASE, STR_FACE_HAIR_TOOLTIP),
973 EndContainer(),
974 NWidget(NWID_HORIZONTAL),
975 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_EYEBROWS_TEXT), SetFill(1, 0),
976 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYEBROWS_L), SetDataTip(AWV_DECREASE, STR_FACE_EYEBROWS_TOOLTIP),
977 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_EYEBROWS), SetDataTip(STR_EMPTY, STR_FACE_EYEBROWS_TOOLTIP),
978 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYEBROWS_R), SetDataTip(AWV_INCREASE, STR_FACE_EYEBROWS_TOOLTIP),
979 EndContainer(),
980 NWidget(NWID_HORIZONTAL),
981 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_EYECOLOUR_TEXT), SetFill(1, 0),
982 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR_L), SetDataTip(AWV_DECREASE, STR_FACE_EYECOLOUR_TOOLTIP),
983 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR), SetDataTip(STR_EMPTY, STR_FACE_EYECOLOUR_TOOLTIP),
984 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR_R), SetDataTip(AWV_INCREASE, STR_FACE_EYECOLOUR_TOOLTIP),
985 EndContainer(),
986 NWidget(NWID_HORIZONTAL),
987 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_GLASSES_TEXT), SetFill(1, 0),
988 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_GLASSES_L), SetDataTip(AWV_DECREASE, STR_FACE_GLASSES_TOOLTIP_2),
989 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_GLASSES), SetDataTip(STR_EMPTY, STR_FACE_GLASSES_TOOLTIP_2),
990 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_GLASSES_R), SetDataTip(AWV_INCREASE, STR_FACE_GLASSES_TOOLTIP_2),
991 EndContainer(),
992 NWidget(NWID_HORIZONTAL),
993 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_NOSE_TEXT), SetFill(1, 0),
994 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_NOSE_L), SetDataTip(AWV_DECREASE, STR_FACE_NOSE_TOOLTIP),
995 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_NOSE), SetDataTip(STR_EMPTY, STR_FACE_NOSE_TOOLTIP),
996 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_NOSE_R), SetDataTip(AWV_INCREASE, STR_FACE_NOSE_TOOLTIP),
997 EndContainer(),
998 NWidget(NWID_HORIZONTAL),
999 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_LIPS_MOUSTACHE_TEXT), SetFill(1, 0),
1000 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE_L), SetDataTip(AWV_DECREASE, STR_FACE_LIPS_MOUSTACHE_TOOLTIP),
1001 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE), SetDataTip(STR_EMPTY, STR_FACE_LIPS_MOUSTACHE_TOOLTIP),
1002 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE_R), SetDataTip(AWV_INCREASE, STR_FACE_LIPS_MOUSTACHE_TOOLTIP),
1003 EndContainer(),
1004 NWidget(NWID_HORIZONTAL),
1005 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_CHIN_TEXT), SetFill(1, 0),
1006 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_CHIN_L), SetDataTip(AWV_DECREASE, STR_FACE_CHIN_TOOLTIP),
1007 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_CHIN), SetDataTip(STR_EMPTY, STR_FACE_CHIN_TOOLTIP),
1008 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_CHIN_R), SetDataTip(AWV_INCREASE, STR_FACE_CHIN_TOOLTIP),
1009 EndContainer(),
1010 NWidget(NWID_HORIZONTAL),
1011 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_JACKET_TEXT), SetFill(1, 0),
1012 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_JACKET_L), SetDataTip(AWV_DECREASE, STR_FACE_JACKET_TOOLTIP),
1013 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_JACKET), SetDataTip(STR_EMPTY, STR_FACE_JACKET_TOOLTIP),
1014 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_JACKET_R), SetDataTip(AWV_INCREASE, STR_FACE_JACKET_TOOLTIP),
1015 EndContainer(),
1016 NWidget(NWID_HORIZONTAL),
1017 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_COLLAR_TEXT), SetFill(1, 0),
1018 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_COLLAR_L), SetDataTip(AWV_DECREASE, STR_FACE_COLLAR_TOOLTIP),
1019 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_COLLAR), SetDataTip(STR_EMPTY, STR_FACE_COLLAR_TOOLTIP),
1020 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_COLLAR_R), SetDataTip(AWV_INCREASE, STR_FACE_COLLAR_TOOLTIP),
1021 EndContainer(),
1022 NWidget(NWID_HORIZONTAL),
1023 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_TIE_EARRING_TEXT), SetFill(1, 0),
1024 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING_L), SetDataTip(AWV_DECREASE, STR_FACE_TIE_EARRING_TOOLTIP),
1025 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING), SetDataTip(STR_EMPTY, STR_FACE_TIE_EARRING_TOOLTIP),
1026 NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING_R), SetDataTip(AWV_INCREASE, STR_FACE_TIE_EARRING_TOOLTIP),
1027 EndContainer(),
1028 NWidget(NWID_SPACER), SetFill(0, 1),
1029 EndContainer(),
1030 EndContainer(),
1031 EndContainer(),
1032 EndContainer(),
1033 NWidget(NWID_SPACER), SetMinimalSize(0, 2),
1034 EndContainer(),
1035 NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
1036 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_CANCEL), SetFill(1, 0), SetDataTip(STR_BUTTON_CANCEL, STR_FACE_CANCEL_TOOLTIP),
1037 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_ACCEPT), SetFill(1, 0), SetDataTip(STR_BUTTON_OK, STR_FACE_OK_TOOLTIP),
1038 EndContainer(),
1039 };
1040
1041 /** Textual names for parts of the face. */
1042 static const StringID PART_TEXTS[] = {
1043 STR_FACE_GLASSES, // WID_SCMF_HAS_GLASSES_TEXT
1044 STR_FACE_HAIR, // WID_SCMF_HAIR_TEXT
1045 STR_FACE_EYEBROWS, // WID_SCMF_EYEBROWS_TEXT
1046 STR_FACE_EYECOLOUR, // WID_SCMF_EYECOLOUR_TEXT
1047 STR_FACE_GLASSES, // WID_SCMF_GLASSES_TEXT
1048 STR_FACE_NOSE, // WID_SCMF_NOSE_TEXT
1049 STR_FACE_CHIN, // WID_SCMF_CHIN_TEXT
1050 STR_FACE_JACKET, // WID_SCMF_JACKET_TEXT
1051 STR_FACE_COLLAR, // WID_SCMF_COLLAR_TEXT
1052 };
1053
1054 /** Management class for customizing the face of the company manager. */
1055 class SelectCompanyManagerFaceWindow : public Window
1056 {
1057 CompanyManagerFace face; ///< company manager face bits
1058 bool advanced; ///< advanced company manager face selection window
1059
1060 GenderEthnicity ge; ///< Gender and ethnicity.
1061 bool is_female; ///< Female face.
1062 bool is_moust_male; ///< Male face with a moustache.
1063
1064 Dimension yesno_dim; ///< Dimension of a yes/no button of a part in the advanced face window.
1065 Dimension number_dim; ///< Dimension of a number widget of a part in the advanced face window.
1066
1067 /** Get the company manager's face bits for the given company manager's face variable. */
1068 uint GetFaceBits (CompanyManagerFaceVariable cmfv) const
1069 {
1070 return GetCompanyManagerFaceBits (this->face, cmfv, this->ge);
1071 }
1072
1073 /**
1074 * Draw dynamic a label to the left of the button and a value in the button
1075 *
1076 * @param widget_index index of this widget in the window
1077 * @param cmfv the variable that will be drawn
1078 * @param is_bool_widget is it a bool button
1079 */
1080 void DrawFaceStringLabel (byte widget_index, CompanyManagerFaceVariable cmfv, bool is_bool_widget) const
1081 {
1082 const NWidgetCore *nwi_widget = this->GetWidget<NWidgetCore>(widget_index);
1083 if (!nwi_widget->IsDisabled()) {
1084 uint val = this->GetFaceBits (cmfv);
1085 StringID str;
1086 if (is_bool_widget) {
1087 /* if it a bool button write yes or no */
1088 str = (val != 0) ? STR_FACE_YES : STR_FACE_NO;
1089 } else {
1090 /* else write the value + 1 */
1091 SetDParam(0, val + 1);
1092 str = STR_JUST_INT;
1093 }
1094
1095 /* Draw the value/bool in white. */
1096 DrawString (nwi_widget->pos_x, nwi_widget->pos_x + nwi_widget->current_x - 1,
1097 nwi_widget->pos_y + 1, str, TC_WHITE, SA_HOR_CENTER);
1098 }
1099 }
1100
1101 void UpdateData()
1102 {
1103 this->ge = (GenderEthnicity)GB(this->face, _cmf_info[CMFV_GEN_ETHN].offset, _cmf_info[CMFV_GEN_ETHN].length); // get the gender and ethnicity
1104 this->is_female = HasBit(this->ge, GENDER_FEMALE); // get the gender: 0 == male and 1 == female
1105 this->is_moust_male = !is_female && this->GetFaceBits(CMFV_HAS_MOUSTACHE) != 0; // is a male face with moustache
1106 }
1107
1108 public:
1109 SelectCompanyManagerFaceWindow(WindowDesc *desc, Window *parent) : Window(desc)
1110 {
1111 this->advanced = false;
1112 this->CreateNestedTree();
1113 this->SelectDisplayPlanes(this->advanced);
1114 this->FinishInitNested(parent->window_number);
1115 this->parent = parent;
1116 this->owner = (Owner)this->window_number;
1117 this->face = Company::Get((CompanyID)this->window_number)->face;
1118
1119 this->UpdateData();
1120 }
1121
1122 /**
1123 * Select planes to display to the user with the #NWID_SELECTION widgets #WID_SCMF_SEL_LOADSAVE, #WID_SCMF_SEL_MALEFEMALE, and #WID_SCMF_SEL_PARTS.
1124 * @param advanced Display advanced face management window.
1125 */
1126 void SelectDisplayPlanes(bool advanced)
1127 {
1128 this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_LOADSAVE)->SetDisplayedPlane(advanced ? 0 : SZSP_NONE);
1129 this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_PARTS)->SetDisplayedPlane(advanced ? 0 : SZSP_NONE);
1130 this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_MALEFEMALE)->SetDisplayedPlane(advanced ? SZSP_NONE : 0);
1131 this->GetWidget<NWidgetCore>(WID_SCMF_RANDOM_NEW_FACE)->widget_data = advanced ? STR_FACE_RANDOM : STR_FACE_NEW_FACE_BUTTON;
1132
1133 NWidgetCore *wi = this->GetWidget<NWidgetCore>(WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON);
1134 if (advanced) {
1135 wi->SetDataTip(STR_FACE_SIMPLE, STR_FACE_SIMPLE_TOOLTIP);
1136 } else {
1137 wi->SetDataTip(STR_FACE_ADVANCED, STR_FACE_ADVANCED_TOOLTIP);
1138 }
1139 }
1140
1141 virtual void OnInit()
1142 {
1143 /* Size of the boolean yes/no button. */
1144 Dimension yesno_dim = maxdim(GetStringBoundingBox(STR_FACE_YES), GetStringBoundingBox(STR_FACE_NO));
1145 yesno_dim.width += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
1146 yesno_dim.height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
1147 /* Size of the number button + arrows. */
1148 Dimension number_dim = {0, 0};
1149 for (int val = 1; val <= 12; val++) {
1150 SetDParam(0, val);
1151 number_dim = maxdim(number_dim, GetStringBoundingBox(STR_JUST_INT));
1152 }
1153 uint arrows_width = GetSpriteSize(SPR_ARROW_LEFT).width + GetSpriteSize(SPR_ARROW_RIGHT).width + 2 * (WD_IMGBTN_LEFT + WD_IMGBTN_RIGHT);
1154 number_dim.width += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT + arrows_width;
1155 number_dim.height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
1156 /* Compute width of both buttons. */
1157 yesno_dim.width = max(yesno_dim.width, number_dim.width);
1158 number_dim.width = yesno_dim.width - arrows_width;
1159
1160 this->yesno_dim = yesno_dim;
1161 this->number_dim = number_dim;
1162 }
1163
1164 virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
1165 {
1166 switch (widget) {
1167 case WID_SCMF_FACE: {
1168 Dimension face_size = GetSpriteSize(SPR_GRADIENT);
1169 size->width = max(size->width, face_size.width);
1170 size->height = max(size->height, face_size.height);
1171 break;
1172 }
1173
1174 case WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT:
1175 case WID_SCMF_TIE_EARRING_TEXT: {
1176 StringID str = (widget == WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT) ? STR_FACE_MOUSTACHE : STR_FACE_TIE;
1177 *size = maxdim (GetStringBoundingBox (str), GetStringBoundingBox (STR_FACE_EARRING));
1178 size->width += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
1179 size->height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
1180 break;
1181 }
1182
1183 case WID_SCMF_LIPS_MOUSTACHE_TEXT:
1184 *size = maxdim(GetStringBoundingBox(STR_FACE_LIPS), GetStringBoundingBox(STR_FACE_MOUSTACHE));
1185 size->width += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
1186 size->height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
1187 break;
1188
1189 case WID_SCMF_HAS_GLASSES_TEXT:
1190 case WID_SCMF_HAIR_TEXT:
1191 case WID_SCMF_EYEBROWS_TEXT:
1192 case WID_SCMF_EYECOLOUR_TEXT:
1193 case WID_SCMF_GLASSES_TEXT:
1194 case WID_SCMF_NOSE_TEXT:
1195 case WID_SCMF_CHIN_TEXT:
1196 case WID_SCMF_JACKET_TEXT:
1197 case WID_SCMF_COLLAR_TEXT:
1198 *size = GetStringBoundingBox(PART_TEXTS[widget - WID_SCMF_HAS_GLASSES_TEXT]);
1199 size->width += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
1200 size->height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
1201 break;
1202
1203 case WID_SCMF_HAS_MOUSTACHE_EARRING:
1204 case WID_SCMF_HAS_GLASSES:
1205 *size = this->yesno_dim;
1206 break;
1207
1208 case WID_SCMF_EYECOLOUR:
1209 case WID_SCMF_CHIN:
1210 case WID_SCMF_EYEBROWS:
1211 case WID_SCMF_LIPS_MOUSTACHE:
1212 case WID_SCMF_NOSE:
1213 case WID_SCMF_HAIR:
1214 case WID_SCMF_JACKET:
1215 case WID_SCMF_COLLAR:
1216 case WID_SCMF_TIE_EARRING:
1217 case WID_SCMF_GLASSES:
1218 *size = this->number_dim;
1219 break;
1220 }
1221 }
1222
1223 virtual void OnPaint()
1224 {
1225 /* lower the non-selected gender button */
1226 this->SetWidgetsLoweredState(!this->is_female, WID_SCMF_MALE, WID_SCMF_MALE2, WIDGET_LIST_END);
1227 this->SetWidgetsLoweredState( this->is_female, WID_SCMF_FEMALE, WID_SCMF_FEMALE2, WIDGET_LIST_END);
1228
1229 /* advanced company manager face selection window */
1230
1231 /* lower the non-selected ethnicity button */
1232 this->SetWidgetLoweredState(WID_SCMF_ETHNICITY_EUR, !HasBit(this->ge, ETHNICITY_BLACK));
1233 this->SetWidgetLoweredState(WID_SCMF_ETHNICITY_AFR, HasBit(this->ge, ETHNICITY_BLACK));
1234
1235
1236 /* Disable dynamically the widgets which CompanyManagerFaceVariable has less than 2 options
1237 * (or in other words you haven't any choice).
1238 * If the widgets depend on a HAS-variable and this is false the widgets will be disabled, too. */
1239
1240 /* Eye colour buttons */
1241 this->SetWidgetsDisabledState(_cmf_info[CMFV_EYE_COLOUR].valid_values[this->ge] < 2,
1242 WID_SCMF_EYECOLOUR, WID_SCMF_EYECOLOUR_L, WID_SCMF_EYECOLOUR_R, WIDGET_LIST_END);
1243
1244 /* Chin buttons */
1245 this->SetWidgetsDisabledState(_cmf_info[CMFV_CHIN].valid_values[this->ge] < 2,
1246 WID_SCMF_CHIN, WID_SCMF_CHIN_L, WID_SCMF_CHIN_R, WIDGET_LIST_END);
1247
1248 /* Eyebrows buttons */
1249 this->SetWidgetsDisabledState(_cmf_info[CMFV_EYEBROWS].valid_values[this->ge] < 2,
1250 WID_SCMF_EYEBROWS, WID_SCMF_EYEBROWS_L, WID_SCMF_EYEBROWS_R, WIDGET_LIST_END);
1251
1252 /* Lips or (if it a male face with a moustache) moustache buttons */
1253 this->SetWidgetsDisabledState(_cmf_info[this->is_moust_male ? CMFV_MOUSTACHE : CMFV_LIPS].valid_values[this->ge] < 2,
1254 WID_SCMF_LIPS_MOUSTACHE, WID_SCMF_LIPS_MOUSTACHE_L, WID_SCMF_LIPS_MOUSTACHE_R, WIDGET_LIST_END);
1255
1256 /* Nose buttons | male faces with moustache haven't any nose options */
1257 this->SetWidgetsDisabledState(_cmf_info[CMFV_NOSE].valid_values[this->ge] < 2 || this->is_moust_male,
1258 WID_SCMF_NOSE, WID_SCMF_NOSE_L, WID_SCMF_NOSE_R, WIDGET_LIST_END);
1259
1260 /* Hair buttons */
1261 this->SetWidgetsDisabledState(_cmf_info[CMFV_HAIR].valid_values[this->ge] < 2,
1262 WID_SCMF_HAIR, WID_SCMF_HAIR_L, WID_SCMF_HAIR_R, WIDGET_LIST_END);
1263
1264 /* Jacket buttons */
1265 this->SetWidgetsDisabledState(_cmf_info[CMFV_JACKET].valid_values[this->ge] < 2,
1266 WID_SCMF_JACKET, WID_SCMF_JACKET_L, WID_SCMF_JACKET_R, WIDGET_LIST_END);
1267
1268 /* Collar buttons */
1269 this->SetWidgetsDisabledState(_cmf_info[CMFV_COLLAR].valid_values[this->ge] < 2,
1270 WID_SCMF_COLLAR, WID_SCMF_COLLAR_L, WID_SCMF_COLLAR_R, WIDGET_LIST_END);
1271
1272 /* Tie/earring buttons | female faces without earring haven't any earring options */
1273 this->SetWidgetsDisabledState(_cmf_info[CMFV_TIE_EARRING].valid_values[this->ge] < 2 ||
1274 (this->is_female && this->GetFaceBits(CMFV_HAS_TIE_EARRING) == 0),
1275 WID_SCMF_TIE_EARRING, WID_SCMF_TIE_EARRING_L, WID_SCMF_TIE_EARRING_R, WIDGET_LIST_END);
1276
1277 /* Glasses buttons | faces without glasses haven't any glasses options */
1278 this->SetWidgetsDisabledState(_cmf_info[CMFV_GLASSES].valid_values[this->ge] < 2 || this->GetFaceBits(CMFV_HAS_GLASSES) == 0,
1279 WID_SCMF_GLASSES, WID_SCMF_GLASSES_L, WID_SCMF_GLASSES_R, WIDGET_LIST_END);
1280
1281 this->DrawWidgets();
1282 }
1283
1284 virtual void DrawWidget(const Rect &r, int widget) const
1285 {
1286 switch (widget) {
1287 case WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT:
1288 case WID_SCMF_TIE_EARRING_TEXT: {
1289 StringID str = this->is_female ? STR_FACE_EARRING :
1290 (widget == WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT) ? STR_FACE_MOUSTACHE : STR_FACE_TIE;
1291 DrawString(r.left + WD_FRAMERECT_LEFT, r.right - WD_FRAMERECT_RIGHT, r.top + WD_FRAMERECT_TOP, str, TC_GOLD, SA_RIGHT);
1292 break;
1293 }
1294
1295 case WID_SCMF_LIPS_MOUSTACHE_TEXT:
1296 DrawString(r.left + WD_FRAMERECT_LEFT, r.right - WD_FRAMERECT_RIGHT, r.top + WD_FRAMERECT_TOP, (this->is_moust_male) ? STR_FACE_MOUSTACHE : STR_FACE_LIPS, TC_GOLD, SA_RIGHT);
1297 break;
1298
1299 case WID_SCMF_HAS_GLASSES_TEXT:
1300 case WID_SCMF_HAIR_TEXT:
1301 case WID_SCMF_EYEBROWS_TEXT:
1302 case WID_SCMF_EYECOLOUR_TEXT:
1303 case WID_SCMF_GLASSES_TEXT:
1304 case WID_SCMF_NOSE_TEXT:
1305 case WID_SCMF_CHIN_TEXT:
1306 case WID_SCMF_JACKET_TEXT:
1307 case WID_SCMF_COLLAR_TEXT:
1308 DrawString(r.left + WD_FRAMERECT_LEFT, r.right - WD_FRAMERECT_RIGHT, r.top + WD_FRAMERECT_TOP, PART_TEXTS[widget - WID_SCMF_HAS_GLASSES_TEXT], TC_GOLD, SA_RIGHT);
1309 break;
1310
1311
1312 case WID_SCMF_HAS_MOUSTACHE_EARRING:
1313 this->DrawFaceStringLabel (WID_SCMF_HAS_MOUSTACHE_EARRING,
1314 this->is_female ? CMFV_HAS_TIE_EARRING : CMFV_HAS_MOUSTACHE, true);
1315 break;
1316
1317 case WID_SCMF_TIE_EARRING:
1318 this->DrawFaceStringLabel (WID_SCMF_TIE_EARRING, CMFV_TIE_EARRING, false);
1319 break;
1320
1321 case WID_SCMF_LIPS_MOUSTACHE:
1322 this->DrawFaceStringLabel (WID_SCMF_LIPS_MOUSTACHE,
1323 this->is_moust_male ? CMFV_MOUSTACHE : CMFV_LIPS, false);
1324 break;
1325
1326 case WID_SCMF_HAS_GLASSES:
1327 this->DrawFaceStringLabel (WID_SCMF_HAS_GLASSES, CMFV_HAS_GLASSES, true );
1328 break;
1329
1330 case WID_SCMF_HAIR:
1331 this->DrawFaceStringLabel (WID_SCMF_HAIR, CMFV_HAIR, false);
1332 break;
1333
1334 case WID_SCMF_EYEBROWS:
1335 this->DrawFaceStringLabel (WID_SCMF_EYEBROWS, CMFV_EYEBROWS, false);
1336 break;
1337
1338 case WID_SCMF_EYECOLOUR:
1339 this->DrawFaceStringLabel (WID_SCMF_EYECOLOUR, CMFV_EYE_COLOUR, false);
1340 break;
1341
1342 case WID_SCMF_GLASSES:
1343 this->DrawFaceStringLabel (WID_SCMF_GLASSES, CMFV_GLASSES, false);
1344 break;
1345
1346 case WID_SCMF_NOSE:
1347 this->DrawFaceStringLabel (WID_SCMF_NOSE, CMFV_NOSE, false);
1348 break;
1349
1350 case WID_SCMF_CHIN:
1351 this->DrawFaceStringLabel (WID_SCMF_CHIN, CMFV_CHIN, false);
1352 break;
1353
1354 case WID_SCMF_JACKET:
1355 this->DrawFaceStringLabel (WID_SCMF_JACKET, CMFV_JACKET, false);
1356 break;
1357
1358 case WID_SCMF_COLLAR:
1359 this->DrawFaceStringLabel (WID_SCMF_COLLAR, CMFV_COLLAR, false);
1360 break;
1361
1362 case WID_SCMF_FACE:
1363 DrawCompanyManagerFace(this->face, Company::Get((CompanyID)this->window_number)->colour, r.left, r.top);
1364 break;
1365 }
1366 }
1367
1368 virtual void OnClick(Point pt, int widget, int click_count)
1369 {
1370 switch (widget) {
1371 /* Toggle size, advanced/simple face selection */
1372 case WID_SCMF_TOGGLE_LARGE_SMALL:
1373 case WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON:
1374 this->advanced = !this->advanced;
1375 this->SelectDisplayPlanes(this->advanced);
1376 this->ReInit();
1377 break;
1378
1379 /* OK button */
1380 case WID_SCMF_ACCEPT:
1381 DoCommandP(0, 0, this->face, CMD_SET_COMPANY_MANAGER_FACE);
1382 /* FALL THROUGH */
1383
1384 /* Cancel button */
1385 case WID_SCMF_CANCEL:
1386 delete this;
1387 break;
1388
1389 /* Load button */
1390 case WID_SCMF_LOAD:
1391 this->face = _company_manager_face;
1392 ScaleAllCompanyManagerFaceBits(this->face);
1393 ShowErrorMessage(STR_FACE_LOAD_DONE, INVALID_STRING_ID, WL_INFO);
1394 this->UpdateData();
1395 this->SetDirty();
1396 break;
1397
1398 /* 'Company manager face number' button, view and/or set company manager face number */
1399 case WID_SCMF_FACECODE:
1400 SetDParam(0, this->face);
1401 ShowQueryString(STR_JUST_INT, STR_FACE_FACECODE_CAPTION, 10 + 1, this, CS_NUMERAL, QSF_NONE);
1402 break;
1403
1404 /* Save button */
1405 case WID_SCMF_SAVE:
1406 _company_manager_face = this->face;
1407 ShowErrorMessage(STR_FACE_SAVE_DONE, INVALID_STRING_ID, WL_INFO);
1408 break;
1409
1410 /* Toggle gender (male/female) button */
1411 case WID_SCMF_MALE:
1412 case WID_SCMF_FEMALE:
1413 case WID_SCMF_MALE2:
1414 case WID_SCMF_FEMALE2:
1415 SetCompanyManagerFaceBits(this->face, CMFV_GENDER, this->ge, (widget == WID_SCMF_FEMALE || widget == WID_SCMF_FEMALE2));
1416 ScaleAllCompanyManagerFaceBits(this->face);
1417 this->UpdateData();
1418 this->SetDirty();
1419 break;
1420
1421 /* Randomize face button */
1422 case WID_SCMF_RANDOM_NEW_FACE:
1423 RandomCompanyManagerFaceBits(this->face, this->ge, this->advanced);
1424 this->UpdateData();
1425 this->SetDirty();
1426 break;
1427
1428 /* Toggle ethnicity (european/african) button */
1429 case WID_SCMF_ETHNICITY_EUR:
1430 case WID_SCMF_ETHNICITY_AFR:
1431 SetCompanyManagerFaceBits(this->face, CMFV_ETHNICITY, this->ge, widget - WID_SCMF_ETHNICITY_EUR);
1432 ScaleAllCompanyManagerFaceBits(this->face);
1433 this->UpdateData();
1434 this->SetDirty();
1435 break;
1436
1437 default:
1438 /* Here all buttons from WID_SCMF_HAS_MOUSTACHE_EARRING to WID_SCMF_GLASSES_R are handled.
1439 * First it checks which CompanyManagerFaceVariable is being changed, and then either
1440 * a: invert the value for boolean variables, or
1441 * b: it checks inside of IncreaseCompanyManagerFaceBits() if a left (_L) butten is pressed and then decrease else increase the variable */
1442 if (widget >= WID_SCMF_HAS_MOUSTACHE_EARRING && widget <= WID_SCMF_GLASSES_R) {
1443 CompanyManagerFaceVariable cmfv; // which CompanyManagerFaceVariable shall be edited
1444
1445 if (widget < WID_SCMF_EYECOLOUR_L) { // Bool buttons
1446 switch (widget - WID_SCMF_HAS_MOUSTACHE_EARRING) {
1447 default: NOT_REACHED();
1448 case 0: cmfv = this->is_female ? CMFV_HAS_TIE_EARRING : CMFV_HAS_MOUSTACHE; break; // Has earring/moustache button
1449 case 1: cmfv = CMFV_HAS_GLASSES; break; // Has glasses button
1450 }
1451 SetCompanyManagerFaceBits(this->face, cmfv, this->ge, !this->GetFaceBits(cmfv));
1452 ScaleAllCompanyManagerFaceBits(this->face);
1453 } else { // Value buttons
1454 switch ((widget - WID_SCMF_EYECOLOUR_L) / 3) {
1455 default: NOT_REACHED();
1456 case 0: cmfv = CMFV_EYE_COLOUR; break; // Eye colour buttons
1457 case 1: cmfv = CMFV_CHIN; break; // Chin buttons
1458 case 2: cmfv = CMFV_EYEBROWS; break; // Eyebrows buttons
1459 case 3: cmfv = this->is_moust_male ? CMFV_MOUSTACHE : CMFV_LIPS; break; // Moustache or lips buttons
1460 case 4: cmfv = CMFV_NOSE; break; // Nose buttons
1461 case 5: cmfv = CMFV_HAIR; break; // Hair buttons
1462 case 6: cmfv = CMFV_JACKET; break; // Jacket buttons
1463 case 7: cmfv = CMFV_COLLAR; break; // Collar buttons
1464 case 8: cmfv = CMFV_TIE_EARRING; break; // Tie/earring buttons
1465 case 9: cmfv = CMFV_GLASSES; break; // Glasses buttons
1466 }
1467 /* 0 == left (_L), 1 == middle or 2 == right (_R) - button click */
1468 IncreaseCompanyManagerFaceBits(this->face, cmfv, this->ge, (((widget - WID_SCMF_EYECOLOUR_L) % 3) != 0) ? 1 : -1);
1469 }
1470 this->UpdateData();
1471 this->SetDirty();
1472 }
1473 break;
1474 }
1475 }
1476
1477 virtual void OnQueryTextFinished(char *str)
1478 {
1479 if (str == NULL) return;
1480 /* Set a new company manager face number */
1481 if (!StrEmpty(str)) {
1482 this->face = strtoul(str, NULL, 10);
1483 ScaleAllCompanyManagerFaceBits(this->face);
1484 ShowErrorMessage(STR_FACE_FACECODE_SET, INVALID_STRING_ID, WL_INFO);
1485 this->UpdateData();
1486 this->SetDirty();
1487 } else {
1488 ShowErrorMessage(STR_FACE_FACECODE_ERR, INVALID_STRING_ID, WL_INFO);
1489 }
1490 }
1491 };
1492
1493 /** Company manager face selection window description */
1494 static WindowDesc _select_company_manager_face_desc(
1495 WDP_AUTO, "company_face", 0, 0,
1496 WC_COMPANY_MANAGER_FACE, WC_NONE,
1497 WDF_CONSTRUCTION,
1498 _nested_select_company_manager_face_widgets, lengthof(_nested_select_company_manager_face_widgets)
1499 );
1500
1501 /**
1502 * Open the simple/advanced company manager face selection window
1503 *
1504 * @param parent the parent company window
1505 * @param adv simple or advanced face selection window
1506 * @param top previous top position of the window
1507 * @param left previous left position of the window
1508 */
1509 static void DoSelectCompanyManagerFace(Window *parent)
1510 {
1511 if (!Company::IsValidID((CompanyID)parent->window_number)) return;
1512
1513 if (BringWindowToFrontById(WC_COMPANY_MANAGER_FACE, parent->window_number)) return;
1514 new SelectCompanyManagerFaceWindow(&_select_company_manager_face_desc, parent);
1515 }
1516
1517 static const NWidgetPart _nested_company_infrastructure_widgets[] = {
1518 NWidget(NWID_HORIZONTAL),
1519 NWidget(WWT_CLOSEBOX, COLOUR_GREY),
1520 NWidget(WWT_CAPTION, COLOUR_GREY, WID_CI_CAPTION), SetDataTip(STR_COMPANY_INFRASTRUCTURE_VIEW_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
1521 NWidget(WWT_SHADEBOX, COLOUR_GREY),
1522 NWidget(WWT_STICKYBOX, COLOUR_GREY),
1523 EndContainer(),
1524 NWidget(WWT_PANEL, COLOUR_GREY),
1525 NWidget(NWID_VERTICAL), SetPIP(WD_FRAMERECT_TOP, 4, WD_FRAMETEXT_BOTTOM),
1526 NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
1527 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_RAIL_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
1528 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_RAIL_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
1529 EndContainer(),
1530 NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
1531 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_ROAD_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
1532 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_ROAD_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
1533 EndContainer(),
1534 NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
1535 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_WATER_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
1536 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_WATER_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
1537 EndContainer(),
1538 NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
1539 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_STATION_DESC), SetMinimalTextLines(3, 0), SetFill(1, 0),
1540 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_STATION_COUNT), SetMinimalTextLines(3, 0), SetFill(0, 1),
1541 EndContainer(),
1542 NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
1543 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_TOTAL_DESC), SetFill(1, 0),
1544 NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_TOTAL), SetFill(0, 1),
1545 EndContainer(),
1546 EndContainer(),
1547 EndContainer(),
1548 };
1549
1550 /**
1551 * Window with detailed information about the company's infrastructure.
1552 */
1553 struct CompanyInfrastructureWindow : Window
1554 {
1555 RailTypes railtypes; ///< Valid railtypes.
1556 RoadTypes roadtypes; ///< Valid roadtypes.
1557
1558 uint total_width; ///< String width of the total cost line.
1559
1560 CompanyInfrastructureWindow(WindowDesc *desc, WindowNumber window_number) : Window(desc)
1561 {
1562 this->UpdateRailRoadTypes();
1563
1564 this->InitNested(window_number);
1565 this->owner = (Owner)this->window_number;
1566 }
1567
1568 void UpdateRailRoadTypes()
1569 {
1570 this->railtypes = RAILTYPES_NONE;
1571 this->roadtypes = ROADTYPES_ROAD; // Road is always available.
1572
1573 /* Find the used railtypes. */
1574 Engine *e;
1575 FOR_ALL_ENGINES_OF_TYPE(e, VEH_TRAIN) {
1576 if (!HasBit(e->info.climates, _settings_game.game_creation.landscape)) continue;
1577
1578 this->railtypes |= GetRailTypeInfo(e->u.rail.railtype)->introduces_railtypes;
1579 }
1580
1581 /* Get the date introduced railtypes as well. */
1582 this->railtypes = AddDateIntroducedRailTypes(this->railtypes, MAX_DAY);
1583
1584 /* Tram is only visible when there will be a tram. */
1585 FOR_ALL_ENGINES_OF_TYPE(e, VEH_ROAD) {
1586 if (!HasBit(e->info.climates, _settings_game.game_creation.landscape)) continue;
1587 if (!HasBit(e->info.misc_flags, EF_ROAD_TRAM)) continue;
1588
1589 this->roadtypes |= ROADTYPES_TRAM;
1590 break;
1591 }
1592 }
1593
1594 /** Get total infrastructure maintenance cost. */
1595 Money GetTotalMaintenanceCost() const
1596 {
1597 const Company *c = Company::Get((CompanyID)this->window_number);
1598 Money total;
1599
1600 uint32 rail_total = c->infrastructure.GetRailTotal();
1601 for (RailType rt = RAILTYPE_BEGIN; rt != RAILTYPE_END; rt++) {
1602 if (HasBit(this->railtypes, rt)) total += RailMaintenanceCost(rt, c->infrastructure.rail[rt], rail_total);
1603 }
1604 total += SignalMaintenanceCost(c->infrastructure.signal);
1605
1606 if (HasBit(this->roadtypes, ROADTYPE_ROAD)) total += RoadMaintenanceCost(ROADTYPE_ROAD, c->infrastructure.road[ROADTYPE_ROAD]);
1607 if (HasBit(this->roadtypes, ROADTYPE_TRAM)) total += RoadMaintenanceCost(ROADTYPE_TRAM, c->infrastructure.road[ROADTYPE_TRAM]);
1608
1609 total += CanalMaintenanceCost(c->infrastructure.water);
1610 total += StationMaintenanceCost(c->infrastructure.station);
1611 total += AirportMaintenanceCost(c->index);
1612
1613 return total;
1614 }
1615
1616 virtual void SetStringParameters(int widget) const
1617 {
1618 switch (widget) {
1619 case WID_CI_CAPTION:
1620 SetDParam(0, (CompanyID)this->window_number);
1621 break;
1622 }
1623 }
1624
1625 virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
1626 {
1627 const Company *c = Company::Get((CompanyID)this->window_number);
1628
1629 switch (widget) {
1630 case WID_CI_RAIL_DESC: {
1631 uint lines = 1;
1632
1633 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_RAIL_SECT).width);
1634
1635 for (RailType rt = RAILTYPE_BEGIN; rt < RAILTYPE_END; rt++) {
1636 if (HasBit(this->railtypes, rt)) {
1637 lines++;
1638 SetDParam(0, GetRailTypeInfo(rt)->strings.name);
1639 size->width = max(size->width, GetStringBoundingBox(STR_WHITE_STRING).width + WD_FRAMERECT_LEFT);
1640 }
1641 }
1642 if (this->railtypes != RAILTYPES_NONE) {
1643 lines++;
1644 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_SIGNALS).width + WD_FRAMERECT_LEFT);
1645 }
1646
1647 size->height = max(size->height, lines * FONT_HEIGHT_NORMAL);
1648 break;
1649 }
1650
1651 case WID_CI_ROAD_DESC: {
1652 uint lines = 1;
1653
1654 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD_SECT).width);
1655
1656 if (HasBit(this->roadtypes, ROADTYPE_ROAD)) {
1657 lines++;
1658 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD).width + WD_FRAMERECT_LEFT);
1659 }
1660 if (HasBit(this->roadtypes, ROADTYPE_TRAM)) {
1661 lines++;
1662 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_TRAMWAY).width + WD_FRAMERECT_LEFT);
1663 }
1664
1665 size->height = max(size->height, lines * FONT_HEIGHT_NORMAL);
1666 break;
1667 }
1668
1669 case WID_CI_WATER_DESC:
1670 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_WATER_SECT).width);
1671 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_CANALS).width + WD_FRAMERECT_LEFT);
1672 break;
1673
1674 case WID_CI_STATION_DESC:
1675 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_STATION_SECT).width);
1676 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_STATIONS).width + WD_FRAMERECT_LEFT);
1677 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_AIRPORTS).width + WD_FRAMERECT_LEFT);
1678 break;
1679
1680 case WID_CI_RAIL_COUNT:
1681 case WID_CI_ROAD_COUNT:
1682 case WID_CI_WATER_COUNT:
1683 case WID_CI_STATION_COUNT:
1684 case WID_CI_TOTAL: {
1685 /* Find the maximum count that is displayed. */
1686 uint32 max_val = 1000; // Some random number to reserve enough space.
1687 Money max_cost = 10000; // Some random number to reserve enough space.
1688 uint32 rail_total = c->infrastructure.GetRailTotal();
1689 for (RailType rt = RAILTYPE_BEGIN; rt < RAILTYPE_END; rt++) {
1690 max_val = max(max_val, c->infrastructure.rail[rt]);
1691 max_cost = max(max_cost, RailMaintenanceCost(rt, c->infrastructure.rail[rt], rail_total));
1692 }
1693 max_val = max(max_val, c->infrastructure.signal);
1694 max_cost = max(max_cost, SignalMaintenanceCost(c->infrastructure.signal));
1695 for (RoadType rt = ROADTYPE_BEGIN; rt < ROADTYPE_END; rt++) {
1696 max_val = max(max_val, c->infrastructure.road[rt]);
1697 max_cost = max(max_cost, RoadMaintenanceCost(rt, c->infrastructure.road[rt]));
1698 }
1699 max_val = max(max_val, c->infrastructure.water);
1700 max_cost = max(max_cost, CanalMaintenanceCost(c->infrastructure.water));
1701 max_val = max(max_val, c->infrastructure.station);
1702 max_cost = max(max_cost, StationMaintenanceCost(c->infrastructure.station));
1703 max_val = max(max_val, c->infrastructure.airport);
1704 max_cost = max(max_cost, AirportMaintenanceCost(c->index));
1705
1706 SetDParamMaxValue(0, max_val);
1707 uint count_width = GetStringBoundingBox(STR_WHITE_COMMA).width + 20; // Reserve some wiggle room
1708
1709 if (_settings_game.economy.infrastructure_maintenance) {
1710 SetDParamMaxValue(0, this->GetTotalMaintenanceCost() * 12); // Convert to per year
1711 this->total_width = GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL).width + 20;
1712 size->width = max(size->width, this->total_width);
1713
1714 SetDParamMaxValue(0, max_cost * 12); // Convert to per year
1715 count_width += max(this->total_width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL).width);
1716 }
1717
1718 size->width = max(size->width, count_width);
1719
1720 /* Set height of the total line. */
1721 if (widget == WID_CI_TOTAL) {
1722 size->height = _settings_game.economy.infrastructure_maintenance ? max(size->height, EXP_LINESPACE + FONT_HEIGHT_NORMAL) : 0;
1723 }
1724 break;
1725 }
1726 }
1727 }
1728
1729 /**
1730 * Helper for drawing the counts line.
1731 * @param r The bounds to draw in.
1732 * @param y The y position to draw at.
1733 * @param count The count to show on this line.
1734 * @param monthly_cost The monthly costs.
1735 */
1736 void DrawCountLine(const Rect &r, int &y, int count, Money monthly_cost) const
1737 {
1738 SetDParam(0, count);
1739 DrawString(r.left, r.right, y += FONT_HEIGHT_NORMAL, STR_WHITE_COMMA, TC_FROMSTRING, SA_RIGHT);
1740
1741 if (_settings_game.economy.infrastructure_maintenance) {
1742 SetDParam(0, monthly_cost * 12); // Convert to per year
1743 int left = _current_text_dir == TD_RTL ? r.right - this->total_width : r.left;
1744 DrawString(left, left + this->total_width, y, STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL, TC_FROMSTRING, SA_RIGHT);
1745 }
1746 }
1747
1748 virtual void DrawWidget(const Rect &r, int widget) const
1749 {
1750 const Company *c = Company::Get((CompanyID)this->window_number);
1751 int y = r.top;
1752
1753 int offs_left = _current_text_dir == TD_LTR ? WD_FRAMERECT_LEFT : 0;
1754 int offs_right = _current_text_dir == TD_LTR ? 0 : WD_FRAMERECT_LEFT;
1755
1756 switch (widget) {
1757 case WID_CI_RAIL_DESC:
1758 DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_RAIL_SECT);
1759
1760 if (this->railtypes != RAILTYPES_NONE) {
1761 /* Draw name of each valid railtype. */
1762 for (RailType rt = RAILTYPE_BEGIN; rt != RAILTYPE_END; rt++) {
1763 if (HasBit(this->railtypes, rt)) {
1764 SetDParam(0, GetRailTypeInfo(rt)->strings.name);
1765 DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_WHITE_STRING);
1766 }
1767 }
1768 DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_SIGNALS);
1769 } else {
1770 /* No valid railtype. */
1771 DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_VIEW_INFRASTRUCTURE_NONE);
1772 }
1773
1774 break;
1775
1776 case WID_CI_RAIL_COUNT: {
1777 /* Draw infrastructure count for each valid railtype. */
1778 uint32 rail_total = c->infrastructure.GetRailTotal();
1779 for (RailType rt = RAILTYPE_BEGIN; rt != RAILTYPE_END; rt++) {
1780 if (HasBit(this->railtypes, rt)) {
1781 this->DrawCountLine(r, y, c->infrastructure.rail[rt], RailMaintenanceCost(rt, c->infrastructure.rail[rt], rail_total));
1782 }
1783 }
1784 if (this->railtypes != RAILTYPES_NONE) {
1785 this->DrawCountLine(r, y, c->infrastructure.signal, SignalMaintenanceCost(c->infrastructure.signal));
1786 }
1787 break;
1788 }
1789
1790 case WID_CI_ROAD_DESC:
1791 DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD_SECT);
1792
1793 if (this->roadtypes != ROADTYPES_NONE) {
1794 if (HasBit(this->roadtypes, ROADTYPE_ROAD)) DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD);
1795 if (HasBit(this->roadtypes, ROADTYPE_TRAM)) DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_TRAMWAY);
1796 } else {
1797 /* No valid roadtypes. */
1798 DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_VIEW_INFRASTRUCTURE_NONE);
1799 }
1800
1801 break;
1802
1803 case WID_CI_ROAD_COUNT:
1804 if (HasBit(this->roadtypes, ROADTYPE_ROAD)) {
1805 this->DrawCountLine(r, y, c->infrastructure.road[ROADTYPE_ROAD], RoadMaintenanceCost(ROADTYPE_ROAD, c->infrastructure.road[ROADTYPE_ROAD]));
1806 }
1807 if (HasBit(this->roadtypes, ROADTYPE_TRAM)) {
1808 this->DrawCountLine(r, y, c->infrastructure.road[ROADTYPE_TRAM], RoadMaintenanceCost(ROADTYPE_TRAM, c->infrastructure.road[ROADTYPE_TRAM]));
1809 }
1810 break;
1811
1812 case WID_CI_WATER_DESC:
1813 DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_WATER_SECT);
1814 DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_CANALS);
1815 break;
1816
1817 case WID_CI_WATER_COUNT:
1818 this->DrawCountLine(r, y, c->infrastructure.water, CanalMaintenanceCost(c->infrastructure.water));
1819 break;
1820
1821 case WID_CI_TOTAL:
1822 if (_settings_game.economy.infrastructure_maintenance) {
1823 int left = _current_text_dir == TD_RTL ? r.right - this->total_width : r.left;
1824 GfxFillRect(left, y, left + this->total_width, y, PC_WHITE);
1825 y += EXP_LINESPACE;
1826 SetDParam(0, this->GetTotalMaintenanceCost() * 12); // Convert to per year
1827 DrawString(left, left + this->total_width, y, STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL, TC_FROMSTRING, SA_RIGHT);
1828 }
1829 break;
1830
1831 case WID_CI_STATION_DESC:
1832 DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_STATION_SECT);
1833 DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_STATIONS);
1834 DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_AIRPORTS);
1835 break;
1836
1837 case WID_CI_STATION_COUNT:
1838 this->DrawCountLine(r, y, c->infrastructure.station, StationMaintenanceCost(c->infrastructure.station));
1839 this->DrawCountLine(r, y, c->infrastructure.airport, AirportMaintenanceCost(c->index));
1840 break;
1841 }
1842 }
1843
1844 /**
1845 * Some data on this window has become invalid.
1846 * @param data Information about the changed data.
1847 * @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.
1848 */
1849 virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
1850 {
1851 if (!gui_scope) return;
1852
1853 this->UpdateRailRoadTypes();
1854 this->ReInit();
1855 }
1856 };
1857
1858 static WindowDesc _company_infrastructure_desc(
1859 WDP_AUTO, "company_infrastructure", 0, 0,
1860 WC_COMPANY_INFRASTRUCTURE, WC_NONE,
1861 0,
1862 _nested_company_infrastructure_widgets, lengthof(_nested_company_infrastructure_widgets)
1863 );
1864
1865 /**
1866 * Open the infrastructure window of a company.
1867 * @param company Company to show infrastructure of.
1868 */
1869 static void ShowCompanyInfrastructure(CompanyID company)
1870 {
1871 if (!Company::IsValidID(company)) return;
1872 AllocateWindowDescFront<CompanyInfrastructureWindow>(&_company_infrastructure_desc, company);
1873 }
1874
1875 static const NWidgetPart _nested_company_widgets[] = {
1876 NWidget(NWID_HORIZONTAL),
1877 NWidget(WWT_CLOSEBOX, COLOUR_GREY),
1878 NWidget(WWT_CAPTION, COLOUR_GREY, WID_C_CAPTION), SetDataTip(STR_COMPANY_VIEW_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
1879 NWidget(WWT_SHADEBOX, COLOUR_GREY),
1880 NWidget(WWT_STICKYBOX, COLOUR_GREY),
1881 EndContainer(),
1882 NWidget(WWT_PANEL, COLOUR_GREY),
1883 NWidget(NWID_HORIZONTAL), SetPIP(4, 6, 4),
1884 NWidget(NWID_VERTICAL), SetPIP(4, 2, 4),
1885 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_FACE), SetMinimalSize(92, 119), SetFill(1, 0),
1886 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_FACE_TITLE), SetFill(1, 1), SetMinimalTextLines(2, 0),
1887 EndContainer(),
1888 NWidget(NWID_VERTICAL),
1889 NWidget(NWID_HORIZONTAL),
1890 NWidget(NWID_VERTICAL), SetPIP(4, 5, 5),
1891 NWidget(WWT_TEXT, COLOUR_GREY, WID_C_DESC_INAUGURATION), SetDataTip(STR_COMPANY_VIEW_INAUGURATED_TITLE, STR_NULL), SetFill(1, 0),
1892 NWidget(NWID_HORIZONTAL), SetPIP(0, 5, 0),
1893 NWidget(WWT_LABEL, COLOUR_GREY, WID_C_DESC_COLOUR_SCHEME), SetDataTip(STR_COMPANY_VIEW_COLOUR_SCHEME_TITLE, STR_NULL),
1894 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_COLOUR_SCHEME_EXAMPLE), SetMinimalSize(30, 0), SetFill(0, 1),
1895 NWidget(NWID_SPACER), SetFill(1, 0),
1896 EndContainer(),
1897 NWidget(NWID_HORIZONTAL), SetPIP(0, 4, 0),
1898 NWidget(NWID_VERTICAL),
1899 NWidget(WWT_TEXT, COLOUR_GREY, WID_C_DESC_VEHICLE), SetDataTip(STR_COMPANY_VIEW_VEHICLES_TITLE, STR_NULL),
1900 NWidget(NWID_SPACER), SetFill(0, 1),
1901 EndContainer(),
1902 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_VEHICLE_COUNTS), SetMinimalTextLines(4, 0),
1903 NWidget(NWID_SPACER), SetFill(1, 0),
1904 EndContainer(),
1905 EndContainer(),
1906 NWidget(NWID_VERTICAL), SetPIP(4, 2, 4),
1907 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_VIEW_BUILD_HQ),
1908 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_VIEW_HQ), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_VIEW_HQ_BUTTON, STR_COMPANY_VIEW_VIEW_HQ_TOOLTIP),
1909 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_C_BUILD_HQ), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_BUILD_HQ_BUTTON, STR_COMPANY_VIEW_BUILD_HQ_TOOLTIP),
1910 EndContainer(),
1911 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_RELOCATE),
1912 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_C_RELOCATE_HQ), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_RELOCATE_HQ, STR_COMPANY_VIEW_RELOCATE_COMPANY_HEADQUARTERS),
1913 NWidget(NWID_SPACER), SetMinimalSize(90, 0),
1914 EndContainer(),
1915 NWidget(NWID_SPACER), SetFill(0, 1),
1916 EndContainer(),
1917 EndContainer(),
1918 NWidget(WWT_TEXT, COLOUR_GREY, WID_C_DESC_COMPANY_VALUE), SetDataTip(STR_COMPANY_VIEW_COMPANY_VALUE, STR_NULL), SetFill(1, 0),
1919 NWidget(NWID_VERTICAL), SetPIP(4, 2, 4),
1920 NWidget(NWID_HORIZONTAL), SetPIP(0, 4, 0),
1921 NWidget(NWID_VERTICAL),
1922 NWidget(WWT_TEXT, COLOUR_GREY, WID_C_DESC_INFRASTRUCTURE), SetDataTip(STR_COMPANY_VIEW_INFRASTRUCTURE, STR_NULL),
1923 NWidget(NWID_SPACER), SetFill(0, 1),
1924 EndContainer(),
1925 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_INFRASTRUCTURE_COUNTS), SetMinimalTextLines(5, 0), SetFill(1, 0),
1926 NWidget(NWID_VERTICAL),
1927 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_VIEW_INFRASTRUCTURE), SetDataTip(STR_COMPANY_VIEW_INFRASTRUCTURE_BUTTON, STR_COMPANY_VIEW_INFRASTRUCTURE_TOOLTIP),
1928 NWidget(NWID_SPACER), SetFill(0, 1), SetMinimalSize(90, 0),
1929 EndContainer(),
1930 EndContainer(),
1931 EndContainer(),
1932 NWidget(NWID_HORIZONTAL),
1933 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_DESC_OWNERS),
1934 NWidget(NWID_VERTICAL), SetPIP(5, 5, 4),
1935 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_OWNERS), SetMinimalTextLines(3, 0),
1936 NWidget(NWID_SPACER), SetFill(0, 1),
1937 EndContainer(),
1938 EndContainer(),
1939 NWidget(NWID_VERTICAL), SetPIP(4, 2, 4),
1940 NWidget(NWID_SPACER), SetMinimalSize(90, 0), SetFill(0, 1),
1941 /* Multi player buttons. */
1942 NWidget(NWID_HORIZONTAL),
1943 NWidget(WWT_EMPTY, COLOUR_GREY, WID_C_HAS_PASSWORD),
1944 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_MULTIPLAYER),
1945 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COMPANY_PASSWORD), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_PASSWORD, STR_COMPANY_VIEW_PASSWORD_TOOLTIP),
1946 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COMPANY_JOIN), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_JOIN, STR_COMPANY_VIEW_JOIN_TOOLTIP),
1947 EndContainer(),
1948 EndContainer(),
1949 EndContainer(),
1950 EndContainer(),
1951 EndContainer(),
1952 EndContainer(),
1953 EndContainer(),
1954 /* Button bars at the bottom. */
1955 NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_BUTTONS),
1956 NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
1957 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_NEW_FACE), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_NEW_FACE_BUTTON, STR_COMPANY_VIEW_NEW_FACE_TOOLTIP),
1958 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COLOUR_SCHEME), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_COLOUR_SCHEME_BUTTON, STR_COMPANY_VIEW_COLOUR_SCHEME_TOOLTIP),
1959 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_PRESIDENT_NAME), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_PRESIDENT_NAME_BUTTON, STR_COMPANY_VIEW_PRESIDENT_NAME_TOOLTIP),
1960 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COMPANY_NAME), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_COMPANY_NAME_BUTTON, STR_COMPANY_VIEW_COMPANY_NAME_TOOLTIP),
1961 EndContainer(),
1962 NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
1963 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_BUY_SHARE), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_BUY_SHARE_BUTTON, STR_COMPANY_VIEW_BUY_SHARE_TOOLTIP),
1964 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_SELL_SHARE), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_SELL_SHARE_BUTTON, STR_COMPANY_VIEW_SELL_SHARE_TOOLTIP),
1965 EndContainer(),
1966 EndContainer(),
1967 };
1968
1969 int GetAmountOwnedBy(const Company *c, Owner owner)
1970 {
1971 return (c->share_owners[0] == owner) +
1972 (c->share_owners[1] == owner) +
1973 (c->share_owners[2] == owner) +
1974 (c->share_owners[3] == owner);
1975 }
1976
1977 /** Strings for the company vehicle counts */
1978 static const StringID _company_view_vehicle_count_strings[] = {
1979 STR_COMPANY_VIEW_TRAINS, STR_COMPANY_VIEW_ROAD_VEHICLES, STR_COMPANY_VIEW_SHIPS, STR_COMPANY_VIEW_AIRCRAFT
1980 };
1981
1982 /**
1983 * Window with general information about a company
1984 */
1985 struct CompanyWindow : Window
1986 {
1987 CompanyWidgets query_widget;
1988
1989 /** Display planes in the company window. */
1990 enum CompanyWindowPlanes {
1991 /* Display planes of the #WID_C_SELECT_MULTIPLAYER selection widget. */
1992 CWP_MP_C_PWD = 0, ///< Display the company password button.
1993 CWP_MP_C_JOIN, ///< Display the join company button.
1994
1995 /* Display planes of the #WID_C_SELECT_VIEW_BUILD_HQ selection widget. */
1996 CWP_VB_VIEW = 0, ///< Display the view button
1997 CWP_VB_BUILD, ///< Display the build button
1998
1999 /* Display planes of the #WID_C_SELECT_RELOCATE selection widget. */
2000 CWP_RELOCATE_SHOW = 0, ///< Show the relocate HQ button.
2001 CWP_RELOCATE_HIDE, ///< Hide the relocate HQ button.
2002
2003 /* Display planes of the #WID_C_SELECT_BUTTONS selection widget. */
2004 CWP_BUTTONS_LOCAL = 0, ///< Buttons of the local company.
2005 CWP_BUTTONS_OTHER, ///< Buttons of the other companies.
2006 };
2007
2008 CompanyWindow(WindowDesc *desc, WindowNumber window_number) : Window(desc)
2009 {
2010 this->InitNested(window_number);
2011 this->owner = (Owner)this->window_number;
2012 this->OnInvalidateData();
2013 }
2014
2015 virtual void OnPaint()
2016 {
2017 const Company *c = Company::Get((CompanyID)this->window_number);
2018 bool local = this->window_number == _local_company;
2019
2020 if (!this->IsShaded()) {
2021 bool reinit = false;
2022
2023 /* Button bar selection. */
2024 int plane = local ? CWP_BUTTONS_LOCAL : CWP_BUTTONS_OTHER;
2025 NWidgetStacked *wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_BUTTONS);
2026 if (plane != wi->shown_plane) {
2027 wi->SetDisplayedPlane(plane);
2028 this->InvalidateData();
2029 return;
2030 }
2031
2032 /* Build HQ button handling. */
2033 plane = (local && c->location_of_HQ == INVALID_TILE) ? CWP_VB_BUILD : CWP_VB_VIEW;
2034 wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_VIEW_BUILD_HQ);
2035 if (plane != wi->shown_plane) {
2036 wi->SetDisplayedPlane(plane);
2037 this->SetDirty();
2038 return;
2039 }
2040
2041 this->SetWidgetDisabledState(WID_C_VIEW_HQ, c->location_of_HQ == INVALID_TILE);
2042
2043 /* Enable/disable 'Relocate HQ' button. */
2044 plane = (!local || c->location_of_HQ == INVALID_TILE) ? CWP_RELOCATE_HIDE : CWP_RELOCATE_SHOW;
2045 wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_RELOCATE);
2046 if (plane != wi->shown_plane) {
2047 wi->SetDisplayedPlane(plane);
2048 this->SetDirty();
2049 return;
2050 }
2051
2052 /* Owners of company */
2053 plane = SZSP_HORIZONTAL;
2054 for (uint i = 0; i < lengthof(c->share_owners); i++) {
2055 if (c->share_owners[i] != INVALID_COMPANY) {
2056 plane = 0;
2057 break;
2058 }
2059 }
2060 wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_DESC_OWNERS);
2061 if (plane != wi->shown_plane) {
2062 wi->SetDisplayedPlane(plane);
2063 reinit = true;
2064 }
2065
2066 /* Multiplayer buttons. */
2067 plane = ((!_networking) ? (int)SZSP_NONE : (int)(local ? CWP_MP_C_PWD : CWP_MP_C_JOIN));
2068 wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_MULTIPLAYER);
2069 if (plane != wi->shown_plane) {
2070 wi->SetDisplayedPlane(plane);
2071 reinit = true;
2072 }
2073 this->SetWidgetDisabledState(WID_C_COMPANY_JOIN, c->is_ai);
2074
2075 if (reinit) {
2076 this->ReInit();
2077 return;
2078 }
2079 }
2080
2081 this->DrawWidgets();
2082 }
2083
2084 virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
2085 {
2086 switch (widget) {
2087 case WID_C_FACE: {
2088 Dimension face_size = GetSpriteSize(SPR_GRADIENT);
2089 size->width = max(size->width, face_size.width);
2090 size->height = max(size->height, face_size.height);
2091 break;
2092 }
2093
2094 case WID_C_DESC_COLOUR_SCHEME_EXAMPLE: {
2095 Point offset;
2096 Dimension d = GetSpriteSize(SPR_VEH_BUS_SW_VIEW, &offset);
2097 d.width -= offset.x;
2098 d.height -= offset.y;
2099 *size = maxdim(*size, d);
2100 break;
2101 }
2102
2103 case WID_C_DESC_COMPANY_VALUE:
2104 SetDParam(0, INT64_MAX); // Arguably the maximum company value
2105 size->width = GetStringBoundingBox(STR_COMPANY_VIEW_COMPANY_VALUE).width;
2106 break;
2107
2108 case WID_C_DESC_VEHICLE_COUNTS:
2109 SetDParamMaxValue(0, 5000); // Maximum number of vehicles
2110 for (uint i = 0; i < lengthof(_company_view_vehicle_count_strings); i++) {
2111 size->width = max(size->width, GetStringBoundingBox(_company_view_vehicle_count_strings[i]).width);
2112 }
2113 break;
2114
2115 case WID_C_DESC_INFRASTRUCTURE_COUNTS:
2116 SetDParamMaxValue(0, UINT_MAX);
2117 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_RAIL).width);
2118 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_ROAD).width);
2119 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_WATER).width);
2120 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_STATION).width);
2121 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_AIRPORT).width);
2122 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_NONE).width);
2123 break;
2124
2125 case WID_C_DESC_OWNERS: {
2126 const Company *c2;
2127
2128 FOR_ALL_COMPANIES(c2) {
2129 SetDParamMaxValue(0, 75);
2130 SetDParam(1, c2->index);
2131
2132 size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_SHARES_OWNED_BY).width);
2133 }
2134 break;
2135 }
2136
2137 #ifdef ENABLE_NETWORK
2138 case WID_C_HAS_PASSWORD:
2139 *size = maxdim(*size, GetSpriteSize(SPR_LOCK));
2140 break;
2141 #endif /* ENABLE_NETWORK */
2142 }
2143 }
2144
2145 virtual void DrawWidget(const Rect &r, int widget) const
2146 {
2147 const Company *c = Company::Get((CompanyID)this->window_number);
2148 switch (widget) {
2149 case WID_C_FACE:
2150 DrawCompanyManagerFace(c->face, c->colour, r.left, r.top);
2151 break;
2152
2153 case WID_C_FACE_TITLE:
2154 SetDParam(0, c->index);
2155 DrawStringMultiLine(r.left, r.right, r.top, r.bottom, STR_COMPANY_VIEW_PRESIDENT_MANAGER_TITLE, TC_FROMSTRING, SA_HOR_CENTER);
2156 break;
2157
2158 case WID_C_DESC_COLOUR_SCHEME_EXAMPLE: {
2159 Point offset;
2160 Dimension d = GetSpriteSize(SPR_VEH_BUS_SW_VIEW, &offset);
2161 d.height -= offset.y;
2162 DrawSprite(SPR_VEH_BUS_SW_VIEW, COMPANY_SPRITE_COLOUR(c->index), r.left - offset.x, (r.top + r.bottom - d.height) / 2 - offset.y);
2163 break;
2164 }
2165
2166 case WID_C_DESC_VEHICLE_COUNTS: {
2167 uint amounts[4];
2168 amounts[0] = c->group_all[VEH_TRAIN].num_vehicle;
2169 amounts[1] = c->group_all[VEH_ROAD].num_vehicle;
2170 amounts[2] = c->group_all[VEH_SHIP].num_vehicle;
2171 amounts[3] = c->group_all[VEH_AIRCRAFT].num_vehicle;
2172
2173 int y = r.top;
2174 if (amounts[0] + amounts[1] + amounts[2] + amounts[3] == 0) {
2175 DrawString(r.left, r.right, y, STR_COMPANY_VIEW_VEHICLES_NONE);
2176 } else {
2177 assert_compile(lengthof(amounts) == lengthof(_company_view_vehicle_count_strings));
2178
2179 for (uint i = 0; i < lengthof(amounts); i++) {
2180 if (amounts[i] != 0) {
2181 SetDParam(0, amounts[i]);
2182 DrawString(r.left, r.right, y, _company_view_vehicle_count_strings[i]);
2183 y += FONT_HEIGHT_NORMAL;
2184 }
2185 }
2186 }
2187 break;
2188 }
2189
2190 case WID_C_DESC_INFRASTRUCTURE_COUNTS: {
2191 uint y = r.top;
2192
2193 /* Collect rail and road counts. */
2194 uint rail_pices = c->infrastructure.signal;
2195 uint road_pieces = 0;
2196 for (uint i = 0; i < lengthof(c->infrastructure.rail); i++) rail_pices += c->infrastructure.rail[i];
2197 for (uint i = 0; i < lengthof(c->infrastructure.road); i++) road_pieces += c->infrastructure.road[i];
2198
2199 if (rail_pices == 0 && road_pieces == 0 && c->infrastructure.water == 0 && c->infrastructure.station == 0 && c->infrastructure.airport == 0) {
2200 DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_NONE);
2201 } else {
2202 if (rail_pices != 0) {
2203 SetDParam(0, rail_pices);
2204 DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_RAIL);
2205 y += FONT_HEIGHT_NORMAL;
2206 }
2207 if (road_pieces != 0) {
2208 SetDParam(0, road_pieces);
2209 DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_ROAD);
2210 y += FONT_HEIGHT_NORMAL;
2211 }
2212 if (c->infrastructure.water != 0) {
2213 SetDParam(0, c->infrastructure.water);
2214 DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_WATER);
2215 y += FONT_HEIGHT_NORMAL;
2216 }
2217 if (c->infrastructure.station != 0) {
2218 SetDParam(0, c->infrastructure.station);
2219 DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_STATION);
2220 y += FONT_HEIGHT_NORMAL;
2221 }
2222 if (c->infrastructure.airport != 0) {
2223 SetDParam(0, c->infrastructure.airport);
2224 DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_AIRPORT);
2225 }
2226 }
2227
2228 break;
2229 }
2230
2231 case WID_C_DESC_OWNERS: {
2232 const Company *c2;
2233 uint y = r.top;
2234
2235 FOR_ALL_COMPANIES(c2) {
2236 uint amt = GetAmountOwnedBy(c, c2->index);
2237 if (amt != 0) {
2238 SetDParam(0, amt * 25);
2239 SetDParam(1, c2->index);
2240
2241 DrawString(r.left, r.right, y, STR_COMPANY_VIEW_SHARES_OWNED_BY);
2242 y += FONT_HEIGHT_NORMAL;
2243 }
2244 }
2245 break;
2246 }
2247
2248 #ifdef ENABLE_NETWORK
2249 case WID_C_HAS_PASSWORD:
2250 if (_networking && NetworkCompanyIsPassworded(c->index)) {
2251 DrawSprite(SPR_LOCK, PAL_NONE, r.left, r.top);
2252 }
2253 break;
2254 #endif /* ENABLE_NETWORK */
2255 }
2256 }
2257
2258 virtual void SetStringParameters(int widget) const
2259 {
2260 switch (widget) {
2261 case WID_C_CAPTION:
2262 SetDParam(0, (CompanyID)this->window_number);
2263 SetDParam(1, (CompanyID)this->window_number);
2264 break;
2265
2266 case WID_C_DESC_INAUGURATION:
2267 SetDParam(0, Company::Get((CompanyID)this->window_number)->inaugurated_year);
2268 break;
2269
2270 case WID_C_DESC_COMPANY_VALUE:
2271 SetDParam(0, CalculateCompanyValue(Company::Get((CompanyID)this->window_number)));
2272 break;
2273 }
2274 }
2275
2276 virtual void OnClick(Point pt, int widget, int click_count)
2277 {
2278 switch (widget) {
2279 case WID_C_NEW_FACE: DoSelectCompanyManagerFace(this); break;
2280
2281 case WID_C_COLOUR_SCHEME:
2282 if (BringWindowToFrontById(WC_COMPANY_COLOUR, this->window_number)) break;
2283 new SelectCompanyLiveryWindow(&_select_company_livery_desc, (CompanyID)this->window_number);
2284 break;
2285
2286 case WID_C_PRESIDENT_NAME:
2287 this->query_widget = WID_C_PRESIDENT_NAME;
2288 SetDParam(0, this->window_number);
2289 ShowQueryString(STR_PRESIDENT_NAME, STR_COMPANY_VIEW_PRESIDENT_S_NAME_QUERY_CAPTION, MAX_LENGTH_PRESIDENT_NAME_CHARS, this, CS_ALPHANUMERAL, QSF_ENABLE_DEFAULT | QSF_LEN_IN_CHARS);
2290 break;
2291
2292 case WID_C_COMPANY_NAME:
2293 this->query_widget = WID_C_COMPANY_NAME;
2294 SetDParam(0, this->window_number);
2295 ShowQueryString(STR_COMPANY_NAME, STR_COMPANY_VIEW_COMPANY_NAME_QUERY_CAPTION, MAX_LENGTH_COMPANY_NAME_CHARS, this, CS_ALPHANUMERAL, QSF_ENABLE_DEFAULT | QSF_LEN_IN_CHARS);
2296 break;
2297
2298 case WID_C_VIEW_HQ: {
2299 TileIndex tile = Company::Get((CompanyID)this->window_number)->location_of_HQ;
2300 if (_ctrl_pressed) {
2301 ShowExtraViewPortWindow(tile);
2302 } else {
2303 ScrollMainWindowToTile(tile);
2304 }
2305 break;
2306 }
2307
2308 case WID_C_BUILD_HQ:
2309 if ((byte)this->window_number != _local_company) return;
2310 if (this->IsWidgetLowered(WID_C_BUILD_HQ)) {
2311 ResetObjectToPlace();
2312 this->RaiseButtons();
2313 break;
2314 }
2315 SetObjectToPlaceWnd(SPR_CURSOR_HQ, PAL_NONE, HT_RECT, this);
2316 SetTileSelectSize(2, 2);
2317 this->LowerWidget(WID_C_BUILD_HQ);
2318 this->SetWidgetDirty(WID_C_BUILD_HQ);
2319 break;
2320
2321 case WID_C_RELOCATE_HQ:
2322 if (this->IsWidgetLowered(WID_C_RELOCATE_HQ)) {
2323 ResetObjectToPlace();
2324 this->RaiseButtons();
2325 break;
2326 }
2327 SetObjectToPlaceWnd(SPR_CURSOR_HQ, PAL_NONE, HT_RECT, this);
2328 SetTileSelectSize(2, 2);
2329 this->LowerWidget(WID_C_RELOCATE_HQ);
2330 this->SetWidgetDirty(WID_C_RELOCATE_HQ);
2331 break;
2332
2333 case WID_C_VIEW_INFRASTRUCTURE:
2334 ShowCompanyInfrastructure((CompanyID)this->window_number);
2335 break;
2336
2337 case WID_C_BUY_SHARE:
2338 DoCommandP(0, this->window_number, 0, CMD_BUY_SHARE_IN_COMPANY);
2339 break;
2340
2341 case WID_C_SELL_SHARE:
2342 DoCommandP(0, this->window_number, 0, CMD_SELL_SHARE_IN_COMPANY);
2343 break;
2344
2345 #ifdef ENABLE_NETWORK
2346 case WID_C_COMPANY_PASSWORD:
2347 if (this->window_number == _local_company) ShowNetworkCompanyPasswordWindow(this);
2348 break;
2349
2350 case WID_C_COMPANY_JOIN: {
2351 this->query_widget = WID_C_COMPANY_JOIN;
2352 CompanyID company = (CompanyID)this->window_number;
2353 if (_network_server) {
2354 NetworkServerDoMove(CLIENT_ID_SERVER, company);
2355 MarkWholeScreenDirty();
2356 } else if (NetworkCompanyIsPassworded(company)) {
2357 /* ask for the password */
2358 ShowQueryString(STR_EMPTY, STR_NETWORK_NEED_COMPANY_PASSWORD_CAPTION, NETWORK_PASSWORD_LENGTH, this, CS_ALPHANUMERAL, QSF_NONE);
2359 } else {
2360 /* just send the join command */
2361 NetworkClientRequestMove(company);
2362 }
2363 break;
2364 }
2365 #endif /* ENABLE_NETWORK */
2366 }
2367 }
2368
2369 virtual void OnHundredthTick()
2370 {
2371 /* redraw the window every now and then */
2372 this->SetDirty();
2373 }
2374
2375 virtual void OnPlaceObject(Point pt, TileIndex tile)
2376 {
2377 if (DoCommandP(tile, OBJECT_HQ, 0, CMD_BUILD_OBJECT)) {
2378 ResetObjectToPlace();
2379 this->RaiseButtons();
2380 }
2381 }
2382
2383 virtual void OnPlaceObjectAbort()
2384 {
2385 this->RaiseButtons();
2386 }
2387
2388 virtual void OnQueryTextFinished(char *str)
2389 {
2390 if (str == NULL) return;
2391
2392 switch (this->query_widget) {
2393 default: NOT_REACHED();
2394
2395 case WID_C_PRESIDENT_NAME:
2396 DoCommandP(0, 0, 0, CMD_RENAME_PRESIDENT, str);
2397 break;
2398
2399 case WID_C_COMPANY_NAME:
2400 DoCommandP(0, 0, 0, CMD_RENAME_COMPANY, str);
2401 break;
2402
2403 #ifdef ENABLE_NETWORK
2404 case WID_C_COMPANY_JOIN:
2405 NetworkClientRequestMove((CompanyID)this->window_number, str);
2406 break;
2407 #endif /* ENABLE_NETWORK */
2408 }
2409 }
2410
2411
2412 /**
2413 * Some data on this window has become invalid.
2414 * @param data Information about the changed data.
2415 * @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.
2416 */
2417 virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
2418 {
2419 if (this->window_number == _local_company) return;
2420
2421 if (_settings_game.economy.allow_shares) { // Shares are allowed
2422 const Company *c = Company::Get(this->window_number);
2423
2424 /* If all shares are owned by someone (none by nobody), disable buy button */
2425 this->SetWidgetDisabledState(WID_C_BUY_SHARE, GetAmountOwnedBy(c, INVALID_OWNER) == 0 ||
2426 /* Only 25% left to buy. If the company is human, disable buying it up.. TODO issues! */
2427 (GetAmountOwnedBy(c, INVALID_OWNER) == 1 && !c->is_ai) ||
2428 /* Spectators cannot do anything of course */
2429 _local_company == COMPANY_SPECTATOR);
2430
2431 /* If the company doesn't own any shares, disable sell button */
2432 this->SetWidgetDisabledState(WID_C_SELL_SHARE, (GetAmountOwnedBy(c, _local_company) == 0) ||
2433 /* Spectators cannot do anything of course */
2434 _local_company == COMPANY_SPECTATOR);
2435 } else { // Shares are not allowed, disable buy/sell buttons
2436 this->DisableWidget(WID_C_BUY_SHARE);
2437 this->DisableWidget(WID_C_SELL_SHARE);
2438 }
2439 }
2440 };
2441
2442 static WindowDesc _company_desc(
2443 WDP_AUTO, "company", 0, 0,
2444 WC_COMPANY, WC_NONE,
2445 0,
2446 _nested_company_widgets, lengthof(_nested_company_widgets)
2447 );
2448
2449 /**
2450 * Show the window with the overview of the company.
2451 * @param company The company to show the window for.
2452 */
2453 void ShowCompany(CompanyID company)
2454 {
2455 if (!Company::IsValidID(company)) return;
2456
2457 AllocateWindowDescFront<CompanyWindow>(&_company_desc, company);
2458 }
2459
2460 /**
2461 * Redraw all windows with company infrastructure counts.
2462 * @param company The company to redraw the windows of.
2463 */
2464 void DirtyCompanyInfrastructureWindows(CompanyID company)
2465 {
2466 SetWindowDirty(WC_COMPANY, company);
2467 SetWindowDirty(WC_COMPANY_INFRASTRUCTURE, company);
2468 }
2469
2470 struct BuyCompanyWindow : Window {
2471 BuyCompanyWindow(WindowDesc *desc, WindowNumber window_number) : Window(desc)
2472 {
2473 this->InitNested(window_number);
2474 }
2475
2476 virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
2477 {
2478 switch (widget) {
2479 case WID_BC_FACE:
2480 *size = GetSpriteSize(SPR_GRADIENT);
2481 break;
2482
2483 case WID_BC_QUESTION:
2484 const Company *c = Company::Get((CompanyID)this->window_number);
2485 SetDParam(0, c->index);
2486 SetDParam(1, c->bankrupt_value);
2487 size->height = GetStringHeight(STR_BUY_COMPANY_MESSAGE, size->width);
2488 break;
2489 }
2490 }
2491
2492 virtual void SetStringParameters(int widget) const
2493 {
2494 switch (widget) {
2495 case WID_BC_CAPTION:
2496 SetDParam(0, STR_COMPANY_NAME);
2497 SetDParam(1, Company::Get((CompanyID)this->window_number)->index);
2498 break;
2499 }
2500 }
2501
2502 virtual void DrawWidget(const Rect &r, int widget) const
2503 {
2504 switch (widget) {
2505 case WID_BC_FACE: {
2506 const Company *c = Company::Get((CompanyID)this->window_number);
2507 DrawCompanyManagerFace(c->face, c->colour, r.left, r.top);
2508 break;
2509 }
2510
2511 case WID_BC_QUESTION: {
2512 const Company *c = Company::Get((CompanyID)this->window_number);
2513 SetDParam(0, c->index);
2514 SetDParam(1, c->bankrupt_value);
2515 DrawStringMultiLine(r.left, r.right, r.top, r.bottom, STR_BUY_COMPANY_MESSAGE, TC_FROMSTRING, SA_CENTER);
2516 break;
2517 }
2518 }
2519 }
2520
2521 virtual void OnClick(Point pt, int widget, int click_count)
2522 {
2523 switch (widget) {
2524 case WID_BC_NO:
2525 delete this;
2526 break;
2527
2528 case WID_BC_YES:
2529 DoCommandP(0, this->window_number, 0, CMD_BUY_COMPANY);
2530 break;
2531 }
2532 }
2533 };
2534
2535 static const NWidgetPart _nested_buy_company_widgets[] = {
2536 NWidget(NWID_HORIZONTAL),
2537 NWidget(WWT_CLOSEBOX, COLOUR_LIGHT_BLUE),
2538 NWidget(WWT_CAPTION, COLOUR_LIGHT_BLUE, WID_BC_CAPTION), SetDataTip(STR_ERROR_MESSAGE_CAPTION_OTHER_COMPANY, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
2539 EndContainer(),
2540 NWidget(WWT_PANEL, COLOUR_LIGHT_BLUE),
2541 NWidget(NWID_VERTICAL), SetPIP(8, 8, 8),
2542 NWidget(NWID_HORIZONTAL), SetPIP(8, 10, 8),
2543 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_BC_FACE), SetFill(0, 1),
2544 NWidget(WWT_EMPTY, INVALID_COLOUR, WID_BC_QUESTION), SetMinimalSize(240, 0), SetFill(1, 1),
2545 EndContainer(),
2546 NWidget(NWID_HORIZONTAL, NC_EQUALSIZE), SetPIP(100, 10, 100),
2547 NWidget(WWT_TEXTBTN, COLOUR_LIGHT_BLUE, WID_BC_NO), SetMinimalSize(60, 12), SetDataTip(STR_QUIT_NO, STR_NULL), SetFill(1, 0),
2548 NWidget(WWT_TEXTBTN, COLOUR_LIGHT_BLUE, WID_BC_YES), SetMinimalSize(60, 12), SetDataTip(STR_QUIT_YES, STR_NULL), SetFill(1, 0),
2549 EndContainer(),
2550 EndContainer(),
2551 EndContainer(),
2552 };
2553
2554 static WindowDesc _buy_company_desc(
2555 WDP_AUTO, NULL, 0, 0,
2556 WC_BUY_COMPANY, WC_NONE,
2557 WDF_CONSTRUCTION,
2558 _nested_buy_company_widgets, lengthof(_nested_buy_company_widgets)
2559 );
2560
2561 /**
2562 * Show the query to buy another company.
2563 * @param company The company to buy.
2564 */
2565 void ShowBuyCompanyDialog(CompanyID company)
2566 {
2567 AllocateWindowDescFront<BuyCompanyWindow>(&_buy_company_desc, company);
2568 }