buttonwithmenu.cpp

Go to the documentation of this file.

/* ============================================================
* Falkon - Qt web browser
* Copyright (C) 2010-2018 David Rosca <nowrep@gmail.com>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program.  If not, see <http://www.gnu.org/licenses/>.
* ============================================================ */
#include "buttonwithmenu.h"
 
#include <QMenu>
#include <QWheelEvent>
 
ButtonWithMenu::ButtonWithMenu(QWidget* parent)
    : ToolButton(parent)
    , m_menu(new QMenu(this))
{
    setCursor(Qt::ArrowCursor);
    setFocusPolicy(Qt::NoFocus);
 
    connect(this, &ToolButton::aboutToShowMenu, this, &ButtonWithMenu::generateMenu);
    connect(m_menu, &QMenu::aboutToShow, this, std::bind(&ButtonWithMenu::setDown, this, true));
    connect(m_menu, &QMenu::aboutToHide, this, std::bind(&ButtonWithMenu::setDown, this, false));
}
 
void ButtonWithMenu::setCurrentItem()
{
    if (auto* action = qobject_cast<QAction*>(sender())) {
        setCurrentItem(action->data().value<Item>());
    }
}
 
void ButtonWithMenu::clearItems()
{
    m_menu->clear();
    m_items.clear();
}
 
void ButtonWithMenu::selectNextItem()
{
    int index = m_items.indexOf(m_currentItem) + 1;
 
    if (index < m_items.size()) {
        setCurrentIndex(index);
    }
}
 
void ButtonWithMenu::selectPreviousItem()
{
    int index = m_items.indexOf(m_currentItem) - 1;
 
    if (index >= 0) {
        setCurrentIndex(index);
    }
}
 
void ButtonWithMenu::addItem(const Item &item)
{
    m_items.append(item);
 
    if (m_items.count() == 1) {
        setCurrentItem(item);
    }
 
    Q_EMIT itemAdded(item);
}
 
void ButtonWithMenu::addItems(const QVector<Item> &items)
{
    for (const Item &item : items) {
        addItem(item);
    }
}
 
void ButtonWithMenu::removeItem(const Item &item)
{
    int index = m_items.indexOf(item);
    if (index < 0) {
        return;
    }
 
    m_items.remove(index);
 
    if (m_items.isEmpty()) {
        setIcon(QIcon());
        return;
    }
 
    if (m_currentItem == item) {
        setCurrentItem(m_items.at(0));
    }
}
 
void ButtonWithMenu::setCurrentItem(const Item &item, bool emitSignal)
{
    int index = m_items.indexOf(item);
    if (index < 0 || m_currentItem == item) {
        return;
    }
 
    m_currentItem = item;
 
    setIcon(m_currentItem.icon);
    setToolTip(m_currentItem.text);
 
    if (emitSignal) {
        Q_EMIT activeItemChanged(m_currentItem);
    }
}
 
void ButtonWithMenu::setCurrentIndex(int index, bool emitSignal)
{
    setCurrentItem(m_items.at(index), emitSignal);
}
 
void ButtonWithMenu::wheelEvent(QWheelEvent* event)
{
    m_wheelHelper.processEvent(event);
    while (WheelHelper::Direction direction = m_wheelHelper.takeDirection()) {
        switch (direction) {
        case WheelHelper::WheelUp:
        case WheelHelper::WheelLeft:
            selectPreviousItem();
            break;
 
        case WheelHelper::WheelDown:
        case WheelHelper::WheelRight:
            selectNextItem();
            break;
 
        default:
            break;
        }
    }
    event->accept();
}
 
ButtonWithMenu::Item ButtonWithMenu::currentItem()
{
    return m_currentItem;
}
 
QMenu* ButtonWithMenu::menu() const
{
    return m_menu;
}
 
void ButtonWithMenu::generateMenu()
{
    m_menu->clear();
 
    for (const Item &item : std::as_const(m_items)) {
        QVariant variant;
        variant.setValue(item);
        m_menu->addAction(item.icon, item.text, this, SLOT(setCurrentItem()))->setData(variant);
    }
}
 
void ButtonWithMenu::mousePressEvent(QMouseEvent *event)
{
    if (event->buttons() == Qt::LeftButton && parentWidget() && parentWidget()->parentWidget()) {
        Q_EMIT aboutToShowMenu();
        QWidget *w = parentWidget()->parentWidget();
        m_menu->popup(w->mapToGlobal(w->rect().bottomLeft()));
    }
 
    ToolButton::mousePressEvent(event);
}
 
ButtonWithMenu::~ButtonWithMenu()
= default;