Add MenuHelper for better menu behavior.

1 files changed, 3 insertions, 5 deletions

diff --git a/ui/addressindicator.h b/ui/addressindicator.h
index 3ad40638..fd9326fe 100644
--- a/ui/addressindicator.h
+++ b/ui/addressindicator.h
@@ -5,12 +5,12 @@
 #include "menus.h"
 #include "uicontext.h"
 
-class BINARYNINJAUIAPI AddressIndicator: public QLabel
+
+class BINARYNINJAUIAPI AddressIndicator: public MenuHelper
 {
 	Q_OBJECT
 
 	uint64_t m_begin, m_end;
-	ContextMenuManager m_contextMenu;
 
 public:
 	AddressIndicator(QWidget* parent);
@@ -18,7 +18,5 @@ public:
 	void setOffsets(uint64_t begin, uint64_t end);
 
 protected:
-	virtual void mousePressEvent(QMouseEvent* event);
-	virtual void enterEvent(QEvent* event);
-	virtual void leaveEvent(QEvent* event);
+	virtual void showMenu();
 };