src/company_gui.cpp

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