glwidget.h

Go to the documentation of this file.

#ifndef GLWIDGET_H
#define GLWIDGET_H




#include <QOpenGLBuffer>
#include <QOpenGLShader>
#include <QOpenGLFunctions>
#include <QOpenGLFramebufferObject>

#include <QGLWidget>
#include <QWidget>

#include "VideoFrameToTextureLoader.h"

#define VV_DRAWMODE_VOLUME      0
#define VV_DRAWMODE_VIDEODIFF   1

#define VV_METRIC_NONE          0
#define VV_METRIC_IQDIF         1
#define VV_METRIC_YDIF          2
#define VV_METRIC_YNDIF         3
#define VV_METRIC_YMSE          4
#define VV_METRIC_YNMSE         5

typedef uint VVMetric;


class GLWidget : public QGLWidget, protected QOpenGLFunctions//, protected QGLFunctions
{
    Q_OBJECT

public:

    GLWidget(QWidget *parent = 0);

    ~GLWidget();

    QSize minimumSizeHint() const;
    QSize sizeHint() const;


    VFLResult openVideoFile(const QString& videoFilePath, int *hRes, int *vRes);


    VFLResult loadVideoFile(int startFrame, int stepSize, int endFrame);


    int getFrameCount();


    void setFrameOffset(int frameOffset);


    void setStyle(bool horseShoeView);


    void setBoxSize(int boxSize);


    void setStepSize(float stepSize);


    void setMetric(VVMetric metric);


    void setTransferFunction(GLfloat *tfArray, int arraySize);

protected:
    void initializeGL();
    void paintGL();
    void resizeGL(int width, int height);
    void mouseMoveEvent(QMouseEvent *event);
    void mousePressEvent(QMouseEvent *event);
    void mouseReleaseEvent(QMouseEvent *event);
    void wheelEvent(QWheelEvent *event);

    PFNGLGENVERTEXARRAYSPROC glGenVertexArrays;
    PFNGLBINDVERTEXARRAYPROC glBindVertexArray;
    PFNGLTEXIMAGE3DPROC glTexImage3D;
private:
    bool loadShaders();
    bool prepareShaderProgram( const QString& vertexShaderPath,const QString& fragmentShaderPath, QOpenGLShaderProgram **shader);
    void createBufferObjects();

    void setupCamera();
    void setupModelMatrix();

    void calculateVideoDiffBuffer(bool normalized);
    void calculateMeanAndStdDev();

    QOpenGLShaderProgram *m_volumeShader;
    QOpenGLShaderProgram *m_videoDiffShader;
    QOpenGLBuffer *m_boxPosVBO;
    QOpenGLBuffer *m_texCoordVBO;
    QOpenGLBuffer *m_boxIBO;
    QOpenGLBuffer *m_squarePosVBO;
    QOpenGLBuffer *m_squareTexCoordVBO;
    QOpenGLBuffer *m_squareIBO;

    GLuint m_videoTexture;
    GLuint m_transferTexture;
    GLuint m_meanTexture;
    GLuint m_varTexture;
    GLuint m_boxVAO;
    GLuint m_squareVAO;

    QMatrix4x4 m_viewMatrix;
    QMatrix4x4 m_projectionMatrix;
    QMatrix4x4 m_modelMatrix;

    CVideoFrameToTextureLoader *p_videoFrameToTextureLoader;
    PBYTE p_videoBuffer;
    PBYTE p_videoDiffBuffer;
    PBYTE p_videoNDiffBuffer;
   // GLfloat *pVideoDiffMeanBuffer;
   // GLfloat *p_videoDiffDeviationBuffer;

    int m_videoWidth;
    int m_videoHeight;
    int m_frameCount;

    uint m_drawMode;

    float m_frameOffset;
    float m_cameraAngleUp;
    float m_cameraAngleSide;
    float m_cameraDistance;

    QVector3D m_cameraCenter;
    QVector3D m_backgroundColor;
    QVector3D m_cameraPosition;

    GLfloat *p_transferFunction;

    bool m_horseShowView;
    float m_stepSize;
    int m_boxSize;

    VVMetric m_metric;

    QPoint m_oldMousePos;
};

#endif // GLWIDGET_H