1 files changed, 20 insertions, 1 deletions
diff --git a/tikzit/src/gui/tikzscene.h b/tikzit/src/gui/tikzscene.h
index dd3cba7..c7039e6 100644
--- a/tikzit/src/gui/tikzscene.h
+++ b/tikzit/src/gui/tikzscene.h
@@ -1,13 +1,32 @@
 #ifndef TIKZSCENE_H
 #define TIKZSCENE_H
 
+#include "graph.h"
+#include "nodeitem.h"
+
 #include <QWidget>
 #include <QGraphicsScene>
+#include <QPainter>
+#include <QRectF>
+#include <QVector>
+#include <QGraphicsEllipseItem>
 
 class TikzScene : public QGraphicsScene
 {
+    Q_OBJECT
 public:
-    TikzScene(QObject *parent);
+    TikzScene(Graph *graph, QObject *parent);
+    Graph *graph() const;
+    void setGraph(Graph *graph);
+public slots:
+    void graphReplaced();
+
+private:
+    Graph *_graph;
+    QVector<NodeItem*> nodeItems;
+
+protected:
+    void drawBackground(QPainter *painter, const QRectF &rect);
 };
 
 #endif // TIKZSCENE_H