Mapping voltages

You can either extend BaseVoltageMapper class or use existing phase mapper classes to map voltage to nodes.

Let’s use transformer voltage mapper class to map voltages to nodes using transformer voltages.

from shift import TransformerVoltageMapper, TransformerVoltageModel, add_voltage_mapper_to_plot
from gdm.quantities import PositiveVoltage
from gdm import DistributionTransformer

voltage_mapper = TransformerVoltageMapper(
    new_graph,
    xfmr_voltage=[
        TransformerVoltageModel(
            name=el.name,
            voltages=[PositiveVoltage(7.2, "kilovolt"), PositiveVoltage(120, "volt")],
        )
        for _, _, el in new_graph.get_edges()
        if el.edge_type is DistributionTransformer
    ],
)

add_voltage_mapper_to_plot(voltage_mapper, plot_manager)
plot_manager.show()