Source code for flopy.modflow.mflpf

"""
mflpf module.  Contains the ModflowLpf class. Note that the user can access
the ModflowLpf class as `flopy.modflow.ModflowLpf`.

Additional information for this MODFLOW package can be found at the `Online
MODFLOW Guide
<https://water.usgs.gov/ogw/modflow/MODFLOW-2005-Guide/lpf.html>`_.

"""

import numpy as np

from ..pakbase import Package
from ..utils import Util2d, Util3d, read1d
from ..utils.flopy_io import line_parse
from .mfpar import ModflowPar as mfpar


[docs]class ModflowLpf(Package): """ MODFLOW Layer Property Flow Package Class. Parameters ---------- model : model object The model object (of type :class:`flopy.modflow.mf.Modflow`) to which this package will be added. ipakcb : int, optional Toggles whether cell-by-cell budget data should be saved. If None or zero, budget data will not be saved (default is None). hdry : float Is the head that is assigned to cells that are converted to dry during a simulation. Although this value plays no role in the model calculations, it is useful as an indicator when looking at the resulting heads that are output from the model. HDRY is thus similar to HNOFLO in the Basic Package, which is the value assigned to cells that are no-flow cells at the start of a model simulation. (default is -1.e30). laytyp : int or array of ints (nlay) Layer type, contains a flag for each layer that specifies the layer type. 0 confined >0 convertible <0 convertible unless the THICKSTRT option is in effect. (default is 0). layavg : int or array of ints (nlay) Layer average 0 is harmonic mean 1 is logarithmic mean 2 is arithmetic mean of saturated thickness and logarithmic mean of of hydraulic conductivity (default is 0). chani : float or array of floats (nlay) contains a value for each layer that is a flag or the horizontal anisotropy. If CHANI is less than or equal to 0, then variable HANI defines horizontal anisotropy. If CHANI is greater than 0, then CHANI is the horizontal anisotropy for the entire layer, and HANI is not read. If any HANI parameters are used, CHANI for all layers must be less than or equal to 0. Use as many records as needed to enter a value of CHANI for each layer. The horizontal anisotropy is the ratio of the hydraulic conductivity along columns (the Y direction) to the hydraulic conductivity along rows (the X direction). (default is 1). layvka : int or array of ints (nlay) a flag for each layer that indicates whether variable VKA is vertical hydraulic conductivity or the ratio of horizontal to vertical hydraulic conductivity. 0: VKA is vertical hydraulic conductivity not 0: VKA is the ratio of horizontal to vertical hydraulic conductivity (default is 0). laywet : int or array of ints (nlay) contains a flag for each layer that indicates if wetting is active. 0 wetting is inactive not 0 wetting is active (default is 0). wetfct : float is a factor that is included in the calculation of the head that is initially established at a cell when it is converted from dry to wet. (default is 0.1). iwetit : int is the iteration interval for attempting to wet cells. Wetting is attempted every IWETIT iteration. If using the PCG solver (Hill, 1990), this applies to outer iterations, not inner iterations. If IWETIT less than or equal to 0, it is changed to 1. (default is 1). ihdwet : int is a flag that determines which equation is used to define the initial head at cells that become wet. (default is 0) hk : float or array of floats (nlay, nrow, ncol) is the hydraulic conductivity along rows. HK is multiplied by horizontal anisotropy (see CHANI and HANI) to obtain hydraulic conductivity along columns. (default is 1.0). hani : float or array of floats (nlay, nrow, ncol) is the ratio of hydraulic conductivity along columns to hydraulic conductivity along rows, where HK of item 10 specifies the hydraulic conductivity along rows. Thus, the hydraulic conductivity along columns is the product of the values in HK and HANI. (default is 1.0). vka : float or array of floats (nlay, nrow, ncol) is either vertical hydraulic conductivity or the ratio of horizontal to vertical hydraulic conductivity depending on the value of LAYVKA. (default is 1.0). ss : float or array of floats (nlay, nrow, ncol) is specific storage unless the STORAGECOEFFICIENT option is used. When STORAGECOEFFICIENT is used, Ss is confined storage coefficient. (default is 1.e-5). sy : float or array of floats (nlay, nrow, ncol) is specific yield. (default is 0.15). vkcb : float or array of floats (nlay, nrow, ncol) is the vertical hydraulic conductivity of a Quasi-three-dimensional confining bed below a layer. (default is 0.0). Note that if an array is passed for vkcb it must be of size (nlay, nrow, ncol) even though the information for the bottom layer is not needed. wetdry : float or array of floats (nlay, nrow, ncol) is a combination of the wetting threshold and a flag to indicate which neighboring cells can cause a cell to become wet. (default is -0.01). storagecoefficient : boolean indicates that variable Ss and SS parameters are read as storage coefficient rather than specific storage. (default is False). constantcv : boolean indicates that vertical conductance for an unconfined cell is computed from the cell thickness rather than the saturated thickness. The CONSTANTCV option automatically invokes the NOCVCORRECTION option. (default is False). thickstrt : boolean indicates that layers having a negative LAYTYP are confined, and their cell thickness for conductance calculations will be computed as STRT-BOT rather than TOP-BOT. (default is False). nocvcorrection : boolean indicates that vertical conductance is not corrected when the vertical flow correction is applied. (default is False). novfc : boolean turns off the vertical flow correction under dewatered conditions. This option turns off the vertical flow calculation described on p. 5-8 of USGS Techniques and Methods Report 6-A16 and the vertical conductance correction described on p. 5-18 of that report. (default is False). extension : string Filename extension (default is 'lpf') unitnumber : int File unit number (default is None). filenames : str or list of str Filenames to use for the package and the output files. If filenames=None the package name will be created using the model name and package extension and the cbc output name will be created using the model name and .cbc extension (for example, modflowtest.cbc), if ipakcb is a number greater than zero. If a single string is passed the package will be set to the string and cbc output name will be created using the model name and .cbc extension, if ipakcb is a number greater than zero. To define the names for all package files (input and output) the length of the list of strings should be 2. Default is None. add_package : bool Flag to add the initialised package object to the parent model object. Default is True. Attributes ---------- Methods ------- See Also -------- Notes ----- Examples -------- >>> import flopy >>> m = flopy.modflow.Modflow() >>> lpf = flopy.modflow.ModflowLpf(m) """ "Layer-property flow package class\n" def __init__( self, model, laytyp=0, layavg=0, chani=1.0, layvka=0, laywet=0, ipakcb=None, hdry=-1e30, iwdflg=0, wetfct=0.1, iwetit=1, ihdwet=0, hk=1.0, hani=1.0, vka=1.0, ss=1e-5, sy=0.15, vkcb=0.0, wetdry=-0.01, storagecoefficient=False, constantcv=False, thickstrt=False, nocvcorrection=False, novfc=False, extension="lpf", unitnumber=None, filenames=None, add_package=True, ): # set default unit number of one is not specified if unitnumber is None: unitnumber = ModflowLpf._defaultunit() # set filenames filenames = self._prepare_filenames(filenames, 2) # cbc output file self.set_cbc_output_file(ipakcb, model, filenames[1]) # call base package constructor super().__init__( model, extension=extension, name=self._ftype(), unit_number=unitnumber, filenames=filenames[0], ) self._generate_heading() self.url = "lpf.html" nrow, ncol, nlay, nper = self.parent.nrow_ncol_nlay_nper # item 1 self.hdry = ( hdry # Head in cells that are converted to dry during a simulation ) self.nplpf = 0 # number of LPF parameters self.ikcflag = 0 # 1 and -1 are not supported. self.laytyp = Util2d(model, (nlay,), np.int32, laytyp, name="laytyp") self.layavg = Util2d(model, (nlay,), np.int32, layavg, name="layavg") self.chani = Util2d(model, (nlay,), np.float32, chani, name="chani") self.layvka = Util2d(model, (nlay,), np.int32, layvka, name="layvka") self.laywet = Util2d(model, (nlay,), np.int32, laywet, name="laywet") # Factor that is included in the calculation of the head when a cell is # converted from dry to wet self.wetfct = wetfct # Iteration interval for attempting to wet cells self.iwetit = iwetit # Flag that determines which equation is used to define the initial # head at cells that become wet self.ihdwet = ihdwet self.options = " " if storagecoefficient: self.options = self.options + "STORAGECOEFFICIENT " if constantcv: self.options = self.options + "CONSTANTCV " if thickstrt: self.options = self.options + "THICKSTRT " if nocvcorrection: self.options = self.options + "NOCVCORRECTION " if novfc: self.options = self.options + "NOVFC " self.hk = Util3d( model, (nlay, nrow, ncol), np.float32, hk, name="hk", locat=self.unit_number[0], ) self.hani = Util3d( model, (nlay, nrow, ncol), np.float32, hani, name="hani", locat=self.unit_number[0], ) keys = [] for k in range(nlay): key = "vka" if self.layvka[k] != 0: key = "vani" keys.append(key) self.vka = Util3d( model, (nlay, nrow, ncol), np.float32, vka, name=keys, locat=self.unit_number[0], ) tag = "ss" if storagecoefficient: tag = "storage" self.ss = Util3d( model, (nlay, nrow, ncol), np.float32, ss, name=tag, locat=self.unit_number[0], ) self.sy = Util3d( model, (nlay, nrow, ncol), np.float32, sy, name="sy", locat=self.unit_number[0], ) self.vkcb = Util3d( model, (nlay, nrow, ncol), np.float32, vkcb, name="vkcb", locat=self.unit_number[0], ) self.wetdry = Util3d( model, (nlay, nrow, ncol), np.float32, wetdry, name="wetdry", locat=self.unit_number[0], ) if add_package: self.parent.add_package(self) return
[docs] def write_file(self, check=True, f=None): """ Write the package file. Parameters ---------- check : boolean Check package data for common errors. (default True) Returns ------- None """ # allows turning off package checks when writing files at model level if check: self.check( f=f"{self.name[0]}.chk", verbose=self.parent.verbose, level=1, ) # get model information nrow, ncol, nlay, nper = self.parent.nrow_ncol_nlay_nper dis = self.parent.get_package("DIS") if dis is None: dis = self.parent.get_package("DISU") # Open file for writing if f is None: f = open(self.fn_path, "w") # Item 0: text f.write(f"{self.heading}\n") # Item 1: IBCFCB, HDRY, NPLPF, <IKCFLAG>, OPTIONS if self.parent.version == "mfusg" and not self.parent.structured: f.write( "{:10d}{:10.6G}{:10d}{:10d} {:s}\n".format( self.ipakcb, self.hdry, self.nplpf, self.ikcflag, self.options, ) ) else: f.write( f"{self.ipakcb:10d}{self.hdry:10.6G}{self.nplpf:10d} {self.options}\n" ) # LAYTYP array f.write(self.laytyp.string) # LAYAVG array f.write(self.layavg.string) # CHANI array f.write(self.chani.string) # LAYVKA array f.write(self.layvka.string) # LAYWET array f.write(self.laywet.string) # Item 7: WETFCT, IWETIT, IHDWET iwetdry = self.laywet.sum() if iwetdry > 0: f.write(f"{self.wetfct:10f}{self.iwetit:10d}{self.ihdwet:10d}\n") transient = not dis.steady.all() for k in range(nlay): f.write(self.hk[k].get_file_entry()) if self.chani[k] <= 0.0: f.write(self.hani[k].get_file_entry()) f.write(self.vka[k].get_file_entry()) if transient: f.write(self.ss[k].get_file_entry()) if self.laytyp[k] != 0: f.write(self.sy[k].get_file_entry()) if dis.laycbd[k] > 0: f.write(self.vkcb[k].get_file_entry()) if self.laywet[k] != 0 and self.laytyp[k] != 0: f.write(self.wetdry[k].get_file_entry()) f.close() return
[docs] @classmethod def load(cls, f, model, ext_unit_dict=None, check=True): """ Load an existing package. Parameters ---------- f : filename or file handle File to load. model : model object The model object (of type :class:`flopy.modflow.mf.Modflow`) to which this package will be added. ext_unit_dict : dictionary, optional If the arrays in the file are specified using EXTERNAL, or older style array control records, then `f` should be a file handle. In this case ext_unit_dict is required, which can be constructed using the function :class:`flopy.utils.mfreadnam.parsenamefile`. check : boolean Check package data for common errors. (default True) Returns ------- lpf : ModflowLpf object ModflowLpf object. Examples -------- >>> import flopy >>> m = flopy.modflow.Modflow() >>> lpf = flopy.modflow.ModflowLpf.load('test.lpf', m) """ if model.verbose: print("loading lpf package file...") openfile = not hasattr(f, "read") if openfile: filename = f f = open(filename, "r") # dataset 0 -- header while True: line = f.readline() if line[0] != "#": break # determine problem dimensions nr, nc, nlay, nper = model.get_nrow_ncol_nlay_nper() dis = model.get_package("DIS") if dis is None: dis = model.get_package("DISU") # Item 1: IBCFCB, HDRY, NPLPF - line already read above if model.verbose: print(" loading IBCFCB, HDRY, NPLPF...") t = line_parse(line) ipakcb, hdry, nplpf = int(t[0]), float(t[1]), int(t[2]) item1_len = 3 if model.version == "mfusg" and not model.structured: ikcflag = int(t[3]) item1_len = 4 # if ipakcb != 0: # model.add_pop_key_list(ipakcb) # ipakcb = 53 # options storagecoefficient = False constantcv = False thickstrt = False nocvcorrection = False novfc = False if len(t) > item1_len: for k in range(item1_len, len(t)): if "STORAGECOEFFICIENT" in t[k].upper(): storagecoefficient = True if "CONSTANTCV" in t[k].upper(): constantcv = True if "THICKSTRT" in t[k].upper(): thickstrt = True if "NOCVCORRECTION" in t[k].upper(): nocvcorrection = True if "NOVFC" in t[k].upper(): novfc = True # LAYTYP array if model.verbose: print(" loading LAYTYP...") laytyp = np.empty((nlay), dtype=np.int32) laytyp = read1d(f, laytyp) # LAYAVG array if model.verbose: print(" loading LAYAVG...") layavg = np.empty((nlay), dtype=np.int32) layavg = read1d(f, layavg) # CHANI array if model.verbose: print(" loading CHANI...") chani = np.empty((nlay), dtype=np.float32) chani = read1d(f, chani) # LAYVKA array if model.verbose: print(" loading LAYVKA...") layvka = np.empty((nlay,), dtype=np.int32) layvka = read1d(f, layvka) # LAYWET array if model.verbose: print(" loading LAYWET...") laywet = np.empty((nlay), dtype=np.int32) laywet = read1d(f, laywet) # Item 7: WETFCT, IWETIT, IHDWET wetfct, iwetit, ihdwet = None, None, None iwetdry = laywet.sum() if iwetdry > 0: if model.verbose: print(" loading WETFCT, IWETIT, IHDWET...") line = f.readline() t = line.strip().split() wetfct, iwetit, ihdwet = float(t[0]), int(t[1]), int(t[2]) # parameters data par_types = [] if nplpf > 0: par_types, parm_dict = mfpar.load(f, nplpf, model.verbose) # print parm_dict # non-parameter data transient = not dis.steady.all() hk = [0] * nlay hani = [0] * nlay vka = [0] * nlay ss = [0] * nlay sy = [0] * nlay vkcb = [0] * nlay wetdry = [0] * nlay # load by layer for k in range(nlay): # allow for unstructured changing nodes per layer if nr is None: nrow = 1 ncol = nc[k] else: nrow = nr ncol = nc # hk if model.verbose: print(f" loading hk layer {k + 1:3d}...") if "hk" not in par_types: t = Util2d.load( f, model, (nrow, ncol), np.float32, "hk", ext_unit_dict ) else: line = f.readline() t = mfpar.parameter_fill( model, (nrow, ncol), "hk", parm_dict, findlayer=k ) hk[k] = t # hani if chani[k] <= 0.0: if model.verbose: print(f" loading hani layer {k + 1:3d}...") if "hani" not in par_types: t = Util2d.load( f, model, (nrow, ncol), np.float32, "hani", ext_unit_dict, ) else: line = f.readline() t = mfpar.parameter_fill( model, (nrow, ncol), "hani", parm_dict, findlayer=k ) hani[k] = t # vka if model.verbose: print(f" loading vka layer {k + 1:3d}...") key = "vk" if layvka[k] != 0: key = "vani" if "vk" not in par_types and "vani" not in par_types: t = Util2d.load( f, model, (nrow, ncol), np.float32, key, ext_unit_dict ) else: line = f.readline() key = "vk" if "vani" in par_types: key = "vani" t = mfpar.parameter_fill( model, (nrow, ncol), key, parm_dict, findlayer=k ) vka[k] = t # storage properties if transient: # ss if model.verbose: print(f" loading ss layer {k + 1:3d}...") if "ss" not in par_types: t = Util2d.load( f, model, (nrow, ncol), np.float32, "ss", ext_unit_dict ) else: line = f.readline() t = mfpar.parameter_fill( model, (nrow, ncol), "ss", parm_dict, findlayer=k ) ss[k] = t # sy if laytyp[k] != 0: if model.verbose: print(f" loading sy layer {k + 1:3d}...") if "sy" not in par_types: t = Util2d.load( f, model, (nrow, ncol), np.float32, "sy", ext_unit_dict, ) else: line = f.readline() t = mfpar.parameter_fill( model, (nrow, ncol), "sy", parm_dict, findlayer=k ) sy[k] = t # vkcb if dis.laycbd[k] > 0: if model.verbose: print(f" loading vkcb layer {k + 1:3d}...") if "vkcb" not in par_types: t = Util2d.load( f, model, (nrow, ncol), np.float32, "vkcb", ext_unit_dict, ) else: line = f.readline() t = mfpar.parameter_fill( model, (nrow, ncol), "vkcb", parm_dict, findlayer=k ) vkcb[k] = t # wetdry if laywet[k] != 0 and laytyp[k] != 0: if model.verbose: print(f" loading wetdry layer {k + 1:3d}...") t = Util2d.load( f, model, (nrow, ncol), np.float32, "wetdry", ext_unit_dict ) wetdry[k] = t if openfile: f.close() # set package unit number unitnumber = None filenames = [None, None] if ext_unit_dict is not None: unitnumber, filenames[0] = model.get_ext_dict_attr( ext_unit_dict, filetype=ModflowLpf._ftype() ) if ipakcb > 0: iu, filenames[1] = model.get_ext_dict_attr( ext_unit_dict, unit=ipakcb ) model.add_pop_key_list(ipakcb) # create instance of lpf class lpf = cls( model, ipakcb=ipakcb, laytyp=laytyp, layavg=layavg, chani=chani, layvka=layvka, laywet=laywet, hdry=hdry, iwdflg=iwetdry, wetfct=wetfct, iwetit=iwetit, ihdwet=ihdwet, hk=hk, hani=hani, vka=vka, ss=ss, sy=sy, vkcb=vkcb, wetdry=wetdry, storagecoefficient=storagecoefficient, constantcv=constantcv, thickstrt=thickstrt, novfc=novfc, nocvcorrection=nocvcorrection, unitnumber=unitnumber, filenames=filenames, ) if check: lpf.check( f=f"{lpf.name[0]}.chk", verbose=lpf.parent.verbose, level=0, ) return lpf
@staticmethod def _ftype(): return "LPF" @staticmethod def _defaultunit(): return 15