//Qt includes

#include <qmenu></qmenu>

#include "toonzqt/popupbutton.h"

//********************************************************************************

//    PopupButton class implementation

//********************************************************************************

PopupButton::PopupButton(QWidget *parent)

	: QPushButton(parent), m_index(-1)

{

	setFocusPolicy(Qt::NoFocus);

	QMenu *menu = new QMenu(this);

	setMenu(menu);

	connect(this, SIGNAL(clicked(bool)), this, SLOT(showMenu()));

}

//--------------------------------------------------------------------------

QAction *PopupButton::addItem(const QString &text)

{

	QAction *action = menu()->addAction(text);

	connect(action, SIGNAL(triggered(bool)), this, SLOT(onIndexChange()));

	m_actions.push_back(action);

	return action;

}

//--------------------------------------------------------------------------

QAction *PopupButton::addItem(const QIcon &icon, const QString &text)

{

	QAction *action = menu()->addAction(icon, text);

	connect(action, SIGNAL(triggered(bool)), this, SLOT(onIndexChange()));

	m_actions.push_back(action);

	return action;

}

//--------------------------------------------------------------------------

QAction *PopupButton::addSeparator()

{

	return menu()->addSeparator();

}

//--------------------------------------------------------------------------

int PopupButton::currentIndex() const

{

	return m_index;

}

//--------------------------------------------------------------------------

const QAction *PopupButton::currentItem() const

{

	return (m_index >= 0) ? m_actions[m_index] : 0;

}

//--------------------------------------------------------------------------

void PopupButton::setCurrentIndex(int index)

{

	if (index < 0 || index >= m_actions.size())

		return;

	m_index = index;

	const QAction *action = currentItem();

	setIcon(action->icon());

	setText(action->text());

	emit currentIndexChanged(index);

}

//--------------------------------------------------------------------------

int PopupButton::count() const

{

	return m_actions.size();

}

//--------------------------------------------------------------------------

int PopupButton::findText(const QString &text, Qt::CaseSensitivity cs) const

{

	int i, size = m_actions.size();

	for (i = 0; i < size; ++i)

		if (QString::compare(m_actions[i]->text(), text, cs) == 0)

			return i;

	return -1;

}

//--------------------------------------------------------------------------

void PopupButton::onIndexChange()

{

	QAction *action = (QAction *)QObject::sender();

	int i, size = m_actions.size();

	for (i = 0; i < size; ++i)

		if (m_actions[i] == action)

			break;

	if (i < size)

		setCurrentIndex(i);

	emit activated(i);

}