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