Add ability to update custom flow graphs

1 files changed, 30 insertions, 0 deletions

diff --git a/flowgraph.cpp b/flowgraph.cpp
index 4bacee81..c779db0c 100644
--- a/flowgraph.cpp
+++ b/flowgraph.cpp
@@ -78,6 +78,16 @@ void FlowGraph::CompleteLayoutCallback(void* ctxt)
 }
 
 
+BNFlowGraph* FlowGraph::UpdateCallback(void* ctxt)
+{
+	FlowGraph* graph = (FlowGraph*)ctxt;
+	Ref<FlowGraph> result = graph->Update();
+	if (!result)
+		return nullptr;
+	return BNNewFlowGraphReference(result->GetGraphObject());
+}
+
+
 void FlowGraph::FinishPrepareForLayout()
 {
 	BNFinishPrepareForLayout(m_graph);
@@ -313,3 +323,23 @@ void FlowGraph::Show(const string& title)
 {
 	ShowGraphReport(title, this);
 }
+
+
+Ref<FlowGraph> FlowGraph::Update()
+{
+	return nullptr;
+}
+
+
+CoreFlowGraph::CoreFlowGraph(BNFlowGraph* graph): FlowGraph(graph)
+{
+}
+
+
+Ref<FlowGraph> CoreFlowGraph::Update()
+{
+	BNFlowGraph* graph = BNUpdateFlowGraph(GetGraphObject());
+	if (!graph)
+		return nullptr;
+	return new CoreFlowGraph(graph);
+}