Add flowgraph API to highlight related instructions.

1 files changed, 5 insertions, 0 deletions

diff --git a/ui/flowgraphwidget.h b/ui/flowgraphwidget.h
index 2c05144d..0a9bf4c1 100644
--- a/ui/flowgraphwidget.h
+++ b/ui/flowgraphwidget.h
@@ -104,6 +104,8 @@ class BINARYNINJAUIAPI FlowGraphWidget: public QAbstractScrollArea, public View,
 	CursorPosition m_cursorPos, m_selectionStartPos;
 	HighlightTokenState m_highlight;
 	bool m_tokenSelection = false;
+	std::set<size_t> m_relatedIndexHighlights;
+	std::set<uint64_t> m_relatedInstructionHighlights;
 
 	ContextMenuManager m_contextMenuManager;
 	QPointer<CommentDialog> m_commentDialog;
@@ -218,6 +220,9 @@ public:
 	bool navigateWithHistoryEntry(FunctionRef func, uint64_t addr, FlowGraphHistoryEntry* entry);
 	void setNavigationTarget(View* target) { m_navigationTarget = target; }
 
+	virtual void setRelatedIndexHighlights(const std::set<size_t>& related);
+	virtual void setRelatedInstructionHighlights(const std::set<uint64_t>& related);
+
 	virtual void zoom(bool direction);
 	virtual void zoomActual();
 	virtual bool event(QEvent* event) override;