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

uint DropDownListItem::Height(uint width) const

    24

{

    25

	return 10;

    26

}

    27

    28

StringID DropDownListStringItem::String() const

    29

{

    30

	return this->string;

    31

}

    32

    33

StringID DropDownListParamStringItem::String() const

    34

{

    35

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

    36

	return this->string;

    37

}

    38

    39

void DropDownListItem::Draw(int x, int y, uint width, uint height, bool sel) const

    40

{

    41

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

    42

}

    43

    44

/**

    45

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

    46

 * @param list List to delete.

    47

 */

    48

static void DeleteDropDownList(DropDownList *list)

    49

{

    50

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

    51

		DropDownListItem *item = *it;

    52

		delete item;

    53

	}

    54

	delete list;

    55

}

    56

    57

struct dropdown_d {

    58

	WindowClass parent_wnd_class;

    59

	WindowNumber parent_wnd_num;

    60

	byte parent_button;

    61

	DropDownList *list;

    62

	int selected_index;

    63

	byte click_delay;

    64

	bool drag_mode;

    65

	int scrolling;

    66

};

    67

assert_compile(WINDOW_CUSTOM_SIZE >= sizeof(dropdown_d));

    68

    69

static const Widget _dropdown_menu_widgets[] = {

    70

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

    71

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

    72

{   WIDGETS_END},

    73

};

    74

    75

static int GetDropDownItem(const Window *w)

    76

{

    77

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

    78

    79

	int y     = _cursor.pos.y - w->top - 2;

    80

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

    81

	int pos   = w->vscroll.pos;

    82

    83

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

    84

    85

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

    86

		/* Skip items that are scrolled up */

    87

		if (--pos >= 0) continue;

    88

    89

		const DropDownListItem *item = *it;

    90

		int item_height = item->Height(width);

    91

    92

		if (y < item_height) {

    93

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

    94

			return item->result;

    95

		}

    96

    97

		y -= item_height;

    98

	}

    99

   100

	return -1;

   101

}

   102

   103

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

   104

{

   105

	switch (e->event) {

   106

		case WE_PAINT: {

   107

			DrawWindowWidgets(w);

   108

   109

			int x = 1;

   110

			int y = 2;

   111

   112

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

   113

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

   114

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

   115

			int pos    = w->vscroll.pos;

   116

   117

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

   118

   119

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

   120

				const DropDownListItem *item = *it;

   121

				int item_height = item->Height(width);

   122

   123

				/* Skip items that are scrolled up */

   124

				if (--pos >= 0) continue;

   125

   126

				if (y + item_height < height) {

   127

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

   128

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

   129

   130

						item->Draw(x, y, width, 10, sel == item->result);

   131

   132

						if (item->masked) {

   133

							GfxFillRect(x, y, x + width, y + item_height - 1,

   134

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

   135

							);

   136

						}

   137

					} else {

   138

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

   139

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

   140

   141

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

   142

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

   143

					}

   144

				}

   145

				y += item_height;

   146

			}

   147

		} break;

   148

   149

		case WE_CLICK: {

   150

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

   151

			int item = GetDropDownItem(w);

   152

			if (item >= 0) {

   153

				WP(w, dropdown_d).click_delay = 4;

   154

				WP(w, dropdown_d).selected_index = item;

   155

				SetWindowDirty(w);

   156

			}

   157

		} break;

   158

   159

		case WE_TICK:

   160

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

   161

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

   162

				SetWindowDirty(w);

   163

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

   164

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

   165

				SetWindowDirty(w);

   166

			}

   167

			WP(w, dropdown_d).scrolling = 0;

   168

			break;

   169

   170

		case WE_MOUSELOOP: {

   171

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

   172

			if (w2 == NULL) {

   173

				DeleteWindow(w);

   174

				return;

   175

			}

   176

   177

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

   178

				WindowEvent e;

   179

				e.event = WE_DROPDOWN_SELECT;

   180

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

   181

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

   182

				w2->wndproc(w2, &e);

   183

				DeleteWindow(w);

   184

				return;

   185

			}

   186

   187

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

   188

				int item = GetDropDownItem(w);

   189

   190

				if (!_left_button_clicked) {

   191

					WP(w, dropdown_d).drag_mode = false;

   192

					if (item < 0) return;

   193

					WP(w, dropdown_d).click_delay = 2;

   194

				} else {

   195

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

   196

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

   197

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

   198

						return;

   199

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

   200

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

   201

						WP(w, dropdown_d).scrolling = 1;

   202

						return;

   203

					}

   204

   205

					if (item < 0) return;

   206

				}

   207

   208

				WP(w, dropdown_d).selected_index = item;

   209

				SetWindowDirty(w);

   210

			}

   211

		} break;

   212

   213

		case WE_DESTROY: {

   214

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

   215

			if (w2 != NULL) {

   216

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

   217

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

   218

			}

   219

   220

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

   221

		} break;

   222

	}

   223

}

   224

   225

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

   226

{

   227

	bool is_dropdown_menu_shown = w->IsWidgetLowered(button);

   228

   229

	DeleteWindowById(WC_DROPDOWN_MENU, 0);

   230

   231

	if (is_dropdown_menu_shown) {

   232

		DeleteDropDownList(list);

   233

		return;

   234

	}

   235

   236

	w->LowerWidget(button);

   237

	w->InvalidateWidget(button);

   238

   239

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

   240

	 * down list window. */

   241

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

   242

   243

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

   244

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

   245

   246

	/* Total length of list */

   247

	int list_height = 0;

   248

   249

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

   250

		DropDownListItem *item = *it;

   251

		list_height += item->Height(width);

   252

	}

   253

   254

	/* Height of window visible */

   255

	int height = list_height;

   256

   257

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

   258

	Window *w3 = FindWindowById(WC_STATUS_BAR, 0);

   259

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

   260

   261

	bool scroll = false;

   262

   263

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

   264

	if (top + height + 4 >= screen_bottom) {

   265

		w3 = FindWindowById(WC_MAIN_TOOLBAR, 0);

   266

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

   267

   268

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

   269

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

   270

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

   271

		} else {

   272

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

   273

			 * list in below the widget */

   274

			int avg_height = list_height / list->size();

   275

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

   276

			height = rows * avg_height;

   277

			scroll = true;

   278

		}

   279

	}

   280

   281

	if (width == 0) width = wi->right - wi->left + 1;

   282

   283

	Window *dw = AllocateWindow(

   284

		w->left + wi->left,

   285

		top,

   286

		width,

   287

		height + 4,

   288

		DropDownMenuWndProc,

   289

		WC_DROPDOWN_MENU,

   290

		_dropdown_menu_widgets);

   291

   292

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

   293

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

   294

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

   295

   296

	dw->SetWidgetHiddenState(1, !scroll);

   297

   298

	if (scroll) {

   299

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

   300

		 * the scrollbar's width */

   301

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

   302

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

   303

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

   304

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

   305

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

   306

   307

		/* Capacity is the average number of items visible */

   308

		dw->vscroll.cap   = height * list->size() / list_height;

   309

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

   310

	}

   311

   312

	dw->desc_flags = WDF_DEF_WIDGET;

   313

	dw->flags4 &= ~WF_WHITE_BORDER_MASK;

   314

   315

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

   316

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

   317

	WP(dw, dropdown_d).parent_button    = button;

   318

	WP(dw, dropdown_d).list             = list;

   319

	WP(dw, dropdown_d).selected_index   = selected;

   320

	WP(dw, dropdown_d).click_delay      = 0;

   321

	WP(dw, dropdown_d).drag_mode        = true;

   322

}

   323

   324

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

   325

{

   326

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

   327

	if (w->IsWidgetLowered(button)) {

   328

		DeleteWindowById(WC_DROPDOWN_MENU, 0);

   329

		return;

   330

	}

   331

   332

	uint result = 0;

   333

	DropDownList *list = new DropDownList();

   334

   335

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

   336

		if (!HasBit(hidden_mask, i)) {

   337

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

   338

		}

   339

		result++;

   340

	}

   341

   342

	/* No entries in the list? */

   343

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

   344

		DeleteDropDownList(list);

   345

		return;

   346

	}

   347

   348

	ShowDropDownList(w, list, selected, button, width);

   349

}

   350

   351

/**

   352

 * Delete the drop-down menu from window \a pw

   353

 * @param pw Parent window of the drop-down menu window

   354

 */

   355

void HideDropDownMenu(Window *pw)

   356

{

   357

	Window **wz;

   358

	FOR_ALL_WINDOWS(wz) {

   359

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

   360

   361

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

   362

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

   363

			DeleteWindow(*wz);

   364

			break;

   365

		}

   366

	}

   367

}

   368