1

/* $Id$ */

     2

     3

#include "../stdafx.h"

     4

#include "../openttd.h"

     5

#include "../strings_type.h"

     6

#include "../window_gui.h"

     7

#include "../strings_func.h"

     8

#include "../strings_type.h"

     9

#include "../gfx_func.h"

    10

#include "../window_func.h"

    11

#include "../core/math_func.hpp"

    12

#include "dropdown_type.h"

    13

#include "dropdown_func.h"

    14

    15

#include "../table/sprites.h"

    16

#include "table/strings.h"

    17

    18

StringID DropDownListItem::String() const

    19

{

    20

	return STR_NULL;

    21

}

    22

    23

StringID DropDownListStringItem::String() const

    24

{

    25

	return this->string;

    26

}

    27

    28

StringID DropDownListParamStringItem::String() const

    29

{

    30

	for (uint i = 0; i < lengthof(this->decode_params); i++) SetDParam(i, this->decode_params[i]);

    31

	return this->string;

    32

}

    33

    34

/**

    35

 * Delete all items of a drop down list and the list itself

    36

 * @param list List to delete.

    37

 */

    38

static void DeleteDropDownList(DropDownList *list)

    39

{

    40

	for (DropDownList::iterator it = list->begin(); it != list->end(); ++it) {

    41

		DropDownListItem *item = *it;

    42

		delete item;

    43

	}

    44

	delete list;

    45

}

    46

    47

struct dropdown_d {

    48

	WindowClass parent_wnd_class;

    49

	WindowNumber parent_wnd_num;

    50

	byte parent_button;

    51

	DropDownList *list;

    52

	int selected_index;

    53

	byte click_delay;

    54

	bool drag_mode;

    55

	int scrolling;

    56

};

    57

assert_compile(WINDOW_CUSTOM_SIZE >= sizeof(dropdown_d));

    58

    59

static const Widget _dropdown_menu_widgets[] = {

    60

{      WWT_PANEL,   RESIZE_NONE,     0,     0, 0,     0, 0, 0x0, STR_NULL},

    61

{  WWT_SCROLLBAR,   RESIZE_NONE,     0,     0, 0,     0, 0, 0x0, STR_0190_SCROLL_BAR_SCROLLS_LIST},

    62

{   WIDGETS_END},

    63

};

    64

    65

static int GetDropDownItem(const Window *w)

    66

{

    67

	if (GetWidgetFromPos(w, _cursor.pos.x - w->left, _cursor.pos.y - w->top) < 0) return -1;

    68

    69

	int y = _cursor.pos.y - w->top - 2 + w->vscroll.pos * 10;

    70

	if (y < 0) return -1;

    71

    72

	uint selected_row = y / 10;

    73

	const DropDownList *list = WP(w, dropdown_d).list;

    74

    75

	if (selected_row >= list->size()) return -1;

    76

    77

	for (DropDownList::const_iterator it = list->begin(); it != list->end(); ++it, selected_row--) {

    78

		if (selected_row == 0) {

    79

			const DropDownListItem *item = *it;

    80

			if (item->masked || item->String() == STR_NULL) return -1;

    81

			return item->result;

    82

		}

    83

	}

    84

    85

	return -1;

    86

}

    87

    88

static void DropDownMenuWndProc(Window *w, WindowEvent *e)

    89

{

    90

	switch (e->event) {

    91

		case WE_PAINT: {

    92

			DrawWindowWidgets(w);

    93

    94

			int x = 1;

    95

			int y = 2 - w->vscroll.pos * 10;

    96

    97

			int sel    = WP(w, dropdown_d).selected_index;

    98

			int width  = w->widget[0].right - 3;

    99

			int height = w->widget[0].bottom - 3;

   100

   101

			DropDownList *list = WP(w, dropdown_d).list;

   102

   103

			for (DropDownList::const_iterator it = list->begin(); it != list->end(); ++it) {

   104

				if (y >= 0 && y <= height) {

   105

					const DropDownListItem *item = *it;

   106

					if (item->String() != STR_NULL) {

   107

						if (sel == item->result) GfxFillRect(x + 1, y, x + width, y + 9, 0);

   108

   109

						DrawStringTruncated(x + 2, y, item->String(), sel == item->result ? TC_WHITE : TC_BLACK, x + width);

   110

   111

						if (item->masked) {

   112

							GfxFillRect(x, y, x + width, y + 9,

   113

								(1 << PALETTE_MODIFIER_GREYOUT) | _colour_gradient[w->widget[0].color][5]

   114

							);

   115

						}

   116

					} else {

   117

						int c1 = _colour_gradient[w->widget[0].color][3];

   118

						int c2 = _colour_gradient[w->widget[0].color][7];

   119

   120

						GfxFillRect(x + 1, y + 3, x + w->width - 5, y + 3, c1);

   121

						GfxFillRect(x + 1, y + 4, x + w->width - 5, y + 4, c2);

   122

					}

   123

				}

   124

				y += 10;

   125

			}

   126

		} break;

   127

   128

		case WE_CLICK: {

   129

			if (e->we.click.widget != 0) break;

   130

			int item = GetDropDownItem(w);

   131

			if (item >= 0) {

   132

				WP(w, dropdown_d).click_delay = 4;

   133

				WP(w, dropdown_d).selected_index = item;

   134

				SetWindowDirty(w);

   135

			}

   136

		} break;

   137

   138

		case WE_TICK:

   139

			if (WP(w, dropdown_d).scrolling == -1) {

   140

				w->vscroll.pos = max(0, w->vscroll.pos - 1);

   141

				SetWindowDirty(w);

   142

			} else if (WP(w, dropdown_d).scrolling == 1) {

   143

				w->vscroll.pos = min(w->vscroll.count - w->vscroll.cap, w->vscroll.pos + 1);

   144

				SetWindowDirty(w);

   145

			}

   146

			WP(w, dropdown_d).scrolling = 0;

   147

			break;

   148

   149

		case WE_MOUSELOOP: {

   150

			Window *w2 = FindWindowById(WP(w, dropdown_d).parent_wnd_class, WP(w,dropdown_d).parent_wnd_num);

   151

			if (w2 == NULL) {

   152

				DeleteWindow(w);

   153

				return;

   154

			}

   155

   156

			if (WP(w, dropdown_d).click_delay != 0 && --WP(w,dropdown_d).click_delay == 0) {

   157

				WindowEvent e;

   158

				e.event = WE_DROPDOWN_SELECT;

   159

				e.we.dropdown.button = WP(w, dropdown_d).parent_button;

   160

				e.we.dropdown.index  = WP(w, dropdown_d).selected_index;

   161

				w2->wndproc(w2, &e);

   162

				DeleteWindow(w);

   163

				return;

   164

			}

   165

   166

			if (WP(w, dropdown_d).drag_mode) {

   167

				int item = GetDropDownItem(w);

   168

   169

				if (!_left_button_clicked) {

   170

					WP(w, dropdown_d).drag_mode = false;

   171

					if (item < 0) return;

   172

					WP(w, dropdown_d).click_delay = 2;

   173

				} else {

   174

					if (_cursor.pos.y <= w->top + 2) {

   175

						/* Cursor is above the list, set scroll up */

   176

						WP(w, dropdown_d).scrolling = -1;

   177

						return;

   178

					} else if (_cursor.pos.y >= w->top + w->height - 2) {

   179

						/* Cursor is below list, set scroll down */

   180

						WP(w, dropdown_d).scrolling = 1;

   181

						return;

   182

					}

   183

   184

					if (item < 0) return;

   185

				}

   186

   187

				WP(w, dropdown_d).selected_index = item;

   188

				SetWindowDirty(w);

   189

			}

   190

		} break;

   191

   192

		case WE_DESTROY: {

   193

			Window *w2 = FindWindowById(WP(w, dropdown_d).parent_wnd_class, WP(w,dropdown_d).parent_wnd_num);

   194

			if (w2 != NULL) {

   195

				w2->RaiseWidget(WP(w, dropdown_d).parent_button);

   196

				w2->InvalidateWidget(WP(w, dropdown_d).parent_button);

   197

			}

   198

   199

			DeleteDropDownList(WP(w, dropdown_d).list);

   200

		} break;

   201

	}

   202

}

   203

   204

void ShowDropDownList(Window *w, DropDownList *list, int selected, int button)

   205

{

   206

	bool is_dropdown_menu_shown = w->IsWidgetLowered(button);

   207

   208

	DeleteWindowById(WC_DROPDOWN_MENU, 0);

   209

   210

	if (is_dropdown_menu_shown) {

   211

		DeleteDropDownList(list);

   212

		return;

   213

	}

   214

   215

	w->LowerWidget(button);

   216

	w->InvalidateWidget(button);

   217

   218

	/* Our parent's button widget is used to determine where to place the drop

   219

	 * down list window. */

   220

	const Widget *wi = &w->widget[button];

   221

   222

	/* The preferred position is just below the dropdown calling widget */

   223

	int top = w->top + wi->bottom + 1;

   224

	int height = list->size() * 10 + 4;

   225

   226

	/* Check if the status bar is visible, as we don't want to draw over it */

   227

	Window *w3 = FindWindowById(WC_STATUS_BAR, 0);

   228

	int screen_bottom = w3 == NULL ? _screen.height : w3->top;

   229

   230

	bool scroll = false;

   231

   232

	/* Check if the dropdown will fully fit below the widget */

   233

	if (top + height >= screen_bottom) {

   234

		w3 = FindWindowById(WC_MAIN_TOOLBAR, 0);

   235

		int screen_top = w3 == NULL ? 0 : w3->top + w3->height;

   236

   237

		/* If not, check if it will fit above the widget */

   238

		if (w->top + wi->top - height > screen_top) {

   239

			top = w->top + wi->top - height;

   240

		} else {

   241

			/* ... and lastly if it won't, enable the scroll bar and fit the

   242

			 * list in below the widget */

   243

			int rows = (screen_bottom - 4 - top) / 10;

   244

			height = rows * 10 + 4;

   245

			scroll = true;

   246

		}

   247

	}

   248

   249

	Window *dw = AllocateWindow(

   250

		w->left + wi->left,

   251

		top,

   252

		wi->right - wi->left + 1,

   253

		height,

   254

		DropDownMenuWndProc,

   255

		WC_DROPDOWN_MENU,

   256

		_dropdown_menu_widgets);

   257

   258

	dw->widget[0].color = wi->color;

   259

	dw->widget[0].right = wi->right - wi->left;

   260

	dw->widget[0].bottom = height - 1;

   261

   262

	dw->SetWidgetHiddenState(1, !scroll);

   263

   264

	if (scroll) {

   265

		/* We're scrolling, so enable the scroll bar and shrink the list by

   266

		 * the scrollbar's width */

   267

		dw->widget[1].color  = wi->color;

   268

		dw->widget[1].right  = dw->widget[0].right;

   269

		dw->widget[1].left   = dw->widget[1].right - 11;

   270

		dw->widget[1].bottom = height - 1;

   271

		dw->widget[0].right -= 12;

   272

   273

		dw->vscroll.cap   = (height - 4) / 10;

   274

		dw->vscroll.count = list->size();

   275

	}

   276

   277

	dw->desc_flags = WDF_DEF_WIDGET;

   278

	dw->flags4 &= ~WF_WHITE_BORDER_MASK;

   279

   280

	WP(dw, dropdown_d).parent_wnd_class = w->window_class;

   281

	WP(dw, dropdown_d).parent_wnd_num   = w->window_number;

   282

	WP(dw, dropdown_d).parent_button    = button;

   283

	WP(dw, dropdown_d).list             = list;

   284

	WP(dw, dropdown_d).selected_index   = selected;

   285

	WP(dw, dropdown_d).click_delay      = 0;

   286

	WP(dw, dropdown_d).drag_mode        = true;

   287

}

   288

   289

void ShowDropDownMenu(Window *w, const StringID *strings, int selected, int button, uint32 disabled_mask, uint32 hidden_mask)

   290

{

   291

	/* Don't create a new list if we're just closing an existing menu */

   292

	if (w->IsWidgetLowered(button)) {

   293

		DeleteWindowById(WC_DROPDOWN_MENU, 0);

   294

		return;

   295

	}

   296

   297

	uint result = 0;

   298

	DropDownList *list = new DropDownList();

   299

   300

	for (uint i = 0; strings[i] != INVALID_STRING_ID; i++) {

   301

		if (!HasBit(hidden_mask, i)) {

   302

			list->push_back(new DropDownListStringItem(strings[i], result, HasBit(disabled_mask, i)));

   303

		}

   304

		result++;

   305

	}

   306

   307

	/* No entries in the list? */

   308

	if (list->size() == 0) {

   309

		DeleteDropDownList(list);

   310

		return;

   311

	}

   312

   313

	ShowDropDownList(w, list, selected, button);

   314

}

   315

   316

void HideDropDownMenu(Window *pw)

   317

{

   318

	Window **wz;

   319

	FOR_ALL_WINDOWS(wz) {

   320

		if ((*wz)->window_class != WC_DROPDOWN_MENU) continue;

   321

   322

		if (pw->window_class == WP(*wz, dropdown_d).parent_wnd_class &&

   323

				pw->window_number == WP(*wz, dropdown_d).parent_wnd_num) {

   324

			DeleteWindow(*wz);

   325

			break;

   326

		}

   327

	}

   328

}

   329