//Qt includes
#include <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);
}