def visualize_th_solution(setup,
graph,
solutions,
th_problem,
withsymbols=False,
withnumbers=True):
'''code by M.Sc. Thomas Licklederer, Technical University of Munich, MSE, all
rights reserved DESCRIPTION TO BE ADDED'''
savepath = './results/'
saveformat = 'png' # jpg, png, eps, svgd
# combine the solution values with the correct edges
dotV_soldict = solutions['dotV']
Deltap_soldict = solutions['Deltap']
DeltaT_soldict = solutions['DeltaT']
T_soldict = solutions['T']
Q_soldict = solutions['Q_trnsf2']
mu_dict = {P: setup.scenario[P]['mu'] for P in setup.PSM}
u_dict = {P: setup.scenario[P]['u'] for P in setup.PSM}
kappa_dict = {P: setup.scenario[P]['kappa'] for P in setup.PSM}
T_sec_dict = {P: setup.scenario[P]['T_sec_in_degree'] for P in setup.PSM}
dotV_sec_dict = {P: setup.scenario[P]['dotV_sec_in'] for P in setup.PSM}
Edges_Temps_dict = th_problem.general_stuff['Edges_Temps_dict']
# assign the solution values to the Graph
for edge in setup.e_vec:
graph.G.edges[edge]['DeltaT_sol'] = DeltaT_soldict[edge]
for node in setup.v_vec:
graph.G.nodes[node]['T_sol'] = {}
for temp in T_soldict:
if temp[0] == node:
graph.G.nodes[node]['T_sol'][temp] = T_soldict[temp]
## initialize new Graph that represents the solution
graph.Gsol_th = nx.DiGraph()
graph.Gsol_th.add_nodes_from(setup.v_vec)
myedgelabels = {}
for i in range(len(setup.e_vec)):
edge = setup.e_vec[i]
if np.sign(dotV_soldict[edge]) == -1:
graph.Gsol_th.add_edge(edge[1], edge[0])
graph.Gsol_th.edges[(
edge[1],
edge[0])]['DeltaT'] = -DeltaT_soldict[(edge[0], edge[1])]
myedgelabels[(edge[1], edge[0])] = []
myedgelabels[(edge[1], edge[0])].append(
'{0:6.4f} K'.format(-DeltaT_soldict[(edge[0], edge[1])]))
elif np.sign(dotV_soldict[edge]) == 1:
graph.Gsol_th.add_edge(edge[0], edge[1])
graph.Gsol_th.edges[(
edge[0], edge[1])]['DeltaT'] = DeltaT_soldict[(edge[0],
edge[1])]
myedgelabels[(edge[0], edge[1])] = []
myedgelabels[(edge[0], edge[1])].append('{0:6.4f} K'.format(
DeltaT_soldict[(edge[0], edge[1])]))
elif np.sign(dotV_soldict[edge]) == 0:
graph.Gsol_th.add_edge(edge[0], edge[1])
graph.Gsol_th.edges[(
edge[0], edge[1])]['DeltaT'] = DeltaT_soldict[(edge[0],
edge[1])]
myedgelabels[(edge[0], edge[1])] = []
myedgelabels[(edge[0], edge[1])].append('{0:6.4f} K'.format(
DeltaT_soldict[(edge[0], edge[1])]))
else:
raise ValueError('')
# plot graph
myfontsize = 16
myfigsize = [2 * 21.0 / 2.54, 2 * 12.98 / 2.54]
myarrowstyle = mpltlib.patches.ArrowStyle.CurveFilledB(
head_length=0.6, head_width=0.4) #CurveB
fig1 = plt.figure(num='thermal solution', figsize=myfigsize)
plt.title('thermal solution', {'fontsize': 30, 'fontweight': 'bold'})
for node in graph.Gsol_th.nodes():
if 'h' in node:
graph.Gsol_th.nodes[node]['color'] = 'r'
if 'c' in node:
graph.Gsol_th.nodes[node]['color'] = 'b'
nodes_colors = {
node: graph.Gsol_th.nodes[node]['color']
for node in graph.Gsol_th.nodes()
}
colorslist = list(nodes_colors.values())
nx.drawing.nx_pylab.draw(graph.Gsol_th,
pos=setup.coordinates,
fontsize='xx-large',
with_labels=True,
node_color=colorslist,
arrowsize=15,
arrowstyle=myarrowstyle,
node_size=700,
width=3,
font_weight='bold',
alpha=1)
xcoords = [setup.coordinates[node][0] for node in setup.coordinates]
ycoords = [setup.coordinates[node][1] for node in setup.coordinates]
plt.gca().set_xlim(min(xcoords) - 30, max(xcoords) + 30)
plt.gca().set_ylim(min(ycoords) - 30, max(ycoords) + 30)
img_valve = mpimg.imread('aux_files/valve_img.png')
img_hx = mpimg.imread('aux_files/hx_img.png')
img_pump = mpimg.imread('aux_files/pump_img.png')
if withsymbols == True:
for PSM in mu_dict:
for i in range(len(setup.e_vec)):
edge = setup.e_vec[i]
if (str(PSM) in edge[0]) and (str(PSM) in edge[1]):
coord_hx = []
coord_hx.append(setup.coordinates[edge[0]][0] +
(setup.coordinates[edge[1]][0] -
setup.coordinates[edge[0]][0]) * (1 / 2))
coord_hx.append(setup.coordinates[edge[0]][1] +
(setup.coordinates[edge[1]][1] -
setup.coordinates[edge[0]][1]) * (2 / 5))
coord_act = [
coord_hx[0], setup.coordinates[edge[0]][1] +
(setup.coordinates[edge[1]][1] -
setup.coordinates[edge[0]][1]) * (3 / 5)
]
coord_act_disp = plt.gca().transData.transform(coord_act)
imagebox_hx = OffsetImage(img_hx, zoom=0.1)
ab_hx = AnnotationBbox(imagebox_hx,
(coord_hx[0], coord_hx[1]),
frameon=False)
plt.gca().add_artist(ab_hx)
fig1.canvas.draw()
renderer = fig1.canvas.renderer
inv1 = plt.gca().transData.inverted()
inv2 = plt.gca().transAxes.inverted()
dotm_sec_str = r'$\dot{V}_{sec} = %4.2f \frac{kg}{min}$' % (
dotV_sec_dict[PSM])
T_sec_in_str = r'$T_{sec,in} = %5.1f ^\circ C$' % (
T_sec_dict[PSM] - 273.15)
T_sec_out_str = r'$T_{sec,out} = %5.1f ^\circ C$' % (
T_soldict[Edges_Temps_dict[PSM]['T_out_PSM']] - 273.15)
Q_sec_str = r'$\dot{Q}_{sec} = %4.2f kW $' % (
Q_soldict[PSM])
dotm_prim_str = r'$\dot{V}_{prim} = %4.2f \frac{kg}{min}$' % (
Edges_Temps_dict[PSM]['dotV_HTNW'])
T_prim_in_str = r'$T_{prim,in} = %5.1f ^\circ C$' % (
T_soldict[Edges_Temps_dict[PSM]['T_in_HTNW']] - 273.15)
T_prim_out_str = r'$T_{prim,out} = %5.1f ^\circ C$' % (
T_soldict[Edges_Temps_dict[PSM]['T_out_HTNW']] -
273.15)
if mu_dict[PSM] == 1:
imagebox_pump = OffsetImage(img_pump, zoom=0.19)
myExtend1 = imagebox_pump.get_extent(renderer)
xpos_text_disp1 = coord_act_disp[
0] + 0.55 * myExtend1[0]
coord_text_data1 = inv1.transform(
(xpos_text_disp1, coord_act_disp[1]))
ab_pump = AnnotationBbox(imagebox_pump,
(coord_act[0], coord_act[1]),
box_alignment=(0.5, 0.5),
pad=0,
frameon=False)
plt.gca().add_artist(ab_pump)
elif mu_dict[PSM] == -1:
# position valve image
imagebox_valve = OffsetImage(img_valve, zoom=0.4)
myExtend2 = imagebox_valve.get_extent(renderer)
xpos_valve_disp = coord_act_disp[
0] - 0.31 * myExtend2[0]
coord_valve_data = inv1.transform(
(xpos_valve_disp, coord_act_disp[1]))
coord_valve_axes = inv2.transform(
(xpos_valve_disp, coord_act_disp[1]))
ab_valve = AnnotationBbox(imagebox_valve,
coord_valve_data,
box_alignment=(0.5, 0.5),
pad=0,
frameon=False)
#xycoords='axes fraction'
plt.gca().add_artist(ab_valve)
# position text for control value U
xpos_text_disp2 = xpos_valve_disp + 0.55 * myExtend2[0]
coord_text_data2 = inv1.transform(
(xpos_text_disp2, coord_act_disp[1]))
if withnumbers == True:
nx.drawing.nx_pylab.draw_networkx_edge_labels(graph.Gsol_th,
pos=setup.coordinates,
font_size=15,
edge_labels=myedgelabels,
node_color=colorslist,
arrowsize=15,
arrowstyle=myarrowstyle,
node_size=700,
width=3,
font_weight='normal',
alpha=1)
for PSM in mu_dict:
for i in range(len(setup.e_vec)):
edge = setup.e_vec[i]
if (str(PSM) in edge[0]) and (str(PSM) in edge[1]):
coord_hx = []
coord_hx.append(setup.coordinates[edge[0]][0] +
(setup.coordinates[edge[1]][0] -
setup.coordinates[edge[0]][0]) * (1 / 2))
coord_hx.append(setup.coordinates[edge[0]][1] +
(setup.coordinates[edge[1]][1] -
setup.coordinates[edge[0]][1]) * (2 / 5))
coord_hx_disp = plt.gca().transData.transform(coord_hx)
imagebox_hx = OffsetImage(img_hx, zoom=0.1)
fig1.canvas.draw()
renderer = fig1.canvas.renderer
inv1 = plt.gca().transData.inverted()
inv2 = plt.gca().transAxes.inverted()
T_netw_h_str = r'$T_{%1.0f h} = %5.1f ^\circ C$' % (
PSM, T_soldict[edge])
T_netw_c_str = r'$T_{%1.0f c} = %5.1f ^\circ C$' % (
PSM, T_soldict[(edge[1], edge[0])])
myExtend0 = imagebox_hx.get_extent(renderer)
xpos_text_disp0a = coord_hx_disp[0] + 0.2 * myExtend0[0]
xpos_text_disp0b = coord_hx_disp[0] + 0.2 * myExtend0[0]
ypos_text_disp0a = coord_hx_disp[1] + 1.0 * myExtend0[0]
ypos_text_disp0b = coord_hx_disp[1] - 2.8 * myExtend0[0]
coord_text_data0a = inv1.transform(
(xpos_text_disp0a, ypos_text_disp0a))
coord_text_data0b = inv1.transform(
(xpos_text_disp0b, ypos_text_disp0b))
plt.text(coord_text_data0a[0],
coord_text_data0a[1],
T_netw_h_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
plt.text(coord_text_data0b[0],
coord_text_data0b[1],
T_netw_c_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
plt.axis('off')
plt.show(block=False)
plt.savefig(''.join((savepath, 'vis_solution_th.', saveformat)),
format=saveformat)
def visualize_prosumer_results(setup,
graph,
solutions,
th_problem,
withnumbers=True):
# initializations
img_valve = mpimg.imread('aux_files/valve_img.png')
img_hx = mpimg.imread('aux_files/hx_img.png')
img_pump = mpimg.imread('aux_files/pump_img.png')
mu_dict = {P: setup.scenario[P]['mu'] for P in setup.PSM}
u_dict = {P: setup.scenario[P]['u'] for P in setup.PSM}
kappa_dict = {P: setup.scenario[P]['kappa'] for P in setup.PSM}
T_sec_dict = {P: setup.scenario[P]['T_sec_in_degree'] for P in setup.PSM}
dotV_sec_dict = {P: setup.scenario[P]['dotV_sec_in'] for P in setup.PSM}
Edges_Temps_dict = th_problem.general_stuff['Edges_Temps_dict']
Q_soldict = solutions['Q_trnsf2']
T_sec_out_dict = {
P: solutions['T'][Edges_Temps_dict[P]['T_out_PSM']]
for P in setup.PSM
}
dotV_soldict = solutions['dotV']
# parameters
myfontsize = 20
myfigsize = [2 * 21.0 / 2.54, 2 * 12.98 / 2.54]
myarrowstyle = mpltlib.patches.ArrowStyle.CurveFilledAB(
head_length=0.6, head_width=0.4) #CurveB, CurveFilledB
savepath = './results/'
saveformat = 'png' # jpg, png, eps, svgd
## initialize new Graph that represents the solution
G = nx.DiGraph()
G.add_nodes_from(setup.v_vec)
myedgelabels = {}
for i in range(len(setup.e_vec)):
edge = setup.e_vec[i]
if np.sign(dotV_soldict[edge]) == -1:
G.add_edge(edge[1], edge[0])
elif np.sign(dotV_soldict[edge]) == 1:
G.add_edge(edge[0], edge[1])
elif np.sign(dotV_soldict[edge]) == 0:
G.add_edge(edge[0], edge[1])
else:
raise ValueError('')
# initialize figure
fig = plt.figure(num='prosumer_results', figsize=myfigsize)
plt.title('prosumer results', {'fontsize': 30, 'fontweight': 'bold'})
# set-up node color
for node in G.nodes():
if 'h' in node:
G.nodes[node]['color'] = 'r'
if 'c' in node:
G.nodes[node]['color'] = 'b'
nodes_colors = {node: G.nodes[node]['color'] for node in G.nodes()}
colorslist = list(nodes_colors.values())
# draw
nx.drawing.nx_pylab.draw(G,
pos=setup.coordinates,
fontsize=20,
with_labels=True,
node_color=colorslist,
arrowsize=15,
arrowstyle=myarrowstyle,
node_size=900,
width=3,
font_weight='bold',
alpha=1)
xcoords = [setup.coordinates[node][0] for node in setup.coordinates]
ycoords = [setup.coordinates[node][1] for node in setup.coordinates]
plt.gca().set_xlim(min(xcoords) - 30, max(xcoords) + 30)
plt.gca().set_ylim(min(ycoords) - 30, max(ycoords) + 30)
plt.axis('off')
plt.show(block=False)
for PSM in mu_dict:
for i in range(len(setup.e_vec)):
edge = setup.e_vec[i]
if (str(PSM) in edge[0]) and (str(PSM) in edge[1]):
coord_hx = []
coord_hx.append(setup.coordinates[edge[0]][0] +
(setup.coordinates[edge[1]][0] -
setup.coordinates[edge[0]][0]) * (1 / 2))
coord_hx.append(setup.coordinates[edge[0]][1] +
(setup.coordinates[edge[1]][1] -
setup.coordinates[edge[0]][1]) * (2 / 5))
coord_act = [
coord_hx[0], setup.coordinates[edge[0]][1] +
(setup.coordinates[edge[1]][1] -
setup.coordinates[edge[0]][1]) * (3 / 5)
]
coord_act_disp = plt.gca().transData.transform(coord_act)
coord_hx_disp = plt.gca().transData.transform(coord_hx)
imagebox_hx = OffsetImage(img_hx, zoom=0.1)
ab_hx = AnnotationBbox(imagebox_hx, (coord_hx[0], coord_hx[1]),
frameon=False)
plt.gca().add_artist(ab_hx)
fig.canvas.draw()
renderer = fig.canvas.renderer
inv1 = plt.gca().transData.inverted()
inv2 = plt.gca().transAxes.inverted()
U_pump_str = r'$u$ = %4.2f' % (u_dict[PSM])
U_valve_str = r'$\kappa = %4.2f$' % (kappa_dict[PSM])
dotm_sec_str = r'$\dot{V}_{sec} = %4.2f \frac{l}{min}$' % (
dotV_sec_dict[PSM])
T_sec_str = r'$T_{sec,in} = %5.1f ^\circ C$' % (
T_sec_dict[PSM])
T_sec_out_str = r'$T_{sec,out} = %5.1f ^\circ C$' % (
T_sec_out_dict[PSM])
dotQ_str = r'$\dot{Q} = %5.1f kW$' % (-Q_soldict[PSM] / 1000)
myExtend0 = imagebox_hx.get_extent(renderer)
xpos_text_disp0a = coord_hx_disp[0] + 0.55 * myExtend0[0]
xpos_text_disp0b = coord_hx_disp[0] + 0.55 * myExtend0[0]
xpos_text_disp0c = coord_hx_disp[0] + 0.55 * myExtend0[0]
xpos_text_disp0d = coord_hx_disp[0] + 0.55 * myExtend0[0]
ypos_text_disp0a = coord_hx_disp[1] + 1.9 * myExtend0[0]
ypos_text_disp0b = coord_hx_disp[1] + 1.1 * myExtend0[0]
ypos_text_disp0c = coord_hx_disp[1] + 0.3 * myExtend0[0]
ypos_text_disp0d = coord_hx_disp[1] - 0.5 * myExtend0[0]
coord_text_data0a = inv1.transform(
(xpos_text_disp0a, ypos_text_disp0a))
coord_text_data0b = inv1.transform(
(xpos_text_disp0b, ypos_text_disp0b))
coord_text_data0c = inv1.transform(
(xpos_text_disp0c, ypos_text_disp0c))
coord_text_data0d = inv1.transform(
(xpos_text_disp0d, ypos_text_disp0d))
if withnumbers == True:
plt.text(coord_text_data0a[0],
coord_text_data0a[1],
dotm_sec_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
plt.text(coord_text_data0b[0],
coord_text_data0b[1],
T_sec_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
plt.text(coord_text_data0c[0],
coord_text_data0c[1],
T_sec_out_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
plt.text(coord_text_data0d[0],
coord_text_data0d[1],
dotQ_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
if mu_dict[PSM] == 1:
imagebox_pump = OffsetImage(img_pump, zoom=0.19)
myExtend1 = imagebox_pump.get_extent(renderer)
xpos_text_disp1 = coord_act_disp[0] + 0.55 * myExtend1[0]
coord_text_data1 = inv1.transform(
(xpos_text_disp1, coord_act_disp[1]))
ab_pump = AnnotationBbox(imagebox_pump,
(coord_act[0], coord_act[1]),
box_alignment=(0.5, 0.5),
pad=0,
frameon=False)
plt.gca().add_artist(ab_pump)
if withnumbers == True:
plt.text(coord_text_data1[0],
coord_text_data1[1],
U_pump_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
elif mu_dict[PSM] == -1:
# position valve image
imagebox_valve = OffsetImage(img_valve, zoom=0.4)
myExtend2 = imagebox_valve.get_extent(renderer)
xpos_valve_disp = coord_act_disp[0] - 0.31 * myExtend2[0]
coord_valve_data = inv1.transform(
(xpos_valve_disp, coord_act_disp[1]))
#coord_valve_axes=inv2.transform((xpos_valve_disp,coord_act_disp[1]))
ab_valve = AnnotationBbox(imagebox_valve,
coord_valve_data,
box_alignment=(0.5, 0.5),
pad=0,
frameon=False)
#xycoords='axes fraction'
plt.gca().add_artist(ab_valve)
if withnumbers == True:
# position text for control value U
xpos_text_disp2 = xpos_valve_disp + 0.55 * myExtend2[0]
coord_text_data2 = inv1.transform(
(xpos_text_disp2, coord_act_disp[1]))
plt.text(coord_text_data2[0],
coord_text_data2[1],
U_valve_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
plt.axis('off')
plt.show(block=False)
plt.savefig(''.join((savepath, 'vis_prosumer_res.', saveformat)),
format=saveformat)
def visualize_hy_solution(setup,
graph,
solutions,
withsymbols=False,
withnumbers=True,
data='flow'):
savepath = './results/'
saveformat = 'png' # jpg, png, eps, svgd
# combine the solution values with the correct edges
dotV_soldict = solutions['dotV']
Deltap_soldict = solutions['Deltap']
mu_dict = {P: setup.scenario[P]['mu'] for P in setup.PSM}
u_dict = {P: setup.scenario[P]['u'] for P in setup.PSM}
kappa_dict = {P: setup.scenario[P]['kappa'] for P in setup.PSM}
T_sec_dict = {P: setup.scenario[P]['T_sec_in_degree'] for P in setup.PSM}
dotV_sec_dict = {P: setup.scenario[P]['dotV_sec_in'] for P in setup.PSM}
# assign the solution values to the Graph
for edge in setup.e_vec:
graph.G.edges[edge]['dotV_sol'] = dotV_soldict[edge]
graph.G.edges[edge]['Deltap_sol'] = Deltap_soldict[edge]
# initialize new Graph that represents the solution
graph.Gsol_hy = nx.DiGraph()
graph.Gsol_hy.add_nodes_from(setup.v_vec)
myedgelabels = {}
for i in range(len(setup.e_vec)):
edge = setup.e_vec[i]
if np.sign(dotV_soldict[edge]) == -1:
graph.Gsol_hy.add_edge(edge[1], edge[0])
graph.Gsol_hy.edges[(
edge[1], edge[0])]['dotV'] = -dotV_soldict[(edge[0], edge[1])]
graph.Gsol_hy.edges[(
edge[1],
edge[0])]['Deltap'] = -Deltap_soldict[(edge[0], edge[1])]
myedgelabels[(edge[1], edge[0])] = []
if data == 'flow':
myedgelabels[(edge[1], edge[0])].append(
'{0:6.2f} l/min'.format(-dotV_soldict[(edge[0], edge[1])]))
elif data == 'pressure':
myedgelabels[(edge[1], edge[0])].append(
'{0:6.1f} hPa'.format(-Deltap_soldict[(edge[0], edge[1])]))
elif data == 'both':
myedgelabels[(edge[1], edge[0])].append(
'{0:6.2f} l/min'.format(-dotV_soldict[(edge[0], edge[1])]))
myedgelabels[(edge[1], edge[0])].append(
'{0:6.1f} hPa'.format(-Deltap_soldict[(edge[0], edge[1])]))
elif np.sign(dotV_soldict[edge]) == 1:
graph.Gsol_hy.add_edge(edge[0], edge[1])
graph.Gsol_hy.edges[(edge[0],
edge[1])]['dotV'] = dotV_soldict[(edge[0],
edge[1])]
graph.Gsol_hy.edges[(
edge[0], edge[1])]['Deltap'] = Deltap_soldict[(edge[0],
edge[1])]
myedgelabels[(edge[0], edge[1])] = []
if data == 'flow':
myedgelabels[(edge[0], edge[1])].append(
'{0:6.2f} l/min'.format(dotV_soldict[(edge[0], edge[1])]))
elif data == 'pressure':
myedgelabels[(edge[0], edge[1])].append('{0:6.1f} hPa'.format(
Deltap_soldict[(edge[0], edge[1])]))
elif data == 'both':
myedgelabels[(edge[0], edge[1])].append(
'{0:6.2f} l/min'.format(dotV_soldict[(edge[0], edge[1])]))
myedgelabels[(edge[0], edge[1])].append('{0:6.1f} hPa'.format(
Deltap_soldict[(edge[0], edge[1])]))
elif np.sign(dotV_soldict[edge]) == 0:
graph.Gsol_hy.add_edge(edge[0], edge[1])
graph.Gsol_hy.edges[(edge[0],
edge[1])]['dotV'] = dotV_soldict[(edge[0],
edge[1])]
graph.Gsol_hy.edges[(
edge[0], edge[1])]['Deltap'] = Deltap_soldict[(edge[0],
edge[1])]
myedgelabels[(edge[0], edge[1])] = []
if data == 'flow':
myedgelabels[(edge[0], edge[1])].append(
'{0:6.2f} l/min'.format(dotV_soldict[(edge[0], edge[1])]))
elif data == 'pressure':
myedgelabels[(edge[0], edge[1])].append('{0:6.1f} hPa'.format(
Deltap_soldict[(edge[0], edge[1])]))
elif data == 'both':
myedgelabels[(edge[0], edge[1])].append(
'{0:6.2f} l/min'.format(dotV_soldict[(edge[0], edge[1])]))
myedgelabels[(edge[0], edge[1])].append('{0:6.1f} hPa'.format(
Deltap_soldict[(edge[0], edge[1])]))
else:
raise ValueError('')
# plot graph
myfontsize = 20
myfigsize = [2 * 21.0 / 2.54, 2 * 12.98 / 2.54]
myarrowstyle = mpltlib.patches.ArrowStyle.CurveFilledB(
head_length=0.6, head_width=0.4) #CurveB
fig1 = plt.figure(num='hydraulic solution', figsize=myfigsize)
plt.title('hydraulic solution', {'fontsize': 30, 'fontweight': 'bold'})
for node in graph.Gsol_hy.nodes():
if 'h' in node:
graph.Gsol_hy.nodes[node]['color'] = 'r'
if 'c' in node:
graph.Gsol_hy.nodes[node]['color'] = 'b'
nodes_colors = {
node: graph.Gsol_hy.nodes[node]['color']
for node in graph.Gsol_hy.nodes()
}
colorslist = list(nodes_colors.values())
nx.drawing.nx_pylab.draw(graph.Gsol_hy,
pos=setup.coordinates,
fontsize='xx-large',
with_labels=True,
node_color=colorslist,
arrowsize=15,
arrowstyle=myarrowstyle,
node_size=700,
width=3,
font_weight='bold',
alpha=1)
xcoords = [setup.coordinates[node][0] for node in setup.coordinates]
ycoords = [setup.coordinates[node][1] for node in setup.coordinates]
plt.gca().set_xlim(min(xcoords) - 30, max(xcoords) + 30)
plt.gca().set_ylim(min(ycoords) - 30, max(ycoords) + 30)
img_valve = mpimg.imread('aux_files/valve_img.png')
img_hx = mpimg.imread('aux_files/hx_img.png')
img_pump = mpimg.imread('aux_files/pump_img.png')
if withsymbols == True:
for PSM in mu_dict:
for i in range(len(setup.e_vec)):
edge = setup.e_vec[i]
if (str(PSM) in edge[0]) and (str(PSM) in edge[1]):
coord_hx = []
coord_hx.append(setup.coordinates[edge[0]][0] +
(setup.coordinates[edge[1]][0] -
setup.coordinates[edge[0]][0]) * (1 / 2))
coord_hx.append(setup.coordinates[edge[0]][1] +
(setup.coordinates[edge[1]][1] -
setup.coordinates[edge[0]][1]) * (2 / 5))
coord_act = [
coord_hx[0], setup.coordinates[edge[0]][1] +
(setup.coordinates[edge[1]][1] -
setup.coordinates[edge[0]][1]) * (3 / 5)
]
coord_act_disp = plt.gca().transData.transform(coord_act)
imagebox_hx = OffsetImage(img_hx, zoom=0.1)
ab_hx = AnnotationBbox(imagebox_hx,
(coord_hx[0], coord_hx[1]),
frameon=False)
plt.gca().add_artist(ab_hx)
fig1.canvas.draw()
renderer = fig1.canvas.renderer
inv1 = plt.gca().transData.inverted()
inv2 = plt.gca().transAxes.inverted()
U_pump_str = r'u = %4.2f' % (u_dict[PSM])
U_valve_str = r'$\kappa$ = %4.2f' % (kappa_dict[PSM])
if mu_dict[PSM] == 1:
imagebox_pump = OffsetImage(img_pump, zoom=0.19)
myExtend1 = imagebox_pump.get_extent(renderer)
xpos_text_disp1 = coord_act_disp[
0] + 0.55 * myExtend1[0]
coord_text_data1 = inv1.transform(
(xpos_text_disp1, coord_act_disp[1]))
ab_pump = AnnotationBbox(imagebox_pump,
(coord_act[0], coord_act[1]),
box_alignment=(0.5, 0.5),
pad=0,
frameon=False)
plt.gca().add_artist(ab_pump)
plt.text(coord_text_data1[0],
coord_text_data1[1],
U_pump_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
elif mu_dict[PSM] == -1:
# position valve image
imagebox_valve = OffsetImage(img_valve, zoom=0.4)
myExtend2 = imagebox_valve.get_extent(renderer)
xpos_valve_disp = coord_act_disp[
0] - 0.31 * myExtend2[0]
coord_valve_data = inv1.transform(
(xpos_valve_disp, coord_act_disp[1]))
coord_valve_axes = inv2.transform(
(xpos_valve_disp, coord_act_disp[1]))
ab_valve = AnnotationBbox(imagebox_valve,
coord_valve_data,
box_alignment=(0.5, 0.5),
pad=0,
frameon=False)
#xycoords='axes fraction'
plt.gca().add_artist(ab_valve)
# position text for control value U
xpos_text_disp2 = xpos_valve_disp + 0.55 * myExtend2[0]
coord_text_data2 = inv1.transform(
(xpos_text_disp2, coord_act_disp[1]))
plt.text(coord_text_data2[0],
coord_text_data2[1],
U_valve_str,
horizontalalignment='left',
verticalalignment='center',
fontsize=myfontsize)
if withnumbers == True:
nx.drawing.nx_pylab.draw_networkx_edge_labels(graph.Gsol_hy,
pos=setup.coordinates,
font_size=14,
edge_labels=myedgelabels,
node_color=colorslist,
arrowsize=15,
arrowstyle=myarrowstyle,
node_size=700,
width=3,
font_weight='normal',
alpha=1)
plt.axis('off')
plt.show(block=False)
plt.savefig(''.join((savepath, 'vis_solution_hy.', saveformat)),
format=saveformat)
