def setup_obs(self):
"""
Sets up the OBS utility.
Parameters
----------
Notes
-----
"""
package = 'obs'
print('\nSetting up {} package...'.format(package.upper()))
t0 = time.time()
# munge the observation data
df = setup_head_observations(self,
format=package,
obsname_column='obsname')
# reformat to flopy input format
obsdata = df[['obsname', 'obstype', 'id']].to_records(index=False)
filename = self.cfg[package]['filename_fmt'].format(self.name)
obsdata = {filename: obsdata}
kwargs = self.cfg[package].copy()
kwargs.update(self.cfg[package]['options'])
kwargs['continuous'] = obsdata
kwargs = get_input_arguments(kwargs, mf6.ModflowUtlobs)
obs = mf6.ModflowUtlobs(self, **kwargs)
print("finished in {:.2f}s\n".format(time.time() - t0))
return obs
def setup_hyd(self):
"""TODO: generalize hydmod setup with specific input requirements"""
package = 'hyd'
print('setting up HYDMOD package...')
t0 = time.time()
# munge the head observation data
df = setup_head_observations(self,
format=package,
obsname_column='hydlbl')
# create observation data recarray
obsdata = fm.ModflowHyd.get_empty(len(df))
for c in obsdata.dtype.names:
assert c in df.columns, "Missing observation data field: {}".format(
c)
obsdata[c] = df[c]
nhyd = len(df)
hyd = flopy.modflow.ModflowHyd(self,
nhyd=nhyd,
hydnoh=-999,
obsdata=obsdata)
print("finished in {:.2f}s\n".format(time.time() - t0))
return hyd