From 525fb8e8681da457d8cc03ce7b5cd2fd821047c4 Mon Sep 17 00:00:00 2001
From: Pierre-Antoine Rouby <pierre-antoine.rouby@inrae.fr>
Date: Fri, 24 Mar 2023 13:42:49 +0100
Subject: [PATCH] model: Add river network to study.
---
src/view/MainWindow.py | 2 +-
src/view/NetworkWindow.py | 9 +++++++--
2 files changed, 8 insertions(+), 3 deletions(-)
diff --git a/src/view/MainWindow.py b/src/view/MainWindow.py
index 62423bad..d1d4e13d 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 4cf74d00..de4a3f99 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
--
GitLab