company_gui.cpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * 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.
 * 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.
 * 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/>.
 */
 
#include "stdafx.h"
#include "currency.h"
#include "error.h"
#include "gui.h"
#include "window_gui.h"
#include "textbuf_gui.h"
#include "viewport_func.h"
#include "company_func.h"
#include "command_func.h"
#include "network/network.h"
#include "network/network_gui.h"
#include "network/network_func.h"
#include "newgrf.h"
#include "company_manager_face.h"
#include "strings_func.h"
#include "timer/timer_game_economy.h"
#include "dropdown_type.h"
#include "dropdown_common_type.h"
#include "tilehighlight_func.h"
#include "company_base.h"
#include "core/geometry_func.hpp"
#include "object_type.h"
#include "rail.h"
#include "road.h"
#include "engine_base.h"
#include "window_func.h"
#include "road_func.h"
#include "water.h"
#include "station_func.h"
#include "zoom_func.h"
#include "sortlist_type.h"
#include "company_cmd.h"
#include "economy_cmd.h"
#include "group_cmd.h"
#include "group_gui.h"
#include "misc_cmd.h"
#include "object_cmd.h"
#include "timer/timer.h"
#include "timer/timer_window.h"
 
#include "widgets/company_widget.h"
 
#include "safeguards.h"
 
 
static void DoSelectCompanyManagerFace(Window *parent);
static void ShowCompanyInfrastructure(CompanyID company);
 
static const std::initializer_list<ExpensesType> _expenses_list_revenue = {
  EXPENSES_TRAIN_REVENUE,
  EXPENSES_ROADVEH_REVENUE,
  EXPENSES_AIRCRAFT_REVENUE,
  EXPENSES_SHIP_REVENUE,
};
 
static const std::initializer_list<ExpensesType> _expenses_list_operating_costs = {
  EXPENSES_TRAIN_RUN,
  EXPENSES_ROADVEH_RUN,
  EXPENSES_AIRCRAFT_RUN,
  EXPENSES_SHIP_RUN,
  EXPENSES_PROPERTY,
  EXPENSES_LOAN_INTEREST,
};
 
static const std::initializer_list<ExpensesType> _expenses_list_capital_costs = {
  EXPENSES_CONSTRUCTION,
  EXPENSES_NEW_VEHICLES,
  EXPENSES_OTHER,
};
 
struct ExpensesList {
  const StringID title; 
  const std::initializer_list<ExpensesType> &items; 
 
  ExpensesList(StringID title, const std::initializer_list<ExpensesType> &list) : title(title), items(list)
  {
  }
 
  uint GetHeight() const
  {
    /* Add up the height of all the lines.  */
    return static_cast<uint>(this->items.size()) * GetCharacterHeight(FS_NORMAL);
  }
 
  uint GetListWidth() const
  {
    uint width = 0;
    for (const ExpensesType &et : this->items) {
      width = std::max(width, GetStringBoundingBox(STR_FINANCES_SECTION_CONSTRUCTION + et).width);
    }
    return width;
  }
};
 
static const std::initializer_list<ExpensesList> _expenses_list_types = {
  { STR_FINANCES_REVENUE_TITLE,            _expenses_list_revenue },
  { STR_FINANCES_OPERATING_EXPENSES_TITLE, _expenses_list_operating_costs },
  { STR_FINANCES_CAPITAL_EXPENSES_TITLE,   _expenses_list_capital_costs },
};
 
static uint GetTotalCategoriesHeight()
{
  /* There's an empty line and blockspace on the year row */
  uint total_height = GetCharacterHeight(FS_NORMAL) + WidgetDimensions::scaled.vsep_wide;
 
  for (const ExpensesList &list : _expenses_list_types) {
    /* Title + expense list + total line + total + blockspace after category */
    total_height += GetCharacterHeight(FS_NORMAL) + list.GetHeight() + WidgetDimensions::scaled.vsep_normal + GetCharacterHeight(FS_NORMAL) + WidgetDimensions::scaled.vsep_wide;
  }
 
  /* Total income */
  total_height += WidgetDimensions::scaled.vsep_normal + GetCharacterHeight(FS_NORMAL) + WidgetDimensions::scaled.vsep_wide;
 
  return total_height;
}
 
static uint GetMaxCategoriesWidth()
{
  uint max_width = GetStringBoundingBox(TimerGameEconomy::UsingWallclockUnits() ? STR_FINANCES_PERIOD_CAPTION : STR_FINANCES_YEAR_CAPTION).width;
 
  /* Loop through categories to check max widths. */
  for (const ExpensesList &list : _expenses_list_types) {
    /* Title of category */
    max_width = std::max(max_width, GetStringBoundingBox(list.title).width);
    /* Entries in category */
    max_width = std::max(max_width, list.GetListWidth() + WidgetDimensions::scaled.hsep_indent);
  }
 
  return max_width;
}
 
static void DrawCategory(const Rect &r, int start_y, const ExpensesList &list)
{
  Rect tr = r.Indent(WidgetDimensions::scaled.hsep_indent, _current_text_dir == TD_RTL);
 
  tr.top = start_y;
 
  for (const ExpensesType &et : list.items) {
    DrawString(tr, STR_FINANCES_SECTION_CONSTRUCTION + et);
    tr.top += GetCharacterHeight(FS_NORMAL);
  }
}
 
static void DrawCategories(const Rect &r)
{
  int y = r.top;
  /* Draw description of 12-minute economic period. */
  DrawString(r.left, r.right, y, (TimerGameEconomy::UsingWallclockUnits() ? STR_FINANCES_PERIOD_CAPTION : STR_FINANCES_YEAR_CAPTION), TC_FROMSTRING, SA_LEFT, true);
  y += GetCharacterHeight(FS_NORMAL) + WidgetDimensions::scaled.vsep_wide;
 
  for (const ExpensesList &list : _expenses_list_types) {
    /* Draw category title and advance y */
    DrawString(r.left, r.right, y, list.title, TC_FROMSTRING, SA_LEFT);
    y += GetCharacterHeight(FS_NORMAL);
 
    /* Draw category items and advance y */
    DrawCategory(r, y, list);
    y += list.GetHeight();
 
    /* Advance y by the height of the horizontal line between amounts and subtotal */
    y += WidgetDimensions::scaled.vsep_normal;
 
    /* Draw category total and advance y */
    DrawString(r.left, r.right, y, STR_FINANCES_TOTAL_CAPTION, TC_FROMSTRING, SA_RIGHT);
    y += GetCharacterHeight(FS_NORMAL);
 
    /* Advance y by a blockspace after this category block */
    y += WidgetDimensions::scaled.vsep_wide;
  }
 
  /* Draw total profit/loss */
  y += WidgetDimensions::scaled.vsep_normal;
  DrawString(r.left, r.right, y, STR_FINANCES_PROFIT, TC_FROMSTRING, SA_LEFT);
}
 
static void DrawPrice(Money amount, int left, int right, int top, TextColour colour)
{
  StringID str = STR_FINANCES_NEGATIVE_INCOME;
  if (amount == 0) {
    str = STR_FINANCES_ZERO_INCOME;
  } else if (amount < 0) {
    amount = -amount;
    str = STR_FINANCES_POSITIVE_INCOME;
  }
  DrawString(left, right, top, GetString(str, amount), colour, SA_RIGHT);
}
 
static Money DrawYearCategory(const Rect &r, int start_y, const ExpensesList &list, const Expenses &tbl)
{
  int y = start_y;
  Money sum = 0;
 
  for (const ExpensesType &et : list.items) {
    Money cost = tbl[et];
    sum += cost;
    if (cost != 0) DrawPrice(cost, r.left, r.right, y, TC_BLACK);
    y += GetCharacterHeight(FS_NORMAL);
  }
 
  /* Draw the total at the bottom of the category. */
  GfxFillRect(r.left, y, r.right, y + WidgetDimensions::scaled.bevel.top - 1, PC_BLACK);
  y += WidgetDimensions::scaled.vsep_normal;
  if (sum != 0) DrawPrice(sum, r.left, r.right, y, TC_WHITE);
 
  /* Return the sum for the yearly total. */
  return sum;
}
 
 
static void DrawYearColumn(const Rect &r, TimerGameEconomy::Year year, const Expenses &tbl)
{
  int y = r.top;
  Money sum;
 
  /* Year header */
  DrawString(r.left, r.right, y, GetString(STR_FINANCES_YEAR, year), TC_FROMSTRING, SA_RIGHT, true);
  y += GetCharacterHeight(FS_NORMAL) + WidgetDimensions::scaled.vsep_wide;
 
  /* Categories */
  for (const ExpensesList &list : _expenses_list_types) {
    y += GetCharacterHeight(FS_NORMAL);
    sum += DrawYearCategory(r, y, list, tbl);
    /* Expense list + expense category title + expense category total + blockspace after category */
    y += list.GetHeight() + WidgetDimensions::scaled.vsep_normal + GetCharacterHeight(FS_NORMAL) + WidgetDimensions::scaled.vsep_wide;
  }
 
  /* Total income. */
  GfxFillRect(r.left, y, r.right, y + WidgetDimensions::scaled.bevel.top - 1, PC_BLACK);
  y += WidgetDimensions::scaled.vsep_normal;
  DrawPrice(sum, r.left, r.right, y, TC_WHITE);
}
 
static constexpr NWidgetPart _nested_company_finances_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_CF_CAPTION),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_CF_TOGGLE_SIZE), SetSpriteTip(SPR_LARGE_SMALL_WINDOW, STR_TOOLTIP_TOGGLE_LARGE_SMALL_WINDOW), SetAspect(WidgetDimensions::ASPECT_TOGGLE_SIZE),
    NWidget(WWT_SHADEBOX, COLOUR_GREY),
    NWidget(WWT_STICKYBOX, COLOUR_GREY),
  EndContainer(),
  NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_PANEL),
    NWidget(WWT_PANEL, COLOUR_GREY),
      NWidget(NWID_HORIZONTAL), SetPadding(WidgetDimensions::unscaled.framerect), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CF_EXPS_CATEGORY), SetMinimalSize(120, 0), SetFill(0, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CF_EXPS_PRICE1), SetMinimalSize(86, 0), SetFill(0, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CF_EXPS_PRICE2), SetMinimalSize(86, 0), SetFill(0, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CF_EXPS_PRICE3), SetMinimalSize(86, 0), SetFill(0, 0),
      EndContainer(),
    EndContainer(),
  EndContainer(),
  NWidget(WWT_PANEL, COLOUR_GREY),
    NWidget(NWID_HORIZONTAL), SetPadding(WidgetDimensions::unscaled.framerect), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0), SetPIPRatio(0, 1, 2),
      NWidget(NWID_VERTICAL), // Vertical column with 'bank balance', 'loan'
        NWidget(WWT_TEXT, INVALID_COLOUR), SetStringTip(STR_FINANCES_OWN_FUNDS_TITLE),
        NWidget(WWT_TEXT, INVALID_COLOUR), SetStringTip(STR_FINANCES_LOAN_TITLE),
        NWidget(WWT_TEXT, INVALID_COLOUR), SetStringTip(STR_FINANCES_BANK_BALANCE_TITLE), SetPadding(WidgetDimensions::unscaled.vsep_normal, 0, 0, 0),
      EndContainer(),
      NWidget(NWID_VERTICAL), // Vertical column with bank balance amount, loan amount, and total.
        NWidget(WWT_TEXT, INVALID_COLOUR, WID_CF_OWN_VALUE), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
        NWidget(WWT_TEXT, INVALID_COLOUR, WID_CF_LOAN_VALUE), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CF_BALANCE_LINE), SetMinimalSize(0, WidgetDimensions::unscaled.vsep_normal),
        NWidget(WWT_TEXT, INVALID_COLOUR, WID_CF_BALANCE_VALUE), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
      EndContainer(),
      NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_MAXLOAN),
        NWidget(NWID_VERTICAL), SetPIPRatio(0, 0, 1), // Max loan information
          NWidget(WWT_TEXT, INVALID_COLOUR, WID_CF_INTEREST_RATE),
          NWidget(WWT_TEXT, INVALID_COLOUR, WID_CF_MAXLOAN_VALUE),
        EndContainer(),
      EndContainer(),
    EndContainer(),
  EndContainer(),
  NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_BUTTONS),
    NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
      NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_INCREASE_LOAN), SetFill(1, 0), SetToolTip(STR_FINANCES_BORROW_TOOLTIP),
      NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_REPAY_LOAN), SetFill(1, 0), SetToolTip(STR_FINANCES_REPAY_TOOLTIP),
      NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_INFRASTRUCTURE), SetFill(1, 0), SetStringTip(STR_FINANCES_INFRASTRUCTURE_BUTTON, STR_COMPANY_VIEW_INFRASTRUCTURE_TOOLTIP),
    EndContainer(),
  EndContainer(),
};
 
struct CompanyFinancesWindow : Window {
  static constexpr int NUM_PERIODS = WID_CF_EXPS_PRICE3 - WID_CF_EXPS_PRICE1 + 1;
 
  static Money max_money; 
  bool small = false; 
  uint8_t first_visible = NUM_PERIODS - 1; 
 
  CompanyFinancesWindow(WindowDesc &desc, CompanyID company) : Window(desc)
  {
    this->CreateNestedTree();
    this->SetupWidgets();
    this->FinishInitNested(company);
 
    this->owner = this->window_number;
    this->InvalidateData();
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    switch (widget) {
      case WID_CF_CAPTION:
        return GetString(STR_FINANCES_CAPTION, this->window_number, this->window_number);
 
      case WID_CF_BALANCE_VALUE: {
        const Company *c = Company::Get(this->window_number);
        return GetString(STR_FINANCES_BANK_BALANCE, c->money);
      }
 
      case WID_CF_LOAN_VALUE: {
        const Company *c = Company::Get(this->window_number);
        return GetString(STR_FINANCES_TOTAL_CURRENCY, c->current_loan);
      }
 
      case WID_CF_OWN_VALUE: {
        const Company *c = Company::Get(this->window_number);
        return GetString(STR_FINANCES_TOTAL_CURRENCY, c->money - c->current_loan);
      }
 
      case WID_CF_INTEREST_RATE:
        return GetString(STR_FINANCES_INTEREST_RATE, _settings_game.difficulty.initial_interest);
 
      case WID_CF_MAXLOAN_VALUE: {
        const Company *c = Company::Get(this->window_number);
        return GetString(STR_FINANCES_MAX_LOAN, c->GetMaxLoan());
      }
 
      case WID_CF_INCREASE_LOAN:
        return GetString(STR_FINANCES_BORROW_BUTTON, LOAN_INTERVAL);
 
      case WID_CF_REPAY_LOAN:
        return GetString(STR_FINANCES_REPAY_BUTTON, LOAN_INTERVAL);
 
      default:
        return this->Window::GetWidgetString(widget, stringid);
    }
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    switch (widget) {
      case WID_CF_EXPS_CATEGORY:
        size.width  = GetMaxCategoriesWidth();
        size.height = GetTotalCategoriesHeight();
        break;
 
      case WID_CF_EXPS_PRICE1:
      case WID_CF_EXPS_PRICE2:
      case WID_CF_EXPS_PRICE3:
        size.height = GetTotalCategoriesHeight();
        [[fallthrough]];
 
      case WID_CF_BALANCE_VALUE:
      case WID_CF_LOAN_VALUE:
      case WID_CF_OWN_VALUE: {
        uint64_t max_value = GetParamMaxValue(CompanyFinancesWindow::max_money);
        size.width = std::max(GetStringBoundingBox(GetString(STR_FINANCES_NEGATIVE_INCOME, max_value)).width, GetStringBoundingBox(GetString(STR_FINANCES_POSITIVE_INCOME, max_value)).width);
        size.width += padding.width;
        break;
      }
 
      case WID_CF_INTEREST_RATE:
        size.height = GetCharacterHeight(FS_NORMAL);
        break;
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    switch (widget) {
      case WID_CF_EXPS_CATEGORY:
        DrawCategories(r);
        break;
 
      case WID_CF_EXPS_PRICE1:
      case WID_CF_EXPS_PRICE2:
      case WID_CF_EXPS_PRICE3: {
        int period = widget - WID_CF_EXPS_PRICE1;
        if (period < this->first_visible) break;
 
        const Company *c = Company::Get(this->window_number);
        const auto &expenses = c->yearly_expenses[NUM_PERIODS - period - 1];
        DrawYearColumn(r, TimerGameEconomy::year - (NUM_PERIODS - period - 1), expenses);
        break;
      }
 
      case WID_CF_BALANCE_LINE:
        GfxFillRect(r.left, r.top, r.right, r.top + WidgetDimensions::scaled.bevel.top - 1, PC_BLACK);
        break;
    }
  }
 
  void SetupWidgets()
  {
    int plane = this->small ? SZSP_NONE : 0;
    this->GetWidget<NWidgetStacked>(WID_CF_SEL_PANEL)->SetDisplayedPlane(plane);
    this->GetWidget<NWidgetStacked>(WID_CF_SEL_MAXLOAN)->SetDisplayedPlane(plane);
 
    CompanyID company = this->window_number;
    plane = (company != _local_company) ? SZSP_NONE : 0;
    this->GetWidget<NWidgetStacked>(WID_CF_SEL_BUTTONS)->SetDisplayedPlane(plane);
  }
 
  void OnPaint() override
  {
    if (!this->IsShaded()) {
      if (!this->small) {
        /* Check that the expenses panel height matches the height needed for the layout. */
        if (GetTotalCategoriesHeight() != this->GetWidget<NWidgetBase>(WID_CF_EXPS_CATEGORY)->current_y) {
          this->SetupWidgets();
          this->ReInit();
          return;
        }
      }
 
      /* Check that the loan buttons are shown only when the user owns the company. */
      CompanyID company = this->window_number;
      int req_plane = (company != _local_company) ? SZSP_NONE : 0;
      if (req_plane != this->GetWidget<NWidgetStacked>(WID_CF_SEL_BUTTONS)->shown_plane) {
        this->SetupWidgets();
        this->ReInit();
        return;
      }
 
      const Company *c = Company::Get(company);
      this->SetWidgetDisabledState(WID_CF_INCREASE_LOAN, c->current_loan >= c->GetMaxLoan()); // Borrow button only shows when there is any more money to loan.
      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.
    }
 
    this->DrawWidgets();
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      case WID_CF_TOGGLE_SIZE: // toggle size
        this->small = !this->small;
        this->SetupWidgets();
        if (this->IsShaded()) {
          /* Finances window is not resizable, so size hints given during unshading have no effect
           * on the changed appearance of the window. */
          this->SetShaded(false);
        } else {
          this->ReInit();
        }
        break;
 
      case WID_CF_INCREASE_LOAN: // increase loan
        Command<CMD_INCREASE_LOAN>::Post(STR_ERROR_CAN_T_BORROW_ANY_MORE_MONEY, _ctrl_pressed ? LoanCommand::Max : LoanCommand::Interval, 0);
        break;
 
      case WID_CF_REPAY_LOAN: // repay loan
        Command<CMD_DECREASE_LOAN>::Post(STR_ERROR_CAN_T_REPAY_LOAN, _ctrl_pressed ? LoanCommand::Max : LoanCommand::Interval, 0);
        break;
 
      case WID_CF_INFRASTRUCTURE: // show infrastructure details
        ShowCompanyInfrastructure(this->window_number);
        break;
    }
  }
 
  void RefreshVisibleColumns()
  {
    for (uint period = 0; period < this->first_visible; ++period) {
      const Company *c = Company::Get(this->window_number);
      const Expenses &expenses = c->yearly_expenses[NUM_PERIODS - period - 1];
      /* Show expenses column if it has any non-zero value in it. */
      if (std::ranges::any_of(expenses, [](const Money &value) { return value != 0; })) {
        this->first_visible = period;
        break;
      }
    }
  }
 
  void OnInvalidateData(int, bool) override
  {
    this->RefreshVisibleColumns();
  }
 
  IntervalTimer<TimerWindow> rescale_interval = {std::chrono::seconds(3), [this](auto) {
    const Company *c = Company::Get(this->window_number);
    if (c->money > CompanyFinancesWindow::max_money) {
      CompanyFinancesWindow::max_money = std::max(c->money * 2, CompanyFinancesWindow::max_money * 4);
      this->SetupWidgets();
      this->ReInit();
    }
  }};
};
 
Money CompanyFinancesWindow::max_money = INT32_MAX;
 
static WindowDesc _company_finances_desc(
  WDP_AUTO, "company_finances", 0, 0,
  WC_FINANCES, WC_NONE,
  {},
  _nested_company_finances_widgets
);
 
void ShowCompanyFinances(CompanyID company)
{
  if (!Company::IsValidID(company)) return;
  if (BringWindowToFrontById(WC_FINANCES, company)) return;
 
  new CompanyFinancesWindow(_company_finances_desc, company);
}
 
/* Association of liveries to livery classes */
static const LiveryClass _livery_class[LS_END] = {
  LC_OTHER,
  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,
  LC_ROAD, LC_ROAD,
  LC_SHIP, LC_SHIP,
  LC_AIRCRAFT, LC_AIRCRAFT, LC_AIRCRAFT,
  LC_ROAD, LC_ROAD,
};
 
template <SpriteID TSprite = SPR_SQUARE>
class DropDownListColourItem : public DropDownIcon<DropDownString<DropDownListItem>> {
public:
  DropDownListColourItem(int colour, bool masked) : DropDownIcon<DropDownString<DropDownListItem>>(TSprite, GENERAL_SPRITE_COLOUR(colour % COLOUR_END), GetString(colour < COLOUR_END ? (STR_COLOUR_DARK_BLUE + colour) : STR_COLOUR_DEFAULT), colour, masked)
  {
  }
};
 
struct SelectCompanyLiveryWindow : public Window {
private:
  uint32_t sel = 0;
  LiveryClass livery_class{};
  Dimension square{};
  uint rows = 0;
  uint line_height = 0;
  GUIGroupList groups{};
  Scrollbar *vscroll = nullptr;
 
  void ShowColourDropDownMenu(uint32_t widget)
  {
    uint32_t used_colours = 0;
    const Livery *livery, *default_livery = nullptr;
    bool primary = widget == WID_SCL_PRI_COL_DROPDOWN;
    uint8_t default_col = 0;
 
    /* Disallow other company colours for the primary colour */
    if (this->livery_class < LC_GROUP_RAIL && HasBit(this->sel, LS_DEFAULT) && primary) {
      for (const Company *c : Company::Iterate()) {
        if (c->index != _local_company) SetBit(used_colours, c->colour);
      }
    }
 
    const Company *c = Company::Get(this->window_number);
 
    if (this->livery_class < LC_GROUP_RAIL) {
      /* Get the first selected livery to use as the default dropdown item */
      LiveryScheme scheme;
      for (scheme = LS_BEGIN; scheme < LS_END; scheme++) {
        if (HasBit(this->sel, scheme)) break;
      }
      if (scheme == LS_END) scheme = LS_DEFAULT;
      livery = &c->livery[scheme];
      if (scheme != LS_DEFAULT) default_livery = &c->livery[LS_DEFAULT];
    } else {
      const Group *g = Group::Get(this->sel);
      livery = &g->livery;
      if (g->parent == GroupID::Invalid()) {
        default_livery = &c->livery[LS_DEFAULT];
      } else {
        const Group *pg = Group::Get(g->parent);
        default_livery = &pg->livery;
      }
    }
 
    DropDownList list;
    if (default_livery != nullptr) {
      /* Add COLOUR_END to put the colour out of range, but also allow us to show what the default is */
      default_col = (primary ? default_livery->colour1 : default_livery->colour2) + COLOUR_END;
      list.push_back(std::make_unique<DropDownListColourItem<>>(default_col, false));
    }
    for (Colours colour = COLOUR_BEGIN; colour != COLOUR_END; colour++) {
      list.push_back(std::make_unique<DropDownListColourItem<>>(colour, HasBit(used_colours, colour)));
    }
 
    uint8_t sel;
    if (default_livery == nullptr || HasBit(livery->in_use, primary ? 0 : 1)) {
      sel = primary ? livery->colour1 : livery->colour2;
    } else {
      sel = default_col;
    }
    ShowDropDownList(this, std::move(list), sel, widget);
  }
 
  void BuildGroupList(CompanyID owner)
  {
    if (!this->groups.NeedRebuild()) return;
 
    this->groups.clear();
 
    if (this->livery_class >= LC_GROUP_RAIL) {
      VehicleType vtype = (VehicleType)(this->livery_class - LC_GROUP_RAIL);
      BuildGuiGroupList(this->groups, false, owner, vtype);
    }
 
    this->groups.RebuildDone();
  }
 
  void SetRows()
  {
    if (this->livery_class < LC_GROUP_RAIL) {
      this->rows = 0;
      for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
        if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
          this->rows++;
        }
      }
    } else {
      this->rows = (uint)this->groups.size();
    }
 
    this->vscroll->SetCount(this->rows);
  }
 
public:
  SelectCompanyLiveryWindow(WindowDesc &desc, CompanyID company, GroupID group) : Window(desc)
  {
    this->CreateNestedTree();
    this->vscroll = this->GetScrollbar(WID_SCL_MATRIX_SCROLLBAR);
 
    if (group == GroupID::Invalid()) {
      this->livery_class = LC_OTHER;
      this->sel = 1;
      this->LowerWidget(WID_SCL_CLASS_GENERAL);
      this->BuildGroupList(company);
      this->SetRows();
    } else {
      this->SetSelectedGroup(company, group);
    }
 
    this->FinishInitNested(company);
    this->owner = company;
    this->InvalidateData(1);
  }
 
  void SetSelectedGroup(CompanyID company, GroupID group)
  {
    this->RaiseWidget(WID_SCL_CLASS_GENERAL + this->livery_class);
    const Group *g = Group::Get(group);
    switch (g->vehicle_type) {
      case VEH_TRAIN: this->livery_class = LC_GROUP_RAIL; break;
      case VEH_ROAD: this->livery_class = LC_GROUP_ROAD; break;
      case VEH_SHIP: this->livery_class = LC_GROUP_SHIP; break;
      case VEH_AIRCRAFT: this->livery_class = LC_GROUP_AIRCRAFT; break;
      default: NOT_REACHED();
    }
    this->sel = group.base();
    this->LowerWidget(WID_SCL_CLASS_GENERAL + this->livery_class);
 
    this->groups.ForceRebuild();
    this->BuildGroupList(company);
    this->SetRows();
 
    /* Position scrollbar to selected group */
    for (uint i = 0; i < this->rows; i++) {
      if (this->groups[i].group->index == sel) {
        this->vscroll->SetPosition(i - this->vscroll->GetCapacity() / 2);
        break;
      }
    }
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    switch (widget) {
      case WID_SCL_SPACER_DROPDOWN: {
        /* The matrix widget below needs enough room to print all the schemes. */
        Dimension d = {0, 0};
        for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
          d = maxdim(d, GetStringBoundingBox(STR_LIVERY_DEFAULT + scheme));
        }
 
        /* And group names */
        for (const Group *g : Group::Iterate()) {
          if (g->owner == this->window_number) {
            d = maxdim(d, GetStringBoundingBox(GetString(STR_GROUP_NAME, g->index)));
          }
        }
 
        size.width = std::max(size.width, 5 + d.width + padding.width);
        break;
      }
 
      case WID_SCL_MATRIX: {
        /* 11 items in the default rail class */
        this->square = GetSpriteSize(SPR_SQUARE);
        this->line_height = std::max(this->square.height, (uint)GetCharacterHeight(FS_NORMAL)) + padding.height;
 
        size.height = 5 * this->line_height;
        resize.width = 1;
        resize.height = this->line_height;
        break;
      }
 
      case WID_SCL_SEC_COL_DROPDOWN:
        if (!_loaded_newgrf_features.has_2CC) {
          size.width = 0;
          break;
        }
        [[fallthrough]];
 
      case WID_SCL_PRI_COL_DROPDOWN: {
        this->square = GetSpriteSize(SPR_SQUARE);
        int string_padding = this->square.width + WidgetDimensions::scaled.hsep_normal + padding.width;
        for (Colours colour = COLOUR_BEGIN; colour != COLOUR_END; colour++) {
          size.width = std::max(size.width, GetStringBoundingBox(STR_COLOUR_DARK_BLUE + colour).width + string_padding);
        }
        size.width = std::max(size.width, GetStringBoundingBox(STR_COLOUR_DEFAULT).width + string_padding);
        break;
      }
    }
  }
 
  void OnPaint() override
  {
    bool local = this->window_number == _local_company;
 
    /* Disable dropdown controls if no scheme is selected */
    bool disabled = this->livery_class < LC_GROUP_RAIL ? (this->sel == 0) : (this->sel == GroupID::Invalid());
    this->SetWidgetDisabledState(WID_SCL_PRI_COL_DROPDOWN, !local || disabled);
    this->SetWidgetDisabledState(WID_SCL_SEC_COL_DROPDOWN, !local || disabled);
 
    this->BuildGroupList(this->window_number);
 
    this->DrawWidgets();
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    switch (widget) {
      case WID_SCL_CAPTION:
        return GetString(STR_LIVERY_CAPTION, this->window_number);
 
      case WID_SCL_PRI_COL_DROPDOWN:
      case WID_SCL_SEC_COL_DROPDOWN: {
        const Company *c = Company::Get(this->window_number);
        bool primary = widget == WID_SCL_PRI_COL_DROPDOWN;
        StringID colour = STR_COLOUR_DEFAULT;
 
        if (this->livery_class < LC_GROUP_RAIL) {
          if (this->sel != 0) {
            LiveryScheme scheme = LS_DEFAULT;
            for (scheme = LS_BEGIN; scheme < LS_END; scheme++) {
              if (HasBit(this->sel, scheme)) break;
            }
            if (scheme == LS_END) scheme = LS_DEFAULT;
            const Livery *livery = &c->livery[scheme];
            if (scheme == LS_DEFAULT || HasBit(livery->in_use, primary ? 0 : 1)) {
              colour = STR_COLOUR_DARK_BLUE + (primary ? livery->colour1 : livery->colour2);
            }
          }
        } else {
          if (this->sel != GroupID::Invalid()) {
            const Group *g = Group::Get(this->sel);
            const Livery *livery = &g->livery;
            if (HasBit(livery->in_use, primary ? 0 : 1)) {
              colour = STR_COLOUR_DARK_BLUE + (primary ? livery->colour1 : livery->colour2);
            }
          }
        }
        return GetString(colour);
      }
 
      default:
        return this->Window::GetWidgetString(widget, stringid);
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    if (widget != WID_SCL_MATRIX) return;
 
    bool rtl = _current_text_dir == TD_RTL;
 
    /* Coordinates of scheme name column. */
    const NWidgetBase *nwi = this->GetWidget<NWidgetBase>(WID_SCL_SPACER_DROPDOWN);
    Rect sch = nwi->GetCurrentRect().Shrink(WidgetDimensions::scaled.framerect);
    /* Coordinates of first dropdown. */
    nwi = this->GetWidget<NWidgetBase>(WID_SCL_PRI_COL_DROPDOWN);
    Rect pri = nwi->GetCurrentRect().Shrink(WidgetDimensions::scaled.framerect);
    /* Coordinates of second dropdown. */
    nwi = this->GetWidget<NWidgetBase>(WID_SCL_SEC_COL_DROPDOWN);
    Rect sec = nwi->GetCurrentRect().Shrink(WidgetDimensions::scaled.framerect);
 
    Rect pri_squ = pri.WithWidth(this->square.width, rtl);
    Rect sec_squ = sec.WithWidth(this->square.width, rtl);
 
    pri = pri.Indent(this->square.width + WidgetDimensions::scaled.hsep_normal, rtl);
    sec = sec.Indent(this->square.width + WidgetDimensions::scaled.hsep_normal, rtl);
 
    Rect ir = r.WithHeight(this->resize.step_height).Shrink(WidgetDimensions::scaled.matrix);
    int square_offs = (ir.Height() - this->square.height) / 2;
    int text_offs   = (ir.Height() - GetCharacterHeight(FS_NORMAL)) / 2;
 
    int y = ir.top;
 
    /* Helper function to draw livery info. */
    auto draw_livery = [&](std::string_view str, const Livery &livery, bool is_selected, bool is_default_scheme, int indent) {
      /* Livery Label. */
      DrawString(sch.left + (rtl ? 0 : indent), sch.right - (rtl ? indent : 0), y + text_offs, str, is_selected ? TC_WHITE : TC_BLACK);
 
      /* Text below the first dropdown. */
      DrawSprite(SPR_SQUARE, GENERAL_SPRITE_COLOUR(livery.colour1), pri_squ.left, y + square_offs);
      DrawString(pri.left, pri.right, y + text_offs, (is_default_scheme || HasBit(livery.in_use, 0)) ? STR_COLOUR_DARK_BLUE + livery.colour1 : STR_COLOUR_DEFAULT, is_selected ? TC_WHITE : TC_GOLD);
 
      /* Text below the second dropdown. */
      if (sec.right > sec.left) { // Second dropdown has non-zero size.
        DrawSprite(SPR_SQUARE, GENERAL_SPRITE_COLOUR(livery.colour2), sec_squ.left, y + square_offs);
        DrawString(sec.left, sec.right, y + text_offs, (is_default_scheme || HasBit(livery.in_use, 1)) ? STR_COLOUR_DARK_BLUE + livery.colour2 : STR_COLOUR_DEFAULT, is_selected ? TC_WHITE : TC_GOLD);
      }
 
      y += this->line_height;
    };
 
    const Company *c = Company::Get(this->window_number);
 
    if (livery_class < LC_GROUP_RAIL) {
      int pos = this->vscroll->GetPosition();
      for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
        if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
          if (pos-- > 0) continue;
          draw_livery(GetString(STR_LIVERY_DEFAULT + scheme), c->livery[scheme], HasBit(this->sel, scheme), scheme == LS_DEFAULT, 0);
        }
      }
    } else {
      auto [first, last] = this->vscroll->GetVisibleRangeIterators(this->groups);
      for (auto it = first; it != last; ++it) {
        const Group *g = it->group;
        draw_livery(GetString(STR_GROUP_NAME, g->index), g->livery, this->sel == g->index, false, it->indent * WidgetDimensions::scaled.hsep_indent);
      }
 
      if (this->vscroll->GetCount() == 0) {
        const StringID empty_labels[] = { STR_LIVERY_TRAIN_GROUP_EMPTY, STR_LIVERY_ROAD_VEHICLE_GROUP_EMPTY, STR_LIVERY_SHIP_GROUP_EMPTY, STR_LIVERY_AIRCRAFT_GROUP_EMPTY };
        VehicleType vtype = (VehicleType)(this->livery_class - LC_GROUP_RAIL);
        DrawString(ir.left, ir.right, y + text_offs, empty_labels[vtype], TC_BLACK);
      }
    }
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      /* Livery Class buttons */
      case WID_SCL_CLASS_GENERAL:
      case WID_SCL_CLASS_RAIL:
      case WID_SCL_CLASS_ROAD:
      case WID_SCL_CLASS_SHIP:
      case WID_SCL_CLASS_AIRCRAFT:
      case WID_SCL_GROUPS_RAIL:
      case WID_SCL_GROUPS_ROAD:
      case WID_SCL_GROUPS_SHIP:
      case WID_SCL_GROUPS_AIRCRAFT:
        this->RaiseWidget(WID_SCL_CLASS_GENERAL + this->livery_class);
        this->livery_class = (LiveryClass)(widget - WID_SCL_CLASS_GENERAL);
        this->LowerWidget(WID_SCL_CLASS_GENERAL + this->livery_class);
 
        /* Select the first item in the list */
        if (this->livery_class < LC_GROUP_RAIL) {
          this->sel = 0;
          for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
            if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
              this->sel = 1 << scheme;
              break;
            }
          }
        } else {
          this->sel = GroupID::Invalid().base();
          this->groups.ForceRebuild();
          this->BuildGroupList(this->window_number);
 
          if (!this->groups.empty()) {
            this->sel = this->groups[0].group->index.base();
          }
        }
 
        this->SetRows();
        this->SetDirty();
        break;
 
      case WID_SCL_PRI_COL_DROPDOWN: // First colour dropdown
        ShowColourDropDownMenu(WID_SCL_PRI_COL_DROPDOWN);
        break;
 
      case WID_SCL_SEC_COL_DROPDOWN: // Second colour dropdown
        ShowColourDropDownMenu(WID_SCL_SEC_COL_DROPDOWN);
        break;
 
      case WID_SCL_MATRIX: {
        if (this->livery_class < LC_GROUP_RAIL) {
          uint row = this->vscroll->GetScrolledRowFromWidget(pt.y, this, widget);
          if (row >= this->rows) return;
 
          LiveryScheme j = (LiveryScheme)row;
 
          for (LiveryScheme scheme = LS_BEGIN; scheme <= j && scheme < LS_END; scheme++) {
            if (_livery_class[scheme] != this->livery_class || !HasBit(_loaded_newgrf_features.used_liveries, scheme)) j++;
          }
          assert(j < LS_END);
 
          if (_ctrl_pressed) {
            ToggleBit(this->sel, j);
          } else {
            this->sel = 1 << j;
          }
        } else {
          auto it = this->vscroll->GetScrolledItemFromWidget(this->groups, pt.y, this, widget);
          if (it == std::end(this->groups)) return;
 
          this->sel = it->group->index.base();
        }
        this->SetDirty();
        break;
      }
    }
  }
 
  void OnResize() override
  {
    this->vscroll->SetCapacityFromWidget(this, WID_SCL_MATRIX);
  }
 
  void OnDropdownSelect(WidgetID widget, int index) override
  {
    bool local = this->window_number == _local_company;
    if (!local) return;
 
    Colours colour = static_cast<Colours>(index);
    if (colour >= COLOUR_END) colour = INVALID_COLOUR;
 
    if (this->livery_class < LC_GROUP_RAIL) {
      /* Set company colour livery */
      for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
        /* Changed colour for the selected scheme, or all visible schemes if CTRL is pressed. */
        if (HasBit(this->sel, scheme) || (_ctrl_pressed && _livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme))) {
          Command<CMD_SET_COMPANY_COLOUR>::Post(scheme, widget == WID_SCL_PRI_COL_DROPDOWN, colour);
        }
      }
    } else {
      /* Setting group livery */
      Command<CMD_SET_GROUP_LIVERY>::Post(static_cast<GroupID>(this->sel), widget == WID_SCL_PRI_COL_DROPDOWN, colour);
    }
  }
 
  void OnInvalidateData([[maybe_unused]] int data = 0, [[maybe_unused]] bool gui_scope = true) override
  {
    if (!gui_scope) return;
 
    if (data != -1) {
      /* data contains a VehicleType, rebuild list if it displayed */
      if (this->livery_class == data + LC_GROUP_RAIL) {
        this->groups.ForceRebuild();
        this->BuildGroupList(this->window_number);
        this->SetRows();
 
        if (!Group::IsValidID(this->sel)) {
          this->sel = GroupID::Invalid().base();
          if (!this->groups.empty()) this->sel = this->groups[0].group->index.base();
        }
 
        this->SetDirty();
      }
      return;
    }
 
    this->SetWidgetsDisabledState(true, WID_SCL_CLASS_RAIL, WID_SCL_CLASS_ROAD, WID_SCL_CLASS_SHIP, WID_SCL_CLASS_AIRCRAFT);
 
    bool current_class_valid = this->livery_class == LC_OTHER || this->livery_class >= LC_GROUP_RAIL;
    if (_settings_client.gui.liveries == LIT_ALL || (_settings_client.gui.liveries == LIT_COMPANY && this->window_number == _local_company)) {
      for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
        if (HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
          if (_livery_class[scheme] == this->livery_class) current_class_valid = true;
          this->EnableWidget(WID_SCL_CLASS_GENERAL + _livery_class[scheme]);
        } else if (this->livery_class < LC_GROUP_RAIL) {
          ClrBit(this->sel, scheme);
        }
      }
    }
 
    if (!current_class_valid) {
      Point pt = {0, 0};
      this->OnClick(pt, WID_SCL_CLASS_GENERAL, 1);
    }
  }
};
 
static constexpr NWidgetPart _nested_select_company_livery_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_SCL_CAPTION),
    NWidget(WWT_SHADEBOX, COLOUR_GREY),
    NWidget(WWT_DEFSIZEBOX, COLOUR_GREY),
    NWidget(WWT_STICKYBOX, COLOUR_GREY),
  EndContainer(),
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_GENERAL), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_IMG_COMPANY_GENERAL, STR_LIVERY_GENERAL_TOOLTIP),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_RAIL), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_IMG_TRAINLIST, STR_LIVERY_TRAIN_TOOLTIP),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_ROAD), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_IMG_TRUCKLIST, STR_LIVERY_ROAD_VEHICLE_TOOLTIP),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_SHIP), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_IMG_SHIPLIST, STR_LIVERY_SHIP_TOOLTIP),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_AIRCRAFT), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_IMG_AIRPLANESLIST, STR_LIVERY_AIRCRAFT_TOOLTIP),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_GROUPS_RAIL), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_GROUP_LIVERY_TRAIN, STR_LIVERY_TRAIN_GROUP_TOOLTIP),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_GROUPS_ROAD), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_GROUP_LIVERY_ROADVEH, STR_LIVERY_ROAD_VEHICLE_GROUP_TOOLTIP),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_GROUPS_SHIP), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_GROUP_LIVERY_SHIP, STR_LIVERY_SHIP_GROUP_TOOLTIP),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_GROUPS_AIRCRAFT), SetMinimalSize(22, 22), SetFill(0, 1), SetSpriteTip(SPR_GROUP_LIVERY_AIRCRAFT, STR_LIVERY_AIRCRAFT_GROUP_TOOLTIP),
    NWidget(WWT_PANEL, COLOUR_GREY), SetFill(1, 1), SetResize(1, 0), EndContainer(),
  EndContainer(),
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_MATRIX, COLOUR_GREY, WID_SCL_MATRIX), SetMinimalSize(275, 0), SetResize(1, 0), SetFill(1, 1), SetMatrixDataTip(1, 0, STR_LIVERY_PANEL_TOOLTIP), SetScrollbar(WID_SCL_MATRIX_SCROLLBAR),
    NWidget(NWID_VSCROLLBAR, COLOUR_GREY, WID_SCL_MATRIX_SCROLLBAR),
  EndContainer(),
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_PANEL, COLOUR_GREY, WID_SCL_SPACER_DROPDOWN), SetFill(1, 1), SetResize(1, 0), EndContainer(),
    NWidget(WWT_DROPDOWN, COLOUR_GREY, WID_SCL_PRI_COL_DROPDOWN), SetFill(0, 1), SetToolTip(STR_LIVERY_PRIMARY_TOOLTIP),
    NWidget(WWT_DROPDOWN, COLOUR_GREY, WID_SCL_SEC_COL_DROPDOWN), SetFill(0, 1), SetToolTip(STR_LIVERY_SECONDARY_TOOLTIP),
    NWidget(WWT_RESIZEBOX, COLOUR_GREY),
  EndContainer(),
};
 
static WindowDesc _select_company_livery_desc(
  WDP_AUTO, "company_color_scheme", 0, 0,
  WC_COMPANY_COLOUR, WC_NONE,
  {},
  _nested_select_company_livery_widgets
);
 
void ShowCompanyLiveryWindow(CompanyID company, GroupID group)
{
  SelectCompanyLiveryWindow *w = (SelectCompanyLiveryWindow *)BringWindowToFrontById(WC_COMPANY_COLOUR, company);
  if (w == nullptr) {
    new SelectCompanyLiveryWindow(_select_company_livery_desc, company, group);
  } else if (group != GroupID::Invalid()) {
    w->SetSelectedGroup(company, group);
  }
}
 
void DrawCompanyManagerFace(CompanyManagerFace cmf, Colours colour, const Rect &r)
{
  GenderEthnicity ge = (GenderEthnicity)GetCompanyManagerFaceBits(cmf, CMFV_GEN_ETHN, GE_WM);
 
  /* Determine offset from centre of drawing rect. */
  Dimension d = GetSpriteSize(SPR_GRADIENT);
  int x = CenterBounds(r.left, r.right, d.width);
  int y = CenterBounds(r.top, r.bottom, d.height);
 
  bool has_moustache   = !HasBit(ge, GENDER_FEMALE) && GetCompanyManagerFaceBits(cmf, CMFV_HAS_MOUSTACHE,   ge) != 0;
  bool has_tie_earring = !HasBit(ge, GENDER_FEMALE) || GetCompanyManagerFaceBits(cmf, CMFV_HAS_TIE_EARRING, ge) != 0;
  bool has_glasses     = GetCompanyManagerFaceBits(cmf, CMFV_HAS_GLASSES, ge) != 0;
  PaletteID pal;
 
  /* Modify eye colour palette only if 2 or more valid values exist */
  if (_cmf_info[CMFV_EYE_COLOUR].valid_values[ge] < 2) {
    pal = PAL_NONE;
  } else {
    switch (GetCompanyManagerFaceBits(cmf, CMFV_EYE_COLOUR, ge)) {
      default: NOT_REACHED();
      case 0: pal = PALETTE_TO_BROWN; break;
      case 1: pal = PALETTE_TO_BLUE;  break;
      case 2: pal = PALETTE_TO_GREEN; break;
    }
  }
 
  /* Draw the gradient (background) */
  DrawSprite(SPR_GRADIENT, GENERAL_SPRITE_COLOUR(colour), x, y);
 
  for (CompanyManagerFaceVariable cmfv = CMFV_CHEEKS; cmfv < CMFV_END; cmfv++) {
    switch (cmfv) {
      case CMFV_MOUSTACHE:   if (!has_moustache)   continue; break;
      case CMFV_LIPS:
      case CMFV_NOSE:        if (has_moustache)    continue; break;
      case CMFV_TIE_EARRING: if (!has_tie_earring) continue; break;
      case CMFV_GLASSES:     if (!has_glasses)     continue; break;
      default: break;
    }
    DrawSprite(GetCompanyManagerFaceSprite(cmf, cmfv, ge), (cmfv == CMFV_EYEBROWS) ? pal : PAL_NONE, x, y);
  }
}
 
static constexpr NWidgetPart _nested_select_company_manager_face_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_SCMF_CAPTION), SetStringTip(STR_FACE_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
    NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCMF_TOGGLE_LARGE_SMALL), SetSpriteTip(SPR_LARGE_SMALL_WINDOW, STR_FACE_ADVANCED_TOOLTIP), SetAspect(WidgetDimensions::ASPECT_TOGGLE_SIZE),
  EndContainer(),
  NWidget(WWT_PANEL, COLOUR_GREY, WID_SCMF_SELECT_FACE),
    NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0), SetPadding(2),
      /* Left side */
      NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
        NWidget(NWID_HORIZONTAL), SetPIPRatio(1, 0, 1),
          NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_FACE), SetMinimalSize(92, 119), SetFill(1, 0),
        EndContainer(),
        NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_RANDOM_NEW_FACE), SetFill(1, 0), SetStringTip(STR_FACE_NEW_FACE_BUTTON, STR_FACE_NEW_FACE_TOOLTIP),
        NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_LOADSAVE), // Load/number/save buttons under the portrait in the advanced view.
          NWidget(NWID_VERTICAL), SetPIP(0, 0, 0), SetPIPRatio(1, 0, 1),
            NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_LOAD), SetFill(1, 0), SetStringTip(STR_FACE_LOAD, STR_FACE_LOAD_TOOLTIP),
            NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_FACECODE), SetFill(1, 0), SetStringTip(STR_FACE_FACECODE, STR_FACE_FACECODE_TOOLTIP),
            NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_SAVE), SetFill(1, 0), SetStringTip(STR_FACE_SAVE, STR_FACE_SAVE_TOOLTIP),
          EndContainer(),
        EndContainer(),
      EndContainer(),
      /* Right side */
      NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
        NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON), SetFill(1, 0), SetStringTip(STR_FACE_ADVANCED, STR_FACE_ADVANCED_TOOLTIP),
        NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_MALEFEMALE), // Simple male/female face setting.
          NWidget(NWID_VERTICAL), SetPIPRatio(1, 0, 1),
            NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_MALE), SetFill(1, 0), SetStringTip(STR_FACE_MALE_BUTTON, STR_FACE_MALE_TOOLTIP),
            NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_FEMALE), SetFill(1, 0), SetStringTip(STR_FACE_FEMALE_BUTTON, STR_FACE_FEMALE_TOOLTIP),
          EndContainer(),
        EndContainer(),
        NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_PARTS), // Advanced face parts setting.
          NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
            NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
              NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_MALE2), SetFill(1, 0), SetStringTip(STR_FACE_MALE_BUTTON, STR_FACE_MALE_TOOLTIP),
              NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_FEMALE2), SetFill(1, 0), SetStringTip(STR_FACE_FEMALE_BUTTON, STR_FACE_FEMALE_TOOLTIP),
            EndContainer(),
            NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
              NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_ETHNICITY_EUR), SetFill(1, 0), SetStringTip(STR_FACE_EUROPEAN, STR_FACE_EUROPEAN_TOOLTIP),
              NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_ETHNICITY_AFR), SetFill(1, 0), SetStringTip(STR_FACE_AFRICAN, STR_FACE_AFRICAN_TOOLTIP),
            EndContainer(),
            NWidget(NWID_VERTICAL),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT), SetFill(1, 0),
                  SetStringTip(STR_FACE_EYECOLOUR), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAS_MOUSTACHE_EARRING), SetToolTip(STR_FACE_MOUSTACHE_EARRING_TOOLTIP), SetTextStyle(TC_WHITE),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_HAS_GLASSES_TEXT), SetFill(1, 0),
                  SetStringTip(STR_FACE_GLASSES), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAS_GLASSES), SetToolTip(STR_FACE_GLASSES_TOOLTIP), SetTextStyle(TC_WHITE),
              EndContainer(),
            EndContainer(),
            NWidget(NWID_VERTICAL),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_HAIR_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_HAIR), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_HAIR_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_HAIR_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAIR), SetToolTip(STR_FACE_HAIR_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_HAIR_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_HAIR_TOOLTIP),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_EYEBROWS_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_EYEBROWS), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYEBROWS_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_EYEBROWS_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_EYEBROWS), SetToolTip(STR_FACE_EYEBROWS_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYEBROWS_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_EYEBROWS_TOOLTIP),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_EYECOLOUR_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_EYECOLOUR), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_EYECOLOUR_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR), SetToolTip(STR_FACE_EYECOLOUR_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_EYECOLOUR_TOOLTIP),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_GLASSES_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_GLASSES), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_GLASSES_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_GLASSES_TOOLTIP_2),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_GLASSES), SetToolTip(STR_FACE_GLASSES_TOOLTIP_2), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_GLASSES_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_GLASSES_TOOLTIP_2),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_NOSE_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_NOSE), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_NOSE_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_NOSE_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_NOSE), SetToolTip(STR_FACE_NOSE_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_NOSE_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_NOSE_TOOLTIP),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_LIPS_MOUSTACHE_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_MOUSTACHE), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_LIPS_MOUSTACHE_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE), SetToolTip(STR_FACE_LIPS_MOUSTACHE_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_LIPS_MOUSTACHE_TOOLTIP),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_CHIN_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_CHIN), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_CHIN_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_CHIN_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_CHIN), SetToolTip(STR_FACE_CHIN_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_CHIN_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_CHIN_TOOLTIP),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_JACKET_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_JACKET), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_JACKET_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_JACKET_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_JACKET), SetToolTip(STR_FACE_JACKET_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_JACKET_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_JACKET_TOOLTIP),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_COLLAR_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_COLLAR), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_COLLAR_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_COLLAR_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_COLLAR), SetToolTip(STR_FACE_COLLAR_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_COLLAR_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_COLLAR_TOOLTIP),
                EndContainer(),
              EndContainer(),
              NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
                NWidget(WWT_TEXT, INVALID_COLOUR, WID_SCMF_TIE_EARRING_TEXT), SetFill(1, 0),
                    SetStringTip(STR_FACE_EARRING), SetTextStyle(TC_GOLD), SetAlignment(SA_VERT_CENTER | SA_RIGHT),
                NWidget(NWID_HORIZONTAL),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING_L), SetArrowWidgetTypeTip(AWV_DECREASE, STR_FACE_TIE_EARRING_TOOLTIP),
                  NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING), SetToolTip(STR_FACE_TIE_EARRING_TOOLTIP), SetTextStyle(TC_WHITE),
                  NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING_R), SetArrowWidgetTypeTip(AWV_INCREASE, STR_FACE_TIE_EARRING_TOOLTIP),
                EndContainer(),
              EndContainer(),
            EndContainer(),
          EndContainer(),
        EndContainer(),
      EndContainer(),
    EndContainer(),
  EndContainer(),
  NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
    NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_CANCEL), SetFill(1, 0), SetStringTip(STR_BUTTON_CANCEL, STR_FACE_CANCEL_TOOLTIP),
    NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_ACCEPT), SetFill(1, 0), SetStringTip(STR_BUTTON_OK, STR_FACE_OK_TOOLTIP),
  EndContainer(),
};
 
class SelectCompanyManagerFaceWindow : public Window
{
  CompanyManagerFace face{}; 
  bool advanced = false; 
 
  GenderEthnicity ge{}; 
  bool is_female = false;     
  bool is_moust_male = false; 
 
  Dimension yesno_dim{};  
  Dimension number_dim{}; 
 
  std::string GetFaceString(WidgetID widget_index, uint8_t val, bool is_bool_widget) const
  {
    const NWidgetCore *nwi_widget = this->GetWidget<NWidgetCore>(widget_index);
    if (nwi_widget->IsDisabled()) return {};
 
    /* If it a bool button write yes or no. */
    if (is_bool_widget) return GetString((val != 0) ? STR_FACE_YES : STR_FACE_NO);
 
    /* Else write the value + 1. */
    return GetString(STR_JUST_INT, val + 1);
  }
 
  void UpdateData()
  {
    this->ge = (GenderEthnicity)GB(this->face, _cmf_info[CMFV_GEN_ETHN].offset, _cmf_info[CMFV_GEN_ETHN].length); // get the gender and ethnicity
    this->is_female = HasBit(this->ge, GENDER_FEMALE); // get the gender: 0 == male and 1 == female
    this->is_moust_male = !is_female && GetCompanyManagerFaceBits(this->face, CMFV_HAS_MOUSTACHE, this->ge) != 0; // is a male face with moustache
 
    this->GetWidget<NWidgetCore>(WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT)->SetString(this->is_female ? STR_FACE_EARRING : STR_FACE_MOUSTACHE);
    this->GetWidget<NWidgetCore>(WID_SCMF_TIE_EARRING_TEXT)->SetString(this->is_female ? STR_FACE_EARRING : STR_FACE_TIE);
    this->GetWidget<NWidgetCore>(WID_SCMF_LIPS_MOUSTACHE_TEXT)->SetString(this->is_moust_male ? STR_FACE_MOUSTACHE : STR_FACE_LIPS);
  }
 
public:
  SelectCompanyManagerFaceWindow(WindowDesc &desc, Window *parent) : Window(desc)
  {
    this->CreateNestedTree();
    this->SelectDisplayPlanes(this->advanced);
    this->FinishInitNested(parent->window_number);
    this->parent = parent;
    this->owner = this->window_number;
    this->face = Company::Get(this->window_number)->face;
 
    this->UpdateData();
  }
 
  void SelectDisplayPlanes(bool advanced)
  {
    this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_LOADSAVE)->SetDisplayedPlane(advanced ? 0 : SZSP_NONE);
    this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_PARTS)->SetDisplayedPlane(advanced ? 0 : SZSP_NONE);
    this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_MALEFEMALE)->SetDisplayedPlane(advanced ? SZSP_NONE : 0);
    this->GetWidget<NWidgetCore>(WID_SCMF_RANDOM_NEW_FACE)->SetString(advanced ? STR_FACE_RANDOM : STR_FACE_NEW_FACE_BUTTON);
 
    NWidgetCore *wi = this->GetWidget<NWidgetCore>(WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON);
    if (advanced) {
      wi->SetStringTip(STR_FACE_SIMPLE, STR_FACE_SIMPLE_TOOLTIP);
    } else {
      wi->SetStringTip(STR_FACE_ADVANCED, STR_FACE_ADVANCED_TOOLTIP);
    }
  }
 
  void OnInit() override
  {
    /* Size of the boolean yes/no button. */
    Dimension yesno_dim = maxdim(GetStringBoundingBox(STR_FACE_YES), GetStringBoundingBox(STR_FACE_NO));
    yesno_dim.width  += WidgetDimensions::scaled.framerect.Horizontal();
    yesno_dim.height += WidgetDimensions::scaled.framerect.Vertical();
    /* Size of the number button + arrows. */
    Dimension number_dim = {0, 0};
    for (int val = 1; val <= 12; val++) {
      number_dim = maxdim(number_dim, GetStringBoundingBox(GetString(STR_JUST_INT, val)));
    }
    uint arrows_width = GetSpriteSize(SPR_ARROW_LEFT).width + GetSpriteSize(SPR_ARROW_RIGHT).width + 2 * (WidgetDimensions::scaled.imgbtn.Horizontal());
    number_dim.width += WidgetDimensions::scaled.framerect.Horizontal() + arrows_width;
    number_dim.height += WidgetDimensions::scaled.framerect.Vertical();
    /* Compute width of both buttons. */
    yesno_dim.width = std::max(yesno_dim.width, number_dim.width);
    number_dim.width = yesno_dim.width - arrows_width;
 
    this->yesno_dim = yesno_dim;
    this->number_dim = number_dim;
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    switch (widget) {
      case WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT:
        size = maxdim(size, GetStringBoundingBox(STR_FACE_EARRING));
        size = maxdim(size, GetStringBoundingBox(STR_FACE_MOUSTACHE));
        break;
 
      case WID_SCMF_TIE_EARRING_TEXT:
        size = maxdim(size, GetStringBoundingBox(STR_FACE_EARRING));
        size = maxdim(size, GetStringBoundingBox(STR_FACE_TIE));
        break;
 
      case WID_SCMF_LIPS_MOUSTACHE_TEXT:
        size = maxdim(size, GetStringBoundingBox(STR_FACE_LIPS));
        size = maxdim(size, GetStringBoundingBox(STR_FACE_MOUSTACHE));
        break;
 
      case WID_SCMF_FACE:
        size = maxdim(size, GetScaledSpriteSize(SPR_GRADIENT));
        break;
 
      case WID_SCMF_HAS_MOUSTACHE_EARRING:
      case WID_SCMF_HAS_GLASSES:
        size = this->yesno_dim;
        break;
 
      case WID_SCMF_EYECOLOUR:
      case WID_SCMF_CHIN:
      case WID_SCMF_EYEBROWS:
      case WID_SCMF_LIPS_MOUSTACHE:
      case WID_SCMF_NOSE:
      case WID_SCMF_HAIR:
      case WID_SCMF_JACKET:
      case WID_SCMF_COLLAR:
      case WID_SCMF_TIE_EARRING:
      case WID_SCMF_GLASSES:
        size = this->number_dim;
        break;
    }
  }
 
  void OnPaint() override
  {
    /* lower the non-selected gender button */
    this->SetWidgetsLoweredState(!this->is_female, WID_SCMF_MALE, WID_SCMF_MALE2);
    this->SetWidgetsLoweredState( this->is_female, WID_SCMF_FEMALE, WID_SCMF_FEMALE2);
 
    /* advanced company manager face selection window */
 
    /* lower the non-selected ethnicity button */
    this->SetWidgetLoweredState(WID_SCMF_ETHNICITY_EUR, !HasBit(this->ge, ETHNICITY_BLACK));
    this->SetWidgetLoweredState(WID_SCMF_ETHNICITY_AFR,  HasBit(this->ge, ETHNICITY_BLACK));
 
 
    /* Disable dynamically the widgets which CompanyManagerFaceVariable has less than 2 options
     * (or in other words you haven't any choice).
     * If the widgets depend on a HAS-variable and this is false the widgets will be disabled, too. */
 
    /* Eye colour buttons */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_EYE_COLOUR].valid_values[this->ge] < 2,
        WID_SCMF_EYECOLOUR, WID_SCMF_EYECOLOUR_L, WID_SCMF_EYECOLOUR_R);
 
    /* Chin buttons */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_CHIN].valid_values[this->ge] < 2,
        WID_SCMF_CHIN, WID_SCMF_CHIN_L, WID_SCMF_CHIN_R);
 
    /* Eyebrows buttons */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_EYEBROWS].valid_values[this->ge] < 2,
        WID_SCMF_EYEBROWS, WID_SCMF_EYEBROWS_L, WID_SCMF_EYEBROWS_R);
 
    /* Lips or (if it a male face with a moustache) moustache buttons */
    this->SetWidgetsDisabledState(_cmf_info[this->is_moust_male ? CMFV_MOUSTACHE : CMFV_LIPS].valid_values[this->ge] < 2,
        WID_SCMF_LIPS_MOUSTACHE, WID_SCMF_LIPS_MOUSTACHE_L, WID_SCMF_LIPS_MOUSTACHE_R);
 
    /* Nose buttons | male faces with moustache haven't any nose options */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_NOSE].valid_values[this->ge] < 2 || this->is_moust_male,
        WID_SCMF_NOSE, WID_SCMF_NOSE_L, WID_SCMF_NOSE_R);
 
    /* Hair buttons */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_HAIR].valid_values[this->ge] < 2,
        WID_SCMF_HAIR, WID_SCMF_HAIR_L, WID_SCMF_HAIR_R);
 
    /* Jacket buttons */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_JACKET].valid_values[this->ge] < 2,
        WID_SCMF_JACKET, WID_SCMF_JACKET_L, WID_SCMF_JACKET_R);
 
    /* Collar buttons */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_COLLAR].valid_values[this->ge] < 2,
        WID_SCMF_COLLAR, WID_SCMF_COLLAR_L, WID_SCMF_COLLAR_R);
 
    /* Tie/earring buttons | female faces without earring haven't any earring options */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_TIE_EARRING].valid_values[this->ge] < 2 ||
          (this->is_female && GetCompanyManagerFaceBits(this->face, CMFV_HAS_TIE_EARRING, this->ge) == 0),
        WID_SCMF_TIE_EARRING, WID_SCMF_TIE_EARRING_L, WID_SCMF_TIE_EARRING_R);
 
    /* Glasses buttons | faces without glasses haven't any glasses options */
    this->SetWidgetsDisabledState(_cmf_info[CMFV_GLASSES].valid_values[this->ge] < 2 || GetCompanyManagerFaceBits(this->face, CMFV_HAS_GLASSES, this->ge) == 0,
        WID_SCMF_GLASSES, WID_SCMF_GLASSES_L, WID_SCMF_GLASSES_R);
 
    this->DrawWidgets();
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    switch (widget) {
      case WID_SCMF_HAS_MOUSTACHE_EARRING: {
        CompanyManagerFaceVariable facepart = this->is_female ? CMFV_HAS_TIE_EARRING : CMFV_HAS_MOUSTACHE;
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, facepart, this->ge), true);
      }
 
      case WID_SCMF_TIE_EARRING:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_TIE_EARRING, this->ge), false);
 
      case WID_SCMF_LIPS_MOUSTACHE: {
        CompanyManagerFaceVariable facepart = this->is_moust_male ? CMFV_MOUSTACHE : CMFV_LIPS;
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, facepart, this->ge), false);
      }
 
      case WID_SCMF_HAS_GLASSES:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_HAS_GLASSES, this->ge), true );
 
      case WID_SCMF_HAIR:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_HAIR,        this->ge), false);
 
      case WID_SCMF_EYEBROWS:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_EYEBROWS,    this->ge), false);
 
      case WID_SCMF_EYECOLOUR:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_EYE_COLOUR,  this->ge), false);
 
      case WID_SCMF_GLASSES:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_GLASSES,     this->ge), false);
 
      case WID_SCMF_NOSE:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_NOSE,        this->ge), false);
 
      case WID_SCMF_CHIN:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_CHIN,        this->ge), false);
 
      case WID_SCMF_JACKET:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_JACKET,      this->ge), false);
 
      case WID_SCMF_COLLAR:
        return this->GetFaceString(widget, GetCompanyManagerFaceBits(this->face, CMFV_COLLAR,      this->ge), false);
 
      default:
        return this->Window::GetWidgetString(widget, stringid);
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    switch (widget) {
      case WID_SCMF_FACE:
        DrawCompanyManagerFace(this->face, Company::Get(this->window_number)->colour, r);
        break;
    }
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      /* Toggle size, advanced/simple face selection */
      case WID_SCMF_TOGGLE_LARGE_SMALL:
      case WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON:
        this->advanced = !this->advanced;
        this->SelectDisplayPlanes(this->advanced);
        this->ReInit();
        break;
 
      /* OK button */
      case WID_SCMF_ACCEPT:
        Command<CMD_SET_COMPANY_MANAGER_FACE>::Post(this->face);
        [[fallthrough]];
 
      /* Cancel button */
      case WID_SCMF_CANCEL:
        this->Close();
        break;
 
      /* Load button */
      case WID_SCMF_LOAD:
        this->face = _company_manager_face;
        ScaleAllCompanyManagerFaceBits(this->face);
        ShowErrorMessage(GetEncodedString(STR_FACE_LOAD_DONE), {}, WL_INFO);
        this->UpdateData();
        this->SetDirty();
        break;
 
      /* 'Company manager face number' button, view and/or set company manager face number */
      case WID_SCMF_FACECODE:
        ShowQueryString(GetString(STR_JUST_INT, this->face), STR_FACE_FACECODE_CAPTION, 10 + 1, this, CS_NUMERAL, {});
        break;
 
      /* Save button */
      case WID_SCMF_SAVE:
        _company_manager_face = this->face;
        ShowErrorMessage(GetEncodedString(STR_FACE_SAVE_DONE), {}, WL_INFO);
        break;
 
      /* Toggle gender (male/female) button */
      case WID_SCMF_MALE:
      case WID_SCMF_FEMALE:
      case WID_SCMF_MALE2:
      case WID_SCMF_FEMALE2:
        SetCompanyManagerFaceBits(this->face, CMFV_GENDER, this->ge, (widget == WID_SCMF_FEMALE || widget == WID_SCMF_FEMALE2));
        ScaleAllCompanyManagerFaceBits(this->face);
        this->UpdateData();
        this->SetDirty();
        break;
 
      /* Randomize face button */
      case WID_SCMF_RANDOM_NEW_FACE:
        RandomCompanyManagerFaceBits(this->face, this->ge, this->advanced, _interactive_random);
        this->UpdateData();
        this->SetDirty();
        break;
 
      /* Toggle ethnicity (european/african) button */
      case WID_SCMF_ETHNICITY_EUR:
      case WID_SCMF_ETHNICITY_AFR:
        SetCompanyManagerFaceBits(this->face, CMFV_ETHNICITY, this->ge, widget - WID_SCMF_ETHNICITY_EUR);
        ScaleAllCompanyManagerFaceBits(this->face);
        this->UpdateData();
        this->SetDirty();
        break;
 
      default:
        /* Here all buttons from WID_SCMF_HAS_MOUSTACHE_EARRING to WID_SCMF_GLASSES_R are handled.
         * First it checks which CompanyManagerFaceVariable is being changed, and then either
         * a: invert the value for boolean variables, or
         * b: it checks inside of IncreaseCompanyManagerFaceBits() if a left (_L) butten is pressed and then decrease else increase the variable */
        if (widget >= WID_SCMF_HAS_MOUSTACHE_EARRING && widget <= WID_SCMF_GLASSES_R) {
          CompanyManagerFaceVariable cmfv; // which CompanyManagerFaceVariable shall be edited
 
          if (widget < WID_SCMF_EYECOLOUR_L) { // Bool buttons
            switch (widget - WID_SCMF_HAS_MOUSTACHE_EARRING) {
              default: NOT_REACHED();
              case 0: cmfv = this->is_female ? CMFV_HAS_TIE_EARRING : CMFV_HAS_MOUSTACHE; break; // Has earring/moustache button
              case 1: cmfv = CMFV_HAS_GLASSES; break; // Has glasses button
            }
            SetCompanyManagerFaceBits(this->face, cmfv, this->ge, !GetCompanyManagerFaceBits(this->face, cmfv, this->ge));
            ScaleAllCompanyManagerFaceBits(this->face);
          } else { // Value buttons
            switch ((widget - WID_SCMF_EYECOLOUR_L) / 3) {
              default: NOT_REACHED();
              case 0: cmfv = CMFV_EYE_COLOUR; break;  // Eye colour buttons
              case 1: cmfv = CMFV_CHIN; break;        // Chin buttons
              case 2: cmfv = CMFV_EYEBROWS; break;    // Eyebrows buttons
              case 3: cmfv = this->is_moust_male ? CMFV_MOUSTACHE : CMFV_LIPS; break; // Moustache or lips buttons
              case 4: cmfv = CMFV_NOSE; break;        // Nose buttons
              case 5: cmfv = CMFV_HAIR; break;        // Hair buttons
              case 6: cmfv = CMFV_JACKET; break;      // Jacket buttons
              case 7: cmfv = CMFV_COLLAR; break;      // Collar buttons
              case 8: cmfv = CMFV_TIE_EARRING; break; // Tie/earring buttons
              case 9: cmfv = CMFV_GLASSES; break;     // Glasses buttons
            }
            /* 0 == left (_L), 1 == middle or 2 == right (_R) - button click */
            IncreaseCompanyManagerFaceBits(this->face, cmfv, this->ge, (((widget - WID_SCMF_EYECOLOUR_L) % 3) != 0) ? 1 : -1);
          }
          this->UpdateData();
          this->SetDirty();
        }
        break;
    }
  }
 
  void OnQueryTextFinished(std::optional<std::string> str) override
  {
    if (!str.has_value()) return;
    /* Set a new company manager face number */
    if (!str->empty()) {
      this->face = std::strtoul(str->c_str(), nullptr, 10);
      ScaleAllCompanyManagerFaceBits(this->face);
      ShowErrorMessage(GetEncodedString(STR_FACE_FACECODE_SET), {}, WL_INFO);
      this->UpdateData();
      this->SetDirty();
    } else {
      ShowErrorMessage(GetEncodedString(STR_FACE_FACECODE_ERR), {}, WL_INFO);
    }
  }
};
 
static WindowDesc _select_company_manager_face_desc(
  WDP_AUTO, nullptr, 0, 0,
  WC_COMPANY_MANAGER_FACE, WC_NONE,
  WindowDefaultFlag::Construction,
  _nested_select_company_manager_face_widgets
);
 
static void DoSelectCompanyManagerFace(Window *parent)
{
  if (!Company::IsValidID(parent->window_number)) return;
 
  if (BringWindowToFrontById(WC_COMPANY_MANAGER_FACE, parent->window_number)) return;
  new SelectCompanyManagerFaceWindow(_select_company_manager_face_desc, parent);
}
 
static constexpr NWidgetPart _nested_company_infrastructure_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_CI_CAPTION),
    NWidget(WWT_SHADEBOX, COLOUR_GREY),
    NWidget(WWT_STICKYBOX, COLOUR_GREY),
  EndContainer(),
  NWidget(WWT_PANEL, COLOUR_GREY),
    NWidget(NWID_VERTICAL), SetPadding(WidgetDimensions::unscaled.framerect), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
      NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_RAIL_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_RAIL_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
      EndContainer(),
      NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_ROAD_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_ROAD_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
      EndContainer(),
      NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_TRAM_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_TRAM_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
      EndContainer(),
      NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_WATER_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_WATER_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
      EndContainer(),
      NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_STATION_DESC), SetMinimalTextLines(3, 0), SetFill(1, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_STATION_COUNT), SetMinimalTextLines(3, 0), SetFill(0, 1),
      EndContainer(),
      NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_TOTAL_DESC), SetFill(1, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_CI_TOTAL), SetFill(0, 1),
      EndContainer(),
    EndContainer(),
  EndContainer(),
};
 
struct CompanyInfrastructureWindow : Window
{
  RailTypes railtypes{}; 
  RoadTypes roadtypes{}; 
 
  uint total_width = 0; 
 
  CompanyInfrastructureWindow(WindowDesc &desc, WindowNumber window_number) : Window(desc)
  {
    this->UpdateRailRoadTypes();
 
    this->InitNested(window_number);
    this->owner = this->window_number;
  }
 
  void UpdateRailRoadTypes()
  {
    this->railtypes = RAILTYPES_NONE;
    this->roadtypes = ROADTYPES_NONE;
 
    /* Find the used railtypes. */
    for (const Engine *e : Engine::IterateType(VEH_TRAIN)) {
      if (!e->info.climates.Test(_settings_game.game_creation.landscape)) continue;
 
      this->railtypes |= GetRailTypeInfo(e->u.rail.railtype)->introduces_railtypes;
    }
 
    /* Get the date introduced railtypes as well. */
    this->railtypes = AddDateIntroducedRailTypes(this->railtypes, CalendarTime::MAX_DATE);
 
    /* Find the used roadtypes. */
    for (const Engine *e : Engine::IterateType(VEH_ROAD)) {
      if (!e->info.climates.Test(_settings_game.game_creation.landscape)) continue;
 
      this->roadtypes |= GetRoadTypeInfo(e->u.road.roadtype)->introduces_roadtypes;
    }
 
    /* Get the date introduced roadtypes as well. */
    this->roadtypes = AddDateIntroducedRoadTypes(this->roadtypes, CalendarTime::MAX_DATE);
    this->roadtypes &= ~_roadtypes_hidden_mask;
  }
 
  Money GetTotalMaintenanceCost() const
  {
    const Company *c = Company::Get(this->window_number);
    Money total;
 
    uint32_t rail_total = c->infrastructure.GetRailTotal();
    for (RailType rt = RAILTYPE_BEGIN; rt != RAILTYPE_END; rt++) {
      if (HasBit(this->railtypes, rt)) total += RailMaintenanceCost(rt, c->infrastructure.rail[rt], rail_total);
    }
    total += SignalMaintenanceCost(c->infrastructure.signal);
 
    uint32_t road_total = c->infrastructure.GetRoadTotal();
    uint32_t tram_total = c->infrastructure.GetTramTotal();
    for (RoadType rt = ROADTYPE_BEGIN; rt != ROADTYPE_END; rt++) {
      if (HasBit(this->roadtypes, rt)) total += RoadMaintenanceCost(rt, c->infrastructure.road[rt], RoadTypeIsRoad(rt) ? road_total : tram_total);
    }
 
    total += CanalMaintenanceCost(c->infrastructure.water);
    total += StationMaintenanceCost(c->infrastructure.station);
    total += AirportMaintenanceCost(c->index);
 
    return total;
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    switch (widget) {
      case WID_CI_CAPTION:
        return GetString(STR_COMPANY_INFRASTRUCTURE_VIEW_CAPTION, this->window_number);
 
      default:
        return this->Window::GetWidgetString(widget, stringid);
    }
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    const Company *c = Company::Get(this->window_number);
 
    switch (widget) {
      case WID_CI_RAIL_DESC: {
        uint lines = 1; // Starts at 1 because a line is also required for the section title
 
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_RAIL_SECT).width + padding.width);
 
        for (const auto &rt : _sorted_railtypes) {
          if (HasBit(this->railtypes, rt)) {
            lines++;
            size.width = std::max(size.width, GetStringBoundingBox(GetRailTypeInfo(rt)->strings.name).width + padding.width + WidgetDimensions::scaled.hsep_indent);
          }
        }
        if (this->railtypes != RAILTYPES_NONE) {
          lines++;
          size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_SIGNALS).width + padding.width + WidgetDimensions::scaled.hsep_indent);
        }
 
        size.height = std::max(size.height, lines * GetCharacterHeight(FS_NORMAL));
        break;
      }
 
      case WID_CI_ROAD_DESC:
      case WID_CI_TRAM_DESC: {
        uint lines = 1; // Starts at 1 because a line is also required for the section title
 
        size.width = std::max(size.width, GetStringBoundingBox(widget == WID_CI_ROAD_DESC ? STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD_SECT : STR_COMPANY_INFRASTRUCTURE_VIEW_TRAM_SECT).width + padding.width);
 
        for (const auto &rt : _sorted_roadtypes) {
          if (HasBit(this->roadtypes, rt) && RoadTypeIsRoad(rt) == (widget == WID_CI_ROAD_DESC)) {
            lines++;
            size.width = std::max(size.width, GetStringBoundingBox(GetRoadTypeInfo(rt)->strings.name).width + padding.width + WidgetDimensions::scaled.hsep_indent);
          }
        }
 
        size.height = std::max(size.height, lines * GetCharacterHeight(FS_NORMAL));
        break;
      }
 
      case WID_CI_WATER_DESC:
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_WATER_SECT).width + padding.width);
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_CANALS).width + padding.width + WidgetDimensions::scaled.hsep_indent);
        break;
 
      case WID_CI_STATION_DESC:
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_STATION_SECT).width + padding.width);
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_STATIONS).width + padding.width + WidgetDimensions::scaled.hsep_indent);
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_AIRPORTS).width + padding.width + WidgetDimensions::scaled.hsep_indent);
        break;
 
      case WID_CI_RAIL_COUNT:
      case WID_CI_ROAD_COUNT:
      case WID_CI_TRAM_COUNT:
      case WID_CI_WATER_COUNT:
      case WID_CI_STATION_COUNT:
      case WID_CI_TOTAL: {
        /* Find the maximum count that is displayed. */
        uint32_t max_val = 1000;  // Some random number to reserve enough space.
        Money max_cost = 10000; // Some random number to reserve enough space.
        uint32_t rail_total = c->infrastructure.GetRailTotal();
        for (RailType rt = RAILTYPE_BEGIN; rt < RAILTYPE_END; rt++) {
          max_val = std::max(max_val, c->infrastructure.rail[rt]);
          max_cost = std::max(max_cost, RailMaintenanceCost(rt, c->infrastructure.rail[rt], rail_total));
        }
        max_val = std::max(max_val, c->infrastructure.signal);
        max_cost = std::max(max_cost, SignalMaintenanceCost(c->infrastructure.signal));
        uint32_t road_total = c->infrastructure.GetRoadTotal();
        uint32_t tram_total = c->infrastructure.GetTramTotal();
        for (RoadType rt = ROADTYPE_BEGIN; rt < ROADTYPE_END; rt++) {
          max_val = std::max(max_val, c->infrastructure.road[rt]);
          max_cost = std::max(max_cost, RoadMaintenanceCost(rt, c->infrastructure.road[rt], RoadTypeIsRoad(rt) ? road_total : tram_total));
 
        }
        max_val = std::max(max_val, c->infrastructure.water);
        max_cost = std::max(max_cost, CanalMaintenanceCost(c->infrastructure.water));
        max_val = std::max(max_val, c->infrastructure.station);
        max_cost = std::max(max_cost, StationMaintenanceCost(c->infrastructure.station));
        max_val = std::max(max_val, c->infrastructure.airport);
        max_cost = std::max(max_cost, AirportMaintenanceCost(c->index));
 
        uint count_width = GetStringBoundingBox(GetString(STR_JUST_COMMA, GetParamMaxValue(max_val))).width + WidgetDimensions::scaled.hsep_indent; // Reserve some wiggle room
 
        if (_settings_game.economy.infrastructure_maintenance) {
          StringID str_total = TimerGameEconomy::UsingWallclockUnits() ? STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL_PERIOD : STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL_YEAR;
          /* Convert to per year */
          this->total_width = GetStringBoundingBox(GetString(str_total, GetParamMaxValue(this->GetTotalMaintenanceCost() * 12))).width + WidgetDimensions::scaled.hsep_indent * 2;
          size.width = std::max(size.width, this->total_width);
 
          /* Convert to per year */
          count_width += std::max(this->total_width, GetStringBoundingBox(GetString(str_total, GetParamMaxValue(max_cost * 12))).width);
        }
 
        size.width = std::max(size.width, count_width);
 
        /* Set height of the total line. */
        if (widget == WID_CI_TOTAL) {
          size.height = _settings_game.economy.infrastructure_maintenance ? std::max<uint>(size.height, WidgetDimensions::scaled.vsep_normal + GetCharacterHeight(FS_NORMAL)) : 0;
        }
        break;
      }
    }
  }
 
  void DrawCountLine(const Rect &r, int &y, int count, Money monthly_cost) const
  {
    DrawString(r.left, r.right, y += GetCharacterHeight(FS_NORMAL), GetString(STR_JUST_COMMA, count), TC_WHITE, SA_RIGHT);
 
    if (_settings_game.economy.infrastructure_maintenance) {
      Rect tr = r.WithWidth(this->total_width, _current_text_dir == TD_RTL);
      DrawString(tr.left, tr.right, y,
        GetString(TimerGameEconomy::UsingWallclockUnits() ? STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL_PERIOD : STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL_YEAR, monthly_cost * 12),
        TC_FROMSTRING, SA_RIGHT);
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    const Company *c = Company::Get(this->window_number);
 
    int y = r.top;
 
    Rect ir = r.Indent(WidgetDimensions::scaled.hsep_indent, _current_text_dir == TD_RTL);
    switch (widget) {
      case WID_CI_RAIL_DESC:
        DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_RAIL_SECT);
 
        if (this->railtypes != RAILTYPES_NONE) {
          /* Draw name of each valid railtype. */
          for (const auto &rt : _sorted_railtypes) {
            if (HasBit(this->railtypes, rt)) {
              DrawString(ir.left, ir.right, y += GetCharacterHeight(FS_NORMAL), GetRailTypeInfo(rt)->strings.name, TC_WHITE);
            }
          }
          DrawString(ir.left, ir.right, y += GetCharacterHeight(FS_NORMAL), STR_COMPANY_INFRASTRUCTURE_VIEW_SIGNALS);
        } else {
          /* No valid railtype. */
          DrawString(ir.left, ir.right, y += GetCharacterHeight(FS_NORMAL), STR_COMPANY_VIEW_INFRASTRUCTURE_NONE);
        }
 
        break;
 
      case WID_CI_RAIL_COUNT: {
        /* Draw infrastructure count for each valid railtype. */
        uint32_t rail_total = c->infrastructure.GetRailTotal();
        for (const auto &rt : _sorted_railtypes) {
          if (HasBit(this->railtypes, rt)) {
            this->DrawCountLine(r, y, c->infrastructure.rail[rt], RailMaintenanceCost(rt, c->infrastructure.rail[rt], rail_total));
          }
        }
        if (this->railtypes != RAILTYPES_NONE) {
          this->DrawCountLine(r, y, c->infrastructure.signal, SignalMaintenanceCost(c->infrastructure.signal));
        }
        break;
      }
 
      case WID_CI_ROAD_DESC:
      case WID_CI_TRAM_DESC: {
        DrawString(r.left, r.right, y, widget == WID_CI_ROAD_DESC ? STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD_SECT : STR_COMPANY_INFRASTRUCTURE_VIEW_TRAM_SECT);
 
        /* Draw name of each valid roadtype. */
        for (const auto &rt : _sorted_roadtypes) {
          if (HasBit(this->roadtypes, rt) && RoadTypeIsRoad(rt) == (widget == WID_CI_ROAD_DESC)) {
            DrawString(ir.left, ir.right, y += GetCharacterHeight(FS_NORMAL), GetRoadTypeInfo(rt)->strings.name, TC_WHITE);
          }
        }
 
        break;
      }
 
      case WID_CI_ROAD_COUNT:
      case WID_CI_TRAM_COUNT: {
        uint32_t road_tram_total = widget == WID_CI_ROAD_COUNT ? c->infrastructure.GetRoadTotal() : c->infrastructure.GetTramTotal();
        for (const auto &rt : _sorted_roadtypes) {
          if (HasBit(this->roadtypes, rt) && RoadTypeIsRoad(rt) == (widget == WID_CI_ROAD_COUNT)) {
            this->DrawCountLine(r, y, c->infrastructure.road[rt], RoadMaintenanceCost(rt, c->infrastructure.road[rt], road_tram_total));
          }
        }
        break;
      }
 
      case WID_CI_WATER_DESC:
        DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_WATER_SECT);
        DrawString(ir.left, ir.right, y += GetCharacterHeight(FS_NORMAL), STR_COMPANY_INFRASTRUCTURE_VIEW_CANALS);
        break;
 
      case WID_CI_WATER_COUNT:
        this->DrawCountLine(r, y, c->infrastructure.water, CanalMaintenanceCost(c->infrastructure.water));
        break;
 
      case WID_CI_TOTAL:
        if (_settings_game.economy.infrastructure_maintenance) {
          Rect tr = r.WithWidth(this->total_width, _current_text_dir == TD_RTL);
          GfxFillRect(tr.left, y, tr.right, y + WidgetDimensions::scaled.bevel.top - 1, PC_WHITE);
          y += WidgetDimensions::scaled.vsep_normal;
          DrawString(tr.left, tr.right, y,
            GetString(TimerGameEconomy::UsingWallclockUnits() ? STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL_PERIOD : STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL_YEAR, this->GetTotalMaintenanceCost() * 12),
            TC_FROMSTRING, SA_RIGHT);
        }
        break;
 
      case WID_CI_STATION_DESC:
        DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_STATION_SECT);
        DrawString(ir.left, ir.right, y += GetCharacterHeight(FS_NORMAL), STR_COMPANY_INFRASTRUCTURE_VIEW_STATIONS);
        DrawString(ir.left, ir.right, y += GetCharacterHeight(FS_NORMAL), STR_COMPANY_INFRASTRUCTURE_VIEW_AIRPORTS);
        break;
 
      case WID_CI_STATION_COUNT:
        this->DrawCountLine(r, y, c->infrastructure.station, StationMaintenanceCost(c->infrastructure.station));
        this->DrawCountLine(r, y, c->infrastructure.airport, AirportMaintenanceCost(c->index));
        break;
    }
  }
 
  void OnInvalidateData([[maybe_unused]] int data = 0, [[maybe_unused]] bool gui_scope = true) override
  {
    if (!gui_scope) return;
 
    this->UpdateRailRoadTypes();
    this->ReInit();
  }
};
 
static WindowDesc _company_infrastructure_desc(
  WDP_AUTO, "company_infrastructure", 0, 0,
  WC_COMPANY_INFRASTRUCTURE, WC_NONE,
  {},
  _nested_company_infrastructure_widgets
);
 
static void ShowCompanyInfrastructure(CompanyID company)
{
  if (!Company::IsValidID(company)) return;
  AllocateWindowDescFront<CompanyInfrastructureWindow>(_company_infrastructure_desc, company);
}
 
static constexpr NWidgetPart _nested_company_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_C_CAPTION),
    NWidget(WWT_SHADEBOX, COLOUR_GREY),
    NWidget(WWT_STICKYBOX, COLOUR_GREY),
  EndContainer(),
  NWidget(WWT_PANEL, COLOUR_GREY),
    NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0), SetPadding(4),
      NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_FACE), SetMinimalSize(92, 119), SetFill(1, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_FACE_TITLE), SetFill(1, 1), SetMinimalTextLines(2, 0),
      EndContainer(),
      NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
        NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
          NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
            NWidget(WWT_TEXT, INVALID_COLOUR, WID_C_DESC_INAUGURATION), SetFill(1, 0),
            NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
              NWidget(WWT_LABEL, INVALID_COLOUR, WID_C_DESC_COLOUR_SCHEME), SetStringTip(STR_COMPANY_VIEW_COLOUR_SCHEME_TITLE),
              NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_COLOUR_SCHEME_EXAMPLE), SetMinimalSize(30, 0), SetFill(1, 1),
            EndContainer(),
            NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
              NWidget(WWT_TEXT, INVALID_COLOUR, WID_C_DESC_VEHICLE), SetStringTip(STR_COMPANY_VIEW_VEHICLES_TITLE), SetAlignment(SA_LEFT | SA_TOP),
              NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_VEHICLE_COUNTS), SetMinimalTextLines(4, 0), SetFill(1, 1),
            EndContainer(),
          EndContainer(),
          NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
            NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_VIEW_BUILD_HQ),
              NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_VIEW_HQ), SetStringTip(STR_COMPANY_VIEW_VIEW_HQ_BUTTON, STR_COMPANY_VIEW_VIEW_HQ_TOOLTIP),
              NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_C_BUILD_HQ), SetStringTip(STR_COMPANY_VIEW_BUILD_HQ_BUTTON, STR_COMPANY_VIEW_BUILD_HQ_TOOLTIP),
            EndContainer(),
            NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_RELOCATE),
              NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_C_RELOCATE_HQ), SetStringTip(STR_COMPANY_VIEW_RELOCATE_HQ, STR_COMPANY_VIEW_RELOCATE_HQ_TOOLTIP),
              NWidget(NWID_SPACER),
            EndContainer(),
          EndContainer(),
        EndContainer(),
 
        NWidget(WWT_TEXT, INVALID_COLOUR, WID_C_DESC_COMPANY_VALUE), SetFill(1, 0),
 
        NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
          NWidget(WWT_TEXT, INVALID_COLOUR, WID_C_DESC_INFRASTRUCTURE), SetStringTip(STR_COMPANY_VIEW_INFRASTRUCTURE),  SetAlignment(SA_LEFT | SA_TOP),
          NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_INFRASTRUCTURE_COUNTS), SetMinimalTextLines(5, 0), SetFill(1, 0),
          NWidget(NWID_VERTICAL), SetPIPRatio(0, 0, 1),
            NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_VIEW_INFRASTRUCTURE), SetStringTip(STR_COMPANY_VIEW_INFRASTRUCTURE_BUTTON, STR_COMPANY_VIEW_INFRASTRUCTURE_TOOLTIP),
          EndContainer(),
        EndContainer(),
 
        /* Multi player buttons. */
        NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0), SetPIPRatio(1, 0, 0),
          NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
            NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_HOSTILE_TAKEOVER),
              NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_HOSTILE_TAKEOVER), SetStringTip(STR_COMPANY_VIEW_HOSTILE_TAKEOVER_BUTTON, STR_COMPANY_VIEW_HOSTILE_TAKEOVER_TOOLTIP),
            EndContainer(),
            NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_GIVE_MONEY),
              NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_GIVE_MONEY), SetStringTip(STR_COMPANY_VIEW_GIVE_MONEY_BUTTON, STR_COMPANY_VIEW_GIVE_MONEY_TOOLTIP),
            EndContainer(),
            NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_MULTIPLAYER),
              NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COMPANY_JOIN), SetStringTip(STR_COMPANY_VIEW_JOIN, STR_COMPANY_VIEW_JOIN_TOOLTIP),
            EndContainer(),
          EndContainer(),
        EndContainer(),
      EndContainer(),
    EndContainer(),
  EndContainer(),
  /* Button bars at the bottom. */
  NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_BUTTONS),
    NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
      NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_NEW_FACE), SetFill(1, 0), SetStringTip(STR_COMPANY_VIEW_NEW_FACE_BUTTON, STR_COMPANY_VIEW_NEW_FACE_TOOLTIP),
      NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COLOUR_SCHEME), SetFill(1, 0), SetStringTip(STR_COMPANY_VIEW_COLOUR_SCHEME_BUTTON, STR_COMPANY_VIEW_COLOUR_SCHEME_TOOLTIP),
      NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_PRESIDENT_NAME), SetFill(1, 0), SetStringTip(STR_COMPANY_VIEW_PRESIDENT_NAME_BUTTON, STR_COMPANY_VIEW_PRESIDENT_NAME_TOOLTIP),
      NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COMPANY_NAME), SetFill(1, 0), SetStringTip(STR_COMPANY_VIEW_COMPANY_NAME_BUTTON, STR_COMPANY_VIEW_COMPANY_NAME_TOOLTIP),
    EndContainer(),
  EndContainer(),
};
 
static const StringID _company_view_vehicle_count_strings[] = {
  STR_COMPANY_VIEW_TRAINS, STR_COMPANY_VIEW_ROAD_VEHICLES, STR_COMPANY_VIEW_SHIPS, STR_COMPANY_VIEW_AIRCRAFT
};
 
struct CompanyWindow : Window
{
  CompanyWidgets query_widget{};
 
  enum CompanyWindowPlanes : uint8_t {
    /* Display planes of the #WID_C_SELECT_VIEW_BUILD_HQ selection widget. */
    CWP_VB_VIEW = 0,  
    CWP_VB_BUILD,     
 
    /* Display planes of the #WID_C_SELECT_RELOCATE selection widget. */
    CWP_RELOCATE_SHOW = 0, 
    CWP_RELOCATE_HIDE,     
  };
 
  CompanyWindow(WindowDesc &desc, WindowNumber window_number) : Window(desc)
  {
    this->InitNested(window_number);
    this->owner = this->window_number;
    this->OnInvalidateData();
  }
 
  void OnPaint() override
  {
    const Company *c = Company::Get(this->window_number);
    bool local = this->window_number == _local_company;
 
    if (!this->IsShaded()) {
      bool reinit = false;
 
      /* Button bar selection. */
      reinit |= this->GetWidget<NWidgetStacked>(WID_C_SELECT_BUTTONS)->SetDisplayedPlane(local ? 0 : SZSP_NONE);
 
      /* Build HQ button handling. */
      reinit |= this->GetWidget<NWidgetStacked>(WID_C_SELECT_VIEW_BUILD_HQ)->SetDisplayedPlane((local && c->location_of_HQ == INVALID_TILE) ? CWP_VB_BUILD : CWP_VB_VIEW);
 
      this->SetWidgetDisabledState(WID_C_VIEW_HQ, c->location_of_HQ == INVALID_TILE);
 
      /* Enable/disable 'Relocate HQ' button. */
      reinit |= this->GetWidget<NWidgetStacked>(WID_C_SELECT_RELOCATE)->SetDisplayedPlane((!local || c->location_of_HQ == INVALID_TILE) ? CWP_RELOCATE_HIDE : CWP_RELOCATE_SHOW);
      /* Enable/disable 'Give money' button. */
      reinit |= this->GetWidget<NWidgetStacked>(WID_C_SELECT_GIVE_MONEY)->SetDisplayedPlane((local || _local_company == COMPANY_SPECTATOR || !_settings_game.economy.give_money) ? SZSP_NONE : 0);
      /* Enable/disable 'Hostile Takeover' button. */
      reinit |= this->GetWidget<NWidgetStacked>(WID_C_SELECT_HOSTILE_TAKEOVER)->SetDisplayedPlane((local || _local_company == COMPANY_SPECTATOR || !c->is_ai || _networking) ? SZSP_NONE : 0);
 
      /* Multiplayer buttons. */
      reinit |= this->GetWidget<NWidgetStacked>(WID_C_SELECT_MULTIPLAYER)->SetDisplayedPlane((!_networking || !NetworkCanJoinCompany(c->index) || _local_company == c->index) ? (int)SZSP_NONE : 0);
 
      this->SetWidgetDisabledState(WID_C_COMPANY_JOIN, c->is_ai);
 
      if (reinit) {
        this->ReInit();
        return;
      }
    }
 
    this->DrawWidgets();
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    switch (widget) {
      case WID_C_FACE:
        size = maxdim(size, GetScaledSpriteSize(SPR_GRADIENT));
        break;
 
      case WID_C_DESC_COLOUR_SCHEME_EXAMPLE: {
        Point offset;
        Dimension d = GetSpriteSize(SPR_VEH_BUS_SW_VIEW, &offset);
        d.width -= offset.x;
        d.height -= offset.y;
        size = maxdim(size, d);
        break;
      }
 
      case WID_C_DESC_COMPANY_VALUE:
        /* INT64_MAX is arguably the maximum company value */
        size.width = GetStringBoundingBox(GetString(STR_COMPANY_VIEW_COMPANY_VALUE, INT64_MAX)).width;
        break;
 
      case WID_C_DESC_VEHICLE_COUNTS: {
        uint64_t max_value = GetParamMaxValue(5000); // Maximum number of vehicles
        for (const auto &count_string : _company_view_vehicle_count_strings) {
          size.width = std::max(size.width, GetStringBoundingBox(GetString(count_string, max_value)).width + padding.width);
        }
        break;
      }
 
      case WID_C_DESC_INFRASTRUCTURE_COUNTS: {
        uint64_t max_value = GetParamMaxValue(UINT_MAX);
        size.width = GetStringBoundingBox(GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_RAIL, max_value)).width;
        size.width = std::max(size.width, GetStringBoundingBox(GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_ROAD, max_value)).width);
        size.width = std::max(size.width, GetStringBoundingBox(GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_WATER, max_value)).width);
        size.width = std::max(size.width, GetStringBoundingBox(GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_STATION, max_value)).width);
        size.width = std::max(size.width, GetStringBoundingBox(GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_AIRPORT, max_value)).width);
        size.width = std::max(size.width, GetStringBoundingBox(GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_NONE, max_value)).width);
        size.width += padding.width;
        break;
      }
 
      case WID_C_VIEW_HQ:
      case WID_C_BUILD_HQ:
      case WID_C_RELOCATE_HQ:
      case WID_C_VIEW_INFRASTRUCTURE:
      case WID_C_GIVE_MONEY:
      case WID_C_HOSTILE_TAKEOVER:
      case WID_C_COMPANY_JOIN:
        size.width = GetStringBoundingBox(STR_COMPANY_VIEW_VIEW_HQ_BUTTON).width;
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_VIEW_BUILD_HQ_BUTTON).width);
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_VIEW_RELOCATE_HQ).width);
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_BUTTON).width);
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_VIEW_GIVE_MONEY_BUTTON).width);
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_VIEW_HOSTILE_TAKEOVER_BUTTON).width);
        size.width = std::max(size.width, GetStringBoundingBox(STR_COMPANY_VIEW_JOIN).width);
        size.width += padding.width;
        break;
    }
  }
 
  void DrawVehicleCountsWidget(const Rect &r, const Company *c) const
  {
    static_assert(VEH_COMPANY_END == lengthof(_company_view_vehicle_count_strings));
 
    int y = r.top;
    for (VehicleType type = VEH_BEGIN; type < VEH_COMPANY_END; type++) {
      uint amount = c->group_all[type].num_vehicle;
      if (amount != 0) {
        DrawString(r.left, r.right, y, GetString(_company_view_vehicle_count_strings[type], amount));
        y += GetCharacterHeight(FS_NORMAL);
      }
    }
 
    if (y == r.top) {
      /* No String was emitted before, so there must be no vehicles at all. */
      DrawString(r.left, r.right, y, STR_COMPANY_VIEW_VEHICLES_NONE);
    }
  }
 
  void DrawInfrastructureCountsWidget(const Rect &r, const Company *c) const
  {
    int y = r.top;
 
    uint rail_pieces = c->infrastructure.signal + c->infrastructure.GetRailTotal();
    if (rail_pieces != 0) {
      DrawString(r.left, r.right, y, GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_RAIL, rail_pieces));
      y += GetCharacterHeight(FS_NORMAL);
    }
 
    /* GetRoadTotal() skips tram pieces, but we actually want road and tram here. */
    uint road_pieces = std::accumulate(std::begin(c->infrastructure.road), std::end(c->infrastructure.road), 0U);
    if (road_pieces != 0) {
      DrawString(r.left, r.right, y, GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_ROAD, road_pieces));
      y += GetCharacterHeight(FS_NORMAL);
    }
 
    if (c->infrastructure.water != 0) {
      DrawString(r.left, r.right, y, GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_WATER, c->infrastructure.water));
      y += GetCharacterHeight(FS_NORMAL);
    }
 
    if (c->infrastructure.station != 0) {
      DrawString(r.left, r.right, y, GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_STATION, c->infrastructure.station));
      y += GetCharacterHeight(FS_NORMAL);
    }
 
    if (c->infrastructure.airport != 0) {
      DrawString(r.left, r.right, y, GetString(STR_COMPANY_VIEW_INFRASTRUCTURE_AIRPORT, c->infrastructure.airport));
      y += GetCharacterHeight(FS_NORMAL);
    }
 
    if (y == r.top) {
      /* No String was emitted before, so there must be no infrastructure at all. */
      DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_NONE);
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    const Company *c = Company::Get(this->window_number);
    switch (widget) {
      case WID_C_FACE:
        DrawCompanyManagerFace(c->face, c->colour, r);
        break;
 
      case WID_C_FACE_TITLE:
        DrawStringMultiLine(r, GetString(STR_COMPANY_VIEW_PRESIDENT_MANAGER_TITLE, c->index), TC_FROMSTRING, SA_HOR_CENTER);
        break;
 
      case WID_C_DESC_COLOUR_SCHEME_EXAMPLE: {
        Point offset;
        Dimension d = GetSpriteSize(SPR_VEH_BUS_SW_VIEW, &offset);
        d.height -= offset.y;
        DrawSprite(SPR_VEH_BUS_SW_VIEW, COMPANY_SPRITE_COLOUR(c->index), r.left - offset.x, CenterBounds(r.top, r.bottom, d.height) - offset.y);
        break;
      }
 
      case WID_C_DESC_VEHICLE_COUNTS:
        DrawVehicleCountsWidget(r, c);
        break;
 
      case WID_C_DESC_INFRASTRUCTURE_COUNTS:
        DrawInfrastructureCountsWidget(r, c);
        break;
    }
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    switch (widget) {
      case WID_C_CAPTION:
        return GetString(STR_COMPANY_VIEW_CAPTION, this->window_number, this->window_number);
 
      case WID_C_DESC_INAUGURATION: {
        const Company &c = *Company::Get(this->window_number);
        if (TimerGameEconomy::UsingWallclockUnits()) {
          return GetString(STR_COMPANY_VIEW_INAUGURATED_TITLE_WALLCLOCK, c.inaugurated_year_calendar, c.inaugurated_year);
        }
        return GetString(STR_COMPANY_VIEW_INAUGURATED_TITLE, c.inaugurated_year);
      }
 
      case WID_C_DESC_COMPANY_VALUE:
        return GetString(STR_COMPANY_VIEW_COMPANY_VALUE, CalculateCompanyValue(Company::Get(this->window_number)));
 
      default:
        return this->Window::GetWidgetString(widget, stringid);
    }
  }
 
  void OnResize() override
  {
    NWidgetResizeBase *wid = this->GetWidget<NWidgetResizeBase>(WID_C_FACE_TITLE);
    int y = GetStringHeight(GetString(STR_COMPANY_VIEW_PRESIDENT_MANAGER_TITLE, this->owner), wid->current_x);
    if (wid->UpdateVerticalSize(y)) this->ReInit(0, 0);
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      case WID_C_NEW_FACE: DoSelectCompanyManagerFace(this); break;
 
      case WID_C_COLOUR_SCHEME:
        ShowCompanyLiveryWindow(this->window_number, GroupID::Invalid());
        break;
 
      case WID_C_PRESIDENT_NAME:
        this->query_widget = WID_C_PRESIDENT_NAME;
        ShowQueryString(GetString(STR_PRESIDENT_NAME, this->window_number), STR_COMPANY_VIEW_PRESIDENT_S_NAME_QUERY_CAPTION, MAX_LENGTH_PRESIDENT_NAME_CHARS, this, CS_ALPHANUMERAL, {QueryStringFlag::EnableDefault, QueryStringFlag::LengthIsInChars});
        break;
 
      case WID_C_COMPANY_NAME:
        this->query_widget = WID_C_COMPANY_NAME;
        ShowQueryString(GetString(STR_COMPANY_NAME, this->window_number), STR_COMPANY_VIEW_COMPANY_NAME_QUERY_CAPTION, MAX_LENGTH_COMPANY_NAME_CHARS, this, CS_ALPHANUMERAL, {QueryStringFlag::EnableDefault, QueryStringFlag::LengthIsInChars});
        break;
 
      case WID_C_VIEW_HQ: {
        TileIndex tile = Company::Get(this->window_number)->location_of_HQ;
        if (_ctrl_pressed) {
          ShowExtraViewportWindow(tile);
        } else {
          ScrollMainWindowToTile(tile);
        }
        break;
      }
 
      case WID_C_BUILD_HQ:
        if (this->window_number != _local_company) return;
        if (this->IsWidgetLowered(WID_C_BUILD_HQ)) {
          ResetObjectToPlace();
          this->RaiseButtons();
          break;
        }
        SetObjectToPlaceWnd(SPR_CURSOR_HQ, PAL_NONE, HT_RECT, this);
        SetTileSelectSize(2, 2);
        this->LowerWidget(WID_C_BUILD_HQ);
        this->SetWidgetDirty(WID_C_BUILD_HQ);
        break;
 
      case WID_C_RELOCATE_HQ:
        if (this->IsWidgetLowered(WID_C_RELOCATE_HQ)) {
          ResetObjectToPlace();
          this->RaiseButtons();
          break;
        }
        SetObjectToPlaceWnd(SPR_CURSOR_HQ, PAL_NONE, HT_RECT, this);
        SetTileSelectSize(2, 2);
        this->LowerWidget(WID_C_RELOCATE_HQ);
        this->SetWidgetDirty(WID_C_RELOCATE_HQ);
        break;
 
      case WID_C_VIEW_INFRASTRUCTURE:
        ShowCompanyInfrastructure(this->window_number);
        break;
 
      case WID_C_GIVE_MONEY:
        this->query_widget = WID_C_GIVE_MONEY;
        ShowQueryString({}, STR_COMPANY_VIEW_GIVE_MONEY_QUERY_CAPTION, 30, this, CS_NUMERAL, {});
        break;
 
      case WID_C_HOSTILE_TAKEOVER:
        ShowBuyCompanyDialog(this->window_number, true);
        break;
 
      case WID_C_COMPANY_JOIN: {
        this->query_widget = WID_C_COMPANY_JOIN;
        CompanyID company = this->window_number;
        if (_network_server) {
          NetworkServerDoMove(CLIENT_ID_SERVER, company);
          MarkWholeScreenDirty();
        } else {
          /* just send the join command */
          NetworkClientRequestMove(company);
        }
        break;
      }
    }
  }
 
  IntervalTimer<TimerWindow> redraw_interval = {std::chrono::seconds(3), [this](auto) {
    this->SetDirty();
  }};
 
  void OnPlaceObject([[maybe_unused]] Point pt, TileIndex tile) override
  {
    if (Command<CMD_BUILD_OBJECT>::Post(STR_ERROR_CAN_T_BUILD_COMPANY_HEADQUARTERS, tile, OBJECT_HQ, 0) && !_shift_pressed) {
      ResetObjectToPlace();
      this->RaiseButtons();
    }
  }
 
  void OnPlaceObjectAbort() override
  {
    this->RaiseButtons();
  }
 
  void OnQueryTextFinished(std::optional<std::string> str) override
  {
    if (!str.has_value()) return;
 
    switch (this->query_widget) {
      default: NOT_REACHED();
 
      case WID_C_GIVE_MONEY: {
        Money money = std::strtoull(str->c_str(), nullptr, 10) / GetCurrency().rate;
        Command<CMD_GIVE_MONEY>::Post(STR_ERROR_CAN_T_GIVE_MONEY, money, this->window_number);
        break;
      }
 
      case WID_C_PRESIDENT_NAME:
        Command<CMD_RENAME_PRESIDENT>::Post(STR_ERROR_CAN_T_CHANGE_PRESIDENT, *str);
        break;
 
      case WID_C_COMPANY_NAME:
        Command<CMD_RENAME_COMPANY>::Post(STR_ERROR_CAN_T_CHANGE_COMPANY_NAME, *str);
        break;
    }
  }
 
  void OnInvalidateData([[maybe_unused]] int data = 0, [[maybe_unused]] bool gui_scope = true) override
  {
    if (gui_scope && data == 1) {
      /* Manually call OnResize to adjust minimum height of president name widget. */
      OnResize();
    }
  }
};
 
static WindowDesc _company_desc(
  WDP_AUTO, "company", 0, 0,
  WC_COMPANY, WC_NONE,
  {},
  _nested_company_widgets
);
 
void ShowCompany(CompanyID company)
{
  if (!Company::IsValidID(company)) return;
 
  AllocateWindowDescFront<CompanyWindow>(_company_desc, company);
}
 
void DirtyCompanyInfrastructureWindows(CompanyID company)
{
  SetWindowDirty(WC_COMPANY, company);
  SetWindowDirty(WC_COMPANY_INFRASTRUCTURE, company);
}
 
struct BuyCompanyWindow : Window {
  BuyCompanyWindow(WindowDesc &desc, WindowNumber window_number, bool hostile_takeover) : Window(desc), hostile_takeover(hostile_takeover)
  {
    this->InitNested(window_number);
 
    const Company *c = Company::Get(this->window_number);
    this->company_value = hostile_takeover ? CalculateHostileTakeoverValue(c) : c->bankrupt_value;
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    switch (widget) {
      case WID_BC_FACE:
        size = GetScaledSpriteSize(SPR_GRADIENT);
        break;
 
      case WID_BC_QUESTION:
        const Company *c = Company::Get(this->window_number);
        size.height = GetStringHeight(GetString(this->hostile_takeover ? STR_BUY_COMPANY_HOSTILE_TAKEOVER : STR_BUY_COMPANY_MESSAGE, c->index, this->company_value), size.width);
        break;
    }
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    switch (widget) {
      case WID_BC_CAPTION:
        return GetString(STR_ERROR_MESSAGE_CAPTION_OTHER_COMPANY, Company::Get(this->window_number)->index);
 
      default:
        return this->Window::GetWidgetString(widget, stringid);
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    switch (widget) {
      case WID_BC_FACE: {
        const Company *c = Company::Get(this->window_number);
        DrawCompanyManagerFace(c->face, c->colour, r);
        break;
      }
 
      case WID_BC_QUESTION: {
        const Company *c = Company::Get(this->window_number);
        DrawStringMultiLine(r, GetString(this->hostile_takeover ? STR_BUY_COMPANY_HOSTILE_TAKEOVER : STR_BUY_COMPANY_MESSAGE, c->index, this->company_value), TC_FROMSTRING, SA_CENTER);
        break;
      }
    }
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      case WID_BC_NO:
        this->Close();
        break;
 
      case WID_BC_YES:
        Command<CMD_BUY_COMPANY>::Post(STR_ERROR_CAN_T_BUY_COMPANY, this->window_number, this->hostile_takeover);
        break;
    }
  }
 
  IntervalTimer<TimerWindow> rescale_interval = {std::chrono::seconds(3), [this](auto) {
    /* Value can't change when in bankruptcy. */
    if (!this->hostile_takeover) return;
 
    const Company *c = Company::Get(this->window_number);
    auto new_value = CalculateHostileTakeoverValue(c);
    if (new_value != this->company_value) {
      this->company_value = new_value;
      this->ReInit();
    }
  }};
 
private:
  bool hostile_takeover = false; 
  Money company_value{}; 
};
 
static constexpr NWidgetPart _nested_buy_company_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_LIGHT_BLUE),
    NWidget(WWT_CAPTION, COLOUR_LIGHT_BLUE, WID_BC_CAPTION),
  EndContainer(),
  NWidget(WWT_PANEL, COLOUR_LIGHT_BLUE),
    NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_wide, 0), SetPadding(WidgetDimensions::unscaled.modalpopup),
      NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_BC_FACE), SetFill(0, 1),
        NWidget(WWT_EMPTY, INVALID_COLOUR, WID_BC_QUESTION), SetMinimalSize(240, 0), SetFill(1, 1),
      EndContainer(),
      NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize), SetPIP(100, WidgetDimensions::unscaled.hsep_wide, 100),
        NWidget(WWT_TEXTBTN, COLOUR_LIGHT_BLUE, WID_BC_NO), SetMinimalSize(60, 12), SetStringTip(STR_QUIT_NO), SetFill(1, 0),
        NWidget(WWT_TEXTBTN, COLOUR_LIGHT_BLUE, WID_BC_YES), SetMinimalSize(60, 12), SetStringTip(STR_QUIT_YES), SetFill(1, 0),
      EndContainer(),
    EndContainer(),
  EndContainer(),
};
 
static WindowDesc _buy_company_desc(
  WDP_AUTO, nullptr, 0, 0,
  WC_BUY_COMPANY, WC_NONE,
  WindowDefaultFlag::Construction,
  _nested_buy_company_widgets
);
 
void ShowBuyCompanyDialog(CompanyID company, bool hostile_takeover)
{
  auto window = BringWindowToFrontById(WC_BUY_COMPANY, company);
  if (window == nullptr) {
    new BuyCompanyWindow(_buy_company_desc, company, hostile_takeover);
  }
}