From 78fc76b12da21dae9be6c57d92e08691a533836f Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Tue, 29 Jan 2019 19:52:52 -0500
Subject: Prepare API for Python UI bindings

---
 flowgraphnode.cpp | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'flowgraphnode.cpp')

diff --git a/flowgraphnode.cpp b/flowgraphnode.cpp
index 9d4ed631..0101d716 100644
--- a/flowgraphnode.cpp
+++ b/flowgraphnode.cpp
@@ -40,6 +40,15 @@ FlowGraphNode::FlowGraphNode(BNFlowGraphNode* node)
 }
 
 
+Ref<FlowGraph> FlowGraphNode::GetGraph() const
+{
+	BNFlowGraph* graph = BNGetFlowGraphNodeOwner(m_object);
+	if (!graph)
+		return nullptr;
+	return new CoreFlowGraph(graph);
+}
+
+
 Ref<BasicBlock> FlowGraphNode::GetBasicBlock() const
 {
 	BNBasicBlock* block = BNGetFlowGraphBasicBlock(m_object);
-- 
cgit v1.3.1