check adjoining nodes in regions adjacent to main channel are dry.

1 files changed, 9 insertions, 0 deletions

diff --git a/makeBoundary.py b/makeBoundary.py
index 78fd532..3e2ebd0 100755
--- a/makeBoundary.py
+++ b/makeBoundary.py
@@ -219,6 +219,7 @@ print('zmax =', zmax)
 #print(h_i)
 
 
+
 p_i = [[] for _ in range(num_panels)]
 A_i = [[] for _ in range(num_panels)]
 r_h = [[] for _ in range(num_panels)]
@@ -226,6 +227,14 @@ h_i = [[] for _ in range(num_panels)]
 K_i = [[] for _ in range(num_panels)]
 Q_i = [[] for _ in range(num_panels)]
 for p in range(num_panels):
+    if p == panel[0]-1 and zregion[p][-1] < zmax:
+        # ensure end node in region to the left of channel is dry:
+        xregion[p] = np.append(xregion[p], xin[marker_ind[p]])
+        zregion[p] = np.append(zregion[p], zin[marker_ind[p]])
+    if p == panel[0]+1 and zregion[p][0] < zmax:
+        # ensure start node in region to the right of channel is dry:
+        xregion[p] = np.insert(xregion[p], 0, xin[marker_ind[p]-1])
+        zregion[p] = np.insert(zregion[p], 0, zin[marker_ind[p]-1])
     if zmax > zmin[p]:
         p_i[p], A_i[p], r_h[p], h_i[p], K_i[p], Q_i[p] = conveyance(
             numH,