1 files changed, 21 insertions, 5 deletions
diff --git a/include/ui/painter.h b/include/ui/painter.h
index df62776..0a2accc 100644
--- a/include/ui/painter.h
+++ b/include/ui/painter.h
@@ -1,11 +1,15 @@
 #ifndef PAINTER_H
 #define PAINTER_H
 
-#include <QWidget>
+#include "diagram/statement.h"
+#include "diagram/structogram.h"
 
-namespace Ui {
-class Painter;
-}
+#include <QWidget>
+#include <QFont>
+#include <QFontMetrics>
+#include <QPaintEvent>
+#include <QPainter>
+#include <QRect>
 
 class Painter : public QWidget
 {
@@ -15,8 +19,20 @@ public:
     explicit Painter(QWidget *parent = 0);
     ~Painter();
 
+    void structogram(samb::Structogram **structogram) { _structogram = structogram; }
+    const samb::Structogram structogram() const { return **_structogram; }
+
+protected:
+    void paintEvent(QPaintEvent *event);
+
 private:
-    Ui::Painter *_ui;
+    samb::Structogram **_structogram = nullptr;
+    QFont _font;
+    QFontMetrics _fontMetrics;
+
+    int _margin = 10;
+
+    void drawStatement(QPainter &qp, samb::Statement &statement, const QRect &rect);
 };
 
 #endif // PAINTER_H