1 files changed, 7 insertions, 2 deletions
diff --git a/scatterplot.h b/scatterplot.h
index f38ce78..3f2fee6 100644
--- a/scatterplot.h
+++ b/scatterplot.h
@@ -5,22 +5,27 @@
 #include <vector>
 #include <QQuickItem>
 
+#include "colorscale.h"
+
 class Scatterplot : public QQuickItem
 {
     Q_OBJECT
 public:
-    Scatterplot();
+    Scatterplot(QQuickItem *parent = 0);
     ~Scatterplot();
 
-    QSGNode *updatePaintNode(QSGNode *oldNode, UpdatePaintNodeData *);
     void setData(const arma::mat &data);
 
 signals:
 
 public slots:
 
+protected:
+    QSGNode *updatePaintNode(QSGNode *oldNode, UpdatePaintNodeData *);
+
 private:
     arma::mat m_data;
+    ColorScale m_colorScale;
 };
 
 #endif // SCATTERPLOT_H