src/Model/Geometry/Point.py

@@ -58,7 +58,7 @@ class Point(SQLSubModel):
         self._name = name
         self.modified()
 
-    def is_named(self):
+    def point_is_named(self):
         """
         Returns:
             True if the point is named.

src/Model/Geometry/PointXYZ.py

@@ -292,9 +292,6 @@ class PointXYZ(Point):
     def __repr__(self):
         return f"({self._x}, {self._y}, {self._z}, {self._name})"
 
-    def __gt__(self, other):
-        return self.y > other.y
-
     @property
     def x(self):
         return self._x
@@ -371,10 +368,5 @@ class PointXYZ(Point):
         a = PointXYZ.distance(p1, p2)
         b = PointXYZ.distance(p2, p3)
         c = PointXYZ.distance(p3, p1)
-
-        s = float((a + b + c) / 2)
-        res = (
-            s * abs(s - a) * abs(s - b) * abs(s - c)
-        ) ** 0.5
-
-        return res
+        s = (a + b + c) / 2
+        return (s*(s-a) * (s-b)*(s-c)) ** 0.5

src/Model/Geometry/Profile.py

@@ -187,7 +187,7 @@ class Profile(object):
             The list of named point
         """
         return [point for point in self.points
-                if point.is_named()]
+                if point.point_is_named()]
 
     def insert_point(self, index: int, point: Point):
         """Insert point at index.

src/Model/Geometry/ProfileXYZ.py

@@ -16,8 +16,6 @@
 
 # -*- coding: utf-8 -*-
 
-import queue
-import random
 import logging
 import numpy as np
 from typing import List
@@ -931,13 +929,13 @@ class ProfileXYZ(Profile, SQLSubModel):
         last_point_not_nan = self._last_point_not_nan(points)
 
         for index, point in enumerate(points):
-            if point.is_named():
+            if point.point_is_named():
                 index_first_named_point = index
                 first_named_point = point
                 break
 
         for point in reversed(points):
-            if point.is_named():
+            if point.point_is_named():
                 last_named_point = point
                 break
 
@@ -1011,46 +1009,46 @@ class ProfileXYZ(Profile, SQLSubModel):
         if (self.nb_points <= np_purge):
             return
 
-        q_area = queue.PriorityQueue()
-        deleted = []
+        nb_named = 2   # we consider the first and last point as named
+        area = [0.0]
 
-        for ind, point in enumerate(self.points):
-            if (ind == 0) or (ind == self.nb_points - 1):
-                continue
-
-            if not point.is_named():
-                q_area.put((
-                    PointXYZ.areatriangle3d(
-                        self.point(ind - 1),
-                        point,
-                        self.point(ind + 1)
-                    ),
-                    ind, point
-                ))
-
-        while self.nb_points > np_purge and not q_area.empty():
-            area, ind, point = q_area.get()
-
-            self.delete_points([point])
-            deleted.append(point)
-
-            # Recompute point neighbourhood
-            for i in [ind-1, ind]:
-                if (i <= 0) or (i >= self.nb_points - 1):
-                    continue
-
-                point = self.point(i)
-                if not point.is_named():
-                      q_area.put((
+        for i in range(1, self.nb_points-1):
+            if self.point(i).point_is_named():
+                area.append(9999999.999)
+                nb_named += 1
+            else:
+                area.append(
                     PointXYZ.areatriangle3d(
                         self.point(i-1),
-                              point,
+                        self.point(i),
+                        self.point(i+1))
+                )
+
+        area.append(0.0)
+
+        while self.nb_points > max(np_purge, nb_named):
+            to_rm = np.argmin(area[1:self.nb_points - 1]) + 1
+
+            self.delete_i([to_rm])
+            area.pop(to_rm)
+
+            for i in [to_rm-1, to_rm]:
+                if (i == 0):
+                    continue
+
+                if (i == self.nb_points - 1):
+                    continue
+
+                if self.point(i).point_is_named():
+                    area[i] = 9999999.999
+                else:
+                    area[i] = PointXYZ.areatriangle3d(
+                        self.point(i-1),
+                        self.point(i),
                         self.point(i+1)
-                          ), i, point
-                      ))
+                    )
 
         self.modified()
-        return deleted
 
     def shift(self, x, y, z):
         for p in self._points:

src/Model/InitialConditions/InitialConditions.py

@@ -375,7 +375,7 @@ class InitialConditions(SQLSubModel):
         return ok
 
     def __len__(self):
-        return len(self.data)
+        return len(self._data)
 
     def lst(self):
         return self._data
@@ -403,7 +403,7 @@ class InitialConditions(SQLSubModel):
         self._data = data
 
     def get(self, index):
-        return self.data[index]
+        return self._data[index]
 
     def set(self, index, data):
         self._data.insert(index, data)
@@ -441,14 +441,11 @@ class InitialConditions(SQLSubModel):
 
     def delete(self, data):
         list(
-            map(
-                lambda x: x.set_as_deleted(),
             filter(
-                    lambda x: x in data,
+                lambda x: x.set_as_deleted(),
                 self._data
             )
         )
-        )
         self.modified()
 
     def delete_i(self, indexes):
@@ -471,7 +468,7 @@ class InitialConditions(SQLSubModel):
         return list(
             map(
                 lambda d: d[key],
-                self.data
+                self._data
             )
         )
 
@@ -479,7 +476,7 @@ class InitialConditions(SQLSubModel):
         return list(
             map(
                 lambda d: d["rk"].rk,
-                self.data
+                self._data
             )
         )
 
@@ -489,10 +486,6 @@ class InitialConditions(SQLSubModel):
     def get_discharge(self):
         return self._data_get("discharge")
 
-    def clear_data(self):
-        for data in self._data:
-            data.set_as_deleted()
-
     def generate_growing_constant_depth(self, height: float,
                                         compute_discharge: bool):
         profiles = self._reach.reach.profiles.copy()
@@ -509,9 +502,7 @@ class InitialConditions(SQLSubModel):
 
         incline = self._reach.reach.get_incline_median_mean()
         logger.debug(f"incline = {incline}")
-
-        self.clear_data()
-
+        self._data = []
         for profile in reversed(profiles):
             width = profile.wet_width(profile.z_min() + height)
             area = profile.wet_area(profile.z_min() + height)
@@ -577,9 +568,7 @@ class InitialConditions(SQLSubModel):
 
         incline = self._reach.reach.get_incline_median_mean()
         logger.debug(f"incline = {incline}")
-
-        self.clear_data()
-
+        self._data = []
         for profile in reversed(profiles):
             width = profile.width_approximation()
             frictions = self._reach.frictions.frictions
@@ -642,7 +631,7 @@ class InitialConditions(SQLSubModel):
                 for data in self._data:
                     data_discharge[data["rk"].rk] = data["discharge"]
 
-        self.clear_data()
+        self._data = []
         for profile in profiles:
 
             if not compute_discharge:

src/View/Geometry/Profile/Table.py

@@ -193,7 +193,7 @@ class GeometryProfileTableModel(PamhyrTableModel):
             DelCommand(
                 self._data, list(
                     map(
-                        lambda r: self._data.point(r),
+                        lambda r: self.get_true_data_row(r),
                         rows
                     )
                 )
@@ -285,8 +285,6 @@ class GeometryProfileTableModel(PamhyrTableModel):
             )
         )
 
-        row = self.get_true_data_row(row)
-
         self._undo.push(
             PasteCommand(
                 self._data, row, points

src/View/Geometry/Profile/UndoCommand.py

@@ -116,19 +116,23 @@ class AddCommand(QUndoCommand):
 
 
 class DelCommand(QUndoCommand):
-    def __init__(self, profile, points):
+    def __init__(self, profile, rows):
         QUndoCommand.__init__(self)
 
         self._profile = profile
-        self._points = points
+        self._rows = rows
+
+        self._points = []
+        for row in rows:
+            self._points.append(self._profile.point(row))
 
     def undo(self):
-        for point in self._points:
-            point.set_as_not_deleted()
+        for row, point in zip(self._rows, self._points):
+            self._profile.insert_point(row, point)
         self._profile.modified()
 
     def redo(self):
-        self._profile.delete_points(self._points)
+        self._profile.delete_i(self._rows)
         self._profile.modified()
 
 
@@ -202,17 +206,16 @@ class PurgeCommand(QUndoCommand):
     def __init__(self, profile, np_purge):
         QUndoCommand.__init__(self)
 
-        self._deleted = []
         self._profile = profile
+        self._old = self._profile.points.copy()
         self._np_purge = np_purge
 
     def undo(self):
-        for point in self._deleted:
-            point.set_as_not_deleted()
+        self._profile._points = self._old.copy()
         self._profile.modified()
 
     def redo(self):
-        self._deleted = self._profile.purge(self._np_purge)
+        self._profile.purge(self._np_purge)
         self._profile.modified()
 
 
@@ -227,7 +230,8 @@ class PasteCommand(QUndoCommand):
         self._points.reverse()
 
     def undo(self):
-        self._profile.delete_points(self._points)
+        for ind in range(len(self._points)):
+            self._profile.delete_i([self._row])
         self._profile.modified()
 
     def redo(self):

src/View/Geometry/UndoCommand.py

@@ -277,17 +277,18 @@ class PurgeCommand(QUndoCommand):
 
         self._reach = reach
         self._np_purge = np_purge
-        self._deleted = {}
+
+        self._old = []
+        for profile in self._reach.profiles:
+            self._old.append(profile.points.copy())
 
     def undo(self):
-        for profile in self._deleted:
-            for point in self._deleted[profile]:
-                point.set_as_not_deleted()
-            profile.modified()
+        for i in range(self._reach.number_profiles):
+            self._reach.profiles[i]._points = self._old[i].copy()
 
     def redo(self):
         for profile in self._reach._profiles:
-            self._deleted[profile] = profile.purge(self._np_purge)
+            profile.purge(self._np_purge)
 
 
 class ChangeReachCommand(QUndoCommand):

src/View/InitialConditions/Table.py

@@ -104,19 +104,6 @@ class InitialConditionTableModel(PamhyrTableModel):
     def _setup_lst(self):
         self._lst = self._data.river.initial_conditions.get(self._reach)
 
-    def get_true_data_row(self, row):
-        data = self._lst.get(row)
-
-        return next(
-            map(
-                lambda e: e[0],
-                filter(
-                    lambda e: e[1] == data,
-                    enumerate(self._lst._data)
-                )
-            ), 0
-        )
-
     def data(self, index, role):
         if role != Qt.ItemDataRole.DisplayRole:
             return QVariant()
@@ -135,11 +122,8 @@ class InitialConditionTableModel(PamhyrTableModel):
                 return f"{velocity:.4f}"
 
             return ""
-
         if self._headers[column] == "rk":
             p = self._lst.get(row)[self._headers[column]]
-            if p is None:
-                return ""
             return f"{p.rk:.4f}"
         elif self._headers[column] not in ["name", "comment"]:
             v = self._lst.get(row)[self._headers[column]]
@@ -173,8 +157,6 @@ class InitialConditionTableModel(PamhyrTableModel):
     def add(self, row, parent=QModelIndex()):
         self.beginInsertRows(parent, row, row - 1)
 
-        row = self.get_true_data_row(row)
-
         self._undo.push(
             AddCommand(
                 self._lst, row
@@ -189,12 +171,7 @@ class InitialConditionTableModel(PamhyrTableModel):
 
         self._undo.push(
             DelCommand(
-                self._lst, list(
-                    map(
-                        lambda row: self._lst.get(row),
-                        rows
-                    )
-                )
+                self._lst, rows
             )
         )
 
@@ -217,10 +194,9 @@ class InitialConditionTableModel(PamhyrTableModel):
         if row <= 0:
             return
 
-        target = row + 1
-        self.beginMoveRows(parent, row - 1, row - 1, parent, target)
+        target = row + 2
 
-        row = self.get_true_data_row(row)
+        self.beginMoveRows(parent, row - 1, row - 1, parent, target)
 
         self._undo.push(
             MoveCommand(
@@ -239,8 +215,6 @@ class InitialConditionTableModel(PamhyrTableModel):
 
         self.beginMoveRows(parent, row + 1, row + 1, parent, target)
 
-        row = self.get_true_data_row(row)
-
         self._undo.push(
             MoveCommand(
                 self._lst, "down", row
@@ -265,8 +239,6 @@ class InitialConditionTableModel(PamhyrTableModel):
 
         self.layoutAboutToBeChanged.emit()
 
-        index = self.get_true_data_row(index)
-
         self._undo.push(
             InsertCommand(
                 self._lst, index,

src/View/InitialConditions/UndoCommand.py

@@ -75,18 +75,23 @@ class AddCommand(QUndoCommand):
 
 
 class DelCommand(QUndoCommand):
-    def __init__(self, ics, data):
+    def __init__(self, ics, rows):
         QUndoCommand.__init__(self)
 
         self._ics = ics
-        self._data = data
+        self._rows = rows
+
+        self._ic = []
+        for row in rows:
+            self._ic.append((row, self._ics.get(row)))
+        self._ic.sort()
 
     def undo(self):
-        for ic in self._data:
-            ic.set_as_not_deleted()
+        for row, el in self._ic:
+            self._ics.insert(row, el)
 
     def redo(self):
-        self._ics.delete(self._data)
+        self._ics.delete_i(self._rows)
 
 
 class SortCommand(QUndoCommand):