From dfe5d28f0aaee75ccad8c733e23879676ed77c37 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Wed, 11 Jul 2018 20:32:22 -0400
Subject: Add APIs for subclassing flow graphs

---
 flowgraph.cpp | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)

(limited to 'flowgraph.cpp')

diff --git a/flowgraph.cpp b/flowgraph.cpp
index e557f540..4bacee81 100644
--- a/flowgraph.cpp
+++ b/flowgraph.cpp
@@ -24,6 +24,17 @@ using namespace BinaryNinja;
 using namespace std;
 
 
+FlowGraph::FlowGraph()
+{
+	BNCustomFlowGraph callbacks;
+	callbacks.context = this;
+	callbacks.prepareForLayout = PrepareForLayoutCallback;
+	callbacks.populateNodes = PopulateNodesCallback;
+	callbacks.completeLayout = CompleteLayoutCallback;
+	m_graph = BNCreateCustomFlowGraph(&callbacks);
+}
+
+
 FlowGraph::FlowGraph(BNFlowGraph* graph): m_graph(graph)
 {
 }
@@ -46,6 +57,49 @@ void FlowGraph::CompleteCallback(void* ctxt)
 }
 
 
+void FlowGraph::PrepareForLayoutCallback(void* ctxt)
+{
+	FlowGraph* graph = (FlowGraph*)ctxt;
+	graph->PrepareForLayout();
+}
+
+
+void FlowGraph::PopulateNodesCallback(void* ctxt)
+{
+	FlowGraph* graph = (FlowGraph*)ctxt;
+	graph->PopulateNodes();
+}
+
+
+void FlowGraph::CompleteLayoutCallback(void* ctxt)
+{
+	FlowGraph* graph = (FlowGraph*)ctxt;
+	graph->CompleteLayout();
+}
+
+
+void FlowGraph::FinishPrepareForLayout()
+{
+	BNFinishPrepareForLayout(m_graph);
+}
+
+
+void FlowGraph::PrepareForLayout()
+{
+	FinishPrepareForLayout();
+}
+
+
+void FlowGraph::PopulateNodes()
+{
+}
+
+
+void FlowGraph::CompleteLayout()
+{
+}
+
+
 Ref<Function> FlowGraph::GetFunction() const
 {
 	BNFunction* func = BNGetFunctionForFlowGraph(m_graph);
-- 
cgit v1.3.1