(svn r28004) -Update from Eints:
[openttd.git] / src / tree_gui.cpp
blobf21eeaef2ed36e7a524b9c69daf099d3425c9bef
1 /* $Id$ */
3 /*
4 * This file is part of OpenTTD.
5 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
6 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
7 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
8 */
10 /** @file tree_gui.cpp GUIs for building trees. */
12 #include "stdafx.h"
13 #include "window_gui.h"
14 #include "gfx_func.h"
15 #include "tilehighlight_func.h"
16 #include "company_func.h"
17 #include "company_base.h"
18 #include "command_func.h"
19 #include "sound_func.h"
20 #include "tree_map.h"
22 #include "widgets/tree_widget.h"
24 #include "table/sprites.h"
25 #include "table/strings.h"
26 #include "table/tree_land.h"
28 #include "safeguards.h"
30 void PlaceTreesRandomly();
32 /** Tree Sprites with their palettes */
33 const PalSpriteID tree_sprites[] = {
34 { 1621, PAL_NONE }, { 1587, PAL_NONE }, { 1656, PAL_NONE }, { 1579, PAL_NONE },
35 { 1607, PAL_NONE }, { 1593, PAL_NONE }, { 1614, PAL_NONE }, { 1586, PAL_NONE },
36 { 1663, PAL_NONE }, { 1677, PAL_NONE }, { 1691, PAL_NONE }, { 1705, PAL_NONE },
37 { 1711, PAL_NONE }, { 1746, PAL_NONE }, { 1753, PAL_NONE }, { 1732, PAL_NONE },
38 { 1739, PAL_NONE }, { 1718, PAL_NONE }, { 1725, PAL_NONE }, { 1760, PAL_NONE },
39 { 1838, PAL_NONE }, { 1844, PAL_NONE }, { 1866, PAL_NONE }, { 1871, PAL_NONE },
40 { 1899, PAL_NONE }, { 1935, PAL_NONE }, { 1928, PAL_NONE }, { 1915, PAL_NONE },
41 { 1887, PAL_NONE }, { 1908, PAL_NONE }, { 1824, PAL_NONE }, { 1943, PAL_NONE },
42 { 1950, PAL_NONE }, { 1957, PALETTE_TO_GREEN }, { 1964, PALETTE_TO_RED }, { 1971, PAL_NONE },
43 { 1978, PAL_NONE }, { 1985, PALETTE_TO_RED, }, { 1992, PALETTE_TO_PALE_GREEN }, { 1999, PALETTE_TO_YELLOW }, { 2006, PALETTE_TO_RED }
47 /**
48 * The build trees window.
50 class BuildTreesWindow : public Window
52 uint16 base; ///< Base tree number used for drawing the window.
53 uint16 count; ///< Number of different trees available.
54 TreeType tree_to_plant; ///< Tree number to plant, \c TREE_INVALID for a random tree.
56 public:
57 BuildTreesWindow(WindowDesc *desc, WindowNumber window_number) : Window(desc)
59 this->InitNested(window_number);
60 ResetObjectToPlace();
63 /**
64 * Calculate the maximum size of all tree sprites
65 * @return Dimension of the largest tree sprite
67 Dimension GetMaxTreeSpriteSize()
69 Dimension size, this_size;
70 Point offset;
71 /* Avoid to use it uninitialized */
72 size.width = 32; // default width - 2
73 size.height = 39; // default height - 7
74 offset.x = 0;
75 offset.y = 0;
77 for (int i = this->base; i < this->base + this->count; i++) {
78 if (i >= (int)lengthof(tree_sprites)) return size;
79 this_size = GetSpriteSize(tree_sprites[i].sprite, &offset);
80 size.width = max<int>(size.width, 2 * max<int>(this_size.width, -offset.x));
81 size.height = max<int>(size.height, max<int>(this_size.height, -offset.y));
84 return size;
87 virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
89 if (widget >= WID_BT_TYPE_11 && widget <= WID_BT_TYPE_34) {
90 Dimension d = GetMaxTreeSpriteSize();
91 /* Allow some pixels extra width and height */
92 size->width = d.width + WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
93 size->height = d.height + WD_FRAMERECT_RIGHT + WD_FRAMERECT_BOTTOM + 7; // we need some more space
94 return;
97 if (widget != WID_BT_MANY_RANDOM) return;
99 if (_game_mode != GM_EDITOR) {
100 size->width = 0;
101 size->height = 0;
105 virtual void DrawWidget(const Rect &r, int widget) const
107 if (widget < WID_BT_TYPE_11 || widget > WID_BT_TYPE_34 || widget - WID_BT_TYPE_11 >= this->count) return;
109 int i = this->base + widget - WID_BT_TYPE_11;
110 /* Trees "grow" in the centre on the bottom line of the buttons */
111 DrawSprite(tree_sprites[i].sprite, tree_sprites[i].pal, (r.left + r.right) / 2 + WD_FRAMERECT_LEFT, r.bottom - 7);
114 virtual void OnClick(Point pt, int widget, int click_count)
116 switch (widget) {
117 case WID_BT_TYPE_11: case WID_BT_TYPE_12: case WID_BT_TYPE_13: case WID_BT_TYPE_14:
118 case WID_BT_TYPE_21: case WID_BT_TYPE_22: case WID_BT_TYPE_23: case WID_BT_TYPE_24:
119 case WID_BT_TYPE_31: case WID_BT_TYPE_32: case WID_BT_TYPE_33: case WID_BT_TYPE_34:
120 if (widget - WID_BT_TYPE_11 >= this->count) break;
122 if (HandlePlacePushButton(this, widget, SPR_CURSOR_TREE, HT_RECT)) {
123 this->tree_to_plant = (TreeType)(this->base + widget - WID_BT_TYPE_11);
125 break;
127 case WID_BT_TYPE_RANDOM: // tree of random type.
128 if (HandlePlacePushButton(this, WID_BT_TYPE_RANDOM, SPR_CURSOR_TREE, HT_RECT)) {
129 this->tree_to_plant = TREE_INVALID;
131 break;
133 case WID_BT_MANY_RANDOM: // place trees randomly over the landscape
134 if (_settings_client.sound.confirm) SndPlayFx(SND_15_BEEP);
135 PlaceTreesRandomly();
136 MarkWholeScreenDirty();
137 break;
141 virtual void OnPlaceObject(Point pt, TileIndex tile)
143 VpStartPlaceSizing(tile, VPM_X_AND_Y, DDSP_PLANT_TREES);
146 virtual void OnPlaceDrag(ViewportPlaceMethod select_method, ViewportDragDropSelectionProcess select_proc, Point pt)
148 VpSelectTilesWithMethod(pt.x, pt.y, select_method);
151 virtual void OnPlaceMouseUp(ViewportPlaceMethod select_method, ViewportDragDropSelectionProcess select_proc, Point pt, TileIndex start_tile, TileIndex end_tile)
153 if (pt.x != -1 && select_proc == DDSP_PLANT_TREES) {
154 DoCommandP(end_tile, this->tree_to_plant, start_tile,
155 CMD_PLANT_TREE | CMD_MSG(STR_ERROR_CAN_T_PLANT_TREE_HERE));
160 * Initialize the window data
162 virtual void OnInit()
164 this->base = _tree_base_by_landscape[_settings_game.game_creation.landscape];
165 this->count = _tree_count_by_landscape[_settings_game.game_creation.landscape];
168 virtual void OnPlaceObjectAbort()
170 this->RaiseButtons();
174 static const NWidgetPart _nested_build_trees_widgets[] = {
175 NWidget(NWID_HORIZONTAL),
176 NWidget(WWT_CLOSEBOX, COLOUR_DARK_GREEN),
177 NWidget(WWT_CAPTION, COLOUR_DARK_GREEN), SetDataTip(STR_PLANT_TREE_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
178 NWidget(WWT_SHADEBOX, COLOUR_DARK_GREEN),
179 NWidget(WWT_STICKYBOX, COLOUR_DARK_GREEN),
180 EndContainer(),
181 NWidget(WWT_PANEL, COLOUR_DARK_GREEN),
182 NWidget(NWID_SPACER), SetMinimalSize(0, 2),
183 NWidget(NWID_HORIZONTAL),
184 NWidget(NWID_SPACER), SetMinimalSize(2, 0),
185 NWidget(NWID_VERTICAL),
186 NWidget(NWID_HORIZONTAL),
187 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_11), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
188 EndContainer(),
189 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
190 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_12), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
191 EndContainer(),
192 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
193 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_13), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
194 EndContainer(),
195 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
196 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_14), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
197 EndContainer(),
198 EndContainer(),
199 NWidget(NWID_SPACER), SetMinimalSize(0, 1),
200 NWidget(NWID_HORIZONTAL),
201 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_21), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
202 EndContainer(),
203 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
204 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_22), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
205 EndContainer(),
206 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
207 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_23), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
208 EndContainer(),
209 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
210 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_24), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
211 EndContainer(),
212 EndContainer(),
213 NWidget(NWID_SPACER), SetMinimalSize(0, 1),
214 NWidget(NWID_HORIZONTAL),
215 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_31), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
216 EndContainer(),
217 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
218 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_32), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
219 EndContainer(),
220 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
221 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_33), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
222 EndContainer(),
223 NWidget(NWID_SPACER), SetMinimalSize(1, 0),
224 NWidget(WWT_PANEL, COLOUR_GREY, WID_BT_TYPE_34), SetMinimalSize(34, 46), SetDataTip(0x0, STR_PLANT_TREE_TOOLTIP),
225 EndContainer(),
226 EndContainer(),
227 NWidget(NWID_SPACER), SetMinimalSize(0, 1),
228 NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_BT_TYPE_RANDOM), SetMinimalSize(139, 12), SetDataTip(STR_TREES_RANDOM_TYPE, STR_TREES_RANDOM_TYPE_TOOLTIP),
229 NWidget(NWID_SPACER), SetMinimalSize(0, 1),
230 NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_BT_MANY_RANDOM), SetMinimalSize(139, 12), SetDataTip(STR_TREES_RANDOM_TREES_BUTTON, STR_TREES_RANDOM_TREES_TOOLTIP),
231 NWidget(NWID_SPACER), SetMinimalSize(0, 2),
232 EndContainer(),
233 NWidget(NWID_SPACER), SetMinimalSize(2, 0),
234 EndContainer(),
235 EndContainer(),
238 static WindowDesc _build_trees_desc(
239 WDP_AUTO, "build_tree", 0, 0,
240 WC_BUILD_TREES, WC_NONE,
241 WDF_CONSTRUCTION,
242 _nested_build_trees_widgets, lengthof(_nested_build_trees_widgets)
245 void ShowBuildTreesToolbar()
247 if (_game_mode != GM_EDITOR && !Company::IsValidID(_local_company)) return;
248 AllocateWindowDescFront<BuildTreesWindow>(&_build_trees_desc, 0);