Add CFGScene QGraphicsScene taking care of BasicBlock stuff

[frida/frida.git] / src / gui / widgets / CFGScene.hxx

diff --git a/src/gui/widgets/CFGScene.hxx b/src/gui/widgets/CFGScene.hxx
new file mode 100644 (file)
index 0000000..6baa618
--- /dev/null
+++ b/src/gui/widgets/CFGScene.hxx
@@ -0,0 +1,30 @@
+#ifndef INCLUDE__CFGScene_hxx
+#define INCLUDE__CFGScene_hxx
+
+#include "gui/qt.hxx"
+#include "BasicBlockWidget.hxx"
+
+#include <vector>
+
+class CFGScene : public QGraphicsScene {
+public:
+    CFGScene(QWidget * parent = 0)
+        : QGraphicsScene(parent) {}
+
+    // Take special care when adding a BasicBlock to the scene as we
+    // need to draw arrows for it later on
+    void addItem(BasicBlockWidget* block) {
+        widgets.push_back(block);
+        QGraphicsScene::addItem(block);
+    }
+
+    virtual void drawBackground(QPainter* painter, const QRectF & rect);
+private:
+    std::vector<BasicBlockWidget*> widgets;
+
+    void drawLine(QPainter* painter, QPointF from, QPointF to, bool left = true);
+
+    void spaceWidgets();
+};
+
+#endif