#include <QModelIndex>
#include <QPainter>
#include <QStyleOptionProgressBarV2>
#include <QProgressBar>
#include <QApplication>

#include "DLListDelegate.h"

DLListDelegate::DLListDelegate(QObject *parent) : QAbstractItemDelegate(parent)
{
    ;
}

DLListDelegate::~DLListDelegate(void)
{
    ;
}

void DLListDelegate::paint(QPainter * painter, const QStyleOptionViewItem & option, const QModelIndex & index) const
{
    QString byteUnits[4] = {tr("B"), tr("KB"), tr("MB"), tr("GB")};
    QStyleOptionViewItem opt = option;
    QStyleOptionProgressBarV2 newopt;
    QRect pixmapRect;
    QPixmap pixmap;
    qlonglong fileSize;
    double progress, dlspeed, multi;
    int minutes, hours, days;
    qlonglong remaining;
    QString temp , status,routequality, host, hash, ptr;
    qlonglong completed;

    //set text color
    QVariant value = index.data(Qt::TextColorRole);
    if(value.isValid() && qvariant_cast<QColor>(value).isValid()) {
        opt.palette.setColor(QPalette::Text, qvariant_cast<QColor>(value));
    }
    QPalette::ColorGroup cg = option.state & QStyle::State_Enabled ? QPalette::Normal : QPalette::Disabled;
    if(option.state & QStyle::State_Selected){
        painter->setPen(opt.palette.color(cg, QPalette::HighlightedText));
    } else {
        painter->setPen(opt.palette.color(cg, QPalette::Text));
    }

    // draw the background color
    if(index.column() != PROGRESS) {
        if(option.showDecorationSelected && (option.state & QStyle::State_Selected)) {
            if(cg == QPalette::Normal && !(option.state & QStyle::State_Active)) {
                cg = QPalette::Inactive;
            }
            painter->fillRect(option.rect, option.palette.brush(cg, QPalette::Highlight));
        } else {
            value = index.data(Qt::BackgroundColorRole);
            if(value.isValid() && qvariant_cast<QColor>(value).isValid()) {
                painter->fillRect(option.rect, qvariant_cast<QColor>(value));
            }
        }
    }
    switch(index.column()) {
        case SIZE:
            fileSize = index.data().toLongLong();
            if(fileSize < 0){
                temp = "Unknown";
            } else {
                multi = 1.0;
                for(int i = 0; i < 5; ++i) {
                    if (fileSize < 1024) {
                        fileSize = index.data().toLongLong();
                        temp.sprintf("%.2f ", fileSize / multi);
                        temp += byteUnits[i];
                        break;
                    }
                    fileSize /= 1024;
                    multi *= 1024.0;
                }
            }
            painter->drawText(option.rect, Qt::AlignCenter, temp);
            break;
        case REMAINING:
            remaining = index.data().toLongLong();
            minutes = remaining / 60;
            hours = minutes / 60;
            minutes = minutes - hours * 60;
            days = hours / 24;
            hours = hours - days * 24;
            if(days > 0) {
                temp.clear();
                temp.sprintf("%dd %2dh %dm", days, hours, minutes);
            } else if(hours > 0 || days > 0) {
                temp.clear();
                temp.sprintf("%dh %dm", hours, minutes);
            } else if(minutes > 0 || hours > 0) {
                temp.clear();
                temp.sprintf("%dm", minutes);
            } else {
                temp = "Unknown";
            }
            painter->drawText(option.rect, Qt::AlignCenter, temp);
            break;
        case COMPLETED:
            completed = index.data().toLongLong();
            if(completed < 0){
                temp = "Unknown";
            } else {
                multi = 1.0;
                for(int i = 0; i < 5; ++i) {
                    if (completed < 1024) {
                        completed = index.data().toLongLong();
                        temp.sprintf("%.2f ", completed / multi);
                        temp += byteUnits[i];
                        break;
                    }
                    completed /= 1024;
                    multi *= 1024.0;
                }
            }
            painter->drawText(option.rect, Qt::AlignCenter, temp);
            break;
        case DLSPEED:
            dlspeed = index.data().toDouble();
            temp.clear();
            temp.sprintf("%.2f", dlspeed/1024.);
            temp += " KB/s";
            painter->drawText(option.rect, Qt::AlignCenter, temp);
            break;
        case PROGRESS:
            progress = index.data().toDouble();
            temp.clear();
            temp.sprintf("%.2f", progress);
            temp += "%";
            newopt.rect = opt.rect;
            //newopt.text = temp;
            newopt.maximum = 100;
            newopt.minimum = 0;
            newopt.progress = (int)progress;
            newopt.state |= QStyle::State_Enabled;
            newopt.textVisible = false;
            QApplication::style()->drawControl(QStyle::CE_ProgressBar, &newopt,
            painter);
                //We prefer to display text manually to control color/font/boldness
            if (option.state & QStyle::State_Selected){
                opt.palette.setColor(QPalette::Text, QColor("grey"));
                painter->setPen(opt.palette.color(cg, QPalette::Text));
            }
            painter->drawText(option.rect, Qt::AlignCenter, newopt.text);
            break;
        case NAME:
                // decoration
            value = index.data(Qt::DecorationRole);
            pixmap = qvariant_cast<QIcon>(value).pixmap(option.decorationSize, option.state & QStyle::State_Enabled ? QIcon::Normal : QIcon::Disabled, option.state & QStyle::State_Open ? QIcon::On : QIcon::Off);
            pixmapRect = (pixmap.isNull() ? QRect(0, 0, 0, 0): QRect(QPoint(0, 0), option.decorationSize));
            if (pixmapRect.isValid()){
                QPoint p = QStyle::alignedRect(option.direction, Qt::AlignLeft, pixmap.size(), option.rect).topLeft();
                painter->drawPixmap(p, pixmap);
            }
            painter->drawText(option.rect.translated(pixmap.size().width(), 0), Qt::AlignLeft, index.data().toString());
            break;
        default:
            painter->drawText(option.rect, Qt::AlignCenter, index.data().toString());
    }
}

QSize DLListDelegate::sizeHint(const QStyleOptionViewItem & option, const QModelIndex & index) const
{
    QVariant value = index.data(Qt::FontRole);
    QFont fnt = value.isValid() ? qvariant_cast<QFont>(value) : option.font;
    QFontMetrics fontMetrics(fnt);
    const QString text = index.data(Qt::DisplayRole).toString();
    QRect textRect = QRect(0, 0, 0, fontMetrics.lineSpacing() * (text.count(QLatin1Char('\n')) + 1));
    return textRect.size();
}


{