Add flow graph options for additional UI features

1 files changed, 27 insertions, 0 deletions

diff --git a/flowgraph.cpp b/flowgraph.cpp
index df1474cd..64bfe364 100644
--- a/flowgraph.cpp
+++ b/flowgraph.cpp
@@ -148,12 +148,27 @@ Ref<Function> FlowGraph::GetFunction() const
 }
 
 
+Ref<BinaryView> FlowGraph::GetView() const
+{
+	BNBinaryView* view = BNGetViewForFlowGraph(m_object);
+	if (!view)
+		return nullptr;
+	return new BinaryView(view);
+}
+
+
 void FlowGraph::SetFunction(Function* func)
 {
 	BNSetFunctionForFlowGraph(m_object, func ? func->GetObject() : nullptr);
 }
 
 
+void FlowGraph::SetView(BinaryView* view)
+{
+	BNSetViewForFlowGraph(m_object, view ? view->GetObject() : nullptr);
+}
+
+
 int FlowGraph::GetHorizontalNodeMargin() const
 {
 	return BNGetHorizontalFlowGraphNodeMargin(m_object);
@@ -344,6 +359,18 @@ Ref<FlowGraph> FlowGraph::Update()
 }
 
 
+void FlowGraph::SetOption(BNFlowGraphOption option, bool value)
+{
+	BNSetFlowGraphOption(m_object, option, value);
+}
+
+
+bool FlowGraph::IsOptionSet(BNFlowGraphOption option)
+{
+	return BNIsFlowGraphOptionSet(m_object, option);
+}
+
+
 CoreFlowGraph::CoreFlowGraph(BNFlowGraph* graph): FlowGraph(graph)
 {
 }