Change from QListWidget to QTreeWidget

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

diff --git a/src/gui/Mainwindow.hxx b/src/gui/Mainwindow.hxx
index 7084e2fec3c6995adb2256d7d208b4db4b6c08f2..a3bd4bdfa88ff354c0ee09a557ceb4d6d8c668c3 100644 (file)
--- a/src/gui/Mainwindow.hxx
+++ b/src/gui/Mainwindow.hxx
@@ -3,26 +3,31 @@
 
 #include <memory>
 #include <map>
-#include <set>
 
 #include <QTextEdit>
 #include <QPushButton>
 #include <QMainWindow>
-#include <QListWidget>
+#include <QTreeWidget>
 #include <QStackedWidget>
 
 #include <log4cxx/logger.h>
 
-#include "disassembler/Disassembler.hxx"
-#include "widgets/BasicBlockWidget.hxx"
-#include "widgets/ScriptingDock.hxx"
-#include "core/InformationManager.hxx"
+class Disassembler;
+class Function;
+class InformationManager;
+class FunctionWidget;
+class BasicBlockWidget;
+class ScriptingDock;
 
 class Mainwindow : public QMainWindow {
        Q_OBJECT
 public:
        Mainwindow(InformationManager* mgr);
 
+public Q_SLOTS:
+       void switchMainPlaneToAddress(uint64_t);
+       void requestNewFunctionByAddress(uint64_t address);
+
 private:
        void addFunction(Function* fun);
        void setGlobalHotkeys();
@@ -32,31 +37,31 @@ private:
        QMenu *fileMenu;
 
        QTabWidget * tabwidget;
-       QListWidget * listWidget;
+       QTreeWidget * listWidget;
        QStackedWidget * stackedWidget;
        QDockWidget * dockWidget;
        ScriptingDock * scripting;
 
        QAction *exitAction;
        QAction *openAction;
+       QAction *saveAction;
 
        std::map<uint64_t, BasicBlockWidget*> blocks;
-       std::map<QListWidgetItem*, QWidget*> objects_list;
-       std::map<uint64_t, QListWidgetItem*> objects_list_by_address;
-       std::set<Function*> functions;
-       InformationManager* manager;
+       std::map<uint64_t, Function*> functions;
+       std::map<QTreeWidgetItem*, FunctionWidget*> objects_list;
+       std::map<uint64_t, QTreeWidgetItem*> objects_list_by_address;
 
+       InformationManager* manager;
        log4cxx::LoggerPtr logger;
-public Q_SLOTS:
-       void switchMainPlaneToAddress(uint64_t);
-       void requestNewFunctionByAddress(uint64_t address);
+
 private Q_SLOTS:
        void quit();
        void open();
-       void switchMainPlane(int);
+       void save();
+       void switchMainPlane(QTreeWidgetItem* item);
        void showListContextMenu(const QPoint&);
        void requestNewFunction();
-       void renameFunction(QListWidgetItem * item);
+       void renameFunction(Function* function);
 };
 
 #endif /* INCLUDE__Mainwindow_hxx_ */