peter1138@8780: /* $Id$ */
peter1138@8780: 
peter1138@8780: #include "../stdafx.h"
peter1138@8780: #include "../openttd.h"
peter1138@8780: #include "../strings_type.h"
peter1138@8780: #include "../window_gui.h"
peter1138@8780: #include "../strings_func.h"
peter1138@8780: #include "../strings_type.h"
peter1138@8780: #include "../gfx_func.h"
peter1138@8780: #include "../window_func.h"
peter1138@8797: #include "../core/math_func.hpp"
peter1138@8780: #include "dropdown_type.h"
peter1138@8780: #include "dropdown_func.h"
peter1138@8780: 
peter1138@8780: #include "../table/sprites.h"
peter1138@8780: #include "table/strings.h"
peter1138@8780: 
peter1138@8780: StringID DropDownListItem::String() const
peter1138@8780: {
peter1138@8780: 	return STR_NULL;
peter1138@8780: }
peter1138@8780: 
peter1138@8780: StringID DropDownListStringItem::String() const
peter1138@8780: {
peter1138@8780: 	return this->string;
peter1138@8780: }
peter1138@8780: 
peter1138@8780: StringID DropDownListParamStringItem::String() const
peter1138@8780: {
peter1138@8780: 	for (uint i = 0; i < lengthof(this->decode_params); i++) SetDParam(i, this->decode_params[i]);
peter1138@8780: 	return this->string;
peter1138@8780: }
peter1138@8780: 
peter1138@8795: /**
peter1138@8795:  * Delete all items of a drop down list and the list itself
peter1138@8795:  * @param list List to delete.
peter1138@8795:  */
peter1138@8795: static void DeleteDropDownList(DropDownList *list)
peter1138@8795: {
peter1138@8795: 	for (DropDownList::iterator it = list->begin(); it != list->end(); ++it) {
peter1138@8795: 		DropDownListItem *item = *it;
peter1138@8795: 		delete item;
peter1138@8795: 	}
peter1138@8795: 	delete list;
peter1138@8795: }
peter1138@8795: 
peter1138@8780: struct dropdown_d {
peter1138@8780: 	WindowClass parent_wnd_class;
peter1138@8780: 	WindowNumber parent_wnd_num;
peter1138@8780: 	byte parent_button;
peter1138@8780: 	DropDownList *list;
peter1138@9039: 	int selected_index;
peter1138@8780: 	byte click_delay;
peter1138@8780: 	bool drag_mode;
peter1138@8797: 	int scrolling;
peter1138@8780: };
peter1138@8780: assert_compile(WINDOW_CUSTOM_SIZE >= sizeof(dropdown_d));
peter1138@8780: 
peter1138@8780: static const Widget _dropdown_menu_widgets[] = {
peter1138@8780: {      WWT_PANEL,   RESIZE_NONE,     0,     0, 0,     0, 0, 0x0, STR_NULL},
peter1138@8780: {  WWT_SCROLLBAR,   RESIZE_NONE,     0,     0, 0,     0, 0, 0x0, STR_0190_SCROLL_BAR_SCROLLS_LIST},
peter1138@8780: {   WIDGETS_END},
peter1138@8780: };
peter1138@8780: 
peter1138@8780: static int GetDropDownItem(const Window *w)
peter1138@8780: {
peter1138@8780: 	if (GetWidgetFromPos(w, _cursor.pos.x - w->left, _cursor.pos.y - w->top) < 0) return -1;
peter1138@8780: 
peter1138@8780: 	int y = _cursor.pos.y - w->top - 2 + w->vscroll.pos * 10;
peter1138@8780: 	if (y < 0) return -1;
peter1138@8780: 
peter1138@8780: 	uint selected_row = y / 10;
peter1138@8780: 	const DropDownList *list = WP(w, dropdown_d).list;
peter1138@8780: 
peter1138@8780: 	if (selected_row >= list->size()) return -1;
peter1138@8780: 
peter1138@8780: 	for (DropDownList::const_iterator it = list->begin(); it != list->end(); ++it, selected_row--) {
peter1138@8780: 		if (selected_row == 0) {
peter1138@8780: 			const DropDownListItem *item = *it;
peter1138@8780: 			if (item->masked || item->String() == STR_NULL) return -1;
peter1138@8780: 			return item->result;
peter1138@8780: 		}
peter1138@8780: 	}
peter1138@8780: 
peter1138@8780: 	return -1;
peter1138@8780: }
peter1138@8780: 
peter1138@8780: static void DropDownMenuWndProc(Window *w, WindowEvent *e)
peter1138@8780: {
peter1138@8780: 	switch (e->event) {
peter1138@8780: 		case WE_PAINT: {
peter1138@8780: 			DrawWindowWidgets(w);
peter1138@8780: 
peter1138@8780: 			int x = 1;
peter1138@8780: 			int y = 2 - w->vscroll.pos * 10;
peter1138@8780: 
peter1138@8780: 			int sel    = WP(w, dropdown_d).selected_index;
peter1138@8780: 			int width  = w->widget[0].right - 3;
peter1138@8780: 			int height = w->widget[0].bottom - 3;
peter1138@8780: 
peter1138@8780: 			DropDownList *list = WP(w, dropdown_d).list;
peter1138@8780: 
peter1138@8780: 			for (DropDownList::const_iterator it = list->begin(); it != list->end(); ++it) {
peter1138@8780: 				if (y >= 0 && y <= height) {
peter1138@8780: 					const DropDownListItem *item = *it;
peter1138@8780: 					if (item->String() != STR_NULL) {
peter1138@8780: 						if (sel == item->result) GfxFillRect(x + 1, y, x + width, y + 9, 0);
peter1138@8780: 
peter1138@8780: 						DrawStringTruncated(x + 2, y, item->String(), sel == item->result ? TC_WHITE : TC_BLACK, x + width);
peter1138@8780: 
peter1138@8780: 						if (item->masked) {
peter1138@8780: 							GfxFillRect(x, y, x + width, y + 9,
peter1138@8790: 								(1 << PALETTE_MODIFIER_GREYOUT) | _colour_gradient[w->widget[0].color][5]
peter1138@8780: 							);
peter1138@8780: 						}
peter1138@8780: 					} else {
peter1138@8790: 						int c1 = _colour_gradient[w->widget[0].color][3];
peter1138@8790: 						int c2 = _colour_gradient[w->widget[0].color][7];
peter1138@8780: 
peter1138@8780: 						GfxFillRect(x + 1, y + 3, x + w->width - 5, y + 3, c1);
peter1138@8780: 						GfxFillRect(x + 1, y + 4, x + w->width - 5, y + 4, c2);
peter1138@8780: 					}
peter1138@8780: 				}
peter1138@8780: 				y += 10;
peter1138@8780: 			}
peter1138@8780: 		} break;
peter1138@8780: 
peter1138@8780: 		case WE_CLICK: {
peter1138@8780: 			if (e->we.click.widget != 0) break;
peter1138@8780: 			int item = GetDropDownItem(w);
peter1138@8780: 			if (item >= 0) {
peter1138@8780: 				WP(w, dropdown_d).click_delay = 4;
peter1138@8780: 				WP(w, dropdown_d).selected_index = item;
peter1138@8780: 				SetWindowDirty(w);
peter1138@8780: 			}
peter1138@8780: 		} break;
peter1138@8780: 
peter1138@8797: 		case WE_TICK:
peter1138@8797: 			if (WP(w, dropdown_d).scrolling == -1) {
peter1138@8797: 				w->vscroll.pos = max(0, w->vscroll.pos - 1);
peter1138@8797: 				SetWindowDirty(w);
peter1138@8797: 			} else if (WP(w, dropdown_d).scrolling == 1) {
peter1138@8797: 				w->vscroll.pos = min(w->vscroll.count - w->vscroll.cap, w->vscroll.pos + 1);
peter1138@8797: 				SetWindowDirty(w);
peter1138@8797: 			}
peter1138@8797: 			WP(w, dropdown_d).scrolling = 0;
peter1138@8797: 			break;
peter1138@8797: 
peter1138@8780: 		case WE_MOUSELOOP: {
peter1138@8780: 			Window *w2 = FindWindowById(WP(w, dropdown_d).parent_wnd_class, WP(w,dropdown_d).parent_wnd_num);
peter1138@8780: 			if (w2 == NULL) {
peter1138@8780: 				DeleteWindow(w);
peter1138@8780: 				return;
peter1138@8780: 			}
peter1138@8780: 
peter1138@8780: 			if (WP(w, dropdown_d).click_delay != 0 && --WP(w,dropdown_d).click_delay == 0) {
peter1138@8780: 				WindowEvent e;
peter1138@8780: 				e.event = WE_DROPDOWN_SELECT;
peter1138@8780: 				e.we.dropdown.button = WP(w, dropdown_d).parent_button;
peter1138@8780: 				e.we.dropdown.index  = WP(w, dropdown_d).selected_index;
peter1138@8780: 				w2->wndproc(w2, &e);
peter1138@8780: 				DeleteWindow(w);
peter1138@8780: 				return;
peter1138@8780: 			}
peter1138@8780: 
peter1138@8780: 			if (WP(w, dropdown_d).drag_mode) {
peter1138@8780: 				int item = GetDropDownItem(w);
peter1138@8780: 
peter1138@8780: 				if (!_left_button_clicked) {
peter1138@8780: 					WP(w, dropdown_d).drag_mode = false;
peter1138@8780: 					if (item < 0) return;
peter1138@8780: 					WP(w, dropdown_d).click_delay = 2;
peter1138@8780: 				} else {
peter1138@8797: 					if (_cursor.pos.y <= w->top + 2) {
peter1138@8797: 						/* Cursor is above the list, set scroll up */
peter1138@8797: 						WP(w, dropdown_d).scrolling = -1;
peter1138@8797: 						return;
peter1138@8797: 					} else if (_cursor.pos.y >= w->top + w->height - 2) {
peter1138@8797: 						/* Cursor is below list, set scroll down */
peter1138@8797: 						WP(w, dropdown_d).scrolling = 1;
peter1138@8797: 						return;
peter1138@8797: 					}
peter1138@8797: 
peter1138@8780: 					if (item < 0) return;
peter1138@8780: 				}
peter1138@8780: 
peter1138@8780: 				WP(w, dropdown_d).selected_index = item;
peter1138@8780: 				SetWindowDirty(w);
peter1138@8780: 			}
peter1138@8780: 		} break;
peter1138@8780: 
peter1138@8780: 		case WE_DESTROY: {
peter1138@8780: 			Window *w2 = FindWindowById(WP(w, dropdown_d).parent_wnd_class, WP(w,dropdown_d).parent_wnd_num);
peter1138@8780: 			if (w2 != NULL) {
peter1138@8780: 				w2->RaiseWidget(WP(w, dropdown_d).parent_button);
peter1138@8780: 				w2->InvalidateWidget(WP(w, dropdown_d).parent_button);
peter1138@8780: 			}
peter1138@8780: 
peter1138@8795: 			DeleteDropDownList(WP(w, dropdown_d).list);
peter1138@8780: 		} break;
peter1138@8780: 	}
peter1138@8780: }
peter1138@8780: 
peter1138@8780: void ShowDropDownList(Window *w, DropDownList *list, int selected, int button)
peter1138@8780: {
peter1138@8780: 	bool is_dropdown_menu_shown = w->IsWidgetLowered(button);
peter1138@8780: 
peter1138@8780: 	DeleteWindowById(WC_DROPDOWN_MENU, 0);
peter1138@8780: 
peter1138@8780: 	if (is_dropdown_menu_shown) {
peter1138@8795: 		DeleteDropDownList(list);
peter1138@8780: 		return;
peter1138@8780: 	}
peter1138@8780: 
peter1138@8780: 	w->LowerWidget(button);
peter1138@8780: 	w->InvalidateWidget(button);
peter1138@8780: 
peter1138@8780: 	/* Our parent's button widget is used to determine where to place the drop
peter1138@8780: 	 * down list window. */
peter1138@8780: 	const Widget *wi = &w->widget[button];
peter1138@8780: 
peter1138@8780: 	/* The preferred position is just below the dropdown calling widget */
peter1138@8831: 	int top = w->top + wi->bottom + 1;
peter1138@8780: 	int height = list->size() * 10 + 4;
peter1138@8780: 
peter1138@8780: 	/* Check if the status bar is visible, as we don't want to draw over it */
peter1138@8780: 	Window *w3 = FindWindowById(WC_STATUS_BAR, 0);
peter1138@8780: 	int screen_bottom = w3 == NULL ? _screen.height : w3->top;
peter1138@8780: 
peter1138@8780: 	bool scroll = false;
peter1138@8780: 
peter1138@8780: 	/* Check if the dropdown will fully fit below the widget */
peter1138@8780: 	if (top + height >= screen_bottom) {
peter1138@8780: 		w3 = FindWindowById(WC_MAIN_TOOLBAR, 0);
peter1138@8780: 		int screen_top = w3 == NULL ? 0 : w3->top + w3->height;
peter1138@8780: 
peter1138@8780: 		/* If not, check if it will fit above the widget */
peter1138@8831: 		if (w->top + wi->top - height > screen_top) {
peter1138@8831: 			top = w->top + wi->top - height;
peter1138@8780: 		} else {
peter1138@8780: 			/* ... and lastly if it won't, enable the scroll bar and fit the
peter1138@8780: 			 * list in below the widget */
peter1138@8780: 			int rows = (screen_bottom - 4 - top) / 10;
peter1138@8780: 			height = rows * 10 + 4;
peter1138@8780: 			scroll = true;
peter1138@8780: 		}
peter1138@8780: 	}
peter1138@8780: 
peter1138@8780: 	Window *dw = AllocateWindow(
peter1138@8854: 		w->left + wi->left,
peter1138@8780: 		top,
peter1138@8854: 		wi->right - wi->left + 1,
peter1138@8780: 		height,
peter1138@8780: 		DropDownMenuWndProc,
peter1138@8780: 		WC_DROPDOWN_MENU,
peter1138@8780: 		_dropdown_menu_widgets);
peter1138@8780: 
peter1138@8780: 	dw->widget[0].color = wi->color;
peter1138@8854: 	dw->widget[0].right = wi->right - wi->left;
peter1138@8780: 	dw->widget[0].bottom = height - 1;
peter1138@8780: 
peter1138@8780: 	dw->SetWidgetHiddenState(1, !scroll);
peter1138@8780: 
peter1138@8780: 	if (scroll) {
peter1138@8780: 		/* We're scrolling, so enable the scroll bar and shrink the list by
peter1138@8780: 		 * the scrollbar's width */
peter1138@8780: 		dw->widget[1].color  = wi->color;
peter1138@8780: 		dw->widget[1].right  = dw->widget[0].right;
peter1138@8780: 		dw->widget[1].left   = dw->widget[1].right - 11;
peter1138@8780: 		dw->widget[1].bottom = height - 1;
peter1138@8780: 		dw->widget[0].right -= 12;
peter1138@8780: 
peter1138@8780: 		dw->vscroll.cap   = (height - 4) / 10;
peter1138@8780: 		dw->vscroll.count = list->size();
peter1138@8780: 	}
peter1138@8780: 
peter1138@8780: 	dw->desc_flags = WDF_DEF_WIDGET;
peter1138@8780: 	dw->flags4 &= ~WF_WHITE_BORDER_MASK;
peter1138@8780: 
peter1138@8780: 	WP(dw, dropdown_d).parent_wnd_class = w->window_class;
peter1138@8780: 	WP(dw, dropdown_d).parent_wnd_num   = w->window_number;
peter1138@8780: 	WP(dw, dropdown_d).parent_button    = button;
peter1138@8780: 	WP(dw, dropdown_d).list             = list;
peter1138@8780: 	WP(dw, dropdown_d).selected_index   = selected;
peter1138@8780: 	WP(dw, dropdown_d).click_delay      = 0;
peter1138@8780: 	WP(dw, dropdown_d).drag_mode        = true;
peter1138@8780: }
peter1138@8780: 
peter1138@8780: void ShowDropDownMenu(Window *w, const StringID *strings, int selected, int button, uint32 disabled_mask, uint32 hidden_mask)
peter1138@8780: {
peter1138@8795: 	/* Don't create a new list if we're just closing an existing menu */
peter1138@8795: 	if (w->IsWidgetLowered(button)) {
peter1138@8795: 		DeleteWindowById(WC_DROPDOWN_MENU, 0);
peter1138@8795: 		return;
peter1138@8795: 	}
peter1138@8795: 
peter1138@8780: 	uint result = 0;
peter1138@8780: 	DropDownList *list = new DropDownList();
peter1138@8780: 
peter1138@8780: 	for (uint i = 0; strings[i] != INVALID_STRING_ID; i++) {
peter1138@8780: 		if (!HasBit(hidden_mask, i)) {
peter1138@8780: 			list->push_back(new DropDownListStringItem(strings[i], result, HasBit(disabled_mask, i)));
peter1138@8780: 		}
peter1138@8780: 		result++;
peter1138@8780: 	}
peter1138@8780: 
peter1138@8780: 	/* No entries in the list? */
peter1138@8780: 	if (list->size() == 0) {
peter1138@8795: 		DeleteDropDownList(list);
peter1138@8780: 		return;
peter1138@8780: 	}
peter1138@8780: 
peter1138@8780: 	ShowDropDownList(w, list, selected, button);
peter1138@8780: }
peter1138@8780: 
peter1138@8780: void HideDropDownMenu(Window *pw)
peter1138@8780: {
peter1138@8780: 	Window **wz;
peter1138@8780: 	FOR_ALL_WINDOWS(wz) {
peter1138@8780: 		if ((*wz)->window_class != WC_DROPDOWN_MENU) continue;
peter1138@8780: 
peter1138@8780: 		if (pw->window_class == WP(*wz, dropdown_d).parent_wnd_class &&
peter1138@8780: 				pw->window_number == WP(*wz, dropdown_d).parent_wnd_num) {
peter1138@8780: 			DeleteWindow(*wz);
peter1138@8780: 			break;
peter1138@8780: 		}
peter1138@8780: 	}
peter1138@8780: }
peter1138@8780: