river, network: Minor fix and change.

This commit is contained in:
Pierre-Antoine Rouby committed 2023-04-05 15:26:14 +02:00
1 parent 150388af12
commit b6dbedf7fa
4 files changed
+49 -13

No files matched your search

+19 -3
View File
@@ -15,7 +15,14 @@ class RiverNode(Node):
x, y
)
self.locker = None
self._locker = None
def locker(self):
return self._locker
def locker(self, locker):
self._locker = locker
class RiverReach(Edge):
def __init__(self, id:str, name:str,
@@ -26,7 +33,14 @@ class RiverReach(Edge):
node1, node2
)
self.reach = None
self._reach = None
def reach(self):
return self._reach
def reach(self, reach:Reach):
self._reach = reach
class River(Graph):
def __init__(self):
@@ -38,9 +52,11 @@ class River(Graph):
self._current_reach = None
def has_current_reach(self):
return self._current_reach is not None
def current_reach(self):
return self._current_reach
def current_reach(self, reach):
def set_current_reach(self, reach):
self._current_reach = reach