graphcolorizer.h

#ifndef GRAPHCOLORIZER_H
#define GRAPHCOLORIZER_H

#include <QObject>
#include "dataset.h"

class GraphColorizer : public QObject
{
    Q_OBJECT

public:
    enum ColoringAttribute { OnsetTime = 0, Volatility, SIZE_OF_ENUM };

    static const char* const coloringAttribute_str[ColoringAttribute::SIZE_OF_ENUM];

    GraphColorizer(QObject *parent);
    ~GraphColorizer();

    void SetColors(QColor const beginColor, QColor const endColor);

    void Set2DColoring(bool enable);

    void Colorize(DataSet& const dataset, ColoringAttribute attr);
private:
    float calculateRatio(DataSet& const dataset, DataItem& const dataitem, ColoringAttribute const attribute);

    QColor mBeginColor;
    QColor mEndColor;
    bool m2DColoring;
};

#endif // GRAPHCOLORIZER_H