/*
  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 <QDesktopWidget>
#include <QApplication>
#include <QGraphicsScene>
#include <QGraphicsPixmapItem>
#include <QImage>
#include <QWheelEvent>
#include <QContextMenuEvent>
#include <QMenu>
#include <QTimer>
#include <QPainter>
#include <QStyleOptionGraphicsItem>
#include <QDate>
#include <QSettings>

#include "pictureviewer2.h"
#include "pictureswidget.h"

PictureViewer2::PictureViewer2(QWidget *parent) : QGraphicsView(parent), mCur(0), mCurPos(0), mDefaultFile(":/picgone.png"), mFnItem(0) {
    //behave like QDialog, but don't be one...
    setWindowFlags(QFlags<Qt::WindowType>(0x1|0x2|0x1000|0x2000|0x10000|0x8000000));

    //initial file list
    PicData initialData;
    for(int i = 0; i < PicFilesModel::NumFields; ++i){
        initialData << QVariant();
    }
    initialData[PicFilesModel::FullPath] = mDefaultFile;
    mFiles << initialData;

    //picture show
    mSlideTimer = new QTimer(this);
    connect(mSlideTimer, SIGNAL(timeout()), this, SLOT(next()));

    //setup gui
    mScene = new QGraphicsScene(this);
    setScene(mScene);
    QDesktopWidget *dw = QApplication::desktop();
    QSize screenSize = dw->availableGeometry().size();
    //TODO: configure!
    mMinimunSize = QSize(screenSize.width() - 100, screenSize.height() - 100);
    setMinimumSize(mMinimunSize);
    move(50, 50);
    setVisible(false);
    readSettings();
    mScene->setSceneRect(QRectF(0, 0, width() - 10, height() - 10));
    showFile(mFiles.at(mCurPos));
}

void PictureViewer2::addFiles(const PicDataList &files, bool clear){
    if(clear){
        mFiles.clear();
        mFiles = files;
        mCurPos = 0;
    }else{
        foreach(const PicData &f, files){
            if(!mFiles.contains(f)){
                mFiles.append(f);
            }
        }
    }
    if(!mCurPos >= mFiles.size()){
        showFile(mFiles.at(mCurPos));
    }else{
        showFile(mFiles.at(0));
    }
}

void PictureViewer2::setFile(const PicData &file){
    addFiles(PicDataList() << file, true);
    showFile(mFiles.at(mCurPos));
}

QSize PictureViewer2::sizeHint() const{
    return mMinimunSize;
}

void PictureViewer2::next(){
    if(mCurPos + 1 >= mFiles.size()){
        mCurPos = 0;
    }else{
        ++mCurPos;
    }
    showFile(mFiles.at(mCurPos));
}

void PictureViewer2::previous(){
    if(mCurPos - 1 < 0){
        mCurPos = mFiles.size() - 1;
    }else{
        --mCurPos;
    }
    showFile(mFiles.at(mCurPos));
}

void PictureViewer2::slide(){
    if(mSlideTimer->isActive()){
        mSlideTimer->stop();
    }else{
        mSlideTimer->start(5000);
    }
}

void PictureViewer2::readSettings(){
    QSettings s;
    mUseGradient = s.value("ui/pvgradient").toBool();
    QVariant t = s.value("ui/pvbgcolor");
    mBgColor = t.value<QColor>();
}

void PictureViewer2::wheelEvent(QWheelEvent *event){
    int steps = event->delta() / 8 / 15;
    if(steps < 0){
        next();
    }else{
        previous();
    }
}

void PictureViewer2::contextMenuEvent(QContextMenuEvent *e){
    QMenu ctxMenu;
    foreach(QAction *a, actions()){
        ctxMenu.addAction(a);
    }
    ctxMenu.exec(e->globalPos());
}

void PictureViewer2::showFile(const PicData &file){
    QPixmap pixmap(file.at(PicFilesModel::FullPath).toString());
    if(pixmap.isNull()){
        return;
    }
    if(mUseGradient){
        setGradient(pixmap);
    }else{
        setBackgroundBrush(QBrush(mBgColor));
    }
    if((pixmap.height() > height()) || (pixmap.width() > width())){
        pixmap = pixmap.scaled(QSize(width() - 3, height() - 10), Qt::KeepAspectRatio, Qt::SmoothTransformation);
    }
    if(mCur){
        mScene->removeItem(mCur);
        delete mCur;
        mCur = 0;
    }
    mCur = new QGraphicsPixmapItem(pixmap);
    float cX = (width() / 2) - (pixmap.width() / 2);
    float cY = height() / 2 - pixmap.height() / 2;
    mScene->addItem(mCur);
    mCur->setPos(cX, cY);
    setWindowTitle(constructWindowTitle());
    //TODO: configure!
    constructInfoItem(file, pixmap.size());
}

void PictureViewer2::constructInfoItem(const PicData &data, QSize picSize){
    if(mFnItem){
        mScene->removeItem(mFnItem);
        delete mFnItem;
    }
    mFnItem = new PictureViewer2FileinfoItem(data, mFiles.size(), picSize);
    mFnItem->setPos(10, 10);
    mScene->addItem(mFnItem);
}

void PictureViewer2::setGradient(const QPixmap &pic){
    QPoint val1 = QPoint(qrand() % pic.width(), qrand() % pic.height());
    QPoint val2 = QPoint(qrand() % pic.width(), qrand() % pic.height());
    QImage img = pic.toImage();
    QColor c1(img.pixel(val1));
    QColor c2(img.pixel(val2));
    QLinearGradient g(QPoint(0,0), QPoint(rect().bottomRight()));
    g.setColorAt(0, c1);
    g.setColorAt(1, c2);
    setBackgroundBrush(QBrush(g));
}

QString PictureViewer2::constructWindowTitle() const {
    QString retval = QString(tr("PicViewer 2 - [%1]")).arg(mFiles.at(mCurPos).at(PicFilesModel::FileName).toString());
    return retval;
}

PictureViewer2FileinfoItem::PictureViewer2FileinfoItem(const PicData &data, const int numSelected, const QSize &picSize, QGraphicsItem *parent) : QGraphicsItem(parent){
    QStringList textList;
    //prevent QStaticText from wrapping filesnames with spaces
    //we shouldn't allow this in the first place
    textList << data.at(PicFilesModel::FileName).toString().replace(' ', '_');
    textList << QString("%1x%2").arg(QString::number(picSize.width())).arg(QString::number(picSize.height()));
    QLocale l;
    textList << QString(QObject::tr("%1 bytes")).arg(l.toString(data.at(PicFilesModel::Size).toInt()));
    textList << l.toString(data.at(PicFilesModel::Added).toDate());
    textList << data.at(PicFilesModel::MimeType).toString();
    textList << QString(QObject::tr("%1 files selected")).arg(QString::number(numSelected));
    mText = QStaticText(textList.join("<br/>"));
}

QRectF PictureViewer2FileinfoItem::boundingRect() const {
    QSizeF size = mText.size();
    size += QSize(10, 10);
    QRectF retval;
    retval.setWidth(size.width());
    retval.setHeight(size.height());
    return retval;
}

void PictureViewer2FileinfoItem::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget){
    Q_UNUSED(option);
    Q_UNUSED(widget);
    painter->save();
    painter->setRenderHint(QPainter::Antialiasing, true);
    painter->setRenderHint(QPainter::TextAntialiasing, true);
    QColor bgColor = QColor(Qt::white);
    bgColor.setAlpha(80);
    QColor fgColor = QColor(Qt::black);
    QBrush brush(bgColor);
    painter->setPen(QColor(Qt::yellow));
    painter->setBrush(brush);
    painter->drawRoundedRect(boundingRect(), 15.0, 15.0);
    painter->setPen(QPen(fgColor));
    painter->drawStaticText(5, 5, mText);
    painter->restore();
}