diff options
| -rwxr-xr-x | makeBoundary.py | 9 |
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, |