path: root/openglscene.cpp

#include "openglscene.h"

#include "model.h"

#include <QtGui>
#include <QtOpenGL>

#ifndef QT_NO_CONCURRENT
#include <QFutureWatcher>
#endif

#ifndef GL_MULTISAMPLE
#define GL_MULTISAMPLE  0x809D
#endif

class Controls : public QGroupBox
{
    Q_OBJECT

public:
    Controls(OpenGLScene *scene);

private slots:
    void loadModel(const QString &model);
    void modelLoaded();
    void setModelColor(bool showDialog = true);
    void setBackgroundColor(bool showDialog = true);

private:
    OpenGLScene *m_scene;
#ifndef QT_NO_CONCURRENT
    QFutureWatcher<Model *> m_modelLoader;
#endif
    QComboBox *m_models;

    QRgb m_modelColor;
    QRgb m_backgroundColor;
    QDir m_dir;
};

Controls::Controls(OpenGLScene *scene)
    : m_scene(scene)
    , m_models(new QComboBox)
    , m_modelColor(qRgb(180, 100, 255))
    , m_backgroundColor(qRgb(0, 0, 0))
    , m_dir("models")
{
    QVBoxLayout *layout = new QVBoxLayout(this);

    layout->addWidget(new QLabel("Model:"));

    m_dir.setNameFilters(QStringList() << "*.obj");
    m_models->addItems(m_dir.entryList());
    connect(m_models, SIGNAL(currentIndexChanged(const QString &)), this, SLOT(loadModel(const QString &)));
#ifndef QT_NO_CONCURRENT
    connect(&m_modelLoader, SIGNAL(finished()), this, SLOT(modelLoaded()));
#endif

    if (m_models->count() > 0)
        loadModel(m_models->currentText());
    layout->addWidget(m_models);

    QCheckBox *autoRotate = new QCheckBox("Auto-rotate");
    autoRotate->setChecked(true);
    connect(autoRotate, SIGNAL(toggled(bool)), m_scene, SLOT(enableAutoRotate(bool)));
    layout->addWidget(autoRotate);

    QCheckBox *wireframe = new QCheckBox("Render as wireframe");
    wireframe->setChecked(true);
    connect(wireframe, SIGNAL(toggled(bool)), m_scene, SLOT(enableWireframe(bool)));
    layout->addWidget(wireframe);

    QCheckBox *normals = new QCheckBox("Display normals vectors");
    wireframe->setChecked(false);
    connect(normals, SIGNAL(toggled(bool)), m_scene, SLOT(enableNormals(bool)));
    layout->addWidget(normals);

    layout->addWidget(new QLabel("Light position:"));
    QSlider *lightPosition = new QSlider(Qt::Horizontal);
    lightPosition->setRange(-100, 100);
    connect(lightPosition, SIGNAL(valueChanged(int)), m_scene, SLOT(setLightPosition(int)));
    layout->addWidget(lightPosition);

    m_scene->setLightPosition(lightPosition->value());

    QPushButton *colorButton = new QPushButton("Choose model color");
    connect(colorButton, SIGNAL(pressed()), this, SLOT(setModelColor()));
    layout->addWidget(colorButton);
    setModelColor(false);

    QPushButton *backgroundButton = new QPushButton("Choose background color");
    connect(backgroundButton, SIGNAL(pressed()), this, SLOT(setBackgroundColor()));
    layout->addWidget(backgroundButton);
    setBackgroundColor(false);
}

Model *loadModel(const QString &filename)
{
    return new Model(filename);
}

void Controls::loadModel(const QString &filename)
{
    m_models->setEnabled(false);
    QApplication::setOverrideCursor(Qt::BusyCursor);
#ifndef QT_NO_CONCURRENT
    m_modelLoader.setFuture(QtConcurrent::run(::loadModel, m_dir.filePath(filename)));
#else
    m_scene->setModel(::loadModel(filename));
    modelLoaded();
#endif
}

void Controls::modelLoaded()
{
#ifndef QT_NO_CONCURRENT
    m_scene->setModel(m_modelLoader.result());
#endif
    m_models->setEnabled(true);
    QApplication::restoreOverrideCursor();
}

void Controls::setModelColor(bool showDialog)
{
    if (showDialog)
        m_modelColor = QColorDialog::getRgba(m_modelColor);

    m_scene->setModelColor(m_modelColor);
}

void Controls::setBackgroundColor(bool showDialog)
{
    if (showDialog)
        m_backgroundColor = QColorDialog::getRgba(m_backgroundColor);

    m_scene->setBackgroundColor(m_backgroundColor);
}

OpenGLScene::OpenGLScene()
    : m_wireframeEnabled(true)
    , m_normalsEnabled(false)
    , m_autoRotate(true)
    , m_rotating(false)
    , m_axis(0.0f, 1.0f, 0.0f)
    , m_angle(20.0f)
    , m_distance(1.5f)
    , m_model(0)
    , m_lastTime(0)
{
    // set identity matrix
    for (int i = 0; i < 4; ++i)
        for (int j = 0; j < 4; ++j)
            m_matrix[i][j] = (i == j);

    Controls *controls = new Controls(this);
    controls->setWindowOpacity(0.8);

    QGraphicsProxyWidget *item = addWidget(controls);
    item->translate(10, 10);
    item->setCacheMode(QGraphicsItem::DeviceCoordinateCache);

    m_time.start();
}

void OpenGLScene::updateMatrix(qreal delta)
{
    if (!QGLContext::currentContext())
        return;

    glMatrixMode(GL_MODELVIEW);
    glPushMatrix();
    glLoadIdentity();
    glRotatef(m_angle * delta, m_axis.x, m_axis.y, m_axis.z);
    glMultMatrixf(&m_matrix[0][0]);
    glGetFloatv(GL_MODELVIEW_MATRIX, &m_matrix[0][0]);
    glPopMatrix();
}

void OpenGLScene::drawBackground(QPainter *painter, const QRectF &)
{
    if (painter->paintEngine()->type() != QPaintEngine::OpenGL) {
        qWarning() << "OpenGLScene: drawBackground needs a QGLWidget to be set as viewport on the graphics view";
        return;
    }

    glClearColor(qRed(m_backgroundColor)/255.0f, qGreen(m_backgroundColor)/255.0f, qBlue(m_backgroundColor)/255.0f, 1);
    glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);

    glMatrixMode(GL_PROJECTION);
    glPushMatrix();
    glLoadIdentity();
    gluPerspective(70, painter->device()->width() / float(painter->device()->height()), 0.01, 1000);

    glMatrixMode(GL_MODELVIEW);
    glPushMatrix();
    glLoadIdentity();
    float pos[] = { m_lightPos, 5, 2, 0 };
    glLightfv(GL_LIGHT0, GL_POSITION, pos);
    glColor4f(qRed(m_modelColor)/255.0f, qGreen(m_modelColor)/255.0f, qBlue(m_modelColor)/255.0f, 1.0f);

    unsigned int current = m_time.elapsed();
    unsigned int delta = current - m_lastTime;
    m_lastTime = current;

    if (m_autoRotate && !m_rotating)
        updateMatrix(delta / 1000.0);

    glLoadIdentity();
    glTranslatef(0, 0, -m_distance);
    glMultMatrixf(&m_matrix[0][0]);

    glEnable(GL_MULTISAMPLE);
    if (m_model)
        m_model->render(m_wireframeEnabled, m_normalsEnabled);
    glDisable(GL_MULTISAMPLE);

    glPopMatrix();

    glMatrixMode(GL_PROJECTION);
    glPopMatrix();

    if (m_autoRotate)
        QTimer::singleShot(20, this, SLOT(update()));

    painter->paintEngine()->setDirty(QPaintEngine::AllDirty);
    painter->paintEngine()->syncState();
}

void OpenGLScene::setModel(Model *model)
{
    delete m_model;
    m_model = model;
    update();
}

void OpenGLScene::enableAutoRotate(bool enabled)
{
    m_autoRotate = enabled;
    update();
}

void OpenGLScene::enableWireframe(bool enabled)
{
    m_wireframeEnabled = enabled;
}

void OpenGLScene::enableNormals(bool enabled)
{
    m_normalsEnabled = enabled;
}

void OpenGLScene::setLightPosition(int pos)
{
    m_lightPos = pos * 0.05;
    update();
}

void OpenGLScene::setModelColor(QRgb color)
{
    m_modelColor = color;
    update();
}

void OpenGLScene::setBackgroundColor(QRgb color)
{
    m_backgroundColor = color;
    update();
}

static Point3d spherical(const QPointF &point, qreal w, qreal h)
{
    qreal R = qMax(w, h);

    Point3d p;
    p.x = -(point.x() - w / 2);
    p.y = point.y() - h / 2;
    p.z = sqrt(R * R - p.x * p.x - p.y * p.y);

    p.x /= R;
    p.y /= R;
    p.z /= R;
    return p;
}

void OpenGLScene::updateRotation(const QPointF &last, const QPointF &current)
{
    Point3d pos = spherical(current, width(), height());
    Point3d lastPos = spherical(last, width(), height());

    m_axis.x = lastPos.y * pos.z - lastPos.z * pos.y;
    m_axis.y = lastPos.z * pos.x - lastPos.x * pos.z;
    m_axis.z = lastPos.x * pos.y - lastPos.y * pos.x;

    qreal length = sqrt(m_axis.x * m_axis.x + m_axis.y * m_axis.y + m_axis.z * m_axis.z);

    if (length == 0) {
        m_angle = 0;
    } else {
        m_angle = -10 * asin(sqrt(length));

        m_axis.x /= length;
        m_axis.y /= length;
        m_axis.z /= length;

        m_accumulated += m_angle;

        updateMatrix();
        update();
    }
}

void OpenGLScene::mouseMoveEvent(QGraphicsSceneMouseEvent *event)
{
    QGraphicsScene::mouseMoveEvent(event);
    if (event->isAccepted() || !m_rotating)
        return;

    updateRotation(event->lastScenePos(), event->scenePos());
    event->accept();
}

void OpenGLScene::mousePressEvent(QGraphicsSceneMouseEvent *event)
{
    QGraphicsScene::mousePressEvent(event);
    if (event->isAccepted())
        return;

    m_startTime = m_time.elapsed();
    m_accumulated = 0;

    event->accept();
    m_rotating = true;
}

void OpenGLScene::mouseReleaseEvent(QGraphicsSceneMouseEvent *event)
{
    QGraphicsScene::mouseReleaseEvent(event);
    if (event->isAccepted())
        return;

    const unsigned int delta = m_time.elapsed() - m_startTime;
    m_angle = m_accumulated / (delta / 1000.0);

    event->accept();
    m_rotating = false;
}

void OpenGLScene::wheelEvent(QGraphicsSceneWheelEvent *event)
{
    QGraphicsScene::wheelEvent(event);
    if (event->isAccepted())
        return;

    m_distance *= qPow(1.2, -event->delta() / 120);
    event->accept();
    update();
}

#include "openglscene.moc"