#include "dialogrender.h"

#include <cmath>
#include <QVBoxLayout>
#include <QImage>
#include <QColor>
#include <QPainter>
#include <QMessageBox>
#include <QFileDialog>
#include <QMessageBox>
#include <QListWidget>
#include <QPushButton>
#include <QMutex>
#include <QThread>
#include <QScrollArea>
#include <QLabel>
#include <QProgressBar>
#include <QComboBox>
#include "tools.h"

#include "Scenery.h"
#include "ColorProfile.h"

static DialogRender* _current_dialog;

static void _renderStart(int width, int height, Color background)
{
    _current_dialog->pixbuf_lock->lock();
    delete _current_dialog->pixbuf;
    _current_dialog->pixbuf = new QImage(width, height, QImage::Format_ARGB32);
    _current_dialog->pixbuf->fill(colorToQColor(background).rgb());
    _current_dialog->pixbuf_lock->unlock();

    _current_dialog->tellRenderSize(width, height);
}

static void _renderDraw(int x, int y, Color col)
{
    _current_dialog->pixbuf->setPixel(x, _current_dialog->pixbuf->height() - 1 - y, colorToQColor(col).rgb());
}

static void _renderUpdate(double progress)
{
    _current_dialog->area->update();
    _current_dialog->tellProgressChange(progress);
}

class RenderThread:public QThread
{
public:
    RenderThread(DialogRender* dialog, Renderer* renderer, RenderParams params):QThread()
    {
        _dialog = dialog;
        _renderer = renderer;
        _params = params;
    }
    void run()
    {
        rendererStart(_renderer, _params);
        _dialog->tellRenderEnded();
    }
private:
    DialogRender* _dialog;
    Renderer* _renderer;
    RenderParams _params;
};

class _RenderArea:public QWidget
{
public:
    _RenderArea(QWidget* parent):
        QWidget(parent)
    {
        setMinimumSize(800, 600);
    }

    void paintEvent(QPaintEvent*)
    {
        QPainter painter(this);
        _current_dialog->pixbuf_lock->lock();
        painter.drawImage(0, 0, *_current_dialog->pixbuf);
        _current_dialog->pixbuf_lock->unlock();
    }
};

DialogRender::DialogRender(QWidget *parent, Renderer* renderer):
    QDialog(parent, Qt::WindowTitleHint | Qt::WindowMaximizeButtonHint | Qt::WindowMinimizeButtonHint | Qt::WindowCloseButtonHint)
{
    pixbuf_lock = new QMutex();
    pixbuf = new QImage(1, 1, QImage::Format_ARGB32);
    _current_dialog = this;
    _render_thread = NULL;
    _renderer = renderer;

    setModal(true);
    setWindowTitle(tr("Paysages 3D - Render"));
    setLayout(new QVBoxLayout());

    _scroll = new QScrollArea(this);
    _scroll->setAlignment(Qt::AlignCenter);
    area = new _RenderArea(_scroll);
    _scroll->setWidget(area);
    layout()->addWidget(_scroll);

    // Status bar
    _info = new QWidget(this);
    _info->setLayout(new QHBoxLayout());
    layout()->addWidget(_info);

    _timer = new QLabel(QString("0:00.00"), _info);
    _info->layout()->addWidget(_timer);

    _progress = new QProgressBar(_info);
    _progress->setMaximumHeight(12);
    _progress->setMinimum(0);
    _progress->setMaximum(1000);
    _progress->setValue(0);
    _info->layout()->addWidget(_progress);

    // Action bar
    _actions = new QWidget(this);
    _actions->setLayout(new QHBoxLayout());
    layout()->addWidget(_actions);

    _actions->layout()->addWidget(new QLabel(tr("Tone-mapping: "), _actions));
    _tonemapping_control = new QComboBox(_actions);
    _tonemapping_control->addItems(QStringList(tr("Uncharted")) << tr("Reinhard"));
    _actions->layout()->addWidget(_tonemapping_control);

    _actions->layout()->addWidget(new QLabel(tr("Exposure: "), _actions));
    _actions->hide();
    _exposure_control = new QSlider(Qt::Horizontal, _actions);
    _exposure_control->setMinimumWidth(200);
    _exposure_control->setRange(0, 1000);
    _exposure_control->setValue(200);
    _actions->layout()->addWidget(_exposure_control);

    _save_button = new QPushButton(QIcon(getDataPath("images/save.png")), tr("Save picture"), _actions);
    _actions->layout()->addWidget(_save_button);

    // Connections
    //connect(this, SIGNAL(renderSizeChanged(int, int)), this, SLOT(applyRenderSize(int, int)));
    connect(this, SIGNAL(progressChanged(double)), this, SLOT(applyProgress(double)));
    connect(this, SIGNAL(renderEnded()), this, SLOT(applyRenderEnded()));
    connect(_save_button, SIGNAL(clicked()), this, SLOT(saveRender()));
    connect(_tonemapping_control, SIGNAL(currentIndexChanged(int)), this, SLOT(toneMappingChanged()));
    connect(_exposure_control, SIGNAL(valueChanged(int)), this, SLOT(toneMappingChanged()));
}

DialogRender::~DialogRender()
{
    if (_render_thread)
    {
        rendererInterrupt(_renderer);
        _render_thread->wait();

        delete _render_thread;
    }
    delete pixbuf;
    delete pixbuf_lock;
}

void DialogRender::tellRenderSize(int width, int height)
{
    emit renderSizeChanged(width, height);
}

void DialogRender::tellProgressChange(double value)
{
    emit progressChanged(value);
}

void DialogRender::tellRenderEnded()
{
    emit renderEnded();
}

void DialogRender::startRender(RenderParams params)
{
    _started = time(NULL);

    applyRenderSize(params.width, params.height);
    rendererSetPreviewCallbacks(_renderer, _renderStart, _renderDraw, _renderUpdate);

    _render_thread = new RenderThread(this, _renderer, params);
    _render_thread->start();

    exec();
}

void DialogRender::applyRenderEnded()
{
    _info->hide();
    _actions->show();

    area->update();
}

void DialogRender::saveRender()
{
    QString filepath;

    filepath = QFileDialog::getSaveFileName(this, tr("Paysages 3D - Choose a filename to save the last render"), QString(), tr("Images (*.png *.jpg)"));
    if (!filepath.isNull())
    {
        if (!filepath.toLower().endsWith(".jpg") && !filepath.toLower().endsWith(".jpeg") && !filepath.toLower().endsWith(".png"))
        {
            filepath = filepath.append(".png");
        }
        if (renderSaveToFile(_renderer->render_area, (char*)filepath.toStdString().c_str()))
        {
            QMessageBox::information(this, "Message", QString(tr("The picture %1 has been saved.")).arg(filepath));
        }
        else
        {
            QMessageBox::critical(this, "Message", QString(tr("Can't write to file : %1")).arg(filepath));
        }
    }
}

void DialogRender::toneMappingChanged()
{
    renderSetToneMapping(_renderer->render_area, ColorProfile((ColorProfile::ToneMappingOperator)_tonemapping_control->currentIndex(), ((double)_exposure_control->value()) * 0.01));
}

void DialogRender::loadLastRender()
{
    applyRenderSize(_renderer->render_width, _renderer->render_height);
    rendererSetPreviewCallbacks(_renderer, _renderStart, _renderDraw, _renderUpdate);
    renderEnded();
    toneMappingChanged();

    exec();
}

void DialogRender::applyRenderSize(int width, int height)
{
    area->setMinimumSize(width, height);
    area->setMaximumSize(width, height);
    area->resize(width, height);
    _scroll->setMinimumSize(width > 800 ? 820 : width + 20, height > 600 ? 620 : height + 20);
}

void DialogRender::applyProgress(double value)
{
    double diff = difftime(time(NULL), _started);
    int hours = (int)floor(diff / 3600.0);
    int minutes = (int)floor((diff - 3600.0 * hours) / 60.0);
    int seconds = (int)floor(diff - 3600.0 * hours - 60.0 * minutes);
    _timer->setText(tr("%1:%2.%3").arg(hours).arg(minutes, 2, 10, QLatin1Char('0')).arg(seconds, 2, 10, QLatin1Char('0')));
    _progress->setValue((int)(value * 1000.0));
    _progress->update();
}