summaryrefslogtreecommitdiff
path: root/flowgraphnode.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'flowgraphnode.cpp')
-rw-r--r--flowgraphnode.cpp7
1 files changed, 5 insertions, 2 deletions
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;
}