company_gui.cpp

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