TransferFunctionEditor.h

Go to the documentation of this file.

#ifndef GlobeEngine_TransferFunctionEditor_h
#define GlobeEngine_TransferFunctionEditor_h

#include "OpenGL_Includes.h"
#include "VBOVertex.h"

#include <QDebug>
#include <QObject>
#include <QMouseEvent>
#include <QPaintEvent>
#include <QPainter>
#include <QColorDialog>
#include <QLabel>
#include <QFileDialog>
#include <QPushButton>
#include <QGroupBox>
#include <QRadioButton>
#include <QHBoxLayout>
#include <QEasingCurve>
#include <QComboBox>

namespace geQT {

    struct TransferFunctionProperties {
        QMap<int, int> points;
        QMap<int, QColor> colors;
        QMap<int, int> gradients;
        QMap<int, QPointF> controls1;
        QMap<int, QPointF> controls2;
    };

    class TransferFunctionEditor : public QWidget
    {
        Q_OBJECT

    public:
        TransferFunctionEditor(QWidget* _parent, QColorDialog* _colorDialog);
        ~TransferFunctionEditor(){};

        unsigned char* getTextureData();

        void setHistogramData(std::vector<std::vector<std::vector<long>>> histograms, std::vector<std::vector<float>> minimumHistValue, std::vector<std::vector<float>> maximumHistValue);
        void initialize(int _numberOfTimesteps);

        void setSomething();

        //std::vector<ge::Color4ub> getColors();
        std::vector<float> getHeights();

        void setCurrent(int _currentAttribute, int _currentTimestep, int _currentMarching) {
            this->activePoint = -1;
            this->currentAttribute = _currentAttribute;
            this->currentTimestep = _currentTimestep;
            this->currentMarching = _currentMarching;
        }

    public slots:
        void test();

        void save();
        void load();
        void loadFromFile(std::string _file);
        void setInterpolationType();


    signals:
        void transferFunctionChanged();

    private:
        void mousePressEvent(QMouseEvent *e)    Q_DECL_OVERRIDE;
        void mouseMoveEvent(QMouseEvent *e)     Q_DECL_OVERRIDE;
        void mouseReleaseEvent(QMouseEvent *e)  Q_DECL_OVERRIDE;

        void paintEvent(QPaintEvent *e)         Q_DECL_OVERRIDE;

        void keepMouseWithinScope(QPoint pos);
        QPoint keepControlPointWithinScope(int left, int up, int right, int down);

        void setGradientType(int _gradientType);
        int getGradientType();

        bool multiplied = false;

        QColor interpolate(QColor color1, QColor color2, float t);

        std::vector<std::vector<TransferFunctionProperties>> tempStore;
        std::vector<std::vector<float>> minimumHistValues;
        std::vector<std::vector<float>> maximumHistValues;
        std::vector<std::vector<std::vector<long>>> histograms;

        int currentAttribute = 0;
        int currentTimestep = 0;
        int currentMarching = 0;

        // user intentions
        QPoint currentPosition;
        QPoint lastPosition; // make int only?!?!

        // current variables
        bool interacting = false;
        bool leftClick;
        bool clickedStartPoint;
        int activePoint;

        // ui settings
        int pointRadius = 5;
        const int border = 20;
        QColorDialog* colorDialog;

        // histogram
        QList<float> list;
        const int binSize = 20;

        QVector<float> binLimits;

        int movedControl1;
        int movedControl2;

        qreal resolution;
        int textureWidth = 1000;

        unsigned char* data;

        QRadioButton* linear;
        QRadioButton* bezier;
        QRadioButton* staircase;
        QComboBox* colorInterpolation;
    };
}
#endif // TRANSFERFUNCTIONEDITORTWO_H