diff --git a/src/view/MainWindow.py b/src/view/MainWindow.py index 62423bad62134b00d8f9e8833f46b245f0ecc0f6..d1d4e13d1c38296e9a83c3cd7774263b9b371a1e 100644 --- a/src/view/MainWindow.py +++ b/src/view/MainWindow.py @@ -274,7 +274,7 @@ class ApplicationWindow(QMainWindow, ListedSubWindow): Nothing """ if not self.model is None: - self.network = NetworkWindow(parent=self) + self.network = NetworkWindow(model=self.model, parent=self) self.network.show() # TODO: Delete me ! diff --git a/src/view/NetworkWindow.py b/src/view/NetworkWindow.py index 4cf74d00e0cd33bf3656105a717e3fd2780bb506..de4a3f99828ff239e1a08a2a25830e28da80cfc8 100644 --- a/src/view/NetworkWindow.py +++ b/src/view/NetworkWindow.py @@ -21,11 +21,16 @@ from PyQt5.QtWidgets import ( ) class NetworkWindow(ASubWindow): - def __init__(self, title="Network", parent=None): + def __init__(self, model=None, title="River network", parent=None): super(NetworkWindow, self).__init__(name=title, ui="Network", parent=parent) self.ui.setWindowTitle(title) - self.graph = Graph() + self.model = model + if not self.model.data["graph"]: + self.graph = Graph() + self.model.data["graph"] = self.graph + else: + self.graph = self.model.data["graph"] # Graph Widget