// 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);

}