auto BC generation

src/Model/BoundaryCondition/BoundaryCondition.py

@@ -371,3 +371,15 @@ class BoundaryCondition(SQLSubModel):
             d = self._data
             d[index], d[prev] = d[prev], d[index]
             self._status.modified()
+
+    def reach(self, river):
+        edges = []
+        if self._node != None:
+            if river != None:
+                for edge in river.edges():
+                    if edge.node1.name == self._node.name:
+                        edges.append(edge.reach)
+                    if edge.node2.name == self._node.name:
+                        edges.append(edge.reach)
+
+        return edges

src/Model/Geometry/ProfileXYZ.py

@@ -436,8 +436,6 @@ class ProfileXYZ(Profile, SQLSubModel):
                 area += poly.area
         return area
 
-        return poly.area
-
     def wet_radius(self, z):
         p = self.wet_perimeter(z)
         a = self.wet_area(z)
@@ -482,13 +480,14 @@ class ProfileXYZ(Profile, SQLSubModel):
             if zz[i] < z:
                 line.append([station[i], zz[i]])
 
-            if zz[i] <= z and zz[i+1] > z:
+            if zz[i] <= z and zz[i+1] >= z:
                 y = np.interp(
                     z,
                     [zz[i], zz[i+1]],
                     [station[i], station[i+1]]
                 )
                 line.append([y, z])
+                if len(line) > 2:
                     lines.append(geometry.LineString(line))
                 line = []
 

src/View/BoundaryCondition/Edit/Table.py

@@ -48,6 +48,7 @@ from Model.BoundaryCondition.BoundaryConditionTypes import (
 from View.BoundaryCondition.Edit.UndoCommand import (
     SetDataCommand, AddCommand, DelCommand,
     SortCommand, MoveCommand, PasteCommand,
+    ReplaceDataCommand,
 )
 
 _translate = QCoreApplication.translate
@@ -200,3 +201,14 @@ class TableModel(PamhyrTableModel):
     def update(self):
         # self.auto_sort()
         self.layoutChanged.emit()
+
+    def replace_data(self, data1, data2):
+        self.layoutAboutToBeChanged.emit()
+        self._undo.push(
+            ReplaceDataCommand(
+                self._data, data1, data2
+            )
+        )
+        self.layoutAboutToBeChanged.emit()
+        self.update()
+

src/View/BoundaryCondition/Edit/UndoCommand.py

@@ -181,3 +181,29 @@ class PasteCommand(QUndoCommand):
     def redo(self):
         for bc in self._bcs:
             self._data.insert(self._row, bc)
+
+class ReplaceDataCommand(QUndoCommand):
+    def __init__(self, data, data1, data2):
+        QUndoCommand.__init__(self)
+        self._data = data
+        self._old_rows = len(data)
+        self._data1 = data1
+        self._data2 = data2
+        self._rows = len(data1)
+
+        self._old_bc = []
+        for row in range(self._old_rows):
+            self._old_bc.append((row, self._data.get_i(row)))
+        self._old_bc.sort()
+
+    def undo(self):
+        self._data.delete_i(list(range(self._rows)))
+        for row, el in self._old_bc:
+            self._data.insert(row, el)
+
+    def redo(self):
+        self._data.delete_i(list(range(self._old_rows)))
+        for row in range(self._rows):
+            self._data.add(row)
+            self._data._set_i_c_v(row, 0, self._data1[row])
+            self._data._set_i_c_v(row, 1, self._data2[row])

src/View/BoundaryCondition/Edit/Window.py

@@ -20,6 +20,8 @@ import logging
 
 from tools import timer, trace
 
+from numpy import sqrt
+
 from View.Tools.PamhyrWindow import PamhyrWindow
 from View.Tools.PamhyrWidget import PamhyrWidget
 from View.Tools.PamhyrDelegate import PamhyrExTimeDelegate
@@ -104,6 +106,7 @@ class EditBoundaryConditionWindow(PamhyrWindow):
         self._data = data
         trad = BCETranslate()
         self._long_types = trad.get_dict("long_types")
+        self._study = study
 
         name = trad[self._pamhyr_name]
         if self._data is not None:
@@ -193,6 +196,14 @@ class EditBoundaryConditionWindow(PamhyrWindow):
         self.find(QAction, "action_add").triggered.connect(self.add)
         self.find(QAction, "action_del").triggered.connect(self.delete)
         self.find(QAction, "action_sort").triggered.connect(self.sort)
+        self.find(QAction, "action_generate_uniform").triggered.connect(self.generate_uniform)
+        self.find(QAction, "action_generate_critical").triggered.connect(self.generate_critical)
+        if self._data.bctype != "ZD" or not self._data.has_node:
+            self.find(QAction, "action_generate_uniform").setVisible(False)
+            self.find(QAction, "action_generate_critical").setVisible(False)
+        else:
+            self.find(QAction, "action_generate_uniform").setVisible(True)
+            self.find(QAction, "action_generate_critical").setVisible(True)
 
         self._table.dataChanged.connect(self.update)
         self._table.layoutChanged.connect(self.update)
@@ -311,3 +322,37 @@ class EditBoundaryConditionWindow(PamhyrWindow):
         self._table.redo()
         self.plot.update()
         self.widget_update()
+
+    def generate_uniform(self):
+        if self._data.has_node:
+            node = self._data.node
+            reach = self._data.reach(self._study.river)[0]
+            profile = reach.profiles[-1]
+            incline = reach.get_incline_median_mean()
+            z_min = profile.z_min()
+            z_max = profile.z_max()
+            strickler = 25
+            height = [(i)*(z_max-z_min)/50 for i in range(51)]
+            q = [((profile.wet_width(z_min + h) * 0.8) * strickler
+                       * (h ** (5/3))
+                       * (abs(incline) ** (0.5)))
+                 for h in height
+                ]
+            self._table.replace_data(height, q)
+
+        return
+
+    def generate_critical(self):
+        if self._data.has_node:
+            node = self._data.node
+            reach = self._data.reach(self._study.river)[0]
+            profile = reach.profiles[-1]
+            incline = reach.get_incline_median_mean()
+            z_min = profile.z_min()
+            z_max = profile.z_max()
+            height = [(i)*(z_max-z_min)/50 for i in range(51)]
+            q = [sqrt(9.81 * (profile.wet_area(z_min + h) ** 3) / profile.wet_width(z_min + h))
+                 for h in height
+                ]
+            self._table.replace_data(height, q)
+        return

src/View/ui/EditBoundaryConditions.ui

@@ -73,6 +73,8 @@
    <addaction name="action_add"/>
    <addaction name="action_del"/>
    <addaction name="action_sort"/>
+   <addaction name="action_generate_uniform"/>
+   <addaction name="action_generate_critical"/>
   </widget>
   <action name="action_add">
    <property name="checkable">
@@ -119,6 +121,25 @@
     <string>Sort points</string>
    </property>
   </action>
+  <action name="action_generate_uniform">
+   <property name="text">
+    <string>Generate uniform</string>
+   </property>
+   <property name="iconText">
+    <string>Generate uniform</string>
+   </property>
+   <property name="toolTip">
+    <string>Generate rating curve from Manning law</string>
+   </property>
+  </action>
+  <action name="action_generate_critical">
+   <property name="text">
+    <string>Generate critical</string>
+   </property>
+   <property name="toolTip">
+    <string>Generate rating curve as Q(z) = Sqrt(g*S(z)^3/L(z))</string>
+   </property>
+  </action>
  </widget>
  <resources/>
  <connections/>