Draw edges between BasicBlockWidgets while handling loops

[frida/frida.git] / src / gui / Mainwindow.hxx

diff --git a/src/gui/Mainwindow.hxx b/src/gui/Mainwindow.hxx
index 3892ebd61ad29572a9517f9a421a9f0c35f12d8b..a0ba231884b485ab53319196d050eab385009451 100644 (file)
--- a/src/gui/Mainwindow.hxx
+++ b/src/gui/Mainwindow.hxx
@@ -1,4 +1,5 @@
 #include <memory>
+#include <map>
 #include <QTextEdit>
 #include <QPushButton>
 #include <QMainWindow>
@@ -6,6 +7,7 @@
 #include <QStackedWidget>
 
 #include "disassembler/Disassembler.hxx"
+#include "widgets/BasicBlockWidget.hxx"
 
 class Mainwindow : public QMainWindow {
     Q_OBJECT
@@ -14,7 +16,7 @@ public:
 private:
        void openBinary(const std::string& filename);
 
-    void populateSymbolInformation(const std::string& sym);
+    void populateSymbolInformation(Function * fun);
 
     QTextEdit *textEdit;
     QPushButton *quitButton;
@@ -29,6 +31,7 @@ private:
     QAction *openAction;
 
     std::shared_ptr<Disassembler> disassembler;
+       std::map<uint64_t, BasicBlockWidget*> blocks;
 private slots:
     void quit();
     void open();