From eb0aed7c4b15c37c46b15b5d9e32ceb93f2e1e95 Mon Sep 17 00:00:00 2001
From: Peter LaFosse <peter@vector35.com>
Date: Tue, 1 Sep 2020 12:43:10 -0400
Subject: Add support for custom flowgraph edges

---
 flowgraphnode.cpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'flowgraphnode.cpp')

diff --git a/flowgraphnode.cpp b/flowgraphnode.cpp
index 3a9359de..344dd8a3 100644
--- a/flowgraphnode.cpp
+++ b/flowgraphnode.cpp
@@ -155,6 +155,9 @@ const vector<FlowGraphEdge>& FlowGraphNode::GetOutgoingEdges()
 		edge.target = edges[i].target ? new FlowGraphNode(BNNewFlowGraphNodeReference(edges[i].target)) : nullptr;
 		edge.points.insert(edge.points.begin(), &edges[i].points[0], &edges[i].points[edges[i].pointCount]);
 		edge.backEdge = edges[i].backEdge;
+		edge.style.color = edges[i].style.color;
+		edge.style.width = edges[i].style.width;
+		edge.style.style = edges[i].style.style;
 		result.push_back(edge);
 	}
 
@@ -192,9 +195,9 @@ const vector<FlowGraphEdge>& FlowGraphNode::GetIncomingEdges()
 }
 
 
-void FlowGraphNode::AddOutgoingEdge(BNBranchType type, FlowGraphNode* target)
+void FlowGraphNode::AddOutgoingEdge(BNBranchType type, FlowGraphNode* target, BNEdgeStyle edgeStyle)
 {
-	BNAddFlowGraphNodeOutgoingEdge(m_object, type, target->GetObject());
+	BNAddFlowGraphNodeOutgoingEdge(m_object, type, target->GetObject(), edgeStyle);
 	m_cachedEdges.clear();
 	m_cachedEdgesValid = false;
 }
-- 
cgit v1.3.1