// Copyright (C) 2002-2012 Nikolaus Gebhardt
// This file is part of the "Irrlicht Engine".
// For conditions of distribution and use, see copyright notice in irrlicht.h

#pragma once

#include "IGUIListBox.h"
#include "irrArray.h"

namespace irr
{
namespace gui
{

class IGUIFont;
class IGUIScrollBar;

class CGUIListBox : public IGUIListBox
{
public:
	//! constructor
	CGUIListBox(IGUIEnvironment *environment, IGUIElement *parent,
			s32 id, core::rect<s32> rectangle, bool clip = true,
			bool drawBack = false, bool moveOverSelect = false);

	//! destructor
	virtual ~CGUIListBox();

	//! returns amount of list items
	u32 getItemCount() const override;

	//! returns string of a list item. the id may be a value from 0 to itemCount-1
	const wchar_t *getListItem(u32 id) const override;

	//! adds an list item, returns id of item
	u32 addItem(const wchar_t *text) override;

	//! clears the list
	void clear() override;

	//! returns id of selected item. returns -1 if no item is selected.
	s32 getSelected() const override;

	//! sets the selected item. Set this to -1 if no item should be selected
	void setSelected(s32 id) override;

	//! sets the selected item. Set this to -1 if no item should be selected
	void setSelected(const wchar_t *item) override;

	//! called if an event happened.
	bool OnEvent(const SEvent &event) override;

	//! draws the element and its children
	void draw() override;

	//! adds an list item with an icon
	//! \param text Text of list entry
	//! \param icon Sprite index of the Icon within the current sprite bank. Set it to -1 if you want no icon
	//! \return
	//! returns the id of the new created item
	u32 addItem(const wchar_t *text, s32 icon) override;

	//! Returns the icon of an item
	s32 getIcon(u32 id) const override;

	//! removes an item from the list
	void removeItem(u32 id) override;

	//! get the the id of the item at the given absolute coordinates
	s32 getItemAt(s32 xpos, s32 ypos) const override;

	//! Sets the sprite bank which should be used to draw list icons. This font is set to the sprite bank of
	//! the built-in-font by default. A sprite can be displayed in front of every list item.
	//! An icon is an index within the icon sprite bank. Several default icons are available in the
	//! skin through getIcon
	void setSpriteBank(IGUISpriteBank *bank) override;

	//! set whether the listbox should scroll to newly selected items
	void setAutoScrollEnabled(bool scroll) override;

	//! returns true if automatic scrolling is enabled, false if not.
	bool isAutoScrollEnabled() const override;

	//! Update the position and size of the listbox, and update the scrollbar
	void updateAbsolutePosition() override;

	//! set all item colors at given index to color
	void setItemOverrideColor(u32 index, video::SColor color) override;

	//! set all item colors of specified type at given index to color
	void setItemOverrideColor(u32 index, EGUI_LISTBOX_COLOR colorType, video::SColor color) override;

	//! clear all item colors at index
	void clearItemOverrideColor(u32 index) override;

	//! clear item color at index for given colortype
	void clearItemOverrideColor(u32 index, EGUI_LISTBOX_COLOR colorType) override;

	//! has the item at index its color overwritten?
	bool hasItemOverrideColor(u32 index, EGUI_LISTBOX_COLOR colorType) const override;

	//! return the overwrite color at given item index.
	video::SColor getItemOverrideColor(u32 index, EGUI_LISTBOX_COLOR colorType) const override;

	//! return the default color which is used for the given colorType
	video::SColor getItemDefaultColor(EGUI_LISTBOX_COLOR colorType) const override;

	//! set the item at the given index
	void setItem(u32 index, const wchar_t *text, s32 icon) override;

	//! Insert the item at the given index
	//! Return the index on success or -1 on failure.
	s32 insertItem(u32 index, const wchar_t *text, s32 icon) override;

	//! Swap the items at the given indices
	void swapItems(u32 index1, u32 index2) override;

	//! set global itemHeight
	void setItemHeight(s32 height) override;

	//! Sets whether to draw the background
	void setDrawBackground(bool draw) override;

	//! Access the vertical scrollbar
	IGUIScrollBar *getVerticalScrollBar() const override;

private:
	struct ListItem
	{
		core::stringw Text;
		s32 Icon = -1;

		// A multicolor extension
		struct ListItemOverrideColor
		{
			bool Use = false;
			video::SColor Color;
		};
		ListItemOverrideColor OverrideColors[EGUI_LBC_COUNT]{};
	};

	void recalculateItemHeight();
	void selectNew(s32 ypos, bool onlyHover = false);
	void recalculateScrollPos();
	void updateScrollBarSize(s32 size);

	// extracted that function to avoid copy&paste code
	void recalculateItemWidth(s32 icon);

	// get labels used for serialization
	bool getSerializationLabels(EGUI_LISTBOX_COLOR colorType, core::stringc &useColorLabel, core::stringc &colorLabel) const;

	core::array<ListItem> Items;
	s32 Selected;
	s32 ItemHeight;
	s32 ItemHeightOverride;
	s32 TotalItemHeight;
	s32 ItemsIconWidth;
	gui::IGUIFont *Font;
	gui::IGUISpriteBank *IconBank;
	gui::IGUIScrollBar *ScrollBar;
	u32 selectTime;
	u32 LastKeyTime;
	core::stringw KeyBuffer;
	bool Selecting;
	bool DrawBack;
	bool MoveOverSelect;
	bool AutoScroll;
	bool HighlightWhenNotFocused;
};

} // end namespace gui
} // end namespace irr