qt5imagegpuwidget.h

#pragma once

#include <QOpenGLWidget>
#include <QOpenGLBuffer>
#include <QOpenGLShaderProgram>
#include <QOpenGLVertexArrayObject>
#include <QOpenGLFunctions_3_3_Core>
#include "../iucore.h"


namespace iu {
class Qt5ImageGpuWidget : public QOpenGLWidget, protected QOpenGLFunctions_3_3_Core
{
    Q_OBJECT
public:
    explicit Qt5ImageGpuWidget(const iu::Size<2> sz, QWidget* parent = NULL);
    virtual ~Qt5ImageGpuWidget();

public slots:
    void update_image(iu::ImageGpu_8u_C1* im);
    void update_image(iu::ImageGpu_8u_C4* im);
    void update_image(iu::ImageGpu_32f_C1 *im, float minVal, float maxVal);
    void update_image_colormap(iu::ImageGpu_32f_C1 *im, float minVal, float maxVal);

protected:

    void paintGL();
    void initializeGL();
    void resizeGL(int w, int h);

    void init_cuda();

    //iu::ImageGpu_8u_C1& img_;
    GLuint texture_;
    cudaGraphicsResource_t cuda_img_;
    QOpenGLBuffer pbo_;
    iu::Size<2> image_size_;

    QOpenGLShaderProgram* shader_program_;
    QOpenGLVertexArrayObject* vao_;
    QOpenGLBuffer* vbo_;
};

} // namespace iu