def get_bc_package_cells(package, exclude_horizontal=True):
"""
Parameters
----------
package : flopy package instance for boundary condition
Returns
-------
k, i, j : 1D numpy arrays of boundary condition package cell locations
"""
if package.package_type == 'sfr':
if package.parent.version == 'mf6':
k, i, j = cellids_to_kij(package.packagedata.array['cellid'])
else:
rd = package.reach_data
k, i, j = rd['k'], rd['i'], rd['j']
elif package.package_type == 'lak':
if package.parent.version == 'mf6':
connectiondata = package.connectiondata.array
if exclude_horizontal:
connectiondata = connectiondata[connectiondata['claktype'] ==
'vertical']
k, i, j = map(np.array, zip(*connectiondata['cellid']))
else:
try:
# todo: figure out why flopy sometimes can't read external files for lakarr
k, i, j = np.where(package.lakarr.array[0, :, :, :] > 0)
except:
k, i, j = np.where(package.parent.lakarr > 0)
else:
df = mftransientlist_to_dataframe(package.stress_period_data,
squeeze=True)
k, i, j = df['k'].values, df['i'].values, df['j'].values
return k, i, j
def test_idomain_above_sfr(model_with_sfr):
m = model_with_sfr
sfr = m.sfr
# get the kij locations of sfr reaches
k, i, j = zip(*sfr.reach_data[['k', 'i', 'j']])
# verify that streambed tops are above layer bottoms
assert np.all(
sfr.packagedata.array['rtp'] > np.all(m.dis.botm.array[k, i, j]))
# test that idomain above sfr cells is being set to 0
# by setting all botms above streambed tops
new_botm = m.dis.botm.array.copy()
new_top = m.dis.top.array.copy()
new_botm[:, i, j] = 9999
new_top[i, j] = 9999
np.savetxt(m.cfg['dis']['griddata']['top'][0]['filename'], new_top)
m.dis.botm = new_botm
#m.dis.top = new_top
m.remove_package(sfr)
m._reset_bc_arrays()
assert not np.any(m._isbc2d == 4)
sfr = m.setup_sfr()
# test loading a 3d array from a filelist
idomain = load_array(m.cfg['dis']['griddata']['idomain'])
assert np.array_equal(m.idomain, idomain)
# dis package idomain of model instance attached to sfrdata
# forms basis for identifying unconnected cells
assert np.array_equal(m.idomain, m.sfrdata.model.idomain)
assert np.array_equal(m.idomain, m.sfrdata.model.dis.idomain.array)
# verify that dis package file still references external file
m.dis.write()
fname = os.path.join(m.model_ws, m.dis.filename)
assert os.path.getsize(fname) < 3e3
# idomain should be zero everywhere there's a sfr reach
# except for in the botm layer
# (verifies that model botm was reset to accomdate SFR reaches)
assert np.array_equal(m.sfr.reach_data.i, i)
assert np.array_equal(m.sfr.reach_data.j, j)
k, i, j = cellids_to_kij(sfr.packagedata.array['cellid'])
assert idomain[:-1, i, j].sum() == 0
active = np.array([
True if c != 'none' else False for c in sfr.packagedata.array['cellid']
])
assert idomain[-1, i, j].sum() == active.sum()
# assert np.all(m.dis.botm.array[:-1, i, j] > 9980)
assert np.all(m.dis.botm.array[-1, i, j] < 100)
def test_idomain_above_sfr(model_with_sfr):
m = model_with_sfr
sfr = m.sfr
# get the kij locations of sfr reaches
k, i, j = zip(*sfr.reach_data[['k', 'i', 'j']])
# verify that streambed tops are above layer bottoms
assert np.all(
sfr.packagedata.array['rtp'] > np.all(m.dis.botm.array[k, i, j]))
# test that idomain above sfr cells is being set to 0
# by setting all botms above streambed tops
new_botm = m.dis.botm.array.copy()
new_top = m.dis.top.array.copy()
new_botm[:-1, i, j] = 9999
new_botm[-1, i, j] = 9990
new_top[i, j] = 9999
np.savetxt(m.cfg['dis']['griddata']['top'][0]['filename'], new_top)
m.dis.botm = new_botm
m.dis.top = new_top
# reset external files for model top
# (that are used to cache an original version of the model top
# prior to any adjustment to lake bottoms)
from pathlib import Path
original_top_file = Path(
m.tmpdir, f"{m.name}_{m.cfg['dis']['top_filename_fmt']}.original")
original_top_file.unlink()
# if original_top_file is not found or invalid,
# the routine in sourcedata.setup_array for setting up the botm array
# attempts to write original_top_file from
# m.cfg['intermediate_data']['top']
# successive calls to sourcedata.setup_array
# in the context of setting up the bottom array
# then reference this "original" top,
# so if adjustments to lake bathymetry are made,
# they are only made relative to the "original" top,
# and not a revised top (which would keep pushing the bottoms downward)
np.savetxt(m.cfg['intermediate_data']['top'][0], new_top)
m.remove_package(sfr)
m._reset_bc_arrays()
assert not np.any(m._isbc2d == 4)
sfr = m.setup_sfr()
# test loading a 3d array from a filelist
idomain = load_array(m.cfg['dis']['griddata']['idomain'])
assert np.array_equal(m.idomain, idomain)
# dis package idomain of model instance attached to sfrdata
# forms basis for identifying unconnected cells
assert np.array_equal(m.idomain, m.sfrdata.model.idomain)
assert np.array_equal(m.idomain, m.sfrdata.model.dis.idomain.array)
# verify that dis package file still references external file
m.dis.write()
fname = os.path.join(m.model_ws, m.dis.filename)
assert os.path.getsize(fname) < 3e3
# idomain should be zero everywhere there's a sfr reach
# except for in the botm layer
# (verifies that model botm was reset to accomdate SFR reaches)
assert np.array_equal(m.sfr.reach_data.i, i)
assert np.array_equal(m.sfr.reach_data.j, j)
k, i, j = cellids_to_kij(sfr.packagedata.array['cellid'])
assert idomain[:-1, i, j].sum() == 0
active = np.array([
True if c != 'none' else False for c in sfr.packagedata.array['cellid']
])
assert idomain[-1, i, j].sum() == active.sum()
# assert np.all(m.dis.botm.array[:-1, i, j] > 9980)
assert np.all(m.dis.botm.array[-1, i, j] < 100)