flopy.utils.gridgen module

class Gridgen(dis, model_ws='.', exe_name='gridgen', surface_interpolation='replicate', vertical_pass_through=False)

Bases: object

Class to work with the gridgen program to create layered quadtree grids.

Parameters:

• dis (flopy.modflow.ModflowDis) – Flopy discretization object
• model_ws (str) – workspace location for creating gridgen files (default is ‘.’)
• exe_name (str) – path and name of the gridgen program. (default is gridgen)
• surface_interpolation (str) – Default gridgen method for interpolating elevations. Valid options include ‘replicate’ (default) and ‘interpolate’
• vertical_pass_through (bool) – If true, Gridgen’s GRID_TO_USGDATA command will connect layers where intermediate layers are inactive. (default is False)

Notes

For the surface elevations, the top of a layer uses the same surface as the bottom of the overlying layer.

add_active_domain(feature, layers)

Parameters:

• feature (str or list) –

  feature can be:

  a string containing the name of a polygon a list of polygons flopy.utils.geometry.Collection object of Polygons shapely.geometry.Collection object of Polygons geojson.GeometryCollection object of Polygons list of shapefile.Shape objects shapefile.Shapes object

• layers (list) – A list of layers (zero based) for which this active domain applies.

Returns:

Return type:

None

add_refinement_features(features, featuretype, level, layers)

Parameters:

• features (str, list, or collection object) –

  features can be

  a string containing the name of a shapefile a list of points, lines, or polygons flopy.utils.geometry.Collection object a list of flopy.utils.geometry objects shapely.geometry.Collection object geojson.GeometryCollection object a list of shapefile.Shape objects shapefile.Shapes object

• featuretype (str) – Must be either ‘point’, ‘line’, or ‘polygon’
• level (int) – The level of refinement for this features
• layers (list) – A list of layers (zero based) for which this refinement features applies.

Returns:

Return type:

None

build(verbose=False)

Build the quadtree grid

Parameters:verbose (bool) – If true, print the results of the gridgen command to the terminal (default is False) Returns: Return type:None

export(verbose=False)

Export the quadtree grid to shapefiles, usgdata, and vtk

Parameters:verbose (bool) – If true, print the results of the gridgen command to the terminal (default is False) Returns: Return type:None

get_angldegx(fldr=None)

Get the angldegx array

Parameters:fldr (ndarray) – Flow direction indicator array. If None, then it is read from gridgen output. Returns:angldegx – A 1D vector indicating the angle (in degrees) between the x axis and an outward normal to the face. Return type:ndarray

get_area()

Get the area array

Returns:area – A 1D vector of cell areas of size nodes Return type:ndarray

get_bot()

Get the bot array

Returns:bot – A 1D vector of cell bottom elevations of size nodes Return type:ndarray

get_cellxy(ncells)

Parameters:ncells (int) – Number of cells for which to create the list of cell centers Returns:cellxy – x and y cell centers. Shape is (ncells, 2) Return type:ndarray

get_center(nodenumber)

Return the cell center x and y coordinates

Parameters:nodenumber – Returns:(x, y) Return type:tuple

get_cl12()

Get the cl12 array

Returns:cl12 – A 1D vector of the cell connection distances, which are from the center of cell n to its shared face will cell m Return type:ndarray

get_disu(model, nper=1, perlen=1, nstp=1, tsmult=1, steady=True, itmuni=4, lenuni=2)

Create a MODFLOW-USG DISU flopy object.

Parameters:

• model (Flopy model object) – The Flopy model object (of type flopy.modflow.mf.Modflow) to which this package will be added.
• nper (int) – Number of model stress periods (default is 1).
• perlen (float or array of floats (nper)) – A single value or array of the stress period lengths (default is 1).
• nstp (int or array of ints (nper)) – Number of time steps in each stress period (default is 1).
• tsmult (float or array of floats (nper)) – Time step multiplier (default is 1.0).
• steady (boolean or array of boolean (nper)) – True or False indicating whether or not stress period is steady state (default is True).
• itmuni (int) – Time units, default is days (4)
• lenuni (int) – Length units, default is meters (2)

Returns:

disu

Return type:

Flopy ModflowDisU object.

get_fahl()

Get the fahl array

Returns:fahl – A 1D vector of the cell connection information, which is flow area for a vertical connection and horizontal length for a horizontal connection Return type:ndarray

get_fldr()

Get the fldr array

Returns:fldr – A 1D vector indicating the direction of the connection 1, 2, and 3 are plus x, y, and z directions. -1, -2, and -3 are negative x, y, and z directions. Return type:ndarray

get_gridprops()

Get a dictionary of information needed to create a MODFLOW-USG DISU Package

Returns:gridprops Return type:dict

get_gridprops_disu6()

Return a dictionary containing all of the information required to create a MODFLOW 6 DISU Package

Returns:gridprops Return type:dict

get_gridprops_disv(verbose=False)

get_hwva(ja=None, ihc=None, fahl=None, top=None, bot=None)

Get the hwva array

Parameters:

• ja (ndarray) – Cell connectivity. If None, it will be read from gridgen output.
• ihc (ndarray) – Connection horizontal indicator array. If None it will be read and calculated from gridgen output.
• fahl (ndarray) – Flow area, horizontal width array required by MODFLOW-USG. If none then it will be read from the gridgen output. Default is None.
• top (ndarray) – Cell top elevation. If None, it will be read from gridgen output.
• bot (ndarray) – Cell bottom elevation. If None, it will be read from gridgen output.

Returns:

fahl – A 1D vector of the cell connection information, which is flow area for a vertical connection and horizontal length for a horizontal connection

Return type:

ndarray

get_iac()

Get the iac array

Returns:iac – A 1D vector of the number of connections (plus 1) for each cell Return type:ndarray

get_ihc(fldr=None)

Get the ihc array

Parameters:fldr (ndarray) – Flow direction indicator array. If None, then it is read from gridgen output. Returns:ihc – A 1D vector indicating the direction of the connection where 0 is vertical, 1 is a regular horizontal connection and 2 is a vertically staggered horizontal connection. Return type:ndarray

get_ivc(fldr=None)

Get the MODFLOW-USG ivc array

Parameters:fldr (ndarray) – Flow direction indicator array. If None, then it is read from gridgen output. Returns:ivc – A 1D vector indicating the direction of the connection where 1 is vertical and 0 is horizontal. Return type:ndarray

get_ja(nja=None)

Get the ja array

Parameters:nja (int) – Number of connections. If None, then it is read from gridgen output. Returns:ja – A 1D vector of the cell connectivity (one-based) Return type:ndarray

get_nlay()

Get the number of layers

Returns:nlay Return type:int

get_nod_recarray()

Load the qtg.nod file and return as a numpy recarray

Returns:node_ra – Recarray representation of the node file with zero-based indexing Return type:ndarray

get_nodelay()

Return the nodelay array, which is an array of size nlay containing the number of nodes in each layer.

Returns:nodelay – Number of nodes in each layer Return type:ndarray

get_nodes()

Get the number of nodes

Returns:nodes Return type:int

get_top()

Get the top array

Returns:top – A 1D vector of cell top elevations of size nodes Return type:ndarray

get_vertices(nodenumber)

Return a list of 5 vertices for the cell. The first vertex should be the same as the last vertex.

Parameters:nodenumber – Returns:list of vertices Return type:list

get_verts_iverts(ncells, verbose=False)

Return a 2d array of x and y vertices and a list of size ncells that has the list of vertices for each cell.

Parameters:

• ncells (int) – The number of entries in iverts. This should be ncpl for a layered model and nodes for a disu model.
• verbose (bool) – Print information as its working

Returns:

verts, iverts – verts is a 2d array of x and y vertex pairs (nvert, 2) and iverts is a list of vertices that comprise each cell

Return type:

tuple

intersect(features, featuretype, layer)

Parameters:

• features (str or list) – features can be either a string containing the name of a shapefile or it can be a list of points, lines, or polygons
• featuretype (str) – Must be either ‘point’, ‘line’, or ‘polygon’
• layer (int) – Layer (zero based) to intersect with. Zero based.

Returns:

result – Recarray of the intersection properties.

Return type:

np.recarray

plot(ax=None, layer=0, edgecolor='k', facecolor='none', cmap='Dark2', a=None, masked_values=None, **kwargs)

Plot the grid. This method will plot the grid using the shapefile that was created as part of the build method.

Note that the layer option is not working yet.

Parameters:

• ax (matplotlib.pyplot axis) – The plot axis. If not provided it, plt.gca() will be used. If there is not a current axis then a new one will be created.
• layer (int) – Layer number to plot
• cmap (string) – Name of colormap to use for polygon shading (default is ‘Dark2’)
• edgecolor (string) – Color name. (Default is ‘scaled’ to scale the edge colors.)
• facecolor (string) – Color name. (Default is ‘scaled’ to scale the face colors.)
• a (numpy.ndarray) – Array to plot.
• masked_values (iterable of floats, ints) – Values to mask.
• kwargs (dictionary) – Keyword arguments that are passed to PatchCollection.set(**kwargs). Some common kwargs would be ‘linewidths’, ‘linestyles’, ‘alpha’, etc.

Returns:

pc

Return type:

matplotlib.collections.PatchCollection

set_surface_interpolation(isurf, type, elev=None, elev_extent=None)

Parameters:

• isurf (int) – surface number where 0 is top and nlay + 1 is bottom
• type (str) – Must be ‘INTERPOLATE’, ‘REPLICATE’ or ‘ASCIIGRID’.
• elev (numpy.ndarray of shape (nr, nc) or str) – Array that is used as an asciigrid. If elev is a string, then it is assumed to be the name of the asciigrid.
• elev_extent (list-like) – List of xmin, xmax, ymin, ymax extents of the elev grid. Must be specified for ASCIIGRID; optional otherwise.

Returns:

Return type:

None

to_disu6(fname, writevertices=True)

Create a MODFLOW 6 DISU file

Parameters:

• fname (str) – name of file to write
• writevertices (bool) – include vertices in the DISU file. (default is True)

to_disv6(fname, verbose=False)

Create a MODFLOW 6 DISV file

Parameters:fname (str) – name of file to write

features_to_shapefile(features, featuretype, filename)

Write a shapefile for the features of type featuretype.

Parameters:

• features (list) –

  point, line, or polygon features. Method accepts feature can be:

  a list of geometries flopy.utils.geometry.Collection object shapely.geometry.Collection object geojson.GeometryCollection object list of shapefile.Shape objects shapefile.Shapes object

• featuretype (str) – Must be ‘point’, ‘line’, ‘linestring’, or ‘polygon’
• filename (string) – name of the shapefile to write

Returns:

Return type:

None

ndarray_to_asciigrid(fname, a, extent, nodata=1e+30)

read1d(f, a)

Quick file to array reader for reading gridgen output. Much faster than the read1d function in util_array