def create_valve_collection(net, valves=None, size=5., junction_geodata=None, color='k',
infofunc=None, picker=False, fill_closed=True,
respect_valves=False, **kwargs):
"""
Creates a matplotlib patch collection of pandapipes junction-junction valves. Valves are
plotted in the center between two junctions with a "helper" line (dashed and thin) being drawn
between the junctions as well.
:param net: The pandapipes network
:type net: pandapipesNet
:param valves: The valves for which the collections are created. If None, all valves which have\
entries in the respective junction geodata will be plotted.
:type valves: list, default None
:param size: Patch size
:type size: float, default 5.
:param junction_geodata: Coordinates to use for plotting. If None, net["junction_geodata"] is used.
:type junction_geodata: pandas.DataFrame, default None
:param colors: Color or list of colors for every valve
:type colors: iterable, float, default None
:param infofunc: infofunction for the patch element
:type infofunc: function, default None
:param picker: Picker argument passed to the patch collection
:type picker: bool, default False
:param fill_closed: If True, valves with parameter opened == False will be filled and those\
with opened == True will have a white facecolor. Vice versa if False.
:type fill_closed: bool, default True
:param kwargs: Keyword arguments are passed to the patch function
:return: lc - line collection, pc - patch collection
"""
valves = get_index_array(
valves, net.valve[net.valve.opened.values].index if respect_valves else net.valve.index)
valve_table = net.valve.loc[valves]
coords, valves_with_geo = coords_from_node_geodata(
valves, valve_table.from_junction.values, valve_table.to_junction.values,
junction_geodata if junction_geodata is not None else net["junction_geodata"], "valve",
"Junction")
if len(valves_with_geo) == 0:
return None
linewidths = kwargs.pop("linewidths", 2.)
linewidths = kwargs.pop("linewidth", linewidths)
linewidths = kwargs.pop("lw", linewidths)
infos = list(np.repeat([infofunc(i) for i in range(len(valves_with_geo))], 2)) \
if infofunc is not None else []
filled = valve_table["opened"].values
if fill_closed:
filled = ~filled
lc, pc = _create_complex_branch_collection(coords, valve_patches, size, infos,
picker=picker, linewidths=linewidths, filled=filled,
patch_facecolor=color, line_color=color,
**kwargs)
return lc, pc
def create_heat_exchanger_collection(net, heat_ex=None, size=5., junction_geodata=None,
infofunc=None, picker=False, **kwargs):
"""
Creates a matplotlib patch collection of pandapipes junction-junction heat_exchangers.
Heat_exchangers are plotted in the center between two junctions with a "helper" line
(dashed and thin) being drawn between the junctions as well.
:param net: The pandapipes network
:type net: pandapipesNet
:param size: Patch size
:type size: float, default 2.
:param helper_line_style: Line style of the "helper" line being plotted between two junctions
connected by a junction-junction heat_exchanger
:type helper_line_style: str, default ":"
:param helper_line_size: Line width of the "helper" line being plotted between two junctions
connected by a junction-junction heat_exchanger
:type helper_line_size: float, default 1.
:param helper_line_color: Line color of the "helper" line being plotted between two junctions
connected by a junction-junction valve
:type helper_line_color: str, default "gray"
:param orientation: Orientation of heat_exchanger collection. pi is directed downwards,
increasing values lead to clockwise direction changes.
:type orientation: float, default np.pi/2
:param kwargs: Keyword arguments are passed to the patch function
:return: heat_exchanger, helper_lines
:rtype: tuple of patch collections
"""
heat_ex = get_index_array(heat_ex, net.heat_exchanger.index)
hex_table = net.heat_exchanger.loc[heat_ex]
coords, hex_with_geo = coords_from_node_geodata(
heat_ex, hex_table.from_junction.values, hex_table.to_junction.values,
junction_geodata if junction_geodata is not None else net["junction_geodata"],
"heat_exchanger", "Junction")
if len(hex_with_geo) == 0:
return None
colors = kwargs.pop("color", "k")
linewidths = kwargs.pop("linewidths", 2.)
linewidths = kwargs.pop("linewidth", linewidths)
linewidths = kwargs.pop("lw", linewidths)
patch_edgecolor = kwargs.pop("patch_edgecolor", colors)
line_color = kwargs.pop("line_color", colors)
infos = list(np.repeat([infofunc(i) for i in range(len(hex_with_geo))], 2)) \
if infofunc is not None else []
lc, pc = _create_complex_branch_collection(
coords, heat_exchanger_patches, size, infos, picker=picker, linewidths=linewidths,
patch_edgecolor=patch_edgecolor, line_color=line_color, **kwargs)
return lc, pc
def create_pump_collection(net, pumps=None, table_name='pump', size=5., junction_geodata=None,
infofunc=None, picker=False, **kwargs):
"""
Creates a matplotlib patch collection of pandapipes junction-junction valves. Valves are
plotted in the center between two junctions with a "helper" line (dashed and thin) being drawn
between the junctions as well.
:param net: The pandapipes network
:type net: pandapipesNet
:param pumps: The pumps for which the collections are created. If None, all pumps which have\
entries in the respective junction geodata will be plotted.
:type pumps: list, default None
:param table_name: Name of the pump table from which to get the data.
:type table_name: str, default 'pump'
:param size: Patch size
:type size: float, default 5.
:param junction_geodata: Coordinates to use for plotting. If None, net["junction_geodata"] is used.
:type junction_geodata: pandas.DataFrame, default None
:param infofunc: infofunction for the patch element
:type infofunc: function, default None
:param picker: Picker argument passed to the patch collection
:type picker: bool, default False
:param kwargs: Keyword arguments are passed to the patch function
:return: lc - line collection, pc - patch collection
"""
pumps = get_index_array(pumps, net[table_name].index)
pump_table = net[table_name].loc[pumps]
coords, pumps_with_geo = coords_from_node_geodata(
pumps, pump_table.from_junction.values, pump_table.to_junction.values,
junction_geodata if junction_geodata is not None else net["junction_geodata"], "pump",
"Junction")
if len(pumps_with_geo) == 0:
return None
colors = kwargs.pop("color", "k")
linewidths = kwargs.pop("linewidths", 2.)
linewidths = kwargs.pop("linewidth", linewidths)
linewidths = kwargs.pop("lw", linewidths)
patch_edgecolor = kwargs.pop("patch_edgecolor", colors)
line_color = kwargs.pop("line_color", colors)
infos = list(np.repeat([infofunc(i) for i in range(len(pumps_with_geo))], 2)) \
if infofunc is not None else []
lc, pc = _create_complex_branch_collection(
coords, pump_patches, size, infos, picker=picker, linewidths=linewidths,
patch_edgecolor=patch_edgecolor, line_color=line_color, **kwargs)
return lc, pc