Skip to content

Commit 579507e

Browse files
update(splitter): updates for advanced transport packages (#2761)
* update(splitter): updates for advanced transport packages * update flow package name mapping for transport models --------- Co-authored-by: jlarsen-usgs <jlarsen@usgs.gov>
1 parent e9f513d commit 579507e

1 file changed

Lines changed: 18 additions & 6 deletions

File tree

flopy/mf6/utils/model_splitter.py

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1677,10 +1677,23 @@ def _remap_adv_transport(self, package, item, pkg_remap, mapped_data):
16771677
"""
16781678
flow_package_name = package.flow_package_name.array
16791679
packagedata = package.packagedata.array
1680-
perioddata = package.perioddata.data
1680+
if isinstance(package, modflow.ModflowGwtsft):
1681+
key = "reachperioddata"
1682+
perioddata = package.reachperioddata.data
1683+
elif isinstance(package, modflow.ModflowGwtlkt):
1684+
key = "lakeperioddata"
1685+
perioddata = package.lakeperioddata.data
1686+
elif isinstance(package, modflow.ModflowGwtuzt):
1687+
key = "uztperioddata"
1688+
perioddata = package.uztperioddata.data
1689+
elif isinstance(package, modflow.ModflowGwtmwt):
1690+
key = "mwtperioddata"
1691+
perioddata = package.mwtperioddata.data
1692+
else:
1693+
key = "perioddata"
1694+
perioddata = package.perioddata.data
16811695

16821696
for mkey in self._model_dict.keys():
1683-
flow_package_const = flow_package_name.split(".")
16841697
new_packagedata = self._remap_adv_tag(
16851698
mkey, packagedata, item, pkg_remap
16861699
)
@@ -1694,10 +1707,9 @@ def _remap_adv_transport(self, package, item, pkg_remap, mapped_data):
16941707
)
16951708
spd[per] = new_recarray
16961709

1697-
flow_package_const[-2] += f"_{mkey :0{self._fdigits}d}"
1698-
new_flow_package_name = ".".join(flow_package_const)
1710+
new_flow_package_name = flow_package_name
16991711
mapped_data[mkey]["packagedata"] = new_packagedata
1700-
mapped_data[mkey]["perioddata"] = spd
1712+
mapped_data[mkey][key] = spd
17011713
mapped_data[mkey]["flow_package_name"] = new_flow_package_name
17021714
return mapped_data
17031715

@@ -2038,7 +2050,7 @@ def _remap_sfr(self, package, mapped_data):
20382050
connectiondata = package.connectiondata.array
20392051
diversions = package.diversions.array
20402052
perioddata = package.perioddata.data
2041-
name = package.filename
2053+
name = package.name[0]
20422054
self._sfr_remaps[name] = {}
20432055
sfr_remaps = {}
20442056
div_mvr_conn = {}

0 commit comments

Comments
 (0)