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

 /* ============================================================
 * 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 "toolbutton.h"
 
 #include <QMenu>
 #include <QStyle>
 #include <QPainter>
 #include <QMouseEvent>
 #include <QApplication>
 #include <QStyleOptionToolButton>
 
 ToolButton::ToolButton(QWidget* parent)
     : QToolButton(parent)
     , m_menu(nullptr)
 {
     setMinimumWidth(16);
 
     QStyleOptionToolButton opt;
     initStyleOption(&opt);
 
     m_pressTimer.setSingleShot(true);
     m_pressTimer.setInterval(QApplication::style()->styleHint(QStyle::SH_ToolButton_PopupDelay, &opt, this));
     connect(&m_pressTimer, SIGNAL(timeout()), this, SLOT(showMenu()));
 }
 
 QImage ToolButton::multiIcon() const
 {
     return m_multiIcon;
 }
 
 void ToolButton::setMultiIcon(const QImage &image)
 {
     m_options |= MultiIconOption;
     m_multiIcon = image;
     setFixedSize(m_multiIcon.width(), m_multiIcon.height() / 4);
 
     update();
 }
 
 QString ToolButton::themeIcon() const
 {
     return m_themeIcon;
 }
 
 void ToolButton::setThemeIcon(const QString &icon)
 {
     const QIcon ic = QIcon::fromTheme(icon);
     if (!ic.isNull()) {
         m_themeIcon = icon;
         setIcon(QIcon::fromTheme(m_themeIcon));
     }
 }
 
 QIcon ToolButton::fallbackIcon() const
 {
     return icon();
 }
 
 void ToolButton::setFallbackIcon(const QIcon &fallbackIcon)
 {
     if (icon().isNull())
         setIcon(fallbackIcon);
 }
 
 QIcon ToolButton::icon() const
 {
     return QToolButton::icon();
 }
 
 void ToolButton::setIcon(const QIcon &icon)
 {
     if (m_options & MultiIconOption)
         setFixedSize(sizeHint());
 
     m_options &= ~MultiIconOption;
     QToolButton::setIcon(icon);
 }
 
 QMenu* ToolButton::menu() const
 {
     return m_menu;
 }
 
 void ToolButton::setMenu(QMenu* menu)
 {
     Q_ASSERT(menu);
 
     if (m_menu)
         disconnect(m_menu, &QMenu::aboutToHide, this, &ToolButton::menuAboutToHide);
 
     m_menu = menu;
     connect(m_menu, &QMenu::aboutToHide, this, &ToolButton::menuAboutToHide);
 }
 
 bool ToolButton::showMenuInside() const
 {
     return m_options & ShowMenuInsideOption;
 }
 
 void ToolButton::setShowMenuInside(bool enable)
 {
     if (enable)
         m_options |= ShowMenuInsideOption;
     else
         m_options &= ~ShowMenuInsideOption;
 }
 
 bool ToolButton::showMenuOnRightClick() const
 {
     return m_options & ShowMenuOnRightClick;
 }
 
 void ToolButton::setShowMenuOnRightClick(bool enable)
 {
     m_options.setFlag(ShowMenuOnRightClick, enable);
 }
 
 bool ToolButton::toolbarButtonLook() const
 {
     return m_options & ToolBarLookOption;
 }
 
 void ToolButton::setToolbarButtonLook(bool enable)
 {
     if (enable) {
         m_options |= ToolBarLookOption;
 
         QStyleOption opt;
         opt.initFrom(this);
         int size = style()->pixelMetric(QStyle::PM_ToolBarIconSize, &opt, this);
         setIconSize(QSize(size, size));
     }
     else {
         m_options &= ~ToolBarLookOption;
     }
 
     setProperty("toolbar-look", QVariant(enable));
     style()->unpolish(this);
     style()->polish(this);
 }
 
 void ToolButton::menuAboutToHide()
 {
     setDown(false);
     Q_EMIT aboutToHideMenu();
 }
 
 void ToolButton::showMenu()
 {
     if (!m_menu || m_menu->isVisible())
         return;
 
     Q_EMIT aboutToShowMenu();
 
     QPoint pos;
 
     if (m_options & ShowMenuInsideOption) {
         pos = mapToGlobal(rect().bottomRight());
         if (QApplication::layoutDirection() == Qt::RightToLeft)
             pos.setX(pos.x() - rect().width());
         else
             pos.setX(pos.x() - m_menu->sizeHint().width());
     }
     else {
         pos = mapToGlobal(rect().bottomLeft());
     }
 
     m_menu->popup(pos);
 }
 
 void ToolButton::mousePressEvent(QMouseEvent* e)
 {
     if (e->buttons() == Qt::LeftButton && popupMode() == QToolButton::DelayedPopup)
         m_pressTimer.start();
 
     if (e->buttons() == Qt::LeftButton && menu() && popupMode() == QToolButton::InstantPopup) {
         setDown(true);
         showMenu();
     }
     else if (e->buttons() == Qt::RightButton && menu() && m_options & ShowMenuOnRightClick) {
         setDown(true);
         showMenu();
     } else {
         QToolButton::mousePressEvent(e);
     }
 }
 
 void ToolButton::mouseReleaseEvent(QMouseEvent* e)
 {
     m_pressTimer.stop();
 
     if (e->button() == Qt::MiddleButton && rect().contains(e->pos())) {
         Q_EMIT middleMouseClicked();
         setDown(false);
     }
     else if (e->button() == Qt::LeftButton && rect().contains(e->pos()) && e->modifiers() == Qt::ControlModifier) {
         Q_EMIT controlClicked();
         setDown(false);
     } else {
         QToolButton::mouseReleaseEvent(e);
     }
 }
 
 void ToolButton::mouseDoubleClickEvent(QMouseEvent* e)
 {
     QToolButton::mouseDoubleClickEvent(e);
 
     m_pressTimer.stop();
 
     if (e->buttons() == Qt::LeftButton) {
         Q_EMIT doubleClicked();
     }
 }
 
 void ToolButton::contextMenuEvent(QContextMenuEvent *e)
 {
     // Block to prevent showing both context menu and button menu
     if (menu() && m_options & ShowMenuOnRightClick)
         return;
 
     QToolButton::contextMenuEvent(e);
 }
 
 void ToolButton::paintEvent(QPaintEvent* e)
 {
     if (!(m_options & MultiIconOption)) {
         QToolButton::paintEvent(e);
         return;
     }
 
     QPainter p(this);
 
     const int w = m_multiIcon.width();
     const int h4 = m_multiIcon.height() / 4;
 
     if (!isEnabled())
         p.drawImage(0, 0, m_multiIcon, 0, h4 * 3, w, h4);
     else if (isDown())
         p.drawImage(0, 0, m_multiIcon, 0, h4 * 2, w, h4);
     else if (underMouse())
         p.drawImage(0, 0, m_multiIcon, 0, h4 * 1, w, h4);
     else
         p.drawImage(0, 0, m_multiIcon, 0, h4 * 0, w, h4);
 }