import numpy as np
import xarray as xr
from imod.logging import init_log_decorator
from imod.mf6.package import Package
from imod.schemata import AllValueSchema, DTypeSchema, OptionSchema
[docs]
class Solution(Package):
"""
Iterative Model Solution.
The model solution will solve all of the models that are added to it, as
specified in the simulation name file, and will include Numerical Exchanges,
if they are present. The iterative model solution requires specification of
both nonlinear and linear settings.
https://water.usgs.gov/water-resources/software/MODFLOW-6/mf6io_6.0.4.pdf#page=147
Three predifined solutions settings are available: SolutionPresetSimple,
SolutionPresetModerate and SolutionPresetComplex. When using one of the
predefined solutions only the print_option, csv_output, and no_ptc have to
be defined. The default values for each are described below.
Parameters
----------
modelnames: list of str
Which models to solve in this solution. Only models of the same type
(GWF or GWT) should be added to the same solution.
outer_dvclose: float
real value defining the head change criterion for convergence of the
outer (nonlinear) iterations, in units of length. When the maximum
absolute value of the head change at all nodes during an iteration is
less than or equal to outer_dvclose, iteration stops. Commonly,
outer_dvclose equals 0.01.
SolutionPresetSimple: 0.001
SolutionPresetModerate: 0.01
SolutionPresetComplex: 0.1
outer_maximum: int
integer value defining the maximum number of outer (nonlinear)
iterations - that is, calls to the solution routine. For a linear
problem outer_maximum should be 1.
SolutionPresetSimple: 25
SolutionPresetModerate: 50
SolutionPresetComplex: 100
inner_maximum: int
integer value defining the maximum number of inner (linear) iterations.
The number typically depends on the characteristics of the matrix
solution scheme being used. For nonlinear problems, inner_maximum
usually ranges from 60 to 600; a value of 100 will be sufficient for
most linear problems.
SolutionPresetSimple: 50
SolutionPresetModerate: 100
SolutionPresetComplex: 500
inner_dvclose: float
real value defining the head change criterion for convergence of the
inner (linear) iterations, in units of length. When the maximum absolute
value of the head change at all nodes during an iteration is less than
or equal to inner_dvclose, the matrix solver assumes convergence.
Commonly, inner_dvclose is set an order of magnitude less than the
outer_dvclose value.
SolutionPresetSimple: 0.001
SolutionPresetModerate: 0.01
SolutionPresetComplex: 0.1
inner_rclose: float
real value that defines the flow residual tolerance for convergence of
the IMS linear solver and specific flow residual criteria used. This
value represents the maximum allowable residual at any single node.
Value is in units of length cubed per time, and must be consistent with
MODFLOW 6 length and time units. Usually a value of 1.0 × 10−1 is
sufficient for the flow-residual criteria when meters and seconds are
the defined MODFLOW 6 length and time.
SolutionPresetSimple: 0.1
SolutionPresetModerate: 0.1
SolutionPresetComplex: 0.1
linear_acceleration: str
options: {"cg", "bicgstab"}
a keyword that defines the linear acceleration method used by the
default IMS linear solvers. CG - preconditioned conjugate gradient
method. BICGSTAB - preconditioned bi-conjugate gradient stabilized
method.
SolutionPresetSimple: "cg"
SolutionPresetModerate: "bicgstab"
SolutionPresetComplex: "bicgstab"
under_relaxation: str, optional
options: {None, "simple", "cooley", "dbd"}
is an optional keyword that defines the nonlinear relative_rclose
schemes used. By default under_relaxation is not used.
None - relative_rclose is not used.
simple - Simple relative_rclose scheme with a fixed relaxation factor is
used.
cooley - Cooley relative_rclose scheme is used.
dbd - delta-bar-delta relative_rclose is used.
Note that the relative_rclose schemes are used in conjunction with
problems that use the Newton-Raphson formulation, however, experience
has indicated that the Cooley relative_rclose and damping work well also
for the Picard scheme with the wet/dry options of MODFLOW 6.
Default value: None
SolutionPresetSimple: None
SolutionPresetModerate: "dbd"
SolutionPresetComplex: "dbd"
under_relaxation_theta: float, optional
real value defining the reduction factor for the learning rate
(underrelaxation term) of the delta-bar-delta algorithm. The value of
under relaxation theta is between zero and one. If the change in the
variable (head) is of opposite sign to that of the previous iteration,
the relative_rclose term is reduced by a factor of under relaxation
theta. The value usually ranges from 0.3 to 0.9; a value of 0.7 works
well for most problems. under relaxation theta only needs to be
specified if under relaxation is dbd.
Default value: None
SolutionPresetSimple: 0.0
SolutionPresetModerate: 0.9
SolutionPresetComplex: 0.8
under_relaxation_kappa: float, optional
real value defining the increment for the learning rate (relative_rclose
term) of the delta-bar-delta algorithm. The value of under relaxation
kappa is between zero and one. If the change in the variable (head) is
of the same sign to that of the previous iteration, the relative_rclose
term is increased by an increment of under_relaxation_kappa. The value
usually ranges from 0.03 to 0.3; a value of 0.1 works well for most
problems. under relaxation kappa only needs to be specified if under
relaxation is dbd.
Default value: None
SolutionPresetSimple: 0.0
SolutionPresetModerate: 0.0001
SolutionPresetComplex: 0.0001
under_relaxation_gamma: float, optional
real value defining the history or memory term factor of the
delta-bardelta algorithm. under relaxation gamma is between zero and 1
but cannot be equal to one. When under relaxation gamma is zero, only
the most recent history (previous iteration value) is maintained. As
under relaxation gamma is increased, past history of iteration changes
has greater influence on the memory term. The memory term is maintained
as an exponential average of past changes. Retaining some past history
can overcome granular behavior in the calculated function surface and
therefore helps to overcome cyclic patterns of nonconvergence. The value
usually ranges from 0.1 to 0.3; a value of 0.2 works well for most
problems. under relaxation gamma only needs to be specified if under
relaxation is not none.
Default value: None
SolutionPresetSimple: 0.0
SolutionPresetModerate: 0.0
SolutionPresetComplex: 0.0
under_relaxation_momentum: float, optional
real value defining the fraction of past history changes that is added
as a momentum term to the step change for a nonlinear iteration. The
value of under relaxation momentum is between zero and one. A large
momentum term should only be used when small learning rates are
expected. Small amounts of the momentum term help convergence. The value
usually ranges from 0.0001 to 0.1; a value of 0.001 works well for most
problems. under relaxation momentum only needs to be specified if under
relaxation is dbd.
Default value: None
SolutionPresetSimple: 0.0
SolutionPresetModerate: 0.0
SolutionPresetComplex: 0.0
backtracking_number: int, optional
integer value defining the maximum number of backtracking iterations
allowed for residual reduction computations. If backtracking number = 0
then the backtracking iterations are omitted. The value usually ranges
from 2 to 20; a value of 10 works well for most problems.
Default value: None
SolutionPresetSimple: 0
SolutionPresetModerate: 0
SolutionPresetComplex: 20
backtracking_tolerance: float, optional
real value defining the tolerance for residual change that is allowed
for residual reduction computations. backtracking tolerance should not
be less than one to avoid getting stuck in local minima. A large value
serves to check for extreme residual increases, while a low value serves
to control step size more severely. The value usually ranges from 1.0 to
106; a value of 104 works well for most problems but lower values like
1.1 may be required for harder problems. backtracking tolerance only
needs to be specified if backtracking_number is greater than zero.
Default value: None
SolutionPresetSimple: 0.0
SolutionPresetModerate: 0.0
SolutionPresetComplex: 1.05
backtracking_reduction_factor: float, optional
real value defining the reduction in step size used for residual
reduction computations. The value of backtracking reduction factor is
between 142 MODFLOW 6 - Description of Input and Output zero and one.
The value usually ranges from 0.1 to 0.3; a value of 0.2 works well for
most problems. backtracking_reduction_factor only needs to be specified
if backtracking number is greater than zero.
Default value: None
SolutionPresetSimple: 0.0
SolutionPresetModerate: 0.0
SolutionPresetComplex: 0.1
backtracking_residual_limit: float, optional
real value defining the limit to which the residual is reduced with
backtracking. If the residual is smaller than
backtracking_residual_limit, then further backtracking is not performed.
A value of 100 is suitable for large problems and residual reduction to
smaller values may only slow down computations. backtracking residual
limit only needs to be specified if backtracking_number is greater than
zero.
Default value: None
SolutionPresetSimple: 0.0
SolutionPresetModerate: 0.0
SolutionPresetComplex: 0.002
rclose_option: str, optional
options: {"strict", "l2norm_rclose", "relative_rclose"}
an optional keyword that defines the specific flow residual criterion
used.
strict: an optional keyword that is used to specify that inner rclose
represents a infinity-norm (absolute convergence criteria) and that the
head and flow convergence criteria must be met on the first inner
iteration (this criteria is equivalent to the criteria used by the
MODFLOW-2005 PCG package (Hill, 1990)).
l2norm_rclose: an optionalkeyword that is used to specify that inner
rclose represents a l-2 norm closure criteria instead of a infinity-norm
(absolute convergence criteria). When l2norm_rclose is specified, a
reasonable initial inner rclose value is 0.1 times the number of active
cells when meters and seconds are the defined MODFLOW 6 length and time.
relative_rclose: an optional keyword that is used to specify that
inner_rclose represents a relative L-2 Norm reduction closure criteria
instead of a infinity-Norm (absolute convergence criteria). When
relative_rclose is specified, a reasonable initial inner_rclose value is
1.0 * 10-4 and convergence is achieved for a given inner (linear)
iteration when ∆h ≤ inner_dvclose and the current L-2 Norm is ≤ the
product of the relativ_rclose and the initial L-2 Norm for the current
inner (linear) iteration. If rclose_option is not specified, an absolute
residual (infinity-norm) criterion is used.
Default value: None
SolutionPresetSimple: "strict"
SolutionPresetModerate: "strict"
SolutionPresetComplex: "strict"
relaxation_factor: float, optional
optional real value that defines the relaxation factor used by the
incomplete LU factorization preconditioners (MILU(0) and MILUT).
relaxation_factor is unitless and should be greater than or equal to 0.0
and less than or equal to 1.0. relaxation_factor Iterative Model
Solution 143 values of about 1.0 are commonly used, and experience
suggests that convergence can be optimized in some cases with relax
values of 0.97. A relaxation_factor value of 0.0 will result in either
ILU(0) or ILUT preconditioning (depending on the value specified for
preconditioner_levels and/or preconditioner_drop_tolerance). By default,
relaxation_factor is zero.
Default value: None
SolutionPresetSimple: 0.0
SolutionPresetModerate: 0
SolutionPresetComplex: 0.0
preconditioner_levels: int, optional
optional integer value defining the level of fill for ILU decomposition
used in the ILUT and MILUT preconditioners. Higher levels of fill
provide more robustness but also require more memory. For optimal
performance, it is suggested that a large level of fill be applied (7 or
8) with use of a drop tolerance. Specification of a
preconditioner_levels value greater than zero results in use of the ILUT
preconditioner. By default, preconditioner_levels is zero and the
zero-fill incomplete LU factorization preconditioners (ILU(0) and
MILU(0)) are used.
Default value: None
SolutionPresetSimple: 0
SolutionPresetModerate: 0
SolutionPresetComplex: 5
preconditioner_drop_tolerance: float, optional
optional real value that defines the drop tolerance used to drop
preconditioner terms based on the magnitude of matrix entries in the
ILUT and MILUT preconditioners. A value of 10−4 works well for most
problems. By default, preconditioner_drop_tolerance is zero and the
zero-fill incomplete LU factorization preconditioners (ILU(0) and
MILU(0)) are used.
Default value: None
SolutionPresetSimple: 0
SolutionPresetModerate: 0.0
SolutionPresetComplex: 0.0001
number_orthogonalizations: int, optional
optional integer value defining the interval used to explicitly
recalculate the residual of the flow equation using the solver
coefficient matrix, the latest head estimates, and the right hand side.
For problems that benefit from explicit recalculation of the residual, a
number between 4 and 10 is appropriate. By default,
number_orthogonalizations is zero.
Default value: None
SolutionPresetSimple: 0
SolutionPresetModerate: 0
SolutionPresetComplex: 2
scaling_method: str
options: {None, "diagonal", "l2norm"}
an optional keyword that defines the matrix scaling approach used. By
default, matrix scaling is not applied.
None - no matrix scaling applied.
diagonal - symmetric matrix scaling using the POLCG preconditioner
scaling method in Hill (1992).
l2norm - symmetric matrix scaling using the L2 norm.
Default value: None
reordering_method: str
options: {None, "rcm", "md"}
an optional keyword that defines the matrix reordering approach used. By
default, matrix reordering is not applied.
None - original ordering.
rcm - reverse Cuthill McKee ordering.
md - minimum degree ordering
Default value: None
print_option: str
options: {"none", "summary", "all"}
is a flag that controls printing of convergence information from the
solver.
None - means print nothing.
summary - means print only the total
number of iterations and nonlinear residual reduction summaries.
all - means print linear matrix solver convergence information to the
solution listing file and model specific linear matrix solver
convergence information to each model listing file in addition to
SUMMARY information.
Default value: "summary"
outer_csvfile: str, optional
None if no csv is to be written for the output, str of filename
if csv is to be written.
Default value: None
inner_csvfile: str, optional
None if no csv is to be written for the output, str of filename
if csv is to be written.
Default value: None
no_ptc: str, optional, either None, "all", or "first".
is a flag that is used to disable pseudo-transient continuation (PTC).
Option only applies to steady-state stress periods for models using the
Newton-Raphson formulation. For many problems, PTC can significantly
improve convergence behavior for steady-state simulations, and for this
reason it is active by default. In some cases, however, PTC can worsen
the convergence behavior, especially when the initial conditions are
similar to the solution. When the initial conditions are similar to, or
exactly the same as, the solution and convergence is slow, then this NO
PTC option should be used to deactivate PTC. This NO PTC option should
also be used in order to compare convergence behavior with other MODFLOW
versions, as PTC is only available in MODFLOW 6.
ats_outer_maximum_fraction: float, optional.
real value defining the fraction of the maximum allowable outer iterations
used with the Adaptive Time Step (ATS) capability if it is active. If this
value is set to zero by the user, then this solution will have no effect
on ATS behavior. This value must be greater than or equal to zero and less
than or equal to 0.5 or the program will terminate with an error. If it is
not specified by the user, then it is assigned a default value of one
third. When the number of outer iterations for this solution is less than
the product of this value and the maximum allowable outer iterations, then
ATS will increase the time step length by a factor of DTADJ in the ATS
input file. When the number of outer iterations for this solution is
greater than the maximum allowable outer iterations minus the product of
this value and the maximum allowable outer iterations, then the ATS (if
active) will decrease the time step length by a factor of 1 / DTADJ.
Default value is None.
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 = "ims"
_keyword_map = {}
_init_schemata = {
"outer_dvclose": [DTypeSchema(np.floating)],
"outer_maximum": [DTypeSchema(np.integer)],
"inner_maximum": [DTypeSchema(np.integer)],
"inner_dvclose": [DTypeSchema(np.floating)],
"inner_rclose": [DTypeSchema(np.floating)],
"linear_acceleration": [OptionSchema(("cg", "bicgstab"))],
"rclose_option": [OptionSchema(("strict", "l2norm_rclose", "relative_rclose"))],
"under_relaxation": [OptionSchema(("simple", "cooley", "dbd"))],
"under_relaxation_theta": [DTypeSchema(np.floating)],
"under_relaxation_kappa": [DTypeSchema(np.floating)],
"under_relaxation_gamma": [DTypeSchema(np.floating)],
"under_relaxation_momentum": [DTypeSchema(np.floating)],
"backtracking_number": [DTypeSchema(np.integer)],
"backtracking_tolerance": [DTypeSchema(np.floating)],
"backtracking_reduction_factor": [DTypeSchema(np.floating)],
"backtracking_residual_limit": [DTypeSchema(np.floating)],
"number_orthogonalizations": [DTypeSchema(np.integer)],
"scaling_method": [OptionSchema(("diagonal", "l2norm"))],
"reordering_method": [OptionSchema(("rcm", "md"))],
"print_option": [OptionSchema(("none", "summary", "all"))],
"no_ptc": [OptionSchema(("first", "all"))],
"ats_outer_maximum_fraction": [
DTypeSchema(np.floating),
AllValueSchema(">=", 0.0),
AllValueSchema("<=", 0.5),
],
}
_template = Package._initialize_template(_pkg_id)
[docs]
@init_log_decorator()
def __init__(
self,
modelnames,
outer_dvclose,
outer_maximum,
inner_maximum,
inner_dvclose,
inner_rclose,
linear_acceleration,
under_relaxation=None,
under_relaxation_theta=None,
under_relaxation_kappa=None,
under_relaxation_gamma=None,
under_relaxation_momentum=None,
backtracking_number=None,
backtracking_tolerance=None,
backtracking_reduction_factor=None,
backtracking_residual_limit=None,
rclose_option=None,
relaxation_factor=None,
preconditioner_levels=None,
preconditioner_drop_tolerance=None,
number_orthogonalizations=None,
scaling_method=None,
reordering_method=None,
print_option="summary",
outer_csvfile=None,
inner_csvfile=None,
no_ptc=None,
ats_outer_maximum_fraction=None,
validate: bool = True,
):
dict_dataset = {
"outer_dvclose": outer_dvclose,
"outer_maximum": outer_maximum,
"under_relaxation": under_relaxation,
"under_relaxation_theta": under_relaxation_theta,
"under_relaxation_kappa": under_relaxation_kappa,
"under_relaxation_gamma": under_relaxation_gamma,
"under_relaxation_momentum": under_relaxation_momentum,
"backtracking_number": backtracking_number,
"backtracking_tolerance": backtracking_tolerance,
"backtracking_reduction_factor": backtracking_reduction_factor,
"backtracking_residual_limit": backtracking_residual_limit,
"inner_maximum": inner_maximum,
"inner_dvclose": inner_dvclose,
"inner_rclose": inner_rclose,
"rclose_option": rclose_option,
"linear_acceleration": linear_acceleration,
"relaxation_factor": relaxation_factor,
"preconditioner_levels": preconditioner_levels,
"preconditioner_drop_tolerance": preconditioner_drop_tolerance,
"number_orthogonalizations": number_orthogonalizations,
"scaling_method": scaling_method,
"reordering_method": reordering_method,
"print_option": print_option,
"outer_csvfile": outer_csvfile,
"inner_csvfile": inner_csvfile,
"ats_outer_maximum_fraction": ats_outer_maximum_fraction,
"no_ptc": no_ptc,
}
# Make sure the modelnames are set as a variable rather than dimension:
if isinstance(modelnames, xr.DataArray):
dict_dataset["modelnames"] = modelnames
else:
dict_dataset["modelnames"] = ("model", modelnames)
super().__init__(dict_dataset)
self._validate_init_schemata(validate)
def remove_model_from_solution(self, modelname: str) -> None:
models_in_solution = self.get_models_in_solution()
if modelname not in models_in_solution:
raise ValueError(
f"attempted to remove model {modelname} from solution, but it was not found."
)
filtered_models = [m for m in models_in_solution if m != modelname]
if len(filtered_models) == 0:
self.dataset = self.dataset.drop_vars("modelnames")
else:
self.dataset.update({"modelnames": ("model", filtered_models)})
def add_model_to_solution(self, modelname: str) -> None:
models_in_solution = self.get_models_in_solution()
if modelname in models_in_solution:
raise ValueError(
f"attempted to add model {modelname} to solution, but it was already in it."
)
models_in_solution.append(modelname)
self.dataset.update({"modelnames": ("model", models_in_solution)})
def get_models_in_solution(self) -> list[str]:
models_in_solution = []
if "modelnames" in self.dataset.keys():
models_in_solution = list(self.dataset["modelnames"].values)
return models_in_solution
[docs]
def SolutionPresetSimple(
modelnames,
print_option="summary",
outer_csvfile=None,
inner_csvfile=None,
no_ptc=None,
):
solution = Solution(
modelnames=modelnames,
print_option=print_option,
outer_csvfile=outer_csvfile,
inner_csvfile=inner_csvfile,
no_ptc=no_ptc,
outer_dvclose=0.001,
outer_maximum=25,
under_relaxation=None,
under_relaxation_theta=0.0,
under_relaxation_kappa=0.0,
under_relaxation_gamma=0.0,
under_relaxation_momentum=0.0,
backtracking_number=0,
backtracking_tolerance=0.0,
backtracking_reduction_factor=0.0,
backtracking_residual_limit=0.0,
inner_maximum=50,
inner_dvclose=0.001,
inner_rclose=0.1,
rclose_option="strict",
linear_acceleration="cg",
relaxation_factor=0.0,
preconditioner_levels=0,
preconditioner_drop_tolerance=0,
number_orthogonalizations=0,
)
return solution
[docs]
def SolutionPresetModerate(
modelnames,
print_option="summary",
outer_csvfile=None,
inner_csvfile=None,
no_ptc=None,
):
solution = Solution(
modelnames=modelnames,
print_option=print_option,
outer_csvfile=outer_csvfile,
inner_csvfile=inner_csvfile,
no_ptc=no_ptc,
outer_dvclose=0.01,
outer_maximum=50,
under_relaxation="dbd",
under_relaxation_theta=0.9,
under_relaxation_kappa=0.0001,
under_relaxation_gamma=0.0,
under_relaxation_momentum=0.0,
backtracking_number=0,
backtracking_tolerance=0.0,
backtracking_reduction_factor=0.0,
backtracking_residual_limit=0.0,
inner_maximum=100,
inner_dvclose=0.01,
inner_rclose=0.1,
rclose_option="strict",
linear_acceleration="bicgstab",
relaxation_factor=0,
preconditioner_levels=0,
preconditioner_drop_tolerance=0.0,
number_orthogonalizations=0,
)
return solution
[docs]
def SolutionPresetComplex(
modelnames,
print_option="summary",
outer_csvfile=None,
inner_csvfile=None,
no_ptc=None,
):
solution = Solution(
modelnames=modelnames,
print_option=print_option,
outer_csvfile=outer_csvfile,
inner_csvfile=inner_csvfile,
no_ptc=no_ptc,
outer_dvclose=0.1,
outer_maximum=100,
under_relaxation="dbd",
under_relaxation_theta=0.8,
under_relaxation_kappa=0.0001,
under_relaxation_gamma=0.0,
under_relaxation_momentum=0.0,
backtracking_number=20,
backtracking_tolerance=1.05,
backtracking_reduction_factor=0.1,
backtracking_residual_limit=0.002,
inner_maximum=500,
inner_dvclose=0.1,
inner_rclose=0.1,
rclose_option="strict",
linear_acceleration="bicgstab",
relaxation_factor=0.0,
preconditioner_levels=5,
preconditioner_drop_tolerance=0.0001,
number_orthogonalizations=2,
)
return solution