You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
2.3 KiB
Python
99 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# Get topologies
|
|
|
|
import sys
|
|
from birdvisu.providers import BirdSocketTopologyProvider, OspfFileTopologyProvider
|
|
|
|
|
|
ref_topo_file = 'reference.ospf'
|
|
if len(sys.argv) > 1:
|
|
ref_topo_file = sys.argv[1]
|
|
|
|
ref_topo = OspfFileTopologyProvider(ref_topo_file).get_topology()
|
|
|
|
try:
|
|
cur_topo = BirdSocketTopologyProvider(instance='ospfv2', area=1).get_topology()
|
|
except OSError as e:
|
|
raise NotImplementedError('Cannot create a mock topology atm') from e
|
|
|
|
# Create combined topology
|
|
|
|
from birdvisu.topo_v3 import TopologyV3
|
|
|
|
combined_topology = TopologyV3.combine_topologies(reference=ref_topo, current=cur_topo)
|
|
combined_topology.freeze()
|
|
|
|
# Annotate it
|
|
|
|
from birdvisu.topo_v3 import VertexID
|
|
from birdvisu.annotations import AnnotatedTopology, AnnotatorID
|
|
from birdvisu.annotations.analysis import TopologyDifference, ShortestPathTree
|
|
from ipaddress import IPv4Address
|
|
|
|
annot_topo = AnnotatedTopology(combined_topology)
|
|
annotators = [
|
|
AnnotatorID(TopologyDifference),
|
|
AnnotatorID(ShortestPathTree, (VertexID(
|
|
family=None,
|
|
is_router=True,
|
|
address=None,
|
|
router_id=int(IPv4Address('172.23.100.10')),
|
|
dr_id=None,
|
|
discriminator=None
|
|
), 'current')),
|
|
]
|
|
for ann_id in annotators:
|
|
annot_topo.run_annotator(ann_id)
|
|
|
|
# ---
|
|
|
|
# Show it
|
|
|
|
from birdvisu.visualisation import annotators
|
|
from birdvisu import maps_new
|
|
|
|
# annotators.create_qgritems does not like being run without Qt initialization.
|
|
from PySide6 import QtCore, QtGui, QtWidgets
|
|
app = QtWidgets.QApplication([])
|
|
|
|
annotated_topology = maps_new.annotate_topology(combined_topology,
|
|
# A semi-canonical set of annotators:
|
|
[
|
|
annotators.extract_positions,
|
|
annotators.random_position,
|
|
annotators.assign_brushes,
|
|
annotators.create_qgritems,
|
|
]
|
|
)
|
|
|
|
|
|
# Render the widget
|
|
|
|
scene = QtWidgets.QGraphicsScene()
|
|
|
|
for tagsrc in [
|
|
annotated_topology.router_annotations.values(),
|
|
annotated_topology.network_annotations.values(),
|
|
annotated_topology.link_annotations.values(),
|
|
]:
|
|
for taglist in tagsrc:
|
|
assert len(taglist) > 0
|
|
assert isinstance(taglist[-1], QtWidgets.QGraphicsItem)
|
|
scene.addItem(taglist[-1])
|
|
|
|
view = QtWidgets.QGraphicsView(scene)
|
|
#view.show()
|
|
|
|
|
|
main_window = QtWidgets.QMainWindow()
|
|
main_window.setCentralWidget(view)
|
|
|
|
menu = main_window.menuBar().addMenu('Hello')
|
|
act = QtGui.QAction('Hi')
|
|
#act.setToolTip('Howdy')
|
|
menu.addAction(act)
|
|
main_window.show()
|
|
|
|
app.exec()
|