Skip to content

Commit 7d4fc6a

Browse files
mjrenomjreno
authored andcommitted
netcdf util uses data type specific interface
1 parent 5df30e0 commit 7d4fc6a

File tree

5 files changed

+106
-113
lines changed

5 files changed

+106
-113
lines changed

flopy/mf6/data/mfdataarray.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1602,10 +1602,10 @@ def _set_storage_netcdf(self, nc_dataset, create=False):
16021602
input_tag = f"{m}/{p}/{n}".lower()
16031603

16041604
if create:
1605-
# cache data and update file before resetting storage
1605+
# add array to netcdf dataset
16061606
nc_varname = f"{p}_{n}".lower()
16071607
data = self._get_data()
1608-
nc_dataset.create_var(nc_varname, input_tag, 0, data, self.structure.shape)
1608+
nc_dataset.create_array(nc_varname, input_tag, 0, data, self.structure.shape)
16091609

16101610
storage._set_storage_netcdf(
16111611
nc_dataset, input_tag, self.structure.layered, storage.layer_storage.get_total_size() - 1

flopy/mf6/data/mffileaccess.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -473,7 +473,7 @@ def load_netcdf_array(
473473
nc_tag,
474474
layer,
475475
):
476-
return nc_dataset.data(nc_tag, layer + 1)
476+
return nc_dataset.array(nc_tag, layer + 1)
477477

478478
def get_data_string(self, data, data_type, data_indent=""):
479479
layer_data_string = [str(data_indent)]

flopy/mf6/data/mfstructure.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1425,7 +1425,7 @@ def __init__(self, data_item, model_data, package_type, dfn_list):
14251425
or "nodes" in data_item.shape
14261426
or len(data_item.layer_dims) > 1
14271427
)
1428-
# TODO: only gwf, gwt, gwe models
1428+
# TODO: only gwf, gwt, gwe models NETCDF-DEV
14291429
self.netcdf = (
14301430
("ncol" in data_item.shape
14311431
or "nrow" in data_item.shape

flopy/mf6/mfmodel.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -937,19 +937,26 @@ def load_base(
937937
dis_str = {
938938
"dis6": "structured",
939939
"disv6": "vertex",
940-
"disu6": "unstructured",
940+
#"disu6": "unstructured",
941941
}
942942
dis_type = None
943943
for t in instance.name_file.packages.get_data():
944944
if t[0].lower().startswith("dis"):
945945
dis_type = t[0].lower()
946946
break
947-
if dis_type:
947+
if dis_type and dis_type in dis_str:
948948
nc_fpth = os.path.join(instance.model_ws, nc_filerecord[0][0])
949949
instance._nc_dataset = open_netcdf_dataset(nc_fpth, dis_type=dis_str[dis_type])
950950
else:
951-
pass
952-
# TODO: set error "invalid dis for netcdf input"
951+
message = (
952+
"Invalid discretization type "
953+
f"provided for model {modelname} "
954+
"NetCDF input"
955+
)
956+
raise MFDataException(
957+
model=modelname,
958+
message=message,
959+
)
953960

954961
# order packages
955962
vnum = mfstructure.MFStructure().get_version_string()

0 commit comments

Comments
 (0)