from typing import Any, Optional
import numpy as np
from imod.common.interfaces.iregridpackage import IRegridPackage
from imod.common.utilities.dataclass_type import (
DataclassType,
)
from imod.common.utilities.regrid import _regrid_package_data
from imod.logging import init_log_decorator
from imod.mf6.package import Package
from imod.mf6.regrid.regrid_schemes import (
InitialConditionsRegridMethod,
)
from imod.mf6.validation import PKG_DIMS_SCHEMA
from imod.schemata import (
AllCoordsValueSchema,
DTypeSchema,
IdentityNoDataSchema,
IndexesSchema,
)
from imod.typing import GridDataArray
from imod.util.regrid import RegridderWeightsCache
[docs]
class InitialConditions(Package, IRegridPackage):
"""
Initial Conditions (IC) Package information is read from the file that is
specified by "IC6" as the file type. Only one IC Package can be specified
for a GWF model.
https://water.usgs.gov/water-resources/software/MODFLOW-6/mf6io_6.0.4.pdf#page=46
Parameters
----------
start: array of floats (xr.DataArray)
is the initial (starting) head or concentration—that is, the simulation's
initial state.
STRT must be specified for all simulations, including steady-state simulations.
One value is read for every model cell. For
simulations in which the first stress period is steady state, the values
used for STRT generally do not affect the simulation (exceptions may
occur if cells go dry and (or) rewet). The execution time, however, will
be less if STRT includes hydraulic heads that are close to the
steadystate solution. A head value lower than the cell bottom can be
provided if a cell should start as dry. (strt)
validate: {True, False}
Flag to indicate whether the package should be validated upon
initialization. This raises a ValidationError if package input is
provided in the wrong manner. Defaults to True.
"""
_pkg_id = "ic"
_grid_data = {"start": np.float64}
_keyword_map = {"start": "strt"}
_template = Package._initialize_template(_pkg_id)
_init_schemata = {
"start": [
DTypeSchema(np.floating),
IndexesSchema(),
PKG_DIMS_SCHEMA,
AllCoordsValueSchema("layer", ">", 0),
],
}
_write_schemata = {
"start": [
IdentityNoDataSchema(other="idomain", is_other_notnull=(">", 0)),
],
}
_regrid_method = InitialConditionsRegridMethod()
[docs]
@init_log_decorator()
def __init__(self, start, validate: bool = True):
dict_dataset = {"start": start}
super().__init__(dict_dataset)
self._validate_init_schemata(validate)
def _render(self, directory, pkgname, globaltimes, binary):
d: dict[str, Any] = {}
icdirectory = directory / pkgname
d["layered"], d["strt"] = self._compose_values(
self["start"], icdirectory, "strt", binary=binary
)
return self._template.render(d)
[docs]
@classmethod
def from_imod5_data(
cls,
imod5_data: dict[str, dict[str, GridDataArray]],
target_grid: GridDataArray,
regridder_types: Optional[DataclassType] = None,
regrid_cache: RegridderWeightsCache = RegridderWeightsCache(),
) -> "InitialConditions":
"""
Construct an InitialConditions-package from iMOD5 data, loaded with the
:func:`imod.formats.prj.open_projectfile_data` function.
.. note::
The method expects the iMOD5 model to be fully 3D, not quasi-3D.
Parameters
----------
imod5_data: dict
Dictionary with iMOD5 data. This can be constructed from the
:func:`imod.formats.prj.open_projectfile_data` method.
target_grid: GridDataArray
The grid that should be used for the new package. Does not
need to be identical to one of the input grids.
regridder_types: InitialConditionsRegridMethod, optional
Optional dataclass with regridder types for a specific variable.
Use this to override default regridding methods.
regrid_cache: RegridderWeightsCache, optional
stores regridder weights for different regridders. Can be used to speed up regridding,
if the same regridders are used several times for regridding different arrays.
Returns
-------
Modflow 6 InitialConditions package.
"""
data = {
"start": imod5_data["shd"]["head"],
}
if regridder_types is None:
regridder_types = cls.get_regrid_methods()
new_package_data = _regrid_package_data(
data, target_grid, regridder_types, regrid_cache, {}
)
return cls(**new_package_data, validate=True)
