1

/* $Id$ */

     2

     3

/** @file dropdown.cpp Implementation of the dropdown widget. */

     4

     5

#include "../stdafx.h"

     6

#include "../openttd.h"

     7

#include "../strings_type.h"

     8

#include "../window_gui.h"

     9

#include "../strings_func.h"

    10

#include "../strings_type.h"

    11

#include "../gfx_func.h"

    12

#include "../window_func.h"

    13

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

    14

#include "dropdown_type.h"

    15

#include "dropdown_func.h"

    16

    17

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

    18

#include "table/strings.h"

    19

    20

StringID DropDownListItem::String() const

    21

{

    22

	return STR_NULL;

    23

}

    24

    25

uint DropDownListItem::Height(uint width) const

    26

{

    27

	return 10;

    28

}

    29

    30

StringID DropDownListStringItem::String() const

    31

{

    32

	return this->string;

    33

}

    34

    35

StringID DropDownListParamStringItem::String() const

    36

{

    37

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

    38

	return this->string;

    39

}

    40

    41

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

    42

{

    43

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

    44

}

    45

    46

/**

    47

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

    48

 * @param list List to delete.

    49

 */

    50

static void DeleteDropDownList(DropDownList *list)

    51

{

    52

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

    53

		DropDownListItem *item = *it;

    54

		delete item;

    55

	}

    56

	delete list;

    57

}

    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

struct DropdownWindow : Window {

    66

	WindowClass parent_wnd_class;

    67

	WindowNumber parent_wnd_num;

    68

	byte parent_button;

    69

	DropDownList *list;

    70

	int selected_index;

    71

	byte click_delay;

    72

	bool drag_mode;

    73

	int scrolling;

    74

    75

	DropdownWindow(int x, int y, int width, int height, const Widget *widget) : Window(x, y, width, height, WC_DROPDOWN_MENU, widget)

    76

	{

    77

	}

    78

    79

	~DropdownWindow()

    80

	{

    81

		Window *w2 = FindWindowById(this->parent_wnd_class, this->parent_wnd_num);

    82

		if (w2 != NULL) {

    83

			w2->RaiseWidget(this->parent_button);

    84

			w2->InvalidateWidget(this->parent_button);

    85

		}

    86

    87

		DeleteDropDownList(this->list);

    88

	}

    89

    90

	bool GetDropDownItem(int &value)

    91

	{

    92

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

    93

    94

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

    95

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

    96

		int pos   = this->vscroll.pos;

    97

    98

		const DropDownList *list = this->list;

    99

   100

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

   101

			/* Skip items that are scrolled up */

   102

			if (--pos >= 0) continue;

   103

   104

			const DropDownListItem *item = *it;

   105

			int item_height = item->Height(width);

   106

   107

			if (y < item_height) {

   108

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

   109

				value = item->result;

   110

				return true;

   111

			}

   112

   113

			y -= item_height;

   114

		}

   115

   116

		return false;

   117

	}

   118

   119

	virtual void OnPaint()

   120

	{

   121

		this->DrawWidgets();

   122

   123

		int x = 1;

   124

		int y = 2;

   125

   126

		int sel    = this->selected_index;

   127

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

   128

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

   129

		int pos    = this->vscroll.pos;

   130

   131

		DropDownList *list = this->list;

   132

   133

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

   134

			const DropDownListItem *item = *it;

   135

			int item_height = item->Height(width);

   136

   137

			/* Skip items that are scrolled up */

   138

			if (--pos >= 0) continue;

   139

   140

			if (y + item_height < height) {

   141

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

   142

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

   143

   144

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

   145

   146

					if (item->masked) {

   147

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

   148

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

   149

						);

   150

					}

   151

				} else {

   152

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

   153

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

   154

   155

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

   156

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

   157

				}

   158

			}

   159

			y += item_height;

   160

		}

   161

	};

   162

   163

	virtual void OnClick(Point pt, int widget)

   164

	{

   165

		if (widget != 0) return;

   166

		int item;

   167

		if (this->GetDropDownItem(item)) {

   168

			this->click_delay = 4;

   169

			this->selected_index = item;

   170

			this->SetDirty();

   171

		}

   172

	}

   173

   174

	virtual void OnTick()

   175

	{

   176

		if (this->scrolling == -1) {

   177

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

   178

			this->SetDirty();

   179

		} else if (this->scrolling == 1) {

   180

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

   181

			this->SetDirty();

   182

		}

   183

		this->scrolling = 0;

   184

	}

   185

   186

	virtual void OnMouseLoop()

   187

	{

   188

		Window *w2 = FindWindowById(this->parent_wnd_class, this->parent_wnd_num);

   189

		if (w2 == NULL) {

   190

			delete this;

   191

			return;

   192

		}

   193

   194

		if (this->click_delay != 0 && --this->click_delay == 0) {

   195

			w2->OnDropdownSelect(this->parent_button, this->selected_index);

   196

			delete this;

   197

			return;

   198

		}

   199

   200

		if (this->drag_mode) {

   201

			int item;

   202

   203

			if (!_left_button_clicked) {

   204

				this->drag_mode = false;

   205

				if (!this->GetDropDownItem(item)) return;

   206

				this->click_delay = 2;

   207

			} else {

   208

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

   209

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

   210

					this->scrolling = -1;

   211

					return;

   212

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

   213

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

   214

					this->scrolling = 1;

   215

					return;

   216

				}

   217

   218

				if (!this->GetDropDownItem(item)) return;

   219

			}

   220

   221

			this->selected_index = item;

   222

			this->SetDirty();

   223

		}

   224

	}

   225

};

   226

   227

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

   228

{

   229

	bool is_dropdown_menu_shown = w->IsWidgetLowered(button);

   230

   231

	DeleteWindowById(WC_DROPDOWN_MENU, 0);

   232

   233

	if (is_dropdown_menu_shown) {

   234

		DeleteDropDownList(list);

   235

		return;

   236

	}

   237

   238

	w->LowerWidget(button);

   239

	w->InvalidateWidget(button);

   240

   241

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

   242

	 * down list window. */

   243

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

   244

   245

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

   246

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

   247

   248

	/* Total length of list */

   249

	int list_height = 0;

   250

   251

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

   252

		DropDownListItem *item = *it;

   253

		list_height += item->Height(width);

   254

	}

   255

   256

	/* Height of window visible */

   257

	int height = list_height;

   258

   259

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

   260

	Window *w3 = FindWindowById(WC_STATUS_BAR, 0);

   261

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

   262

   263

	bool scroll = false;

   264

   265

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

   266

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

   267

		w3 = FindWindowById(WC_MAIN_TOOLBAR, 0);

   268

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

   269

   270

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

   271

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

   272

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

   273

		} else {

   274

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

   275

			 * list in below the widget */

   276

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

   277

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

   278

			height = rows * avg_height;

   279

			scroll = true;

   280

		}

   281

	}

   282

   283

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

   284

   285

	DropdownWindow *dw = new DropdownWindow(

   286

		w->left + wi->left,

   287

		top,

   288

		width,

   289

		height + 4,

   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

	dw->parent_wnd_class = w->window_class;

   316

	dw->parent_wnd_num   = w->window_number;

   317

	dw->parent_button    = button;

   318

	dw->list             = list;

   319

	dw->selected_index   = selected;

   320

	dw->click_delay      = 0;

   321

	dw->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

		DropdownWindow *dw = dynamic_cast<DropdownWindow*>(*wz);

   362

		if (pw->window_class == dw->parent_wnd_class &&

   363

				pw->window_number == dw->parent_wnd_num) {

   364

			delete dw;

   365

			break;

   366

		}

   367

	}

   368

}

   369