Add file-backed only mode toggle to feature map context menu.

2 files changed, 6 insertions, 5 deletions

diff --git a/ui/featuremap.h b/ui/featuremap.h
index e01b5f50..e023cff4 100644
--- a/ui/featuremap.h
+++ b/ui/featuremap.h
@@ -6,6 +6,7 @@
 #include <QtWidgets/QMenu>
 #include <QtGui/QPainter>
 #include <QtCore/QRect>
+#include <QtCore/QPointer>
 #include <QtCore/QTimer>
 #include <QtCore/QVector>
 #include <QtWidgets/QWidget>
@@ -54,7 +55,7 @@ class BINARYNINJAUIAPI FeatureMap: public QWidget, public BinaryNinja::BinaryDat
 	{
 		std::mutex m_mutex;
 		bool m_valid;
-		FeatureMap* m_featureMap;
+		QPointer<FeatureMap> m_featureMap;
 
 	public:
 		BackgroundRefresh(FeatureMap* featureMap);
@@ -62,7 +63,7 @@ class BINARYNINJAUIAPI FeatureMap: public QWidget, public BinaryNinja::BinaryDat
 		void abort();
 	};
 
-	BinaryNinja::Ref<BackgroundRefresh> m_backgroundRefresh;
+	BinaryNinja::Ref<BackgroundRefresh> m_backgroundRefresh = nullptr;
 
 public:
 	FeatureMap(ViewFrame* frame, BinaryViewRef data);
diff --git a/ui/viewframe.h b/ui/viewframe.h
index fefc72d2..a4c3940d 100644
--- a/ui/viewframe.h
+++ b/ui/viewframe.h
@@ -211,7 +211,7 @@ public:
 	bool isGraphViewPreferred() { return m_graphViewPreferred; }
 	void setGraphViewPreferred(bool graphViewPreferred) { m_graphViewPreferred = graphViewPreferred; }
 	void focus();
-	void closeFeatureMap();
+	void closeFeatureMap(bool recreate = false);
 	QWidget* createFeatureMap();
 	void refreshFeatureMap();
 
@@ -262,10 +262,10 @@ public Q_SLOTS:
 	virtual void compile();
 
 private Q_SLOTS:
-	void deleteFeatureMap();
+	void deleteFeatureMap(bool recreate);
 
 Q_SIGNALS:
-	void notifyCloseFeatureMap();
+	void notifyCloseFeatureMap(bool recreate);
 	void notifyViewChanged(ViewFrame* frame);
 };