File indexing completed on 2024-05-12 04:58:26

 /* ============================================================
 * 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;