group_gui.cpp

Go to the documentation of this file.
00001 /* $Id: group_gui.cpp 20429 2010-08-09 21:38:18Z rubidium $ */
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 "openttd.h"
00014 #include "textbuf_gui.h"
00015 #include "command_func.h"
00016 #include "vehicle_gui.h"
00017 #include "vehicle_base.h"
00018 #include "group.h"
00019 #include "strings_func.h"
00020 #include "window_func.h"
00021 #include "vehicle_func.h"
00022 #include "autoreplace_gui.h"
00023 #include "company_func.h"
00024 #include "widgets/dropdown_type.h"
00025 #include "widgets/dropdown_func.h"
00026 #include "tilehighlight_func.h"
00027 #include "vehicle_gui_base.h"
00028 #include "core/geometry_func.hpp"
00029 #include "company_base.h"
00030 
00031 #include "table/strings.h"
00032 #include "table/sprites.h"
00033 
00034 typedef GUIList<const Group*> GUIGroupList;
00035 
00036 enum GroupListWidgets {
00037   GRP_WIDGET_CAPTION,
00038   GRP_WIDGET_SORT_BY_ORDER,
00039   GRP_WIDGET_SORT_BY_DROPDOWN,
00040   GRP_WIDGET_LIST_VEHICLE,
00041   GRP_WIDGET_LIST_VEHICLE_SCROLLBAR,
00042   GRP_WIDGET_AVAILABLE_VEHICLES,
00043   GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN,
00044   GRP_WIDGET_STOP_ALL,
00045   GRP_WIDGET_START_ALL,
00046 
00047   GRP_WIDGET_ALL_VEHICLES,
00048   GRP_WIDGET_DEFAULT_VEHICLES,
00049   GRP_WIDGET_LIST_GROUP,
00050   GRP_WIDGET_LIST_GROUP_SCROLLBAR,
00051   GRP_WIDGET_CREATE_GROUP,
00052   GRP_WIDGET_DELETE_GROUP,
00053   GRP_WIDGET_RENAME_GROUP,
00054   GRP_WIDGET_REPLACE_PROTECTION,
00055 };
00056 
00057 enum GroupActionListFunction {
00058   GALF_REPLACE,
00059   GALF_SERVICE,
00060   GALF_DEPOT,
00061   GALF_ADD_SHARED,
00062   GALF_REMOVE_ALL,
00063 };
00064 
00070 static void ShowGroupActionDropdown(Window *w, GroupID gid)
00071 {
00072   DropDownList *list = new DropDownList();
00073 
00074   list->push_back(new DropDownListStringItem(STR_VEHICLE_LIST_REPLACE_VEHICLES,    GALF_REPLACE, false));
00075   list->push_back(new DropDownListStringItem(STR_VEHICLE_LIST_SEND_FOR_SERVICING,  GALF_SERVICE, false));
00076   list->push_back(new DropDownListStringItem(STR_VEHICLE_LIST_SEND_TRAIN_TO_DEPOT, GALF_DEPOT,   false));
00077 
00078   if (Group::IsValidID(gid)) {
00079     list->push_back(new DropDownListStringItem(STR_GROUP_ADD_SHARED_VEHICLE,  GALF_ADD_SHARED, false));
00080     list->push_back(new DropDownListStringItem(STR_GROUP_REMOVE_ALL_VEHICLES, GALF_REMOVE_ALL, false));
00081   }
00082 
00083   ShowDropDownList(w, list, 0, GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN);
00084 }
00085 
00086 static const NWidgetPart _nested_group_widgets[] = {
00087   NWidget(NWID_HORIZONTAL), // Window header
00088     NWidget(WWT_CLOSEBOX, COLOUR_GREY),
00089     NWidget(WWT_CAPTION, COLOUR_GREY, GRP_WIDGET_CAPTION),
00090     NWidget(WWT_SHADEBOX, COLOUR_GREY),
00091     NWidget(WWT_STICKYBOX, COLOUR_GREY),
00092   EndContainer(),
00093   NWidget(NWID_HORIZONTAL),
00094     /* left part */
00095     NWidget(NWID_VERTICAL),
00096       NWidget(WWT_PANEL, COLOUR_GREY), SetMinimalTextLines(1, WD_DROPDOWNTEXT_TOP + WD_DROPDOWNTEXT_BOTTOM), SetFill(1, 0), EndContainer(),
00097       NWidget(WWT_PANEL, COLOUR_GREY, GRP_WIDGET_ALL_VEHICLES), SetMinimalSize(200, 13), SetFill(1, 0), EndContainer(),
00098       NWidget(WWT_PANEL, COLOUR_GREY, GRP_WIDGET_DEFAULT_VEHICLES), SetMinimalSize(200, 13), SetFill(1, 0), EndContainer(),
00099       NWidget(NWID_HORIZONTAL),
00100         NWidget(WWT_MATRIX, COLOUR_GREY, GRP_WIDGET_LIST_GROUP), SetMinimalSize(188, 0), SetDataTip(0x701, STR_GROUPS_CLICK_ON_GROUP_FOR_TOOLTIP),
00101             SetFill(1, 0), SetResize(0, 1),
00102         NWidget(WWT_SCROLL2BAR, COLOUR_GREY, GRP_WIDGET_LIST_GROUP_SCROLLBAR),
00103       EndContainer(),
00104       NWidget(NWID_HORIZONTAL),
00105         NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, GRP_WIDGET_CREATE_GROUP), SetMinimalSize(24, 25), SetFill(0, 1),
00106             SetDataTip(SPR_GROUP_CREATE_TRAIN, STR_GROUP_CREATE_TOOLTIP),
00107         NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, GRP_WIDGET_DELETE_GROUP), SetMinimalSize(24, 25), SetFill(0, 1),
00108             SetDataTip(SPR_GROUP_DELETE_TRAIN, STR_GROUP_DELETE_TOOLTIP),
00109         NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, GRP_WIDGET_RENAME_GROUP), SetMinimalSize(24, 25), SetFill(0, 1),
00110             SetDataTip(SPR_GROUP_RENAME_TRAIN, STR_GROUP_RENAME_TOOLTIP),
00111         NWidget(WWT_PANEL, COLOUR_GREY), SetMinimalSize(92, 25), SetFill(1, 1), EndContainer(),
00112         NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, GRP_WIDGET_REPLACE_PROTECTION), SetMinimalSize(24, 25), SetFill(0, 1),
00113             SetDataTip(SPR_GROUP_REPLACE_OFF_TRAIN, STR_GROUP_REPLACE_PROTECTION_TOOLTIP),
00114         NWidget(WWT_PANEL, COLOUR_GREY), SetMinimalSize(12, 25), SetFill(0, 1), EndContainer(),
00115       EndContainer(),
00116     EndContainer(),
00117     /* right part */
00118     NWidget(NWID_VERTICAL),
00119       NWidget(NWID_HORIZONTAL),
00120         NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, GRP_WIDGET_SORT_BY_ORDER), SetMinimalSize(81, 12), SetDataTip(STR_BUTTON_SORT_BY, STR_TOOLTIP_SORT_ORDER),
00121         NWidget(WWT_DROPDOWN, COLOUR_GREY, GRP_WIDGET_SORT_BY_DROPDOWN), SetMinimalSize(167, 12), SetDataTip(0x0, STR_TOOLTIP_SORT_CRITERIA),
00122         NWidget(WWT_PANEL, COLOUR_GREY), SetMinimalSize(12, 12), SetResize(1, 0), EndContainer(),
00123       EndContainer(),
00124       NWidget(NWID_HORIZONTAL),
00125         NWidget(WWT_MATRIX, COLOUR_GREY, GRP_WIDGET_LIST_VEHICLE), SetMinimalSize(248, 0), SetDataTip(0x701, STR_NULL), SetResize(1, 1), SetFill(1, 0),
00126         NWidget(WWT_SCROLLBAR, COLOUR_GREY, GRP_WIDGET_LIST_VEHICLE_SCROLLBAR),
00127       EndContainer(),
00128       NWidget(NWID_HORIZONTAL),
00129         NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, GRP_WIDGET_AVAILABLE_VEHICLES), SetMinimalSize(106, 12), SetFill(0, 1),
00130             SetDataTip(STR_BLACK_STRING, STR_VEHICLE_LIST_AVAILABLE_ENGINES_TOOLTIP),
00131         NWidget(WWT_DROPDOWN, COLOUR_GREY, GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN), SetMinimalSize(118, 12), SetFill(0, 1),
00132             SetDataTip(STR_VEHICLE_LIST_MANAGE_LIST, STR_VEHICLE_LIST_MANAGE_LIST_TOOLTIP),
00133         NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, GRP_WIDGET_STOP_ALL), SetMinimalSize(12, 12), SetFill(0, 1),
00134             SetDataTip(SPR_FLAG_VEH_STOPPED, STR_VEHICLE_LIST_MASS_STOP_LIST_TOOLTIP),
00135         NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, GRP_WIDGET_START_ALL), SetMinimalSize(12, 12), SetFill(0, 1),
00136             SetDataTip(SPR_FLAG_VEH_RUNNING, STR_VEHICLE_LIST_MASS_START_LIST_TOOLTIP),
00137         NWidget(WWT_PANEL, COLOUR_GREY), SetMinimalSize(0, 12), SetFill(1, 1), SetResize(1, 0), EndContainer(),
00138         NWidget(WWT_RESIZEBOX, COLOUR_GREY),
00139       EndContainer(),
00140     EndContainer(),
00141   EndContainer(),
00142 };
00143 
00144 class VehicleGroupWindow : public BaseVehicleListWindow {
00145 private:
00146   GroupID group_sel;     
00147   VehicleID vehicle_sel; 
00148   GroupID group_rename;  
00149   GUIGroupList groups;   
00150   uint tiny_step_height; 
00151 
00157   void BuildGroupList(Owner owner)
00158   {
00159     if (!this->groups.NeedRebuild()) return;
00160 
00161     this->groups.Clear();
00162 
00163     const Group *g;
00164     FOR_ALL_GROUPS(g) {
00165       if (g->owner == owner && g->vehicle_type == this->vehicle_type) {
00166         *this->groups.Append() = g;
00167       }
00168     }
00169 
00170     this->groups.Compact();
00171     this->groups.RebuildDone();
00172   }
00173 
00175   static int CDECL GroupNameSorter(const Group * const *a, const Group * const *b)
00176   {
00177     static const Group *last_group[2] = { NULL, NULL };
00178     static char         last_name[2][64] = { "", "" };
00179 
00180     if (*a != last_group[0]) {
00181       last_group[0] = *a;
00182       SetDParam(0, (*a)->index);
00183       GetString(last_name[0], STR_GROUP_NAME, lastof(last_name[0]));
00184     }
00185 
00186     if (*b != last_group[1]) {
00187       last_group[1] = *b;
00188       SetDParam(0, (*b)->index);
00189       GetString(last_name[1], STR_GROUP_NAME, lastof(last_name[1]));
00190     }
00191 
00192     int r = strcmp(last_name[0], last_name[1]); // sort by name
00193     if (r == 0) return (*a)->index - (*b)->index;
00194     return r;
00195   }
00196 
00197 public:
00198   VehicleGroupWindow(const WindowDesc *desc, WindowNumber window_number) : BaseVehicleListWindow()
00199   {
00200     this->CreateNestedTree(desc);
00201 
00202     this->vehicle_type = (VehicleType)GB(window_number, 11, 5);
00203     switch (this->vehicle_type) {
00204       default: NOT_REACHED();
00205       case VEH_TRAIN:    this->sorting = &_sorting.train;    break;
00206       case VEH_ROAD:     this->sorting = &_sorting.roadveh;  break;
00207       case VEH_SHIP:     this->sorting = &_sorting.ship;     break;
00208       case VEH_AIRCRAFT: this->sorting = &_sorting.aircraft; break;
00209     }
00210 
00211     this->group_sel = ALL_GROUP;
00212     this->vehicle_sel = INVALID_VEHICLE;
00213     this->group_rename = INVALID_GROUP;
00214 
00215     const Owner owner = (Owner)GB(window_number, 0, 8);
00216     this->vehicles.SetListing(*this->sorting);
00217     this->vehicles.ForceRebuild();
00218     this->vehicles.NeedResort();
00219     this->BuildVehicleList(owner, this->group_sel, IsAllGroupID(this->group_sel) ? VLW_STANDARD : VLW_GROUP_LIST);
00220     this->SortVehicleList();
00221 
00222     this->groups.ForceRebuild();
00223     this->groups.NeedResort();
00224     this->BuildGroupList(owner);
00225     this->groups.Sort(&GroupNameSorter);
00226 
00227     this->GetWidget<NWidgetCore>(GRP_WIDGET_CAPTION)->widget_data = STR_VEHICLE_LIST_TRAIN_CAPTION + this->vehicle_type;
00228     this->GetWidget<NWidgetCore>(GRP_WIDGET_LIST_VEHICLE)->tool_tip = STR_VEHICLE_LIST_TRAIN_LIST_TOOLTIP + this->vehicle_type;
00229 
00230     this->GetWidget<NWidgetCore>(GRP_WIDGET_CREATE_GROUP)->widget_data += this->vehicle_type;
00231     this->GetWidget<NWidgetCore>(GRP_WIDGET_RENAME_GROUP)->widget_data += this->vehicle_type;
00232     this->GetWidget<NWidgetCore>(GRP_WIDGET_DELETE_GROUP)->widget_data += this->vehicle_type;
00233     this->GetWidget<NWidgetCore>(GRP_WIDGET_REPLACE_PROTECTION)->widget_data += this->vehicle_type;
00234 
00235     this->FinishInitNested(desc, window_number);
00236     this->owner = owner;
00237   }
00238 
00239   ~VehicleGroupWindow()
00240   {
00241     *this->sorting = this->vehicles.GetListing();
00242   }
00243 
00244   virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
00245   {
00246     switch (widget) {
00247       case GRP_WIDGET_LIST_GROUP:
00248         this->tiny_step_height = FONT_HEIGHT_NORMAL + WD_MATRIX_TOP;
00249         resize->height = this->tiny_step_height;
00250         /* Minimum height is the height of the list widget minus all and default vehicles and a bit for the bottom bar */
00251         size->height =  4 * GetVehicleListHeight(this->vehicle_type, this->tiny_step_height) - (this->tiny_step_height > 25 ? 2 : 3) * this->tiny_step_height;
00252         break;
00253 
00254       case GRP_WIDGET_ALL_VEHICLES:
00255       case GRP_WIDGET_DEFAULT_VEHICLES:
00256         size->height = FONT_HEIGHT_NORMAL + WD_MATRIX_TOP;
00257         size->width = max(GetStringBoundingBox(STR_GROUP_DEFAULT_TRAINS + this->vehicle_type).width, GetStringBoundingBox(STR_GROUP_ALL_TRAINS + this->vehicle_type).width);
00258         size->width += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT + 8 + 8;
00259         break;
00260 
00261       case GRP_WIDGET_LIST_VEHICLE:
00262         resize->height = GetVehicleListHeight(this->vehicle_type, FONT_HEIGHT_NORMAL + WD_MATRIX_TOP);
00263         size->height = 4 * resize->height;
00264         break;
00265 
00266       case GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN: {
00267         static const StringID _dropdown_text[] = {STR_VEHICLE_LIST_REPLACE_VEHICLES, STR_VEHICLE_LIST_SEND_FOR_SERVICING, STR_VEHICLE_LIST_SEND_TRAIN_TO_DEPOT, STR_GROUP_ADD_SHARED_VEHICLE, STR_GROUP_REMOVE_ALL_VEHICLES};
00268         Dimension d = {0, 0};
00269         for (const StringID *sid = _dropdown_text; sid != endof(_dropdown_text); sid++) {
00270           d = maxdim(d, GetStringBoundingBox(*sid));
00271         }
00272         d.height += padding.height;
00273         d.width  += padding.width;
00274         *size = maxdim(*size, d);
00275       } break;
00276     }
00277   }
00278 
00279   virtual void OnInvalidateData(int data)
00280   {
00281     if (data == 0) {
00282       this->vehicles.ForceRebuild();
00283       this->groups.ForceRebuild();
00284     } else {
00285       this->vehicles.ForceResort();
00286       this->groups.ForceResort();
00287     }
00288 
00289     if (this->group_rename != INVALID_GROUP && !Group::IsValidID(this->group_rename)) {
00290       DeleteWindowByClass(WC_QUERY_STRING);
00291       this->group_rename = INVALID_GROUP;
00292     }
00293 
00294     if (!(IsAllGroupID(this->group_sel) || IsDefaultGroupID(this->group_sel) || Group::IsValidID(this->group_sel))) {
00295       this->group_sel = ALL_GROUP;
00296       HideDropDownMenu(this);
00297     }
00298     this->SetDirty();
00299   }
00300 
00301   virtual void SetStringParameters(int widget) const
00302   {
00303     switch (widget) {
00304       case GRP_WIDGET_AVAILABLE_VEHICLES:
00305         SetDParam(0, STR_VEHICLE_LIST_AVAILABLE_TRAINS + this->vehicle_type);
00306         break;
00307 
00308       case GRP_WIDGET_CAPTION:
00309         /* If selected_group == DEFAULT_GROUP || ALL_GROUP, draw the standard caption
00310          * We list all vehicles or ungrouped vehicles */
00311         if (IsDefaultGroupID(this->group_sel) || IsAllGroupID(this->group_sel)) {
00312           SetDParam(0, STR_COMPANY_NAME);
00313           SetDParam(1, GB(this->window_number, 0, 8));
00314           SetDParam(2, this->vehicles.Length());
00315           SetDParam(3, this->vehicles.Length());
00316         } else {
00317           const Group *g = Group::Get(this->group_sel);
00318 
00319           SetDParam(0, STR_GROUP_NAME);
00320           SetDParam(1, g->index);
00321           SetDParam(2, g->num_vehicle);
00322           SetDParam(3, g->num_vehicle);
00323         }
00324         break;
00325     }
00326   }
00327 
00328   virtual void OnPaint()
00329   {
00330     const Owner owner = (Owner)GB(this->window_number, 0, 8);
00331 
00332     /* If we select the all vehicles, this->list will contain all vehicles of the owner
00333      * else this->list will contain all vehicles which belong to the selected group */
00334     this->BuildVehicleList(owner, this->group_sel, IsAllGroupID(this->group_sel) ? VLW_STANDARD : VLW_GROUP_LIST);
00335     this->SortVehicleList();
00336 
00337     this->BuildGroupList(owner);
00338     this->groups.Sort(&GroupNameSorter);
00339 
00340     this->vscroll2.SetCount(this->groups.Length());
00341     this->vscroll.SetCount(this->vehicles.Length());
00342 
00343     /* The drop down menu is out, *but* it may not be used, retract it. */
00344     if (this->vehicles.Length() == 0 && this->IsWidgetLowered(GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN)) {
00345       this->RaiseWidget(GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN);
00346       HideDropDownMenu(this);
00347     }
00348 
00349     /* Disable all lists management button when the list is empty */
00350     this->SetWidgetsDisabledState(this->vehicles.Length() == 0 || _local_company != owner,
00351         GRP_WIDGET_STOP_ALL,
00352         GRP_WIDGET_START_ALL,
00353         GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN,
00354         WIDGET_LIST_END);
00355 
00356     /* Disable the group specific function when we select the default group or all vehicles */
00357     this->SetWidgetsDisabledState(IsDefaultGroupID(this->group_sel) || IsAllGroupID(this->group_sel) || _local_company != owner,
00358         GRP_WIDGET_DELETE_GROUP,
00359         GRP_WIDGET_RENAME_GROUP,
00360         GRP_WIDGET_REPLACE_PROTECTION,
00361         WIDGET_LIST_END);
00362 
00363     /* Disable remaining buttons for non-local companies
00364      * Needed while changing _local_company, eg. by cheats
00365      * All procedures (eg. move vehicle to another group)
00366      *  verify, whether you are the owner of the vehicle,
00367      *  so it doesn't have to be disabled
00368      */
00369     this->SetWidgetsDisabledState(_local_company != owner,
00370         GRP_WIDGET_CREATE_GROUP,
00371         GRP_WIDGET_AVAILABLE_VEHICLES,
00372         WIDGET_LIST_END);
00373 
00374     /* If not a default group and the group has replace protection, show an enabled replace sprite. */
00375     uint16 protect_sprite = SPR_GROUP_REPLACE_OFF_TRAIN;
00376     if (!IsDefaultGroupID(this->group_sel) && !IsAllGroupID(this->group_sel) && Group::Get(this->group_sel)->replace_protection) protect_sprite = SPR_GROUP_REPLACE_ON_TRAIN;
00377     this->GetWidget<NWidgetCore>(GRP_WIDGET_REPLACE_PROTECTION)->widget_data = protect_sprite + this->vehicle_type;
00378 
00379     /* Set text of sort by dropdown */
00380     this->GetWidget<NWidgetCore>(GRP_WIDGET_SORT_BY_DROPDOWN)->widget_data = this->vehicle_sorter_names[this->vehicles.SortType()];
00381 
00382     this->DrawWidgets();
00383   }
00384 
00385   virtual void DrawWidget(const Rect &r, int widget) const
00386   {
00387     switch (widget) {
00388       case GRP_WIDGET_ALL_VEHICLES:
00389         DrawString(r.left + WD_FRAMERECT_LEFT + 8, r.right - WD_FRAMERECT_RIGHT - 8, r.top + WD_FRAMERECT_TOP + 1,
00390             STR_GROUP_ALL_TRAINS + this->vehicle_type, IsAllGroupID(this->group_sel) ? TC_WHITE : TC_BLACK);
00391         break;
00392 
00393       case GRP_WIDGET_DEFAULT_VEHICLES:
00394         DrawString(r.left + WD_FRAMERECT_LEFT + 8, r.right - WD_FRAMERECT_RIGHT - 8, r.top + WD_FRAMERECT_TOP + 1,
00395             STR_GROUP_DEFAULT_TRAINS + this->vehicle_type, IsDefaultGroupID(this->group_sel) ? TC_WHITE : TC_BLACK);
00396         break;
00397 
00398       case GRP_WIDGET_LIST_GROUP: {
00399         int y1 = r.top + WD_FRAMERECT_TOP + 1;
00400         int max = min(this->vscroll2.GetPosition() + this->vscroll2.GetCapacity(), this->groups.Length());
00401         for (int i = this->vscroll2.GetPosition(); i < max; ++i) {
00402           const Group *g = this->groups[i];
00403 
00404           assert(g->owner == this->owner);
00405 
00406           /* draw the selected group in white, else we draw it in black */
00407           SetDParam(0, g->index);
00408           DrawString(r.left + WD_FRAMERECT_LEFT + 8, r.right - WD_FRAMERECT_RIGHT - 8, y1, STR_GROUP_NAME, (this->group_sel == g->index) ? TC_WHITE : TC_BLACK);
00409 
00410           /* draw the number of vehicles of the group */
00411           SetDParam(0, g->num_vehicle);
00412           DrawString(r.left + WD_FRAMERECT_LEFT, r.right - WD_FRAMERECT_RIGHT, y1 + 1, STR_TINY_COMMA, (this->group_sel == g->index) ? TC_WHITE : TC_BLACK, SA_RIGHT);
00413 
00414           y1 += this->tiny_step_height;
00415         }
00416         break;
00417       }
00418 
00419       case GRP_WIDGET_SORT_BY_ORDER:
00420         this->DrawSortButtonState(GRP_WIDGET_SORT_BY_ORDER, this->vehicles.IsDescSortOrder() ? SBS_DOWN : SBS_UP);
00421         break;
00422 
00423       case GRP_WIDGET_LIST_VEHICLE:
00424         this->DrawVehicleListItems(this->vehicle_sel, this->resize.step_height, r);
00425         break;
00426     }
00427   }
00428 
00429   virtual void OnClick(Point pt, int widget, int click_count)
00430   {
00431     switch (widget) {
00432       case GRP_WIDGET_SORT_BY_ORDER: // Flip sorting method ascending/descending
00433         this->vehicles.ToggleSortOrder();
00434         this->SetDirty();
00435         break;
00436 
00437       case GRP_WIDGET_SORT_BY_DROPDOWN: // Select sorting criteria dropdown menu
00438         ShowDropDownMenu(this, this->vehicle_sorter_names, this->vehicles.SortType(),  GRP_WIDGET_SORT_BY_DROPDOWN, 0, (this->vehicle_type == VEH_TRAIN || this->vehicle_type == VEH_ROAD) ? 0 : (1 << 10));
00439         return;
00440 
00441       case GRP_WIDGET_ALL_VEHICLES: // All vehicles button
00442         if (!IsAllGroupID(this->group_sel)) {
00443           this->group_sel = ALL_GROUP;
00444           this->vehicles.ForceRebuild();
00445           this->SetDirty();
00446         }
00447         break;
00448 
00449       case GRP_WIDGET_DEFAULT_VEHICLES: // Ungrouped vehicles button
00450         if (!IsDefaultGroupID(this->group_sel)) {
00451           this->group_sel = DEFAULT_GROUP;
00452           this->vehicles.ForceRebuild();
00453           this->SetDirty();
00454         }
00455         break;
00456 
00457       case GRP_WIDGET_LIST_GROUP: { // Matrix Group
00458         uint16 id_g = (pt.y - this->GetWidget<NWidgetBase>(GRP_WIDGET_LIST_GROUP)->pos_y) / (int)this->tiny_step_height;
00459 
00460         if (id_g >= this->vscroll2.GetCapacity()) return;
00461 
00462         id_g += this->vscroll2.GetPosition();
00463 
00464         if (id_g >= this->groups.Length()) return;
00465 
00466         this->group_sel = this->groups[id_g]->index;
00467 
00468         this->vehicles.ForceRebuild();
00469         this->SetDirty();
00470         break;
00471       }
00472 
00473       case GRP_WIDGET_LIST_VEHICLE: { // Matrix Vehicle
00474         uint32 id_v = (pt.y - this->GetWidget<NWidgetBase>(GRP_WIDGET_LIST_VEHICLE)->pos_y) / (int)this->resize.step_height;
00475         if (id_v >= this->vscroll.GetCapacity()) return; // click out of bounds
00476 
00477         id_v += this->vscroll.GetPosition();
00478 
00479         if (id_v >= this->vehicles.Length()) return; // click out of list bound
00480 
00481         const Vehicle *v = this->vehicles[id_v];
00482 
00483         this->vehicle_sel = v->index;
00484 
00485         SetObjectToPlaceWnd(v->GetImage(DIR_W), GetVehiclePalette(v), HT_DRAG, this);
00486         _cursor.vehchain = true;
00487 
00488         this->SetDirty();
00489         break;
00490       }
00491 
00492       case GRP_WIDGET_CREATE_GROUP: { // Create a new group
00493         DoCommandP(0, this->vehicle_type, 0, CMD_CREATE_GROUP | CMD_MSG(STR_ERROR_GROUP_CAN_T_CREATE), CcCreateGroup);
00494         break;
00495       }
00496 
00497       case GRP_WIDGET_DELETE_GROUP: { // Delete the selected group
00498         GroupID group = this->group_sel;
00499         this->group_sel = ALL_GROUP;
00500 
00501         DoCommandP(0, group, 0, CMD_DELETE_GROUP | CMD_MSG(STR_ERROR_GROUP_CAN_T_DELETE));
00502         break;
00503       }
00504 
00505       case GRP_WIDGET_RENAME_GROUP: // Rename the selected roup
00506         this->ShowRenameGroupWindow(this->group_sel, false);
00507         break;
00508 
00509       case GRP_WIDGET_AVAILABLE_VEHICLES:
00510         ShowBuildVehicleWindow(INVALID_TILE, this->vehicle_type);
00511         break;
00512 
00513       case GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN:
00514         ShowGroupActionDropdown(this, this->group_sel);
00515         break;
00516 
00517       case GRP_WIDGET_START_ALL:
00518       case GRP_WIDGET_STOP_ALL: { // Start/stop all vehicles of the list
00519         DoCommandP(0, this->group_sel, ((IsAllGroupID(this->group_sel) ? VLW_STANDARD : VLW_GROUP_LIST) & VLW_MASK)
00520                           | (1 << 6)
00521                           | (widget == GRP_WIDGET_START_ALL ? (1 << 5) : 0)
00522                           | this->vehicle_type, CMD_MASS_START_STOP);
00523 
00524         break;
00525       }
00526 
00527       case GRP_WIDGET_REPLACE_PROTECTION: {
00528         const Group *g = Group::GetIfValid(this->group_sel);
00529         if (g != NULL) {
00530           DoCommandP(0, this->group_sel, !g->replace_protection, CMD_SET_GROUP_REPLACE_PROTECTION);
00531         }
00532         break;
00533       }
00534     }
00535   }
00536 
00537   virtual void OnDragDrop(Point pt, int widget)
00538   {
00539     switch (widget) {
00540       case GRP_WIDGET_ALL_VEHICLES: // All vehicles
00541       case GRP_WIDGET_DEFAULT_VEHICLES: // Ungrouped vehicles
00542         DoCommandP(0, DEFAULT_GROUP, this->vehicle_sel, CMD_ADD_VEHICLE_GROUP | CMD_MSG(STR_ERROR_GROUP_CAN_T_ADD_VEHICLE));
00543 
00544         this->vehicle_sel = INVALID_VEHICLE;
00545 
00546         this->SetDirty();
00547         break;
00548 
00549       case GRP_WIDGET_LIST_GROUP: { // Maxtrix group
00550         uint16 id_g = (pt.y - this->GetWidget<NWidgetBase>(GRP_WIDGET_LIST_GROUP)->pos_y) / (int)this->tiny_step_height;
00551         const VehicleID vindex = this->vehicle_sel;
00552 
00553         this->vehicle_sel = INVALID_VEHICLE;
00554 
00555         this->SetDirty();
00556 
00557         if (id_g >= this->vscroll2.GetCapacity()) return;
00558 
00559         id_g += this->vscroll2.GetPosition();
00560 
00561         if (id_g >= this->groups.Length()) return;
00562 
00563         DoCommandP(0, this->groups[id_g]->index, vindex, CMD_ADD_VEHICLE_GROUP | CMD_MSG(STR_ERROR_GROUP_CAN_T_ADD_VEHICLE));
00564         break;
00565       }
00566 
00567       case GRP_WIDGET_LIST_VEHICLE: { // Maxtrix vehicle
00568         uint32 id_v = (pt.y - this->GetWidget<NWidgetBase>(GRP_WIDGET_LIST_VEHICLE)->pos_y) / (int)this->resize.step_height;
00569         const VehicleID vindex = this->vehicle_sel;
00570 
00571         this->vehicle_sel = INVALID_VEHICLE;
00572 
00573         this->SetDirty();
00574 
00575         if (id_v >= this->vscroll.GetCapacity()) return; // click out of bounds
00576 
00577         id_v += this->vscroll.GetPosition();
00578 
00579         if (id_v >= this->vehicles.Length()) return; // click out of list bound
00580 
00581         const Vehicle *v = this->vehicles[id_v];
00582         if (vindex == v->index) {
00583           ShowVehicleViewWindow(v);
00584         }
00585         break;
00586       }
00587     }
00588     _cursor.vehchain = false;
00589   }
00590 
00591   virtual void OnQueryTextFinished(char *str)
00592   {
00593     if (str != NULL) DoCommandP(0, this->group_rename, 0, CMD_RENAME_GROUP | CMD_MSG(STR_ERROR_GROUP_CAN_T_RENAME), NULL, str);
00594     this->group_rename = INVALID_GROUP;
00595   }
00596 
00597   virtual void OnResize()
00598   {
00599     NWidgetCore *nwi = this->GetWidget<NWidgetCore>(GRP_WIDGET_LIST_GROUP);
00600     this->vscroll2.SetCapacity(nwi->current_y / this->tiny_step_height);
00601     nwi->widget_data = (this->vscroll2.GetCapacity() << MAT_ROW_START) + (1 << MAT_COL_START);
00602 
00603     nwi = this->GetWidget<NWidgetCore>(GRP_WIDGET_LIST_VEHICLE);
00604     this->vscroll.SetCapacityFromWidget(this, GRP_WIDGET_LIST_VEHICLE);
00605     nwi->widget_data = (this->vscroll.GetCapacity() << MAT_ROW_START) + (1 << MAT_COL_START);
00606   }
00607 
00608   virtual void OnDropdownSelect(int widget, int index)
00609   {
00610     switch (widget) {
00611       case GRP_WIDGET_SORT_BY_DROPDOWN:
00612         this->vehicles.SetSortType(index);
00613         break;
00614 
00615       case GRP_WIDGET_MANAGE_VEHICLES_DROPDOWN:
00616         assert(this->vehicles.Length() != 0);
00617 
00618         switch (index) {
00619           case GALF_REPLACE: // Replace window
00620             ShowReplaceGroupVehicleWindow(this->group_sel, this->vehicle_type);
00621             break;
00622           case GALF_SERVICE: // Send for servicing
00623             DoCommandP(0, this->group_sel, ((IsAllGroupID(this->group_sel) ? VLW_STANDARD : VLW_GROUP_LIST) & VLW_MASK)
00624                   | DEPOT_MASS_SEND
00625                   | DEPOT_SERVICE, GetCmdSendToDepot(this->vehicle_type));
00626             break;
00627           case GALF_DEPOT: // Send to Depots
00628             DoCommandP(0, this->group_sel, ((IsAllGroupID(this->group_sel) ? VLW_STANDARD : VLW_GROUP_LIST) & VLW_MASK)
00629                   | DEPOT_MASS_SEND, GetCmdSendToDepot(this->vehicle_type));
00630             break;
00631           case GALF_ADD_SHARED: // Add shared Vehicles
00632             assert(Group::IsValidID(this->group_sel));
00633 
00634             DoCommandP(0, this->group_sel, this->vehicle_type, CMD_ADD_SHARED_VEHICLE_GROUP | CMD_MSG(STR_ERROR_GROUP_CAN_T_ADD_SHARED_VEHICLE));
00635             break;
00636           case GALF_REMOVE_ALL: // Remove all Vehicles from the selected group
00637             assert(Group::IsValidID(this->group_sel));
00638 
00639             DoCommandP(0, this->group_sel, this->vehicle_type, CMD_REMOVE_ALL_VEHICLES_GROUP | CMD_MSG(STR_ERROR_GROUP_CAN_T_REMOVE_ALL_VEHICLES));
00640             break;
00641           default: NOT_REACHED();
00642         }
00643         break;
00644 
00645       default: NOT_REACHED();
00646     }
00647 
00648     this->SetDirty();
00649   }
00650 
00651   virtual void OnTick()
00652   {
00653     if (_pause_mode != PM_UNPAUSED) return;
00654     if (this->groups.NeedResort() || this->vehicles.NeedResort()) {
00655       this->SetDirty();
00656     }
00657   }
00658 
00659   virtual void OnPlaceObjectAbort()
00660   {
00661     /* abort drag & drop */
00662     this->vehicle_sel = INVALID_VEHICLE;
00663     this->SetWidgetDirty(GRP_WIDGET_LIST_VEHICLE);
00664   }
00665 
00666   void ShowRenameGroupWindow(GroupID group, bool empty)
00667   {
00668     assert(Group::IsValidID(group));
00669     this->group_rename = group;
00670     /* Show empty query for new groups */
00671     StringID str = STR_EMPTY;
00672     if (!empty) {
00673       SetDParam(0, group);
00674       str = STR_GROUP_NAME;
00675     }
00676     ShowQueryString(str, STR_GROUP_RENAME_CAPTION, MAX_LENGTH_GROUP_NAME_BYTES, MAX_LENGTH_GROUP_NAME_PIXELS, this, CS_ALPHANUMERAL, QSF_ENABLE_DEFAULT);
00677   }
00678 
00684   void UnselectVehicle(VehicleID vehicle)
00685   {
00686     if (this->vehicle_sel == vehicle) ResetObjectToPlace();
00687   }
00688 };
00689 
00690 
00691 static WindowDesc _other_group_desc(
00692   WDP_AUTO, 460, 246,
00693   WC_INVALID, WC_NONE,
00694   WDF_UNCLICK_BUTTONS,
00695   _nested_group_widgets, lengthof(_nested_group_widgets)
00696 );
00697 
00698 static const WindowDesc _train_group_desc(
00699   WDP_AUTO, 525, 246,
00700   WC_TRAINS_LIST, WC_NONE,
00701   WDF_UNCLICK_BUTTONS,
00702   _nested_group_widgets, lengthof(_nested_group_widgets)
00703 );
00704 
00705 void ShowCompanyGroup(CompanyID company, VehicleType vehicle_type)
00706 {
00707   if (!Company::IsValidID(company)) return;
00708 
00709   WindowNumber num = (vehicle_type << 11) | VLW_GROUP_LIST | company;
00710   if (vehicle_type == VEH_TRAIN) {
00711     AllocateWindowDescFront<VehicleGroupWindow>(&_train_group_desc, num);
00712   } else {
00713     _other_group_desc.cls = GetWindowClassForVehicleType(vehicle_type);
00714     AllocateWindowDescFront<VehicleGroupWindow>(&_other_group_desc, num);
00715   }
00716 }
00717 
00724 static inline VehicleGroupWindow *FindVehicleGroupWindow(VehicleType vt, Owner owner)
00725 {
00726   return (VehicleGroupWindow *)FindWindowById(GetWindowClassForVehicleType(vt), (vt << 11) | VLW_GROUP_LIST | owner);
00727 }
00728 
00737 void CcCreateGroup(const CommandCost &result, TileIndex tile, uint32 p1, uint32 p2)
00738 {
00739   if (result.Failed()) return;
00740   assert(p1 <= VEH_AIRCRAFT);
00741 
00742   VehicleGroupWindow *w = FindVehicleGroupWindow((VehicleType)p1, _current_company);
00743   if (w != NULL) w->ShowRenameGroupWindow(_new_group_id, true);
00744 }
00745 
00750 void DeleteGroupHighlightOfVehicle(const Vehicle *v)
00751 {
00752   /* If we haven't got any vehicles on the mouse pointer, we haven't got any highlighted in any group windows either
00753    * If that is the case, we can skip looping though the windows and save time
00754    */
00755   if (_special_mouse_mode != WSM_DRAGDROP) return;
00756 
00757   VehicleGroupWindow *w = FindVehicleGroupWindow(v->type, v->owner);
00758   if (w != NULL) w->UnselectVehicle(v->index);
00759 }

Generated on Tue Sep 14 17:06:49 2010 for OpenTTD by  doxygen 1.6.1