Updating Branch Type

In order to use right equipment, you need to update the branch types in the DistributionGraph.

Let’s say, you want to use MatrixImpedanceBranch you can update branches in graph with following snippet.

from shift import DistributionGraph, add_distribution_graph_to_plot, PlotManager
from gdm import DistributionBranchBase, MatrixImpedanceBranch

new_graph = DistributionGraph()

for node in graph.get_nodes():
    new_graph.add_node(node)

for from_node, to_node, edge in graph.get_edges():
    if edge.edge_type == DistributionBranchBase:
        edge.edge_type = MatrixImpedanceBranch
    new_graph.add_edge(from_node, to_node, edge_data=edge)

add_distribution_graph_to_plot(new_graph, plot_manager)
plot_manager.show()