company_gui.cpp

Go to the documentation of this file.
00001 /* $Id: company_gui.cpp 23531 2011-12-16 16:27:45Z truebrain $ */
00002 
00003 /*
00004  * This file is part of OpenTTD.
00005  * 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.
00006  * 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.
00007  * 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/>.
00008  */
00009 
00012 #include "stdafx.h"
00013 #include "error.h"
00014 #include "gui.h"
00015 #include "window_gui.h"
00016 #include "textbuf_gui.h"
00017 #include "viewport_func.h"
00018 #include "company_func.h"
00019 #include "command_func.h"
00020 #include "network/network.h"
00021 #include "network/network_gui.h"
00022 #include "network/network_func.h"
00023 #include "vehicle_func.h"
00024 #include "newgrf.h"
00025 #include "company_manager_face.h"
00026 #include "strings_func.h"
00027 #include "date_func.h"
00028 #include "widgets/dropdown_type.h"
00029 #include "tilehighlight_func.h"
00030 #include "sprite.h"
00031 #include "company_base.h"
00032 #include "core/geometry_func.hpp"
00033 #include "economy_func.h"
00034 #include "object_type.h"
00035 #include "rail.h"
00036 #include "engine_base.h"
00037 #include "window_func.h"
00038 #include "road_func.h"
00039 #include "water.h"
00040 #include "station_func.h"
00041 
00042 #include "widgets/company_widget.h"
00043 
00044 #include "table/strings.h"
00045 
00047 static const uint EXP_LINESPACE  = 2;      
00048 static const uint EXP_BLOCKSPACE = 10;     
00049 
00050 static void DoSelectCompanyManagerFace(Window *parent);
00051 static void ShowCompanyInfrastructure(CompanyID company);
00052 
00054 static ExpensesType _expenses_list_1[] = {
00055   EXPENSES_CONSTRUCTION,
00056   EXPENSES_NEW_VEHICLES,
00057   EXPENSES_TRAIN_RUN,
00058   EXPENSES_ROADVEH_RUN,
00059   EXPENSES_AIRCRAFT_RUN,
00060   EXPENSES_SHIP_RUN,
00061   EXPENSES_PROPERTY,
00062   EXPENSES_TRAIN_INC,
00063   EXPENSES_ROADVEH_INC,
00064   EXPENSES_AIRCRAFT_INC,
00065   EXPENSES_SHIP_INC,
00066   EXPENSES_LOAN_INT,
00067   EXPENSES_OTHER,
00068 };
00069 
00071 static ExpensesType _expenses_list_2[] = {
00072   EXPENSES_TRAIN_INC,
00073   EXPENSES_ROADVEH_INC,
00074   EXPENSES_AIRCRAFT_INC,
00075   EXPENSES_SHIP_INC,
00076   INVALID_EXPENSES,
00077   EXPENSES_TRAIN_RUN,
00078   EXPENSES_ROADVEH_RUN,
00079   EXPENSES_AIRCRAFT_RUN,
00080   EXPENSES_SHIP_RUN,
00081   EXPENSES_PROPERTY,
00082   EXPENSES_LOAN_INT,
00083   INVALID_EXPENSES,
00084   EXPENSES_CONSTRUCTION,
00085   EXPENSES_NEW_VEHICLES,
00086   EXPENSES_OTHER,
00087   INVALID_EXPENSES,
00088 };
00089 
00091 struct ExpensesList {
00092   const ExpensesType *et;   
00093   const uint length;        
00094   const uint num_subtotals; 
00095 
00096   ExpensesList(ExpensesType *et, int length, int num_subtotals) : et(et), length(length), num_subtotals(num_subtotals)
00097   {
00098   }
00099 
00100   uint GetHeight() const
00101   {
00102     /* heading + line + texts of expenses + sub-totals + total line + total text */
00103     return FONT_HEIGHT_NORMAL + EXP_LINESPACE + this->length * FONT_HEIGHT_NORMAL + num_subtotals * (EXP_BLOCKSPACE + EXP_LINESPACE) + EXP_LINESPACE + FONT_HEIGHT_NORMAL;
00104   }
00105 
00107   uint GetCategoriesWidth() const
00108   {
00109     uint width = 0;
00110     bool invalid_expenses_measured = false; // Measure 'Total' width only once.
00111     for (uint i = 0; i < this->length; i++) {
00112       ExpensesType et = this->et[i];
00113       if (et == INVALID_EXPENSES) {
00114         if (!invalid_expenses_measured) {
00115           width = max(width, GetStringBoundingBox(STR_FINANCES_TOTAL_CAPTION).width);
00116           invalid_expenses_measured = true;
00117         }
00118       } else {
00119         width = max(width, GetStringBoundingBox(STR_FINANCES_SECTION_CONSTRUCTION + et).width);
00120       }
00121     }
00122     return width;
00123   }
00124 };
00125 
00126 static const ExpensesList _expenses_list_types[] = {
00127   ExpensesList(_expenses_list_1, lengthof(_expenses_list_1), 0),
00128   ExpensesList(_expenses_list_2, lengthof(_expenses_list_2), 3),
00129 };
00130 
00136 static void DrawCategories(const Rect &r)
00137 {
00138   int y = r.top;
00139 
00140   DrawString(r.left, r.right, y, STR_FINANCES_EXPENDITURE_INCOME_TITLE, TC_FROMSTRING, SA_HOR_CENTER, true);
00141   y += FONT_HEIGHT_NORMAL + EXP_LINESPACE;
00142 
00143   int type = _settings_client.gui.expenses_layout;
00144   for (uint i = 0; i < _expenses_list_types[type].length; i++) {
00145     const ExpensesType et = _expenses_list_types[type].et[i];
00146     if (et == INVALID_EXPENSES) {
00147       y += EXP_LINESPACE;
00148       DrawString(r.left, r.right, y, STR_FINANCES_TOTAL_CAPTION, TC_FROMSTRING, SA_RIGHT);
00149       y += FONT_HEIGHT_NORMAL + EXP_BLOCKSPACE;
00150     } else {
00151       DrawString(r.left, r.right, y, STR_FINANCES_SECTION_CONSTRUCTION + et);
00152       y += FONT_HEIGHT_NORMAL;
00153     }
00154   }
00155 
00156   DrawString(r.left, r.right, y + EXP_LINESPACE, STR_FINANCES_TOTAL_CAPTION, TC_FROMSTRING, SA_RIGHT);
00157 }
00158 
00166 static void DrawPrice(Money amount, int left, int right, int top)
00167 {
00168   StringID str = STR_FINANCES_NEGATIVE_INCOME;
00169   if (amount < 0) {
00170     amount = -amount;
00171     str++;
00172   }
00173   SetDParam(0, amount);
00174   DrawString(left, right, top, str, TC_FROMSTRING, SA_RIGHT);
00175 }
00176 
00184 static void DrawYearColumn(const Rect &r, int year, const Money (*tbl)[EXPENSES_END])
00185 {
00186   int y = r.top;
00187 
00188   SetDParam(0, year);
00189   DrawString(r.left, r.right, y, STR_FINANCES_YEAR, TC_FROMSTRING, SA_RIGHT, true);
00190   y += FONT_HEIGHT_NORMAL + EXP_LINESPACE;
00191 
00192   Money sum = 0;
00193   Money subtotal = 0;
00194   int type = _settings_client.gui.expenses_layout;
00195   for (uint i = 0; i < _expenses_list_types[type].length; i++) {
00196     const ExpensesType et = _expenses_list_types[type].et[i];
00197     if (et == INVALID_EXPENSES) {
00198       Money cost = subtotal;
00199       subtotal = 0;
00200       GfxFillRect(r.left, y, r.right, y, PC_BLACK);
00201       y += EXP_LINESPACE;
00202       DrawPrice(cost, r.left, r.right, y);
00203       y += FONT_HEIGHT_NORMAL + EXP_BLOCKSPACE;
00204     } else {
00205       Money cost = (*tbl)[et];
00206       subtotal += cost;
00207       sum += cost;
00208       if (cost != 0) DrawPrice(cost, r.left, r.right, y);
00209       y += FONT_HEIGHT_NORMAL;
00210     }
00211   }
00212 
00213   GfxFillRect(r.left, y, r.right, y, PC_BLACK);
00214   y += EXP_LINESPACE;
00215   DrawPrice(sum, r.left, r.right, y);
00216 }
00217 
00218 static const NWidgetPart _nested_company_finances_widgets[] = {
00219   NWidget(NWID_HORIZONTAL),
00220     NWidget(WWT_CLOSEBOX, COLOUR_GREY),
00221     NWidget(WWT_CAPTION, COLOUR_GREY, WID_CF_CAPTION), SetDataTip(STR_FINANCES_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
00222     NWidget(WWT_IMGBTN, COLOUR_GREY, WID_CF_TOGGLE_SIZE), SetDataTip(SPR_LARGE_SMALL_WINDOW, STR_TOOLTIP_TOGGLE_LARGE_SMALL_WINDOW),
00223     NWidget(WWT_SHADEBOX, COLOUR_GREY),
00224     NWidget(WWT_STICKYBOX, COLOUR_GREY),
00225   EndContainer(),
00226   NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_PANEL),
00227     NWidget(WWT_PANEL, COLOUR_GREY),
00228       NWidget(NWID_HORIZONTAL), SetPadding(WD_FRAMERECT_TOP, WD_FRAMERECT_RIGHT, WD_FRAMERECT_BOTTOM, WD_FRAMERECT_LEFT), SetPIP(0, 9, 0),
00229         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_EXPS_CATEGORY), SetMinimalSize(120, 0), SetFill(0, 0),
00230         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_EXPS_PRICE1), SetMinimalSize(86, 0), SetFill(0, 0),
00231         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_EXPS_PRICE2), SetMinimalSize(86, 0), SetFill(0, 0),
00232         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_EXPS_PRICE3), SetMinimalSize(86, 0), SetFill(0, 0),
00233       EndContainer(),
00234     EndContainer(),
00235   EndContainer(),
00236   NWidget(WWT_PANEL, COLOUR_GREY),
00237     NWidget(NWID_HORIZONTAL), SetPadding(WD_FRAMERECT_TOP, WD_FRAMERECT_RIGHT, WD_FRAMERECT_BOTTOM, WD_FRAMERECT_LEFT),
00238       NWidget(NWID_VERTICAL), // Vertical column with 'bank balance', 'loan'
00239         NWidget(WWT_TEXT, COLOUR_GREY), SetDataTip(STR_FINANCES_BANK_BALANCE_TITLE, STR_NULL), SetFill(1, 0),
00240         NWidget(WWT_TEXT, COLOUR_GREY), SetDataTip(STR_FINANCES_LOAN_TITLE, STR_NULL), SetFill(1, 0),
00241         NWidget(NWID_SPACER), SetFill(0, 1),
00242       EndContainer(),
00243       NWidget(NWID_SPACER), SetFill(0, 0), SetMinimalSize(30, 0),
00244       NWidget(NWID_VERTICAL), // Vertical column with bank balance amount, loan amount, and total.
00245         NWidget(WWT_TEXT, COLOUR_GREY, WID_CF_BALANCE_VALUE), SetDataTip(STR_NULL, STR_NULL),
00246         NWidget(WWT_TEXT, COLOUR_GREY, WID_CF_LOAN_VALUE), SetDataTip(STR_NULL, STR_NULL),
00247         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_LOAN_LINE), SetMinimalSize(0, 2), SetFill(1, 0),
00248         NWidget(WWT_TEXT, COLOUR_GREY, WID_CF_TOTAL_VALUE), SetDataTip(STR_NULL, STR_NULL),
00249       EndContainer(),
00250       NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_MAXLOAN),
00251         NWidget(NWID_HORIZONTAL),
00252           NWidget(NWID_SPACER), SetFill(0, 1), SetMinimalSize(25, 0),
00253           NWidget(NWID_VERTICAL), // Max loan information
00254             NWidget(WWT_EMPTY, COLOUR_GREY, WID_CF_MAXLOAN_GAP), SetFill(0, 0),
00255             NWidget(WWT_TEXT, COLOUR_GREY, WID_CF_MAXLOAN_VALUE), SetDataTip(STR_FINANCES_MAX_LOAN, STR_NULL),
00256             NWidget(NWID_SPACER), SetFill(0, 1),
00257           EndContainer(),
00258         EndContainer(),
00259       EndContainer(),
00260       NWidget(NWID_SPACER), SetFill(1, 1),
00261     EndContainer(),
00262   EndContainer(),
00263   NWidget(NWID_SELECTION, INVALID_COLOUR, WID_CF_SEL_BUTTONS),
00264     NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
00265       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_INCREASE_LOAN), SetFill(1, 0), SetDataTip(STR_FINANCES_BORROW_BUTTON, STR_FINANCES_BORROW_TOOLTIP),
00266       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_REPAY_LOAN), SetFill(1, 0), SetDataTip(STR_FINANCES_REPAY_BUTTON, STR_FINANCES_REPAY_TOOLTIP),
00267       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CF_INFRASTRUCTURE), SetFill(1, 0), SetDataTip(STR_FINANCES_INFRASTRUCTURE_BUTTON, STR_COMPANY_VIEW_INFRASTRUCTURE_TOOLTIP),
00268     EndContainer(),
00269   EndContainer(),
00270 };
00271 
00276 struct CompanyFinancesWindow : Window {
00277   static Money max_money; 
00278   bool small;             
00279 
00280   CompanyFinancesWindow(const WindowDesc *desc, CompanyID company) : Window()
00281   {
00282     this->small = false;
00283     this->CreateNestedTree(desc);
00284     this->SetupWidgets();
00285     this->FinishInitNested(desc, company);
00286 
00287     this->owner = (Owner)this->window_number;
00288   }
00289 
00290   virtual void SetStringParameters(int widget) const
00291   {
00292     switch (widget) {
00293       case WID_CF_CAPTION:
00294         SetDParam(0, (CompanyID)this->window_number);
00295         SetDParam(1, (CompanyID)this->window_number);
00296         break;
00297 
00298       case WID_CF_MAXLOAN_VALUE:
00299         SetDParam(0, _economy.max_loan);
00300         break;
00301 
00302       case WID_CF_INCREASE_LOAN:
00303       case WID_CF_REPAY_LOAN:
00304         SetDParam(0, LOAN_INTERVAL);
00305         break;
00306     }
00307   }
00308 
00309   virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
00310   {
00311     int type = _settings_client.gui.expenses_layout;
00312     switch (widget) {
00313       case WID_CF_EXPS_CATEGORY:
00314         size->width  = _expenses_list_types[type].GetCategoriesWidth();
00315         size->height = _expenses_list_types[type].GetHeight();
00316         break;
00317 
00318       case WID_CF_EXPS_PRICE1:
00319       case WID_CF_EXPS_PRICE2:
00320       case WID_CF_EXPS_PRICE3:
00321         size->height = _expenses_list_types[type].GetHeight();
00322         /* FALL THROUGH */
00323       case WID_CF_BALANCE_VALUE:
00324       case WID_CF_LOAN_VALUE:
00325       case WID_CF_TOTAL_VALUE:
00326         SetDParam(0, CompanyFinancesWindow::max_money);
00327         size->width = max(GetStringBoundingBox(STR_FINANCES_NEGATIVE_INCOME).width, GetStringBoundingBox(STR_FINANCES_POSITIVE_INCOME).width) + padding.width;
00328         break;
00329 
00330       case WID_CF_MAXLOAN_GAP:
00331         size->height = FONT_HEIGHT_NORMAL;
00332         break;
00333     }
00334   }
00335 
00336   virtual void DrawWidget(const Rect &r, int widget) const
00337   {
00338     switch (widget) {
00339       case WID_CF_EXPS_CATEGORY:
00340         DrawCategories(r);
00341         break;
00342 
00343       case WID_CF_EXPS_PRICE1:
00344       case WID_CF_EXPS_PRICE2:
00345       case WID_CF_EXPS_PRICE3: {
00346         const Company *c = Company::Get((CompanyID)this->window_number);
00347         int age = min(_cur_year - c->inaugurated_year, 2);
00348         int wid_offset = widget - WID_CF_EXPS_PRICE1;
00349         if (wid_offset <= age) {
00350           DrawYearColumn(r, _cur_year - (age - wid_offset), c->yearly_expenses + (age - wid_offset));
00351         }
00352         break;
00353       }
00354 
00355       case WID_CF_BALANCE_VALUE: {
00356         const Company *c = Company::Get((CompanyID)this->window_number);
00357         SetDParam(0, c->money);
00358         DrawString(r.left, r.right, r.top, STR_FINANCES_TOTAL_CURRENCY, TC_FROMSTRING, SA_RIGHT);
00359         break;
00360       }
00361 
00362       case WID_CF_LOAN_VALUE: {
00363         const Company *c = Company::Get((CompanyID)this->window_number);
00364         SetDParam(0, c->current_loan);
00365         DrawString(r.left, r.right, r.top, STR_FINANCES_TOTAL_CURRENCY, TC_FROMSTRING, SA_RIGHT);
00366         break;
00367       }
00368 
00369       case WID_CF_TOTAL_VALUE: {
00370         const Company *c = Company::Get((CompanyID)this->window_number);
00371         SetDParam(0, c->money - c->current_loan);
00372         DrawString(r.left, r.right, r.top, STR_FINANCES_TOTAL_CURRENCY, TC_FROMSTRING, SA_RIGHT);
00373         break;
00374       }
00375 
00376       case WID_CF_LOAN_LINE:
00377         GfxFillRect(r.left, r.top, r.right, r.top, PC_BLACK);
00378         break;
00379     }
00380   }
00381 
00386   void SetupWidgets()
00387   {
00388     int plane = this->small ? SZSP_NONE : 0;
00389     this->GetWidget<NWidgetStacked>(WID_CF_SEL_PANEL)->SetDisplayedPlane(plane);
00390     this->GetWidget<NWidgetStacked>(WID_CF_SEL_MAXLOAN)->SetDisplayedPlane(plane);
00391 
00392     CompanyID company = (CompanyID)this->window_number;
00393     plane = (company != _local_company) ? SZSP_NONE : 0;
00394     this->GetWidget<NWidgetStacked>(WID_CF_SEL_BUTTONS)->SetDisplayedPlane(plane);
00395   }
00396 
00397   virtual void OnPaint()
00398   {
00399     if (!this->IsShaded()) {
00400       if (!this->small) {
00401         /* Check that the expenses panel height matches the height needed for the layout. */
00402         int type = _settings_client.gui.expenses_layout;
00403         if (_expenses_list_types[type].GetHeight() != this->GetWidget<NWidgetBase>(WID_CF_EXPS_CATEGORY)->current_y) {
00404           this->SetupWidgets();
00405           this->ReInit();
00406           return;
00407         }
00408       }
00409 
00410       /* Check that the loan buttons are shown only when the user owns the company. */
00411       CompanyID company = (CompanyID)this->window_number;
00412       int req_plane = (company != _local_company) ? SZSP_NONE : 0;
00413       if (req_plane != this->GetWidget<NWidgetStacked>(WID_CF_SEL_BUTTONS)->shown_plane) {
00414         this->SetupWidgets();
00415         this->ReInit();
00416         return;
00417       }
00418 
00419       const Company *c = Company::Get(company);
00420       this->SetWidgetDisabledState(WID_CF_INCREASE_LOAN, c->current_loan == _economy.max_loan); // Borrow button only shows when there is any more money to loan.
00421       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.
00422     }
00423 
00424     this->DrawWidgets();
00425   }
00426 
00427   virtual void OnClick(Point pt, int widget, int click_count)
00428   {
00429     switch (widget) {
00430       case WID_CF_TOGGLE_SIZE: // toggle size
00431         this->small = !this->small;
00432         this->SetupWidgets();
00433         if (this->IsShaded()) {
00434           /* Finances window is not resizable, so size hints given during unshading have no effect
00435            * on the changed appearance of the window. */
00436           this->SetShaded(false);
00437         } else {
00438           this->ReInit();
00439         }
00440         break;
00441 
00442       case WID_CF_INCREASE_LOAN: // increase loan
00443         DoCommandP(0, 0, _ctrl_pressed, CMD_INCREASE_LOAN | CMD_MSG(STR_ERROR_CAN_T_BORROW_ANY_MORE_MONEY));
00444         break;
00445 
00446       case WID_CF_REPAY_LOAN: // repay loan
00447         DoCommandP(0, 0, _ctrl_pressed, CMD_DECREASE_LOAN | CMD_MSG(STR_ERROR_CAN_T_REPAY_LOAN));
00448         break;
00449 
00450       case WID_CF_INFRASTRUCTURE: // show infrastructure details
00451         ShowCompanyInfrastructure((CompanyID)this->window_number);
00452         break;
00453     }
00454   }
00455 
00456   virtual void OnHundredthTick()
00457   {
00458     const Company *c = Company::Get((CompanyID)this->window_number);
00459     if (c->money > CompanyFinancesWindow::max_money) {
00460       CompanyFinancesWindow::max_money = max(c->money * 2, CompanyFinancesWindow::max_money * 4);
00461       this->SetupWidgets();
00462       this->ReInit();
00463     }
00464   }
00465 };
00466 
00468 Money CompanyFinancesWindow::max_money = INT32_MAX;
00469 
00470 static const WindowDesc _company_finances_desc(
00471   WDP_AUTO, 0, 0,
00472   WC_FINANCES, WC_NONE,
00473   WDF_UNCLICK_BUTTONS,
00474   _nested_company_finances_widgets, lengthof(_nested_company_finances_widgets)
00475 );
00476 
00482 void ShowCompanyFinances(CompanyID company)
00483 {
00484   if (!Company::IsValidID(company)) return;
00485   if (BringWindowToFrontById(WC_FINANCES, company)) return;
00486 
00487   new CompanyFinancesWindow(&_company_finances_desc, company);
00488 }
00489 
00490 /* List of colours for the livery window */
00491 static const StringID _colour_dropdown[] = {
00492   STR_COLOUR_DARK_BLUE,
00493   STR_COLOUR_PALE_GREEN,
00494   STR_COLOUR_PINK,
00495   STR_COLOUR_YELLOW,
00496   STR_COLOUR_RED,
00497   STR_COLOUR_LIGHT_BLUE,
00498   STR_COLOUR_GREEN,
00499   STR_COLOUR_DARK_GREEN,
00500   STR_COLOUR_BLUE,
00501   STR_COLOUR_CREAM,
00502   STR_COLOUR_MAUVE,
00503   STR_COLOUR_PURPLE,
00504   STR_COLOUR_ORANGE,
00505   STR_COLOUR_BROWN,
00506   STR_COLOUR_GREY,
00507   STR_COLOUR_WHITE,
00508 };
00509 
00510 /* Association of liveries to livery classes */
00511 static const LiveryClass _livery_class[LS_END] = {
00512   LC_OTHER,
00513   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,
00514   LC_ROAD, LC_ROAD,
00515   LC_SHIP, LC_SHIP,
00516   LC_AIRCRAFT, LC_AIRCRAFT, LC_AIRCRAFT,
00517   LC_ROAD, LC_ROAD,
00518 };
00519 
00520 class DropDownListColourItem : public DropDownListItem {
00521 public:
00522   DropDownListColourItem(int result, bool masked) : DropDownListItem(result, masked) {}
00523 
00524   virtual ~DropDownListColourItem() {}
00525 
00526   StringID String() const
00527   {
00528     return _colour_dropdown[this->result];
00529   }
00530 
00531   uint Height(uint width) const
00532   {
00533     return max(FONT_HEIGHT_NORMAL, (byte)14);
00534   }
00535 
00536   bool Selectable() const
00537   {
00538     return true;
00539   }
00540 
00541   void Draw(int left, int right, int top, int bottom, bool sel, int bg_colour) const
00542   {
00543     bool rtl = _current_text_dir == TD_RTL;
00544     DrawSprite(SPR_VEH_BUS_SIDE_VIEW, PALETTE_RECOLOUR_START + this->result, rtl ? right - 16 : left + 16, top + 7);
00545     DrawString(rtl ? left + 2 : left + 32, rtl ? right - 32 : right - 2, top + max(0, 13 - FONT_HEIGHT_NORMAL), this->String(), sel ? TC_WHITE : TC_BLACK);
00546   }
00547 };
00548 
00550 struct SelectCompanyLiveryWindow : public Window {
00551 private:
00552   uint32 sel;
00553   LiveryClass livery_class;
00554   Dimension square;
00555   Dimension box;
00556   uint line_height;
00557 
00558   void ShowColourDropDownMenu(uint32 widget)
00559   {
00560     uint32 used_colours = 0;
00561     const Livery *livery;
00562     LiveryScheme scheme;
00563 
00564     /* Disallow other company colours for the primary colour */
00565     if (HasBit(this->sel, LS_DEFAULT) && widget == WID_SCL_PRI_COL_DROPDOWN) {
00566       const Company *c;
00567       FOR_ALL_COMPANIES(c) {
00568         if (c->index != _local_company) SetBit(used_colours, c->colour);
00569       }
00570     }
00571 
00572     /* Get the first selected livery to use as the default dropdown item */
00573     for (scheme = LS_BEGIN; scheme < LS_END; scheme++) {
00574       if (HasBit(this->sel, scheme)) break;
00575     }
00576     if (scheme == LS_END) scheme = LS_DEFAULT;
00577     livery = &Company::Get((CompanyID)this->window_number)->livery[scheme];
00578 
00579     DropDownList *list = new DropDownList();
00580     for (uint i = 0; i < lengthof(_colour_dropdown); i++) {
00581       list->push_back(new DropDownListColourItem(i, HasBit(used_colours, i)));
00582     }
00583 
00584     ShowDropDownList(this, list, widget == WID_SCL_PRI_COL_DROPDOWN ? livery->colour1 : livery->colour2, widget);
00585   }
00586 
00587 public:
00588   SelectCompanyLiveryWindow(const WindowDesc *desc, CompanyID company) : Window()
00589   {
00590     this->livery_class = LC_OTHER;
00591     this->sel = 1;
00592 
00593     this->square = GetSpriteSize(SPR_SQUARE);
00594     this->box    = maxdim(GetSpriteSize(SPR_BOX_CHECKED), GetSpriteSize(SPR_BOX_EMPTY));
00595     this->line_height = max(max(this->square.height, this->box.height), (uint)FONT_HEIGHT_NORMAL) + 4;
00596 
00597     this->InitNested(desc, company);
00598     this->owner = company;
00599     this->LowerWidget(WID_SCL_CLASS_GENERAL);
00600     this->InvalidateData(1);
00601   }
00602 
00603   virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
00604   {
00605     switch (widget) {
00606       case WID_SCL_SPACER_DROPDOWN: {
00607         /* The matrix widget below needs enough room to print all the schemes. */
00608         Dimension d = {0, 0};
00609         for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
00610           d = maxdim(d, GetStringBoundingBox(STR_LIVERY_DEFAULT + scheme));
00611         }
00612         size->width = max(size->width, 5 + this->box.width + d.width + WD_FRAMERECT_RIGHT);
00613         break;
00614       }
00615 
00616       case WID_SCL_MATRIX: {
00617         uint livery_height = 0;
00618         for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
00619           if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
00620             livery_height++;
00621           }
00622         }
00623         size->height = livery_height * this->line_height;
00624         this->GetWidget<NWidgetCore>(WID_SCL_MATRIX)->widget_data = (livery_height << MAT_ROW_START) | (1 << MAT_COL_START);
00625         break;
00626       }
00627 
00628       case WID_SCL_SEC_COL_DROPDOWN:
00629         if (!_loaded_newgrf_features.has_2CC) {
00630           size->width = 0;
00631           break;
00632         }
00633         /* FALL THROUGH */
00634       case WID_SCL_PRI_COL_DROPDOWN: {
00635         int padding = this->square.width + NWidgetScrollbar::GetVerticalDimension().width + 10;
00636         for (const StringID *id = _colour_dropdown; id != endof(_colour_dropdown); id++) {
00637           size->width = max(size->width, GetStringBoundingBox(*id).width + padding);
00638         }
00639         break;
00640       }
00641     }
00642   }
00643 
00644   virtual void OnPaint()
00645   {
00646     /* Disable dropdown controls if no scheme is selected */
00647     this->SetWidgetDisabledState(WID_SCL_PRI_COL_DROPDOWN, this->sel == 0);
00648     this->SetWidgetDisabledState(WID_SCL_SEC_COL_DROPDOWN, this->sel == 0);
00649 
00650     this->DrawWidgets();
00651   }
00652 
00653   virtual void SetStringParameters(int widget) const
00654   {
00655     switch (widget) {
00656       case WID_SCL_PRI_COL_DROPDOWN:
00657       case WID_SCL_SEC_COL_DROPDOWN: {
00658         const Company *c = Company::Get((CompanyID)this->window_number);
00659         LiveryScheme scheme = LS_DEFAULT;
00660 
00661         if (this->sel != 0) {
00662           for (scheme = LS_BEGIN; scheme < LS_END; scheme++) {
00663             if (HasBit(this->sel, scheme)) break;
00664           }
00665           if (scheme == LS_END) scheme = LS_DEFAULT;
00666         }
00667         SetDParam(0, STR_COLOUR_DARK_BLUE + ((widget == WID_SCL_PRI_COL_DROPDOWN) ? c->livery[scheme].colour1 : c->livery[scheme].colour2));
00668         break;
00669       }
00670     }
00671   }
00672 
00673   virtual void DrawWidget(const Rect &r, int widget) const
00674   {
00675     if (widget != WID_SCL_MATRIX) return;
00676 
00677     bool rtl = _current_text_dir == TD_RTL;
00678 
00679     /* Horizontal coordinates of scheme name column. */
00680     const NWidgetBase *nwi = this->GetWidget<NWidgetBase>(WID_SCL_SPACER_DROPDOWN);
00681     int sch_left = nwi->pos_x;
00682     int sch_right = sch_left + nwi->current_x - 1;
00683     /* Horizontal coordinates of first dropdown. */
00684     nwi = this->GetWidget<NWidgetBase>(WID_SCL_PRI_COL_DROPDOWN);
00685     int pri_left = nwi->pos_x;
00686     int pri_right = pri_left + nwi->current_x - 1;
00687     /* Horizontal coordinates of second dropdown. */
00688     nwi = this->GetWidget<NWidgetBase>(WID_SCL_SEC_COL_DROPDOWN);
00689     int sec_left = nwi->pos_x;
00690     int sec_right = sec_left + nwi->current_x - 1;
00691 
00692     int text_left  = (rtl ? (uint)WD_FRAMERECT_LEFT : (this->box.width + 5));
00693     int text_right = (rtl ? (this->box.width + 5) : (uint)WD_FRAMERECT_RIGHT);
00694 
00695     int box_offs    = (this->line_height - this->box.height) / 2;
00696     int square_offs = (this->line_height - this->square.height) / 2 + 1;
00697     int text_offs   = (this->line_height - FONT_HEIGHT_NORMAL) / 2 + 1;
00698 
00699     int y = r.top;
00700     const Company *c = Company::Get((CompanyID)this->window_number);
00701     for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
00702       if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
00703         bool sel = HasBit(this->sel, scheme) != 0;
00704 
00705         /* Optional check box + scheme name. */
00706         if (scheme != LS_DEFAULT) {
00707           DrawSprite(c->livery[scheme].in_use ? SPR_BOX_CHECKED : SPR_BOX_EMPTY, PAL_NONE, (rtl ? sch_right - (this->box.width + 5) + WD_FRAMERECT_RIGHT : sch_left) + WD_FRAMERECT_LEFT, y + box_offs);
00708         }
00709         DrawString(sch_left + text_left, sch_right - text_right, y + text_offs, STR_LIVERY_DEFAULT + scheme, sel ? TC_WHITE : TC_BLACK);
00710 
00711         /* Text below the first dropdown. */
00712         DrawSprite(SPR_SQUARE, GENERAL_SPRITE_COLOUR(c->livery[scheme].colour1), (rtl ? pri_right - (this->box.width + 5) + WD_FRAMERECT_RIGHT : pri_left) + WD_FRAMERECT_LEFT, y + square_offs);
00713         DrawString(pri_left + text_left, pri_right - text_right, y + text_offs, STR_COLOUR_DARK_BLUE + c->livery[scheme].colour1, sel ? TC_WHITE : TC_GOLD);
00714 
00715         /* Text below the second dropdown. */
00716         if (sec_right > sec_left) { // Second dropdown has non-zero size.
00717           DrawSprite(SPR_SQUARE, GENERAL_SPRITE_COLOUR(c->livery[scheme].colour2), (rtl ? sec_right - (this->box.width + 5) + WD_FRAMERECT_RIGHT : sec_left) + WD_FRAMERECT_LEFT, y + square_offs);
00718           DrawString(sec_left + text_left, sec_right - text_right, y + text_offs, STR_COLOUR_DARK_BLUE + c->livery[scheme].colour2, sel ? TC_WHITE : TC_GOLD);
00719         }
00720 
00721         y += this->line_height;
00722       }
00723     }
00724   }
00725 
00726   virtual void OnClick(Point pt, int widget, int click_count)
00727   {
00728     switch (widget) {
00729       /* Livery Class buttons */
00730       case WID_SCL_CLASS_GENERAL:
00731       case WID_SCL_CLASS_RAIL:
00732       case WID_SCL_CLASS_ROAD:
00733       case WID_SCL_CLASS_SHIP:
00734       case WID_SCL_CLASS_AIRCRAFT:
00735         this->RaiseWidget(this->livery_class + WID_SCL_CLASS_GENERAL);
00736         this->livery_class = (LiveryClass)(widget - WID_SCL_CLASS_GENERAL);
00737         this->LowerWidget(this->livery_class + WID_SCL_CLASS_GENERAL);
00738 
00739         /* Select the first item in the list */
00740         this->sel = 0;
00741         for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
00742           if (_livery_class[scheme] == this->livery_class && HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
00743             this->sel = 1 << scheme;
00744             break;
00745           }
00746         }
00747 
00748         this->ReInit();
00749         break;
00750 
00751       case WID_SCL_PRI_COL_DROPDOWN: // First colour dropdown
00752         ShowColourDropDownMenu(WID_SCL_PRI_COL_DROPDOWN);
00753         break;
00754 
00755       case WID_SCL_SEC_COL_DROPDOWN: // Second colour dropdown
00756         ShowColourDropDownMenu(WID_SCL_SEC_COL_DROPDOWN);
00757         break;
00758 
00759       case WID_SCL_MATRIX: {
00760         const NWidgetBase *wid = this->GetWidget<NWidgetBase>(WID_SCL_MATRIX);
00761         LiveryScheme j = (LiveryScheme)((pt.y - wid->pos_y) / this->line_height);
00762 
00763         for (LiveryScheme scheme = LS_BEGIN; scheme <= j; scheme++) {
00764           if (_livery_class[scheme] != this->livery_class || !HasBit(_loaded_newgrf_features.used_liveries, scheme)) j++;
00765           if (scheme >= LS_END) return;
00766         }
00767         if (j >= LS_END) return;
00768 
00769         /* If clicking on the left edge, toggle using the livery */
00770         if (_current_text_dir == TD_RTL ? pt.x - wid->pos_x > wid->current_x - (this->box.width + 5) : pt.x - wid->pos_x < (this->box.width + 5)) {
00771           DoCommandP(0, j | (2 << 8), !Company::Get((CompanyID)this->window_number)->livery[j].in_use, CMD_SET_COMPANY_COLOUR);
00772         }
00773 
00774         if (_ctrl_pressed) {
00775           ToggleBit(this->sel, j);
00776         } else {
00777           this->sel = 1 << j;
00778         }
00779         this->SetDirty();
00780         break;
00781       }
00782     }
00783   }
00784 
00785   virtual void OnDropdownSelect(int widget, int index)
00786   {
00787     for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
00788       if (HasBit(this->sel, scheme)) {
00789         DoCommandP(0, scheme | (widget == WID_SCL_PRI_COL_DROPDOWN ? 0 : 256), index, CMD_SET_COMPANY_COLOUR);
00790       }
00791     }
00792   }
00793 
00799   virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
00800   {
00801     if (!gui_scope) return;
00802     this->SetWidgetsDisabledState(true, WID_SCL_CLASS_RAIL, WID_SCL_CLASS_ROAD, WID_SCL_CLASS_SHIP, WID_SCL_CLASS_AIRCRAFT, WIDGET_LIST_END);
00803 
00804     bool current_class_valid = this->livery_class == LC_OTHER;
00805     if (_settings_client.gui.liveries == LIT_ALL || (_settings_client.gui.liveries == LIT_COMPANY && this->window_number == _local_company)) {
00806       for (LiveryScheme scheme = LS_DEFAULT; scheme < LS_END; scheme++) {
00807         if (HasBit(_loaded_newgrf_features.used_liveries, scheme)) {
00808           if (_livery_class[scheme] == this->livery_class) current_class_valid = true;
00809           this->EnableWidget(WID_SCL_CLASS_GENERAL + _livery_class[scheme]);
00810         } else {
00811           ClrBit(this->sel, scheme);
00812         }
00813       }
00814     }
00815 
00816     if (!current_class_valid) {
00817       Point pt = {0, 0};
00818       this->OnClick(pt, WID_SCL_CLASS_GENERAL, 1);
00819     } else if (data == 0) {
00820       this->ReInit();
00821     }
00822   }
00823 };
00824 
00825 static const NWidgetPart _nested_select_company_livery_widgets [] = {
00826   NWidget(NWID_HORIZONTAL),
00827     NWidget(WWT_CLOSEBOX, COLOUR_GREY),
00828     NWidget(WWT_CAPTION, COLOUR_GREY, WID_SCL_CAPTION), SetDataTip(STR_LIVERY_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
00829   EndContainer(),
00830   NWidget(NWID_HORIZONTAL),
00831     NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_GENERAL), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_COMPANY_GENERAL, STR_LIVERY_GENERAL_TOOLTIP),
00832     NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_RAIL), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_TRAINLIST, STR_LIVERY_TRAIN_TOOLTIP),
00833     NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_ROAD), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_TRUCKLIST, STR_LIVERY_ROAD_VEHICLE_TOOLTIP),
00834     NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_SHIP), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_SHIPLIST, STR_LIVERY_SHIP_TOOLTIP),
00835     NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCL_CLASS_AIRCRAFT), SetMinimalSize(22, 22), SetFill(0, 1), SetDataTip(SPR_IMG_AIRPLANESLIST, STR_LIVERY_AIRCRAFT_TOOLTIP),
00836     NWidget(WWT_PANEL, COLOUR_GREY), SetMinimalSize(90, 22), SetFill(1, 1), EndContainer(),
00837   EndContainer(),
00838   NWidget(NWID_HORIZONTAL),
00839     NWidget(WWT_PANEL, COLOUR_GREY, WID_SCL_SPACER_DROPDOWN), SetMinimalSize(150, 12), SetFill(1, 1), EndContainer(),
00840     NWidget(WWT_DROPDOWN, COLOUR_GREY, WID_SCL_PRI_COL_DROPDOWN), SetMinimalSize(125, 12), SetFill(0, 1), SetDataTip(STR_BLACK_STRING, STR_LIVERY_PRIMARY_TOOLTIP),
00841     NWidget(WWT_DROPDOWN, COLOUR_GREY, WID_SCL_SEC_COL_DROPDOWN), SetMinimalSize(125, 12), SetFill(0, 1),
00842         SetDataTip(STR_BLACK_STRING, STR_LIVERY_SECONDARY_TOOLTIP),
00843   EndContainer(),
00844   NWidget(WWT_MATRIX, COLOUR_GREY, WID_SCL_MATRIX), SetMinimalSize(275, 15), SetFill(1, 0), SetDataTip((1 << MAT_ROW_START) | (1 << MAT_COL_START), STR_LIVERY_PANEL_TOOLTIP),
00845 };
00846 
00847 static const WindowDesc _select_company_livery_desc(
00848   WDP_AUTO, 0, 0,
00849   WC_COMPANY_COLOUR, WC_NONE,
00850   0,
00851   _nested_select_company_livery_widgets, lengthof(_nested_select_company_livery_widgets)
00852 );
00853 
00861 void DrawCompanyManagerFace(CompanyManagerFace cmf, int colour, int x, int y)
00862 {
00863   GenderEthnicity ge = (GenderEthnicity)GetCompanyManagerFaceBits(cmf, CMFV_GEN_ETHN, GE_WM);
00864 
00865   bool has_moustache   = !HasBit(ge, GENDER_FEMALE) && GetCompanyManagerFaceBits(cmf, CMFV_HAS_MOUSTACHE,   ge) != 0;
00866   bool has_tie_earring = !HasBit(ge, GENDER_FEMALE) || GetCompanyManagerFaceBits(cmf, CMFV_HAS_TIE_EARRING, ge) != 0;
00867   bool has_glasses     = GetCompanyManagerFaceBits(cmf, CMFV_HAS_GLASSES, ge) != 0;
00868   PaletteID pal;
00869 
00870   /* Modify eye colour palette only if 2 or more valid values exist */
00871   if (_cmf_info[CMFV_EYE_COLOUR].valid_values[ge] < 2) {
00872     pal = PAL_NONE;
00873   } else {
00874     switch (GetCompanyManagerFaceBits(cmf, CMFV_EYE_COLOUR, ge)) {
00875       default: NOT_REACHED();
00876       case 0: pal = PALETTE_TO_BROWN; break;
00877       case 1: pal = PALETTE_TO_BLUE;  break;
00878       case 2: pal = PALETTE_TO_GREEN; break;
00879     }
00880   }
00881 
00882   /* Draw the gradient (background) */
00883   DrawSprite(SPR_GRADIENT, GENERAL_SPRITE_COLOUR(colour), x, y);
00884 
00885   for (CompanyManagerFaceVariable cmfv = CMFV_CHEEKS; cmfv < CMFV_END; cmfv++) {
00886     switch (cmfv) {
00887       case CMFV_MOUSTACHE:   if (!has_moustache)   continue; break;
00888       case CMFV_LIPS:        // FALL THROUGH
00889       case CMFV_NOSE:        if (has_moustache)    continue; break;
00890       case CMFV_TIE_EARRING: if (!has_tie_earring) continue; break;
00891       case CMFV_GLASSES:     if (!has_glasses)     continue; break;
00892       default: break;
00893     }
00894     DrawSprite(GetCompanyManagerFaceSprite(cmf, cmfv, ge), (cmfv == CMFV_EYEBROWS) ? pal : PAL_NONE, x, y);
00895   }
00896 }
00897 
00899 static const NWidgetPart _nested_select_company_manager_face_widgets[] = {
00900   NWidget(NWID_HORIZONTAL),
00901     NWidget(WWT_CLOSEBOX, COLOUR_GREY),
00902     NWidget(WWT_CAPTION, COLOUR_GREY, WID_SCMF_CAPTION), SetDataTip(STR_FACE_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
00903     NWidget(WWT_IMGBTN, COLOUR_GREY, WID_SCMF_TOGGLE_LARGE_SMALL), SetDataTip(SPR_LARGE_SMALL_WINDOW, STR_FACE_ADVANCED_TOOLTIP),
00904   EndContainer(),
00905   NWidget(WWT_PANEL, COLOUR_GREY, WID_SCMF_SELECT_FACE),
00906     NWidget(NWID_SPACER), SetMinimalSize(0, 2),
00907     NWidget(NWID_HORIZONTAL), SetPIP(2, 2, 2),
00908       NWidget(NWID_VERTICAL),
00909         NWidget(NWID_HORIZONTAL),
00910           NWidget(NWID_SPACER), SetFill(1, 0),
00911           NWidget(WWT_EMPTY, COLOUR_GREY, WID_SCMF_FACE), SetMinimalSize(92, 119),
00912           NWidget(NWID_SPACER), SetFill(1, 0),
00913         EndContainer(),
00914         NWidget(NWID_SPACER), SetMinimalSize(0, 2),
00915         NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_RANDOM_NEW_FACE), SetFill(1, 0), SetDataTip(STR_FACE_NEW_FACE_BUTTON, STR_FACE_NEW_FACE_TOOLTIP),
00916         NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_LOADSAVE), // Load/number/save buttons under the portrait in the advanced view.
00917           NWidget(NWID_VERTICAL),
00918             NWidget(NWID_SPACER), SetMinimalSize(0, 5), SetFill(0, 1),
00919             NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_LOAD), SetFill(1, 0), SetDataTip(STR_FACE_LOAD, STR_FACE_LOAD_TOOLTIP),
00920             NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_FACECODE), SetFill(1, 0), SetDataTip(STR_FACE_FACECODE, STR_FACE_FACECODE_TOOLTIP),
00921             NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_SAVE), SetFill(1, 0), SetDataTip(STR_FACE_SAVE, STR_FACE_SAVE_TOOLTIP),
00922             NWidget(NWID_SPACER), SetMinimalSize(0, 5), SetFill(0, 1),
00923           EndContainer(),
00924         EndContainer(),
00925       EndContainer(),
00926       NWidget(NWID_VERTICAL),
00927         NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON), SetFill(1, 0), SetDataTip(STR_FACE_ADVANCED, STR_FACE_ADVANCED_TOOLTIP),
00928         NWidget(NWID_SPACER), SetMinimalSize(0, 2),
00929         NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_MALEFEMALE), // Simple male/female face setting.
00930           NWidget(NWID_VERTICAL),
00931             NWidget(NWID_SPACER), SetFill(0, 1),
00932             NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_MALE), SetFill(1, 0), SetDataTip(STR_FACE_MALE_BUTTON, STR_FACE_MALE_TOOLTIP),
00933             NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_FEMALE), SetFill(1, 0), SetDataTip(STR_FACE_FEMALE_BUTTON, STR_FACE_FEMALE_TOOLTIP),
00934             NWidget(NWID_SPACER), SetFill(0, 1),
00935           EndContainer(),
00936         EndContainer(),
00937         NWidget(NWID_SELECTION, INVALID_COLOUR, WID_SCMF_SEL_PARTS), // Advanced face parts setting.
00938           NWidget(NWID_VERTICAL),
00939             NWidget(NWID_SPACER), SetMinimalSize(0, 2),
00940             NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
00941               NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_MALE2), SetFill(1, 0), SetDataTip(STR_FACE_MALE_BUTTON, STR_FACE_MALE_TOOLTIP),
00942               NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_FEMALE2), SetFill(1, 0), SetDataTip(STR_FACE_FEMALE_BUTTON, STR_FACE_FEMALE_TOOLTIP),
00943             EndContainer(),
00944             NWidget(NWID_SPACER), SetMinimalSize(0, 2),
00945             NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
00946               NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_ETHNICITY_EUR), SetFill(1, 0), SetDataTip(STR_FACE_EUROPEAN, STR_FACE_SELECT_EUROPEAN),
00947               NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SCMF_ETHNICITY_AFR), SetFill(1, 0), SetDataTip(STR_FACE_AFRICAN, STR_FACE_SELECT_AFRICAN),
00948             EndContainer(),
00949             NWidget(NWID_SPACER), SetMinimalSize(0, 4),
00950             NWidget(NWID_HORIZONTAL),
00951               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT), SetFill(1, 0),
00952               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAS_MOUSTACHE_EARRING), SetDataTip(STR_EMPTY, STR_FACE_MOUSTACHE_EARRING_TOOLTIP),
00953             EndContainer(),
00954             NWidget(NWID_HORIZONTAL),
00955               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_HAS_GLASSES_TEXT), SetFill(1, 0),
00956               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAS_GLASSES), SetDataTip(STR_EMPTY, STR_FACE_GLASSES_TOOLTIP),
00957             EndContainer(),
00958             NWidget(NWID_SPACER), SetMinimalSize(0, 2), SetFill(1, 0),
00959             NWidget(NWID_HORIZONTAL),
00960               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_HAIR_TEXT), SetFill(1, 0),
00961               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_HAIR_L), SetDataTip(AWV_DECREASE, STR_FACE_HAIR_TOOLTIP),
00962               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_HAIR), SetDataTip(STR_EMPTY, STR_FACE_HAIR_TOOLTIP),
00963               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_HAIR_R), SetDataTip(AWV_INCREASE, STR_FACE_HAIR_TOOLTIP),
00964             EndContainer(),
00965             NWidget(NWID_HORIZONTAL),
00966               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_EYEBROWS_TEXT), SetFill(1, 0),
00967               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYEBROWS_L), SetDataTip(AWV_DECREASE, STR_FACE_EYEBROWS_TOOLTIP),
00968               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_EYEBROWS), SetDataTip(STR_EMPTY, STR_FACE_EYEBROWS_TOOLTIP),
00969               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYEBROWS_R), SetDataTip(AWV_INCREASE, STR_FACE_EYEBROWS_TOOLTIP),
00970             EndContainer(),
00971             NWidget(NWID_HORIZONTAL),
00972               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_EYECOLOUR_TEXT), SetFill(1, 0),
00973               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR_L), SetDataTip(AWV_DECREASE, STR_FACE_EYECOLOUR_TOOLTIP),
00974               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR), SetDataTip(STR_EMPTY, STR_FACE_EYECOLOUR_TOOLTIP),
00975               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_EYECOLOUR_R), SetDataTip(AWV_INCREASE, STR_FACE_EYECOLOUR_TOOLTIP),
00976             EndContainer(),
00977             NWidget(NWID_HORIZONTAL),
00978               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_GLASSES_TEXT), SetFill(1, 0),
00979               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_GLASSES_L), SetDataTip(AWV_DECREASE, STR_FACE_GLASSES_TOOLTIP_2),
00980               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_GLASSES), SetDataTip(STR_EMPTY, STR_FACE_GLASSES_TOOLTIP_2),
00981               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_GLASSES_R), SetDataTip(AWV_INCREASE, STR_FACE_GLASSES_TOOLTIP_2),
00982             EndContainer(),
00983             NWidget(NWID_HORIZONTAL),
00984               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_NOSE_TEXT), SetFill(1, 0),
00985               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_NOSE_L), SetDataTip(AWV_DECREASE, STR_FACE_NOSE_TOOLTIP),
00986               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_NOSE), SetDataTip(STR_EMPTY, STR_FACE_NOSE_TOOLTIP),
00987               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_NOSE_R), SetDataTip(AWV_INCREASE, STR_FACE_NOSE_TOOLTIP),
00988             EndContainer(),
00989             NWidget(NWID_HORIZONTAL),
00990               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_LIPS_MOUSTACHE_TEXT), SetFill(1, 0),
00991               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE_L), SetDataTip(AWV_DECREASE, STR_FACE_LIPS_MOUSTACHE_TOOLTIP),
00992               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE), SetDataTip(STR_EMPTY, STR_FACE_LIPS_MOUSTACHE_TOOLTIP),
00993               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_LIPS_MOUSTACHE_R), SetDataTip(AWV_INCREASE, STR_FACE_LIPS_MOUSTACHE_TOOLTIP),
00994             EndContainer(),
00995             NWidget(NWID_HORIZONTAL),
00996               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_CHIN_TEXT), SetFill(1, 0),
00997               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_CHIN_L), SetDataTip(AWV_DECREASE, STR_FACE_CHIN_TOOLTIP),
00998               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_CHIN), SetDataTip(STR_EMPTY, STR_FACE_CHIN_TOOLTIP),
00999               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_CHIN_R), SetDataTip(AWV_INCREASE, STR_FACE_CHIN_TOOLTIP),
01000             EndContainer(),
01001             NWidget(NWID_HORIZONTAL),
01002               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_JACKET_TEXT), SetFill(1, 0),
01003               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_JACKET_L), SetDataTip(AWV_DECREASE, STR_FACE_JACKET_TOOLTIP),
01004               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_JACKET), SetDataTip(STR_EMPTY, STR_FACE_JACKET_TOOLTIP),
01005               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_JACKET_R), SetDataTip(AWV_INCREASE, STR_FACE_JACKET_TOOLTIP),
01006             EndContainer(),
01007             NWidget(NWID_HORIZONTAL),
01008               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_COLLAR_TEXT), SetFill(1, 0),
01009               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_COLLAR_L), SetDataTip(AWV_DECREASE, STR_FACE_COLLAR_TOOLTIP),
01010               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_COLLAR), SetDataTip(STR_EMPTY, STR_FACE_COLLAR_TOOLTIP),
01011               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_COLLAR_R), SetDataTip(AWV_INCREASE, STR_FACE_COLLAR_TOOLTIP),
01012             EndContainer(),
01013             NWidget(NWID_HORIZONTAL),
01014               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_SCMF_TIE_EARRING_TEXT), SetFill(1, 0),
01015               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING_L), SetDataTip(AWV_DECREASE, STR_FACE_TIE_EARRING_TOOLTIP),
01016               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING), SetDataTip(STR_EMPTY, STR_FACE_TIE_EARRING_TOOLTIP),
01017               NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_SCMF_TIE_EARRING_R), SetDataTip(AWV_INCREASE, STR_FACE_TIE_EARRING_TOOLTIP),
01018             EndContainer(),
01019             NWidget(NWID_SPACER), SetFill(0, 1),
01020           EndContainer(),
01021         EndContainer(),
01022       EndContainer(),
01023     EndContainer(),
01024     NWidget(NWID_SPACER), SetMinimalSize(0, 2),
01025   EndContainer(),
01026   NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
01027     NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_CANCEL), SetFill(1, 0), SetDataTip(STR_BUTTON_CANCEL, STR_FACE_CANCEL_TOOLTIP),
01028     NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SCMF_ACCEPT), SetFill(1, 0), SetDataTip(STR_BUTTON_OK, STR_FACE_OK_TOOLTIP),
01029   EndContainer(),
01030 };
01031 
01033 class SelectCompanyManagerFaceWindow : public Window
01034 {
01035   CompanyManagerFace face; 
01036   bool advanced; 
01037 
01038   GenderEthnicity ge; 
01039   bool is_female;     
01040   bool is_moust_male; 
01041 
01042   Dimension yesno_dim;  
01043   Dimension number_dim; 
01044 
01045   static const StringID PART_TEXTS_IS_FEMALE[]; 
01046   static const StringID PART_TEXTS[];           
01047 
01055   void DrawFaceStringLabel(byte widget_index, uint8 val, bool is_bool_widget) const
01056   {
01057     StringID str;
01058     const NWidgetCore *nwi_widget = this->GetWidget<NWidgetCore>(widget_index);
01059     if (!nwi_widget->IsDisabled()) {
01060       if (is_bool_widget) {
01061         /* if it a bool button write yes or no */
01062         str = (val != 0) ? STR_FACE_YES : STR_FACE_NO;
01063       } else {
01064         /* else write the value + 1 */
01065         SetDParam(0, val + 1);
01066         str = STR_JUST_INT;
01067       }
01068 
01069       /* Draw the value/bool in white (0xC). If the button clicked adds 1px to x and y text coordinates (IsWindowWidgetLowered()). */
01070       DrawString(nwi_widget->pos_x + nwi_widget->IsLowered(), nwi_widget->pos_x + nwi_widget->current_x - 1 - nwi_widget->IsLowered(),
01071           nwi_widget->pos_y + 1 + nwi_widget->IsLowered(), str, TC_WHITE, SA_HOR_CENTER);
01072     }
01073   }
01074 
01075   void UpdateData()
01076   {
01077     this->ge = (GenderEthnicity)GB(this->face, _cmf_info[CMFV_GEN_ETHN].offset, _cmf_info[CMFV_GEN_ETHN].length); // get the gender and ethnicity
01078     this->is_female = HasBit(this->ge, GENDER_FEMALE); // get the gender: 0 == male and 1 == female
01079     this->is_moust_male = !is_female && GetCompanyManagerFaceBits(this->face, CMFV_HAS_MOUSTACHE, this->ge) != 0; // is a male face with moustache
01080   }
01081 
01082 public:
01083   SelectCompanyManagerFaceWindow(const WindowDesc *desc, Window *parent) : Window()
01084   {
01085     this->advanced = false;
01086     this->CreateNestedTree(desc);
01087     this->SelectDisplayPlanes(this->advanced);
01088     this->FinishInitNested(desc, parent->window_number);
01089     this->parent = parent;
01090     this->owner = (Owner)this->window_number;
01091     this->face = Company::Get((CompanyID)this->window_number)->face;
01092 
01093     this->UpdateData();
01094   }
01095 
01100   void SelectDisplayPlanes(bool advanced)
01101   {
01102     this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_LOADSAVE)->SetDisplayedPlane(advanced ? 0 : SZSP_NONE);
01103     this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_PARTS)->SetDisplayedPlane(advanced ? 0 : SZSP_NONE);
01104     this->GetWidget<NWidgetStacked>(WID_SCMF_SEL_MALEFEMALE)->SetDisplayedPlane(advanced ? SZSP_NONE : 0);
01105     this->GetWidget<NWidgetCore>(WID_SCMF_RANDOM_NEW_FACE)->widget_data = advanced ? STR_MAPGEN_RANDOM : STR_FACE_NEW_FACE_BUTTON;
01106 
01107     NWidgetCore *wi = this->GetWidget<NWidgetCore>(WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON);
01108     if (advanced) {
01109       wi->SetDataTip(STR_FACE_SIMPLE, STR_FACE_SIMPLE_TOOLTIP);
01110     } else {
01111       wi->SetDataTip(STR_FACE_ADVANCED, STR_FACE_ADVANCED_TOOLTIP);
01112     }
01113   }
01114 
01115   virtual void OnInit()
01116   {
01117     /* Size of the boolean yes/no button. */
01118     Dimension yesno_dim = maxdim(GetStringBoundingBox(STR_FACE_YES), GetStringBoundingBox(STR_FACE_NO));
01119     yesno_dim.width  += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
01120     yesno_dim.height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
01121     /* Size of the number button + arrows. */
01122     Dimension number_dim = {0, 0};
01123     for (int val = 1; val <= 12; val++) {
01124       SetDParam(0, val);
01125       number_dim = maxdim(number_dim, GetStringBoundingBox(STR_JUST_INT));
01126     }
01127     uint arrows_width = GetSpriteSize(SPR_ARROW_LEFT).width + GetSpriteSize(SPR_ARROW_RIGHT).width + 2 * (WD_IMGBTN_LEFT + WD_IMGBTN_RIGHT);
01128     number_dim.width += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT + arrows_width;
01129     number_dim.height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
01130     /* Compute width of both buttons. */
01131     yesno_dim.width = max(yesno_dim.width, number_dim.width);
01132     number_dim.width = yesno_dim.width - arrows_width;
01133 
01134     this->yesno_dim = yesno_dim;
01135     this->number_dim = number_dim;
01136   }
01137 
01138   virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
01139   {
01140     switch (widget) {
01141       case WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT:
01142       case WID_SCMF_TIE_EARRING_TEXT: {
01143         int offset = (widget - WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT) * 2;
01144         *size = maxdim(GetStringBoundingBox(PART_TEXTS_IS_FEMALE[offset]), GetStringBoundingBox(PART_TEXTS_IS_FEMALE[offset + 1]));
01145         size->width  += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
01146         size->height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
01147         break;
01148       }
01149 
01150       case WID_SCMF_LIPS_MOUSTACHE_TEXT:
01151         *size = maxdim(GetStringBoundingBox(STR_FACE_LIPS), GetStringBoundingBox(STR_FACE_MOUSTACHE));
01152         size->width  += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
01153         size->height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
01154         break;
01155 
01156       case WID_SCMF_HAS_GLASSES_TEXT:
01157       case WID_SCMF_HAIR_TEXT:
01158       case WID_SCMF_EYEBROWS_TEXT:
01159       case WID_SCMF_EYECOLOUR_TEXT:
01160       case WID_SCMF_GLASSES_TEXT:
01161       case WID_SCMF_NOSE_TEXT:
01162       case WID_SCMF_CHIN_TEXT:
01163       case WID_SCMF_JACKET_TEXT:
01164       case WID_SCMF_COLLAR_TEXT:
01165         *size = GetStringBoundingBox(PART_TEXTS[widget - WID_SCMF_HAS_GLASSES_TEXT]);
01166         size->width  += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
01167         size->height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
01168         break;
01169 
01170       case WID_SCMF_HAS_MOUSTACHE_EARRING:
01171       case WID_SCMF_HAS_GLASSES:
01172         *size = this->yesno_dim;
01173         break;
01174 
01175       case WID_SCMF_EYECOLOUR:
01176       case WID_SCMF_CHIN:
01177       case WID_SCMF_EYEBROWS:
01178       case WID_SCMF_LIPS_MOUSTACHE:
01179       case WID_SCMF_NOSE:
01180       case WID_SCMF_HAIR:
01181       case WID_SCMF_JACKET:
01182       case WID_SCMF_COLLAR:
01183       case WID_SCMF_TIE_EARRING:
01184       case WID_SCMF_GLASSES:
01185         *size = this->number_dim;
01186         break;
01187     }
01188   }
01189 
01190   virtual void OnPaint()
01191   {
01192     /* lower the non-selected gender button */
01193     this->SetWidgetsLoweredState(!this->is_female, WID_SCMF_MALE, WID_SCMF_MALE2, WIDGET_LIST_END);
01194     this->SetWidgetsLoweredState( this->is_female, WID_SCMF_FEMALE, WID_SCMF_FEMALE2, WIDGET_LIST_END);
01195 
01196     /* advanced company manager face selection window */
01197 
01198     /* lower the non-selected ethnicity button */
01199     this->SetWidgetLoweredState(WID_SCMF_ETHNICITY_EUR, !HasBit(this->ge, ETHNICITY_BLACK));
01200     this->SetWidgetLoweredState(WID_SCMF_ETHNICITY_AFR,  HasBit(this->ge, ETHNICITY_BLACK));
01201 
01202 
01203     /* Disable dynamically the widgets which CompanyManagerFaceVariable has less than 2 options
01204      * (or in other words you haven't any choice).
01205      * If the widgets depend on a HAS-variable and this is false the widgets will be disabled, too. */
01206 
01207     /* Eye colour buttons */
01208     this->SetWidgetsDisabledState(_cmf_info[CMFV_EYE_COLOUR].valid_values[this->ge] < 2,
01209         WID_SCMF_EYECOLOUR, WID_SCMF_EYECOLOUR_L, WID_SCMF_EYECOLOUR_R, WIDGET_LIST_END);
01210 
01211     /* Chin buttons */
01212     this->SetWidgetsDisabledState(_cmf_info[CMFV_CHIN].valid_values[this->ge] < 2,
01213         WID_SCMF_CHIN, WID_SCMF_CHIN_L, WID_SCMF_CHIN_R, WIDGET_LIST_END);
01214 
01215     /* Eyebrows buttons */
01216     this->SetWidgetsDisabledState(_cmf_info[CMFV_EYEBROWS].valid_values[this->ge] < 2,
01217         WID_SCMF_EYEBROWS, WID_SCMF_EYEBROWS_L, WID_SCMF_EYEBROWS_R, WIDGET_LIST_END);
01218 
01219     /* Lips or (if it a male face with a moustache) moustache buttons */
01220     this->SetWidgetsDisabledState(_cmf_info[this->is_moust_male ? CMFV_MOUSTACHE : CMFV_LIPS].valid_values[this->ge] < 2,
01221         WID_SCMF_LIPS_MOUSTACHE, WID_SCMF_LIPS_MOUSTACHE_L, WID_SCMF_LIPS_MOUSTACHE_R, WIDGET_LIST_END);
01222 
01223     /* Nose buttons | male faces with moustache haven't any nose options */
01224     this->SetWidgetsDisabledState(_cmf_info[CMFV_NOSE].valid_values[this->ge] < 2 || this->is_moust_male,
01225         WID_SCMF_NOSE, WID_SCMF_NOSE_L, WID_SCMF_NOSE_R, WIDGET_LIST_END);
01226 
01227     /* Hair buttons */
01228     this->SetWidgetsDisabledState(_cmf_info[CMFV_HAIR].valid_values[this->ge] < 2,
01229         WID_SCMF_HAIR, WID_SCMF_HAIR_L, WID_SCMF_HAIR_R, WIDGET_LIST_END);
01230 
01231     /* Jacket buttons */
01232     this->SetWidgetsDisabledState(_cmf_info[CMFV_JACKET].valid_values[this->ge] < 2,
01233         WID_SCMF_JACKET, WID_SCMF_JACKET_L, WID_SCMF_JACKET_R, WIDGET_LIST_END);
01234 
01235     /* Collar buttons */
01236     this->SetWidgetsDisabledState(_cmf_info[CMFV_COLLAR].valid_values[this->ge] < 2,
01237         WID_SCMF_COLLAR, WID_SCMF_COLLAR_L, WID_SCMF_COLLAR_R, WIDGET_LIST_END);
01238 
01239     /* Tie/earring buttons | female faces without earring haven't any earring options */
01240     this->SetWidgetsDisabledState(_cmf_info[CMFV_TIE_EARRING].valid_values[this->ge] < 2 ||
01241           (this->is_female && GetCompanyManagerFaceBits(this->face, CMFV_HAS_TIE_EARRING, this->ge) == 0),
01242         WID_SCMF_TIE_EARRING, WID_SCMF_TIE_EARRING_L, WID_SCMF_TIE_EARRING_R, WIDGET_LIST_END);
01243 
01244     /* Glasses buttons | faces without glasses haven't any glasses options */
01245     this->SetWidgetsDisabledState(_cmf_info[CMFV_GLASSES].valid_values[this->ge] < 2 || GetCompanyManagerFaceBits(this->face, CMFV_HAS_GLASSES, this->ge) == 0,
01246         WID_SCMF_GLASSES, WID_SCMF_GLASSES_L, WID_SCMF_GLASSES_R, WIDGET_LIST_END);
01247 
01248     this->DrawWidgets();
01249   }
01250 
01251   virtual void DrawWidget(const Rect &r, int widget) const
01252   {
01253     switch (widget) {
01254       case WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT:
01255       case WID_SCMF_TIE_EARRING_TEXT: {
01256         StringID str = PART_TEXTS_IS_FEMALE[(widget - WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT) * 2 + this->is_female];
01257         DrawString(r.left + WD_FRAMERECT_LEFT, r.right - WD_FRAMERECT_RIGHT, r.top + WD_FRAMERECT_TOP, str, TC_GOLD, SA_RIGHT);
01258         break;
01259       }
01260 
01261       case WID_SCMF_LIPS_MOUSTACHE_TEXT:
01262         DrawString(r.left + WD_FRAMERECT_LEFT, r.right - WD_FRAMERECT_RIGHT, r.top + WD_FRAMERECT_TOP, (this->is_moust_male) ? STR_FACE_MOUSTACHE : STR_FACE_LIPS, TC_GOLD, SA_RIGHT);
01263         break;
01264 
01265       case WID_SCMF_HAS_GLASSES_TEXT:
01266       case WID_SCMF_HAIR_TEXT:
01267       case WID_SCMF_EYEBROWS_TEXT:
01268       case WID_SCMF_EYECOLOUR_TEXT:
01269       case WID_SCMF_GLASSES_TEXT:
01270       case WID_SCMF_NOSE_TEXT:
01271       case WID_SCMF_CHIN_TEXT:
01272       case WID_SCMF_JACKET_TEXT:
01273       case WID_SCMF_COLLAR_TEXT:
01274         DrawString(r.left + WD_FRAMERECT_LEFT, r.right - WD_FRAMERECT_RIGHT, r.top + WD_FRAMERECT_TOP, PART_TEXTS[widget - WID_SCMF_HAS_GLASSES_TEXT], TC_GOLD, SA_RIGHT);
01275         break;
01276 
01277 
01278       case WID_SCMF_HAS_MOUSTACHE_EARRING:
01279         if (this->is_female) { // Only for female faces
01280           this->DrawFaceStringLabel(WID_SCMF_HAS_MOUSTACHE_EARRING, GetCompanyManagerFaceBits(this->face, CMFV_HAS_TIE_EARRING, this->ge), true);
01281         } else { // Only for male faces
01282           this->DrawFaceStringLabel(WID_SCMF_HAS_MOUSTACHE_EARRING, GetCompanyManagerFaceBits(this->face, CMFV_HAS_MOUSTACHE,   this->ge), true);
01283         }
01284         break;
01285 
01286       case WID_SCMF_TIE_EARRING:
01287         if (this->is_female) { // Only for female faces
01288           this->DrawFaceStringLabel(WID_SCMF_TIE_EARRING, GetCompanyManagerFaceBits(this->face, CMFV_TIE_EARRING, this->ge), false);
01289         } else { // Only for male faces
01290           this->DrawFaceStringLabel(WID_SCMF_TIE_EARRING, GetCompanyManagerFaceBits(this->face, CMFV_TIE_EARRING, this->ge), false);
01291         }
01292         break;
01293 
01294       case WID_SCMF_LIPS_MOUSTACHE:
01295         if (this->is_moust_male) { // Only for male faces with moustache
01296           this->DrawFaceStringLabel(WID_SCMF_LIPS_MOUSTACHE, GetCompanyManagerFaceBits(this->face, CMFV_MOUSTACHE, this->ge), false);
01297         } else { // Only for female faces or male faces without moustache
01298           this->DrawFaceStringLabel(WID_SCMF_LIPS_MOUSTACHE, GetCompanyManagerFaceBits(this->face, CMFV_LIPS,      this->ge), false);
01299         }
01300         break;
01301 
01302       case WID_SCMF_HAS_GLASSES:
01303         this->DrawFaceStringLabel(WID_SCMF_HAS_GLASSES, GetCompanyManagerFaceBits(this->face, CMFV_HAS_GLASSES, this->ge), true );
01304         break;
01305 
01306       case WID_SCMF_HAIR:
01307         this->DrawFaceStringLabel(WID_SCMF_HAIR,        GetCompanyManagerFaceBits(this->face, CMFV_HAIR,        this->ge), false);
01308         break;
01309 
01310       case WID_SCMF_EYEBROWS:
01311         this->DrawFaceStringLabel(WID_SCMF_EYEBROWS,    GetCompanyManagerFaceBits(this->face, CMFV_EYEBROWS,    this->ge), false);
01312         break;
01313 
01314       case WID_SCMF_EYECOLOUR:
01315         this->DrawFaceStringLabel(WID_SCMF_EYECOLOUR,   GetCompanyManagerFaceBits(this->face, CMFV_EYE_COLOUR,  this->ge), false);
01316         break;
01317 
01318       case WID_SCMF_GLASSES:
01319         this->DrawFaceStringLabel(WID_SCMF_GLASSES,     GetCompanyManagerFaceBits(this->face, CMFV_GLASSES,     this->ge), false);
01320         break;
01321 
01322       case WID_SCMF_NOSE:
01323         this->DrawFaceStringLabel(WID_SCMF_NOSE,        GetCompanyManagerFaceBits(this->face, CMFV_NOSE,        this->ge), false);
01324         break;
01325 
01326       case WID_SCMF_CHIN:
01327         this->DrawFaceStringLabel(WID_SCMF_CHIN,        GetCompanyManagerFaceBits(this->face, CMFV_CHIN,        this->ge), false);
01328         break;
01329 
01330       case WID_SCMF_JACKET:
01331         this->DrawFaceStringLabel(WID_SCMF_JACKET,      GetCompanyManagerFaceBits(this->face, CMFV_JACKET,      this->ge), false);
01332         break;
01333 
01334       case WID_SCMF_COLLAR:
01335         this->DrawFaceStringLabel(WID_SCMF_COLLAR,      GetCompanyManagerFaceBits(this->face, CMFV_COLLAR,      this->ge), false);
01336         break;
01337 
01338       case WID_SCMF_FACE:
01339         DrawCompanyManagerFace(this->face, Company::Get((CompanyID)this->window_number)->colour, r.left, r.top);
01340         break;
01341     }
01342   }
01343 
01344   virtual void OnClick(Point pt, int widget, int click_count)
01345   {
01346     switch (widget) {
01347       /* Toggle size, advanced/simple face selection */
01348       case WID_SCMF_TOGGLE_LARGE_SMALL:
01349       case WID_SCMF_TOGGLE_LARGE_SMALL_BUTTON:
01350         this->advanced = !this->advanced;
01351         this->SelectDisplayPlanes(this->advanced);
01352         this->ReInit();
01353         break;
01354 
01355       /* OK button */
01356       case WID_SCMF_ACCEPT:
01357         DoCommandP(0, 0, this->face, CMD_SET_COMPANY_MANAGER_FACE);
01358         /* FALL THROUGH */
01359 
01360       /* Cancel button */
01361       case WID_SCMF_CANCEL:
01362         delete this;
01363         break;
01364 
01365       /* Load button */
01366       case WID_SCMF_LOAD:
01367         this->face = _company_manager_face;
01368         ScaleAllCompanyManagerFaceBits(this->face);
01369         ShowErrorMessage(STR_FACE_LOAD_DONE, INVALID_STRING_ID, WL_INFO);
01370         this->UpdateData();
01371         this->SetDirty();
01372         break;
01373 
01374       /* 'Company manager face number' button, view and/or set company manager face number */
01375       case WID_SCMF_FACECODE:
01376         SetDParam(0, this->face);
01377         ShowQueryString(STR_JUST_INT, STR_FACE_FACECODE_CAPTION, 10 + 1, this, CS_NUMERAL, QSF_NONE);
01378         break;
01379 
01380       /* Save button */
01381       case WID_SCMF_SAVE:
01382         _company_manager_face = this->face;
01383         ShowErrorMessage(STR_FACE_SAVE_DONE, INVALID_STRING_ID, WL_INFO);
01384         break;
01385 
01386       /* Toggle gender (male/female) button */
01387       case WID_SCMF_MALE:
01388       case WID_SCMF_FEMALE:
01389       case WID_SCMF_MALE2:
01390       case WID_SCMF_FEMALE2:
01391         SetCompanyManagerFaceBits(this->face, CMFV_GENDER, this->ge, (widget == WID_SCMF_FEMALE || widget == WID_SCMF_FEMALE2));
01392         ScaleAllCompanyManagerFaceBits(this->face);
01393         this->UpdateData();
01394         this->SetDirty();
01395         break;
01396 
01397       /* Randomize face button */
01398       case WID_SCMF_RANDOM_NEW_FACE:
01399         RandomCompanyManagerFaceBits(this->face, this->ge, this->advanced);
01400         this->UpdateData();
01401         this->SetDirty();
01402         break;
01403 
01404       /* Toggle ethnicity (european/african) button */
01405       case WID_SCMF_ETHNICITY_EUR:
01406       case WID_SCMF_ETHNICITY_AFR:
01407         SetCompanyManagerFaceBits(this->face, CMFV_ETHNICITY, this->ge, widget - WID_SCMF_ETHNICITY_EUR);
01408         ScaleAllCompanyManagerFaceBits(this->face);
01409         this->UpdateData();
01410         this->SetDirty();
01411         break;
01412 
01413       default:
01414         /* Here all buttons from WID_SCMF_HAS_MOUSTACHE_EARRING to WID_SCMF_GLASSES_R are handled.
01415          * First it checks which CompanyManagerFaceVariable is being changed, and then either
01416          * a: invert the value for boolean variables, or
01417          * b: it checks inside of IncreaseCompanyManagerFaceBits() if a left (_L) butten is pressed and then decrease else increase the variable */
01418         if (widget >= WID_SCMF_HAS_MOUSTACHE_EARRING && widget <= WID_SCMF_GLASSES_R) {
01419           CompanyManagerFaceVariable cmfv; // which CompanyManagerFaceVariable shall be edited
01420 
01421           if (widget < WID_SCMF_EYECOLOUR_L) { // Bool buttons
01422             switch (widget - WID_SCMF_HAS_MOUSTACHE_EARRING) {
01423               default: NOT_REACHED();
01424               case 0: cmfv = this->is_female ? CMFV_HAS_TIE_EARRING : CMFV_HAS_MOUSTACHE; break; // Has earring/moustache button
01425               case 1: cmfv = CMFV_HAS_GLASSES; break; // Has glasses button
01426             }
01427             SetCompanyManagerFaceBits(this->face, cmfv, this->ge, !GetCompanyManagerFaceBits(this->face, cmfv, this->ge));
01428             ScaleAllCompanyManagerFaceBits(this->face);
01429           } else { // Value buttons
01430             switch ((widget - WID_SCMF_EYECOLOUR_L) / 3) {
01431               default: NOT_REACHED();
01432               case 0: cmfv = CMFV_EYE_COLOUR; break;  // Eye colour buttons
01433               case 1: cmfv = CMFV_CHIN; break;        // Chin buttons
01434               case 2: cmfv = CMFV_EYEBROWS; break;    // Eyebrows buttons
01435               case 3: cmfv = this->is_moust_male ? CMFV_MOUSTACHE : CMFV_LIPS; break; // Moustache or lips buttons
01436               case 4: cmfv = CMFV_NOSE; break;        // Nose buttons
01437               case 5: cmfv = CMFV_HAIR; break;        // Hair buttons
01438               case 6: cmfv = CMFV_JACKET; break;      // Jacket buttons
01439               case 7: cmfv = CMFV_COLLAR; break;      // Collar buttons
01440               case 8: cmfv = CMFV_TIE_EARRING; break; // Tie/earring buttons
01441               case 9: cmfv = CMFV_GLASSES; break;     // Glasses buttons
01442             }
01443             /* 0 == left (_L), 1 == middle or 2 == right (_R) - button click */
01444             IncreaseCompanyManagerFaceBits(this->face, cmfv, this->ge, (((widget - WID_SCMF_EYECOLOUR_L) % 3) != 0) ? 1 : -1);
01445           }
01446           this->UpdateData();
01447           this->SetDirty();
01448         }
01449         break;
01450     }
01451   }
01452 
01453   virtual void OnQueryTextFinished(char *str)
01454   {
01455     if (str == NULL) return;
01456     /* Set a new company manager face number */
01457     if (!StrEmpty(str)) {
01458       this->face = strtoul(str, NULL, 10);
01459       ScaleAllCompanyManagerFaceBits(this->face);
01460       ShowErrorMessage(STR_FACE_FACECODE_SET, INVALID_STRING_ID, WL_INFO);
01461       this->UpdateData();
01462       this->SetDirty();
01463     } else {
01464       ShowErrorMessage(STR_FACE_FACECODE_ERR, INVALID_STRING_ID, WL_INFO);
01465     }
01466   }
01467 };
01468 
01470 const StringID SelectCompanyManagerFaceWindow::PART_TEXTS_IS_FEMALE[] = {
01471   STR_FACE_MOUSTACHE, STR_FACE_EARRING, // WID_SCMF_HAS_MOUSTACHE_EARRING_TEXT
01472   STR_FACE_TIE,       STR_FACE_EARRING, // WID_SCMF_TIE_EARRING_TEXT
01473 };
01474 
01476 const StringID SelectCompanyManagerFaceWindow::PART_TEXTS[] = {
01477   STR_FACE_GLASSES,   // WID_SCMF_HAS_GLASSES_TEXT
01478   STR_FACE_HAIR,      // WID_SCMF_HAIR_TEXT
01479   STR_FACE_EYEBROWS,  // WID_SCMF_EYEBROWS_TEXT
01480   STR_FACE_EYECOLOUR, // WID_SCMF_EYECOLOUR_TEXT
01481   STR_FACE_GLASSES,   // WID_SCMF_GLASSES_TEXT
01482   STR_FACE_NOSE,      // WID_SCMF_NOSE_TEXT
01483   STR_FACE_CHIN,      // WID_SCMF_CHIN_TEXT
01484   STR_FACE_JACKET,    // WID_SCMF_JACKET_TEXT
01485   STR_FACE_COLLAR,    // WID_SCMF_COLLAR_TEXT
01486 };
01487 
01489 static const WindowDesc _select_company_manager_face_desc(
01490   WDP_AUTO, 0, 0,
01491   WC_COMPANY_MANAGER_FACE, WC_NONE,
01492   WDF_UNCLICK_BUTTONS | WDF_CONSTRUCTION,
01493   _nested_select_company_manager_face_widgets, lengthof(_nested_select_company_manager_face_widgets)
01494 );
01495 
01504 static void DoSelectCompanyManagerFace(Window *parent)
01505 {
01506   if (!Company::IsValidID((CompanyID)parent->window_number)) return;
01507 
01508   if (BringWindowToFrontById(WC_COMPANY_MANAGER_FACE, parent->window_number)) return;
01509   new SelectCompanyManagerFaceWindow(&_select_company_manager_face_desc, parent);
01510 }
01511 
01512 static const NWidgetPart _nested_company_infrastructure_widgets[] = {
01513   NWidget(NWID_HORIZONTAL),
01514     NWidget(WWT_CLOSEBOX, COLOUR_GREY),
01515     NWidget(WWT_CAPTION, COLOUR_GREY, WID_CI_CAPTION), SetDataTip(STR_COMPANY_INFRASTRUCTURE_VIEW_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
01516     NWidget(WWT_SHADEBOX, COLOUR_GREY),
01517     NWidget(WWT_STICKYBOX, COLOUR_GREY),
01518   EndContainer(),
01519   NWidget(WWT_PANEL, COLOUR_GREY),
01520     NWidget(NWID_VERTICAL), SetPIP(WD_FRAMERECT_TOP, 4, WD_FRAMETEXT_BOTTOM),
01521       NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
01522         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_RAIL_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
01523         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_RAIL_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
01524       EndContainer(),
01525       NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
01526         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_ROAD_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
01527         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_ROAD_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
01528       EndContainer(),
01529       NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
01530         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_WATER_DESC), SetMinimalTextLines(2, 0), SetFill(1, 0),
01531         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_WATER_COUNT), SetMinimalTextLines(2, 0), SetFill(0, 1),
01532       EndContainer(),
01533       NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
01534         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_STATION_DESC), SetMinimalTextLines(3, 0), SetFill(1, 0),
01535         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_STATION_COUNT), SetMinimalTextLines(3, 0), SetFill(0, 1),
01536       EndContainer(),
01537       NWidget(NWID_HORIZONTAL), SetPIP(2, 4, 2),
01538         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_TOTAL_DESC), SetFill(1, 0),
01539         NWidget(WWT_EMPTY, COLOUR_GREY, WID_CI_TOTAL), SetFill(0, 1),
01540       EndContainer(),
01541     EndContainer(),
01542   EndContainer(),
01543 };
01544 
01548 struct CompanyInfrastructureWindow : Window
01549 {
01550   RailTypes railtypes; 
01551   RoadTypes roadtypes; 
01552 
01553   uint total_width; 
01554 
01555   CompanyInfrastructureWindow(const WindowDesc *desc, WindowNumber window_number) : Window()
01556   {
01557     this->UpdateRailRoadTypes();
01558 
01559     this->InitNested(desc, window_number);
01560     this->owner = (Owner)this->window_number;
01561   }
01562 
01563   void UpdateRailRoadTypes()
01564   {
01565     this->railtypes = RAILTYPES_NONE;
01566     this->roadtypes = ROADTYPES_ROAD; // Road is always available.
01567 
01568     /* Find the used railtypes. */
01569     Engine *e;
01570     FOR_ALL_ENGINES_OF_TYPE(e, VEH_TRAIN) {
01571       if (!HasBit(e->info.climates, _settings_game.game_creation.landscape)) continue;
01572 
01573       this->railtypes |= GetRailTypeInfo(e->u.rail.railtype)->introduces_railtypes;
01574     }
01575 
01576     /* Get the date introduced railtypes as well. */
01577     this->railtypes = AddDateIntroducedRailTypes(this->railtypes, MAX_DAY);
01578 
01579     /* Tram is only visible when there will be a tram. */
01580     FOR_ALL_ENGINES_OF_TYPE(e, VEH_ROAD) {
01581       if (!HasBit(e->info.climates, _settings_game.game_creation.landscape)) continue;
01582       if (!HasBit(e->info.misc_flags, EF_ROAD_TRAM)) continue;
01583 
01584       this->roadtypes |= ROADTYPES_TRAM;
01585       break;
01586     }
01587   }
01588 
01590   Money GetTotalMaintenanceCost() const
01591   {
01592     const Company *c = Company::Get((CompanyID)this->window_number);
01593     Money total;
01594 
01595     for (RailType rt = RAILTYPE_BEGIN; rt != RAILTYPE_END; rt++) {
01596       if (HasBit(this->railtypes, rt)) total += RailMaintenanceCost(rt, c->infrastructure.rail[rt]);
01597     }
01598     total += SignalMaintenanceCost(c->infrastructure.signal);
01599 
01600     if (HasBit(this->roadtypes, ROADTYPE_ROAD)) total += RoadMaintenanceCost(ROADTYPE_ROAD, c->infrastructure.road[ROADTYPE_ROAD]);
01601     if (HasBit(this->roadtypes, ROADTYPE_TRAM)) total += RoadMaintenanceCost(ROADTYPE_TRAM, c->infrastructure.road[ROADTYPE_TRAM]);
01602 
01603     total += CanalMaintenanceCost(c->infrastructure.water);
01604     total += StationMaintenanceCost(c->infrastructure.station);
01605     total += AirportMaintenanceCost(c->index);
01606 
01607     return total;
01608   }
01609 
01610   virtual void SetStringParameters(int widget) const
01611   {
01612     switch (widget) {
01613       case WID_CI_CAPTION:
01614         SetDParam(0, (CompanyID)this->window_number);
01615         break;
01616     }
01617   }
01618 
01619   virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
01620   {
01621     const Company *c = Company::Get((CompanyID)this->window_number);
01622 
01623     switch (widget) {
01624       case WID_CI_RAIL_DESC: {
01625         uint lines = 1;
01626 
01627         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_RAIL_SECT).width);
01628 
01629         for (RailType rt = RAILTYPE_BEGIN; rt < RAILTYPE_END; rt++) {
01630           if (HasBit(this->railtypes, rt)) {
01631             lines++;
01632             SetDParam(0, GetRailTypeInfo(rt)->strings.name);
01633             size->width = max(size->width, GetStringBoundingBox(STR_WHITE_STRING).width + WD_FRAMERECT_LEFT);
01634           }
01635         }
01636         if (this->railtypes != RAILTYPES_NONE) {
01637           lines++;
01638           size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_SIGNALS).width + WD_FRAMERECT_LEFT);
01639         }
01640 
01641         size->height = max(size->height, lines * FONT_HEIGHT_NORMAL);
01642         break;
01643       }
01644 
01645       case WID_CI_ROAD_DESC: {
01646         uint lines = 1;
01647 
01648         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD_SECT).width);
01649 
01650         if (HasBit(this->roadtypes, ROADTYPE_ROAD)) {
01651           lines++;
01652           size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD).width + WD_FRAMERECT_LEFT);
01653         }
01654         if (HasBit(this->roadtypes, ROADTYPE_TRAM)) {
01655           lines++;
01656           size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_TRAMWAY).width + WD_FRAMERECT_LEFT);
01657         }
01658 
01659         size->height = max(size->height, lines * FONT_HEIGHT_NORMAL);
01660         break;
01661       }
01662 
01663       case WID_CI_WATER_DESC:
01664         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_WATER_SECT).width);
01665         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_CANALS).width + WD_FRAMERECT_LEFT);
01666         break;
01667 
01668       case WID_CI_STATION_DESC:
01669         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_STATION_SECT).width);
01670         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_STATIONS).width + WD_FRAMERECT_LEFT);
01671         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_AIRPORTS).width + WD_FRAMERECT_LEFT);
01672         break;
01673 
01674       case WID_CI_RAIL_COUNT:
01675       case WID_CI_ROAD_COUNT:
01676       case WID_CI_WATER_COUNT:
01677       case WID_CI_STATION_COUNT:
01678       case WID_CI_TOTAL: {
01679         /* Find the maximum count that is displayed. */
01680         uint32 max_val = 1000;  // Some random number to reserve enough space.
01681         Money max_cost = 10000; // Some random number to reserve enough space.
01682         for (RailType rt = RAILTYPE_BEGIN; rt < RAILTYPE_END; rt++) {
01683           max_val = max(max_val, c->infrastructure.rail[rt]);
01684           max_cost = max(max_cost, RailMaintenanceCost(rt, c->infrastructure.rail[rt]));
01685         }
01686         max_val = max(max_val, c->infrastructure.signal);
01687         max_cost = max(max_cost, SignalMaintenanceCost(c->infrastructure.signal));
01688         for (RoadType rt = ROADTYPE_BEGIN; rt < ROADTYPE_END; rt++) {
01689           max_val = max(max_val, c->infrastructure.road[rt]);
01690           max_cost = max(max_cost, RoadMaintenanceCost(rt, c->infrastructure.road[rt]));
01691         }
01692         max_val = max(max_val, c->infrastructure.water);
01693         max_cost = max(max_cost, CanalMaintenanceCost(c->infrastructure.water));
01694         max_val = max(max_val, c->infrastructure.station);
01695         max_cost = max(max_cost, StationMaintenanceCost(c->infrastructure.station));
01696         max_val = max(max_val, c->infrastructure.airport);
01697         max_cost = max(max_cost, AirportMaintenanceCost(c->index));
01698 
01699         SetDParam(0, max_val);
01700         SetDParam(1, max_cost * 12); // Convert to per year
01701         size->width = max(size->width, GetStringBoundingBox(_settings_game.economy.infrastructure_maintenance ? STR_COMPANY_INFRASTRUCTURE_VIEW_COST : STR_WHITE_COMMA).width + 20); // Reserve some wiggle room.
01702 
01703         if (_settings_game.economy.infrastructure_maintenance) {
01704           SetDParam(0, this->GetTotalMaintenanceCost() * 12); // Convert to per year
01705           this->total_width = GetStringBoundingBox(STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL).width + 20;
01706           size->width = max(size->width, this->total_width);
01707         }
01708 
01709         /* Set height of the total line. */
01710         if (widget == WID_CI_TOTAL) {
01711           size->height = _settings_game.economy.infrastructure_maintenance ? max(size->height, EXP_LINESPACE + FONT_HEIGHT_NORMAL) : 0;
01712         }
01713         break;
01714       }
01715     }
01716   }
01717 
01718   virtual void DrawWidget(const Rect &r, int widget) const
01719   {
01720     const Company *c = Company::Get((CompanyID)this->window_number);
01721     int y = r.top;
01722 
01723     int offs_left = _current_text_dir == TD_LTR ? WD_FRAMERECT_LEFT : 0;
01724     int offs_right = _current_text_dir == TD_LTR ? 0 : WD_FRAMERECT_LEFT;
01725 
01726     switch (widget) {
01727       case WID_CI_RAIL_DESC:
01728         DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_RAIL_SECT);
01729 
01730         if (this->railtypes != RAILTYPES_NONE) {
01731           /* Draw name of each valid railtype. */
01732           for (RailType rt = RAILTYPE_BEGIN; rt != RAILTYPE_END; rt++) {
01733             if (HasBit(this->railtypes, rt)) {
01734               SetDParam(0, GetRailTypeInfo(rt)->strings.name);
01735               DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_WHITE_STRING);
01736             }
01737           }
01738           DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_SIGNALS);
01739         } else {
01740           /* No valid railtype. */
01741           DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_VIEW_INFRASTRUCTURE_NONE);
01742         }
01743 
01744         break;
01745 
01746       case WID_CI_RAIL_COUNT:
01747         /* Draw infrastructure count for each valid railtype. */
01748         for (RailType rt = RAILTYPE_BEGIN; rt != RAILTYPE_END; rt++) {
01749           if (HasBit(this->railtypes, rt)) {
01750             SetDParam(0, c->infrastructure.rail[rt]);
01751             SetDParam(1, RailMaintenanceCost(rt, c->infrastructure.rail[rt]) * 12); // Convert to per year
01752             DrawString(r.left, r.right, y += FONT_HEIGHT_NORMAL, _settings_game.economy.infrastructure_maintenance ? STR_COMPANY_INFRASTRUCTURE_VIEW_COST : STR_WHITE_COMMA);
01753           }
01754         }
01755         if (this->railtypes != RAILTYPES_NONE) {
01756           SetDParam(0, c->infrastructure.signal);
01757           SetDParam(1, SignalMaintenanceCost(c->infrastructure.signal) * 12); // Convert to per year
01758           DrawString(r.left, r.right, y += FONT_HEIGHT_NORMAL, _settings_game.economy.infrastructure_maintenance ? STR_COMPANY_INFRASTRUCTURE_VIEW_COST : STR_WHITE_COMMA);
01759         }
01760         break;
01761 
01762       case WID_CI_ROAD_DESC:
01763         DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD_SECT);
01764 
01765         if (this->roadtypes != ROADTYPES_NONE) {
01766           if (HasBit(this->roadtypes, ROADTYPE_ROAD)) DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_ROAD);
01767           if (HasBit(this->roadtypes, ROADTYPE_TRAM)) DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_TRAMWAY);
01768         } else {
01769           /* No valid roadtypes. */
01770           DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_VIEW_INFRASTRUCTURE_NONE);
01771         }
01772 
01773         break;
01774 
01775       case WID_CI_ROAD_COUNT:
01776         if (HasBit(this->roadtypes, ROADTYPE_ROAD)) {
01777           SetDParam(0, c->infrastructure.road[ROADTYPE_ROAD]);
01778           SetDParam(1, RoadMaintenanceCost(ROADTYPE_ROAD, c->infrastructure.road[ROADTYPE_ROAD]) * 12); // Convert to per year
01779           DrawString(r.left, r.right, y += FONT_HEIGHT_NORMAL, _settings_game.economy.infrastructure_maintenance ? STR_COMPANY_INFRASTRUCTURE_VIEW_COST : STR_WHITE_COMMA);
01780         }
01781         if (HasBit(this->roadtypes, ROADTYPE_TRAM)) {
01782           SetDParam(0, c->infrastructure.road[ROADTYPE_TRAM]);
01783           SetDParam(1, RoadMaintenanceCost(ROADTYPE_TRAM, c->infrastructure.road[ROADTYPE_TRAM]) * 12); // Convert to per year
01784           DrawString(r.left, r.right, y += FONT_HEIGHT_NORMAL, _settings_game.economy.infrastructure_maintenance ? STR_COMPANY_INFRASTRUCTURE_VIEW_COST : STR_WHITE_COMMA);
01785         }
01786         break;
01787 
01788       case WID_CI_WATER_DESC:
01789         DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_WATER_SECT);
01790         DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_CANALS);
01791         break;
01792 
01793       case WID_CI_WATER_COUNT:
01794         SetDParam(0, c->infrastructure.water);
01795         SetDParam(1, CanalMaintenanceCost(c->infrastructure.water) * 12); // Convert to per year
01796         DrawString(r.left, r.right, y += FONT_HEIGHT_NORMAL, _settings_game.economy.infrastructure_maintenance ? STR_COMPANY_INFRASTRUCTURE_VIEW_COST : STR_WHITE_COMMA);
01797         break;
01798 
01799       case WID_CI_TOTAL:
01800         if (_settings_game.economy.infrastructure_maintenance) {
01801           GfxFillRect(r.left, y, r.left + this->total_width, y, PC_WHITE);
01802           y += EXP_LINESPACE;
01803           SetDParam(0, this->GetTotalMaintenanceCost() * 12); // Convert to per year
01804           DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_TOTAL);
01805         }
01806         break;
01807 
01808       case WID_CI_STATION_DESC:
01809         DrawString(r.left, r.right, y, STR_COMPANY_INFRASTRUCTURE_VIEW_STATION_SECT);
01810         DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_STATIONS);
01811         DrawString(r.left + offs_left, r.right - offs_right, y += FONT_HEIGHT_NORMAL, STR_COMPANY_INFRASTRUCTURE_VIEW_AIRPORTS);
01812         break;
01813 
01814       case WID_CI_STATION_COUNT:
01815         SetDParam(0, c->infrastructure.station);
01816         SetDParam(1, StationMaintenanceCost(c->infrastructure.station) * 12); // Convert to per year
01817         DrawString(r.left, r.right, y += FONT_HEIGHT_NORMAL, _settings_game.economy.infrastructure_maintenance ? STR_COMPANY_INFRASTRUCTURE_VIEW_COST : STR_WHITE_COMMA);
01818         SetDParam(0, c->infrastructure.airport);
01819         SetDParam(1, AirportMaintenanceCost(c->index) * 12); // Convert to per year
01820         DrawString(r.left, r.right, y += FONT_HEIGHT_NORMAL, _settings_game.economy.infrastructure_maintenance ? STR_COMPANY_INFRASTRUCTURE_VIEW_COST : STR_WHITE_COMMA);
01821         break;
01822     }
01823   }
01824 
01830   virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
01831   {
01832     if (!gui_scope) return;
01833 
01834     this->UpdateRailRoadTypes();
01835     this->ReInit();
01836   }
01837 };
01838 
01839 static const WindowDesc _company_infrastructure_desc(
01840   WDP_AUTO, 0, 0,
01841   WC_COMPANY_INFRASTRUCTURE, WC_NONE,
01842   WDF_UNCLICK_BUTTONS,
01843   _nested_company_infrastructure_widgets, lengthof(_nested_company_infrastructure_widgets)
01844 );
01845 
01850 static void ShowCompanyInfrastructure(CompanyID company)
01851 {
01852   if (!Company::IsValidID(company)) return;
01853   AllocateWindowDescFront<CompanyInfrastructureWindow>(&_company_infrastructure_desc, company);
01854 }
01855 
01856 static const NWidgetPart _nested_company_widgets[] = {
01857   NWidget(NWID_HORIZONTAL),
01858     NWidget(WWT_CLOSEBOX, COLOUR_GREY),
01859     NWidget(WWT_CAPTION, COLOUR_GREY, WID_C_CAPTION), SetDataTip(STR_COMPANY_VIEW_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
01860     NWidget(WWT_SHADEBOX, COLOUR_GREY),
01861     NWidget(WWT_STICKYBOX, COLOUR_GREY),
01862   EndContainer(),
01863   NWidget(WWT_PANEL, COLOUR_GREY),
01864     NWidget(NWID_HORIZONTAL), SetPIP(4, 6, 4),
01865       NWidget(NWID_VERTICAL), SetPIP(4, 2, 4),
01866         NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_FACE), SetMinimalSize(92, 119), SetFill(1, 0),
01867         NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_FACE_TITLE), SetFill(1, 1), SetMinimalTextLines(2, 0),
01868       EndContainer(),
01869       NWidget(NWID_VERTICAL),
01870         NWidget(NWID_HORIZONTAL),
01871           NWidget(NWID_VERTICAL), SetPIP(4, 5, 5),
01872             NWidget(WWT_TEXT, COLOUR_GREY, WID_C_DESC_INAUGURATION), SetDataTip(STR_COMPANY_VIEW_INAUGURATED_TITLE, STR_NULL), SetFill(1, 0),
01873             NWidget(NWID_HORIZONTAL), SetPIP(0, 5, 0),
01874               NWidget(WWT_LABEL, COLOUR_GREY, WID_C_DESC_COLOUR_SCHEME), SetDataTip(STR_COMPANY_VIEW_COLOUR_SCHEME_TITLE, STR_NULL),
01875               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_COLOUR_SCHEME_EXAMPLE), SetMinimalSize(30, 0), SetFill(0, 1),
01876               NWidget(NWID_SPACER), SetFill(1, 0),
01877             EndContainer(),
01878             NWidget(NWID_HORIZONTAL), SetPIP(0, 4, 0),
01879               NWidget(NWID_VERTICAL),
01880                 NWidget(WWT_TEXT, COLOUR_GREY, WID_C_DESC_VEHICLE), SetDataTip(STR_COMPANY_VIEW_VEHICLES_TITLE, STR_NULL),
01881                 NWidget(NWID_SPACER), SetFill(0, 1),
01882               EndContainer(),
01883               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_VEHICLE_COUNTS), SetMinimalTextLines(4, 0),
01884               NWidget(NWID_SPACER), SetFill(1, 0),
01885             EndContainer(),
01886           EndContainer(),
01887           NWidget(NWID_VERTICAL), SetPIP(4, 2, 4),
01888             NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_VIEW_BUILD_HQ),
01889               NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_VIEW_HQ), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_VIEW_HQ_BUTTON, STR_COMPANY_VIEW_VIEW_HQ_TOOLTIP),
01890               NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_C_BUILD_HQ), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_BUILD_HQ_BUTTON, STR_COMPANY_VIEW_BUILD_HQ_TOOLTIP),
01891             EndContainer(),
01892             NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_RELOCATE),
01893               NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_C_RELOCATE_HQ), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_RELOCATE_HQ, STR_COMPANY_VIEW_RELOCATE_COMPANY_HEADQUARTERS),
01894               NWidget(NWID_SPACER), SetMinimalSize(90, 0),
01895             EndContainer(),
01896             NWidget(NWID_SPACER), SetFill(0, 1),
01897           EndContainer(),
01898         EndContainer(),
01899         NWidget(WWT_TEXT, COLOUR_GREY, WID_C_DESC_COMPANY_VALUE), SetDataTip(STR_COMPANY_VIEW_COMPANY_VALUE, STR_NULL), SetFill(1, 0),
01900           NWidget(NWID_VERTICAL), SetPIP(4, 2, 4),
01901             NWidget(NWID_HORIZONTAL), SetPIP(0, 4, 0),
01902               NWidget(NWID_VERTICAL),
01903                 NWidget(WWT_TEXT, COLOUR_GREY, WID_C_DESC_INFRASTRUCTURE), SetDataTip(STR_COMPANY_VIEW_INFRASTRUCTURE, STR_NULL),
01904                 NWidget(NWID_SPACER), SetFill(0, 1),
01905               EndContainer(),
01906               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_INFRASTRUCTURE_COUNTS), SetMinimalTextLines(5, 0), SetFill(1, 0),
01907               NWidget(NWID_VERTICAL),
01908                 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_VIEW_INFRASTRUCTURE), SetDataTip(STR_COMPANY_VIEW_INFRASTRUCTURE_BUTTON, STR_COMPANY_VIEW_INFRASTRUCTURE_TOOLTIP),
01909                 NWidget(NWID_SPACER), SetFill(0, 1), SetMinimalSize(90, 0),
01910               EndContainer(),
01911             EndContainer(),
01912           EndContainer(),
01913         NWidget(NWID_HORIZONTAL),
01914           NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_DESC_OWNERS),
01915             NWidget(NWID_VERTICAL), SetPIP(5, 5, 4),
01916               NWidget(WWT_EMPTY, INVALID_COLOUR, WID_C_DESC_OWNERS), SetMinimalTextLines(3, 0),
01917               NWidget(NWID_SPACER), SetFill(0, 1),
01918             EndContainer(),
01919           EndContainer(),
01920           NWidget(NWID_VERTICAL), SetPIP(4, 2, 4),
01921             NWidget(NWID_SPACER), SetMinimalSize(90, 0), SetFill(0, 1),
01922             /* Multi player buttons. */
01923             NWidget(NWID_HORIZONTAL),
01924               NWidget(WWT_EMPTY, COLOUR_GREY, WID_C_HAS_PASSWORD),
01925               NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_MULTIPLAYER),
01926                 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COMPANY_PASSWORD), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_PASSWORD, STR_COMPANY_VIEW_PASSWORD_TOOLTIP),
01927                 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COMPANY_JOIN), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_JOIN, STR_COMPANY_VIEW_JOIN_TOOLTIP),
01928               EndContainer(),
01929             EndContainer(),
01930           EndContainer(),
01931         EndContainer(),
01932       EndContainer(),
01933     EndContainer(),
01934   EndContainer(),
01935   /* Button bars at the bottom. */
01936   NWidget(NWID_SELECTION, INVALID_COLOUR, WID_C_SELECT_BUTTONS),
01937     NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
01938       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_NEW_FACE), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_NEW_FACE_BUTTON, STR_COMPANY_VIEW_NEW_FACE_TOOLTIP),
01939       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COLOUR_SCHEME), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_COLOUR_SCHEME_BUTTON, STR_COMPANY_VIEW_COLOUR_SCHEME_TOOLTIP),
01940       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_PRESIDENT_NAME), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_PRESIDENT_NAME_BUTTON, STR_COMPANY_VIEW_PRESIDENT_NAME_TOOLTIP),
01941       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_COMPANY_NAME), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_COMPANY_NAME_BUTTON, STR_COMPANY_VIEW_COMPANY_NAME_TOOLTIP),
01942     EndContainer(),
01943     NWidget(NWID_HORIZONTAL, NC_EQUALSIZE),
01944       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_BUY_SHARE), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_BUY_SHARE_BUTTON, STR_COMPANY_VIEW_BUY_SHARE_TOOLTIP),
01945       NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_C_SELL_SHARE), SetFill(1, 0), SetDataTip(STR_COMPANY_VIEW_SELL_SHARE_BUTTON, STR_COMPANY_VIEW_SELL_SHARE_TOOLTIP),
01946     EndContainer(),
01947   EndContainer(),
01948 };
01949 
01950 int GetAmountOwnedBy(const Company *c, Owner owner)
01951 {
01952   return (c->share_owners[0] == owner) +
01953          (c->share_owners[1] == owner) +
01954          (c->share_owners[2] == owner) +
01955          (c->share_owners[3] == owner);
01956 }
01957 
01959 static const StringID _company_view_vehicle_count_strings[] = {
01960   STR_COMPANY_VIEW_TRAINS, STR_COMPANY_VIEW_ROAD_VEHICLES, STR_COMPANY_VIEW_SHIPS, STR_COMPANY_VIEW_AIRCRAFT
01961 };
01962 
01966 struct CompanyWindow : Window
01967 {
01968   CompanyWidgets query_widget;
01969 
01971   enum CompanyWindowPlanes {
01972     /* Display planes of the #WID_C_SELECT_MULTIPLAYER selection widget. */
01973     CWP_MP_C_PWD = 0, 
01974     CWP_MP_C_JOIN,    
01975 
01976     /* Display planes of the #WID_C_SELECT_VIEW_BUILD_HQ selection widget. */
01977     CWP_VB_VIEW = 0,  
01978     CWP_VB_BUILD,     
01979 
01980     /* Display planes of the #WID_C_SELECT_RELOCATE selection widget. */
01981     CWP_RELOCATE_SHOW = 0, 
01982     CWP_RELOCATE_HIDE,     
01983 
01984     /* Display planes of the #WID_C_SELECT_BUTTONS selection widget. */
01985     CWP_BUTTONS_LOCAL = 0, 
01986     CWP_BUTTONS_OTHER,     
01987   };
01988 
01989   CompanyWindow(const WindowDesc *desc, WindowNumber window_number) : Window()
01990   {
01991     this->InitNested(desc, window_number);
01992     this->owner = (Owner)this->window_number;
01993   }
01994 
01995   virtual void OnPaint()
01996   {
01997     const Company *c = Company::Get((CompanyID)this->window_number);
01998     bool local = this->window_number == _local_company;
01999 
02000     if (!this->IsShaded()) {
02001       bool reinit = false;
02002 
02003       /* Button bar selection. */
02004       int plane = local ? CWP_BUTTONS_LOCAL : CWP_BUTTONS_OTHER;
02005       NWidgetStacked *wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_BUTTONS);
02006       if (plane != wi->shown_plane) {
02007         wi->SetDisplayedPlane(plane);
02008         this->SetDirty();
02009         return;
02010       }
02011 
02012       /* Build HQ button handling. */
02013       plane = (local && c->location_of_HQ == INVALID_TILE) ? CWP_VB_BUILD : CWP_VB_VIEW;
02014       wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_VIEW_BUILD_HQ);
02015       if (plane != wi->shown_plane) {
02016         wi->SetDisplayedPlane(plane);
02017         this->SetDirty();
02018         return;
02019       }
02020 
02021       this->SetWidgetDisabledState(WID_C_VIEW_HQ, c->location_of_HQ == INVALID_TILE);
02022 
02023       /* Enable/disable 'Relocate HQ' button. */
02024       plane = (!local || c->location_of_HQ == INVALID_TILE) ? CWP_RELOCATE_HIDE : CWP_RELOCATE_SHOW;
02025       wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_RELOCATE);
02026       if (plane != wi->shown_plane) {
02027         wi->SetDisplayedPlane(plane);
02028         this->SetDirty();
02029         return;
02030       }
02031 
02032       /* Owners of company */
02033       plane = SZSP_HORIZONTAL;
02034       for (uint i = 0; i < lengthof(c->share_owners); i++) {
02035         if (c->share_owners[i] != INVALID_COMPANY) {
02036           plane = 0;
02037           break;
02038         }
02039       }
02040       wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_DESC_OWNERS);
02041       if (plane != wi->shown_plane) {
02042         wi->SetDisplayedPlane(plane);
02043         reinit = true;
02044       }
02045 
02046       /* Multiplayer buttons. */
02047       plane = ((!_networking) ? (int)SZSP_NONE : (int)(local ? CWP_MP_C_PWD : CWP_MP_C_JOIN));
02048       wi = this->GetWidget<NWidgetStacked>(WID_C_SELECT_MULTIPLAYER);
02049       if (plane != wi->shown_plane) {
02050         wi->SetDisplayedPlane(plane);
02051         reinit = true;
02052       }
02053       this->SetWidgetDisabledState(WID_C_COMPANY_JOIN,   c->is_ai);
02054 
02055       if (reinit) {
02056         this->ReInit();
02057         return;
02058       }
02059     }
02060 
02061     if (!local) {
02062       if (_settings_game.economy.allow_shares) { // Shares are allowed
02063         /* If all shares are owned by someone (none by nobody), disable buy button */
02064         this->SetWidgetDisabledState(WID_C_BUY_SHARE, GetAmountOwnedBy(c, INVALID_OWNER) == 0 ||
02065             /* Only 25% left to buy. If the company is human, disable buying it up.. TODO issues! */
02066             (GetAmountOwnedBy(c, INVALID_OWNER) == 1 && !c->is_ai) ||
02067             /* Spectators cannot do anything of course */
02068             _local_company == COMPANY_SPECTATOR);
02069 
02070         /* If the company doesn't own any shares, disable sell button */
02071         this->SetWidgetDisabledState(WID_C_SELL_SHARE, (GetAmountOwnedBy(c, _local_company) == 0) ||
02072             /* Spectators cannot do anything of course */
02073             _local_company == COMPANY_SPECTATOR);
02074       } else { // Shares are not allowed, disable buy/sell buttons
02075         this->DisableWidget(WID_C_BUY_SHARE);
02076         this->DisableWidget(WID_C_SELL_SHARE);
02077       }
02078     }
02079 
02080     this->DrawWidgets();
02081   }
02082 
02083   virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
02084   {
02085     switch (widget) {
02086       case WID_C_DESC_COLOUR_SCHEME_EXAMPLE: {
02087         Point offset;
02088         Dimension d = GetSpriteSize(SPR_VEH_BUS_SW_VIEW, &offset);
02089         d.width -= offset.x;
02090         d.height -= offset.y;
02091         *size = maxdim(*size, d);
02092         break;
02093       }
02094 
02095       case WID_C_DESC_COMPANY_VALUE:
02096         SetDParam(0, INT64_MAX); // Arguably the maximum company value
02097         size->width = GetStringBoundingBox(STR_COMPANY_VIEW_COMPANY_VALUE).width;
02098         break;
02099 
02100       case WID_C_DESC_VEHICLE_COUNTS:
02101         SetDParam(0, 5000); // Maximum number of vehicles
02102         for (uint i = 0; i < lengthof(_company_view_vehicle_count_strings); i++) {
02103           size->width = max(size->width, GetStringBoundingBox(_company_view_vehicle_count_strings[i]).width);
02104         }
02105         break;
02106 
02107       case WID_C_DESC_INFRASTRUCTURE_COUNTS:
02108         SetDParam(0, UINT_MAX);
02109         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_RAIL).width);
02110         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_ROAD).width);
02111         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_WATER).width);
02112         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_STATION).width);
02113         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_AIRPORT).width);
02114         size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_INFRASTRUCTURE_NONE).width);
02115         break;
02116 
02117       case WID_C_DESC_OWNERS: {
02118         const Company *c2;
02119 
02120         FOR_ALL_COMPANIES(c2) {
02121           SetDParam(0, 25);
02122           SetDParam(1, c2->index);
02123 
02124           size->width = max(size->width, GetStringBoundingBox(STR_COMPANY_VIEW_SHARES_OWNED_BY).width);
02125         }
02126         break;
02127       }
02128 
02129 #ifdef ENABLE_NETWORK
02130       case WID_C_HAS_PASSWORD:
02131         *size = maxdim(*size, GetSpriteSize(SPR_LOCK));
02132         break;
02133 #endif /* ENABLE_NETWORK */
02134     }
02135   }
02136 
02137   virtual void DrawWidget(const Rect &r, int widget) const
02138   {
02139     const Company *c = Company::Get((CompanyID)this->window_number);
02140     switch (widget) {
02141       case WID_C_FACE:
02142         DrawCompanyManagerFace(c->face, c->colour, r.left, r.top);
02143         break;
02144 
02145       case WID_C_FACE_TITLE:
02146         SetDParam(0, c->index);
02147         DrawStringMultiLine(r.left, r.right, r.top, r.bottom, STR_COMPANY_VIEW_PRESIDENT_MANAGER_TITLE, TC_FROMSTRING, SA_HOR_CENTER);
02148         break;
02149 
02150       case WID_C_DESC_COLOUR_SCHEME_EXAMPLE: {
02151         Point offset;
02152         Dimension d = GetSpriteSize(SPR_VEH_BUS_SW_VIEW, &offset);
02153         d.height -= offset.y;
02154         DrawSprite(SPR_VEH_BUS_SW_VIEW, COMPANY_SPRITE_COLOUR(c->index), r.left - offset.x, (r.top + r.bottom - d.height) / 2 - offset.y);
02155         break;
02156       }
02157 
02158       case WID_C_DESC_VEHICLE_COUNTS: {
02159         uint amounts[4];
02160         amounts[0] = c->group_all[VEH_TRAIN].num_vehicle;
02161         amounts[1] = c->group_all[VEH_ROAD].num_vehicle;
02162         amounts[2] = c->group_all[VEH_SHIP].num_vehicle;
02163         amounts[3] = c->group_all[VEH_AIRCRAFT].num_vehicle;
02164 
02165         int y = r.top;
02166         if (amounts[0] + amounts[1] + amounts[2] + amounts[3] == 0) {
02167           DrawString(r.left, r.right, y, STR_COMPANY_VIEW_VEHICLES_NONE);
02168         } else {
02169           assert_compile(lengthof(amounts) == lengthof(_company_view_vehicle_count_strings));
02170 
02171           for (uint i = 0; i < lengthof(amounts); i++) {
02172             if (amounts[i] != 0) {
02173               SetDParam(0, amounts[i]);
02174               DrawString(r.left, r.right, y, _company_view_vehicle_count_strings[i]);
02175               y += FONT_HEIGHT_NORMAL;
02176             }
02177           }
02178         }
02179         break;
02180       }
02181 
02182       case WID_C_DESC_INFRASTRUCTURE_COUNTS: {
02183         uint y = r.top;
02184 
02185         /* Collect rail and road counts. */
02186         uint rail_pices = c->infrastructure.signal;
02187         uint road_pieces = 0;
02188         for (uint i = 0; i < lengthof(c->infrastructure.rail); i++) rail_pices += c->infrastructure.rail[i];
02189         for (uint i = 0; i < lengthof(c->infrastructure.road); i++) road_pieces += c->infrastructure.road[i];
02190 
02191         if (rail_pices == 0 && road_pieces == 0 && c->infrastructure.water == 0 && c->infrastructure.station == 0 && c->infrastructure.airport == 0) {
02192           DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_NONE);
02193         } else {
02194           if (rail_pices != 0) {
02195             SetDParam(0, rail_pices);
02196             DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_RAIL);
02197             y += FONT_HEIGHT_NORMAL;
02198           }
02199           if (road_pieces != 0) {
02200             SetDParam(0, road_pieces);
02201             DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_ROAD);
02202             y += FONT_HEIGHT_NORMAL;
02203           }
02204           if (c->infrastructure.water != 0) {
02205             SetDParam(0, c->infrastructure.water);
02206             DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_WATER);
02207             y += FONT_HEIGHT_NORMAL;
02208           }
02209           if (c->infrastructure.station != 0) {
02210             SetDParam(0, c->infrastructure.station);
02211             DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_STATION);
02212             y += FONT_HEIGHT_NORMAL;
02213           }
02214           if (c->infrastructure.airport != 0) {
02215             SetDParam(0, c->infrastructure.airport);
02216             DrawString(r.left, r.right, y, STR_COMPANY_VIEW_INFRASTRUCTURE_AIRPORT);
02217           }
02218         }
02219 
02220         break;
02221       }
02222 
02223       case WID_C_DESC_OWNERS: {
02224         const Company *c2;
02225         uint y = r.top;
02226 
02227         FOR_ALL_COMPANIES(c2) {
02228           uint amt = GetAmountOwnedBy(c, c2->index);
02229           if (amt != 0) {
02230             SetDParam(0, amt * 25);
02231             SetDParam(1, c2->index);
02232 
02233             DrawString(r.left, r.right, y, STR_COMPANY_VIEW_SHARES_OWNED_BY);
02234             y += FONT_HEIGHT_NORMAL;
02235           }
02236         }
02237         break;
02238       }
02239 
02240 #ifdef ENABLE_NETWORK
02241       case WID_C_HAS_PASSWORD:
02242         if (_networking && NetworkCompanyIsPassworded(c->index)) {
02243           DrawSprite(SPR_LOCK, PAL_NONE, r.left, r.top);
02244         }
02245         break;
02246 #endif /* ENABLE_NETWORK */
02247     }
02248   }
02249 
02250   virtual void SetStringParameters(int widget) const
02251   {
02252     switch (widget) {
02253       case WID_C_CAPTION:
02254         SetDParam(0, (CompanyID)this->window_number);
02255         SetDParam(1, (CompanyID)this->window_number);
02256         break;
02257 
02258       case WID_C_DESC_INAUGURATION:
02259         SetDParam(0, Company::Get((CompanyID)this->window_number)->inaugurated_year);
02260         break;
02261 
02262       case WID_C_DESC_COMPANY_VALUE:
02263         SetDParam(0, CalculateCompanyValue(Company::Get((CompanyID)this->window_number)));
02264         break;
02265     }
02266   }
02267 
02268   virtual void OnClick(Point pt, int widget, int click_count)
02269   {
02270     switch (widget) {
02271       case WID_C_NEW_FACE: DoSelectCompanyManagerFace(this); break;
02272 
02273       case WID_C_COLOUR_SCHEME:
02274         if (BringWindowToFrontById(WC_COMPANY_COLOUR, this->window_number)) break;
02275         new SelectCompanyLiveryWindow(&_select_company_livery_desc, (CompanyID)this->window_number);
02276         break;
02277 
02278       case WID_C_PRESIDENT_NAME:
02279         this->query_widget = WID_C_PRESIDENT_NAME;
02280         SetDParam(0, this->window_number);
02281         ShowQueryString(STR_PRESIDENT_NAME, STR_COMPANY_VIEW_PRESIDENT_S_NAME_QUERY_CAPTION, MAX_LENGTH_PRESIDENT_NAME_CHARS, this, CS_ALPHANUMERAL, QSF_ENABLE_DEFAULT | QSF_LEN_IN_CHARS);
02282         break;
02283 
02284       case WID_C_COMPANY_NAME:
02285         this->query_widget = WID_C_COMPANY_NAME;
02286         SetDParam(0, this->window_number);
02287         ShowQueryString(STR_COMPANY_NAME, STR_COMPANY_VIEW_COMPANY_NAME_QUERY_CAPTION, MAX_LENGTH_COMPANY_NAME_CHARS, this, CS_ALPHANUMERAL, QSF_ENABLE_DEFAULT | QSF_LEN_IN_CHARS);
02288         break;
02289 
02290       case WID_C_VIEW_HQ: {
02291         TileIndex tile = Company::Get((CompanyID)this->window_number)->location_of_HQ;
02292         if (_ctrl_pressed) {
02293           ShowExtraViewPortWindow(tile);
02294         } else {
02295           ScrollMainWindowToTile(tile);
02296         }
02297         break;
02298       }
02299 
02300       case WID_C_BUILD_HQ:
02301         if ((byte)this->window_number != _local_company) return;
02302         if (this->IsWidgetLowered(WID_C_BUILD_HQ)) {
02303           ResetObjectToPlace();
02304           this->RaiseButtons();
02305           break;
02306         }
02307         SetObjectToPlaceWnd(SPR_CURSOR_HQ, PAL_NONE, HT_RECT, this);
02308         SetTileSelectSize(2, 2);
02309         this->LowerWidget(WID_C_BUILD_HQ);
02310         this->SetWidgetDirty(WID_C_BUILD_HQ);
02311         break;
02312 
02313       case WID_C_RELOCATE_HQ:
02314         if (this->IsWidgetLowered(WID_C_RELOCATE_HQ)) {
02315           ResetObjectToPlace();
02316           this->RaiseButtons();
02317           break;
02318         }
02319         SetObjectToPlaceWnd(SPR_CURSOR_HQ, PAL_NONE, HT_RECT, this);
02320         SetTileSelectSize(2, 2);
02321         this->LowerWidget(WID_C_RELOCATE_HQ);
02322         this->SetWidgetDirty(WID_C_RELOCATE_HQ);
02323         break;
02324 
02325       case WID_C_VIEW_INFRASTRUCTURE:
02326         ShowCompanyInfrastructure((CompanyID)this->window_number);
02327         break;
02328 
02329       case WID_C_BUY_SHARE:
02330         DoCommandP(0, this->window_number, 0, CMD_BUY_SHARE_IN_COMPANY | CMD_MSG(STR_ERROR_CAN_T_BUY_25_SHARE_IN_THIS));
02331         break;
02332 
02333       case WID_C_SELL_SHARE:
02334         DoCommandP(0, this->window_number, 0, CMD_SELL_SHARE_IN_COMPANY | CMD_MSG(STR_ERROR_CAN_T_SELL_25_SHARE_IN));
02335         break;
02336 
02337 #ifdef ENABLE_NETWORK
02338       case WID_C_COMPANY_PASSWORD:
02339         if (this->window_number == _local_company) ShowNetworkCompanyPasswordWindow(this);
02340         break;
02341 
02342       case WID_C_COMPANY_JOIN: {
02343         this->query_widget = WID_C_COMPANY_JOIN;
02344         CompanyID company = (CompanyID)this->window_number;
02345         if (_network_server) {
02346           NetworkServerDoMove(CLIENT_ID_SERVER, company);
02347           MarkWholeScreenDirty();
02348         } else if (NetworkCompanyIsPassworded(company)) {
02349           /* ask for the password */
02350           ShowQueryString(STR_EMPTY, STR_NETWORK_NEED_COMPANY_PASSWORD_CAPTION, NETWORK_PASSWORD_LENGTH, this, CS_ALPHANUMERAL, QSF_NONE);
02351         } else {
02352           /* just send the join command */
02353           NetworkClientRequestMove(company);
02354         }
02355         break;
02356       }
02357 #endif /* ENABLE_NETWORK */
02358     }
02359   }
02360 
02361   virtual void OnHundredthTick()
02362   {
02363     /* redraw the window every now and then */
02364     this->SetDirty();
02365   }
02366 
02367   virtual void OnPlaceObject(Point pt, TileIndex tile)
02368   {
02369     if (DoCommandP(tile, OBJECT_HQ, 0, CMD_BUILD_OBJECT | CMD_MSG(STR_ERROR_CAN_T_BUILD_COMPANY_HEADQUARTERS))) {
02370       ResetObjectToPlace();
02371       this->RaiseButtons();
02372     }
02373   }
02374 
02375   virtual void OnPlaceObjectAbort()
02376   {
02377     this->RaiseButtons();
02378   }
02379 
02380   virtual void OnQueryTextFinished(char *str)
02381   {
02382     if (str == NULL) return;
02383 
02384     switch (this->query_widget) {
02385       default: NOT_REACHED();
02386 
02387       case WID_C_PRESIDENT_NAME:
02388         DoCommandP(0, 0, 0, CMD_RENAME_PRESIDENT | CMD_MSG(STR_ERROR_CAN_T_CHANGE_PRESIDENT), NULL, str);
02389         break;
02390 
02391       case WID_C_COMPANY_NAME:
02392         DoCommandP(0, 0, 0, CMD_RENAME_COMPANY | CMD_MSG(STR_ERROR_CAN_T_CHANGE_COMPANY_NAME), NULL, str);
02393         break;
02394 
02395 #ifdef ENABLE_NETWORK
02396       case WID_C_COMPANY_JOIN:
02397         NetworkClientRequestMove((CompanyID)this->window_number, str);
02398         break;
02399 #endif /* ENABLE_NETWORK */
02400     }
02401   }
02402 };
02403 
02404 static const WindowDesc _company_desc(
02405   WDP_AUTO, 0, 0,
02406   WC_COMPANY, WC_NONE,
02407   WDF_UNCLICK_BUTTONS,
02408   _nested_company_widgets, lengthof(_nested_company_widgets)
02409 );
02410 
02415 void ShowCompany(CompanyID company)
02416 {
02417   if (!Company::IsValidID(company)) return;
02418 
02419   AllocateWindowDescFront<CompanyWindow>(&_company_desc, company);
02420 }
02421 
02426 void DirtyCompanyInfrastructureWindows(CompanyID company)
02427 {
02428   SetWindowDirty(WC_COMPANY, company);
02429   SetWindowDirty(WC_COMPANY_INFRASTRUCTURE, company);
02430 }
02431 
02432 struct BuyCompanyWindow : Window {
02433   BuyCompanyWindow(const WindowDesc *desc, WindowNumber window_number) : Window()
02434   {
02435     this->InitNested(desc, window_number);
02436   }
02437 
02438   virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
02439   {
02440     switch (widget) {
02441       case WID_BC_FACE:
02442         *size = GetSpriteSize(SPR_GRADIENT);
02443         break;
02444 
02445       case WID_BC_QUESTION:
02446         const Company *c = Company::Get((CompanyID)this->window_number);
02447         SetDParam(0, c->index);
02448         SetDParam(1, c->bankrupt_value);
02449         size->height = GetStringHeight(STR_BUY_COMPANY_MESSAGE, size->width);
02450         break;
02451     }
02452   }
02453 
02454   virtual void SetStringParameters(int widget) const
02455   {
02456     switch (widget) {
02457       case WID_BC_CAPTION:
02458         SetDParam(0, STR_COMPANY_NAME);
02459         SetDParam(1, Company::Get((CompanyID)this->window_number)->index);
02460         break;
02461     }
02462   }
02463 
02464   virtual void DrawWidget(const Rect &r, int widget) const
02465   {
02466     switch (widget) {
02467       case WID_BC_FACE: {
02468         const Company *c = Company::Get((CompanyID)this->window_number);
02469         DrawCompanyManagerFace(c->face, c->colour, r.left, r.top);
02470         break;
02471       }
02472 
02473       case WID_BC_QUESTION: {
02474         const Company *c = Company::Get((CompanyID)this->window_number);
02475         SetDParam(0, c->index);
02476         SetDParam(1, c->bankrupt_value);
02477         DrawStringMultiLine(r.left, r.right, r.top, r.bottom, STR_BUY_COMPANY_MESSAGE, TC_FROMSTRING, SA_CENTER);
02478         break;
02479       }
02480     }
02481   }
02482 
02483   virtual void OnClick(Point pt, int widget, int click_count)
02484   {
02485     switch (widget) {
02486       case WID_BC_NO:
02487         delete this;
02488         break;
02489 
02490       case WID_BC_YES:
02491         DoCommandP(0, this->window_number, 0, CMD_BUY_COMPANY | CMD_MSG(STR_ERROR_CAN_T_BUY_COMPANY));
02492         break;
02493     }
02494   }
02495 };
02496 
02497 static const NWidgetPart _nested_buy_company_widgets[] = {
02498   NWidget(NWID_HORIZONTAL),
02499     NWidget(WWT_CLOSEBOX, COLOUR_LIGHT_BLUE),
02500     NWidget(WWT_CAPTION, COLOUR_LIGHT_BLUE, WID_BC_CAPTION), SetDataTip(STR_ERROR_MESSAGE_CAPTION_OTHER_COMPANY, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
02501   EndContainer(),
02502   NWidget(WWT_PANEL, COLOUR_LIGHT_BLUE),
02503     NWidget(NWID_VERTICAL), SetPIP(8, 8, 8),
02504       NWidget(NWID_HORIZONTAL), SetPIP(8, 10, 8),
02505         NWidget(WWT_EMPTY, INVALID_COLOUR, WID_BC_FACE), SetFill(0, 1),
02506         NWidget(WWT_EMPTY, INVALID_COLOUR, WID_BC_QUESTION), SetMinimalSize(240, 0), SetFill(1, 1),
02507       EndContainer(),
02508       NWidget(NWID_HORIZONTAL, NC_EQUALSIZE), SetPIP(100, 10, 100),
02509         NWidget(WWT_TEXTBTN, COLOUR_LIGHT_BLUE, WID_BC_NO), SetMinimalSize(60, 12), SetDataTip(STR_QUIT_NO, STR_NULL), SetFill(1, 0),
02510         NWidget(WWT_TEXTBTN, COLOUR_LIGHT_BLUE, WID_BC_YES), SetMinimalSize(60, 12), SetDataTip(STR_QUIT_YES, STR_NULL), SetFill(1, 0),
02511       EndContainer(),
02512     EndContainer(),
02513   EndContainer(),
02514 };
02515 
02516 static const WindowDesc _buy_company_desc(
02517   WDP_AUTO, 0, 0,
02518   WC_BUY_COMPANY, WC_NONE,
02519   WDF_CONSTRUCTION,
02520   _nested_buy_company_widgets, lengthof(_nested_buy_company_widgets)
02521 );
02522 
02527 void ShowBuyCompanyDialog(CompanyID company)
02528 {
02529   AllocateWindowDescFront<BuyCompanyWindow>(&_buy_company_desc, company);
02530 }