/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the demonstration applications of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QDebug>

#include "scene.h"

#include <QtGui/qmatrix4x4.h>

#include <QtGui/qvector3d.h>

#include "3rdparty/fbm.h"

void checkGLErrors(const QString& prefix)

{

    switch (glGetError()) {

    case GL_NO_ERROR:

        //qDebug() << prefix << tr("No error.");

        break;

    case GL_INVALID_ENUM:

        qDebug() << prefix << QObject::tr("Invalid enum.");

        break;

    case GL_INVALID_VALUE:

        qDebug() << prefix << QObject::tr("Invalid value.");

        break;

    case GL_INVALID_OPERATION:

        qDebug() << prefix << QObject::tr("Invalid operation.");

        break;

    case GL_STACK_OVERFLOW:

        qDebug() << prefix << QObject::tr("Stack overflow.");

        break;

    case GL_STACK_UNDERFLOW:

        qDebug() << prefix << QObject::tr("Stack underflow.");

        break;

    case GL_OUT_OF_MEMORY:

        qDebug() << prefix << QObject::tr("Out of memory.");

        break;

    default:

        qDebug() << prefix << QObject::tr("Unknown error.");

        break;

    }

}

//============================================================================//

//                                  ColorEdit                                 //

//============================================================================//

ColorEdit::ColorEdit(QRgb initialColor, int id)

    : m_color(initialColor), m_id(id)

{

    QHBoxLayout *layout = new QHBoxLayout;

    setLayout(layout);

    layout->setContentsMargins(0, 0, 0, 0);

    m_lineEdit = new QLineEdit(QString::number(m_color, 16));

    layout->addWidget(m_lineEdit);

    m_button = new QFrame;

    QPalette palette = m_button->palette();

    palette.setColor(QPalette::Window, QColor(m_color));

    m_button->setPalette(palette);

    m_button->setAutoFillBackground(true);

    m_button->setMinimumSize(32, 0);

    m_button->setSizePolicy(QSizePolicy::Fixed, QSizePolicy::Preferred);

    m_button->setFrameStyle(QFrame::StyledPanel | QFrame::Sunken);

    layout->addWidget(m_button);

    connect(m_lineEdit, SIGNAL(editingFinished()), this, SLOT(editDone()));

}

void ColorEdit::editDone()

{

    bool ok;

    QRgb newColor = m_lineEdit->text().toUInt(&ok, 16);

    if (ok)

        setColor(newColor);

}

void ColorEdit::mousePressEvent(QMouseEvent *event)

{

    if (event->button() == Qt::LeftButton) {

        QColor color(m_color);

        QColorDialog dialog(color, 0);

        dialog.setOption(QColorDialog::ShowAlphaChannel, true);

// The ifdef block is a workaround for the beta, TODO: remove when bug 238525 is fixed

#ifdef Q_WS_MAC

        dialog.setOption(QColorDialog::DontUseNativeDialog, true);

#endif

        dialog.move(280, 120);

        if (dialog.exec() == QDialog::Rejected)

            return;

        QRgb newColor = dialog.selectedColor().rgba();

        if (newColor == m_color)

            return;

        setColor(newColor);

    }

}

void ColorEdit::setColor(QRgb color)

{

    m_color = color;

    m_lineEdit->setText(QString::number(m_color, 16)); // "Clean up" text

    QPalette palette = m_button->palette();

    palette.setColor(QPalette::Window, QColor(m_color));

    m_button->setPalette(palette);

    emit colorChanged(m_color, m_id);

}

//============================================================================//

//                                  FloatEdit                                 //

//============================================================================//

FloatEdit::FloatEdit(float initialValue, int id)

    : m_value(initialValue), m_id(id)

{

    QHBoxLayout *layout = new QHBoxLayout;

    setLayout(layout);

    layout->setContentsMargins(0, 0, 0, 0);

    m_lineEdit = new QLineEdit(QString::number(m_value));

    layout->addWidget(m_lineEdit);

    connect(m_lineEdit, SIGNAL(editingFinished()), this, SLOT(editDone()));

}

void FloatEdit::editDone()

{

    bool ok;

    float newValue = m_lineEdit->text().toFloat(&ok);

    if (ok) {

        m_value = newValue;

        m_lineEdit->setText(QString::number(m_value)); // "Clean up" text

        emit valueChanged(m_value, m_id);

    }

}

//============================================================================//

//                           TwoSidedGraphicsWidget                           //

//============================================================================//

TwoSidedGraphicsWidget::TwoSidedGraphicsWidget(QGraphicsScene *scene)

    : QObject(scene)

    , m_current(0)

    , m_angle(0)

    , m_delta(0)

{

    for (int i = 0; i < 2; ++i)

        m_proxyWidgets[i] = 0;

}

void TwoSidedGraphicsWidget::setWidget(int index, QWidget *widget)

{

    if (index < 0 || index >= 2)

	{

		qWarning("TwoSidedGraphicsWidget::setWidget: Index out of bounds, index == %d", index);

		return;

	}

    GraphicsWidget *proxy = new GraphicsWidget;

    proxy->setWidget(widget);

    if (m_proxyWidgets[index])

        delete m_proxyWidgets[index];

    m_proxyWidgets[index] = proxy;

    proxy->setCacheMode(QGraphicsItem::ItemCoordinateCache);

    proxy->setZValue(1e30); // Make sure the dialog is drawn on top of all other (OpenGL) items

    if (index != m_current)

        proxy->setVisible(false);

    qobject_cast<QGraphicsScene *>(parent())->addItem(proxy);

}

QWidget *TwoSidedGraphicsWidget::widget(int index)

{

    if (index < 0 || index >= 2)

	{

		qWarning("TwoSidedGraphicsWidget::widget: Index out of bounds, index == %d", index);

		return 0;

	}

    return m_proxyWidgets[index]->widget();

}

void TwoSidedGraphicsWidget::flip()

{

    m_delta = (m_current == 0 ? 9 : -9);

    animateFlip();

}

void TwoSidedGraphicsWidget::animateFlip()

{

    m_angle += m_delta;

    if (m_angle == 90) {

        int old = m_current;

        m_current ^= 1;

        m_proxyWidgets[old]->setVisible(false);

        m_proxyWidgets[m_current]->setVisible(true);

        m_proxyWidgets[m_current]->setGeometry(m_proxyWidgets[old]->geometry());

    }

    QRectF r = m_proxyWidgets[m_current]->boundingRect();

    m_proxyWidgets[m_current]->setTransform(QTransform()

        .translate(r.width() / 2, r.height() / 2)

        .rotate(m_angle - 180 * m_current, Qt::YAxis)

        .translate(-r.width() / 2, -r.height() / 2));

    if ((m_current == 0 && m_angle > 0) || (m_current == 1 && m_angle < 180))

        QTimer::singleShot(25, this, SLOT(animateFlip()));

}

QVariant GraphicsWidget::itemChange(GraphicsItemChange change, const QVariant &value)

{

    if (change == ItemPositionChange && scene()) {

        QRectF rect = boundingRect();

        QPointF pos = value.toPointF();

        QRectF sceneRect = scene()->sceneRect();

        if (pos.x() + rect.left() < sceneRect.left())

            pos.setX(sceneRect.left() - rect.left());

        else if (pos.x() + rect.right() >= sceneRect.right())

            pos.setX(sceneRect.right() - rect.right());

        if (pos.y() + rect.top() < sceneRect.top())

            pos.setY(sceneRect.top() - rect.top());

        else if (pos.y() + rect.bottom() >= sceneRect.bottom())

            pos.setY(sceneRect.bottom() - rect.bottom());

        return pos;

    }

    return QGraphicsProxyWidget::itemChange(change, value);

}

void GraphicsWidget::resizeEvent(QGraphicsSceneResizeEvent *event)

{

    setCacheMode(QGraphicsItem::NoCache);

    setCacheMode(QGraphicsItem::ItemCoordinateCache);

    QGraphicsProxyWidget::resizeEvent(event);

}

void GraphicsWidget::paint(QPainter *painter, const QStyleOptionGraphicsItem *option, QWidget *widget)

{

    painter->setRenderHint(QPainter::Antialiasing, false);

    QGraphicsProxyWidget::paint(painter, option, widget);

    //painter->setRenderHint(QPainter::Antialiasing, true);

}

//============================================================================//

//                             RenderOptionsDialog                            //

//============================================================================//

RenderOptionsDialog::RenderOptionsDialog()

    : QDialog(0, Qt::CustomizeWindowHint | Qt::WindowTitleHint)

{

    setWindowOpacity(0.75);

    setWindowTitle(tr("Options (double click to flip)"));

    QGridLayout *layout = new QGridLayout;

    setLayout(layout);

    layout->setColumnStretch(1, 1);

    int row = 0;

    QCheckBox *check = new QCheckBox(tr("Dynamic cube map"));

    check->setCheckState(Qt::Unchecked);

    // Dynamic cube maps are only enabled when multi-texturing and render to texture are available.

    check->setEnabled(glActiveTexture && glGenFramebuffersEXT);

    connect(check, SIGNAL(stateChanged(int)), this, SIGNAL(dynamicCubemapToggled(int)));

    layout->addWidget(check, 0, 0, 1, 2);

    ++row;

    QPalette palette;

    // Load all .par files

    // .par files have a simple syntax for specifying user adjustable uniform variables.

    QSet<QByteArray> uniforms;

    QList<QString> filter = QStringList("*.par");

    QList<QFileInfo> files = QDir(":/res/boxes/").entryInfoList(filter, QDir::Files | QDir::Readable);

    foreach (QFileInfo fileInfo, files) {

        QFile file(fileInfo.absoluteFilePath());

        if (file.open(QIODevice::ReadOnly)) {

            while (!file.atEnd()) {

                QList<QByteArray> tokens = file.readLine().simplified().split(' ');

                QList<QByteArray>::const_iterator it = tokens.begin();

                if (it == tokens.end())

                    continue;

                QByteArray type = *it;

                if (++it == tokens.end())

                    continue;

                QByteArray name = *it;

                bool singleElement = (tokens.size() == 3); // type, name and one value

                char counter[10] = "000000000";

                int counterPos = 8; // position of last digit

                while (++it != tokens.end()) {

                    m_parameterNames << name;

                    if (!singleElement) {

                        m_parameterNames.back() += "[";

                        m_parameterNames.back() += counter + counterPos;

                        m_parameterNames.back() += "]";

                        int j = 8; // position of last digit

                        ++counter[j];

                        while (j > 0 && counter[j] > '9') {

                            counter[j] = '0';

                            ++counter[--j];

                        }

                        if (j < counterPos)

                            counterPos = j;

                    }

                    if (type == "color") {

                        layout->addWidget(new QLabel(m_parameterNames.back()));

                        bool ok;

                        ColorEdit *colorEdit = new ColorEdit(it->toUInt(&ok, 16), m_parameterNames.size() - 1);

                        m_parameterEdits << colorEdit;

                        layout->addWidget(colorEdit);

                        connect(colorEdit, SIGNAL(colorChanged(QRgb,int)), this, SLOT(setColorParameter(QRgb,int)));

                        ++row;

                    } else if (type == "float") {

                        layout->addWidget(new QLabel(m_parameterNames.back()));

                        bool ok;

                        FloatEdit *floatEdit = new FloatEdit(it->toFloat(&ok), m_parameterNames.size() - 1);

                        m_parameterEdits << floatEdit;

                        layout->addWidget(floatEdit);

                        connect(floatEdit, SIGNAL(valueChanged(float,int)), this, SLOT(setFloatParameter(float,int)));

                        ++row;

                    }

                }

            }

            file.close();

        }

    }

    layout->addWidget(new QLabel(tr("Texture:")));

    m_textureCombo = new QComboBox;

    connect(m_textureCombo, SIGNAL(currentIndexChanged(int)), this, SIGNAL(textureChanged(int)));

    layout->addWidget(m_textureCombo);

    ++row;

    layout->addWidget(new QLabel(tr("Shader:")));

    m_shaderCombo = new QComboBox;

    connect(m_shaderCombo, SIGNAL(currentIndexChanged(int)), this, SIGNAL(shaderChanged(int)));

    layout->addWidget(m_shaderCombo);

    ++row;

    layout->setRowStretch(row, 1);

}

int RenderOptionsDialog::addTexture(const QString &name)

{

    m_textureCombo->addItem(name);

    return m_textureCombo->count() - 1;

}

int RenderOptionsDialog::addShader(const QString &name)

{

    m_shaderCombo->addItem(name);

    return m_shaderCombo->count() - 1;

}

void RenderOptionsDialog::emitParameterChanged()

{

    foreach (ParameterEdit *edit, m_parameterEdits)

        edit->emitChange();

}

void RenderOptionsDialog::setColorParameter(QRgb color, int id)

{

    emit colorParameterChanged(m_parameterNames[id], color);

}

void RenderOptionsDialog::setFloatParameter(float value, int id)

{

    emit floatParameterChanged(m_parameterNames[id], value);

}

void RenderOptionsDialog::mouseDoubleClickEvent(QMouseEvent *event)

{

    if (event->button() == Qt::LeftButton)

        emit doubleClicked();

}

//============================================================================//

//                                 ItemDialog                                 //

//============================================================================//

ItemDialog::ItemDialog()

    : QDialog(0, Qt::CustomizeWindowHint | Qt::WindowTitleHint)

{

    setWindowTitle(tr("Items (double click to flip)"));

    setWindowOpacity(0.75);

    resize(160, 100);

    QVBoxLayout *layout = new QVBoxLayout;

    setLayout(layout);

    QPushButton *button;

    button = new QPushButton(tr("Add Qt box"));

    layout->addWidget(button);

    connect(button, SIGNAL(clicked()), this, SLOT(triggerNewQtBox()));

    button = new QPushButton(tr("Add circle"));

    layout->addWidget(button);

    connect(button, SIGNAL(clicked()), this, SLOT(triggerNewCircleItem()));

    button = new QPushButton(tr("Add square"));

    layout->addWidget(button);

    connect(button, SIGNAL(clicked()), this, SLOT(triggerNewSquareItem()));

    layout->addStretch(1);

}

void ItemDialog::triggerNewQtBox()

{

    emit newItemTriggered(QtBoxItem);

}

void ItemDialog::triggerNewCircleItem()

{

    emit newItemTriggered(CircleItem);

}

void ItemDialog::triggerNewSquareItem()

{

    emit newItemTriggered(SquareItem);

}

void ItemDialog::mouseDoubleClickEvent(QMouseEvent *event)

{

    if (event->button() == Qt::LeftButton)

        emit doubleClicked();

}

//============================================================================//

//                                    Scene                                   //

//============================================================================//

const static char environmentShaderText[] =

    "uniform samplerCube env;"

    "void main() {"

        "gl_FragColor = textureCube(env, gl_TexCoord[1].xyz);"

    "}";

Scene::Scene(int width, int height, int maxTextureSize)

    : m_distExp(600)

    , m_frame(0)

    , m_maxTextureSize(maxTextureSize)

    , m_currentShader(0)

    , m_currentTexture(0)

    , m_dynamicCubemap(false)

    , m_updateAllCubemaps(true)

    , m_box(0)

    , m_vertexShader(0)

    , m_environmentShader(0)

    , m_environmentProgram(0)

{

    setSceneRect(0, 0, width, height);

    m_trackBalls[0] = TrackBall(0.05f, QVector3D(0, 1, 0), TrackBall::Sphere);

    m_trackBalls[1] = TrackBall(0.005f, QVector3D(0, 0, 1), TrackBall::Sphere);

    m_trackBalls[2] = TrackBall(0.0f, QVector3D(0, 1, 0), TrackBall::Plane);

    m_renderOptions = new RenderOptionsDialog;

    m_renderOptions->move(20, 120);

    m_renderOptions->resize(m_renderOptions->sizeHint());

    connect(m_renderOptions, SIGNAL(dynamicCubemapToggled(int)), this, SLOT(toggleDynamicCubemap(int)));

    connect(m_renderOptions, SIGNAL(colorParameterChanged(QString,QRgb)), this, SLOT(setColorParameter(QString,QRgb)));

    connect(m_renderOptions, SIGNAL(floatParameterChanged(QString,float)), this, SLOT(setFloatParameter(QString,float)));

    connect(m_renderOptions, SIGNAL(textureChanged(int)), this, SLOT(setTexture(int)));

    connect(m_renderOptions, SIGNAL(shaderChanged(int)), this, SLOT(setShader(int)));

    m_itemDialog = new ItemDialog;

    connect(m_itemDialog, SIGNAL(newItemTriggered(ItemDialog::ItemType)), this, SLOT(newItem(ItemDialog::ItemType)));

    TwoSidedGraphicsWidget *twoSided = new TwoSidedGraphicsWidget(this);

    twoSided->setWidget(0, m_renderOptions);

    twoSided->setWidget(1, m_itemDialog);

    connect(m_renderOptions, SIGNAL(doubleClicked()), twoSided, SLOT(flip()));

    connect(m_itemDialog, SIGNAL(doubleClicked()), twoSided, SLOT(flip()));

    addItem(new QtBox(64, width - 64, height - 64));

    addItem(new QtBox(64, width - 64, 64));

    addItem(new QtBox(64, 64, height - 64));

    addItem(new QtBox(64, 64, 64));

    initGL();

    m_timer = new QTimer(this);

    m_timer->setInterval(20);

    connect(m_timer, SIGNAL(timeout()), this, SLOT(update()));

    m_timer->start();

    m_time.start();

}