AdisTS: OutputRK: Fix save methode.

src/Meshing/Internal.py

@@ -214,7 +214,7 @@ class InternalMeshing(AMeshingTool):
                             ratio = (alpha[j0] - beta[j0-1]) \
                                     / (beta[j0] - beta[j0-1])
                             if ratio < 0.0:
-                                print(f"ratio négatif {ratio}")
+                                logger.warning(f"ratio négatif {ratio}")
                                 # on double le point a gauche
                                 p = sect2.point(start2+j0-1).copy()
                                 sect2.insert_point(start2+j0-1, p)

src/Model/BoundaryConditionsAdisTS/BoundaryConditionAdisTS.py

@@ -304,8 +304,10 @@ class BoundaryConditionAdisTS(SQLSubModel):
 
             if pol_id not in pid_pol:
                 # ⚠️ cas important : probablement déjà migré
-                print(f"[WARN] pol_id {pol_id} not in pid_pol " +
-                      f"→ probably already migrated")
+                logger.warning(
+                    f"pol_id {pol_id} not in pid_pol " +
+                    "→ probably already migrated"
+                )
                 continue
 
             execute(

src/Model/Geometry/PointXYZ.py

@@ -338,6 +338,13 @@ class PointXYZ(Point):
         self._z = float(value)
         self.modified()
 
+    def shift(self, x, y, z):
+        self.x += x
+        self.y += y
+        self.z += z
+
+        self.modified()
+
     def get_coordinate(self):
         return (self._x, self._y, self._z)
 

src/Model/Geometry/ProfileXYZ.py

@@ -24,7 +24,7 @@ from typing import List
 from functools import reduce
 from dataclasses import dataclass
 
-from tools import timer, flatten
+from tools import trace, timer, flatten
 from shapely import geometry
 
 from Model.Tools.PamhyrDB import SQLSubModel
@@ -396,7 +396,6 @@ class ProfileXYZ(Profile, SQLSubModel):
 
         for point in self._points:
             p = point.copy()
-            print(p)
             new_p._points.append(p)
 
         new_p.modified()
@@ -1074,16 +1073,10 @@ class ProfileXYZ(Profile, SQLSubModel):
 
     def shift(self, x, y, z):
         for p in self._points:
-            p.x = p.x + x
-            p.y = p.y + y
-            p.z = p.z + z
+            p.shift(x, y, z)
 
         self.modified()
 
-    def modified(self):
-        self.tab_up_to_date = False
-        self.station_up_to_date = False
-
     def add_npoints(self, npoints):
         # add npoints in a profile
         for k in range(npoints):

src/Model/Geometry/Reach.py

@@ -52,6 +52,11 @@ class Reach(SQLSubModel):
         self._guidelines_is_valid = False
         self._guidelines = {}
 
+        super(Reach, self).__init__(
+            id=parent.pamhyr_id,
+            status=status,
+        )
+
     @property
     def pamhyr_id(self):
         return self._parent.pamhyr_id
@@ -60,6 +65,10 @@ class Reach(SQLSubModel):
     def _pamhyr_id(self):
         return self._parent.pamhyr_id
 
+    @_pamhyr_id.setter
+    def _pamhyr_id(self, id):
+        return
+
     @classmethod
     def _db_create(cls, execute):
         return cls._create_submodel(execute)
@@ -202,7 +211,6 @@ class Reach(SQLSubModel):
         else:
             gi = self.get_global_profil_index(index)
             profile.num = gi
-            print(f"gi = {gi}")
             self._profiles.insert(gi, profile)
 
         self.modified()

src/Model/HydraulicStructures/Basic/HydraulicStructures.py

@@ -183,7 +183,6 @@ class BasicHS(SQLSubModel):
             bhs._data = BHSValue._db_load(
                 execute, data
             )
-            print(f"{bhs_pid} : {deleted}")
             if deleted:
                 bhs.set_as_deleted()
 

src/Model/InitialConditions/InitialConditions.py

@@ -588,8 +588,6 @@ class InitialConditions(SQLSubModel):
                 for data in self._data:
                     data_height[data["rk"].rk] = data["height"]
 
-        print(data_height)
-
         incline = self._reach.reach.get_incline_median_mean()
         logger.debug(f"incline = {incline}")
 

src/Model/OutputRKAdists/OutputRKAdists.py

@@ -193,7 +193,7 @@ class OutputRKAdists(SQLSubModel):
             "output_rk_adists(pamhyr_id, deleted, " +
             "reach, rk, title, scenario) " +
             "VALUES (" +
-            f"{self.id}, {self._db_format(self.is_deleted())}" +
+            f"{self.id}, {self._db_format(self.is_deleted())}, " +
             f"{self._reach}, {self._rk}, " +
             f"'{self._db_format(self._title)}', " +
             f"{self._status.scenario_id}" +

src/Model/Reservoir/Reservoir.py

@@ -263,8 +263,6 @@ class Reservoir(SQLSubModel):
         cls._db_update_to_0_2_0_set_node_pid(execute, table, nodes)
         Scenario.update_db_add_scenario(execute, table)
 
-        print(execute(f"SELECT * FROM {table}"))
-
         cls._db_create(execute, ext="_tmp")
         execute(
             f"INSERT INTO {table}_tmp " +

src/Model/Scenarios.py

@@ -16,6 +16,8 @@
 
 # -*- coding: utf-8 -*-
 
+from functools import reduce
+
 from tools import logger_exception
 
 from Model.Tools.PamhyrDict import PamhyrModelDict
@@ -78,3 +80,12 @@ class Scenarios(PamhyrModelDict):
 
         self._dict[key].set_as_deleted()
         self._status.modified()
+
+    def is_leaf(self, scenario):
+        return not reduce(
+            lambda acc, s: (
+                acc or (not s.is_deleted() and s.parent is scenario)
+            ),
+            self._dict.values(),
+            False
+        )

src/Model/Tools/PamhyrDB.py

@@ -26,6 +26,8 @@ from functools import reduce
 from SQL import SQL
 from Model.Except import NotImplementedMethodeError
 
+from tools import trace
+
 from Model.Tools.PamhyrID import PamhyrID
 
 logger = logging.getLogger()
@@ -48,11 +50,9 @@ class SQLModel(SQL):
         self._cur = self._db.cursor()
 
         if is_new:
-            logger.info("Create database")
             self._create()      # Create db
             # self._save()        # Save
         else:
-            logger.info("Update database")
             self._update()      # Update db scheme if necessary
             # self._load()        # Load data
 
@@ -313,8 +313,10 @@ class SQLSubModel(PamhyrID):
 
             if node_id not in nodes:
                 # ⚠️ cas important : probablement déjà migré
-                print(f"[WARN] node_id {node_id} not in nodes " +
-                      f"→ probably already migrated")
+                logger.warning(
+                    f"node_id {node_id} not in nodes " +
+                    "→ probably already migrated"
+                )
                 continue
 
             execute(

src/Solver/RubarBE.py

@@ -426,7 +426,6 @@ class Rubar3(CommandLineSolver):
                         coeff = coeff_min
                     else:
                         for s in lst:
-                            print(s.begin_rk, s.end_rk)
                             if (rk >= s.begin_rk and rk <= s.end_rk or
                                     rk <= s.begin_rk and rk >= s.end_rk):
                                 coeff = s.begin_strickler  # TODO: inerpolate
@@ -473,7 +472,6 @@ class Rubar3(CommandLineSolver):
                 last = profiles[-1]
 
                 if first not in data or last not in data:
-                    print(data)
                     logger.error(
                         "Study initial condition is not fully defined"
                     )

src/View/Geometry/Table.py

@@ -251,7 +251,6 @@ class GeometryReachTableModel(PamhyrTableModel):
         self.layoutChanged.emit()
 
     def meshing(self, mesher, data, tableView):
-
         new_profiles = mesher.meshing(
                 self._data,
                 **data
@@ -287,7 +286,6 @@ class GeometryReachTableModel(PamhyrTableModel):
         self.layoutChanged.emit()
 
     def purge(self, np_purge):
-
         self._undo.push(
             PurgeCommand(
                 self._data, np_purge
@@ -296,7 +294,6 @@ class GeometryReachTableModel(PamhyrTableModel):
         self.layoutChanged.emit()
 
     def shift(self, rows, dx, dy, dz):
-
         self._undo.push(
             ShiftCommand(
                 self._data, rows, dx, dy, dz
@@ -305,7 +302,6 @@ class GeometryReachTableModel(PamhyrTableModel):
         self.layoutChanged.emit()
 
     def change_reach(self, new_reach, parent):
-
         self._undo.push(
             ChangeReachCommand(
                 new_reach,

src/View/Geometry/UndoCommand.py

@@ -411,13 +411,13 @@ class ShiftCommand(QUndoCommand):
     def undo(self):
         for i in self._rows:
             profile = self._reach.profiles[i]
-            self._reach.profiles[i].shift(
+            profile.shift(
                 -self._dx, -self._dy, -self._dz
             )
 
     def redo(self):
         for i in self._rows:
             profile = self._reach.profiles[i]
-            self._reach.profiles[i].shift(
+            profile.shift(
                 self._dx, self._dy, self._dz
             )

src/View/Network/GraphWidget.py

@@ -925,9 +925,11 @@ class GraphWidget(QGraphicsView):
         self.scale(scaleFactor, scaleFactor)
 
     def mousePressEvent(self, event):
-        if self._only_display or self.graph._status.is_read_only():
+        if self._only_display:
             return
 
+        locked = self.graph._status.is_read_only()
+
         pos = self.mapToScene(event.pos())
         self.clicked = True
 
@@ -936,9 +938,6 @@ class GraphWidget(QGraphicsView):
             super(GraphWidget, self).mousePressEvent(event)
             return
 
-        if self.graph._status.is_read_only():
-            return
-
         # Move item and select edge item
         if self._state == "move":
             self._selected_new_edge_src_node = None
@@ -948,39 +947,41 @@ class GraphWidget(QGraphicsView):
                 edge = items[0]
                 if edge:
                     self.set_current_edge(edge)
-            elif items and type(items[0]) is NodeItem:
-                self._mouse_origin_x = pos.x()
-                self._mouse_origin_y = pos.y()
-                self._current_moved_node = items[0]
+            if not locked:
+                if items and type(items[0]) is NodeItem:
+                    self._mouse_origin_x = pos.x()
+                    self._mouse_origin_y = pos.y()
+                    self._current_moved_node = items[0]
 
-        # Add nodes and edges
-        elif self._state == "add":
-            items = self.items(event.pos())
-            nodes = list(filter(lambda i: type(i) is NodeItem, items))
-            if not nodes:
-                self.add_node(pos)
-                self.set_selected_new_edge_src_node(None)
-            else:
-                if self.selected_new_edge_src_node() is None:
-                    self.set_selected_new_edge_src_node(nodes[0])
+        if not locked:
+            # Add nodes and edges
+            if self._state == "add":
+                items = self.items(event.pos())
+                nodes = list(filter(lambda i: type(i) is NodeItem, items))
+                if not nodes:
+                    self.add_node(pos)
+                    self.set_selected_new_edge_src_node(None)
                 else:
-                    self.add_edge(self.selected_new_edge_src_node(), nodes[0])
+                    if self.selected_new_edge_src_node() is None:
+                        self.set_selected_new_edge_src_node(nodes[0])
+                    else:
+                        self.add_edge(self.selected_new_edge_src_node(), nodes[0])
 
-        # Delete nodes and edges
-        elif self._state == "del":
-            self._selected_new_edge_src_node = None
-            items = list(
-                filter(
-                    lambda i: type(i) is NodeItem or type(i) is EdgeItem,
-                    self.items(event.pos())
+            # Delete nodes and edges
+            elif self._state == "del":
+                self._selected_new_edge_src_node = None
+                items = list(
+                    filter(
+                        lambda i: type(i) is NodeItem or type(i) is EdgeItem,
+                        self.items(event.pos())
+                    )
                 )
-            )
-            if len(items) > 0:
-                item = items[0]
-                if type(item) is NodeItem:
-                    self.del_node(item)
-                elif type(item) is EdgeItem:
-                    self.del_edge(item)
+                if len(items) > 0:
+                    item = items[0]
+                    if type(item) is NodeItem:
+                        self.del_node(item)
+                    elif type(item) is EdgeItem:
+                        self.del_edge(item)
 
         self.update()
         super(GraphWidget, self).mousePressEvent(event)
@@ -988,10 +989,12 @@ class GraphWidget(QGraphicsView):
     def mouseReleaseEvent(self, event):
         self.clicked = False
 
-        if self._only_display or self.graph._status.is_read_only():
+        if self._only_display:
             return
 
-        if self._state == "move":
+        locked = self.graph._status.is_read_only()
+
+        if not locked and self._state == "move":
             if self._current_moved_node is not None:
                 pos = self.mapToScene(event.pos())
                 self._undo.push(
@@ -1007,10 +1010,8 @@ class GraphWidget(QGraphicsView):
         super(GraphWidget, self).mouseReleaseEvent(event)
 
     def mouseMoveEvent(self, event):
-        if self.graph._status.is_read_only():
-            return
-
         pos = self.mapToScene(event.pos())
+        locked = self.graph._status.is_read_only()
 
         # Selecte item on the fly
         items = self.items(event.pos())

src/View/Scenarios/ContextMenu.py

@@ -57,6 +57,8 @@ class DefaultMenu(AbstractMenu):
 class ScenarioMenu(AbstractMenu):
     def run(self):
         item = self._items[0]
+        scenarios = item.graph._study.scenarios
+
         current_scenario = item.graph._study.status.scenario.id
 
         select = self._menu.addAction(self._trad["menu_select_scenario"])
@@ -64,7 +66,9 @@ class ScenarioMenu(AbstractMenu):
         delete = None
 
         if item.scenario.id != 0:
-            delete = self._menu.addAction(self._trad["menu_del_scenario"])
+            if scenarios.is_leaf(item.scenario):
+                delete = self._menu.addAction(self._trad["menu_del_scenario"])
+
             if item.scenario.id == current_scenario:
                 duplicate = self._menu.addAction(
                     self._trad["menu_dup_scenario"]

src/View/Scenarios/GraphWidget.py

@@ -470,9 +470,16 @@ class GraphWidget(QGraphicsView):
         self.changeScenario.emit(self.sender())
 
     def new_scenario(self, pos):
+        must_save = self.dialog_save()
+        if must_save == "Cancel":
+            return
+
         def fn():
             self._close_other_window()
-            self._study.save()
+
+            if must_save == "Save":
+                self._study.save()
+
             self._undo.push(
                 AddScenariosCommand(
                     self._study,
@@ -500,9 +507,16 @@ class GraphWidget(QGraphicsView):
         self.changeScenario.emit(self.sender())
 
     def duplicate_scenario(self, item):
+        must_save = self.dialog_save()
+        if must_save == "Cancel":
+            return
+
         def fn():
             self._close_other_window()
-            # self._study.save()
+
+            if must_save == "Save":
+                self._study.save()
+
             self._undo.push(
                 DuplicateScenariosCommand(
                     self._study,