GLWidget.hpp

#ifndef INCLUDE_ONCE_71D92E37_E297_472C_8495_1BF8EA61DC99
#define INCLUDE_ONCE_71D92E37_E297_472C_8495_1BF8EA61DC99

#include <memory>
#include <QOpenGLWidget>
#include <QOpenGLTexture>
#include <QOpenGLFunctions_3_3_Core>
#include "AtmosphereRenderer.hpp"
#include "../common/AtmosphereParameters.hpp"

class ToolsWidget;
class GLWidget : public QOpenGLWidget, public QOpenGLFunctions_3_3_Core
{
    Q_OBJECT

public:
    enum class Projection
    {
        Equirectangular,
        Perspective,
        Fisheye,
    };
    enum class ColorMode
    {
        sRGB,
        ScotopicLuminance,
        PhotopicLuminance,
        XYZChromaticity,
        sRGBlChromaticity,
        sRGBlChromaticityToMax,
    };

private:
    std::unique_ptr<ShowMySky::AtmosphereRenderer> renderer;
    std::unique_ptr<QOpenGLShaderProgram> luminanceToScreenRGB_;
    std::unique_ptr<QOpenGLShaderProgram> glareProgram_;
    QOpenGLTexture ditherPatternTexture_;
    GLuint glareTextures_[2] = {};
    GLuint glareFBOs_[2] = {};
    QString pathToData;
    ToolsWidget* tools;
    GLuint vao_=0, vbo_=0;
    QPoint lastRadianceCapturePosition{-1,-1};
    decltype(::ShowMySky_AtmosphereRenderer_create)* ShowMySky_AtmosphereRenderer_create=nullptr;
    Projection currentProjection_ = Projection::Equirectangular;
    ColorMode currentColorMode_ = ColorMode::sRGB;

    enum class DragMode
    {
        None,
        Sun,
        Camera,
    } dragMode_=DragMode::None;
    int prevMouseX_, prevMouseY_;

public:
    enum class DitheringMode
    {
        Color565,    
        Color666,    
        Color888,    
        Color101010, 
    };
    enum class DitheringMethod
    {
        NoDithering,                
        Bayer,                      
        BlueNoiseTriangleRemapped,  
    };
public:
    explicit GLWidget(QString const& pathToData, ToolsWidget* tools, QWidget* parent=nullptr);
    ~GLWidget();

protected:
    void initializeGL() override;
    void paintGL() override;
    void resizeGL(int w, int h) override;
    void wheelEvent(QWheelEvent* event) override;
    void mouseMoveEvent(QMouseEvent* event) override;
    void mousePressEvent(QMouseEvent* event) override;
    void mouseReleaseEvent(QMouseEvent* event) override;
    bool eventFilter(QObject* object, QEvent* event) override;
    void keyPressEvent(QKeyEvent* event) override;

private:
    void setupBuffers();
    void reloadShaders();
    void stepDataLoading();
    void stepShaderReloading();
    void stepPreparationToDraw(bool emitProgressStatus);
    QVector3D rgbMaxValue() const;
    void makeGlareRenderTarget();
    void makeDitherPatternTexture();
    void updateSpectralRadiance(QPoint const& pixelPos);
    void setDragMode(DragMode mode, int x=0, int y=0) { dragMode_=mode; prevMouseX_=x; prevMouseY_=y; }
    void setFlatSolarSpectrum();
    void resetSolarSpectrum();
    void setBlackBodySolarSpectrum(double temperature);
    void saveScreenshot();
    Projection currentProjection() const { return currentProjection_; }
    ColorMode  currentColorMode () const { return currentColorMode_; }

signals:
    void frameFinished(long long timeInUS);
    void loadProgress(QString const& currentActivity, int stepsDone, int stepsToDo);
};

#endif