Properly rename function

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

diff --git a/src/gui/Mainwindow.hxx b/src/gui/Mainwindow.hxx
index 633c6478a8ce28002f85b47c6c2455584fc04ad4..ba0d8b4944529138ff78fbf2e9b995133c677dab 100644 (file)
--- a/src/gui/Mainwindow.hxx
+++ b/src/gui/Mainwindow.hxx
@@ -3,7 +3,6 @@
 
 #include <memory>
 #include <map>
-#include <set>
 
 #include <QTextEdit>
 #include <QPushButton>
@@ -16,6 +15,7 @@
 class Disassembler;
 class Function;
 class InformationManager;
+class FunctionWidget;
 class BasicBlockWidget;
 class ScriptingDock;
 
@@ -47,11 +47,11 @@ private:
        QAction *saveAction;
 
        std::map<uint64_t, BasicBlockWidget*> blocks;
-       std::map<QListWidgetItem*, QWidget*> objects_list;
+       std::map<uint64_t, Function*> functions;
+       std::map<QListWidgetItem*, FunctionWidget*> objects_list;
        std::map<uint64_t, QListWidgetItem*> objects_list_by_address;
-       std::set<Function*> functions;
-       InformationManager* manager;
 
+       InformationManager* manager;
        log4cxx::LoggerPtr logger;
 
 private Q_SLOTS:
@@ -61,7 +61,7 @@ private Q_SLOTS:
        void switchMainPlane(int);
        void showListContextMenu(const QPoint&);
        void requestNewFunction();
-       void renameFunction(QListWidgetItem * item);
+       void renameFunction(Function* function);
 };
 
 #endif /* INCLUDE__Mainwindow_hxx_ */