qttrackballwidget.hpp

Go to the documentation of this file.

#ifndef __QTTRACKBALLWIDGET__
#define __QTTRACKBALLWIDGET__

#include <GL/glew.h>

#include "objimporter.hpp"
#include "plyimporter.hpp"

#include <tucano.hpp>
#include <utils/trackball.hpp>

#include <QGLWidget>
#include <QMouseEvent>
#include <QFileDialog>

using namespace std;

namespace Tucano
{

class QtGlewInitializer : public QGLWidget
{
public:
    QtGlewInitializer (QWidget* parent) : QGLWidget(parent)
    {
        makeCurrent();
        Misc::initGlew();
    }

};

class QtTrackballWidget : public QtGlewInitializer
{
    Q_OBJECT

protected:

    Mesh mesh;

    Trackball camera;

    Trackball light_trackball;

public:

    explicit QtTrackballWidget(QWidget *parent) : QtGlewInitializer(parent) {}

    ~QtTrackballWidget () {}

    virtual void initializeGL (void)
    {
        makeCurrent();

        #ifdef TUCANODEBUG
        QGLFormat glCurrentFormat = this->format();
        cout << "QT GL version : " << glCurrentFormat.majorVersion() << " , " << glCurrentFormat.minorVersion() << endl;
        #endif
    }

    virtual void resizeGL (void)
    {
        camera.setViewport(Eigen::Vector2f ((float)this->width(), (float)this->height()));
        camera.setPerspectiveMatrix(camera.getFovy(), (float)this->width()/(float)this->height(), 0.1f, 100.0f);
        light_trackball.setViewport(Eigen::Vector2f ((float)this->width(), (float)this->height()));
        updateGL();
    }

    void initialize (void)
    {
        camera.setPerspectiveMatrix(60.0, (float)this->width()/(float)this->height(), 0.1f, 100.0f);
        camera.setRenderFlag(true);
        camera.setViewport(Eigen::Vector2f ((float)this->width(), (float)this->height()));

        light_trackball.setRenderFlag(false);
        light_trackball.setViewport(Eigen::Vector2f ((float)this->width(), (float)this->height()));

        updateGL();
    }

    virtual void openMesh (string filename)
    {
        QString str (filename.c_str());
        QStringList strlist = str.split(".");
        QString extension = strlist[strlist.size()-1];

        if (extension.compare("ply") != 0 && extension.compare("obj") != 0)
        {
            cerr << "file format [" << extension.toStdString() << "] not supported" << endl;
            return;
        }

                mesh = Mesh();

        if (extension.compare("ply") == 0)
        {
            MeshImporter::loadPlyFile(&mesh, filename);
        }
        if (extension.compare("obj") == 0)
        {
            MeshImporter::loadObjFile(&mesh, filename);
        }

        mesh.normalizeModelMatrix();
    }

protected:

    void keyPressEvent (QKeyEvent * event)
    {
        if (event->key() == Qt::Key_O)
        {
            QString filename = QFileDialog::getOpenFileName(this, tr("Open File"), "", tr("Mesh Files (*.obj *.ply)"));
            if (!filename.isEmpty())
            {
                openMesh (filename.toStdString());
            }
        }
        event->ignore();
        updateGL();
    }

    void mousePressEvent (QMouseEvent * event)
    {
        setFocus ();
        Eigen::Vector2f screen_pos (event->x(), event->y());
        if (event->modifiers() & Qt::ShiftModifier)
        {
            if (event->button() == Qt::LeftButton)
            {
                camera.translateCamera(screen_pos);
            }
        }
        else
        {
            if (event->button() == Qt::LeftButton)
            {
                camera.rotateCamera(screen_pos);
            }
            if (event->button() == Qt::RightButton)
            {
                light_trackball.rotateCamera(screen_pos);
            }
        }
        updateGL ();
    }

    void mouseMoveEvent (QMouseEvent * event)
    {
        Eigen::Vector2f screen_pos (event->x(), event->y());
        if (event->modifiers() & Qt::ShiftModifier && event->buttons() & Qt::LeftButton)
        {
            camera.translateCamera(screen_pos);
        }
        else
        {
            if (event->buttons() & Qt::LeftButton)
            {
                camera.rotateCamera(screen_pos);
            }
            if (event->buttons() & Qt::RightButton)
            {
                light_trackball.rotateCamera(screen_pos);
            }
        }

        updateGL ();

    }

    void mouseReleaseEvent (QMouseEvent * event)
    {
        if (event->button() == Qt::LeftButton)
        {
            camera.endTranslation();
            camera.endRotation();
        }
        if (event->button() == Qt::RightButton)
        {
            light_trackball.endRotation();
        }

        updateGL ();
    }

    void wheelEvent (QWheelEvent * event)
    {
        const int WHEEL_STEP = 120;

        float pos = event->delta () / float (WHEEL_STEP);

        if (event->modifiers() & Qt::ShiftModifier) // change FOV
        {
            camera.incrementFov(pos);
        }
        else // change ZOOM
        {
            if( (pos > 0) )
            {
                camera.increaseZoom(1.05);
            }

            else if(pos < 0)
            {
                camera.increaseZoom(1.0/1.05);
            }
        }
        updateGL ();
    }

signals:

public slots:

};

}
#endif