mirror of
https://github.com/Telecominfraproject/oopt-gnpy.git
synced 2025-10-31 01:57:54 +00:00
c86ea206d9
bugs from the last refactor merge - fix #1: update fiber length after split - fix #2: add egress amp after fiber split Signed-off-by: Jean-Luc Auge <jeanluc.auge@orange.com>
153 lines
6.4 KiB
Python
Executable File
153 lines
6.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
'''
|
|
Transmission setup example:
|
|
reads from network json (default = examples/edfa/edfa_example_network.json)
|
|
propagates a 96 channels comb
|
|
'''
|
|
|
|
from gnpy.core.equipment import load_equipment
|
|
from gnpy.core.utils import db2lin, lin2db
|
|
from argparse import ArgumentParser
|
|
from sys import exit
|
|
from pathlib import Path
|
|
from json import loads
|
|
from collections import Counter
|
|
from logging import getLogger, basicConfig, INFO, ERROR, DEBUG
|
|
|
|
from matplotlib.pyplot import show, axis, figure, title
|
|
from networkx import (draw_networkx_nodes, draw_networkx_edges,
|
|
draw_networkx_labels, dijkstra_path)
|
|
from gnpy.core.network import load_network, build_network, set_roadm_loss, set_edfa_dp
|
|
from gnpy.core.elements import Transceiver, Fiber, Edfa, Roadm
|
|
from gnpy.core.info import SpectralInformation, Channel, Power, Pref
|
|
|
|
logger = getLogger(__name__)
|
|
|
|
def plot_results(network, path, source, sink):
|
|
path_edges = set(zip(path[:-1], path[1:]))
|
|
edges = set(network.edges()) - path_edges
|
|
pos = {n: (n.lng, n.lat) for n in network.nodes()}
|
|
labels = {n: n.location.city for n in network.nodes() if isinstance(n, Transceiver)}
|
|
city_labels = set(labels.values())
|
|
for n in network.nodes():
|
|
if n.location.city and n.location.city not in city_labels:
|
|
labels[n] = n.location.city
|
|
city_labels.add(n.location.city)
|
|
label_pos = pos
|
|
|
|
fig = figure()
|
|
kwargs = {'figure': fig, 'pos': pos}
|
|
plot = draw_networkx_nodes(network, nodelist=network.nodes(), node_color='#ababab', **kwargs)
|
|
draw_networkx_nodes(network, nodelist=path, node_color='#ff0000', **kwargs)
|
|
draw_networkx_edges(network, edgelist=edges, edge_color='#ababab', **kwargs)
|
|
draw_networkx_edges(network, edgelist=path_edges, edge_color='#ff0000', **kwargs)
|
|
draw_networkx_labels(network, labels=labels, font_size=14, **{**kwargs, 'pos': label_pos})
|
|
title(f'Propagating from {source.loc.city} to {sink.loc.city}')
|
|
axis('off')
|
|
show()
|
|
|
|
|
|
def main(network, equipment, source, sink):
|
|
build_network(network, equipment=equipment)
|
|
path = dijkstra_path(network, source, sink)
|
|
gain_mode = False
|
|
if not gain_mode:
|
|
path_amps = [amp for amp in path if isinstance(amp, Edfa)]
|
|
set_edfa_dp(network, path_amps)
|
|
spans = [s.length for s in path if isinstance(s, Fiber)]
|
|
print(f'\nThere are {len(spans)} fiber spans over {sum(spans):.0f}m between {source.uid} and {sink.uid}')
|
|
print(f'\nNow propagating between {source.uid} and {sink.uid}:')
|
|
|
|
pref_span0_db = 2
|
|
bounds = range(1, 2) #power sweep
|
|
pref_roadm_db = -20
|
|
pref_span0 = db2lin(pref_span0_db)*1e-3
|
|
pref_roadm = db2lin(pref_roadm_db)*1e-3
|
|
for p_db in range(pref_span0_db+bounds.start, pref_span0_db+bounds.stop): #change range to sweep results across several powers in dBm
|
|
p = db2lin(p_db)*1e-3
|
|
roadm_loss = pref_span0_db - pref_roadm_db
|
|
path_roadms = [roadm for roadm in path if isinstance(roadm, Roadm)]
|
|
set_roadm_loss(path_roadms, roadm_loss)
|
|
spacing = 0.05 # THz
|
|
si = SpectralInformation(pref=Pref(pref_span0_db, pref_span0_db))
|
|
si = si.update(carriers=[
|
|
Channel(f, (191.3 + spacing * f) * 1e12, 32e9, 0.15, Power(p, 0, 0))
|
|
for f in range(1,97)
|
|
])
|
|
print(f'\nPorpagating with input power = {lin2db(p*1e3):.2f}dBm :')
|
|
for el in path:
|
|
si = el(si)
|
|
print(el) #remove this line when sweeping across several powers
|
|
print(f'\nTransmission result for input power = {lin2db(p*1e3):.2f}dBm :')
|
|
print(sink)
|
|
|
|
return path
|
|
|
|
parser = ArgumentParser()
|
|
parser.add_argument('-e', '--equipment', type=Path,
|
|
default=Path(__file__).parent / 'eqpt_config.json')
|
|
parser.add_argument('-p', '--plot', action='store_true', default=False)
|
|
parser.add_argument('-v', '--verbose', action='count')
|
|
parser.add_argument('-l', '--list-nodes', action='store_true', default=False, help='list all transceiver nodes')
|
|
parser.add_argument('filename', nargs='?', type=Path,
|
|
default=Path(__file__).parent / 'edfa_example_network.json')
|
|
parser.add_argument('source', nargs='?', help='source node')
|
|
parser.add_argument('sink', nargs='?', help='sink node')
|
|
|
|
if __name__ == '__main__':
|
|
args = parser.parse_args()
|
|
basicConfig(level={0: ERROR, 1: INFO, 2: DEBUG}.get(args.verbose, ERROR))
|
|
|
|
equipment = load_equipment(args.equipment)
|
|
# logger.info(equipment)
|
|
|
|
network = load_network(args.filename, equipment)
|
|
print(network)
|
|
|
|
transceivers = {n.uid: n for n in network.nodes() if isinstance(n, Transceiver)}
|
|
|
|
if not transceivers:
|
|
exit('Network has no transceivers!')
|
|
if len(transceivers) < 2:
|
|
exit('Network has only one transceiver!')
|
|
|
|
if args.list_nodes:
|
|
for uid in transceivers:
|
|
print(uid)
|
|
exit()
|
|
|
|
if args.source:
|
|
try:
|
|
source = next(transceivers[uid] for uid in transceivers if uid == args.source)
|
|
except StopIteration as e:
|
|
#TODO code a more advanced regex to find nodes match
|
|
nodes_suggestion = [uid for uid in transceivers if args.source.lower() in uid.lower()]
|
|
source = transceivers[nodes_suggestion[0]] if len(nodes_suggestion)>0 else transceivers[0]
|
|
print(f'invalid souce node specified, did you mean:\
|
|
\n{nodes_suggestion}?\
|
|
\n{args.source!r}, replaced with {source.uid}')
|
|
del transceivers[source.uid]
|
|
else:
|
|
logger.info('No source node specified: picking random transceiver')
|
|
source = list(transceivers.values())[0]
|
|
|
|
if args.sink:
|
|
try:
|
|
sink = next(transceivers[uid] for uid in transceivers if uid == args.sink)
|
|
except StopIteration as e:
|
|
nodes_suggestion = [uid for uid in transceivers if args.sink.lower() in uid.lower()]
|
|
sink = transceivers[nodes_suggestion[0]] if len(nodes_suggestion)>0 else tansceivers[-1]
|
|
print(f'invalid destination node specified, did you mean:\
|
|
\n{nodes_suggestion}?\
|
|
\n{args.sink!r}, replaced with {sink.uid}')
|
|
else:
|
|
logger.info('No source node specified: picking random transceiver')
|
|
sink = list(transceivers.values())[-1]
|
|
|
|
logger.info(f'source = {args.source!r}')
|
|
logger.info(f'sink = {args.sink!r}')
|
|
path = main(network, equipment, source, sink)
|
|
|
|
if args.plot:
|
|
plot_results(network, path, source, sink)
|