File indexing completed on 2024-04-21 04:57:23

 /* This file is part of the KDE project
 
    Copyright (C) 2006 Dario Massarin <nekkar@libero.it>
    Copyright (C) 2007 by Javier Goday <jgoday@gmail.com>
    Copyright (C) 2008 Lukas Appelhans <l.appelhans@gmx.de>
    Copyright (C) 2010 Matthias Fuchs <mat69@gmx.net>
 
    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 2 of the License, or (at your option) any later version.
 */
 
 #include "ui/transfersviewdelegate.h"
 
 #include "core/kget.h"
 #include "core/transfergrouphandler.h"
 #include "core/transferhandler.h"
 #include "core/transfertreemodel.h"
 #include "core/transfertreeselectionmodel.h"
 #include "settings.h"
 #include "transferdetails.h"
 #include "ui/contextmenu.h"
 
 #include "kget_debug.h"
 #include <KLocalizedString>
 
 #include <QAbstractItemView>
 #include <QApplication>
 #include <QButtonGroup>
 #include <QDebug>
 #include <QHBoxLayout>
 #include <QIcon>
 #include <QMenu>
 #include <QMouseEvent>
 #include <QPainter>
 
 GroupStatusButton::GroupStatusButton(const QModelIndex &index, QWidget *parent)
     : QToolButton(parent)
     , m_status(None)
     , m_index(index)
     , m_timerId(-1)
     , m_iconSize(22)
     , m_gradientId(0)
 {
     setAttribute(Qt::WA_NoSystemBackground);
 }
 
 void GroupStatusButton::checkStateSet()
 {
     //     qCDebug(KGET_DEBUG) << "GroupStatusButton::checkStateSet";
 
     QToolButton::checkStateSet();
 
     if (isChecked()) {
         if (m_status == None)
             m_gradientId = 0.9;
         m_status = Selecting;
     } else {
         if (m_status == None)
             m_gradientId = 1;
         m_status = Deselecting;
     }
 
     setMouseTracking(!isChecked());
 
     if (m_timerId == -1)
         m_timerId = startTimer(100);
 }
 
 void GroupStatusButton::enterEvent(QEnterEvent *event)
 {
     Q_UNUSED(event)
     if (!isChecked()) {
         m_status = Blinking;
 
         if (m_timerId == -1) {
             m_timerId = startTimer(100);
 
             if (m_status == !BlinkingExiting)
                 m_gradientId = 1;
         }
     }
 }
 
 void GroupStatusButton::leaveEvent(QEvent *event)
 {
     Q_UNUSED(event)
     if (m_status == Blinking)
         m_status = BlinkingExiting;
 }
 
 void GroupStatusButton::paintEvent(QPaintEvent *event)
 {
     Q_UNUSED(event)
 
     QPainter p(this);
 
     const int offset = (rect().width() - m_iconSize) / 2;
 
     if (m_gradientId == 0)
         m_gradientId = isChecked() ? 1 : 0.7;
 
     QRadialGradient gradient(height() / 2.0, height() / 2.0, height() / 2);
 
     QPen pen;
 
     if (KGet::selectionModel()->isSelected(m_index)) {
         gradient.setColorAt(0, palette().color(QPalette::AlternateBase));
         gradient.setColorAt(m_gradientId, Qt::transparent);
         gradient.setColorAt(1, Qt::transparent);
         pen.setColor(palette().color(QPalette::AlternateBase));
     } else {
         gradient.setColorAt(0, palette().color(QPalette::Highlight));
         gradient.setColorAt(m_gradientId, Qt::transparent);
         gradient.setColorAt(1, Qt::transparent);
         pen.setColor(palette().color(QPalette::Highlight));
     }
 
     QRect r = rect().adjusted(0, 0, 0, 1);
 
     p.fillRect(r, gradient);
 
     p.setRenderHint(QPainter::Antialiasing);
 
     if (isChecked()) {
         pen.setWidth(1);
         p.setPen(pen);
         p.drawEllipse(rect().x() + 5, rect().y() + 4, rect().width() - 10, rect().width() - 10);
     }
 
     p.drawPixmap(rect().topLeft() + QPoint(offset, offset - 1),
                  icon().pixmap(m_iconSize, isChecked() || m_status == Blinking ? QIcon::Normal : QIcon::Disabled));
 }
 
 void GroupStatusButton::timerEvent(QTimerEvent *event)
 {
     Q_UNUSED(event)
 
     if (m_status == Selecting) {
         m_gradientId += 0.04;
 
         if (m_gradientId >= 1) {
             m_status = None;
             m_gradientId = 1;
             killTimer(m_timerId);
             m_timerId = -1;
         }
     } else if (m_status == Deselecting) {
         m_gradientId -= 0.04;
 
         if (m_gradientId <= 0.7) {
             m_status = None;
             m_gradientId = 0.7;
             killTimer(m_timerId);
             m_timerId = -1;
         }
     } else if (m_status == Blinking) {
         if (isChecked()) {
             m_status = Selecting;
             m_gradientId = 0.9;
             return;
         }
 
         m_gradientId -= 0.04;
 
         if (m_gradientId <= 0.7) {
             m_gradientId = 1;
         }
     } else if (m_status == BlinkingExiting) {
         m_gradientId -= 0.04;
 
         if (m_gradientId <= 0.7) {
             m_status = None;
             m_gradientId = 0.7;
             killTimer(m_timerId);
             m_timerId = -1;
         }
     }
 
     update();
 }
 
 GroupStatusEditor::GroupStatusEditor(const QModelIndex &index, QWidget *parent)
     : QWidget(parent)
     , m_index(index)
 {
     setMinimumWidth(80);
 
     m_layout = new QHBoxLayout();
     m_layout->addStretch();
     setLayout(m_layout);
 
     m_btGroup = new QButtonGroup(this);
     m_btGroup->setExclusive(true);
 
     m_startBt = new GroupStatusButton(m_index, this);
     m_startBt->setCheckable(true);
     m_startBt->setAutoRaise(true);
     m_startBt->setIcon(QIcon::fromTheme("media-playback-start"));
     m_startBt->setFixedSize(36, 36);
     m_startBt->setIconSize(QSize(22, 22));
     m_layout->addWidget(m_startBt);
     m_btGroup->addButton(m_startBt);
 
     m_stopBt = new GroupStatusButton(m_index, this);
     m_stopBt->setCheckable(true);
     m_stopBt->setAutoRaise(true);
     m_stopBt->setIcon(QIcon::fromTheme("media-playback-pause"));
     m_stopBt->setFixedSize(36, 36);
     m_stopBt->setIconSize(QSize(22, 22));
     m_layout->addWidget(m_stopBt);
     m_btGroup->addButton(m_stopBt);
 
     m_stopBt->setChecked(true);
 
     m_layout->addStretch();
     m_layout->setContentsMargins(1, 1, 1, 1);
 
     connect(m_startBt, &GroupStatusButton::toggled, this, &GroupStatusEditor::slotStatusChanged);
 }
 
 void GroupStatusEditor::setRunning(bool running)
 {
     if (running == m_startBt->isChecked())
         return;
 
     if (running)
         m_startBt->setChecked(true);
     else
         m_stopBt->setChecked(true);
 }
 
 bool GroupStatusEditor::isRunning()
 {
     return m_startBt->isChecked();
 }
 
 void GroupStatusEditor::slotStatusChanged()
 {
     Q_EMIT changedStatus(this);
 }
 
 BasicTransfersViewDelegate::BasicTransfersViewDelegate(QAbstractItemView *parent)
     : KExtendableItemDelegate(parent)
 {
 }
 
 QWidget *BasicTransfersViewDelegate::createEditor(QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index) const
 {
     if (index.column() == TransferTreeModel::Status) {
         auto *qroupStatusEditor = new GroupStatusEditor(index, parent);
         connect(qroupStatusEditor, &GroupStatusEditor::changedStatus, this, &BasicTransfersViewDelegate::slotGroupStatusChanged);
         return qroupStatusEditor;
     } else {
         return KExtendableItemDelegate::createEditor(parent, option, index);
     }
 }
 
 void BasicTransfersViewDelegate::slotGroupStatusChanged(GroupStatusEditor *editor)
 {
     commitData(editor);
 }
 
 void BasicTransfersViewDelegate::setEditorData(QWidget *editor, const QModelIndex &index) const
 {
     if (index.column() == TransferTreeModel::Status) {
         auto *groupEditor = static_cast<GroupStatusEditor *>(editor);
         groupEditor->setRunning(KGet::model()->itemFromIndex(index)->asGroup()->groupHandler()->status() == JobQueue::Running);
     } else {
         KExtendableItemDelegate::setEditorData(editor, index);
     }
 }
 
 void BasicTransfersViewDelegate::setModelData(QWidget *editor, QAbstractItemModel *model, const QModelIndex &index) const
 {
     if (index.column() == TransferTreeModel::Status) {
         auto *groupEditor = static_cast<GroupStatusEditor *>(editor);
         TransferGroupHandler *groupHandler = KGet::model()->itemFromIndex(index)->asGroup()->groupHandler();
 
         if (groupEditor->isRunning()) {
             groupHandler->start();
         } else {
             groupHandler->stop();
         }
     } else {
         KExtendableItemDelegate::setModelData(editor, model, index);
     }
 }
 
 TransfersViewDelegate::TransfersViewDelegate(QAbstractItemView *parent)
     : BasicTransfersViewDelegate(parent)
 {
     Q_ASSERT(qobject_cast<QAbstractItemView *>(parent));
     setExtendPixmap(QIcon::fromTheme("arrow-right").pixmap(16));
     setContractPixmap(QIcon::fromTheme("arrow-down").pixmap(16));
 }
 
 void TransfersViewDelegate::paint(QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index) const
 {
     TransferTreeModel *transferTreeModel = KGet::model();
 
     ModelItem *item = transferTreeModel->itemFromIndex(index);
 
     if (item->isGroup()) {
         painter->save();
 
         if (option.state & QStyle::State_Selected) {
         } else {
             static bool backgroundInitialized = false;
             static QPixmap groupBackground(64, 35);
             static QPalette palette(QApplication::palette());
 
             if (!backgroundInitialized || palette != QApplication::palette()) {
                 const QRect rect = groupBackground.rect();
                 QPainter p(&groupBackground);
 
                 QLinearGradient gradient(rect.x(), rect.y(), rect.x(), (rect.y() + rect.height()));
 
                 gradient.setColorAt(0, QApplication::palette().color(QPalette::Base));
                 gradient.setColorAt(0.5, QApplication::palette().color(QPalette::AlternateBase).darker(110));
                 gradient.setColorAt(1, QApplication::palette().color(QPalette::Base));
 
                 p.fillRect(rect, gradient);
                 backgroundInitialized = true;
             }
 
             painter->drawTiledPixmap(option.rect, groupBackground);
         }
 
         KExtendableItemDelegate::paint(painter, option, index);
 
         painter->restore();
     } else {
         if (KGet::selectionModel()->isSelected(index))
             painter->fillRect(option.rect,
                               QApplication::palette().color(option.state & QStyle::State_Active ? QPalette::Active : QPalette::Inactive, QPalette::Highlight));
 
         KExtendableItemDelegate::paint(painter, option, index);
 
         if (index.column() == 3 && !isExtended(transferTreeModel->index(index.row(), 0, index.parent()))) { // the percent column
             TransferHandler *transferHandler = item->asTransfer()->transferHandler();
 
             // following progressbar code has mostly been taken from Qt4 examples/network/torrent/mainview.cpp
             // Set up a QStyleOptionProgressBar to precisely mimic the
             // environment of a progress bar.
             QStyleOptionProgressBar progressBarOption;
             progressBarOption.state = QStyle::State_Enabled | QStyle::State_Horizontal;
             progressBarOption.direction = QApplication::layoutDirection();
             progressBarOption.rect = option.rect;
             progressBarOption.fontMetrics = QApplication::fontMetrics();
             progressBarOption.minimum = 0;
             progressBarOption.maximum = 100;
             progressBarOption.textAlignment = Qt::AlignCenter;
             progressBarOption.textVisible = true;
 
             // Set the progress and text values of the style option.
             int progress = transferHandler->percent();
             if (progress >= 0 && progress <= 100) {
                 progressBarOption.progress = progress;
                 progressBarOption.text = QStringLiteral("%1%").arg(progressBarOption.progress);
             } else {
                 progressBarOption.text = i18nc("not available", "n/a");
             }
 
             progressBarOption.rect.setY(progressBarOption.rect.y() + (option.rect.height() - QApplication::fontMetrics().height()) / 2);
             progressBarOption.rect.setHeight(QApplication::fontMetrics().height());
 
             // Draw the progress bar onto the view.
             QApplication::style()->drawControl(QStyle::CE_ProgressBar, &progressBarOption, painter);
         }
     }
 
     ///    These lines are just for testing purposes. Uncomment them to show on the view the repaint events.
     //     static int i=0;
     //     qCDebug(KGET_DEBUG) << "paint!!! " << i++ << " " << index.internalPointer() << " " << index.column();
     //
     //     painter->drawRect(option.rect);
     //     painter->drawText(option.rect.topLeft(), QString::number(i));
 }
 
 void TransfersViewDelegate::drawFocus(QPainter *painter, const QStyleOptionViewItem &option, const QRect &rect) const
 {
     Q_UNUSED(painter)
     Q_UNUSED(option)
     Q_UNUSED(rect)
 }
 
 QSize TransfersViewDelegate::sizeHint(const QStyleOptionViewItem &option, const QModelIndex &index) const
 {
     Q_UNUSED(option)
 
     TransferTreeModel *transferTreeModel = KGet::model();
     ModelItem *item = transferTreeModel->itemFromIndex(index);
 
     if (!item) {
         qCWarning(KGET_DEBUG) << "Sizehint for non-existing item.";
         return QSize();
     }
 
     if (transferTreeModel->itemFromIndex(index)->isGroup()) {
         return QSize(0, 35);
     } else {
         QSize ret(KExtendableItemDelegate::sizeHint(option, index));
         ret.rheight() += 8;
         return ret;
     }
 }
 
 bool TransfersViewDelegate::editorEvent(QEvent *event, QAbstractItemModel *model, const QStyleOptionViewItem &option, const QModelIndex &index)
 {
     Q_UNUSED(model)
     Q_UNUSED(option)
 
     if (event->type() == QEvent::MouseButtonRelease) {
         auto *mouseEvent = static_cast<QMouseEvent *>(event);
         if (mouseEvent->button() == Qt::RightButton) {
             //             qCDebug(KGET_DEBUG) << "TransfersViewDelegate::editorEvent() -> rightClick";
 
             QMenu *popup = nullptr;
 
             TransferTreeModel *transferTreeModel = KGet::model();
 
             ModelItem *item = transferTreeModel->itemFromIndex(index);
             if (item->isGroup()) {
                 //                 qCDebug(KGET_DEBUG) << "isTransferGroup = true";
                 TransferGroupHandler *transferGroupHandler = item->asGroup()->groupHandler();
 
                 popup = ContextMenu::createTransferGroupContextMenu(transferGroupHandler, qobject_cast<QWidget *>(this));
             } else {
                 //                 qCDebug(KGET_DEBUG) << "isTransferGroup = false";
 
                 TransferHandler *transferHandler = item->asTransfer()->transferHandler();
 
                 popup = ContextMenu::createTransferContextMenu(transferHandler, qobject_cast<QWidget *>(this));
             }
 
             if (popup) {
                 popup->exec(QCursor::pos());
                 popup->deleteLater();
             }
         }
     }
 
     return false;
 }
 
 #include "moc_transfersviewdelegate.cpp"