mirror of
https://github.com/Telecominfraproject/oopt-gnpy.git
synced 2025-11-04 03:57:47 +00:00
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from gnpy import Network
|
||
from gnpy.utils import read_config
|
||
from os.path import realpath, join, dirname
|
||
|
||
if __name__ == '__main__':
|
||
basedir = dirname(realpath(__file__))
|
||
filename = join(basedir, 'config/config_ex1.json')
|
||
config = read_config(filename)
|
||
nw = Network(config)
|
||
nw.propagate_all_paths()
|
||
|
||
# output OSNR propagation
|
||
for path in nw.tr_paths:
|
||
print(' → '.join(x.id for x in path.path))
|
||
for u, v in path.edge_list:
|
||
channels = nw.g[u][v]['channels']
|
||
channel_info = ('\n' + ' ' * 24).join(
|
||
' '.join([f'freq: {x["frequency"]:7.2f}',
|
||
f'osnr: {x["osnr"]:7.2f}',
|
||
f'power: {x["power"]:7.2f}'])
|
||
for x in channels)
|
||
print(f'{u.id:^10s} → {v.id:^10s} {channel_info}')
|
||
|
||
exit()
|
||
|
||
|
||
config_fn = './gnpy/examples/config/config_ex1.json'
|
||
nw = gnpy.Network(config_fn)
|
||
nw.propagate_all_paths()
|
||
|
||
# output OSNR propagation
|
||
edge_fmt = "ν: {frequency}\nOSNR:{osnr:^7.2f}\npow:{power:^8.1f}"
|
||
elabels = {}
|
||
for path in nw.tr_paths:
|
||
print("Path", path.path)
|
||
print("{:^8}{:^11}{:^9}{:^5}{:^11}".format("n0", "n1", "ν", "OSNR", "pow"))
|
||
for edge in path.edge_list:
|
||
spectrum = nw.g[edge[0]][edge[1]]['channels']
|
||
pspec = [edge_fmt.format(**s) for s in spectrum]
|
||
elabels[edge] = '\n'.join(pspec)
|
||
print("{:^8}..{:^8}" .format(str(edge[0]), str(edge[1])),
|
||
pspec[0].replace('\n', ' ').replace('OSNR', '')
|
||
.replace('pow', "").replace("ν", "").replace(":", ""))
|
||
|
||
if 0:
|
||
import networkx as nx
|
||
layout = nx.spring_layout(nw.g)
|
||
nx.draw_networkx_nodes(nw.g, layout, node_size=1000,
|
||
node_color='b', alpha=0.2, node_shape='s')
|
||
nx.draw_networkx_labels(nw.g, layout)
|
||
nx.draw_networkx_edges(nw.g, layout, width=2,
|
||
alpha=0.3, edge_color='green')
|
||
nx.draw_networkx_edge_labels(
|
||
nw.g, layout, edge_labels=elabels, font_size=10)
|
||
plt.rcdefaults()
|
||
plt.axis('off')
|
||
plt.show()
|