Gael-de-Sailly
/
flopy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor(flopy): run black formatting with latest (20.8b1) (#975)

develop
jdhughes-usgs 2020-08-27 10:17:23 -04:00 committed by GitHub
parent cab9f63f52
commit cb6b98cf85
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
78 changed files with 2556 additions and 1668 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
.docs/conf.py
View File

@ -79,9 +79,9 @@ if stderr:
# -- Project information ----------------------------------------------------- # -- Project information -----------------------------------------------------
project = 'flopy' project = "flopy"
copyright = '2020, Bakker, Mark, Post, Vincent, Langevin, C. D., Hughes, J. D., White, J. T., Leaf, A. T., Paulinski, S. R., Larsen, J. D., Toews, M. W., Morway, E. D., Bellino, J. C., Starn, J. J., and Fienen, M. N.' copyright = "2020, Bakker, Mark, Post, Vincent, Langevin, C. D., Hughes, J. D., White, J. T., Leaf, A. T., Paulinski, S. R., Larsen, J. D., Toews, M. W., Morway, E. D., Bellino, J. C., Starn, J. J., and Fienen, M. N."
author = 'Bakker, Mark, Post, Vincent, Langevin, C. D., Hughes, J. D., White, J. T., Leaf, A. T., Paulinski, S. R., Larsen, J. D., Toews, M. W., Morway, E. D., Bellino, J. C., Starn, J. J., and Fienen, M. N.' author = "Bakker, Mark, Post, Vincent, Langevin, C. D., Hughes, J. D., White, J. T., Leaf, A. T., Paulinski, S. R., Larsen, J. D., Toews, M. W., Morway, E. D., Bellino, J. C., Starn, J. J., and Fienen, M. N."
# The version. # The version.
version = __version__ version = __version__
@ -112,7 +112,6 @@ extensions = [
] ]
# Add any paths that contain templates here, relative to this directory. # Add any paths that contain templates here, relative to this directory.
templates_path = ["_templates"] templates_path = ["_templates"]

64
.docs/pysrc/tutorial1.py
View File

@ -3,80 +3,82 @@ import sys
import numpy as np import numpy as np
import flopy import flopy
#Assign name and create modflow model object # Assign name and create modflow model object
modelname = 'tutorial1' modelname = "tutorial1"
exe_name = os.path.join("..", ".bin", "mf2005") exe_name = os.path.join("..", ".bin", "mf2005")
mf = flopy.modflow.Modflow(modelname, exe_name=exe_name) mf = flopy.modflow.Modflow(modelname, exe_name=exe_name)
#model domain and grid definition # model domain and grid definition
Lx = 1000. Lx = 1000.0
Ly = 1000. Ly = 1000.0
ztop = 0. ztop = 0.0
zbot = -50. zbot = -50.0
nlay = 1 nlay = 1
nrow = 10 nrow = 10
ncol = 10 ncol = 10
delr = Lx/ncol delr = Lx / ncol
delc = Ly/nrow delc = Ly / nrow
delv = (ztop - zbot) / nlay delv = (ztop - zbot) / nlay
botm = np.linspace(ztop, zbot, nlay + 1) botm = np.linspace(ztop, zbot, nlay + 1)
# Create the discretization object # Create the discretization object
dis = flopy.modflow.ModflowDis(mf, nlay, nrow, ncol, delr=delr, delc=delc, dis = flopy.modflow.ModflowDis(
top=ztop, botm=botm[1:]) mf, nlay, nrow, ncol, delr=delr, delc=delc, top=ztop, botm=botm[1:]
)
# Variables for the BAS package # Variables for the BAS package
ibound = np.ones((nlay, nrow, ncol), dtype=np.int32) ibound = np.ones((nlay, nrow, ncol), dtype=np.int32)
ibound[:, :, 0] = -1 ibound[:, :, 0] = -1
ibound[:, :, -1] = -1 ibound[:, :, -1] = -1
strt = np.ones((nlay, nrow, ncol), dtype=np.float32) strt = np.ones((nlay, nrow, ncol), dtype=np.float32)
strt[:, :, 0] = 10. strt[:, :, 0] = 10.0
strt[:, :, -1] = 0. strt[:, :, -1] = 0.0
bas = flopy.modflow.ModflowBas(mf, ibound=ibound, strt=strt) bas = flopy.modflow.ModflowBas(mf, ibound=ibound, strt=strt)
#Add LPF package to the MODFLOW model # Add LPF package to the MODFLOW model
lpf = flopy.modflow.ModflowLpf(mf, hk=10., vka=10., ipakcb=53) lpf = flopy.modflow.ModflowLpf(mf, hk=10.0, vka=10.0, ipakcb=53)
# Add OC package to the MODFLOW model # Add OC package to the MODFLOW model
spd = {(0, 0): ['print head', 'print budget', 'save head', 'save budget']} spd = {(0, 0): ["print head", "print budget", "save head", "save budget"]}
oc = flopy.modflow.ModflowOc(mf, stress_period_data=spd, compact=True) oc = flopy.modflow.ModflowOc(mf, stress_period_data=spd, compact=True)
#Add PCG package to the MODFLOW model # Add PCG package to the MODFLOW model
pcg = flopy.modflow.ModflowPcg(mf) pcg = flopy.modflow.ModflowPcg(mf)
#Write the MODFLOW model input files # Write the MODFLOW model input files
mf.write_input() mf.write_input()
#Run the MODFLOW model # Run the MODFLOW model
mf.run_model() mf.run_model()
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import flopy.utils.binaryfile as bf import flopy.utils.binaryfile as bf
plt.subplot(1,1,1,aspect='equal')
hds = bf.HeadFile(modelname+'.hds') plt.subplot(1, 1, 1, aspect="equal")
hds = bf.HeadFile(modelname + ".hds")
head = hds.get_data(totim=1.0) head = hds.get_data(totim=1.0)
levels = np.arange(1,10,1) levels = np.arange(1, 10, 1)
extent = (delr/2., Lx - delc/2., Ly - delc/2., delc/2.) extent = (delr / 2.0, Lx - delc / 2.0, Ly - delc / 2.0, delc / 2.0)
plt.contour(head[0, :, :], levels=levels, extent=extent) plt.contour(head[0, :, :], levels=levels, extent=extent)
plt.savefig('tutorial1a.png') plt.savefig("tutorial1a.png")
fig = plt.figure(figsize=(10,10)) fig = plt.figure(figsize=(10, 10))
ax = fig.add_subplot(1, 1, 1, aspect='equal') ax = fig.add_subplot(1, 1, 1, aspect="equal")
hds = bf.HeadFile(modelname + '.hds') hds = bf.HeadFile(modelname + ".hds")
times = hds.get_times() times = hds.get_times()
head = hds.get_data(totim=times[-1]) head = hds.get_data(totim=times[-1])
levels = np.linspace(0, 10, 11) levels = np.linspace(0, 10, 11)
cbb = bf.CellBudgetFile(modelname + '.cbc') cbb = bf.CellBudgetFile(modelname + ".cbc")
kstpkper_list = cbb.get_kstpkper() kstpkper_list = cbb.get_kstpkper()
frf = cbb.get_data(text='FLOW RIGHT FACE', totim=times[-1])[0] frf = cbb.get_data(text="FLOW RIGHT FACE", totim=times[-1])[0]
fff = cbb.get_data(text='FLOW FRONT FACE', totim=times[-1])[0] fff = cbb.get_data(text="FLOW FRONT FACE", totim=times[-1])[0]
pmv = flopy.plot.PlotMapView(model=mf, layer=0) pmv = flopy.plot.PlotMapView(model=mf, layer=0)
qm = pmv.plot_ibound() qm = pmv.plot_ibound()
lc = pmv.plot_grid() lc = pmv.plot_grid()
cs = pmv.contour_array(head, levels=levels) cs = pmv.contour_array(head, levels=levels)
quiver = pmv.plot_discharge(frf, fff, head=head) quiver = pmv.plot_discharge(frf, fff, head=head)
plt.savefig('tutorial1b.png') plt.savefig("tutorial1b.png")

170
.docs/pysrc/tutorial2.py
View File

@ -3,10 +3,10 @@ import numpy as np
import flopy import flopy
# Model domain and grid definition # Model domain and grid definition
Lx = 1000. Lx = 1000.0
Ly = 1000. Ly = 1000.0
ztop = 10. ztop = 10.0
zbot = -50. zbot = -50.0
nlay = 1 nlay = 1
nrow = 10 nrow = 10
ncol = 10 ncol = 10
@ -14,16 +14,16 @@ delr = Lx / ncol
delc = Ly / nrow delc = Ly / nrow
delv = (ztop - zbot) / nlay delv = (ztop - zbot) / nlay
botm = np.linspace(ztop, zbot, nlay + 1) botm = np.linspace(ztop, zbot, nlay + 1)
hk = 1. hk = 1.0
vka = 1. vka = 1.0
sy = 0.1 sy = 0.1
ss = 1.e-4 ss = 1.0e-4
laytyp = 1 laytyp = 1
# Variables for the BAS package # Variables for the BAS package
# Note that changes from the previous tutorial! # Note that changes from the previous tutorial!
ibound = np.ones((nlay, nrow, ncol), dtype=np.int32) ibound = np.ones((nlay, nrow, ncol), dtype=np.int32)
strt = 10. * np.ones((nlay, nrow, ncol), dtype=np.float32) strt = 10.0 * np.ones((nlay, nrow, ncol), dtype=np.float32)
# Time step parameters # Time step parameters
nper = 3 nper = 3
@ -32,21 +32,32 @@ nstp = [1, 100, 100]
steady = [True, False, False] steady = [True, False, False]
# Flopy objects # Flopy objects
modelname = 'tutorial2' modelname = "tutorial2"
exe_name = os.path.join("..", ".bin", "mf2005") exe_name = os.path.join("..", ".bin", "mf2005")
mf = flopy.modflow.Modflow(modelname, exe_name=exe_name) mf = flopy.modflow.Modflow(modelname, exe_name=exe_name)
dis = flopy.modflow.ModflowDis(mf, nlay, nrow, ncol, delr=delr, delc=delc, dis = flopy.modflow.ModflowDis(
top=ztop, botm=botm[1:], mf,
nper=nper, perlen=perlen, nstp=nstp, nlay,
steady=steady) nrow,
ncol,
delr=delr,
delc=delc,
top=ztop,
botm=botm[1:],
nper=nper,
perlen=perlen,
nstp=nstp,
steady=steady,
)
bas = flopy.modflow.ModflowBas(mf, ibound=ibound, strt=strt) bas = flopy.modflow.ModflowBas(mf, ibound=ibound, strt=strt)
lpf = flopy.modflow.ModflowLpf(mf, hk=hk, vka=vka, sy=sy, ss=ss, laytyp=laytyp, lpf = flopy.modflow.ModflowLpf(
ipakcb=53) mf, hk=hk, vka=vka, sy=sy, ss=ss, laytyp=laytyp, ipakcb=53
)
pcg = flopy.modflow.ModflowPcg(mf) pcg = flopy.modflow.ModflowPcg(mf)
# Make list for stress period 1 # Make list for stress period 1
stageleft = 10. stageleft = 10.0
stageright = 10. stageright = 10.0
bound_sp1 = [] bound_sp1 = []
for il in range(nlay): for il in range(nlay):
condleft = hk * (stageleft - zbot) * delc condleft = hk * (stageleft - zbot) * delc
@ -54,11 +65,11 @@ for il in range(nlay):
for ir in range(nrow): for ir in range(nrow):
bound_sp1.append([il, ir, 0, stageleft, condleft]) bound_sp1.append([il, ir, 0, stageleft, condleft])
bound_sp1.append([il, ir, ncol - 1, stageright, condright]) bound_sp1.append([il, ir, ncol - 1, stageright, condright])
print('Adding ', len(bound_sp1), 'GHBs for stress period 1.') print("Adding ", len(bound_sp1), "GHBs for stress period 1.")
# Make list for stress period 2 # Make list for stress period 2
stageleft = 10. stageleft = 10.0
stageright = 0. stageright = 0.0
condleft = hk * (stageleft - zbot) * delc condleft = hk * (stageleft - zbot) * delc
condright = hk * (stageright - zbot) * delc condright = hk * (stageright - zbot) * delc
bound_sp2 = [] bound_sp2 = []
@ -66,7 +77,7 @@ for il in range(nlay):
for ir in range(nrow): for ir in range(nrow):
bound_sp2.append([il, ir, 0, stageleft, condleft]) bound_sp2.append([il, ir, 0, stageleft, condleft])
bound_sp2.append([il, ir, ncol - 1, stageright, condright]) bound_sp2.append([il, ir, ncol - 1, stageright, condright])
print('Adding ', len(bound_sp2), 'GHBs for stress period 2.') print("Adding ", len(bound_sp2), "GHBs for stress period 2.")
# We do not need to add a dictionary entry for stress period 3. # We do not need to add a dictionary entry for stress period 3.
# Flopy will automatically take the list from stress period 2 and apply it # Flopy will automatically take the list from stress period 2 and apply it
@ -78,9 +89,9 @@ ghb = flopy.modflow.ModflowGhb(mf, stress_period_data=stress_period_data)
# Create the well package # Create the well package
# Remember to use zero-based layer, row, column indices! # Remember to use zero-based layer, row, column indices!
pumping_rate = -100. pumping_rate = -100.0
wel_sp1 = [[0, nrow / 2 - 1, ncol / 2 - 1, 0.]] wel_sp1 = [[0, nrow / 2 - 1, ncol / 2 - 1, 0.0]]
wel_sp2 = [[0, nrow / 2 - 1, ncol / 2 - 1, 0.]] wel_sp2 = [[0, nrow / 2 - 1, ncol / 2 - 1, 0.0]]
wel_sp3 = [[0, nrow / 2 - 1, ncol / 2 - 1, pumping_rate]] wel_sp3 = [[0, nrow / 2 - 1, ncol / 2 - 1, pumping_rate]]
stress_period_data = {0: wel_sp1, 1: wel_sp2, 2: wel_sp3} stress_period_data = {0: wel_sp1, 1: wel_sp2, 2: wel_sp3}
wel = flopy.modflow.ModflowWel(mf, stress_period_data=stress_period_data) wel = flopy.modflow.ModflowWel(mf, stress_period_data=stress_period_data)
@ -88,13 +99,16 @@ wel = flopy.modflow.ModflowWel(mf, stress_period_data=stress_period_data)
stress_period_data = {} stress_period_data = {}
for kper in range(nper): for kper in range(nper):
for kstp in range(nstp[kper]): for kstp in range(nstp[kper]):
stress_period_data[(kper, kstp)] = ['save head', stress_period_data[(kper, kstp)] = [
'save drawdown', "save head",
'save budget', "save drawdown",
'print head', "save budget",
'print budget'] "print head",
oc = flopy.modflow.ModflowOc(mf, stress_period_data=stress_period_data, "print budget",
compact=True) ]
oc = flopy.modflow.ModflowOc(
mf, stress_period_data=stress_period_data, compact=True
)
# Write the model input files # Write the model input files
mf.write_input() mf.write_input()
@ -102,7 +116,7 @@ mf.write_input()
# Run the model # Run the model
success, mfoutput = mf.run_model(silent=True, pause=False, report=True) success, mfoutput = mf.run_model(silent=True, pause=False, report=True)
if not success: if not success:
raise Exception('MODFLOW did not terminate normally.') raise Exception("MODFLOW did not terminate normally.")
# Imports # Imports
@ -110,66 +124,72 @@ import matplotlib.pyplot as plt
import flopy.utils.binaryfile as bf import flopy.utils.binaryfile as bf
# Create the headfile and budget file objects # Create the headfile and budget file objects
headobj = bf.HeadFile(modelname+'.hds') headobj = bf.HeadFile(modelname + ".hds")
times = headobj.get_times() times = headobj.get_times()
cbb = bf.CellBudgetFile(modelname+'.cbc') cbb = bf.CellBudgetFile(modelname + ".cbc")
# Setup contour parameters # Setup contour parameters
levels = np.linspace(0, 10, 11) levels = np.linspace(0, 10, 11)
extent = (delr/2., Lx - delr/2., delc/2., Ly - delc/2.) extent = (delr / 2.0, Lx - delr / 2.0, delc / 2.0, Ly - delc / 2.0)
print('Levels: ', levels) print("Levels: ", levels)
print('Extent: ', extent) print("Extent: ", extent)
# Well point # Well point
wpt = ((float(ncol/2)-0.5)*delr, (float(nrow/2-1)+0.5)*delc) wpt = ((float(ncol / 2) - 0.5) * delr, (float(nrow / 2 - 1) + 0.5) * delc)
wpt = (450., 550.) wpt = (450.0, 550.0)
# Make the plots # Make the plots
mytimes = [1.0, 101.0, 201.0] mytimes = [1.0, 101.0, 201.0]
for iplot, time in enumerate(mytimes): for iplot, time in enumerate(mytimes):
print('*****Processing time: ', time) print("*****Processing time: ", time)
head = headobj.get_data(totim=time) head = headobj.get_data(totim=time)
#Print statistics # Print statistics
print('Head statistics') print("Head statistics")
print(' min: ', head.min()) print(" min: ", head.min())
print(' max: ', head.max()) print(" max: ", head.max())
print(' std: ', head.std()) print(" std: ", head.std())
# Extract flow right face and flow front face # Extract flow right face and flow front face
frf = cbb.get_data(text='FLOW RIGHT FACE', totim=time)[0] frf = cbb.get_data(text="FLOW RIGHT FACE", totim=time)[0]
fff = cbb.get_data(text='FLOW FRONT FACE', totim=time)[0] fff = cbb.get_data(text="FLOW FRONT FACE", totim=time)[0]
#Create the plot # Create the plot
f = plt.figure() f = plt.figure()
plt.subplot(1, 1, 1, aspect='equal') plt.subplot(1, 1, 1, aspect="equal")
plt.title('stress period ' + str(iplot + 1)) plt.title("stress period " + str(iplot + 1))
modelmap = flopy.plot.PlotMapView(model=mf, layer=0)
qm = modelmap.plot_ibound()
lc = modelmap.plot_grid()
qm = modelmap.plot_bc("GHB", alpha=0.5)
cs = modelmap.contour_array(head, levels=levels)
plt.clabel(cs, inline=1, fontsize=10, fmt="%1.1f")
quiver = modelmap.plot_discharge(frf, fff, head=head)
modelmap = flopy.plot.PlotMapView(model=mf, layer=0) mfc = "None"
qm = modelmap.plot_ibound() if (iplot + 1) == len(mytimes):
lc = modelmap.plot_grid() mfc = "black"
qm = modelmap.plot_bc('GHB', alpha=0.5) plt.plot(
cs = modelmap.contour_array(head, levels=levels) wpt[0],
plt.clabel(cs, inline=1, fontsize=10, fmt='%1.1f') wpt[1],
quiver = modelmap.plot_discharge(frf, fff, head=head) lw=0,
marker="o",
markersize=8,
mfc = 'None' markeredgewidth=0.5,
if (iplot+1) == len(mytimes): markeredgecolor="black",
mfc='black' markerfacecolor=mfc,
plt.plot(wpt[0], wpt[1], lw=0, marker='o', markersize=8, zorder=9,
markeredgewidth=0.5, )
markeredgecolor='black', markerfacecolor=mfc, zorder=9) plt.text(wpt[0] + 25, wpt[1] - 25, "well", size=12, zorder=12)
plt.text(wpt[0]+25, wpt[1]-25, 'well', size=12, zorder=12) plt.savefig("tutorial2-{}.png".format(iplot))
plt.savefig('tutorial2-{}.png'.format(iplot))
# Plot the head versus time # Plot the head versus time
idx = (0, int(nrow / 2) - 1, int(ncol / 2) - 1) idx = (0, int(nrow / 2) - 1, int(ncol / 2) - 1)
ts = headobj.get_ts(idx) ts = headobj.get_ts(idx)
plt.subplot(1, 1, 1) plt.subplot(1, 1, 1)
ttl = 'Head at cell ({0},{1},{2})'.format(idx[0] + 1, idx[1] + 1, idx[2] + 1) ttl = "Head at cell ({0},{1},{2})".format(idx[0] + 1, idx[1] + 1, idx[2] + 1)
plt.title(ttl) plt.title(ttl)
plt.xlabel('time') plt.xlabel("time")
plt.ylabel('head') plt.ylabel("head")
plt.plot(ts[:, 0], ts[:, 1], 'bo-') plt.plot(ts[:, 0], ts[:, 1], "bo-")
plt.savefig('tutorial2-ts.png') plt.savefig("tutorial2-ts.png")

343
examples/Notebooks/setup_pmv_demo.py
View File

@ -11,21 +11,22 @@ def run():
try: try:
import flopy import flopy
except: except:
fpth = os.path.abspath(os.path.join('..', '..')) fpth = os.path.abspath(os.path.join("..", ".."))
sys.path.append(fpth) sys.path.append(fpth)
import flopy import flopy
print(sys.version) print(sys.version)
print('numpy version: {}'.format(np.__version__)) print("numpy version: {}".format(np.__version__))
print('matplotlib version: {}'.format(mpl.__version__)) print("matplotlib version: {}".format(mpl.__version__))
print('flopy version: {}'.format(flopy.__version__)) print("flopy version: {}".format(flopy.__version__))
if not os.path.exists("data"): if not os.path.exists("data"):
os.mkdir("data") os.mkdir("data")
from flopy.utils.gridgen import Gridgen from flopy.utils.gridgen import Gridgen
Lx = 10000.
Ly = 10500. Lx = 10000.0
Ly = 10500.0
nlay = 3 nlay = 3
nrow = 21 nrow = 21
ncol = 20 ncol = 20
@ -35,180 +36,267 @@ def run():
botm = [220, 200, 0] botm = [220, 200, 0]
ms = flopy.modflow.Modflow() ms = flopy.modflow.Modflow()
dis5 = flopy.modflow.ModflowDis(ms, nlay=nlay, nrow=nrow, ncol=ncol, delr=delr, dis5 = flopy.modflow.ModflowDis(
delc=delc, top=top, botm=botm) ms,
nlay=nlay,
nrow=nrow,
ncol=ncol,
delr=delr,
delc=delc,
top=top,
botm=botm,
)
model_name = 'mp7p2' model_name = "mp7p2"
model_ws = os.path.join('data', 'mp7_ex2', 'mf6') model_ws = os.path.join("data", "mp7_ex2", "mf6")
gridgen_ws = os.path.join(model_ws, 'gridgen') gridgen_ws = os.path.join(model_ws, "gridgen")
g = Gridgen(dis5, model_ws=gridgen_ws) g = Gridgen(dis5, model_ws=gridgen_ws)
rf0shp = os.path.join(gridgen_ws, 'rf0') rf0shp = os.path.join(gridgen_ws, "rf0")
xmin = 7 * delr xmin = 7 * delr
xmax = 12 * delr xmax = 12 * delr
ymin = 8 * delc ymin = 8 * delc
ymax = 13 * delc ymax = 13 * delc
rfpoly = [[[(xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax), (xmin, ymin)]]] rfpoly = [
g.add_refinement_features(rfpoly, 'polygon', 1, range(nlay)) [
[
(xmin, ymin),
(xmax, ymin),
(xmax, ymax),
(xmin, ymax),
(xmin, ymin),
]
]
]
g.add_refinement_features(rfpoly, "polygon", 1, range(nlay))
rf1shp = os.path.join(gridgen_ws, 'rf1') rf1shp = os.path.join(gridgen_ws, "rf1")
xmin = 8 * delr xmin = 8 * delr
xmax = 11 * delr xmax = 11 * delr
ymin = 9 * delc ymin = 9 * delc
ymax = 12 * delc ymax = 12 * delc
rfpoly = [[[(xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax), (xmin, ymin)]]] rfpoly = [
g.add_refinement_features(rfpoly, 'polygon', 2, range(nlay)) [
[
(xmin, ymin),
(xmax, ymin),
(xmax, ymax),
(xmin, ymax),
(xmin, ymin),
]
]
]
g.add_refinement_features(rfpoly, "polygon", 2, range(nlay))
rf2shp = os.path.join(gridgen_ws, 'rf2') rf2shp = os.path.join(gridgen_ws, "rf2")
xmin = 9 * delr xmin = 9 * delr
xmax = 10 * delr xmax = 10 * delr
ymin = 10 * delc ymin = 10 * delc
ymax = 11 * delc ymax = 11 * delc
rfpoly = [[[(xmin, ymin), (xmax, ymin), (xmax, ymax), (xmin, ymax), (xmin, ymin)]]] rfpoly = [
g.add_refinement_features(rfpoly, 'polygon', 3, range(nlay)) [
[
(xmin, ymin),
(xmax, ymin),
(xmax, ymax),
(xmin, ymax),
(xmin, ymin),
]
]
]
g.add_refinement_features(rfpoly, "polygon", 3, range(nlay))
g.build(verbose=False) g.build(verbose=False)
gridprops = g.get_gridprops_disv() gridprops = g.get_gridprops_disv()
ncpl = gridprops['ncpl'] ncpl = gridprops["ncpl"]
top = gridprops['top'] top = gridprops["top"]
botm = gridprops['botm'] botm = gridprops["botm"]
nvert = gridprops['nvert'] nvert = gridprops["nvert"]
vertices = gridprops['vertices'] vertices = gridprops["vertices"]
cell2d = gridprops['cell2d'] cell2d = gridprops["cell2d"]
# cellxy = gridprops['cellxy'] # cellxy = gridprops['cellxy']
# create simulation # create simulation
sim = flopy.mf6.MFSimulation(sim_name=model_name, version='mf6', exe_name='mf6', sim = flopy.mf6.MFSimulation(
sim_ws=model_ws) sim_name=model_name, version="mf6", exe_name="mf6", sim_ws=model_ws
)
# create tdis package # create tdis package
tdis_rc = [(1000.0, 1, 1.0)] tdis_rc = [(1000.0, 1, 1.0)]
tdis = flopy.mf6.ModflowTdis(sim, pname='tdis', time_units='DAYS', tdis = flopy.mf6.ModflowTdis(
perioddata=tdis_rc) sim, pname="tdis", time_units="DAYS", perioddata=tdis_rc
)
# create gwf model # create gwf model
gwf = flopy.mf6.ModflowGwf(sim, modelname=model_name, gwf = flopy.mf6.ModflowGwf(
model_nam_file='{}.nam'.format(model_name)) sim, modelname=model_name, model_nam_file="{}.nam".format(model_name)
)
gwf.name_file.save_flows = True gwf.name_file.save_flows = True
# create iterative model solution and register the gwf model with it # create iterative model solution and register the gwf model with it
ims = flopy.mf6.ModflowIms(sim, pname='ims', print_option='SUMMARY', ims = flopy.mf6.ModflowIms(
complexity='SIMPLE', outer_hclose=1.e-5, sim,
outer_maximum=100, under_relaxation='NONE', pname="ims",
inner_maximum=100, inner_hclose=1.e-6, print_option="SUMMARY",
rcloserecord=0.1, linear_acceleration='BICGSTAB', complexity="SIMPLE",
scaling_method='NONE', reordering_method='NONE', outer_hclose=1.0e-5,
relaxation_factor=0.99) outer_maximum=100,
under_relaxation="NONE",
inner_maximum=100,
inner_hclose=1.0e-6,
rcloserecord=0.1,
linear_acceleration="BICGSTAB",
scaling_method="NONE",
reordering_method="NONE",
relaxation_factor=0.99,
)
sim.register_ims_package(ims, [gwf.name]) sim.register_ims_package(ims, [gwf.name])
# disv # disv
disv = flopy.mf6.ModflowGwfdisv(gwf, nlay=nlay, ncpl=ncpl, disv = flopy.mf6.ModflowGwfdisv(
top=top, botm=botm, gwf,
nvert=nvert, vertices=vertices, nlay=nlay,
cell2d=cell2d) ncpl=ncpl,
top=top,
botm=botm,
nvert=nvert,
vertices=vertices,
cell2d=cell2d,
)
# initial conditions # initial conditions
ic = flopy.mf6.ModflowGwfic(gwf, pname='ic', strt=320.) ic = flopy.mf6.ModflowGwfic(gwf, pname="ic", strt=320.0)
# node property flow # node property flow
npf = flopy.mf6.ModflowGwfnpf(gwf, xt3doptions=[('xt3d')], npf = flopy.mf6.ModflowGwfnpf(
save_specific_discharge=True, gwf,
icelltype=[1, 0, 0], xt3doptions=[("xt3d")],
k=[50.0, 0.01, 200.0], save_specific_discharge=True,
k33=[10., 0.01, 20.]) icelltype=[1, 0, 0],
k=[50.0, 0.01, 200.0],
k33=[10.0, 0.01, 20.0],
)
# wel # wel
wellpoints = [(4750., 5250.)] wellpoints = [(4750.0, 5250.0)]
welcells = g.intersect(wellpoints, 'point', 0) welcells = g.intersect(wellpoints, "point", 0)
# welspd = flopy.mf6.ModflowGwfwel.stress_period_data.empty(gwf, maxbound=1, aux_vars=['iface']) # welspd = flopy.mf6.ModflowGwfwel.stress_period_data.empty(gwf, maxbound=1, aux_vars=['iface'])
welspd = [[(2, icpl), -150000, 0] for icpl in welcells['nodenumber']] welspd = [[(2, icpl), -150000, 0] for icpl in welcells["nodenumber"]]
wel = flopy.mf6.ModflowGwfwel(gwf, print_input=True, wel = flopy.mf6.ModflowGwfwel(
auxiliary=[('iface',)], gwf,
stress_period_data=welspd) print_input=True,
auxiliary=[("iface",)],
stress_period_data=welspd,
)
# rch # rch
aux = [np.ones(ncpl, dtype=np.int) * 6] aux = [np.ones(ncpl, dtype=np.int) * 6]
rch = flopy.mf6.ModflowGwfrcha(gwf, recharge=0.005, rch = flopy.mf6.ModflowGwfrcha(
auxiliary=[('iface',)], gwf, recharge=0.005, auxiliary=[("iface",)], aux={0: [6]}
aux={0: [6]}) )
# riv # riv
riverline = [[[(Lx - 1., Ly), (Lx - 1., 0.)]]] riverline = [[[(Lx - 1.0, Ly), (Lx - 1.0, 0.0)]]]
rivcells = g.intersect(riverline, 'line', 0) rivcells = g.intersect(riverline, "line", 0)
rivspd = [[(0, icpl), 320., 100000., 318] for icpl in rivcells['nodenumber']] rivspd = [
[(0, icpl), 320.0, 100000.0, 318] for icpl in rivcells["nodenumber"]
]
riv = flopy.mf6.ModflowGwfriv(gwf, stress_period_data=rivspd) riv = flopy.mf6.ModflowGwfriv(gwf, stress_period_data=rivspd)
# output control # output control
oc = flopy.mf6.ModflowGwfoc(gwf, pname='oc', budget_filerecord='{}.cbb'.format(model_name), oc = flopy.mf6.ModflowGwfoc(
head_filerecord='{}.hds'.format(model_name), gwf,
headprintrecord=[('COLUMNS', 10, 'WIDTH', 15, pname="oc",
'DIGITS', 6, 'GENERAL')], budget_filerecord="{}.cbb".format(model_name),
saverecord=[('HEAD', 'ALL'), ('BUDGET', 'ALL')], head_filerecord="{}.hds".format(model_name),
printrecord=[('HEAD', 'ALL'), ('BUDGET', 'ALL')]) headprintrecord=[("COLUMNS", 10, "WIDTH", 15, "DIGITS", 6, "GENERAL")],
saverecord=[("HEAD", "ALL"), ("BUDGET", "ALL")],
printrecord=[("HEAD", "ALL"), ("BUDGET", "ALL")],
)
sim.write_simulation() sim.write_simulation()
sim.run_simulation() sim.run_simulation()
mp_namea = model_name + 'a_mp' mp_namea = model_name + "a_mp"
mp_nameb = model_name + 'b_mp' mp_nameb = model_name + "b_mp"
pcoord = np.array([[0.000, 0.125, 0.500], pcoord = np.array(
[0.000, 0.375, 0.500], [
[0.000, 0.625, 0.500], [0.000, 0.125, 0.500],
[0.000, 0.875, 0.500], [0.000, 0.375, 0.500],
[1.000, 0.125, 0.500], [0.000, 0.625, 0.500],
[1.000, 0.375, 0.500], [0.000, 0.875, 0.500],
[1.000, 0.625, 0.500], [1.000, 0.125, 0.500],
[1.000, 0.875, 0.500], [1.000, 0.375, 0.500],
[0.125, 0.000, 0.500], [1.000, 0.625, 0.500],
[0.375, 0.000, 0.500], [1.000, 0.875, 0.500],
[0.625, 0.000, 0.500], [0.125, 0.000, 0.500],
[0.875, 0.000, 0.500], [0.375, 0.000, 0.500],
[0.125, 1.000, 0.500], [0.625, 0.000, 0.500],
[0.375, 1.000, 0.500], [0.875, 0.000, 0.500],
[0.625, 1.000, 0.500], [0.125, 1.000, 0.500],
[0.875, 1.000, 0.500]]) [0.375, 1.000, 0.500],
nodew = gwf.disv.ncpl.array * 2 + welcells['nodenumber'][0] [0.625, 1.000, 0.500],
[0.875, 1.000, 0.500],
]
)
nodew = gwf.disv.ncpl.array * 2 + welcells["nodenumber"][0]
plocs = [nodew for i in range(pcoord.shape[0])] plocs = [nodew for i in range(pcoord.shape[0])]
# create particle data # create particle data
pa = flopy.modpath.ParticleData(plocs, structured=False, pa = flopy.modpath.ParticleData(
localx=pcoord[:, 0], plocs,
localy=pcoord[:, 1], structured=False,
localz=pcoord[:, 2], localx=pcoord[:, 0],
drape=0) localy=pcoord[:, 1],
localz=pcoord[:, 2],
drape=0,
)
# create backward particle group # create backward particle group
fpth = mp_namea + '.sloc' fpth = mp_namea + ".sloc"
pga = flopy.modpath.ParticleGroup(particlegroupname='BACKWARD1', particledata=pa, pga = flopy.modpath.ParticleGroup(
filename=fpth) particlegroupname="BACKWARD1", particledata=pa, filename=fpth
)
facedata = flopy.modpath.FaceDataType(drape=0, facedata = flopy.modpath.FaceDataType(
verticaldivisions1=10, horizontaldivisions1=10, drape=0,
verticaldivisions2=10, horizontaldivisions2=10, verticaldivisions1=10,
verticaldivisions3=10, horizontaldivisions3=10, horizontaldivisions1=10,
verticaldivisions4=10, horizontaldivisions4=10, verticaldivisions2=10,
rowdivisions5=0, columndivisions5=0, horizontaldivisions2=10,
rowdivisions6=4, columndivisions6=4) verticaldivisions3=10,
horizontaldivisions3=10,
verticaldivisions4=10,
horizontaldivisions4=10,
rowdivisions5=0,
columndivisions5=0,
rowdivisions6=4,
columndivisions6=4,
)
pb = flopy.modpath.NodeParticleData(subdivisiondata=facedata, nodes=nodew) pb = flopy.modpath.NodeParticleData(subdivisiondata=facedata, nodes=nodew)
# create forward particle group # create forward particle group
fpth = mp_nameb + '.sloc' fpth = mp_nameb + ".sloc"
pgb = flopy.modpath.ParticleGroupNodeTemplate(particlegroupname='BACKWARD2', pgb = flopy.modpath.ParticleGroupNodeTemplate(
particledata=pb, particlegroupname="BACKWARD2", particledata=pb, filename=fpth
filename=fpth) )
# create modpath files # create modpath files
mp = flopy.modpath.Modpath7(modelname=mp_namea, flowmodel=gwf, mp = flopy.modpath.Modpath7(
exe_name='mp7', model_ws=model_ws) modelname=mp_namea, flowmodel=gwf, exe_name="mp7", model_ws=model_ws
)
flopy.modpath.Modpath7Bas(mp, porosity=0.1) flopy.modpath.Modpath7Bas(mp, porosity=0.1)
flopy.modpath.Modpath7Sim(mp, simulationtype='combined', flopy.modpath.Modpath7Sim(
trackingdirection='backward', mp,
weaksinkoption='pass_through', simulationtype="combined",
weaksourceoption='pass_through', trackingdirection="backward",
referencetime=0., weaksinkoption="pass_through",
stoptimeoption='extend', weaksourceoption="pass_through",
timepointdata=[500, 1000.], referencetime=0.0,
particlegroups=pga) stoptimeoption="extend",
timepointdata=[500, 1000.0],
particlegroups=pga,
)
# write modpath datasets # write modpath datasets
mp.write_input() mp.write_input()
@ -217,16 +305,20 @@ def run():
mp.run_model() mp.run_model()
# create modpath files # create modpath files
mp = flopy.modpath.Modpath7(modelname=mp_nameb, flowmodel=gwf, mp = flopy.modpath.Modpath7(
exe_name='mp7', model_ws=model_ws) modelname=mp_nameb, flowmodel=gwf, exe_name="mp7", model_ws=model_ws
)
flopy.modpath.Modpath7Bas(mp, porosity=0.1) flopy.modpath.Modpath7Bas(mp, porosity=0.1)
flopy.modpath.Modpath7Sim(mp, simulationtype='endpoint', flopy.modpath.Modpath7Sim(
trackingdirection='backward', mp,
weaksinkoption='pass_through', simulationtype="endpoint",
weaksourceoption='pass_through', trackingdirection="backward",
referencetime=0., weaksinkoption="pass_through",
stoptimeoption='extend', weaksourceoption="pass_through",
particlegroups=pgb) referencetime=0.0,
stoptimeoption="extend",
particlegroups=pgb,
)
# write modpath datasets # write modpath datasets
mp.write_input() mp.write_input()
@ -235,5 +327,6 @@ def run():
mp.run_model() mp.run_model()
return return
if __name__ == "__main__": if __name__ == "__main__":
run() run()

55
examples/Tutorials/Tutorial01/tutorial01.py
View File

@ -1,42 +1,42 @@
import numpy as np import numpy as np
import flopy import flopy
# Assign name and create modflow model object # Assign name and create modflow model object
modelname = 'tutorial1' modelname = "tutorial1"
mf = flopy.modflow.Modflow(modelname, exe_name='mf2005') mf = flopy.modflow.Modflow(modelname, exe_name="mf2005")
# Model domain and grid definition # Model domain and grid definition
Lx = 1000. Lx = 1000.0
Ly = 1000. Ly = 1000.0
ztop = 0. ztop = 0.0
zbot = -50. zbot = -50.0
nlay = 1 nlay = 1
nrow = 10 nrow = 10
ncol = 10 ncol = 10
delr = Lx/ncol delr = Lx / ncol
delc = Ly/nrow delc = Ly / nrow
delv = (ztop - zbot) / nlay delv = (ztop - zbot) / nlay
botm = np.linspace(ztop, zbot, nlay + 1) botm = np.linspace(ztop, zbot, nlay + 1)
# Create the discretization object # Create the discretization object
dis = flopy.modflow.ModflowDis(mf, nlay, nrow, ncol, delr=delr, delc=delc, dis = flopy.modflow.ModflowDis(
top=ztop, botm=botm[1:]) mf, nlay, nrow, ncol, delr=delr, delc=delc, top=ztop, botm=botm[1:]
)
# Variables for the BAS package # Variables for the BAS package
ibound = np.ones((nlay, nrow, ncol), dtype=np.int32) ibound = np.ones((nlay, nrow, ncol), dtype=np.int32)
ibound[:, :, 0] = -1 ibound[:, :, 0] = -1
ibound[:, :, -1] = -1 ibound[:, :, -1] = -1
strt = np.ones((nlay, nrow, ncol), dtype=np.float32) strt = np.ones((nlay, nrow, ncol), dtype=np.float32)
strt[:, :, 0] = 10. strt[:, :, 0] = 10.0
strt[:, :, -1] = 0. strt[:, :, -1] = 0.0
bas = flopy.modflow.ModflowBas(mf, ibound=ibound, strt=strt) bas = flopy.modflow.ModflowBas(mf, ibound=ibound, strt=strt)
# Add LPF package to the MODFLOW model # Add LPF package to the MODFLOW model
lpf = flopy.modflow.ModflowLpf(mf, hk=10., vka=10., ipakcb=53) lpf = flopy.modflow.ModflowLpf(mf, hk=10.0, vka=10.0, ipakcb=53)
# Add OC package to the MODFLOW model # Add OC package to the MODFLOW model
spd = {(0, 0): ['print head', 'print budget', 'save head', 'save budget']} spd = {(0, 0): ["print head", "print budget", "save head", "save budget"]}
oc = flopy.modflow.ModflowOc(mf, stress_period_data=spd, compact=True) oc = flopy.modflow.ModflowOc(mf, stress_period_data=spd, compact=True)
# Add PCG package to the MODFLOW model # Add PCG package to the MODFLOW model
@ -52,33 +52,30 @@ success, buff = mf.run_model()
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import flopy.utils.binaryfile as bf import flopy.utils.binaryfile as bf
plt.subplot(1, 1, 1, aspect='equal') plt.subplot(1, 1, 1, aspect="equal")
hds = bf.HeadFile(modelname + '.hds') hds = bf.HeadFile(modelname + ".hds")
head = hds.get_data(totim=1.0) head = hds.get_data(totim=1.0)
levels = np.arange(1, 10, 1) levels = np.arange(1, 10, 1)
extent = (delr / 2., Lx - delr / 2., Ly - delc / 2., delc / 2.) extent = (delr / 2.0, Lx - delr / 2.0, Ly - delc / 2.0, delc / 2.0)
plt.contour(head[0, :, :], levels=levels, extent=extent) plt.contour(head[0, :, :], levels=levels, extent=extent)
plt.savefig('tutorial1a.png') plt.savefig("tutorial1a.png")
fig = plt.figure(figsize=(10,10)) fig = plt.figure(figsize=(10, 10))
ax = fig.add_subplot(1, 1, 1, aspect='equal') ax = fig.add_subplot(1, 1, 1, aspect="equal")
hds = bf.HeadFile(modelname+'.hds') hds = bf.HeadFile(modelname + ".hds")
times = hds.get_times() times = hds.get_times()
head = hds.get_data(totim=times[-1]) head = hds.get_data(totim=times[-1])
levels = np.linspace(0, 10, 11) levels = np.linspace(0, 10, 11)
cbb = bf.CellBudgetFile(modelname+'.cbc') cbb = bf.CellBudgetFile(modelname + ".cbc")
kstpkper_list = cbb.get_kstpkper() kstpkper_list = cbb.get_kstpkper()
frf = cbb.get_data(text='FLOW RIGHT FACE', totim=times[-1])[0] frf = cbb.get_data(text="FLOW RIGHT FACE", totim=times[-1])[0]
fff = cbb.get_data(text='FLOW FRONT FACE', totim=times[-1])[0] fff = cbb.get_data(text="FLOW FRONT FACE", totim=times[-1])[0]
modelmap = flopy.plot.ModelMap(model=mf, layer=0) modelmap = flopy.plot.ModelMap(model=mf, layer=0)
qm = modelmap.plot_ibound() qm = modelmap.plot_ibound()
lc = modelmap.plot_grid() lc = modelmap.plot_grid()
cs = modelmap.contour_array(head, levels=levels) cs = modelmap.contour_array(head, levels=levels)
quiver = modelmap.plot_discharge(frf, fff, head=head) quiver = modelmap.plot_discharge(frf, fff, head=head)
plt.savefig('tutorial1b.png') plt.savefig("tutorial1b.png")

163
examples/Tutorials/Tutorial02/tutorial02.py
View File

@ -1,12 +1,11 @@
import numpy as np import numpy as np
import flopy import flopy
# Model domain and grid definition # Model domain and grid definition
Lx = 1000. Lx = 1000.0
Ly = 1000. Ly = 1000.0
ztop = 10. ztop = 10.0
zbot = -50. zbot = -50.0
nlay = 1 nlay = 1
nrow = 10 nrow = 10
ncol = 10 ncol = 10
@ -14,16 +13,16 @@ delr = Lx / ncol
delc = Ly / nrow delc = Ly / nrow
delv = (ztop - zbot) / nlay delv = (ztop - zbot) / nlay
botm = np.linspace(ztop, zbot, nlay + 1) botm = np.linspace(ztop, zbot, nlay + 1)
hk = 1. hk = 1.0
vka = 1. vka = 1.0
sy = 0.1 sy = 0.1
ss = 1.e-4 ss = 1.0e-4
laytyp = 1 laytyp = 1
# Variables for the BAS package # Variables for the BAS package
# Note that changes from the previous tutorial! # Note that changes from the previous tutorial!
ibound = np.ones((nlay, nrow, ncol), dtype=np.int32) ibound = np.ones((nlay, nrow, ncol), dtype=np.int32)
strt = 10. * np.ones((nlay, nrow, ncol), dtype=np.float32) strt = 10.0 * np.ones((nlay, nrow, ncol), dtype=np.float32)
# Time step parameters # Time step parameters
nper = 3 nper = 3
@ -32,20 +31,31 @@ nstp = [1, 100, 100]
steady = [True, False, False] steady = [True, False, False]
# Flopy objects # Flopy objects
modelname = 'tutorial2' modelname = "tutorial2"
mf = flopy.modflow.Modflow(modelname, exe_name='mf2005') mf = flopy.modflow.Modflow(modelname, exe_name="mf2005")
dis = flopy.modflow.ModflowDis(mf, nlay, nrow, ncol, delr=delr, delc=delc, dis = flopy.modflow.ModflowDis(
top=ztop, botm=botm[1:], mf,
nper=nper, perlen=perlen, nstp=nstp, nlay,
steady=steady) nrow,
ncol,
delr=delr,
delc=delc,
top=ztop,
botm=botm[1:],
nper=nper,
perlen=perlen,
nstp=nstp,
steady=steady,
)
bas = flopy.modflow.ModflowBas(mf, ibound=ibound, strt=strt) bas = flopy.modflow.ModflowBas(mf, ibound=ibound, strt=strt)
lpf = flopy.modflow.ModflowLpf(mf, hk=hk, vka=vka, sy=sy, ss=ss, laytyp=laytyp, lpf = flopy.modflow.ModflowLpf(
ipakcb=53) mf, hk=hk, vka=vka, sy=sy, ss=ss, laytyp=laytyp, ipakcb=53
)
pcg = flopy.modflow.ModflowPcg(mf) pcg = flopy.modflow.ModflowPcg(mf)
# Make list for stress period 1 # Make list for stress period 1
stageleft = 10. stageleft = 10.0
stageright = 10. stageright = 10.0
bound_sp1 = [] bound_sp1 = []
for il in range(nlay): for il in range(nlay):
condleft = hk * (stageleft - zbot) * delc condleft = hk * (stageleft - zbot) * delc
@ -53,11 +63,11 @@ for il in range(nlay):
for ir in range(nrow): for ir in range(nrow):
bound_sp1.append([il, ir, 0, stageleft, condleft]) bound_sp1.append([il, ir, 0, stageleft, condleft])
bound_sp1.append([il, ir, ncol - 1, stageright, condright]) bound_sp1.append([il, ir, ncol - 1, stageright, condright])
print('Adding ', len(bound_sp1), 'GHBs for stress period 1.') print("Adding ", len(bound_sp1), "GHBs for stress period 1.")
# Make list for stress period 2 # Make list for stress period 2
stageleft = 10. stageleft = 10.0
stageright = 0. stageright = 0.0
condleft = hk * (stageleft - zbot) * delc condleft = hk * (stageleft - zbot) * delc
condright = hk * (stageright - zbot) * delc condright = hk * (stageright - zbot) * delc
bound_sp2 = [] bound_sp2 = []
@ -65,7 +75,7 @@ for il in range(nlay):
for ir in range(nrow): for ir in range(nrow):
bound_sp2.append([il, ir, 0, stageleft, condleft]) bound_sp2.append([il, ir, 0, stageleft, condleft])
bound_sp2.append([il, ir, ncol - 1, stageright, condright]) bound_sp2.append([il, ir, ncol - 1, stageright, condright])
print('Adding ', len(bound_sp2), 'GHBs for stress period 2.') print("Adding ", len(bound_sp2), "GHBs for stress period 2.")
# We do not need to add a dictionary entry for stress period 3. # We do not need to add a dictionary entry for stress period 3.
# Flopy will automatically take the list from stress period 2 and apply it # Flopy will automatically take the list from stress period 2 and apply it
@ -77,10 +87,10 @@ ghb = flopy.modflow.ModflowGhb(mf, stress_period_data=stress_period_data)
# Create the well package # Create the well package
# Remember to use zero-based layer, row, column indices! # Remember to use zero-based layer, row, column indices!
pumping_rate = -500. pumping_rate = -500.0
wel_sp1 = [[0, nrow/2 - 1, ncol/2 - 1, 0.]] wel_sp1 = [[0, nrow / 2 - 1, ncol / 2 - 1, 0.0]]
wel_sp2 = [[0, nrow/2 - 1, ncol/2 - 1, 0.]] wel_sp2 = [[0, nrow / 2 - 1, ncol / 2 - 1, 0.0]]
wel_sp3 = [[0, nrow/2 - 1, ncol/2 - 1, pumping_rate]] wel_sp3 = [[0, nrow / 2 - 1, ncol / 2 - 1, pumping_rate]]
stress_period_data = {0: wel_sp1, 1: wel_sp2, 2: wel_sp3} stress_period_data = {0: wel_sp1, 1: wel_sp2, 2: wel_sp3}
wel = flopy.modflow.ModflowWel(mf, stress_period_data=stress_period_data) wel = flopy.modflow.ModflowWel(mf, stress_period_data=stress_period_data)
@ -88,13 +98,16 @@ wel = flopy.modflow.ModflowWel(mf, stress_period_data=stress_period_data)
stress_period_data = {} stress_period_data = {}
for kper in range(nper): for kper in range(nper):
for kstp in range(nstp[kper]): for kstp in range(nstp[kper]):
stress_period_data[(kper, kstp)] = ['save head', stress_period_data[(kper, kstp)] = [
'save drawdown', "save head",
'save budget', "save drawdown",
'print head', "save budget",
'print budget'] "print head",
oc = flopy.modflow.ModflowOc(mf, stress_period_data=stress_period_data, "print budget",
compact=True) ]
oc = flopy.modflow.ModflowOc(
mf, stress_period_data=stress_period_data, compact=True
)
# Write the model input files # Write the model input files
mf.write_input() mf.write_input()
@ -102,7 +115,7 @@ mf.write_input()
# Run the model # Run the model
success, mfoutput = mf.run_model(silent=False, pause=False) success, mfoutput = mf.run_model(silent=False, pause=False)
if not success: if not success:
raise Exception('MODFLOW did not terminate normally.') raise Exception("MODFLOW did not terminate normally.")
# Imports # Imports
@ -110,67 +123,73 @@ import matplotlib.pyplot as plt
import flopy.utils.binaryfile as bf import flopy.utils.binaryfile as bf
# Create the headfile and budget file objects # Create the headfile and budget file objects
headobj = bf.HeadFile(modelname+'.hds') headobj = bf.HeadFile(modelname + ".hds")
times = headobj.get_times() times = headobj.get_times()
cbb = bf.CellBudgetFile(modelname+'.cbc') cbb = bf.CellBudgetFile(modelname + ".cbc")
# Setup contour parameters # Setup contour parameters
levels = np.linspace(0, 10, 11) levels = np.linspace(0, 10, 11)
extent = (delr/2., Lx - delr/2., delc/2., Ly - delc/2.) extent = (delr / 2.0, Lx - delr / 2.0, delc / 2.0, Ly - delc / 2.0)
print('Levels: ', levels) print("Levels: ", levels)
print('Extent: ', extent) print("Extent: ", extent)
# Well point # Well point
wpt = ((float(ncol/2)-0.5)*delr, (float(nrow/2-1)+0.5)*delc) wpt = ((float(ncol / 2) - 0.5) * delr, (float(nrow / 2 - 1) + 0.5) * delc)
wpt = (450., 550.) wpt = (450.0, 550.0)
# Make the plots # Make the plots
mytimes = [1.0, 101.0, 201.0] mytimes = [1.0, 101.0, 201.0]
for iplot, time in enumerate(mytimes): for iplot, time in enumerate(mytimes):
print('*****Processing time: ', time) print("*****Processing time: ", time)
head = headobj.get_data(totim=time) head = headobj.get_data(totim=time)
#Print statistics # Print statistics
print('Head statistics') print("Head statistics")
print(' min: ', head.min()) print(" min: ", head.min())
print(' max: ', head.max()) print(" max: ", head.max())
print(' std: ', head.std()) print(" std: ", head.std())
# Extract flow right face and flow front face # Extract flow right face and flow front face
frf = cbb.get_data(text='FLOW RIGHT FACE', totim=time)[0] frf = cbb.get_data(text="FLOW RIGHT FACE", totim=time)[0]
fff = cbb.get_data(text='FLOW FRONT FACE', totim=time)[0] fff = cbb.get_data(text="FLOW FRONT FACE", totim=time)[0]
#Create the plot
#plt.subplot(1, len(mytimes), iplot + 1, aspect='equal')
plt.subplot(1, 1, 1, aspect='equal')
plt.title('stress period ' + str(iplot + 1))
# Create the plot
# plt.subplot(1, len(mytimes), iplot + 1, aspect='equal')
plt.subplot(1, 1, 1, aspect="equal")
plt.title("stress period " + str(iplot + 1))
modelmap = flopy.plot.ModelMap(model=mf, layer=0) modelmap = flopy.plot.ModelMap(model=mf, layer=0)
qm = modelmap.plot_ibound() qm = modelmap.plot_ibound()
lc = modelmap.plot_grid() lc = modelmap.plot_grid()
qm = modelmap.plot_bc('GHB', alpha=0.5) qm = modelmap.plot_bc("GHB", alpha=0.5)
cs = modelmap.contour_array(head, levels=levels) cs = modelmap.contour_array(head, levels=levels)
plt.clabel(cs, inline=1, fontsize=10, fmt='%1.1f') plt.clabel(cs, inline=1, fontsize=10, fmt="%1.1f")
quiver = modelmap.plot_discharge(frf, fff, head=head) quiver = modelmap.plot_discharge(frf, fff, head=head)
mfc = "None"
mfc = 'None' if (iplot + 1) == len(mytimes):
if (iplot+1) == len(mytimes): mfc = "black"
mfc='black' plt.plot(
plt.plot(wpt[0], wpt[1], lw=0, marker='o', markersize=8, wpt[0],
markeredgewidth=0.5, wpt[1],
markeredgecolor='black', markerfacecolor=mfc, zorder=9) lw=0,
plt.text(wpt[0]+25, wpt[1]-25, 'well', size=12, zorder=12) marker="o",
plt.savefig('tutorial2-{}.png'.format(iplot)) markersize=8,
markeredgewidth=0.5,
markeredgecolor="black",
markerfacecolor=mfc,
zorder=9,
)
plt.text(wpt[0] + 25, wpt[1] - 25, "well", size=12, zorder=12)
plt.savefig("tutorial2-{}.png".format(iplot))
# Plot the head versus time # Plot the head versus time
idx = (0, int(nrow/2) - 1, int(ncol/2) - 1) idx = (0, int(nrow / 2) - 1, int(ncol / 2) - 1)
ts = headobj.get_ts(idx) ts = headobj.get_ts(idx)
plt.subplot(1, 1, 1) plt.subplot(1, 1, 1)
ttl = 'Head at cell ({0},{1},{2})'.format(idx[0] + 1, idx[1] + 1, idx[2] + 1) ttl = "Head at cell ({0},{1},{2})".format(idx[0] + 1, idx[1] + 1, idx[2] + 1)
plt.title(ttl) plt.title(ttl)
plt.xlabel('time') plt.xlabel("time")
plt.ylabel('head') plt.ylabel("head")
plt.plot(ts[:, 0], ts[:, 1], 'bo-') plt.plot(ts[:, 0], ts[:, 1], "bo-")
plt.savefig('tutorial2-ts.png') plt.savefig("tutorial2-ts.png")

1
examples/groundwater_paper/scripts/ex01-script.py
View File

@ -1,4 +1,5 @@
import flopy import flopy
model = flopy.modflow.Modflow() model = flopy.modflow.Modflow()
dis = flopy.modflow.ModflowDis() dis = flopy.modflow.ModflowDis()
bas = flopy.modflow.ModflowBas() bas = flopy.modflow.ModflowBas()

136
examples/groundwater_paper/scripts/uspb_capture.py
View File

@ -5,19 +5,24 @@ import numpy as np
import flopy import flopy
base_pth = os.path.join('data', 'uspb', 'flopy') base_pth = os.path.join("data", "uspb", "flopy")
cf_pth = os.path.join('data', 'uspb', 'cf') cf_pth = os.path.join("data", "uspb", "cf")
res_pth = os.path.join('data', 'uspb', 'results') res_pth = os.path.join("data", "uspb", "results")
def cf_model(model, k, i, j, base, Q=-100): def cf_model(model, k, i, j, base, Q=-100):
wd = {1: [[k, i, j, Q]]} wd = {1: [[k, i, j, Q]]}
model.remove_package('WEL') model.remove_package("WEL")
wel = flopy.modflow.ModflowWel(model=model, stress_period_data=wd) wel = flopy.modflow.ModflowWel(model=model, stress_period_data=wd)
wel.write_file() wel.write_file()
model.run_model(silent=True) model.run_model(silent=True)
# get the results # get the results
hedObj = flopy.utils.HeadFile(os.path.join(cf_pth, 'DG.hds'), precision='double') hedObj = flopy.utils.HeadFile(
cbcObj = flopy.utils.CellBudgetFile(os.path.join(cf_pth, 'DG.cbc'), precision='double') os.path.join(cf_pth, "DG.hds"), precision="double"
)
cbcObj = flopy.utils.CellBudgetFile(
os.path.join(cf_pth, "DG.cbc"), precision="double"
)
kk = hedObj.get_kstpkper() kk = hedObj.get_kstpkper()
v = np.zeros((len(kk)), dtype=np.float) v = np.zeros((len(kk)), dtype=np.float)
h = hedObj.get_ts((k, i, j)) h = hedObj.get_ts((k, i, j))
@ -25,15 +30,22 @@ def cf_model(model, k, i, j, base, Q=-100):
if h[idx, 1] == model.lpf.hdry: if h[idx, 1] == model.lpf.hdry:
v[idx] = np.nan v[idx] = np.nan
else: else:
v1 = cbcObj.get_data(kstpkper=kon, text='DRAINS', full3D=True)[0] v1 = cbcObj.get_data(kstpkper=kon, text="DRAINS", full3D=True)[0]
v2 = cbcObj.get_data(kstpkper=kon, text='STREAM LEAKAGE', full3D=True)[0] v2 = cbcObj.get_data(
v3 = cbcObj.get_data(kstpkper=kon, text='ET', full3D=True)[0] kstpkper=kon, text="STREAM LEAKAGE", full3D=True
)[0]
v3 = cbcObj.get_data(kstpkper=kon, text="ET", full3D=True)[0]
v[idx] = ((v1.sum() + v2.sum() + v3.sum()) - base) / (-Q) v[idx] = ((v1.sum() + v2.sum() + v3.sum()) - base) / (-Q)
return v return v
ml = flopy.modflow.Modflow.load('DG.nam', version='mf2005', exe_name='mf2005dbl', ml = flopy.modflow.Modflow.load(
verbose=True, model_ws=base_pth) "DG.nam",
version="mf2005",
exe_name="mf2005dbl",
verbose=True,
model_ws=base_pth,
)
# set a few variables from the model # set a few variables from the model
nrow, ncol = ml.dis.nrow, ml.dis.ncol nrow, ncol = ml.dis.nrow, ml.dis.ncol
@ -45,34 +57,38 @@ ml.write_input()
ml.run_model() ml.run_model()
# get base model results # get base model results
cbcObj = flopy.utils.CellBudgetFile(os.path.join(cf_pth, 'DG.cbc'), precision='double') cbcObj = flopy.utils.CellBudgetFile(
v1 = cbcObj.get_data(kstpkper=(0, 0), text='DRAINS', full3D=True)[0] os.path.join(cf_pth, "DG.cbc"), precision="double"
v2 = cbcObj.get_data(kstpkper=(0, 0), text='STREAM LEAKAGE', full3D=True)[0] )
v3 = cbcObj.get_data(kstpkper=(0, 0), text='ET', full3D=True)[0] v1 = cbcObj.get_data(kstpkper=(0, 0), text="DRAINS", full3D=True)[0]
v2 = cbcObj.get_data(kstpkper=(0, 0), text="STREAM LEAKAGE", full3D=True)[0]
v3 = cbcObj.get_data(kstpkper=(0, 0), text="ET", full3D=True)[0]
baseQ = v1.sum() + v2.sum() + v3.sum() baseQ = v1.sum() + v2.sum() + v3.sum()
# modify OC # modify OC
ml.remove_package('OC') ml.remove_package("OC")
stress_period_data = {(1,9): ['save head', 'save budget', 'print budget'], stress_period_data = {
(1,10): [], (1, 9): ["save head", "save budget", "print budget"],
(1,19): ['save head', 'save budget', 'print budget'], (1, 10): [],
(1,20): [], (1, 19): ["save head", "save budget", "print budget"],
(1,29): ['save head', 'save budget', 'print budget'], (1, 20): [],
(1,30): [], (1, 29): ["save head", "save budget", "print budget"],
(1,39): ['save head', 'save budget', 'print budget'], (1, 30): [],
(1,40): [], (1, 39): ["save head", "save budget", "print budget"],
(1,49): ['save head', 'save budget', 'print budget'], (1, 40): [],
(1,50): [], (1, 49): ["save head", "save budget", "print budget"],
(1,59): ['save head', 'save budget', 'print budget'], (1, 50): [],
(1,60): [], (1, 59): ["save head", "save budget", "print budget"],
(1,69): ['save head', 'save budget', 'print budget'], (1, 60): [],
(1,70): [], (1, 69): ["save head", "save budget", "print budget"],
(1,79): ['save head', 'save budget', 'print budget'], (1, 70): [],
(1,80): [], (1, 79): ["save head", "save budget", "print budget"],
(1,89): ['save head', 'save budget', 'print budget'], (1, 80): [],
(1,90): [], (1, 89): ["save head", "save budget", "print budget"],
(1,99): ['save head', 'save budget', 'print budget'], (1, 90): [],
(1,100): []} (1, 99): ["save head", "save budget", "print budget"],
(1, 100): [],
}
oc = flopy.modflow.ModflowOc(ml, stress_period_data=stress_period_data) oc = flopy.modflow.ModflowOc(ml, stress_period_data=stress_period_data)
oc.write_file() oc.write_file()
@ -82,12 +98,22 @@ nrow2 = nrow // nstep
ncol2 = ncol // nstep ncol2 = ncol // nstep
# open summary file # open summary file
fs = open(os.path.join('data', 'uspb', 'uspb_capture_{}.out'.format(nstep)), 'w', 0) fs = open(
os.path.join("data", "uspb", "uspb_capture_{}.out".format(nstep)), "w", 0
)
# write some summary information # write some summary information
fs.write('Problem size: {} rows and {} columns.\n'.format(nrow, ncol)) fs.write("Problem size: {} rows and {} columns.\n".format(nrow, ncol))
fs.write('Capture fraction analysis performed every {} rows and columns.\n'.format(nstep)) fs.write(
fs.write('Maximum number of analyses: {} rows and {} columns.\n'.format(nrow2, ncol2)) "Capture fraction analysis performed every {} rows and columns.\n".format(
nstep
)
)
fs.write(
"Maximum number of analyses: {} rows and {} columns.\n".format(
nrow2, ncol2
)
)
# create array to store capture fraction data (subset of model) # create array to store capture fraction data (subset of model)
cf_array = np.empty((10, nrow2, ncol2), dtype=np.float) cf_array = np.empty((10, nrow2, ncol2), dtype=np.float)
@ -104,15 +130,17 @@ for i in range(0, nrow, nstep):
jcnt = 0 jcnt = 0
for j in range(0, ncol, nstep): for j in range(0, ncol, nstep):
if ibound[i, j] < 1: if ibound[i, j] < 1:
sys.stdout.write('.') sys.stdout.write(".")
else: else:
line = '\nrow {} of {} - col {} of {}\n'.format(icnt+1, nrow2, jcnt+1, ncol2) line = "\nrow {} of {} - col {} of {}\n".format(
icnt + 1, nrow2, jcnt + 1, ncol2
)
fs.write(line) fs.write(line)
sys.stdout.write(line) sys.stdout.write(line)
s0 = time.time() s0 = time.time()
cf = cf_model(ml, 3, i, j, baseQ) cf = cf_model(ml, 3, i, j, baseQ)
s1 = time.time() s1 = time.time()
line = ' model {} run time: {} seconds\n'.format(idx, s1-s0) line = " model {} run time: {} seconds\n".format(idx, s1 - s0)
fs.write(line) fs.write(line)
sys.stdout.write(line) sys.stdout.write(line)
idx += 1 idx += 1
@ -127,14 +155,16 @@ for i in range(0, nrow, nstep):
# end timer for capture fraction analysis # end timer for capture fraction analysis
end = time.time() end = time.time()
ets = end - start ets = end - start
line = '\n' + \ line = (
'streamflow capture analysis took {} seconds.\n'.format(ets) + \ "\n"
'streamflow capture analysis took {} minutes.\n'.format(ets/60.) + \ + "streamflow capture analysis took {} seconds.\n".format(ets)
'streamflow capture analysis took {} hours.\n'.format(ets/3600.) + "streamflow capture analysis took {} minutes.\n".format(ets / 60.0)
+ "streamflow capture analysis took {} hours.\n".format(ets / 3600.0)
)
fs.write(line) fs.write(line)
sys.stdout.write(line) sys.stdout.write(line)
#close summary file # close summary file
fs.close() fs.close()
# clean up working directory # clean up working directory
@ -146,7 +176,9 @@ for f in filelist:
if not os.path.exists(res_pth): if not os.path.exists(res_pth):
os.makedirs(res_pth) os.makedirs(res_pth)
for idx in range(10): for idx in range(10):
fn = os.path.join(res_pth, 'USPB_capture_fraction_{:02d}_{:02d}.dat'.format(nstep, idx+1)) fn = os.path.join(
print('saving capture fraction data to...{}'.format(os.path.basename(fn))) res_pth,
np.savetxt(fn, cf_array[idx, :, :], delimiter=' ') "USPB_capture_fraction_{:02d}_{:02d}.dat".format(nstep, idx + 1),
)
print("saving capture fraction data to...{}".format(os.path.basename(fn)))
np.savetxt(fn, cf_array[idx, :, :], delimiter=" ")

343
examples/groundwater_paper/scripts/uspb_capture_par.py
View File

@ -15,83 +15,102 @@ import multiprocessing as mp
import flopy import flopy
# global executable name and output precision # global executable name and output precision
precision = 'double' precision = "double"
exe_name = 'mf2005dbl' exe_name = "mf2005dbl"
if platform.system() == 'Windows': if platform.system() == "Windows":
exe_name = 'mf2005dbl_x64.exe' exe_name = "mf2005dbl_x64.exe"
# functions that do all of the work # functions that do all of the work
def load_base_model(klay): def load_base_model(klay):
# paths # paths
base_pth = os.path.join('data', 'uspb', 'flopy') base_pth = os.path.join("data", "uspb", "flopy")
sys.stdout.write("loading base model\n")
ml = flopy.modflow.Modflow.load(
"DG.nam",
version="mf2005",
exe_name=exe_name,
verbose=True,
model_ws=base_pth,
)
sys.stdout.write('loading base model\n')
ml = flopy.modflow.Modflow.load('DG.nam', version='mf2005', exe_name=exe_name,
verbose=True, model_ws=base_pth)
# set a few variables from the model # set a few variables from the model
nrow, ncol = ml.dis.nrow, ml.dis.ncol nrow, ncol = ml.dis.nrow, ml.dis.ncol
ibound = ml.bas6.ibound[klay, :, :] ibound = ml.bas6.ibound[klay, :, :]
return ml, nrow, ncol, ibound
def get_baseQ(model):
sys.stdout.write('\nrunning base model to get base head-dependent flow\n\n')
success, report = model.run_model(silent=True, report=True)
sys.stdout.write('Base model run: {}\n'.format(report[-3]))
# get base model results
cbcObj = flopy.utils.CellBudgetFile(os.path.join(model.model_ws, 'DG.cbc'), precision=precision)
v1 = cbcObj.get_data(kstpkper=(0, 0), text='DRAINS', full3D=True)[0]
v2 = cbcObj.get_data(kstpkper=(0, 0), text='STREAM LEAKAGE', full3D=True)[0]
v3 = cbcObj.get_data(kstpkper=(0, 0), text='ET', full3D=True)[0]
return v1.sum() + v2.sum() + v3.sum()
def copy_files(ml, nproc):
# path
cf_base = os.path.join('data', 'uspb')
exclude = ['hds', 'cbc', 'list', 'ddn'] return ml, nrow, ncol, ibound
def get_baseQ(model):
sys.stdout.write(
"\nrunning base model to get base head-dependent flow\n\n"
)
success, report = model.run_model(silent=True, report=True)
sys.stdout.write("Base model run: {}\n".format(report[-3]))
# get base model results
cbcObj = flopy.utils.CellBudgetFile(
os.path.join(model.model_ws, "DG.cbc"), precision=precision
)
v1 = cbcObj.get_data(kstpkper=(0, 0), text="DRAINS", full3D=True)[0]
v2 = cbcObj.get_data(kstpkper=(0, 0), text="STREAM LEAKAGE", full3D=True)[
0
]
v3 = cbcObj.get_data(kstpkper=(0, 0), text="ET", full3D=True)[0]
return v1.sum() + v2.sum() + v3.sum()
def copy_files(ml, nproc):
# path
cf_base = os.path.join("data", "uspb")
exclude = ["hds", "cbc", "list", "ddn"]
cf_pths = [] cf_pths = []
for idx in range(nproc): for idx in range(nproc):
cf_pths.append(os.path.join(cf_base, 'cf{:02d}'.format(idx))) cf_pths.append(os.path.join(cf_base, "cf{:02d}".format(idx)))
# create base model in each directory # create base model in each directory
if idx == 0: if idx == 0:
ml.model_ws = cf_pths[idx] ml.model_ws = cf_pths[idx]
# modify the oc file # modify the oc file
ml.remove_package('OC') ml.remove_package("OC")
stress_period_data = {(1,9): ['save head', 'save budget', 'print budget'], stress_period_data = {
(1,10): [], (1, 9): ["save head", "save budget", "print budget"],
(1,19): ['save head', 'save budget', 'print budget'], (1, 10): [],
(1,20): [], (1, 19): ["save head", "save budget", "print budget"],
(1,29): ['save head', 'save budget', 'print budget'], (1, 20): [],
(1,30): [], (1, 29): ["save head", "save budget", "print budget"],
(1,39): ['save head', 'save budget', 'print budget'], (1, 30): [],
(1,40): [], (1, 39): ["save head", "save budget", "print budget"],
(1,49): ['save head', 'save budget', 'print budget'], (1, 40): [],
(1,50): [], (1, 49): ["save head", "save budget", "print budget"],
(1,59): ['save head', 'save budget', 'print budget'], (1, 50): [],
(1,60): [], (1, 59): ["save head", "save budget", "print budget"],
(1,69): ['save head', 'save budget', 'print budget'], (1, 60): [],
(1,70): [], (1, 69): ["save head", "save budget", "print budget"],
(1,79): ['save head', 'save budget', 'print budget'], (1, 70): [],
(1,80): [], (1, 79): ["save head", "save budget", "print budget"],
(1,89): ['save head', 'save budget', 'print budget'], (1, 80): [],
(1,90): [], (1, 89): ["save head", "save budget", "print budget"],
(1,99): ['save head', 'save budget', 'print budget'], (1, 90): [],
(1,100): []} (1, 99): ["save head", "save budget", "print budget"],
oc = flopy.modflow.ModflowOc(ml, stress_period_data=stress_period_data) (1, 100): [],
}
oc = flopy.modflow.ModflowOc(
ml, stress_period_data=stress_period_data
)
# write the input files # write the input files
ml.write_input() ml.write_input()
else: else:
if not os.path.exists(cf_pths[idx]): if not os.path.exists(cf_pths[idx]):
os.makedirs(cf_pths[idx]) os.makedirs(cf_pths[idx])
filelist = [f for f in os.listdir(cf_pths[0])] filelist = [f for f in os.listdir(cf_pths[0])]
sys.stdout.write('copying files from {} to {}\n'.format(cf_pths[0], cf_pths[idx])) sys.stdout.write(
"copying files from {} to {}\n".format(
cf_pths[0], cf_pths[idx]
)
)
for f in filelist: for f in filelist:
if os.path.splitext(f)[1].lower() in exclude: if os.path.splitext(f)[1].lower() in exclude:
continue continue
@ -100,46 +119,48 @@ def copy_files(ml, nproc):
shutil.copyfile(src, dst) shutil.copyfile(src, dst)
return ml, cf_pths return ml, cf_pths
# functions to run the models in parallel # functions to run the models in parallel
def unpack_args(args): def unpack_args(args):
try: try:
f, idx, nmax, k, i, j, Qt, base, hdry = args f, idx, nmax, k, i, j, Qt, base, hdry = args
except: except:
sys.stdout.write('could not unpack args\n') sys.stdout.write("could not unpack args\n")
raise raise
try: try:
current = mp.current_process() current = mp.current_process()
imod = current._identity[0]-1 imod = current._identity[0] - 1
except: except:
sys.stdout.write('could not get current process\n') sys.stdout.write("could not get current process\n")
raise raise
return f(imod, idx, nmax, k, i, j, Qt, base, hdry) return f(imod, idx, nmax, k, i, j, Qt, base, hdry)
def make_well(pth, k, i, j, Qt): def make_well(pth, k, i, j, Qt):
fn = os.path.join(pth, 'DG.wel') fn = os.path.join(pth, "DG.wel")
f = open(fn, 'w') f = open(fn, "w")
f.write('# Well file for MODFLOW, generated by Flopy.\n') f.write("# Well file for MODFLOW, generated by Flopy.\n")
f.write(' 1 0\n') f.write(" 1 0\n")
f.write(' 0 0 # stress period 0\n') f.write(" 0 0 # stress period 0\n")
f.write(' 1 0 # stress period 1\n') f.write(" 1 0 # stress period 1\n")
f.write('{:10d}{:10d}{:10d}{:10.2f}\n'.format(k+1, i+1, j+1, Qt)) f.write("{:10d}{:10d}{:10d}{:10.2f}\n".format(k + 1, i + 1, j + 1, Qt))
f.close() f.close()
def run_model(pth): def run_model(pth):
proc = sp.Popen([exe_name, 'DG.nam'], proc = sp.Popen([exe_name, "DG.nam"], stdout=sp.PIPE, cwd=pth)
stdout=sp.PIPE, cwd=pth) sys.stdout.write(" running {} in {}\n".format(exe_name, pth))
sys.stdout.write(' running {} in {}\n'.format(exe_name, pth))
success = False success = False
buff = [] buff = []
elt = 'Normal model termination did not occur' elt = "Normal model termination did not occur"
while True: while True:
line = proc.stdout.readline() line = proc.stdout.readline()
c = line.decode('utf-8') c = line.decode("utf-8")
if c != '': if c != "":
if 'normal termination of simulation' in c.lower(): if "normal termination of simulation" in c.lower():
success = True success = True
c = c.rstrip('\r\n') c = c.rstrip("\r\n")
buff.append(c) buff.append(c)
else: else:
break break
@ -148,96 +169,136 @@ def run_model(pth):
elt = buff[-3].strip() elt = buff[-3].strip()
except: except:
pass pass
return ([success, elt]) return [success, elt]
# function to create well file, run model, and extract results # function to create well file, run model, and extract results
def cf_model(imod, ion, nmax, k, i, j, Qt, base, hdry): def cf_model(imod, ion, nmax, k, i, j, Qt, base, hdry):
pth = os.path.join('data', 'uspb', 'cf{:02d}'.format(imod)) pth = os.path.join("data", "uspb", "cf{:02d}".format(imod))
sys.stdout.write('\nRunning model number: {}\n'.format(imod)) sys.stdout.write("\nRunning model number: {}\n".format(imod))
sys.stdout.write(' model run: {} of {}\n'.format(ion+1, nmax)) sys.stdout.write(" model run: {} of {}\n".format(ion + 1, nmax))
sys.stdout.write(' model number {} working directory: {}\n'.format(imod, pth)) sys.stdout.write(
" model number {} working directory: {}\n".format(imod, pth)
)
make_well(pth, k, i, j, Qt) make_well(pth, k, i, j, Qt)
success, elt = run_model(pth) success, elt = run_model(pth)
line = '\nModel run: {} of {} (model number {})\n'.format(ion+1, nmax, imod) line = "\nModel run: {} of {} (model number {})\n".format(
line += ' row {} - col {}\n'.format(i+1, j+1) ion + 1, nmax, imod
line += ' {}\n'.format(elt) )
line += " row {} - col {}\n".format(i + 1, j + 1)
line += " {}\n".format(elt)
# get the results # get the results
v = np.zeros((10), dtype=np.float) v = np.zeros((10), dtype=np.float)
if success: if success:
try: try:
hedObj = flopy.utils.HeadFile(os.path.join(pth, 'DG.hds'), precision=precision) hedObj = flopy.utils.HeadFile(
cbcObj = flopy.utils.CellBudgetFile(os.path.join(pth, 'DG.cbc'), precision=precision) os.path.join(pth, "DG.hds"), precision=precision
)
cbcObj = flopy.utils.CellBudgetFile(
os.path.join(pth, "DG.cbc"), precision=precision
)
kk = hedObj.get_kstpkper() kk = hedObj.get_kstpkper()
h = hedObj.get_ts((k, i, j)) h = hedObj.get_ts((k, i, j))
for idx, kon in enumerate(kk): for idx, kon in enumerate(kk):
if h[idx, 1] == hdry: if h[idx, 1] == hdry:
v[idx] = np.nan v[idx] = np.nan
else: else:
v1 = cbcObj.get_data(kstpkper=kon, text='DRAINS', full3D=True)[0] v1 = cbcObj.get_data(
v2 = cbcObj.get_data(kstpkper=kon, text='STREAM LEAKAGE', full3D=True)[0] kstpkper=kon, text="DRAINS", full3D=True
v3 = cbcObj.get_data(kstpkper=kon, text='ET', full3D=True)[0] )[0]
v2 = cbcObj.get_data(
kstpkper=kon, text="STREAM LEAKAGE", full3D=True
)[0]
v3 = cbcObj.get_data(kstpkper=kon, text="ET", full3D=True)[
0
]
v[idx] = ((v1.sum() + v2.sum() + v3.sum()) - base) / (-Qt) v[idx] = ((v1.sum() + v2.sum() + v3.sum()) - base) / (-Qt)
except: except:
line += ' Error: Model run: {} of {} (model number {}) - '.format(ion+1, nmax, imod) line += " Error: Model run: {} of {} (model number {}) - ".format(
line += 'could not process model results.\n' ion + 1, nmax, imod
)
line += "could not process model results.\n"
v[:] = np.nan v[:] = np.nan
else: else:
line += ' Error: Model run: {} of {} (model number {}) '.format(ion+1, nmax, imod) line += " Error: Model run: {} of {} (model number {}) ".format(
line += 'did not execute successfully\n' ion + 1, nmax, imod
)
line += "did not execute successfully\n"
v[:] = np.nan v[:] = np.nan
sys.stdout.write(line) sys.stdout.write(line)
return (v, line) return (v, line)
def doit(): def doit():
# multi processing information # multi processing information
nproc = 3 nproc = 3
ncores = mp.cpu_count() ncores = mp.cpu_count()
if nproc > ncores: if nproc > ncores:
sys.stdout.write('Requested {} cores but only {} cores are available.\n\n\n'.format(nproc, ncores)) sys.stdout.write(
"Requested {} cores but only {} cores are available.\n\n\n".format(
nproc, ncores
)
)
else: else:
sys.stdout.write('Requested {} cores and {} cores are available.\n\n\n'.format(nproc, ncores)) sys.stdout.write(
"Requested {} cores and {} cores are available.\n\n\n".format(
#paths nproc, ncores
res_pth = os.path.join('data', 'uspb', 'results') )
)
#model data
# paths
res_pth = os.path.join("data", "uspb", "results")
# model data
klay = 3 klay = 3
Qcf = -100. Qcf = -100.0
nstep = 4 nstep = 4
# load base model # load base model
ml, nrow, ncol, ibound = load_base_model(klay) ml, nrow, ncol, ibound = load_base_model(klay)
# run first model created to get base model results # run first model created to get base model results
baseQ = get_baseQ(ml) baseQ = get_baseQ(ml)
sys.stdout.write('Base head-dependent flux = {}'.format(baseQ)) sys.stdout.write("Base head-dependent flux = {}".format(baseQ))
# modify oc file copy model files # modify oc file copy model files
ml, cf_pths = copy_files(ml, nproc) ml, cf_pths = copy_files(ml, nproc)
# calculate subset of model to run # calculate subset of model to run
nrow2 = nrow // nstep nrow2 = nrow // nstep
ncol2 = ncol // nstep ncol2 = ncol // nstep
# open summary file # open summary file
fs = open(os.path.join('data', 'uspb', 'uspb_capture_{}.out'.format(nstep)), 'w', 0) fs = open(
os.path.join("data", "uspb", "uspb_capture_{}.out".format(nstep)),
"w",
0,
)
# write some summary information # write some summary information
fs.write('Problem size: {} rows and {} columns.\n'.format(nrow, ncol)) fs.write("Problem size: {} rows and {} columns.\n".format(nrow, ncol))
fs.write('Capture fraction analysis performed every {} rows and columns.\n'.format(nstep)) fs.write(
fs.write('Maximum number of analyses: {} rows and {} columns.\n'.format(nrow2, ncol2)) "Capture fraction analysis performed every {} rows and columns.\n".format(
nstep
)
)
fs.write(
"Maximum number of analyses: {} rows and {} columns.\n".format(
nrow2, ncol2
)
)
# create array to store capture fraction data (subset of model) # create array to store capture fraction data (subset of model)
cf_array = np.empty((10, nrow2, ncol2), dtype=np.float) cf_array = np.empty((10, nrow2, ncol2), dtype=np.float)
cf_array.fill(np.nan) cf_array.fill(np.nan)
# timer for capture fraction analysis # timer for capture fraction analysis
start = time.time() start = time.time()
# capture fraction analysis # capture fraction analysis
icnt = 0 icnt = 0
jcnt = 0 jcnt = 0
# build tuple with list of cells # build tuple with list of cells
cells = [] cells = []
cellmap = [] cellmap = []
@ -252,57 +313,67 @@ def doit():
jcnt += 1 jcnt += 1
# increment icnt # increment icnt
icnt += 1 icnt += 1
## test cg_model function ## test cg_model function
#t = cf_model(models[0], klay, cells[0][0], cells[0][1], Qcf, baseQ) # t = cf_model(models[0], klay, cells[0][0], cells[0][1], Qcf, baseQ)
#sys.stdout.write(t) # sys.stdout.write(t)
# create multiprocessing pool # create multiprocessing pool
pool = mp.Pool(processes=nproc) pool = mp.Pool(processes=nproc)
args = [(cf_model, idx, len(cells), klay, i, j, Qcf, baseQ, ml.lpf.hdry) for idx, (i, j) in enumerate(cells)] args = [
#sys.stdout.write(args) (cf_model, idx, len(cells), klay, i, j, Qcf, baseQ, ml.lpf.hdry)
for idx, (i, j) in enumerate(cells)
]
# sys.stdout.write(args)
output = pool.map(unpack_args, args, nproc) output = pool.map(unpack_args, args, nproc)
pool.close() pool.close()
pool.join() pool.join()
for v in output: for v in output:
fs.write(v[1]) fs.write(v[1])
for idx, (icnt, jcnt) in enumerate(cellmap): for idx, (icnt, jcnt) in enumerate(cellmap):
# add values to the array # add values to the array
if icnt < nrow2 and jcnt < ncol2: if icnt < nrow2 and jcnt < ncol2:
cf_array[:, icnt, jcnt] = output[idx][0].copy() cf_array[:, icnt, jcnt] = output[idx][0].copy()
# end timer for capture fraction analysis # end timer for capture fraction analysis
end = time.time() end = time.time()
ets = end - start ets = end - start
line = '\n' + \ line = (
'streamflow capture analysis took {} seconds.\n'.format(ets) + \ "\n"
'streamflow capture analysis took {} minutes.\n'.format(ets/60.) + \ + "streamflow capture analysis took {} seconds.\n".format(ets)
'streamflow capture analysis took {} hours.\n'.format(ets/3600.) + "streamflow capture analysis took {} minutes.\n".format(ets / 60.0)
+ "streamflow capture analysis took {} hours.\n".format(ets / 3600.0)
)
fs.write(line) fs.write(line)
sys.stdout.write(line) sys.stdout.write(line)
#close summary file # close summary file
fs.close() fs.close()
# clean up working directories # clean up working directories
for idx in range(nproc): for idx in range(nproc):
filelist = [f for f in os.listdir(cf_pths[idx])] filelist = [f for f in os.listdir(cf_pths[idx])]
for f in filelist: for f in filelist:
os.remove(os.path.join(cf_pths[idx], f)) os.remove(os.path.join(cf_pths[idx], f))
# create res_pth (if it doesn't exist) and save data # create res_pth (if it doesn't exist) and save data
if not os.path.exists(res_pth): if not os.path.exists(res_pth):
os.makedirs(res_pth) os.makedirs(res_pth)
for idx in range(10): for idx in range(10):
fn = os.path.join(res_pth, 'USPB_capture_fraction_{:02d}_{:02d}.dat'.format(nstep, idx+1)) fn = os.path.join(
sys.stdout.write('saving capture fraction data to...{}\n'.format(os.path.basename(fn))) res_pth,
np.savetxt(fn, cf_array[idx, :, :], delimiter=' ') "USPB_capture_fraction_{:02d}_{:02d}.dat".format(nstep, idx + 1),
)
sys.stdout.write(
"saving capture fraction data to...{}\n".format(
os.path.basename(fn)
)
)
np.savetxt(fn, cf_array[idx, :, :], delimiter=" ")
if __name__ == "__main__":
if __name__ == '__main__':
doit() doit()

143
examples/scripts/flopy_henry.py
View File

@ -6,44 +6,55 @@ import flopy
def run(): def run():
workspace = os.path.join('henry') workspace = os.path.join("henry")
# make sure workspace directory exists # make sure workspace directory exists
if not os.path.exists(workspace): if not os.path.exists(workspace):
os.makedirs(workspace) os.makedirs(workspace)
fext = 'png' fext = "png"
narg = len(sys.argv) narg = len(sys.argv)
iarg = 0 iarg = 0
if narg > 1: if narg > 1:
while iarg < narg - 1: while iarg < narg - 1:
iarg += 1 iarg += 1
basearg = sys.argv[iarg].lower() basearg = sys.argv[iarg].lower()
if basearg == '--pdf': if basearg == "--pdf":
fext = 'pdf' fext = "pdf"
# Input variables for the Henry Problem # Input variables for the Henry Problem
Lx = 2. Lx = 2.0
Lz = 1. Lz = 1.0
nlay = 50 nlay = 50
nrow = 1 nrow = 1
ncol = 100 ncol = 100
delr = Lx / ncol delr = Lx / ncol
delc = 1.0 delc = 1.0
delv = Lz / nlay delv = Lz / nlay
henry_top = 1. henry_top = 1.0
henry_botm = np.linspace(henry_top - delv, 0., nlay) henry_botm = np.linspace(henry_top - delv, 0.0, nlay)
qinflow = 5.702 # m3/day qinflow = 5.702 # m3/day
dmcoef = 0.57024 # m2/day Could also try 1.62925 as another case of the Henry problem dmcoef = 0.57024 # m2/day Could also try 1.62925 as another case of the Henry problem
hk = 864. # m/day hk = 864.0 # m/day
# Create the basic MODFLOW model data # Create the basic MODFLOW model data
modelname = 'henry' modelname = "henry"
m = flopy.seawat.Seawat(modelname, exe_name="swtv4", model_ws=workspace) m = flopy.seawat.Seawat(modelname, exe_name="swtv4", model_ws=workspace)
# Add DIS package to the MODFLOW model # Add DIS package to the MODFLOW model
dis = flopy.modflow.ModflowDis(m, nlay, nrow, ncol, nper=1, delr=delr, dis = flopy.modflow.ModflowDis(
delc=delc, laycbd=0, top=henry_top, m,
botm=henry_botm, perlen=1.5, nstp=15) nlay,
nrow,
ncol,
nper=1,
delr=delr,
delc=delc,
laycbd=0,
top=henry_top,
botm=henry_botm,
perlen=1.5,
nstp=15,
)
# Variables for the BAS package # Variables for the BAS package
ibound = np.ones((nlay, nrow, ncol), dtype=np.int32) ibound = np.ones((nlay, nrow, ncol), dtype=np.int32)
@ -54,13 +65,14 @@ def run():
lpf = flopy.modflow.ModflowLpf(m, hk=hk, vka=hk, ipakcb=53) lpf = flopy.modflow.ModflowLpf(m, hk=hk, vka=hk, ipakcb=53)
# Add PCG Package to the MODFLOW model # Add PCG Package to the MODFLOW model
pcg = flopy.modflow.ModflowPcg(m, hclose=1.e-8) pcg = flopy.modflow.ModflowPcg(m, hclose=1.0e-8)
# Add OC package to the MODFLOW model # Add OC package to the MODFLOW model
oc = flopy.modflow.ModflowOc(m, oc = flopy.modflow.ModflowOc(
stress_period_data={ m,
(0, 0): ['save head', 'save budget']}, stress_period_data={(0, 0): ["save head", "save budget"]},
compact=True) compact=True,
)
# Create WEL and SSM data # Create WEL and SSM data
itype = flopy.mt3d.Mt3dSsm.itype_dict() itype = flopy.mt3d.Mt3dSsm.itype_dict()
@ -70,32 +82,48 @@ def run():
ssm_sp1 = [] ssm_sp1 = []
for k in range(nlay): for k in range(nlay):
wel_sp1.append([k, 0, 0, qinflow / nlay]) wel_sp1.append([k, 0, 0, qinflow / nlay])
ssm_sp1.append([k, 0, 0, 0., itype['WEL']]) ssm_sp1.append([k, 0, 0, 0.0, itype["WEL"]])
ssm_sp1.append([k, 0, ncol - 1, 35., itype['BAS6']]) ssm_sp1.append([k, 0, ncol - 1, 35.0, itype["BAS6"]])
wel_data[0] = wel_sp1 wel_data[0] = wel_sp1
ssm_data[0] = ssm_sp1 ssm_data[0] = ssm_sp1
wel = flopy.modflow.ModflowWel(m, stress_period_data=wel_data) wel = flopy.modflow.ModflowWel(m, stress_period_data=wel_data)
# Create the basic MT3DMS model data # Create the basic MT3DMS model data
btn = flopy.mt3d.Mt3dBtn(m, nprs=-5, prsity=0.35, sconc=35., ifmtcn=0, btn = flopy.mt3d.Mt3dBtn(
chkmas=False, nprobs=10, nprmas=10, dt0=0.001) m,
nprs=-5,
prsity=0.35,
sconc=35.0,
ifmtcn=0,
chkmas=False,
nprobs=10,
nprmas=10,
dt0=0.001,
)
adv = flopy.mt3d.Mt3dAdv(m, mixelm=0) adv = flopy.mt3d.Mt3dAdv(m, mixelm=0)
dsp = flopy.mt3d.Mt3dDsp(m, al=0., trpt=1., trpv=1., dmcoef=dmcoef) dsp = flopy.mt3d.Mt3dDsp(m, al=0.0, trpt=1.0, trpv=1.0, dmcoef=dmcoef)
gcg = flopy.mt3d.Mt3dGcg(m, iter1=500, mxiter=1, isolve=1, cclose=1e-7) gcg = flopy.mt3d.Mt3dGcg(m, iter1=500, mxiter=1, isolve=1, cclose=1e-7)
ssm = flopy.mt3d.Mt3dSsm(m, stress_period_data=ssm_data) ssm = flopy.mt3d.Mt3dSsm(m, stress_period_data=ssm_data)
# Create the SEAWAT model data # Create the SEAWAT model data
vdf = flopy.seawat.SeawatVdf(m, iwtable=0, densemin=0, densemax=0, vdf = flopy.seawat.SeawatVdf(
denseref=1000., denseslp=0.7143, firstdt=1e-3) m,
iwtable=0,
densemin=0,
densemax=0,
denseref=1000.0,
denseslp=0.7143,
firstdt=1e-3,
)
# Write the input files # Write the input files
m.write_input() m.write_input()
# Try to delete the output files, to prevent accidental use of older files # Try to delete the output files, to prevent accidental use of older files
try: try:
os.remove(os.path.join(workspace, 'MT3D001.UCN')) os.remove(os.path.join(workspace, "MT3D001.UCN"))
os.remove(os.path.join(workspace, modelname + '.hds')) os.remove(os.path.join(workspace, modelname + ".hds"))
os.remove(os.path.join(workspace, modelname + '.cbc')) os.remove(os.path.join(workspace, modelname + ".cbc"))
except: except:
pass pass
@ -106,61 +134,72 @@ def run():
# Load data # Load data
ucnobj = flopy.utils.binaryfile.UcnFile( ucnobj = flopy.utils.binaryfile.UcnFile(
os.path.join(workspace, 'MT3D001.UCN'), os.path.join(workspace, "MT3D001.UCN"), model=m
model=m) )
times = ucnobj.get_times() times = ucnobj.get_times()
concentration = ucnobj.get_data(totim=times[-1]) concentration = ucnobj.get_data(totim=times[-1])
cbbobj = flopy.utils.binaryfile.CellBudgetFile( cbbobj = flopy.utils.binaryfile.CellBudgetFile(
os.path.join(workspace, 'henry.cbc')) os.path.join(workspace, "henry.cbc")
)
times = cbbobj.get_times() times = cbbobj.get_times()
qx = cbbobj.get_data(text='flow right face', totim=times[-1])[0] qx = cbbobj.get_data(text="flow right face", totim=times[-1])[0]
qz = cbbobj.get_data(text='flow lower face', totim=times[-1])[0] qz = cbbobj.get_data(text="flow lower face", totim=times[-1])[0]
# Average flows to cell centers # Average flows to cell centers
qx_avg = np.empty(qx.shape, dtype=qx.dtype) qx_avg = np.empty(qx.shape, dtype=qx.dtype)
qx_avg[:, :, 1:] = 0.5 * (qx[:, :, 0:ncol - 1] + qx[:, :, 1:ncol]) qx_avg[:, :, 1:] = 0.5 * (qx[:, :, 0 : ncol - 1] + qx[:, :, 1:ncol])
qx_avg[:, :, 0] = 0.5 * qx[:, :, 0] qx_avg[:, :, 0] = 0.5 * qx[:, :, 0]
qz_avg = np.empty(qz.shape, dtype=qz.dtype) qz_avg = np.empty(qz.shape, dtype=qz.dtype)
qz_avg[1:, :, :] = 0.5 * (qz[0:nlay - 1, :, :] + qz[1:nlay, :, :]) qz_avg[1:, :, :] = 0.5 * (qz[0 : nlay - 1, :, :] + qz[1:nlay, :, :])
qz_avg[0, :, :] = 0.5 * qz[0, :, :] qz_avg[0, :, :] = 0.5 * qz[0, :, :]
# Make the plot # Make the plot
# import matplotlib.pyplot as plt # import matplotlib.pyplot as plt
fig = plt.figure(figsize=(10, 10)) fig = plt.figure(figsize=(10, 10))
ax = fig.add_subplot(1, 1, 1, aspect='equal') ax = fig.add_subplot(1, 1, 1, aspect="equal")
ax.imshow(concentration[:, 0, :], interpolation='nearest', ax.imshow(
extent=(0, Lx, 0, Lz)) concentration[:, 0, :], interpolation="nearest", extent=(0, Lx, 0, Lz)
)
y, x, z = dis.get_node_coordinates() y, x, z = dis.get_node_coordinates()
X, Z = np.meshgrid(x, z[:, 0, 0]) X, Z = np.meshgrid(x, z[:, 0, 0])
iskip = 3 iskip = 3
ax.quiver(X[::iskip, ::iskip], Z[::iskip, ::iskip], ax.quiver(
qx_avg[::iskip, 0, ::iskip], -qz_avg[::iskip, 0, ::iskip], X[::iskip, ::iskip],
color='w', scale=5, headwidth=3, headlength=2, Z[::iskip, ::iskip],
headaxislength=2, width=0.0025) qx_avg[::iskip, 0, ::iskip],
-qz_avg[::iskip, 0, ::iskip],
color="w",
scale=5,
headwidth=3,
headlength=2,
headaxislength=2,
width=0.0025,
)
outfig = os.path.join(workspace, 'henry_flows.{0}'.format(fext)) outfig = os.path.join(workspace, "henry_flows.{0}".format(fext))
fig.savefig(outfig, dpi=300) fig.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
# Extract the heads # Extract the heads
fname = os.path.join(workspace, 'henry.hds') fname = os.path.join(workspace, "henry.hds")
headobj = flopy.utils.binaryfile.HeadFile(fname) headobj = flopy.utils.binaryfile.HeadFile(fname)
times = headobj.get_times() times = headobj.get_times()
head = headobj.get_data(totim=times[-1]) head = headobj.get_data(totim=times[-1])
# Make a simple head plot # Make a simple head plot
fig = plt.figure(figsize=(10, 10)) fig = plt.figure(figsize=(10, 10))
ax = fig.add_subplot(1, 1, 1, aspect='equal') ax = fig.add_subplot(1, 1, 1, aspect="equal")
im = ax.imshow(head[:, 0, :], interpolation='nearest', im = ax.imshow(
extent=(0, Lx, 0, Lz)) head[:, 0, :], interpolation="nearest", extent=(0, Lx, 0, Lz)
ax.set_title('Simulated Heads') )
ax.set_title("Simulated Heads")
outfig = os.path.join(workspace, 'henry_heads.{0}'.format(fext)) outfig = os.path.join(workspace, "henry_heads.{0}".format(fext))
fig.savefig(outfig, dpi=300) fig.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
return 0 return 0
if __name__ == '__main__': if __name__ == "__main__":
success = run() success = run()

62
examples/scripts/flopy_lake_example.py
View File

@ -7,20 +7,20 @@ import flopy
def run(): def run():
workspace = os.path.join('lake') workspace = os.path.join("lake")
# make sure workspace directory exists # make sure workspace directory exists
if not os.path.exists(workspace): if not os.path.exists(workspace):
os.makedirs(workspace) os.makedirs(workspace)
fext = 'png' fext = "png"
narg = len(sys.argv) narg = len(sys.argv)
iarg = 0 iarg = 0
if narg > 1: if narg > 1:
while iarg < narg - 1: while iarg < narg - 1:
iarg += 1 iarg += 1
basearg = sys.argv[iarg].lower() basearg = sys.argv[iarg].lower()
if basearg == '--pdf': if basearg == "--pdf":
fext = 'pdf' fext = "pdf"
# save the starting path # save the starting path
cwdpth = os.getcwd() cwdpth = os.getcwd()
@ -33,7 +33,7 @@ def run():
# of the model and the parameters of the model: the number of layers `Nlay`, the number of rows # of the model and the parameters of the model: the number of layers `Nlay`, the number of rows
# and columns `N`, lengths of the sides of the model `L`, aquifer thickness `H`, hydraulic # and columns `N`, lengths of the sides of the model `L`, aquifer thickness `H`, hydraulic
# conductivity `Kh` # conductivity `Kh`
name = 'lake_example' name = "lake_example"
h1 = 100 h1 = 100
h2 = 90 h2 = 90
Nlay = 10 Nlay = 10
@ -46,8 +46,9 @@ def run():
# whatever you want). The modelname will be the name given to all MODFLOW files (input and output). # whatever you want). The modelname will be the name given to all MODFLOW files (input and output).
# The exe_name should be the full path to your MODFLOW executable. The version is either 'mf2k' # The exe_name should be the full path to your MODFLOW executable. The version is either 'mf2k'
# for MODFLOW2000 or 'mf2005'for MODFLOW2005. # for MODFLOW2000 or 'mf2005'for MODFLOW2005.
ml = flopy.modflow.Modflow(modelname=name, exe_name='mf2005', ml = flopy.modflow.Modflow(
version='mf2005') modelname=name, exe_name="mf2005", version="mf2005"
)
# Define the discretization of the model. All layers are given equal thickness. The `bot` array # Define the discretization of the model. All layers are given equal thickness. The `bot` array
# is build from the `Hlay` values to indicate top and bottom of each layer, and `delrow` and # is build from the `Hlay` values to indicate top and bottom of each layer, and `delrow` and
@ -55,8 +56,17 @@ def run():
# the Discretization file is built. # the Discretization file is built.
bot = np.linspace(-H / Nlay, -H, Nlay) bot = np.linspace(-H / Nlay, -H, Nlay)
delrow = delcol = L / (N - 1) delrow = delcol = L / (N - 1)
dis = flopy.modflow.ModflowDis(ml, nlay=Nlay, nrow=N, ncol=N, delr=delrow, dis = flopy.modflow.ModflowDis(
delc=delcol, top=0.0, botm=bot, laycbd=0) ml,
nlay=Nlay,
nrow=N,
ncol=N,
delr=delrow,
delc=delcol,
top=0.0,
botm=bot,
laycbd=0,
)
# Next we specify the boundary conditions and starting heads with the Basic package. The `ibound` # Next we specify the boundary conditions and starting heads with the Basic package. The `ibound`
# array will be `1` in all cells in all layers, except for along the boundary and in the cell at # array will be `1` in all cells in all layers, except for along the boundary and in the cell at
@ -77,14 +87,14 @@ def run():
# create external ibound array and starting head files # create external ibound array and starting head files
files = [] files = []
hfile = '{}_strt.ref'.format(name) hfile = "{}_strt.ref".format(name)
np.savetxt(hfile, start) np.savetxt(hfile, start)
hfiles = [] hfiles = []
for kdx in range(Nlay): for kdx in range(Nlay):
file = '{}_ib{:02d}.ref'.format(name, kdx + 1) file = "{}_ib{:02d}.ref".format(name, kdx + 1)
files.append(file) files.append(file)
hfiles.append(hfile) hfiles.append(hfile)
np.savetxt(file, ibound[kdx, :, :], fmt='%5d') np.savetxt(file, ibound[kdx, :, :], fmt="%5d")
bas = flopy.modflow.ModflowBas(ml, ibound=files, strt=hfiles) bas = flopy.modflow.ModflowBas(ml, ibound=files, strt=hfiles)
@ -108,39 +118,39 @@ def run():
# specifying, in this case, the step number and period number for which we want to retrieve data. # specifying, in this case, the step number and period number for which we want to retrieve data.
# A three-dimensional array is returned of size `nlay, nrow, ncol`. Matplotlib contouring functions # A three-dimensional array is returned of size `nlay, nrow, ncol`. Matplotlib contouring functions
# are used to make contours of the layers or a cross-section. # are used to make contours of the layers or a cross-section.
hds = flopy.utils.HeadFile(os.path.join(workspace, name + '.hds')) hds = flopy.utils.HeadFile(os.path.join(workspace, name + ".hds"))
h = hds.get_data(kstpkper=(0, 0)) h = hds.get_data(kstpkper=(0, 0))
x = y = np.linspace(0, L, N) x = y = np.linspace(0, L, N)
c = plt.contour(x, y, h[0], np.arange(90, 100.1, 0.2)) c = plt.contour(x, y, h[0], np.arange(90, 100.1, 0.2))
plt.clabel(c, fmt='%2.1f') plt.clabel(c, fmt="%2.1f")
plt.axis('scaled') plt.axis("scaled")
outfig = os.path.join(workspace, 'lake1.{0}'.format(fext)) outfig = os.path.join(workspace, "lake1.{0}".format(fext))
fig = plt.gcf() fig = plt.gcf()
fig.savefig(outfig, dpi=300) fig.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
x = y = np.linspace(0, L, N) x = y = np.linspace(0, L, N)
c = plt.contour(x, y, h[-1], np.arange(90, 100.1, 0.2)) c = plt.contour(x, y, h[-1], np.arange(90, 100.1, 0.2))
plt.clabel(c, fmt='%1.1f') plt.clabel(c, fmt="%1.1f")
plt.axis('scaled') plt.axis("scaled")
outfig = os.path.join(workspace, 'lake2.{0}'.format(fext)) outfig = os.path.join(workspace, "lake2.{0}".format(fext))
fig = plt.gcf() fig = plt.gcf()
fig.savefig(outfig, dpi=300) fig.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
z = np.linspace(-H / Nlay / 2, -H + H / Nlay / 2, Nlay) z = np.linspace(-H / Nlay / 2, -H + H / Nlay / 2, Nlay)
c = plt.contour(x, z, h[:, 50, :], np.arange(90, 100.1, .2)) c = plt.contour(x, z, h[:, 50, :], np.arange(90, 100.1, 0.2))
plt.axis('scaled') plt.axis("scaled")
outfig = os.path.join(workspace, 'lake3.{0}'.format(fext)) outfig = os.path.join(workspace, "lake3.{0}".format(fext))
fig = plt.gcf() fig = plt.gcf()
fig.savefig(outfig, dpi=300) fig.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
return 0 return 0
if __name__ == '__main__': if __name__ == "__main__":
success = run() success = run()

213
examples/scripts/flopy_swi2_ex1.py
View File

@ -11,54 +11,56 @@ import flopy
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
# --modify default matplotlib settings # --modify default matplotlib settings
updates = {'font.family': ['Univers 57 Condensed', 'Arial'], updates = {
'mathtext.default': 'regular', "font.family": ["Univers 57 Condensed", "Arial"],
'pdf.compression': 0, "mathtext.default": "regular",
'pdf.fonttype': 42, "pdf.compression": 0,
'legend.fontsize': 7, "pdf.fonttype": 42,
'axes.labelsize': 8, "legend.fontsize": 7,
'xtick.labelsize': 7, "axes.labelsize": 8,
'ytick.labelsize': 7} "xtick.labelsize": 7,
"ytick.labelsize": 7,
}
plt.rcParams.update(updates) plt.rcParams.update(updates)
def run(): def run():
workspace = 'swiex1' workspace = "swiex1"
cleanFiles = False cleanFiles = False
fext = 'png' fext = "png"
narg = len(sys.argv) narg = len(sys.argv)
iarg = 0 iarg = 0
if narg > 1: if narg > 1:
while iarg < narg - 1: while iarg < narg - 1:
iarg += 1 iarg += 1
basearg = sys.argv[iarg].lower() basearg = sys.argv[iarg].lower()
if basearg == '--clean': if basearg == "--clean":
cleanFiles = True cleanFiles = True
elif basearg == '--pdf': elif basearg == "--pdf":
fext = 'pdf' fext = "pdf"
if cleanFiles: if cleanFiles:
print('cleaning all files') print("cleaning all files")
print('excluding *.py files') print("excluding *.py files")
files = os.listdir(workspace) files = os.listdir(workspace)
for f in files: for f in files:
if os.path.isdir(f): if os.path.isdir(f):
continue continue
if '.py' != os.path.splitext(f)[1].lower(): if ".py" != os.path.splitext(f)[1].lower():
print(' removing...{}'.format(os.path.basename(f))) print(" removing...{}".format(os.path.basename(f)))
os.remove(os.path.join(workspace, f)) os.remove(os.path.join(workspace, f))
return 1 return 1
modelname = 'swiex1' modelname = "swiex1"
exe_name = 'mf2005' exe_name = "mf2005"
nlay = 1 nlay = 1
nrow = 1 nrow = 1
ncol = 50 ncol = 50
delr = 5. delr = 5.0
delc = 1. delc = 1.0
ibound = np.ones((nrow, ncol), np.int) ibound = np.ones((nrow, ncol), np.int)
ibound[0, -1] = -1 ibound[0, -1] = -1
@ -75,26 +77,45 @@ def run():
ocdict = {} ocdict = {}
for idx in range(49, 200, 50): for idx in range(49, 200, 50):
key = (0, idx) key = (0, idx)
ocdict[key] = ['save head', 'save budget'] ocdict[key] = ["save head", "save budget"]
key = (0, idx + 1) key = (0, idx + 1)
ocdict[key] = [] ocdict[key] = []
# create flopy modflow object # create flopy modflow object
ml = flopy.modflow.Modflow(modelname, version='mf2005', exe_name=exe_name, ml = flopy.modflow.Modflow(
model_ws=workspace) modelname, version="mf2005", exe_name=exe_name, model_ws=workspace
)
# create flopy modflow package objects # create flopy modflow package objects
discret = flopy.modflow.ModflowDis(ml, nlay=nlay, nrow=nrow, ncol=ncol, discret = flopy.modflow.ModflowDis(
delr=delr, delc=delc, ml,
top=0, botm=[-40.0], nlay=nlay,
perlen=400, nstp=200) nrow=nrow,
ncol=ncol,
delr=delr,
delc=delc,
top=0,
botm=[-40.0],
perlen=400,
nstp=200,
)
bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=0.0) bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=0.0)
lpf = flopy.modflow.ModflowLpf(ml, hk=2., vka=2.0, vkcb=0, laytyp=0, lpf = flopy.modflow.ModflowLpf(
layavg=0) ml, hk=2.0, vka=2.0, vkcb=0, laytyp=0, layavg=0
)
wel = flopy.modflow.ModflowWel(ml, stress_period_data={0: [(0, 0, 0, 1)]}) wel = flopy.modflow.ModflowWel(ml, stress_period_data={0: [(0, 0, 0, 1)]})
swi = flopy.modflow.ModflowSwi2(ml, iswizt=55, npln=1, istrat=1, swi = flopy.modflow.ModflowSwi2(
toeslope=0.2, tipslope=0.2, nu=[0, 0.025], ml,
zeta=z, ssz=0.2, isource=isource, iswizt=55,
nsolver=1) npln=1,
istrat=1,
toeslope=0.2,
tipslope=0.2,
nu=[0, 0.025],
zeta=z,
ssz=0.2,
isource=isource,
nsolver=1,
)
oc = flopy.modflow.ModflowOc(ml, stress_period_data=ocdict) oc = flopy.modflow.ModflowOc(ml, stress_period_data=ocdict)
pcg = flopy.modflow.ModflowPcg(ml) pcg = flopy.modflow.ModflowPcg(ml)
# create model files # create model files
@ -102,17 +123,17 @@ def run():
# run the model # run the model
m = ml.run_model(silent=False) m = ml.run_model(silent=False)
# read model heads # read model heads
headfile = os.path.join(workspace, '{}.hds'.format(modelname)) headfile = os.path.join(workspace, "{}.hds".format(modelname))
hdobj = flopy.utils.HeadFile(headfile) hdobj = flopy.utils.HeadFile(headfile)
head = hdobj.get_alldata() head = hdobj.get_alldata()
head = np.array(head) head = np.array(head)
# read model zeta # read model zeta
zetafile = os.path.join(workspace, '{}.zta'.format(modelname)) zetafile = os.path.join(workspace, "{}.zta".format(modelname))
zobj = flopy.utils.CellBudgetFile(zetafile) zobj = flopy.utils.CellBudgetFile(zetafile)
zkstpkper = zobj.get_kstpkper() zkstpkper = zobj.get_kstpkper()
zeta = [] zeta = []
for kk in zkstpkper: for kk in zkstpkper:
zeta.append(zobj.get_data(kstpkper=kk, text=' ZETASRF 1')[0]) zeta.append(zobj.get_data(kstpkper=kk, text=" ZETASRF 1")[0])
zeta = np.array(zeta) zeta = np.array(zeta)
x = np.arange(0.5 * delr, ncol * delr, delr) x = np.arange(0.5 * delr, ncol * delr, delr)
@ -134,70 +155,116 @@ def run():
fbot = 0.125 fbot = 0.125
ftop = 0.925 ftop = 0.925
fig = plt.figure(figsize=(fwid, fhgt), facecolor='w') fig = plt.figure(figsize=(fwid, fhgt), facecolor="w")
fig.subplots_adjust(wspace=0.25, hspace=0.25, left=flft, right=frgt, fig.subplots_adjust(
bottom=fbot, top=ftop) wspace=0.25, hspace=0.25, left=flft, right=frgt, bottom=fbot, top=ftop
)
ax = fig.add_subplot(211) ax = fig.add_subplot(211)
ax.text(-0.075, 1.05, 'A', transform=ax.transAxes, va='center', ax.text(
ha='center', -0.075,
size='8') 1.05,
ax.plot([80, 120], [0, -40], 'k') "A",
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
ax.plot([80, 120], [0, -40], "k")
ax.set_xlim(0, 250) ax.set_xlim(0, 250)
ax.set_ylim(-40, 0) ax.set_ylim(-40, 0)
ax.set_yticks(np.arange(-40, 1, 10)) ax.set_yticks(np.arange(-40, 1, 10))
ax.text(50, -10, 'salt') ax.text(50, -10, "salt")
ax.text(130, -10, 'fresh') ax.text(130, -10, "fresh")
a = ax.annotate("", xy=(50, -25), xytext=(30, -25), a = ax.annotate(
arrowprops=dict(arrowstyle='->', fc='k')) "",
ax.text(40, -22, 'groundwater flow velocity=0.125 m/d', ha='center', xy=(50, -25),
size=7) xytext=(30, -25),
ax.set_ylabel('Elevation, in meters') arrowprops=dict(arrowstyle="->", fc="k"),
)
ax.text(
40, -22, "groundwater flow velocity=0.125 m/d", ha="center", size=7
)
ax.set_ylabel("Elevation, in meters")
ax = fig.add_subplot(212) ax = fig.add_subplot(212)
ax.text(-0.075, 1.05, 'B', transform=ax.transAxes, va='center', ax.text(
ha='center', -0.075,
size='8') 1.05,
"B",
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
for i in range(4): for i in range(4):
Ltoe[i] = H * math.sqrt(k * nu * (t[i] + tzero) / n / H) Ltoe[i] = H * math.sqrt(k * nu * (t[i] + tzero) / n / H)
ax.plot([100 - Ltoe[i] + v * t[i], 100 + Ltoe[i] + v * t[i]], [0, -40], ax.plot(
'k', label='_None') [100 - Ltoe[i] + v * t[i], 100 + Ltoe[i] + v * t[i]],
[0, -40],
"k",
label="_None",
)
for i in range(4): for i in range(4):
zi = zeta[i, 0, 0, :] zi = zeta[i, 0, 0, :]
p = (zi < 0) & (zi > -39.9) p = (zi < 0) & (zi > -39.9)
ax.plot(x[p], zeta[i, 0, 0, p], 'bo', ax.plot(
markersize=3, markeredgecolor='blue', markerfacecolor='None', x[p],
label='_None') zeta[i, 0, 0, p],
"bo",
markersize=3,
markeredgecolor="blue",
markerfacecolor="None",
label="_None",
)
ipos = 0 ipos = 0
for jdx, t in enumerate(zeta[i, 0, 0, :]): for jdx, t in enumerate(zeta[i, 0, 0, :]):
if t > -39.9: if t > -39.9:
ipos = jdx ipos = jdx
ax.text(x[ipos], -37.75, '{0} days'.format(((i + 1) * 100)), size=5, ax.text(
ha='left', va='center') x[ipos],
-37.75,
"{0} days".format(((i + 1) * 100)),
size=5,
ha="left",
va="center",
)
# fake items for labels # fake items for labels
ax.plot([-100., -100], [-100., -100], 'k', label='Analytical solution') ax.plot([-100.0, -100], [-100.0, -100], "k", label="Analytical solution")
ax.plot([-100., -100], [-100., -100], 'bo', markersize=3, ax.plot(
markeredgecolor='blue', markerfacecolor='None', label='SWI2') [-100.0, -100],
[-100.0, -100],
"bo",
markersize=3,
markeredgecolor="blue",
markerfacecolor="None",
label="SWI2",
)
# legend # legend
leg = ax.legend(loc='upper right', numpoints=1) leg = ax.legend(loc="upper right", numpoints=1)
leg._drawFrame = False leg._drawFrame = False
# axes # axes
ax.set_xlim(0, 250) ax.set_xlim(0, 250)
ax.set_ylim(-40, 0) ax.set_ylim(-40, 0)
ax.set_yticks(np.arange(-40, 1, 10)) ax.set_yticks(np.arange(-40, 1, 10))
a = ax.annotate("", xy=(50, -25), xytext=(30, -25), a = ax.annotate(
arrowprops=dict(arrowstyle='->', fc='k')) "",
ax.text(40, -22, 'groundwater flow velocity=0.125 m/d', ha='center', xy=(50, -25),
size=7) xytext=(30, -25),
ax.set_ylabel('Elevation, in meters') arrowprops=dict(arrowstyle="->", fc="k"),
ax.set_xlabel('Horizontal distance, in meters') )
ax.text(
40, -22, "groundwater flow velocity=0.125 m/d", ha="center", size=7
)
ax.set_ylabel("Elevation, in meters")
ax.set_xlabel("Horizontal distance, in meters")
outfig = os.path.join(workspace, 'Figure06_swi2ex1.{0}'.format(fext)) outfig = os.path.join(workspace, "Figure06_swi2ex1.{0}".format(fext))
fig.savefig(outfig, dpi=300) fig.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
return 0 return 0

401
examples/scripts/flopy_swi2_ex2.py
View File

@ -11,57 +11,59 @@ import flopy
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
# --modify default matplotlib settings # --modify default matplotlib settings
updates = {'font.family': ['Univers 57 Condensed', 'Arial'], updates = {
'mathtext.default': 'regular', "font.family": ["Univers 57 Condensed", "Arial"],
'pdf.compression': 0, "mathtext.default": "regular",
'pdf.fonttype': 42, "pdf.compression": 0,
'legend.fontsize': 7, "pdf.fonttype": 42,
'axes.labelsize': 8, "legend.fontsize": 7,
'xtick.labelsize': 7, "axes.labelsize": 8,
'ytick.labelsize': 7} "xtick.labelsize": 7,
"ytick.labelsize": 7,
}
plt.rcParams.update(updates) plt.rcParams.update(updates)
def run(): def run():
workspace = 'swiex2' workspace = "swiex2"
if not os.path.isdir(workspace): if not os.path.isdir(workspace):
os.mkdir(workspace) os.mkdir(workspace)
cleanFiles = False cleanFiles = False
skipRuns = False skipRuns = False
fext = 'png' fext = "png"
narg = len(sys.argv) narg = len(sys.argv)
iarg = 0 iarg = 0
if narg > 1: if narg > 1:
while iarg < narg - 1: while iarg < narg - 1:
iarg += 1 iarg += 1
basearg = sys.argv[iarg].lower() basearg = sys.argv[iarg].lower()
if basearg == '--clean': if basearg == "--clean":
cleanFiles = True cleanFiles = True
elif basearg == '--skipruns': elif basearg == "--skipruns":
skipRuns = True skipRuns = True
elif basearg == '--pdf': elif basearg == "--pdf":
fext = 'pdf' fext = "pdf"
dirs = [os.path.join(workspace, 'SWI2'), os.path.join(workspace, 'SEAWAT')] dirs = [os.path.join(workspace, "SWI2"), os.path.join(workspace, "SEAWAT")]
if cleanFiles: if cleanFiles:
print('cleaning all files') print("cleaning all files")
print('excluding *.py files') print("excluding *.py files")
file_dict = collections.OrderedDict() file_dict = collections.OrderedDict()
file_dict[0] = os.listdir(dirs[0]) file_dict[0] = os.listdir(dirs[0])
file_dict[1] = os.listdir(dirs[1]) file_dict[1] = os.listdir(dirs[1])
file_dict[-1] = os.listdir(workspace) file_dict[-1] = os.listdir(workspace)
for key, files in list(file_dict.items()): for key, files in list(file_dict.items()):
pth = '.' pth = "."
if key >= 0: if key >= 0:
pth = dirs[key] pth = dirs[key]
for f in files: for f in files:
fpth = os.path.join(pth, f) fpth = os.path.join(pth, f)
if os.path.isdir(fpth): if os.path.isdir(fpth):
continue continue
if '.py' != os.path.splitext(f)[1].lower(): if ".py" != os.path.splitext(f)[1].lower():
print(' removing...{}'.format(os.path.basename(f))) print(" removing...{}".format(os.path.basename(f)))
try: try:
os.remove(fpth) os.remove(fpth)
except: except:
@ -76,15 +78,15 @@ def run():
if not os.path.exists(d): if not os.path.exists(d):
os.mkdir(d) os.mkdir(d)
modelname = 'swiex2' modelname = "swiex2"
mf_name = 'mf2005' mf_name = "mf2005"
# problem data # problem data
nper = 1 nper = 1
perlen = 2000 perlen = 2000
nstp = 1000 nstp = 1000
nlay, nrow, ncol = 1, 1, 60 nlay, nrow, ncol = 1, 1, 60
delr = 5. delr = 5.0
nsurf = 2 nsurf = 2
x = np.arange(0.5 * delr, ncol * delr, delr) x = np.arange(0.5 * delr, ncol * delr, delr)
xedge = np.linspace(0, float(ncol) * delr, len(x) + 1) xedge = np.linspace(0, float(ncol) * delr, len(x) + 1)
@ -101,75 +103,111 @@ def run():
z.append(z0) z.append(z0)
z.append(z1) z.append(z1)
ssz = 0.2 ssz = 0.2
isource = np.ones((nrow, ncol), 'int') isource = np.ones((nrow, ncol), "int")
isource[0, 0] = 2 isource[0, 0] = 2
# stratified model # stratified model
modelname = 'swiex2_strat' modelname = "swiex2_strat"
print('creating...', modelname) print("creating...", modelname)
ml = flopy.modflow.Modflow(modelname, version='mf2005', exe_name=mf_name, ml = flopy.modflow.Modflow(
model_ws=dirs[0]) modelname, version="mf2005", exe_name=mf_name, model_ws=dirs[0]
discret = flopy.modflow.ModflowDis(ml, nlay=1, ncol=ncol, nrow=nrow, )
delr=delr, discret = flopy.modflow.ModflowDis(
delc=1, ml,
top=0, botm=[-40.0], nlay=1,
nper=nper, perlen=perlen, nstp=nstp) ncol=ncol,
nrow=nrow,
delr=delr,
delc=1,
top=0,
botm=[-40.0],
nper=nper,
perlen=perlen,
nstp=nstp,
)
bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=0.05) bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=0.05)
bcf = flopy.modflow.ModflowBcf(ml, laycon=0, tran=2 * 40) bcf = flopy.modflow.ModflowBcf(ml, laycon=0, tran=2 * 40)
swi = flopy.modflow.ModflowSwi2(ml, iswizt=55, nsrf=nsurf, istrat=1, swi = flopy.modflow.ModflowSwi2(
toeslope=0.2, tipslope=0.2, ml,
nu=[0, 0.0125, 0.025], iswizt=55,
zeta=z, ssz=ssz, isource=isource, nsrf=nsurf,
nsolver=1) istrat=1,
oc = flopy.modflow.ModflowOc(ml, toeslope=0.2,
stress_period_data={(0, 999): ['save head']}) tipslope=0.2,
nu=[0, 0.0125, 0.025],
zeta=z,
ssz=ssz,
isource=isource,
nsolver=1,
)
oc = flopy.modflow.ModflowOc(
ml, stress_period_data={(0, 999): ["save head"]}
)
pcg = flopy.modflow.ModflowPcg(ml) pcg = flopy.modflow.ModflowPcg(ml)
ml.write_input() ml.write_input()
# run stratified model # run stratified model
if not skipRuns: if not skipRuns:
m = ml.run_model(silent=False) m = ml.run_model(silent=False)
# read stratified results # read stratified results
zetafile = os.path.join(dirs[0], '{}.zta'.format(modelname)) zetafile = os.path.join(dirs[0], "{}.zta".format(modelname))
zobj = flopy.utils.CellBudgetFile(zetafile) zobj = flopy.utils.CellBudgetFile(zetafile)
zkstpkper = zobj.get_kstpkper() zkstpkper = zobj.get_kstpkper()
zeta = zobj.get_data(kstpkper=zkstpkper[-1], text='ZETASRF 1')[0] zeta = zobj.get_data(kstpkper=zkstpkper[-1], text="ZETASRF 1")[0]
zeta2 = zobj.get_data(kstpkper=zkstpkper[-1], text='ZETASRF 2')[0] zeta2 = zobj.get_data(kstpkper=zkstpkper[-1], text="ZETASRF 2")[0]
# #
# vd model # vd model
modelname = 'swiex2_vd' modelname = "swiex2_vd"
print('creating...', modelname) print("creating...", modelname)
ml = flopy.modflow.Modflow(modelname, version='mf2005', exe_name=mf_name, ml = flopy.modflow.Modflow(
model_ws=dirs[0]) modelname, version="mf2005", exe_name=mf_name, model_ws=dirs[0]
discret = flopy.modflow.ModflowDis(ml, nlay=1, ncol=ncol, nrow=nrow, )
delr=delr, discret = flopy.modflow.ModflowDis(
delc=1, ml,
top=0, botm=[-40.0], nlay=1,
nper=nper, perlen=perlen, nstp=nstp) ncol=ncol,
nrow=nrow,
delr=delr,
delc=1,
top=0,
botm=[-40.0],
nper=nper,
perlen=perlen,
nstp=nstp,
)
bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=0.05) bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=0.05)
bcf = flopy.modflow.ModflowBcf(ml, laycon=0, tran=2 * 40) bcf = flopy.modflow.ModflowBcf(ml, laycon=0, tran=2 * 40)
swi = flopy.modflow.ModflowSwi2(ml, iswizt=55, nsrf=nsurf, istrat=0, swi = flopy.modflow.ModflowSwi2(
toeslope=0.2, tipslope=0.2, ml,
nu=[0, 0, 0.025, 0.025], iswizt=55,
zeta=z, ssz=ssz, isource=isource, nsrf=nsurf,
nsolver=1) istrat=0,
oc = flopy.modflow.ModflowOc(ml, toeslope=0.2,
stress_period_data={(0, 999): ['save head']}) tipslope=0.2,
nu=[0, 0, 0.025, 0.025],
zeta=z,
ssz=ssz,
isource=isource,
nsolver=1,
)
oc = flopy.modflow.ModflowOc(
ml, stress_period_data={(0, 999): ["save head"]}
)
pcg = flopy.modflow.ModflowPcg(ml) pcg = flopy.modflow.ModflowPcg(ml)
ml.write_input() ml.write_input()
# run vd model # run vd model
if not skipRuns: if not skipRuns:
m = ml.run_model(silent=False) m = ml.run_model(silent=False)
# read vd model data # read vd model data
zetafile = os.path.join(dirs[0], '{}.zta'.format(modelname)) zetafile = os.path.join(dirs[0], "{}.zta".format(modelname))
zobj = flopy.utils.CellBudgetFile(zetafile) zobj = flopy.utils.CellBudgetFile(zetafile)
zkstpkper = zobj.get_kstpkper() zkstpkper = zobj.get_kstpkper()
zetavd = zobj.get_data(kstpkper=zkstpkper[-1], text='ZETASRF 1')[0] zetavd = zobj.get_data(kstpkper=zkstpkper[-1], text="ZETASRF 1")[0]
zetavd2 = zobj.get_data(kstpkper=zkstpkper[-1], text='ZETASRF 2')[0] zetavd2 = zobj.get_data(kstpkper=zkstpkper[-1], text="ZETASRF 2")[0]
# #
# seawat model # seawat model
swtexe_name = 'swtv4' swtexe_name = "swtv4"
modelname = 'swiex2_swt' modelname = "swiex2_swt"
print('creating...', modelname) print("creating...", modelname)
swt_xmax = 300.0 swt_xmax = 300.0
swt_zmax = 40.0 swt_zmax = 40.0
swt_delr = 1.0 swt_delr = 1.0
@ -184,7 +222,7 @@ def run():
swt_ibound[0, 0, 0] = -1 swt_ibound[0, 0, 0] = -1
swt_x = np.arange(0.5 * swt_delr, swt_ncol * swt_delr, swt_delr) swt_x = np.arange(0.5 * swt_delr, swt_ncol * swt_delr, swt_delr)
swt_xedge = np.linspace(0, float(ncol) * delr, len(swt_x) + 1) swt_xedge = np.linspace(0, float(ncol) * delr, len(swt_x) + 1)
swt_top = 0. swt_top = 0.0
z0 = swt_top z0 = swt_top
swt_botm = np.zeros((swt_nlay), np.float) swt_botm = np.zeros((swt_nlay), np.float)
swt_z = np.zeros((swt_nlay), np.float) swt_z = np.zeros((swt_nlay), np.float)
@ -217,60 +255,91 @@ def run():
# ssm data # ssm data
itype = flopy.mt3d.Mt3dSsm.itype_dict() itype = flopy.mt3d.Mt3dSsm.itype_dict()
ssm_data = {0: [0, 0, 0, 35., itype['BAS6']]} ssm_data = {0: [0, 0, 0, 35.0, itype["BAS6"]]}
# print sconcp # print sconcp
# mt3d print times # mt3d print times
timprs = (np.arange(5) + 1) * 2000. timprs = (np.arange(5) + 1) * 2000.0
nprs = len(timprs) nprs = len(timprs)
# create the MODFLOW files # create the MODFLOW files
m = flopy.seawat.Seawat(modelname, exe_name=swtexe_name, model_ws=dirs[1]) m = flopy.seawat.Seawat(modelname, exe_name=swtexe_name, model_ws=dirs[1])
discret = flopy.modflow.ModflowDis(m, nrow=swt_nrow, ncol=swt_ncol, discret = flopy.modflow.ModflowDis(
nlay=swt_nlay, m,
delr=swt_delr, delc=swt_delc, laycbd=0, nrow=swt_nrow,
top=swt_top, ncol=swt_ncol,
botm=swt_botm, nlay=swt_nlay,
nper=nper, perlen=perlen, nstp=1, delr=swt_delr,
steady=False) delc=swt_delc,
laycbd=0,
top=swt_top,
botm=swt_botm,
nper=nper,
perlen=perlen,
nstp=1,
steady=False,
)
bas = flopy.modflow.ModflowBas(m, ibound=swt_ibound, strt=0.05) bas = flopy.modflow.ModflowBas(m, ibound=swt_ibound, strt=0.05)
lpf = flopy.modflow.ModflowLpf(m, hk=2.0, vka=2.0, ss=0.0, sy=0.0, lpf = flopy.modflow.ModflowLpf(
laytyp=0, m, hk=2.0, vka=2.0, ss=0.0, sy=0.0, laytyp=0, layavg=0
layavg=0) )
oc = flopy.modflow.ModflowOc(m, save_every=1, save_types=['save head']) oc = flopy.modflow.ModflowOc(m, save_every=1, save_types=["save head"])
pcg = flopy.modflow.ModflowPcg(m) pcg = flopy.modflow.ModflowPcg(m)
# Create the MT3DMS model files # Create the MT3DMS model files
adv = flopy.mt3d.Mt3dAdv(m, mixelm=-1, # -1 is TVD adv = flopy.mt3d.Mt3dAdv(
percel=0.05, m,
nadvfd=0, mixelm=-1, # -1 is TVD
# 0 or 1 is upstream; 2 is central in space percel=0.05,
# particle based methods nadvfd=0,
nplane=4, # 0 or 1 is upstream; 2 is central in space
mxpart=1e7, # particle based methods
itrack=2, nplane=4,
dceps=1e-4, mxpart=1e7,
npl=16, itrack=2,
nph=16, dceps=1e-4,
npmin=8, npl=16,
npmax=256) nph=16,
btn = flopy.mt3d.Mt3dBtn(m, icbund=1, prsity=ssz, sconc=sconc, ifmtcn=-1, npmin=8,
chkmas=False, nprobs=10, nprmas=10, dt0=0.0, npmax=256,
ttsmult=1.2, )
ttsmax=100.0, btn = flopy.mt3d.Mt3dBtn(
ncomp=1, nprs=nprs, timprs=timprs, mxstrn=1e8) m,
dsp = flopy.mt3d.Mt3dDsp(m, al=0., trpt=1., trpv=1., dmcoef=0.) icbund=1,
gcg = flopy.mt3d.Mt3dGcg(m, mxiter=1, iter1=50, isolve=3, cclose=1e-6, prsity=ssz,
iprgcg=5) sconc=sconc,
ifmtcn=-1,
chkmas=False,
nprobs=10,
nprmas=10,
dt0=0.0,
ttsmult=1.2,
ttsmax=100.0,
ncomp=1,
nprs=nprs,
timprs=timprs,
mxstrn=1e8,
)
dsp = flopy.mt3d.Mt3dDsp(m, al=0.0, trpt=1.0, trpv=1.0, dmcoef=0.0)
gcg = flopy.mt3d.Mt3dGcg(
m, mxiter=1, iter1=50, isolve=3, cclose=1e-6, iprgcg=5
)
ssm = flopy.mt3d.Mt3dSsm(m, stress_period_data=ssm_data) ssm = flopy.mt3d.Mt3dSsm(m, stress_period_data=ssm_data)
# Create the SEAWAT model files # Create the SEAWAT model files
vdf = flopy.seawat.SeawatVdf(m, nswtcpl=1, iwtable=0, densemin=0, vdf = flopy.seawat.SeawatVdf(
densemax=0, m,
denseref=1000., denseslp=25., firstdt=1.0e-03) nswtcpl=1,
iwtable=0,
densemin=0,
densemax=0,
denseref=1000.0,
denseslp=25.0,
firstdt=1.0e-03,
)
m.write_input() m.write_input()
# run seawat model # run seawat model
if not skipRuns: if not skipRuns:
m = m.run_model(silent=False) m = m.run_model(silent=False)
# read seawat model data # read seawat model data
ucnfile = os.path.join(dirs[1], 'MT3D001.UCN') ucnfile = os.path.join(dirs[1], "MT3D001.UCN")
uobj = flopy.utils.UcnFile(ucnfile) uobj = flopy.utils.UcnFile(ucnfile)
times = uobj.get_times() times = uobj.get_times()
print(times) print(times)
@ -290,85 +359,125 @@ def run():
fbot = 0.125 fbot = 0.125
ftop = 0.925 ftop = 0.925
print('creating cross-section figure...') print("creating cross-section figure...")
xsf = plt.figure(figsize=(fwid, fhgt), facecolor='w') xsf = plt.figure(figsize=(fwid, fhgt), facecolor="w")
xsf.subplots_adjust(wspace=0.25, hspace=0.25, left=flft, right=frgt, xsf.subplots_adjust(
bottom=fbot, top=ftop) wspace=0.25, hspace=0.25, left=flft, right=frgt, bottom=fbot, top=ftop
)
# plot initial conditions # plot initial conditions
ax = xsf.add_subplot(3, 1, 1) ax = xsf.add_subplot(3, 1, 1)
ax.text(-0.075, 1.05, 'A', transform=ax.transAxes, va='center', ax.text(
ha='center', -0.075,
size='8') 1.05,
"A",
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
# text(.975, .1, '(a)', transform = ax.transAxes, va = 'center', ha = 'center') # text(.975, .1, '(a)', transform = ax.transAxes, va = 'center', ha = 'center')
ax.plot([110, 150], [0, -40], 'k') ax.plot([110, 150], [0, -40], "k")
ax.plot([150, 190], [0, -40], 'k') ax.plot([150, 190], [0, -40], "k")
ax.set_xlim(0, 300) ax.set_xlim(0, 300)
ax.set_ylim(-40, 0) ax.set_ylim(-40, 0)
ax.set_yticks(np.arange(-40, 1, 10)) ax.set_yticks(np.arange(-40, 1, 10))
ax.text(50, -20, 'salt', va='center', ha='center') ax.text(50, -20, "salt", va="center", ha="center")
ax.text(150, -20, 'brackish', va='center', ha='center') ax.text(150, -20, "brackish", va="center", ha="center")
ax.text(250, -20, 'fresh', va='center', ha='center') ax.text(250, -20, "fresh", va="center", ha="center")
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
# plot stratified swi2 and seawat results # plot stratified swi2 and seawat results
ax = xsf.add_subplot(3, 1, 2) ax = xsf.add_subplot(3, 1, 2)
ax.text(-0.075, 1.05, 'B', transform=ax.transAxes, va='center', ax.text(
ha='center', -0.075,
size='8') 1.05,
"B",
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
# #
zp = zeta[0, 0, :] zp = zeta[0, 0, :]
p = (zp < 0.0) & (zp > -40.) p = (zp < 0.0) & (zp > -40.0)
ax.plot(x[p], zp[p], 'b', linewidth=1.5, drawstyle='steps-mid') ax.plot(x[p], zp[p], "b", linewidth=1.5, drawstyle="steps-mid")
zp = zeta2[0, 0, :] zp = zeta2[0, 0, :]
p = (zp < 0.0) & (zp > -40.) p = (zp < 0.0) & (zp > -40.0)
ax.plot(x[p], zp[p], 'b', linewidth=1.5, drawstyle='steps-mid') ax.plot(x[p], zp[p], "b", linewidth=1.5, drawstyle="steps-mid")
# seawat data # seawat data
cc = ax.contour(swt_X, swt_Z, conc, levels=[0.25, 0.75], colors='k', cc = ax.contour(
linestyles='solid', linewidths=0.75, zorder=101) swt_X,
swt_Z,
conc,
levels=[0.25, 0.75],
colors="k",
linestyles="solid",
linewidths=0.75,
zorder=101,
)
# fake figures # fake figures
ax.plot([-100., -100], [-100., -100], 'b', linewidth=1.5, label='SWI2') ax.plot([-100.0, -100], [-100.0, -100], "b", linewidth=1.5, label="SWI2")
ax.plot([-100., -100], [-100., -100], 'k', linewidth=0.75, label='SEAWAT') ax.plot(
[-100.0, -100], [-100.0, -100], "k", linewidth=0.75, label="SEAWAT"
)
# legend # legend
leg = ax.legend(loc='lower left', numpoints=1) leg = ax.legend(loc="lower left", numpoints=1)
leg._drawFrame = False leg._drawFrame = False
# axes # axes
ax.set_xlim(0, 300) ax.set_xlim(0, 300)
ax.set_ylim(-40, 0) ax.set_ylim(-40, 0)
ax.set_yticks(np.arange(-40, 1, 10)) ax.set_yticks(np.arange(-40, 1, 10))
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
# plot vd model # plot vd model
ax = xsf.add_subplot(3, 1, 3) ax = xsf.add_subplot(3, 1, 3)
ax.text(-0.075, 1.05, 'C', transform=ax.transAxes, va='center', ax.text(
ha='center', size='8') -0.075,
1.05,
"C",
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
dr = zeta[0, 0, :] dr = zeta[0, 0, :]
ax.plot(x, dr, 'b', linewidth=1.5, drawstyle='steps-mid') ax.plot(x, dr, "b", linewidth=1.5, drawstyle="steps-mid")
dr = zeta2[0, 0, :] dr = zeta2[0, 0, :]
ax.plot(x, dr, 'b', linewidth=1.5, drawstyle='steps-mid') ax.plot(x, dr, "b", linewidth=1.5, drawstyle="steps-mid")
dr = zetavd[0, 0, :] dr = zetavd[0, 0, :]
ax.plot(x, dr, 'r', linewidth=0.75, drawstyle='steps-mid') ax.plot(x, dr, "r", linewidth=0.75, drawstyle="steps-mid")
dr = zetavd2[0, 0, :] dr = zetavd2[0, 0, :]
ax.plot(x, dr, 'r', linewidth=0.75, drawstyle='steps-mid') ax.plot(x, dr, "r", linewidth=0.75, drawstyle="steps-mid")
# fake figures # fake figures
ax.plot([-100., -100], [-100., -100], 'b', linewidth=1.5, ax.plot(
label='SWI2 stratified option') [-100.0, -100],
ax.plot([-100., -100], [-100., -100], 'r', linewidth=0.75, [-100.0, -100],
label='SWI2 continuous option') "b",
linewidth=1.5,
label="SWI2 stratified option",
)
ax.plot(
[-100.0, -100],
[-100.0, -100],
"r",
linewidth=0.75,
label="SWI2 continuous option",
)
# legend # legend
leg = ax.legend(loc='lower left', numpoints=1) leg = ax.legend(loc="lower left", numpoints=1)
leg._drawFrame = False leg._drawFrame = False
# axes # axes
ax.set_xlim(0, 300) ax.set_xlim(0, 300)
ax.set_ylim(-40, 0) ax.set_ylim(-40, 0)
ax.set_yticks(np.arange(-40, 1, 10)) ax.set_yticks(np.arange(-40, 1, 10))
ax.set_xlabel('Horizontal distance, in meters') ax.set_xlabel("Horizontal distance, in meters")
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
outfig = os.path.join(workspace, 'Figure07_swi2ex2.{0}'.format(fext)) outfig = os.path.join(workspace, "Figure07_swi2ex2.{0}".format(fext))
xsf.savefig(outfig, dpi=300) xsf.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
return 0 return 0
if __name__ == '__main__': if __name__ == "__main__":
success = run() success = run()
sys.exit(success) sys.exit(success)

268
examples/scripts/flopy_swi2_ex3.py
View File

@ -10,14 +10,16 @@ import flopy
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
# --modify default matplotlib settings # --modify default matplotlib settings
updates = {'font.family': ['Univers 57 Condensed', 'Arial'], updates = {
'mathtext.default': 'regular', "font.family": ["Univers 57 Condensed", "Arial"],
'pdf.compression': 0, "mathtext.default": "regular",
'pdf.fonttype': 42, "pdf.compression": 0,
'legend.fontsize': 7, "pdf.fonttype": 42,
'axes.labelsize': 8, "legend.fontsize": 7,
'xtick.labelsize': 7, "axes.labelsize": 8,
'ytick.labelsize': 7} "xtick.labelsize": 7,
"ytick.labelsize": 7,
}
plt.rcParams.update(updates) plt.rcParams.update(updates)
@ -42,104 +44,120 @@ def MergeData(ndim, zdata, tb):
def LegBar(ax, x0, y0, t0, dx, dy, dt, cc): def LegBar(ax, x0, y0, t0, dx, dy, dt, cc):
for c in cc: for c in cc:
ax.plot([x0, x0 + dx], [y0, y0], color=c, linewidth=4) ax.plot([x0, x0 + dx], [y0, y0], color=c, linewidth=4)
ctxt = '{0:=3d} years'.format(t0) ctxt = "{0:=3d} years".format(t0)
ax.text(x0 + 2. * dx, y0 + dy / 2., ctxt, size=5) ax.text(x0 + 2.0 * dx, y0 + dy / 2.0, ctxt, size=5)
y0 += dy y0 += dy
t0 += dt t0 += dt
return return
def run(): def run():
workspace = 'swiex3' workspace = "swiex3"
cleanFiles = False cleanFiles = False
fext = 'png' fext = "png"
narg = len(sys.argv) narg = len(sys.argv)
iarg = 0 iarg = 0
if narg > 1: if narg > 1:
while iarg < narg - 1: while iarg < narg - 1:
iarg += 1 iarg += 1
basearg = sys.argv[iarg].lower() basearg = sys.argv[iarg].lower()
if basearg == '--clean': if basearg == "--clean":
cleanFiles = True cleanFiles = True
elif basearg == '--pdf': elif basearg == "--pdf":
fext = 'pdf' fext = "pdf"
if cleanFiles: if cleanFiles:
print('cleaning all files') print("cleaning all files")
print('excluding *.py files') print("excluding *.py files")
files = os.listdir(workspace) files = os.listdir(workspace)
for f in files: for f in files:
fpth = os.path.join(workspace, f) fpth = os.path.join(workspace, f)
if os.path.isdir(fpth): if os.path.isdir(fpth):
continue continue
if '.py' != os.path.splitext(f)[1].lower(): if ".py" != os.path.splitext(f)[1].lower():
print(' removing...{}'.format(os.path.basename(f))) print(" removing...{}".format(os.path.basename(f)))
try: try:
os.remove(fpth) os.remove(fpth)
except: except:
pass pass
return 0 return 0
modelname = 'swiex3' modelname = "swiex3"
exe_name = 'mf2005' exe_name = "mf2005"
nlay = 3 nlay = 3
nrow = 1 nrow = 1
ncol = 200 ncol = 200
delr = 20.0 delr = 20.0
delc = 1. delc = 1.0
# well data # well data
lrcQ1 = np.array([(0, 0, 199, 0.01), (2, 0, 199, 0.02)]) lrcQ1 = np.array([(0, 0, 199, 0.01), (2, 0, 199, 0.02)])
lrcQ2 = np.array([(0, 0, 199, 0.01 * 0.5), (2, 0, 199, 0.02 * 0.5)]) lrcQ2 = np.array([(0, 0, 199, 0.01 * 0.5), (2, 0, 199, 0.02 * 0.5)])
# ghb data # ghb data
lrchc = np.zeros((30, 5)) lrchc = np.zeros((30, 5))
lrchc[:, [0, 1, 3, 4]] = [0, 0, 0., 0.8 / 2.0] lrchc[:, [0, 1, 3, 4]] = [0, 0, 0.0, 0.8 / 2.0]
lrchc[:, 2] = np.arange(0, 30) lrchc[:, 2] = np.arange(0, 30)
# swi2 data # swi2 data
zini = np.hstack( zini = np.hstack(
(-9 * np.ones(24), np.arange(-9, -50, -0.5), -50 * np.ones(94)))[ (-9 * np.ones(24), np.arange(-9, -50, -0.5), -50 * np.ones(94))
np.newaxis, :] )[np.newaxis, :]
iso = np.zeros((1, 200), dtype=np.int) iso = np.zeros((1, 200), dtype=np.int)
iso[:, :30] = -2 iso[:, :30] = -2
# model objects # model objects
ml = flopy.modflow.Modflow(modelname, version='mf2005', exe_name=exe_name, ml = flopy.modflow.Modflow(
model_ws=workspace) modelname, version="mf2005", exe_name=exe_name, model_ws=workspace
discret = flopy.modflow.ModflowDis(ml, nrow=nrow, ncol=ncol, nlay=3, )
delr=delr, discret = flopy.modflow.ModflowDis(
delc=delc, ml,
laycbd=[0, 0, 0], top=-9.0, nrow=nrow,
botm=[-29, -30, -50], ncol=ncol,
nper=2, perlen=[365 * 1000, 1000 * 365], nlay=3,
nstp=[500, 500]) delr=delr,
delc=delc,
laycbd=[0, 0, 0],
top=-9.0,
botm=[-29, -30, -50],
nper=2,
perlen=[365 * 1000, 1000 * 365],
nstp=[500, 500],
)
bas = flopy.modflow.ModflowBas(ml, ibound=1, strt=1.0) bas = flopy.modflow.ModflowBas(ml, ibound=1, strt=1.0)
bcf = flopy.modflow.ModflowBcf(ml, laycon=[0, 0, 0], tran=[40.0, 1, 80.0], bcf = flopy.modflow.ModflowBcf(
vcont=[0.005, 0.005]) ml, laycon=[0, 0, 0], tran=[40.0, 1, 80.0], vcont=[0.005, 0.005]
)
wel = flopy.modflow.ModflowWel(ml, stress_period_data={0: lrcQ1, 1: lrcQ2}) wel = flopy.modflow.ModflowWel(ml, stress_period_data={0: lrcQ1, 1: lrcQ2})
ghb = flopy.modflow.ModflowGhb(ml, stress_period_data={0: lrchc}) ghb = flopy.modflow.ModflowGhb(ml, stress_period_data={0: lrchc})
swi = flopy.modflow.ModflowSwi2(ml, iswizt=55, nsrf=1, istrat=1, swi = flopy.modflow.ModflowSwi2(
toeslope=0.01, tipslope=0.04, ml,
nu=[0, 0.025], iswizt=55,
zeta=[zini, zini, zini], ssz=0.2, nsrf=1,
isource=iso, istrat=1,
nsolver=1) toeslope=0.01,
oc = flopy.modflow.ModflowOc(ml, save_every=100, save_types=['save head']) tipslope=0.04,
nu=[0, 0.025],
zeta=[zini, zini, zini],
ssz=0.2,
isource=iso,
nsolver=1,
)
oc = flopy.modflow.ModflowOc(ml, save_every=100, save_types=["save head"])
pcg = flopy.modflow.ModflowPcg(ml) pcg = flopy.modflow.ModflowPcg(ml)
# write the model files # write the model files
ml.write_input() ml.write_input()
# run the model # run the model
m = ml.run_model(silent=True) m = ml.run_model(silent=True)
headfile = os.path.join(workspace, '{}.hds'.format(modelname)) headfile = os.path.join(workspace, "{}.hds".format(modelname))
hdobj = flopy.utils.HeadFile(headfile) hdobj = flopy.utils.HeadFile(headfile)
head = hdobj.get_data(totim=3.65000E+05) head = hdobj.get_data(totim=3.65000e05)
zetafile = os.path.join(workspace, '{}.zta'.format(modelname)) zetafile = os.path.join(workspace, "{}.zta".format(modelname))
zobj = flopy.utils.CellBudgetFile(zetafile) zobj = flopy.utils.CellBudgetFile(zetafile)
zkstpkper = zobj.get_kstpkper() zkstpkper = zobj.get_kstpkper()
zeta = [] zeta = []
for kk in zkstpkper: for kk in zkstpkper:
zeta.append(zobj.get_data(kstpkper=kk, text='ZETASRF 1')[0]) zeta.append(zobj.get_data(kstpkper=kk, text="ZETASRF 1")[0])
zeta = np.array(zeta) zeta = np.array(zeta)
fwid, fhgt = 7.00, 4.50 fwid, fhgt = 7.00, 4.50
@ -154,106 +172,150 @@ def run():
lw = 0.5 lw = 0.5
x = np.arange(-30 * delr + 0.5 * delr, (ncol - 30) * delr, delr) x = np.arange(-30 * delr + 0.5 * delr, (ncol - 30) * delr, delr)
xedge = np.linspace(-30. * delr, (ncol - 30.) * delr, len(x) + 1) xedge = np.linspace(-30.0 * delr, (ncol - 30.0) * delr, len(x) + 1)
zedge = [[-9., -29.], [-29., -30.], [-30., -50.]] zedge = [[-9.0, -29.0], [-29.0, -30.0], [-30.0, -50.0]]
fig = plt.figure(figsize=(fwid, fhgt), facecolor='w') fig = plt.figure(figsize=(fwid, fhgt), facecolor="w")
fig.subplots_adjust(wspace=0.25, hspace=0.25, left=flft, right=frgt, fig.subplots_adjust(
bottom=fbot, top=ftop) wspace=0.25, hspace=0.25, left=flft, right=frgt, bottom=fbot, top=ftop
)
ax = fig.add_subplot(311) ax = fig.add_subplot(311)
ax.text(-0.075, 1.05, 'A', transform=ax.transAxes, va='center', ax.text(
ha='center', -0.075,
size='8') 1.05,
"A",
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
# confining unit # confining unit
ax.fill([-600, 3400, 3400, -600], [-29, -29, -30, -30], fc=[.8, .8, .8], ax.fill(
ec=[.8, .8, .8]) [-600, 3400, 3400, -600],
[-29, -29, -30, -30],
fc=[0.8, 0.8, 0.8],
ec=[0.8, 0.8, 0.8],
)
# #
z = np.copy(zini[0, :]) z = np.copy(zini[0, :])
zr = z.copy() zr = z.copy()
p = (zr < -9.) & (zr > -50.0) p = (zr < -9.0) & (zr > -50.0)
ax.plot(x[p], zr[p], color=cc[0], linewidth=lw, drawstyle='steps-mid') ax.plot(x[p], zr[p], color=cc[0], linewidth=lw, drawstyle="steps-mid")
# #
for i in range(5): for i in range(5):
zt = MergeData(ncol, zt = MergeData(
[zeta[i, 0, 0, :], zeta[i, 1, 0, :], zeta[i, 2, 0, :]], ncol, [zeta[i, 0, 0, :], zeta[i, 1, 0, :], zeta[i, 2, 0, :]], zedge
zedge) )
dr = zt.copy() dr = zt.copy()
ax.plot(x, dr, color=cc[i + 1], linewidth=lw, drawstyle='steps-mid') ax.plot(x, dr, color=cc[i + 1], linewidth=lw, drawstyle="steps-mid")
# Manufacture a legend bar # Manufacture a legend bar
LegBar(ax, -200., -33.75, 0, 25, -2.5, 200, cc[0:6]) LegBar(ax, -200.0, -33.75, 0, 25, -2.5, 200, cc[0:6])
# axes # axes
ax.set_ylim(-50, -9) ax.set_ylim(-50, -9)
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
ax.set_xlim(-250., 2500.) ax.set_xlim(-250.0, 2500.0)
ax = fig.add_subplot(312) ax = fig.add_subplot(312)
ax.text(-0.075, 1.05, 'B', transform=ax.transAxes, va='center', ax.text(
ha='center', -0.075,
size='8') 1.05,
"B",
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
# confining unit # confining unit
ax.fill([-600, 3400, 3400, -600], [-29, -29, -30, -30], fc=[.8, .8, .8], ax.fill(
ec=[.8, .8, .8]) [-600, 3400, 3400, -600],
[-29, -29, -30, -30],
fc=[0.8, 0.8, 0.8],
ec=[0.8, 0.8, 0.8],
)
# #
for i in range(4, 10): for i in range(4, 10):
zt = MergeData(ncol, zt = MergeData(
[zeta[i, 0, 0, :], zeta[i, 1, 0, :], zeta[i, 2, 0, :]], ncol, [zeta[i, 0, 0, :], zeta[i, 1, 0, :], zeta[i, 2, 0, :]], zedge
zedge) )
dr = zt.copy() dr = zt.copy()
ax.plot(x, dr, color=cc[i + 1], linewidth=lw, drawstyle='steps-mid') ax.plot(x, dr, color=cc[i + 1], linewidth=lw, drawstyle="steps-mid")
# Manufacture a legend bar # Manufacture a legend bar
LegBar(ax, -200., -33.75, 1000, 25, -2.5, 200, cc[5:11]) LegBar(ax, -200.0, -33.75, 1000, 25, -2.5, 200, cc[5:11])
# axes # axes
ax.set_ylim(-50, -9) ax.set_ylim(-50, -9)
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
ax.set_xlim(-250., 2500.) ax.set_xlim(-250.0, 2500.0)
ax = fig.add_subplot(313) ax = fig.add_subplot(313)
ax.text(-0.075, 1.05, 'C', transform=ax.transAxes, va='center', ax.text(
ha='center', -0.075,
size='8') 1.05,
"C",
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
# confining unit # confining unit
ax.fill([-600, 3400, 3400, -600], [-29, -29, -30, -30], fc=[.8, .8, .8], ax.fill(
ec=[.8, .8, .8]) [-600, 3400, 3400, -600],
[-29, -29, -30, -30],
fc=[0.8, 0.8, 0.8],
ec=[0.8, 0.8, 0.8],
)
# #
zt = MergeData(ncol, zt = MergeData(
[zeta[4, 0, 0, :], zeta[4, 1, 0, :], zeta[4, 2, 0, :]], ncol, [zeta[4, 0, 0, :], zeta[4, 1, 0, :], zeta[4, 2, 0, :]], zedge
zedge) )
ax.plot(x, zt, marker='o', markersize=3, linewidth=0.0, ax.plot(
markeredgecolor='blue', x,
markerfacecolor='None') zt,
marker="o",
markersize=3,
linewidth=0.0,
markeredgecolor="blue",
markerfacecolor="None",
)
# ghyben herzberg # ghyben herzberg
zeta1 = -9 - 40. * (head[0, 0, :]) zeta1 = -9 - 40.0 * (head[0, 0, :])
gbh = np.empty(len(zeta1), np.float) gbh = np.empty(len(zeta1), np.float)
gbho = np.empty(len(zeta1), np.float) gbho = np.empty(len(zeta1), np.float)
for idx, z1 in enumerate(zeta1): for idx, z1 in enumerate(zeta1):
if z1 >= -9.0 or z1 <= -50.0: if z1 >= -9.0 or z1 <= -50.0:
gbh[idx] = np.nan gbh[idx] = np.nan
gbho[idx] = 0. gbho[idx] = 0.0
else: else:
gbh[idx] = z1 gbh[idx] = z1
gbho[idx] = z1 gbho[idx] = z1
ax.plot(x, gbh, 'r') ax.plot(x, gbh, "r")
np.savetxt(os.path.join(workspace, 'Ghyben-Herzberg.out'), gbho) np.savetxt(os.path.join(workspace, "Ghyben-Herzberg.out"), gbho)
# fake figures # fake figures
ax.plot([-100., -100], [-100., -100], 'r', label='Ghyben-Herzberg') ax.plot([-100.0, -100], [-100.0, -100], "r", label="Ghyben-Herzberg")
ax.plot([-100., -100], [-100., -100], 'bo', markersize=3, ax.plot(
markeredgecolor='blue', markerfacecolor='None', label='SWI2') [-100.0, -100],
[-100.0, -100],
"bo",
markersize=3,
markeredgecolor="blue",
markerfacecolor="None",
label="SWI2",
)
# legend # legend
leg = ax.legend(loc='lower left', numpoints=1) leg = ax.legend(loc="lower left", numpoints=1)
leg._drawFrame = False leg._drawFrame = False
# axes # axes
ax.set_ylim(-50, -9) ax.set_ylim(-50, -9)
ax.set_xlabel('Horizontal distance, in meters') ax.set_xlabel("Horizontal distance, in meters")
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
ax.set_xlim(-250., 2500.) ax.set_xlim(-250.0, 2500.0)
outfig = os.path.join(workspace, 'Figure08_swi2ex3.{0}'.format(fext)) outfig = os.path.join(workspace, "Figure08_swi2ex3.{0}".format(fext))
fig.savefig(outfig, dpi=300) fig.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
return 0 return 0
if __name__ == '__main__': if __name__ == "__main__":
success = run() success = run()

510
examples/scripts/flopy_swi2_ex4.py
View File

@ -11,56 +11,58 @@ import flopy
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
# --modify default matplotlib settings # --modify default matplotlib settings
updates = {'font.family': ['Univers 57 Condensed', 'Arial'], updates = {
'mathtext.default': 'regular', "font.family": ["Univers 57 Condensed", "Arial"],
'pdf.compression': 0, "mathtext.default": "regular",
'pdf.fonttype': 42, "pdf.compression": 0,
'legend.fontsize': 7, "pdf.fonttype": 42,
'axes.labelsize': 8, "legend.fontsize": 7,
'xtick.labelsize': 7, "axes.labelsize": 8,
'ytick.labelsize': 7} "xtick.labelsize": 7,
"ytick.labelsize": 7,
}
plt.rcParams.update(updates) plt.rcParams.update(updates)
def LegBar(ax, x0, y0, t0, dx, dy, dt, cc): def LegBar(ax, x0, y0, t0, dx, dy, dt, cc):
for c in cc: for c in cc:
ax.plot([x0, x0 + dx], [y0, y0], color=c, linewidth=4) ax.plot([x0, x0 + dx], [y0, y0], color=c, linewidth=4)
ctxt = '{0:=3d} years'.format(t0) ctxt = "{0:=3d} years".format(t0)
ax.text(x0 + 2. * dx, y0 + dy / 2., ctxt, size=5) ax.text(x0 + 2.0 * dx, y0 + dy / 2.0, ctxt, size=5)
y0 += dy y0 += dy
t0 += dt t0 += dt
return return
def run(): def run():
workspace = 'swiex4' workspace = "swiex4"
cleanFiles = False cleanFiles = False
skipRuns = False skipRuns = False
fext = 'png' fext = "png"
narg = len(sys.argv) narg = len(sys.argv)
iarg = 0 iarg = 0
if narg > 1: if narg > 1:
while iarg < narg - 1: while iarg < narg - 1:
iarg += 1 iarg += 1
basearg = sys.argv[iarg].lower() basearg = sys.argv[iarg].lower()
if basearg == '--clean': if basearg == "--clean":
cleanFiles = True cleanFiles = True
elif basearg == '--skipruns': elif basearg == "--skipruns":
skipRuns = True skipRuns = True
elif basearg == '--pdf': elif basearg == "--pdf":
fext = 'pdf' fext = "pdf"
if cleanFiles: if cleanFiles:
print('cleaning all files') print("cleaning all files")
print('excluding *.py files') print("excluding *.py files")
files = os.listdir(workspace) files = os.listdir(workspace)
for f in files: for f in files:
fpth = os.path.join(workspace, f) fpth = os.path.join(workspace, f)
if os.path.isdir(fpth): if os.path.isdir(fpth):
continue continue
if '.py' != os.path.splitext(f)[1].lower(): if ".py" != os.path.splitext(f)[1].lower():
print(' removing...{}'.format(os.path.basename(f))) print(" removing...{}".format(os.path.basename(f)))
try: try:
os.remove(fpth) os.remove(fpth)
except: except:
@ -68,45 +70,45 @@ def run():
return 0 return 0
# Set path and name of MODFLOW exe # Set path and name of MODFLOW exe
exe_name = 'mf2005' exe_name = "mf2005"
if platform.system() == 'Windows': if platform.system() == "Windows":
exe_name = 'mf2005' exe_name = "mf2005"
ncol = 61 ncol = 61
nrow = 61 nrow = 61
nlay = 2 nlay = 2
nper = 3 nper = 3
perlen = [365.25 * 200., 365.25 * 12., 365.25 * 18.] perlen = [365.25 * 200.0, 365.25 * 12.0, 365.25 * 18.0]
nstp = [1000, 120, 180] nstp = [1000, 120, 180]
save_head = [200, 60, 60] save_head = [200, 60, 60]
steady = True steady = True
# dis data # dis data
delr, delc = 50.0, 50.0 delr, delc = 50.0, 50.0
botm = np.array([-10., -30., -50.]) botm = np.array([-10.0, -30.0, -50.0])
# oc data # oc data
ocspd = {} ocspd = {}
ocspd[(0, 199)] = ['save head'] ocspd[(0, 199)] = ["save head"]
ocspd[(0, 200)] = [] ocspd[(0, 200)] = []
ocspd[(0, 399)] = ['save head'] ocspd[(0, 399)] = ["save head"]
ocspd[(0, 400)] = [] ocspd[(0, 400)] = []
ocspd[(0, 599)] = ['save head'] ocspd[(0, 599)] = ["save head"]
ocspd[(0, 600)] = [] ocspd[(0, 600)] = []
ocspd[(0, 799)] = ['save head'] ocspd[(0, 799)] = ["save head"]
ocspd[(0, 800)] = [] ocspd[(0, 800)] = []
ocspd[(0, 999)] = ['save head'] ocspd[(0, 999)] = ["save head"]
ocspd[(1, 0)] = [] ocspd[(1, 0)] = []
ocspd[(1, 59)] = ['save head'] ocspd[(1, 59)] = ["save head"]
ocspd[(1, 60)] = [] ocspd[(1, 60)] = []
ocspd[(1, 119)] = ['save head'] ocspd[(1, 119)] = ["save head"]
ocspd[(2, 0)] = [] ocspd[(2, 0)] = []
ocspd[(2, 59)] = ['save head'] ocspd[(2, 59)] = ["save head"]
ocspd[(2, 60)] = [] ocspd[(2, 60)] = []
ocspd[(2, 119)] = ['save head'] ocspd[(2, 119)] = ["save head"]
modelname = 'swiex4_2d_2layer' modelname = "swiex4_2d_2layer"
# bas data # bas data
# ibound - active except for the corners # ibound - active except for the corners
@ -120,7 +122,7 @@ def run():
# lpf data # lpf data
laytyp = 0 laytyp = 0
hk = 10. hk = 10.0
vka = 0.2 vka = 0.2
# boundary condition data # boundary condition data
@ -136,7 +138,7 @@ def run():
lrchc[:, 0] = 0 lrchc[:, 0] = 0
lrchc[:, 1] = rowcell[index == 1] lrchc[:, 1] = rowcell[index == 1]
lrchc[:, 2] = colcell[index == 1] lrchc[:, 2] = colcell[index == 1]
lrchc[:, 3] = 0. lrchc[:, 3] = 0.0
lrchc[:, 4] = 50.0 * 50.0 / 40.0 lrchc[:, 4] = 50.0 * 50.0 / 40.0
# create ghb dictionary # create ghb dictionary
ghb_data = {0: lrchc} ghb_data = {0: lrchc}
@ -155,8 +157,8 @@ def run():
lrcqw = lrcq.copy() lrcqw = lrcq.copy()
lrcqw[0, 3] = -250 lrcqw[0, 3] = -250
lrcqsw = lrcq.copy() lrcqsw = lrcq.copy()
lrcqsw[0, 3] = -250. lrcqsw[0, 3] = -250.0
lrcqsw[1, 3] = -25. lrcqsw[1, 3] = -25.0
# create well dictionary # create well dictionary
base_well_data = {0: lrcq, 1: lrcqw} base_well_data = {0: lrcq, 1: lrcqw}
swwells_well_data = {0: lrcq, 1: lrcqw, 2: lrcqsw} swwells_well_data = {0: lrcq, 1: lrcqw, 2: lrcqsw}
@ -178,78 +180,117 @@ def run():
iso[1, 30, 35] = 2 iso[1, 30, 35] = 2
ssz = 0.2 ssz = 0.2
# swi2 observations # swi2 observations
obsnam = ['layer1_', 'layer2_'] obsnam = ["layer1_", "layer2_"]
obslrc = [[0, 30, 35], [1, 30, 35]] obslrc = [[0, 30, 35], [1, 30, 35]]
nobs = len(obsnam) nobs = len(obsnam)
iswiobs = 1051 iswiobs = 1051
modelname = 'swiex4_s1' modelname = "swiex4_s1"
if not skipRuns: if not skipRuns:
ml = flopy.modflow.Modflow(modelname, version='mf2005', ml = flopy.modflow.Modflow(
exe_name=exe_name, modelname, version="mf2005", exe_name=exe_name, model_ws=workspace
model_ws=workspace) )
discret = flopy.modflow.ModflowDis(ml, nlay=nlay, nrow=nrow, ncol=ncol, discret = flopy.modflow.ModflowDis(
laycbd=0, ml,
delr=delr, delc=delc, top=botm[0], nlay=nlay,
botm=botm[1:], nrow=nrow,
nper=nper, perlen=perlen, nstp=nstp, ncol=ncol,
steady=steady) laycbd=0,
delr=delr,
delc=delc,
top=botm[0],
botm=botm[1:],
nper=nper,
perlen=perlen,
nstp=nstp,
steady=steady,
)
bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=ihead) bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=ihead)
lpf = flopy.modflow.ModflowLpf(ml, laytyp=laytyp, hk=hk, vka=vka) lpf = flopy.modflow.ModflowLpf(ml, laytyp=laytyp, hk=hk, vka=vka)
wel = flopy.modflow.ModflowWel(ml, stress_period_data=base_well_data) wel = flopy.modflow.ModflowWel(ml, stress_period_data=base_well_data)
ghb = flopy.modflow.ModflowGhb(ml, stress_period_data=ghb_data) ghb = flopy.modflow.ModflowGhb(ml, stress_period_data=ghb_data)
rch = flopy.modflow.ModflowRch(ml, rech=rch_data) rch = flopy.modflow.ModflowRch(ml, rech=rch_data)
swi = flopy.modflow.ModflowSwi2(ml, iswizt=55, nsrf=1, istrat=1, swi = flopy.modflow.ModflowSwi2(
toeslope=toeslope, ml,
tipslope=tipslope, nu=nu, iswizt=55,
zeta=z, ssz=ssz, isource=iso, nsrf=1,
nsolver=1, istrat=1,
nadptmx=nadptmx, nadptmn=nadptmn, toeslope=toeslope,
nobs=nobs, iswiobs=iswiobs, tipslope=tipslope,
obsnam=obsnam, nu=nu,
obslrc=obslrc) zeta=z,
ssz=ssz,
isource=iso,
nsolver=1,
nadptmx=nadptmx,
nadptmn=nadptmn,
nobs=nobs,
iswiobs=iswiobs,
obsnam=obsnam,
obslrc=obslrc,
)
oc = flopy.modflow.ModflowOc(ml, stress_period_data=ocspd) oc = flopy.modflow.ModflowOc(ml, stress_period_data=ocspd)
pcg = flopy.modflow.ModflowPcg(ml, hclose=1.0e-6, rclose=3.0e-3, pcg = flopy.modflow.ModflowPcg(
mxiter=100, iter1=50) ml, hclose=1.0e-6, rclose=3.0e-3, mxiter=100, iter1=50
)
# create model files # create model files
ml.write_input() ml.write_input()
# run the model # run the model
m = ml.run_model(silent=False) m = ml.run_model(silent=False)
# model with saltwater wells # model with saltwater wells
modelname2 = 'swiex4_s2' modelname2 = "swiex4_s2"
if not skipRuns: if not skipRuns:
ml2 = flopy.modflow.Modflow(modelname2, version='mf2005', ml2 = flopy.modflow.Modflow(
exe_name=exe_name, modelname2, version="mf2005", exe_name=exe_name, model_ws=workspace
model_ws=workspace) )
discret = flopy.modflow.ModflowDis(ml2, nlay=nlay, nrow=nrow, discret = flopy.modflow.ModflowDis(
ncol=ncol, ml2,
laycbd=0, nlay=nlay,
delr=delr, delc=delc, top=botm[0], nrow=nrow,
botm=botm[1:], ncol=ncol,
nper=nper, perlen=perlen, nstp=nstp, laycbd=0,
steady=steady) delr=delr,
delc=delc,
top=botm[0],
botm=botm[1:],
nper=nper,
perlen=perlen,
nstp=nstp,
steady=steady,
)
bas = flopy.modflow.ModflowBas(ml2, ibound=ibound, strt=ihead) bas = flopy.modflow.ModflowBas(ml2, ibound=ibound, strt=ihead)
lpf = flopy.modflow.ModflowLpf(ml2, laytyp=laytyp, hk=hk, vka=vka) lpf = flopy.modflow.ModflowLpf(ml2, laytyp=laytyp, hk=hk, vka=vka)
wel = flopy.modflow.ModflowWel(ml2, wel = flopy.modflow.ModflowWel(
stress_period_data=swwells_well_data) ml2, stress_period_data=swwells_well_data
)
ghb = flopy.modflow.ModflowGhb(ml2, stress_period_data=ghb_data) ghb = flopy.modflow.ModflowGhb(ml2, stress_period_data=ghb_data)
rch = flopy.modflow.ModflowRch(ml2, rech=rch_data) rch = flopy.modflow.ModflowRch(ml2, rech=rch_data)
swi = flopy.modflow.ModflowSwi2(ml2, iswizt=55, nsrf=1, istrat=1, swi = flopy.modflow.ModflowSwi2(
toeslope=toeslope, ml2,
tipslope=tipslope, nu=nu, iswizt=55,
zeta=z, ssz=ssz, isource=iso, nsrf=1,
nsolver=1, istrat=1,
nadptmx=nadptmx, nadptmn=nadptmn, toeslope=toeslope,
nobs=nobs, iswiobs=iswiobs, tipslope=tipslope,
obsnam=obsnam, nu=nu,
obslrc=obslrc) zeta=z,
ssz=ssz,
isource=iso,
nsolver=1,
nadptmx=nadptmx,
nadptmn=nadptmn,
nobs=nobs,
iswiobs=iswiobs,
obsnam=obsnam,
obslrc=obslrc,
)
oc = flopy.modflow.ModflowOc(ml2, stress_period_data=ocspd) oc = flopy.modflow.ModflowOc(ml2, stress_period_data=ocspd)
pcg = flopy.modflow.ModflowPcg(ml2, hclose=1.0e-6, rclose=3.0e-3, pcg = flopy.modflow.ModflowPcg(
mxiter=100, ml2, hclose=1.0e-6, rclose=3.0e-3, mxiter=100, iter1=50
iter1=50) )
# create model files # create model files
ml2.write_input() ml2.write_input()
# run the model # run the model
@ -258,33 +299,37 @@ def run():
# Load the simulation 1 `ZETA` data and `ZETA` observations. # Load the simulation 1 `ZETA` data and `ZETA` observations.
# read base model zeta # read base model zeta
zfile = flopy.utils.CellBudgetFile( zfile = flopy.utils.CellBudgetFile(
os.path.join(workspace, modelname + '.zta')) os.path.join(workspace, modelname + ".zta")
)
kstpkper = zfile.get_kstpkper() kstpkper = zfile.get_kstpkper()
zeta = [] zeta = []
for kk in kstpkper: for kk in kstpkper:
zeta.append(zfile.get_data(kstpkper=kk, text='ZETASRF 1')[0]) zeta.append(zfile.get_data(kstpkper=kk, text="ZETASRF 1")[0])
zeta = np.array(zeta) zeta = np.array(zeta)
# read swi obs # read swi obs
zobs = np.genfromtxt(os.path.join(workspace, modelname + '.zobs.out'), zobs = np.genfromtxt(
names=True) os.path.join(workspace, modelname + ".zobs.out"), names=True
)
# Load the simulation 2 `ZETA` data and `ZETA` observations. # Load the simulation 2 `ZETA` data and `ZETA` observations.
# read saltwater well model zeta # read saltwater well model zeta
zfile2 = flopy.utils.CellBudgetFile( zfile2 = flopy.utils.CellBudgetFile(
os.path.join(workspace, modelname2 + '.zta')) os.path.join(workspace, modelname2 + ".zta")
)
kstpkper = zfile2.get_kstpkper() kstpkper = zfile2.get_kstpkper()
zeta2 = [] zeta2 = []
for kk in kstpkper: for kk in kstpkper:
zeta2.append(zfile2.get_data(kstpkper=kk, text='ZETASRF 1')[0]) zeta2.append(zfile2.get_data(kstpkper=kk, text="ZETASRF 1")[0])
zeta2 = np.array(zeta2) zeta2 = np.array(zeta2)
# read swi obs # read swi obs
zobs2 = np.genfromtxt(os.path.join(workspace, modelname2 + '.zobs.out'), zobs2 = np.genfromtxt(
names=True) os.path.join(workspace, modelname2 + ".zobs.out"), names=True
)
# Create arrays for the x-coordinates and the output years # Create arrays for the x-coordinates and the output years
x = np.linspace(-1500, 1500, 61) x = np.linspace(-1500, 1500, 61)
xcell = np.linspace(-1500, 1500, 61) + delr / 2. xcell = np.linspace(-1500, 1500, 61) + delr / 2.0
xedge = np.linspace(-1525, 1525, 62) xedge = np.linspace(-1525, 1525, 62)
years = [40, 80, 120, 160, 200, 6, 12, 18, 24, 30] years = [40, 80, 120, 160, 200, 6, 12, 18, 24, 30]
@ -304,10 +349,11 @@ def run():
# Recreate **Figure 9** from the SWI2 documentation (http://pubs.usgs.gov/tm/6a46/). # Recreate **Figure 9** from the SWI2 documentation (http://pubs.usgs.gov/tm/6a46/).
plt.rcParams.update({'legend.fontsize': 6, 'legend.frameon': False}) plt.rcParams.update({"legend.fontsize": 6, "legend.frameon": False})
fig = plt.figure(figsize=(fwid, fhgt), facecolor='w') fig = plt.figure(figsize=(fwid, fhgt), facecolor="w")
fig.subplots_adjust(wspace=0.25, hspace=0.25, left=flft, right=frgt, fig.subplots_adjust(
bottom=fbot, top=ftop) wspace=0.25, hspace=0.25, left=flft, right=frgt, bottom=fbot, top=ftop
)
# first plot # first plot
ax = fig.add_subplot(2, 2, 1) ax = fig.add_subplot(2, 2, 1)
# axes limits # axes limits
@ -315,27 +361,56 @@ def run():
ax.set_ylim(-50, -10) ax.set_ylim(-50, -10)
for idx in range(5): for idx in range(5):
# layer 1 # layer 1
ax.plot(xcell, zeta[idx, 0, 30, :], drawstyle='steps-mid', ax.plot(
linewidth=0.5, color=cc[idx], xcell,
label='{:2d} years'.format(years[idx])) zeta[idx, 0, 30, :],
drawstyle="steps-mid",
linewidth=0.5,
color=cc[idx],
label="{:2d} years".format(years[idx]),
)
# layer 2 # layer 2
ax.plot(xcell, zeta[idx, 1, 30, :], drawstyle='steps-mid', ax.plot(
linewidth=0.5, color=cc[idx], label='_None') xcell,
ax.plot([-1500, 1500], [-30, -30], color='k', linewidth=1.0) zeta[idx, 1, 30, :],
drawstyle="steps-mid",
linewidth=0.5,
color=cc[idx],
label="_None",
)
ax.plot([-1500, 1500], [-30, -30], color="k", linewidth=1.0)
# legend # legend
plt.legend(loc='lower left') plt.legend(loc="lower left")
# axes labels and text # axes labels and text
ax.set_xlabel('Horizontal distance, in meters') ax.set_xlabel("Horizontal distance, in meters")
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
ax.text(0.025, .55, 'Layer 1', transform=ax.transAxes, va='center', ax.text(
ha='left', 0.025,
size='7') 0.55,
ax.text(0.025, .45, 'Layer 2', transform=ax.transAxes, va='center', "Layer 1",
ha='left', transform=ax.transAxes,
size='7') va="center",
ax.text(0.975, .1, 'Recharge conditions', transform=ax.transAxes, ha="left",
va='center', size="7",
ha='right', size='8') )
ax.text(
0.025,
0.45,
"Layer 2",
transform=ax.transAxes,
va="center",
ha="left",
size="7",
)
ax.text(
0.975,
0.1,
"Recharge conditions",
transform=ax.transAxes,
va="center",
ha="right",
size="8",
)
# second plot # second plot
ax = fig.add_subplot(2, 2, 2) ax = fig.add_subplot(2, 2, 2)
@ -344,26 +419,56 @@ def run():
ax.set_ylim(-50, -10) ax.set_ylim(-50, -10)
for idx in range(5, len(years) - 1): for idx in range(5, len(years) - 1):
# layer 1 # layer 1
ax.plot(xcell, zeta[idx, 0, 30, :], drawstyle='steps-mid', ax.plot(
linewidth=0.5, color=cc[idx - 5], xcell,
label='{:2d} years'.format(years[idx])) zeta[idx, 0, 30, :],
drawstyle="steps-mid",
linewidth=0.5,
color=cc[idx - 5],
label="{:2d} years".format(years[idx]),
)
# layer 2 # layer 2
ax.plot(xcell, zeta[idx, 1, 30, :], drawstyle='steps-mid', ax.plot(
linewidth=0.5, color=cc[idx - 5], label='_None') xcell,
ax.plot([-1500, 1500], [-30, -30], color='k', linewidth=1.0) zeta[idx, 1, 30, :],
drawstyle="steps-mid",
linewidth=0.5,
color=cc[idx - 5],
label="_None",
)
ax.plot([-1500, 1500], [-30, -30], color="k", linewidth=1.0)
# legend # legend
plt.legend(loc='lower left') plt.legend(loc="lower left")
# axes labels and text # axes labels and text
ax.set_xlabel('Horizontal distance, in meters') ax.set_xlabel("Horizontal distance, in meters")
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
ax.text(0.025, .55, 'Layer 1', transform=ax.transAxes, va='center', ax.text(
ha='left', 0.025,
size='7') 0.55,
ax.text(0.025, .45, 'Layer 2', transform=ax.transAxes, va='center', "Layer 1",
ha='left', transform=ax.transAxes,
size='7') va="center",
ax.text(0.975, .1, 'Freshwater well withdrawal', transform=ax.transAxes, ha="left",
va='center', ha='right', size='8') size="7",
)
ax.text(
0.025,
0.45,
"Layer 2",
transform=ax.transAxes,
va="center",
ha="left",
size="7",
)
ax.text(
0.975,
0.1,
"Freshwater well withdrawal",
transform=ax.transAxes,
va="center",
ha="right",
size="8",
)
# third plot # third plot
ax = fig.add_subplot(2, 2, 3) ax = fig.add_subplot(2, 2, 3)
@ -372,64 +477,117 @@ def run():
ax.set_ylim(-50, -10) ax.set_ylim(-50, -10)
for idx in range(5, len(years) - 1): for idx in range(5, len(years) - 1):
# layer 1 # layer 1
ax.plot(xcell, zeta2[idx, 0, 30, :], drawstyle='steps-mid', ax.plot(
linewidth=0.5, color=cc[idx - 5], xcell,
label='{:2d} years'.format(years[idx])) zeta2[idx, 0, 30, :],
drawstyle="steps-mid",
linewidth=0.5,
color=cc[idx - 5],
label="{:2d} years".format(years[idx]),
)
# layer 2 # layer 2
ax.plot(xcell, zeta2[idx, 1, 30, :], drawstyle='steps-mid', ax.plot(
linewidth=0.5, color=cc[idx - 5], label='_None') xcell,
ax.plot([-1500, 1500], [-30, -30], color='k', linewidth=1.0) zeta2[idx, 1, 30, :],
drawstyle="steps-mid",
linewidth=0.5,
color=cc[idx - 5],
label="_None",
)
ax.plot([-1500, 1500], [-30, -30], color="k", linewidth=1.0)
# legend # legend
plt.legend(loc='lower left') plt.legend(loc="lower left")
# axes labels and text # axes labels and text
ax.set_xlabel('Horizontal distance, in meters') ax.set_xlabel("Horizontal distance, in meters")
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
ax.text(0.025, .55, 'Layer 1', transform=ax.transAxes, va='center', ax.text(
ha='left', 0.025,
size='7') 0.55,
ax.text(0.025, .45, 'Layer 2', transform=ax.transAxes, va='center', "Layer 1",
ha='left', transform=ax.transAxes,
size='7') va="center",
ax.text(0.975, .1, 'Freshwater and saltwater\nwell withdrawals', ha="left",
transform=ax.transAxes, size="7",
va='center', ha='right', size='8') )
ax.text(
0.025,
0.45,
"Layer 2",
transform=ax.transAxes,
va="center",
ha="left",
size="7",
)
ax.text(
0.975,
0.1,
"Freshwater and saltwater\nwell withdrawals",
transform=ax.transAxes,
va="center",
ha="right",
size="8",
)
# fourth plot # fourth plot
ax = fig.add_subplot(2, 2, 4) ax = fig.add_subplot(2, 2, 4)
# axes limits # axes limits
ax.set_xlim(0, 30) ax.set_xlim(0, 30)
ax.set_ylim(-50, -10) ax.set_ylim(-50, -10)
t = zobs['TOTIM'][999:] / 365 - 200. t = zobs["TOTIM"][999:] / 365 - 200.0
tz2 = zobs['layer1_001'][999:] tz2 = zobs["layer1_001"][999:]
tz3 = zobs2['layer1_001'][999:] tz3 = zobs2["layer1_001"][999:]
for i in range(len(t)): for i in range(len(t)):
if zobs['layer2_001'][i + 999] < -30. - 0.1: if zobs["layer2_001"][i + 999] < -30.0 - 0.1:
tz2[i] = zobs['layer2_001'][i + 999] tz2[i] = zobs["layer2_001"][i + 999]
if zobs2['layer2_001'][i + 999] < 20. - 0.1: if zobs2["layer2_001"][i + 999] < 20.0 - 0.1:
tz3[i] = zobs2['layer2_001'][i + 999] tz3[i] = zobs2["layer2_001"][i + 999]
ax.plot(t, tz2, linestyle='solid', color='r', linewidth=0.75, ax.plot(
label='Freshwater well') t,
ax.plot(t, tz3, linestyle='dotted', color='r', linewidth=0.75, tz2,
label='Freshwater and saltwater well') linestyle="solid",
ax.plot([0, 30], [-30, -30], 'k', linewidth=1.0, label='_None') color="r",
linewidth=0.75,
label="Freshwater well",
)
ax.plot(
t,
tz3,
linestyle="dotted",
color="r",
linewidth=0.75,
label="Freshwater and saltwater well",
)
ax.plot([0, 30], [-30, -30], "k", linewidth=1.0, label="_None")
# legend # legend
leg = plt.legend(loc='lower right', numpoints=1) leg = plt.legend(loc="lower right", numpoints=1)
# axes labels and text # axes labels and text
ax.set_xlabel('Time, in years') ax.set_xlabel("Time, in years")
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
ax.text(0.025, .55, 'Layer 1', transform=ax.transAxes, va='center', ax.text(
ha='left', 0.025,
size='7') 0.55,
ax.text(0.025, .45, 'Layer 2', transform=ax.transAxes, va='center', "Layer 1",
ha='left', transform=ax.transAxes,
size='7') va="center",
ha="left",
size="7",
)
ax.text(
0.025,
0.45,
"Layer 2",
transform=ax.transAxes,
va="center",
ha="left",
size="7",
)
outfig = os.path.join(workspace, 'Figure09_swi2ex4.{0}'.format(fext)) outfig = os.path.join(workspace, "Figure09_swi2ex4.{0}".format(fext))
fig.savefig(outfig, dpi=300) fig.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
return 0 return 0
if __name__ == '__main__': if __name__ == "__main__":
success = run() success = run()

451
examples/scripts/flopy_swi2_ex5.py
View File

@ -12,57 +12,59 @@ import flopy
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
# --modify default matplotlib settings # --modify default matplotlib settings
updates = {'font.family': ['Univers 57 Condensed', 'Arial'], updates = {
'mathtext.default': 'regular', "font.family": ["Univers 57 Condensed", "Arial"],
'pdf.compression': 0, "mathtext.default": "regular",
'pdf.fonttype': 42, "pdf.compression": 0,
'legend.fontsize': 7, "pdf.fonttype": 42,
'axes.labelsize': 8, "legend.fontsize": 7,
'xtick.labelsize': 7, "axes.labelsize": 8,
'ytick.labelsize': 7} "xtick.labelsize": 7,
"ytick.labelsize": 7,
}
plt.rcParams.update(updates) plt.rcParams.update(updates)
def run(): def run():
workspace = 'swiex5' workspace = "swiex5"
if not os.path.isdir(workspace): if not os.path.isdir(workspace):
os.mkdir(workspace) os.mkdir(workspace)
cleanFiles = False cleanFiles = False
skipRuns = False skipRuns = False
fext = 'png' fext = "png"
narg = len(sys.argv) narg = len(sys.argv)
iarg = 0 iarg = 0
if narg > 1: if narg > 1:
while iarg < narg - 1: while iarg < narg - 1:
iarg += 1 iarg += 1
basearg = sys.argv[iarg].lower() basearg = sys.argv[iarg].lower()
if basearg == '--clean': if basearg == "--clean":
cleanFiles = True cleanFiles = True
elif basearg == '--skipruns': elif basearg == "--skipruns":
skipRuns = True skipRuns = True
elif basearg == '--pdf': elif basearg == "--pdf":
fext = 'pdf' fext = "pdf"
dirs = [os.path.join(workspace, 'SWI2'), os.path.join(workspace, 'SEAWAT')] dirs = [os.path.join(workspace, "SWI2"), os.path.join(workspace, "SEAWAT")]
if cleanFiles: if cleanFiles:
print('cleaning all files') print("cleaning all files")
print('excluding *.py files') print("excluding *.py files")
file_dict = collections.OrderedDict() file_dict = collections.OrderedDict()
file_dict[0] = os.listdir(dirs[0]) file_dict[0] = os.listdir(dirs[0])
file_dict[1] = os.listdir(dirs[1]) file_dict[1] = os.listdir(dirs[1])
file_dict[-1] = os.listdir(workspace) file_dict[-1] = os.listdir(workspace)
for key, files in list(file_dict.items()): for key, files in list(file_dict.items()):
pth = '.' pth = "."
if key >= 0: if key >= 0:
pth = dirs[key] pth = dirs[key]
for f in files: for f in files:
fpth = os.path.join(pth, f) fpth = os.path.join(pth, f)
if os.path.isdir(fpth): if os.path.isdir(fpth):
continue continue
if '.py' != os.path.splitext(f)[1].lower(): if ".py" != os.path.splitext(f)[1].lower():
print(' removing...{}'.format(os.path.basename(f))) print(" removing...{}".format(os.path.basename(f)))
try: try:
os.remove(fpth) os.remove(fpth)
except: except:
@ -82,7 +84,7 @@ def run():
nrow = 1 nrow = 1
ncol = 113 ncol = 113
delr = np.zeros((ncol), np.float) delr = np.zeros((ncol), np.float)
delc = 1. delc = 1.0
r = np.zeros((ncol), np.float) r = np.zeros((ncol), np.float)
x = np.zeros((ncol), np.float) x = np.zeros((ncol), np.float)
edge = np.zeros((ncol), np.float) edge = np.zeros((ncol), np.float)
@ -111,11 +113,11 @@ def run():
for k in range(0, nlay): for k in range(0, nlay):
ibound[k, 0, ncol - 1] = -1 ibound[k, 0, ncol - 1] = -1
bot = np.zeros((nlay, nrow, ncol), np.float) bot = np.zeros((nlay, nrow, ncol), np.float)
dz = 100. / float(nlay - 1) dz = 100.0 / float(nlay - 1)
zall = -np.arange(0, 100 + dz, dz) zall = -np.arange(0, 100 + dz, dz)
zall = np.append(zall, -120.) zall = np.append(zall, -120.0)
tb = -np.arange(dz, 100 + dz, dz) tb = -np.arange(dz, 100 + dz, dz)
tb = np.append(tb, -120.) tb = np.append(tb, -120.0)
for k in range(0, nlay): for k in range(0, nlay):
for i in range(0, ncol): for i in range(0, ncol):
bot[k, 0, i] = tb[k] bot[k, 0, i] = tb[k]
@ -123,8 +125,8 @@ def run():
isource[:, 0, ncol - 1] = 1 isource[:, 0, ncol - 1] = 1
isource[nlay - 1, 0, ncol - 1] = 2 isource[nlay - 1, 0, ncol - 1] = 2
khb = (0.0000000000256 * 1000. * 9.81 / 0.001) * 60 * 60 * 24 khb = (0.0000000000256 * 1000.0 * 9.81 / 0.001) * 60 * 60 * 24
kvb = (0.0000000000100 * 1000. * 9.81 / 0.001) * 60 * 60 * 24 kvb = (0.0000000000100 * 1000.0 * 9.81 / 0.001) * 60 * 60 * 24
ssb = 1e-5 ssb = 1e-5
sszb = 0.2 sszb = 0.2
kh = np.zeros((nlay, nrow, ncol), np.float) kh = np.zeros((nlay, nrow, ncol), np.float)
@ -139,14 +141,14 @@ def run():
ss[k, 0, i] = ssb * f ss[k, 0, i] = ssb * f
ssz[k, 0, i] = sszb * f ssz[k, 0, i] = sszb * f
z = np.ones((nlay), np.float) z = np.ones((nlay), np.float)
z = -100. * z z = -100.0 * z
nwell = 1 nwell = 1
for k in range(0, nlay): for k in range(0, nlay):
if zall[k] > -20. and zall[k + 1] <= -20: if zall[k] > -20.0 and zall[k + 1] <= -20:
nwell = k + 1 nwell = k + 1
print('nlay={} dz={} nwell={}'.format(nlay, dz, nwell)) print("nlay={} dz={} nwell={}".format(nlay, dz, nwell))
wellQ = -2400. wellQ = -2400.0
wellbtm = -20.0 wellbtm = -20.0
wellQpm = wellQ / abs(wellbtm) wellQpm = wellQ / abs(wellbtm)
well_data = {} well_data = {}
@ -174,40 +176,68 @@ def run():
for j in range(0, nstp[i]): for j in range(0, nstp[i]):
icnt += 1 icnt += 1
if icnt == 365: if icnt == 365:
ocspd[(i, j)] = ['save head'] ocspd[(i, j)] = ["save head"]
icnt = 0 icnt = 0
else: else:
ocspd[(i, j)] = [] ocspd[(i, j)] = []
solver2params = {'mxiter': 100, 'iter1': 20, 'npcond': 1, 'zclose': 1.0e-6, solver2params = {
'rclose': 3e-3, 'relax': 1.0, "mxiter": 100,
'nbpol': 2, 'damp': 1.0, 'dampt': 1.0} "iter1": 20,
"npcond": 1,
"zclose": 1.0e-6,
"rclose": 3e-3,
"relax": 1.0,
"nbpol": 2,
"damp": 1.0,
"dampt": 1.0,
}
# --create model file and run model # --create model file and run model
modelname = 'swi2ex5' modelname = "swi2ex5"
mf_name = 'mf2005' mf_name = "mf2005"
if not skipRuns: if not skipRuns:
ml = flopy.modflow.Modflow(modelname, version='mf2005', ml = flopy.modflow.Modflow(
exe_name=mf_name, modelname, version="mf2005", exe_name=mf_name, model_ws=dirs[0]
model_ws=dirs[0]) )
discret = flopy.modflow.ModflowDis(ml, nrow=nrow, ncol=ncol, nlay=nlay, discret = flopy.modflow.ModflowDis(
delr=delr, delc=delc, top=0, ml,
botm=bot, nrow=nrow,
laycbd=0, nper=nper, perlen=perlen, ncol=ncol,
nstp=nstp, steady=steady) nlay=nlay,
delr=delr,
delc=delc,
top=0,
botm=bot,
laycbd=0,
nper=nper,
perlen=perlen,
nstp=nstp,
steady=steady,
)
bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=ihead) bas = flopy.modflow.ModflowBas(ml, ibound=ibound, strt=ihead)
lpf = flopy.modflow.ModflowLpf(ml, hk=kh, vka=kv, ss=ss, sy=ssz, lpf = flopy.modflow.ModflowLpf(
vkcb=0, ml, hk=kh, vka=kv, ss=ss, sy=ssz, vkcb=0, laytyp=0, layavg=1
laytyp=0, layavg=1) )
wel = flopy.modflow.ModflowWel(ml, stress_period_data=well_data) wel = flopy.modflow.ModflowWel(ml, stress_period_data=well_data)
swi = flopy.modflow.ModflowSwi2(ml, iswizt=55, npln=1, istrat=1, swi = flopy.modflow.ModflowSwi2(
toeslope=0.025, tipslope=0.025, ml,
nu=[0, 0.025], zeta=z, ssz=ssz, iswizt=55,
isource=isource, nsolver=2, npln=1,
solver2params=solver2params) istrat=1,
toeslope=0.025,
tipslope=0.025,
nu=[0, 0.025],
zeta=z,
ssz=ssz,
isource=isource,
nsolver=2,
solver2params=solver2params,
)
oc = flopy.modflow.ModflowOc(ml, stress_period_data=ocspd) oc = flopy.modflow.ModflowOc(ml, stress_period_data=ocspd)
pcg = flopy.modflow.ModflowPcg(ml, hclose=1.0e-6, rclose=3.0e-3, pcg = flopy.modflow.ModflowPcg(
mxiter=100, iter1=50) ml, hclose=1.0e-6, rclose=3.0e-3, mxiter=100, iter1=50
)
# --write the modflow files # --write the modflow files
ml.write_input() ml.write_input()
m = ml.run_model(silent=False) m = ml.run_model(silent=False)
@ -216,53 +246,53 @@ def run():
get_stp = [364, 729, 1094, 1459, 364, 729, 1094, 1459] get_stp = [364, 729, 1094, 1459, 364, 729, 1094, 1459]
get_per = [0, 0, 0, 0, 1, 1, 1, 1] get_per = [0, 0, 0, 0, 1, 1, 1, 1]
nswi_times = len(get_per) nswi_times = len(get_per)
zetafile = os.path.join(dirs[0], '{}.zta'.format(modelname)) zetafile = os.path.join(dirs[0], "{}.zta".format(modelname))
zobj = flopy.utils.CellBudgetFile(zetafile) zobj = flopy.utils.CellBudgetFile(zetafile)
zeta = [] zeta = []
for kk in zip(get_stp, get_per): for kk in zip(get_stp, get_per):
zeta.append(zobj.get_data(kstpkper=kk, text='ZETASRF 1')[0]) zeta.append(zobj.get_data(kstpkper=kk, text="ZETASRF 1")[0])
zeta = np.array(zeta) zeta = np.array(zeta)
# --seawat input - redefine input data that differ from SWI2 # --seawat input - redefine input data that differ from SWI2
nlay_swt = 120 nlay_swt = 120
# --mt3d print times # --mt3d print times
timprs = (np.arange(8) + 1) * 365. timprs = (np.arange(8) + 1) * 365.0
nprs = len(timprs) nprs = len(timprs)
# -- # --
ndecay = 4 ndecay = 4
ibound = np.ones((nlay_swt, nrow, ncol), 'int') ibound = np.ones((nlay_swt, nrow, ncol), "int")
for k in range(0, nlay_swt): for k in range(0, nlay_swt):
ibound[k, 0, ncol - 1] = -1 ibound[k, 0, ncol - 1] = -1
bot = np.zeros((nlay_swt, nrow, ncol), np.float) bot = np.zeros((nlay_swt, nrow, ncol), np.float)
zall = [0, -20., -40., -60., -80., -100., -120.] zall = [0, -20.0, -40.0, -60.0, -80.0, -100.0, -120.0]
dz = 120. / nlay_swt dz = 120.0 / nlay_swt
tb = np.arange(nlay_swt) * -dz - dz tb = np.arange(nlay_swt) * -dz - dz
sconc = np.zeros((nlay_swt, nrow, ncol), np.float) sconc = np.zeros((nlay_swt, nrow, ncol), np.float)
icbund = np.ones((nlay_swt, nrow, ncol), np.int) icbund = np.ones((nlay_swt, nrow, ncol), np.int)
strt = np.zeros((nlay_swt, nrow, ncol), np.float) strt = np.zeros((nlay_swt, nrow, ncol), np.float)
pressure = 0. pressure = 0.0
g = 9.81 g = 9.81
z = - dz / 2. # cell center z = -dz / 2.0 # cell center
for k in range(0, nlay_swt): for k in range(0, nlay_swt):
for i in range(0, ncol): for i in range(0, ncol):
bot[k, 0, i] = tb[k] bot[k, 0, i] = tb[k]
if bot[k, 0, 0] >= -100.: if bot[k, 0, 0] >= -100.0:
sconc[k, 0, :] = 0. / 3. * .025 * 1000. / .7143 sconc[k, 0, :] = 0.0 / 3.0 * 0.025 * 1000.0 / 0.7143
else: else:
sconc[k, 0, :] = 3. / 3. * .025 * 1000. / .7143 sconc[k, 0, :] = 3.0 / 3.0 * 0.025 * 1000.0 / 0.7143
icbund[k, 0, -1] = -1 icbund[k, 0, -1] = -1
dense = 1000. + 0.7143 * sconc[k, 0, 0] dense = 1000.0 + 0.7143 * sconc[k, 0, 0]
pressure += 0.5 * dz * dense * g pressure += 0.5 * dz * dense * g
if k > 0: if k > 0:
z = z - dz z = z - dz
denseup = 1000. + 0.7143 * sconc[k - 1, 0, 0] denseup = 1000.0 + 0.7143 * sconc[k - 1, 0, 0]
pressure += 0.5 * dz * denseup * g pressure += 0.5 * dz * denseup * g
strt[k, 0, :] = z + pressure / dense / g strt[k, 0, :] = z + pressure / dense / g
# print z, pressure, strt[k, 0, 0], sconc[k, 0, 0] # print z, pressure, strt[k, 0, 0], sconc[k, 0, 0]
khb = (0.0000000000256 * 1000. * 9.81 / 0.001) * 60 * 60 * 24 khb = (0.0000000000256 * 1000.0 * 9.81 / 0.001) * 60 * 60 * 24
kvb = (0.0000000000100 * 1000. * 9.81 / 0.001) * 60 * 60 * 24 kvb = (0.0000000000100 * 1000.0 * 9.81 / 0.001) * 60 * 60 * 24
ssb = 1e-5 ssb = 1e-5
sszb = 0.2 sszb = 0.2
kh = np.zeros((nlay_swt, nrow, ncol), np.float) kh = np.zeros((nlay_swt, nrow, ncol), np.float)
@ -280,12 +310,12 @@ def run():
itype = flopy.mt3d.Mt3dSsm.itype_dict() itype = flopy.mt3d.Mt3dSsm.itype_dict()
nwell = 1 nwell = 1
for k in range(0, nlay_swt): for k in range(0, nlay_swt):
if bot[k, 0, 0] >= -20.: if bot[k, 0, 0] >= -20.0:
nwell = k + 1 nwell = k + 1
print('nlay_swt={} dz={} nwell={}'.format(nlay_swt, dz, nwell)) print("nlay_swt={} dz={} nwell={}".format(nlay_swt, dz, nwell))
well_data = {} well_data = {}
ssm_data = {} ssm_data = {}
wellQ = -2400. wellQ = -2400.0
wellbtm = -20.0 wellbtm = -20.0
wellQpm = wellQ / abs(wellbtm) wellQpm = wellQ / abs(wellbtm)
for ip in range(0, nper): for ip in range(0, nper):
@ -300,59 +330,90 @@ def run():
welllist[iw, 1] = 0 welllist[iw, 1] = 0
welllist[iw, 2] = 0 welllist[iw, 2] = 0
welllist[iw, 3] = q welllist[iw, 3] = q
ssmlist.append([iw, 0, 0, 0., itype['WEL']]) ssmlist.append([iw, 0, 0, 0.0, itype["WEL"]])
well_data[ip] = welllist.copy() well_data[ip] = welllist.copy()
ssm_data[ip] = ssmlist ssm_data[ip] = ssmlist
# Define model name for SEAWAT model # Define model name for SEAWAT model
modelname = 'swi2ex5_swt' modelname = "swi2ex5_swt"
swtexe_name = 'swtv4' swtexe_name = "swtv4"
# Create the MODFLOW model data # Create the MODFLOW model data
if not skipRuns: if not skipRuns:
m = flopy.seawat.Seawat(modelname, exe_name=swtexe_name, m = flopy.seawat.Seawat(
model_ws=dirs[1]) modelname, exe_name=swtexe_name, model_ws=dirs[1]
discret = flopy.modflow.ModflowDis(m, nrow=nrow, ncol=ncol, )
nlay=nlay_swt, discret = flopy.modflow.ModflowDis(
delr=delr, delc=delc, top=0, m,
botm=bot, nrow=nrow,
laycbd=0, nper=nper, perlen=perlen, ncol=ncol,
nstp=nstp, steady=True) nlay=nlay_swt,
delr=delr,
delc=delc,
top=0,
botm=bot,
laycbd=0,
nper=nper,
perlen=perlen,
nstp=nstp,
steady=True,
)
bas = flopy.modflow.ModflowBas(m, ibound=ibound, strt=strt) bas = flopy.modflow.ModflowBas(m, ibound=ibound, strt=strt)
lpf = flopy.modflow.ModflowLpf(m, hk=kh, vka=kv, ss=ss, sy=ssz, vkcb=0, lpf = flopy.modflow.ModflowLpf(
laytyp=0, layavg=1) m, hk=kh, vka=kv, ss=ss, sy=ssz, vkcb=0, laytyp=0, layavg=1
)
wel = flopy.modflow.ModflowWel(m, stress_period_data=well_data) wel = flopy.modflow.ModflowWel(m, stress_period_data=well_data)
oc = flopy.modflow.ModflowOc(m, save_every=365, oc = flopy.modflow.ModflowOc(
save_types=['save head']) m, save_every=365, save_types=["save head"]
pcg = flopy.modflow.ModflowPcg(m, hclose=1.0e-5, rclose=3.0e-3, )
mxiter=100, pcg = flopy.modflow.ModflowPcg(
iter1=50) m, hclose=1.0e-5, rclose=3.0e-3, mxiter=100, iter1=50
)
# Create the basic MT3DMS model data # Create the basic MT3DMS model data
adv = flopy.mt3d.Mt3dAdv(m, mixelm=-1, adv = flopy.mt3d.Mt3dAdv(
percel=0.5, m,
nadvfd=0, mixelm=-1,
# 0 or 1 is upstream; 2 is central in space percel=0.5,
# particle based methods nadvfd=0,
nplane=4, # 0 or 1 is upstream; 2 is central in space
mxpart=1e7, # particle based methods
itrack=2, nplane=4,
dceps=1e-4, mxpart=1e7,
npl=16, itrack=2,
nph=16, dceps=1e-4,
npmin=8, npl=16,
npmax=256) nph=16,
btn = flopy.mt3d.Mt3dBtn(m, icbund=icbund, prsity=ssz, ncomp=1, npmin=8,
sconc=sconc, npmax=256,
ifmtcn=-1, )
chkmas=False, nprobs=10, nprmas=10, dt0=1.0, btn = flopy.mt3d.Mt3dBtn(
ttsmult=1.0, m,
nprs=nprs, timprs=timprs, mxstrn=1e8) icbund=icbund,
dsp = flopy.mt3d.Mt3dDsp(m, al=0., trpt=1., trpv=1., dmcoef=0.) prsity=ssz,
ncomp=1,
sconc=sconc,
ifmtcn=-1,
chkmas=False,
nprobs=10,
nprmas=10,
dt0=1.0,
ttsmult=1.0,
nprs=nprs,
timprs=timprs,
mxstrn=1e8,
)
dsp = flopy.mt3d.Mt3dDsp(m, al=0.0, trpt=1.0, trpv=1.0, dmcoef=0.0)
gcg = flopy.mt3d.Mt3dGcg(m, mxiter=1, iter1=50, isolve=1, cclose=1e-7) gcg = flopy.mt3d.Mt3dGcg(m, mxiter=1, iter1=50, isolve=1, cclose=1e-7)
ssm = flopy.mt3d.Mt3dSsm(m, stress_period_data=ssm_data) ssm = flopy.mt3d.Mt3dSsm(m, stress_period_data=ssm_data)
# Create the SEAWAT model data # Create the SEAWAT model data
vdf = flopy.seawat.SeawatVdf(m, iwtable=0, densemin=0, densemax=0, vdf = flopy.seawat.SeawatVdf(
denseref=1000., denseslp=0.7143, m,
firstdt=1e-3) iwtable=0,
densemin=0,
densemax=0,
denseref=1000.0,
denseslp=0.7143,
firstdt=1e-3,
)
# write seawat files # write seawat files
m.write_input() m.write_input()
@ -361,7 +422,7 @@ def run():
# plot the results # plot the results
# read seawat model data # read seawat model data
ucnfile = os.path.join(dirs[1], 'MT3D001.UCN') ucnfile = os.path.join(dirs[1], "MT3D001.UCN")
uobj = flopy.utils.UcnFile(ucnfile) uobj = flopy.utils.UcnFile(ucnfile)
times = uobj.get_times() times = uobj.get_times()
print(times) print(times)
@ -375,16 +436,16 @@ def run():
# spatial data # spatial data
# swi2 # swi2
bot = np.zeros((1, ncol, nlay), np.float) bot = np.zeros((1, ncol, nlay), np.float)
dz = 100. / float(nlay - 1) dz = 100.0 / float(nlay - 1)
zall = -np.arange(0, 100 + dz, dz) zall = -np.arange(0, 100 + dz, dz)
zall = np.append(zall, -120.) zall = np.append(zall, -120.0)
tb = -np.arange(dz, 100 + dz, dz) tb = -np.arange(dz, 100 + dz, dz)
tb = np.append(tb, -120.) tb = np.append(tb, -120.0)
for k in range(0, nlay): for k in range(0, nlay):
for i in range(0, ncol): for i in range(0, ncol):
bot[0, i, k] = tb[k] bot[0, i, k] = tb[k]
# seawat # seawat
swt_dz = 120. / nlay_swt swt_dz = 120.0 / nlay_swt
swt_tb = np.zeros((nlay_swt), np.float) swt_tb = np.zeros((nlay_swt), np.float)
zc = -swt_dz / 2.0 zc = -swt_dz / 2.0
for klay in range(0, nlay_swt): for klay in range(0, nlay_swt):
@ -398,32 +459,73 @@ def run():
eps = 1.0e-3 eps = 1.0e-3
lc = ['r', 'c', 'g', 'b', 'k'] lc = ["r", "c", "g", "b", "k"]
cfig = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'] cfig = ["A", "B", "C", "D", "E", "F", "G", "H"]
inc = 1.0e-3 inc = 1.0e-3
xsf = plt.figure(figsize=(fwid, fhgt), facecolor='w') xsf = plt.figure(figsize=(fwid, fhgt), facecolor="w")
xsf.subplots_adjust(wspace=0.25, hspace=0.25, left=flft, right=frgt, xsf.subplots_adjust(
bottom=fbot, top=ftop) wspace=0.25, hspace=0.25, left=flft, right=frgt, bottom=fbot, top=ftop
)
# withdrawal and recovery titles # withdrawal and recovery titles
ax = xsf.add_subplot(4, 2, 1) ax = xsf.add_subplot(4, 2, 1)
ax.text(0.0, 1.03, 'Withdrawal', transform=ax.transAxes, va='bottom', ax.text(
ha='left', size='8') 0.0,
1.03,
"Withdrawal",
transform=ax.transAxes,
va="bottom",
ha="left",
size="8",
)
ax = xsf.add_subplot(4, 2, 2) ax = xsf.add_subplot(4, 2, 2)
ax.text(0.0, 1.03, 'Recovery', transform=ax.transAxes, va='bottom', ax.text(
ha='left', 0.0,
size='8') 1.03,
"Recovery",
transform=ax.transAxes,
va="bottom",
ha="left",
size="8",
)
# dummy items for legend # dummy items for legend
ax = xsf.add_subplot(4, 2, 1) ax = xsf.add_subplot(4, 2, 1)
ax.plot([-1, -1], [-1, -1], 'bo', markersize=3, markeredgecolor='blue', ax.plot(
markerfacecolor='None', label='SWI2 interface') [-1, -1],
ax.plot([-1, -1], [-1, -1], color='k', linewidth=0.75, linestyle='solid', [-1, -1],
label='SEAWAT 50% seawater') "bo",
ax.plot([-1, -1], [-1, -1], marker='s', color='k', linewidth=0, markersize=3,
linestyle='none', markeredgecolor='w', markeredgecolor="blue",
markerfacecolor='0.75', label='SEAWAT 5-95% seawater') markerfacecolor="None",
leg = ax.legend(loc='upper left', numpoints=1, ncol=1, labelspacing=0.5, label="SWI2 interface",
borderaxespad=1, handlelength=3) )
ax.plot(
[-1, -1],
[-1, -1],
color="k",
linewidth=0.75,
linestyle="solid",
label="SEAWAT 50% seawater",
)
ax.plot(
[-1, -1],
[-1, -1],
marker="s",
color="k",
linewidth=0,
linestyle="none",
markeredgecolor="w",
markerfacecolor="0.75",
label="SEAWAT 5-95% seawater",
)
leg = ax.legend(
loc="upper left",
numpoints=1,
ncol=1,
labelspacing=0.5,
borderaxespad=1,
handlelength=3,
)
leg._drawFrame = False leg._drawFrame = False
# data items # data items
for itime in range(0, nswi_times): for itime in range(0, nswi_times):
@ -432,8 +534,8 @@ def run():
for icol in range(0, ncol): for icol in range(0, ncol):
for klay in range(0, nlay): for klay in range(0, nlay):
# top and bottom of layer # top and bottom of layer
ztop = float('{0:10.3e}'.format(zall[klay])) ztop = float("{0:10.3e}".format(zall[klay]))
zbot = float('{0:10.3e}'.format(zall[klay + 1])) zbot = float("{0:10.3e}".format(zall[klay + 1]))
# fresh-salt zeta surface # fresh-salt zeta surface
zt = zeta[itime, klay, 0, icol] zt = zeta[itime, klay, 0, icol]
if (ztop - zt) > eps: if (ztop - zt) > eps:
@ -447,29 +549,56 @@ def run():
isp = (ic * 2) + 2 isp = (ic * 2) + 2
ax = xsf.add_subplot(4, 2, isp) ax = xsf.add_subplot(4, 2, isp)
# figure title # figure title
ax.text(-0.15, 1.025, cfig[itime], transform=ax.transAxes, va='center', ax.text(
ha='center', size='8') -0.15,
1.025,
cfig[itime],
transform=ax.transAxes,
va="center",
ha="center",
size="8",
)
# swi2 # swi2
ax.plot(x, zs, 'bo', markersize=3, markeredgecolor='blue', ax.plot(
markerfacecolor='None', label='_None') x,
zs,
"bo",
markersize=3,
markeredgecolor="blue",
markerfacecolor="None",
label="_None",
)
# seawat # seawat
sc = ax.contour(X, Z, conc[itime, :, :], levels=[17.5], colors='k', sc = ax.contour(
linestyles='solid', linewidths=0.75, zorder=30) X,
cc = ax.contourf(X, Z, conc[itime, :, :], levels=[0.0, 1.75, 33.250], Z,
colors=['w', '0.75', 'w']) conc[itime, :, :],
levels=[17.5],
colors="k",
linestyles="solid",
linewidths=0.75,
zorder=30,
)
cc = ax.contourf(
X,
Z,
conc[itime, :, :],
levels=[0.0, 1.75, 33.250],
colors=["w", "0.75", "w"],
)
# set graph limits # set graph limits
ax.set_xlim(0, 500) ax.set_xlim(0, 500)
ax.set_ylim(-100, -65) ax.set_ylim(-100, -65)
if itime < ndecay: if itime < ndecay:
ax.set_ylabel('Elevation, in meters') ax.set_ylabel("Elevation, in meters")
# x labels # x labels
ax = xsf.add_subplot(4, 2, 7) ax = xsf.add_subplot(4, 2, 7)
ax.set_xlabel('Horizontal distance, in meters') ax.set_xlabel("Horizontal distance, in meters")
ax = xsf.add_subplot(4, 2, 8) ax = xsf.add_subplot(4, 2, 8)
ax.set_xlabel('Horizontal distance, in meters') ax.set_xlabel("Horizontal distance, in meters")
# simulation time titles # simulation time titles
for itime in range(0, nswi_times): for itime in range(0, nswi_times):
@ -483,19 +612,25 @@ def run():
ax = xsf.add_subplot(4, 2, isp) ax = xsf.add_subplot(4, 2, isp)
iyr = itime + 1 iyr = itime + 1
if iyr > 1: if iyr > 1:
ctxt = '{} years'.format(iyr) ctxt = "{} years".format(iyr)
else: else:
ctxt = '{} year'.format(iyr) ctxt = "{} year".format(iyr)
ax.text(0.95, 0.925, ctxt, transform=ax.transAxes, va='top', ax.text(
ha='right', 0.95,
size='8') 0.925,
ctxt,
transform=ax.transAxes,
va="top",
ha="right",
size="8",
)
outfig = os.path.join(workspace, 'Figure11_swi2ex5.{0}'.format(fext)) outfig = os.path.join(workspace, "Figure11_swi2ex5.{0}".format(fext))
xsf.savefig(outfig, dpi=300) xsf.savefig(outfig, dpi=300)
print('created...', outfig) print("created...", outfig)
return 0 return 0
if __name__ == '__main__': if __name__ == "__main__":
success = run() success = run()

2
flopy/discretization/structuredgrid.py
View File

@ -436,7 +436,7 @@ class StructuredGrid(Grid):
@property @property
def grid_lines(self): def grid_lines(self):
""" """
Get the grid lines as a list Get the grid lines as a list
""" """
# get edges initially in model coordinates # get edges initially in model coordinates

6
flopy/export/metadata.py
View File

@ -39,8 +39,10 @@ class acdd:
self.model = model self.model = model
self.model_grid = model.modelgrid self.model_grid = model.modelgrid
self.model_time = model.modeltime self.model_time = model.modeltime
self.sciencebase_url = "https://www.sciencebase.gov/catalog/item/{}".format( self.sciencebase_url = (
sciencebase_id "https://www.sciencebase.gov/catalog/item/{}".format(
sciencebase_id
)
) )
self.sb = self.get_sciencebase_metadata(sciencebase_id) self.sb = self.get_sciencebase_metadata(sciencebase_id)
if self.sb is None: if self.sb is None:

23
flopy/export/netcdf.py
View File

@ -85,14 +85,18 @@ class Logger(object):
+ "\n" + "\n"
) )
if self.echo: if self.echo:
print(s,) print(
s,
)
if self.filename: if self.filename:
self.f.write(s) self.f.write(s)
self.items.pop(phrase) self.items.pop(phrase)
else: else:
s = str(t) + " starting: " + str(phrase) + "\n" s = str(t) + " starting: " + str(phrase) + "\n"
if self.echo: if self.echo:
print(s,) print(
s,
)
if self.filename: if self.filename:
self.f.write(s) self.f.write(s)
self.items[phrase] = copy.deepcopy(t) self.items[phrase] = copy.deepcopy(t)
@ -109,7 +113,9 @@ class Logger(object):
""" """
s = str(datetime.now()) + " WARNING: " + message + "\n" s = str(datetime.now()) + " WARNING: " + message + "\n"
if self.echo: if self.echo:
print(s,) print(
s,
)
if self.filename: if self.filename:
self.f.write(s) self.f.write(s)
return return
@ -624,8 +630,7 @@ class NetCdf(object):
self.log("processing variable {0}".format(vname)) self.log("processing variable {0}".format(vname))
def _dt_str(self, dt): def _dt_str(self, dt):
""" for datetime to string for year < 1900 """for datetime to string for year < 1900"""
"""
dt_str = "{0:04d}-{1:02d}-{2:02d}T{3:02d}:{4:02d}:{5:02}Z".format( dt_str = "{0:04d}-{1:02d}-{2:02d}T{3:02d}:{4:02d}:{5:02}Z".format(
dt.year, dt.month, dt.day, dt.hour, dt.minute, dt.second dt.year, dt.month, dt.day, dt.hour, dt.minute, dt.second
) )
@ -655,7 +660,7 @@ class NetCdf(object):
def _initialize_attributes(self): def _initialize_attributes(self):
"""private method to initial the attributes """private method to initial the attributes
of the NetCdf instance of the NetCdf instance
""" """
assert ( assert (
"nc" not in self.__dict__.keys() "nc" not in self.__dict__.keys()
@ -707,8 +712,8 @@ class NetCdf(object):
self.nc = None self.nc = None
def initialize_geometry(self): def initialize_geometry(self):
""" initialize the geometric information """initialize the geometric information
needed for the netcdf file needed for the netcdf file
""" """
try: try:
import pyproj import pyproj
@ -1354,7 +1359,7 @@ class NetCdf(object):
return var return var
def add_global_attributes(self, attr_dict): def add_global_attributes(self, attr_dict):
""" add global attribute to an initialized file """add global attribute to an initialized file
Parameters Parameters
---------- ----------

3
flopy/export/shapefile_utils.py
View File

@ -418,8 +418,7 @@ def enforce_10ch_limit(names):
def get_pyshp_field_info(dtypename): def get_pyshp_field_info(dtypename):
"""Get pyshp dtype information for a given numpy dtype. """Get pyshp dtype information for a given numpy dtype."""
"""
fields = { fields = {
"int": ("N", 18, 0), "int": ("N", 18, 0),
"<i": ("N", 18, 0), "<i": ("N", 18, 0),

6
flopy/mf6/data/mfdatalist.py
View File

@ -931,9 +931,9 @@ class MFList(mfdata.MFMultiDimVar, DataListInterface):
keystr_struct, mfstructure.MFDataStructure keystr_struct, mfstructure.MFDataStructure
): ):
# data items following keystring # data items following keystring
ks_structs = keystr_struct.data_item_structures[ ks_structs = (
1: keystr_struct.data_item_structures[1:]
] )
else: else:
# key string stands alone # key string stands alone
ks_structs = [keystr_struct] ks_structs = [keystr_struct]

6
flopy/mf6/data/mfdatastorage.py
View File

@ -2423,8 +2423,10 @@ class DataStorage(object):
model_dim = self.data_dimensions.get_model_dim( model_dim = self.data_dimensions.get_model_dim(
None None
) )
expression_array = model_dim.build_shape_expression( expression_array = (
data_item.shape model_dim.build_shape_expression(
data_item.shape
)
) )
if ( if (
isinstance(expression_array, list) isinstance(expression_array, list)

12
flopy/mf6/data/mffileaccess.py
View File

@ -1652,9 +1652,9 @@ class MFFileAccessList(MFFileAccess):
) )
) )
break break
data_item_ks = data_item.keystring_dict[ data_item_ks = (
name_data data_item.keystring_dict[name_data]
] )
if data_item_ks == 0: if data_item_ks == 0:
comment = ( comment = (
"Could not find " "Could not find "
@ -1688,9 +1688,9 @@ class MFFileAccessList(MFFileAccess):
if data_item.name in self._temp_dict: if data_item.name in self._temp_dict:
# used cached data item for # used cached data item for
# performance # performance
keyword_data_item = self._temp_dict[ keyword_data_item = (
data_item.name self._temp_dict[data_item.name]
] )
else: else:
keyword_data_item = deepcopy( keyword_data_item = deepcopy(
data_item data_item

2
flopy/mf6/modflow/mfgnc.py
View File

@ -92,7 +92,7 @@ class ModflowGnc(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfchd.py
View File

@ -96,7 +96,7 @@ class ModflowGwfchd(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfcsub.py
View File

@ -266,7 +266,7 @@ class ModflowGwfcsub(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfdis.py
View File

@ -74,7 +74,7 @@ class ModflowGwfdis(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfdisu.py
View File

@ -168,7 +168,7 @@ class ModflowGwfdisu(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfdisv.py
View File

@ -102,7 +102,7 @@ class ModflowGwfdisv(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfdrn.py
View File

@ -118,7 +118,7 @@ class ModflowGwfdrn(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfevt.py
View File

@ -136,7 +136,7 @@ class ModflowGwfevt(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfevta.py
View File

@ -95,7 +95,7 @@ class ModflowGwfevta(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfghb.py
View File

@ -107,7 +107,7 @@ class ModflowGwfghb(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfgnc.py
View File

@ -92,7 +92,7 @@ class ModflowGwfgnc(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfgwf.py
View File

@ -144,7 +144,7 @@ class ModflowGwfgwf(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfhfb.py
View File

@ -60,7 +60,7 @@ class ModflowGwfhfb(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfic.py
View File

@ -33,7 +33,7 @@ class ModflowGwfic(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwflak.py
View File

@ -408,7 +408,7 @@ class ModflowGwflak(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfmaw.py
View File

@ -335,7 +335,7 @@ class ModflowGwfmaw(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfmvr.py
View File

@ -102,7 +102,7 @@ class ModflowGwfmvr(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfnam.py
View File

@ -64,7 +64,7 @@ class ModflowGwfnam(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfnpf.py
View File

@ -188,7 +188,7 @@ class ModflowGwfnpf(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfoc.py
View File

@ -81,7 +81,7 @@ class ModflowGwfoc(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfrch.py
View File

@ -102,7 +102,7 @@ class ModflowGwfrch(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfrcha.py
View File

@ -95,7 +95,7 @@ class ModflowGwfrcha(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfriv.py
View File

@ -108,7 +108,7 @@ class ModflowGwfriv(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfsfr.py
View File

@ -368,7 +368,7 @@ class ModflowGwfsfr(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfsto.py
View File

@ -55,7 +55,7 @@ class ModflowGwfsto(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfuzf.py
View File

@ -235,7 +235,7 @@ class ModflowGwfuzf(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfgwfwel.py
View File

@ -112,7 +112,7 @@ class ModflowGwfwel(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfims.py
View File

@ -325,7 +325,7 @@ class ModflowIms(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfmvr.py
View File

@ -102,7 +102,7 @@ class ModflowMvr(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfnam.py
View File

@ -67,7 +67,7 @@ class ModflowNam(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mftdis.py
View File

@ -44,7 +44,7 @@ class ModflowTdis(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfutllaktab.py
View File

@ -41,7 +41,7 @@ class ModflowUtllaktab(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfutlobs.py
View File

@ -62,7 +62,7 @@ class ModflowUtlobs(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfutltas.py
View File

@ -36,7 +36,7 @@ class ModflowUtltas(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/modflow/mfutlts.py
View File

@ -46,7 +46,7 @@ class ModflowUtlts(mfpackage.MFPackage):
Package name for this package. Package name for this package.
parent_file : MFPackage parent_file : MFPackage
Parent package file that references this package. Only needed for Parent package file that references this package. Only needed for
utility packages (mfutl*). For example, mfutllaktab package must have utility packages (mfutl*). For example, mfutllaktab package must have
a mfgwflak package parent_file. a mfgwflak package parent_file.
""" """

2
flopy/mf6/utils/binaryfile_utils.py
View File

@ -71,7 +71,7 @@ class MFOutputRequester:
-------- --------
>>> data = MFOutputRequester(mfdict, path, key) >>> data = MFOutputRequester(mfdict, path, key)
>>> data.querybinarydata >>> data.querybinarydata
""" """
def __init__(self, mfdict, path, key): def __init__(self, mfdict, path, key):
self.path = path self.path = path

2
flopy/mf6/utils/mfobservation.py
View File

@ -62,7 +62,7 @@ class Observations:
idx = (int), (slice(start, stop)) integer or slice of data to be idx = (int), (slice(start, stop)) integer or slice of data to be
returned. corresponds to kstp*kper - 1 returned. corresponds to kstp*kper - 1
totim = (float) model time value to return data from totim = (float) model time value to return data from
list_records(): prints a list of all valid record names contained within list_records(): prints a list of all valid record names contained within
the Obs.out file the Obs.out file
get_times(): (list) returns list of time values contained in Obs.out get_times(): (list) returns list of time values contained in Obs.out

15
flopy/mf6/utils/reference.py
View File

@ -66,7 +66,7 @@ class StructuredSpatialReference(object):
xul and yul can be explicitly (re)set after SpatialReference xul and yul can be explicitly (re)set after SpatialReference
instantiation, but only before any of the other attributes and methods are instantiation, but only before any of the other attributes and methods are
accessed accessed
""" """
def __init__( def __init__(
@ -259,7 +259,7 @@ class StructuredSpatialReference(object):
def set_spatialreference(self, xul=None, yul=None, rotation=0.0): def set_spatialreference(self, xul=None, yul=None, rotation=0.0):
""" """
set spatial reference - can be called from model instance set spatial reference - can be called from model instance
""" """
# Set origin and rotation # Set origin and rotation
@ -405,7 +405,7 @@ class StructuredSpatialReference(object):
def get_grid_lines(self): def get_grid_lines(self):
""" """
get the grid lines as a list get the grid lines as a list
""" """
xmin = self.xedge[0] xmin = self.xedge[0]
xmax = self.xedge[-1] xmax = self.xedge[-1]
@ -487,8 +487,7 @@ class StructuredSpatialReference(object):
return yedge return yedge
def write_gridSpec(self, filename): def write_gridSpec(self, filename):
""" write a PEST-style grid specification file """write a PEST-style grid specification file"""
"""
f = open(filename, "w") f = open(filename, "w")
f.write( f.write(
"{0:10d} {1:10d}\n".format(self.delc.shape[0], self.delr.shape[0]) "{0:10d} {1:10d}\n".format(self.delc.shape[0], self.delr.shape[0])
@ -722,9 +721,9 @@ class VertexSpatialReference(object):
def set_spatialreference(self, xadj=0.0, yadj=0.0, rotation=0.0): def set_spatialreference(self, xadj=0.0, yadj=0.0, rotation=0.0):
""" """
set spatial reference - can be called from model instance set spatial reference - can be called from model instance
xadj, yadj should be named xadj, yadj since they represent an xadj, yadj should be named xadj, yadj since they represent an
adjustment factor adjustment factor
""" """
# Set origin and rotation # Set origin and rotation

12
flopy/modflow/mfbct.py
View File

@ -65,7 +65,11 @@ class ModflowBct(Package):
self.izod = izod self.izod = izod
self.ifod = ifod self.ifod = ifod
self.icbund = Util3d( self.icbund = Util3d(
model, (nlay, nrow, ncol), np.float32, icbund, "icbund", model,
(nlay, nrow, ncol),
np.float32,
icbund,
"icbund",
) )
self.porosity = Util3d( self.porosity = Util3d(
model, (nlay, nrow, ncol), np.float32, porosity, "porosity" model, (nlay, nrow, ncol), np.float32, porosity, "porosity"
@ -77,7 +81,11 @@ class ModflowBct(Package):
self.dth = Util3d(model, (nlay, nrow, ncol), np.float32, dth, "dth") self.dth = Util3d(model, (nlay, nrow, ncol), np.float32, dth, "dth")
self.dtv = Util3d(model, (nlay, nrow, ncol), np.float32, dth, "dtv") self.dtv = Util3d(model, (nlay, nrow, ncol), np.float32, dth, "dtv")
self.sconc = Util3d( self.sconc = Util3d(
model, (nlay, nrow, ncol), np.float32, sconc, "sconc", model,
(nlay, nrow, ncol),
np.float32,
sconc,
"sconc",
) )
self.parent.add_package(self) self.parent.add_package(self)
return return

18
flopy/modflow/mfdis.py
View File

@ -599,16 +599,16 @@ class ModflowDis(Package):
def get_layer(self, i, j, elev): def get_layer(self, i, j, elev):
"""Return the layer for an elevation at an i, j location. """Return the layer for an elevation at an i, j location.
Parameters Parameters
---------- ----------
i : row index (zero-based) i : row index (zero-based)
j : column index j : column index
elev : elevation (in same units as model) elev : elevation (in same units as model)
Returns Returns
------- -------
k : zero-based layer index k : zero-based layer index
""" """
return get_layer(self, i, j, elev) return get_layer(self, i, j, elev)
def gettop(self): def gettop(self):

18
flopy/modflow/mfsfr2.py
View File

@ -3156,11 +3156,13 @@ class check:
def slope(self, minimum_slope=1e-4, maximum_slope=1.0): def slope(self, minimum_slope=1e-4, maximum_slope=1.0):
"""Checks that streambed slopes are greater than or equal to a specified minimum value. """Checks that streambed slopes are greater than or equal to a specified minimum value.
Low slope values can cause "backup" or unrealistic stream stages with icalc options Low slope values can cause "backup" or unrealistic stream stages with icalc options
where stage is computed. where stage is computed.
""" """
headertxt = "Checking for streambed slopes of less than {}...\n".format( headertxt = (
minimum_slope "Checking for streambed slopes of less than {}...\n".format(
minimum_slope
)
) )
txt = "" txt = ""
if self.verbose: if self.verbose:
@ -3191,8 +3193,10 @@ class check:
passed = True passed = True
self._txt_footer(headertxt, txt, "minimum slope", passed) self._txt_footer(headertxt, txt, "minimum slope", passed)
headertxt = "Checking for streambed slopes of greater than {}...\n".format( headertxt = (
maximum_slope "Checking for streambed slopes of greater than {}...\n".format(
maximum_slope
)
) )
txt = "" txt = ""
if self.verbose: if self.verbose:

4
flopy/modpath/mp7.py
View File

@ -171,7 +171,9 @@ class Modpath7(BaseModel):
shape = (nlay, ncpl) shape = (nlay, ncpl)
elif dis.package_name.lower() == "disu": elif dis.package_name.lower() == "disu":
nodes = dis.nodes.array nodes = dis.nodes.array
shape = tuple(nodes,) shape = tuple(
nodes,
)
else: else:
msg = ( msg = (
"DIS, DISV, or DISU packages must be " "DIS, DISV, or DISU packages must be "

2
flopy/modpath/mpsim.py
View File

@ -444,7 +444,7 @@ class StartingLocationsFile(Package):
@staticmethod @staticmethod
def get_dtypes(): def get_dtypes():
""" """
Build numpy dtype for the MODPATH 6 starting locations file. Build numpy dtype for the MODPATH 6 starting locations file.
""" """
dtype = np.dtype( dtype = np.dtype(
[ [

6
flopy/mt3d/mt.py
View File

@ -936,8 +936,10 @@ class Mt3dms(BaseModel):
msg = "First line in file must be \n{}\nFound {}".format( msg = "First line in file must be \n{}\nFound {}".format(
firstline, line.strip() firstline, line.strip()
) )
msg += "\n{} does not appear to be a valid MT3D OBS file".format( msg += (
fname "\n{} does not appear to be a valid MT3D OBS file".format(
fname
)
) )
raise Exception(msg) raise Exception(msg)

4
flopy/mt3d/mtcts.py
View File

@ -140,7 +140,9 @@ class Mt3dCts(Package):
""" """
def __init__(self,): def __init__(
self,
):
raise NotImplementedError() raise NotImplementedError()
# # unit number # # unit number
# if unitnumber is None: # if unitnumber is None:

64
flopy/utils/datafile.py
View File

@ -206,7 +206,9 @@ class LayerFile(object):
if self.mg is None: if self.mg is None:
self.mg = StructuredGrid( self.mg = StructuredGrid(
delc=np.ones((self.nrow,)), delc=np.ones((self.nrow,)),
delr=np.ones(self.ncol,), delr=np.ones(
self.ncol,
),
xoff=0.0, xoff=0.0,
yoff=0.0, yoff=0.0,
angrot=0.0, angrot=0.0,
@ -222,41 +224,41 @@ class LayerFile(object):
attrib_name="lf_data", attrib_name="lf_data",
): ):
""" """
Export model output data to a shapefile at a specific location Export model output data to a shapefile at a specific location
in LayerFile instance. in LayerFile instance.
Parameters Parameters
---------- ----------
filename : str filename : str
Shapefile name to write Shapefile name to write
kstpkper : tuple of ints kstpkper : tuple of ints
A tuple containing the time step and stress period (kstp, kper). A tuple containing the time step and stress period (kstp, kper).
These are zero-based kstp and kper values. These are zero-based kstp and kper values.
totim : float totim : float
The simulation time. The simulation time.
mflay : integer mflay : integer
MODFLOW zero-based layer number to return. If None, then layer 1 MODFLOW zero-based layer number to return. If None, then layer 1
will be written will be written
attrib_name : str attrib_name : str
Base name of attribute columns. (default is 'lf_data') Base name of attribute columns. (default is 'lf_data')
Returns Returns
---------- ----------
None None
See Also See Also
-------- --------
Notes Notes
----- -----
Examples Examples
-------- --------
>>> import flopy >>> import flopy
>>> hdobj = flopy.utils.HeadFile('test.hds') >>> hdobj = flopy.utils.HeadFile('test.hds')
>>> times = hdobj.get_times() >>> times = hdobj.get_times()
>>> hdobj.to_shapefile('test_heads_sp6.shp', totim=times[-1]) >>> hdobj.to_shapefile('test_heads_sp6.shp', totim=times[-1])
""" """
plotarray = np.atleast_3d( plotarray = np.atleast_3d(
self.get_data( self.get_data(

12
flopy/utils/gridintersect.py
View File

@ -1223,12 +1223,16 @@ class GridIntersect:
v_realworld = [] v_realworld = []
if intersect.geom_type.startswith("Multi"): if intersect.geom_type.startswith("Multi"):
for ipoly in intersect: for ipoly in intersect:
v_realworld += self._transform_geo_interface_polygon( v_realworld += (
ipoly self._transform_geo_interface_polygon(
ipoly
)
) )
else: else:
v_realworld += self._transform_geo_interface_polygon( v_realworld += (
intersect self._transform_geo_interface_polygon(
intersect
)
) )
intersect_realworld = rotate( intersect_realworld = rotate(
intersect, self.mfgrid.angrot, origin=(0.0, 0.0) intersect, self.mfgrid.angrot, origin=(0.0, 0.0)

22
flopy/utils/mflistfile.py
View File

@ -241,7 +241,7 @@ class ListBudget(object):
>>> mf_list = MfListBudget("my_model.list") >>> mf_list = MfListBudget("my_model.list")
>>> cumulative = mf_list.get_cumulative() >>> cumulative = mf_list.get_cumulative()
""" """
if not self._isvalid: if not self._isvalid:
return None return None
if names is None: if names is None:
@ -1005,9 +1005,7 @@ class ListBudget(object):
class SwtListBudget(ListBudget): class SwtListBudget(ListBudget):
""" """"""
"""
def set_budget_key(self): def set_budget_key(self):
self.budgetkey = "MASS BUDGET FOR ENTIRE MODEL" self.budgetkey = "MASS BUDGET FOR ENTIRE MODEL"
@ -1015,9 +1013,7 @@ class SwtListBudget(ListBudget):
class MfListBudget(ListBudget): class MfListBudget(ListBudget):
""" """"""
"""
def set_budget_key(self): def set_budget_key(self):
self.budgetkey = "VOLUMETRIC BUDGET FOR ENTIRE MODEL" self.budgetkey = "VOLUMETRIC BUDGET FOR ENTIRE MODEL"
@ -1025,9 +1021,7 @@ class MfListBudget(ListBudget):
class Mf6ListBudget(ListBudget): class Mf6ListBudget(ListBudget):
""" """"""
"""
def set_budget_key(self): def set_budget_key(self):
self.budgetkey = "VOLUME BUDGET FOR ENTIRE MODEL" self.budgetkey = "VOLUME BUDGET FOR ENTIRE MODEL"
@ -1035,9 +1029,7 @@ class Mf6ListBudget(ListBudget):
class MfusgListBudget(ListBudget): class MfusgListBudget(ListBudget):
""" """"""
"""
def set_budget_key(self): def set_budget_key(self):
self.budgetkey = "VOLUMETRIC BUDGET FOR ENTIRE MODEL" self.budgetkey = "VOLUMETRIC BUDGET FOR ENTIRE MODEL"
@ -1045,9 +1037,7 @@ class MfusgListBudget(ListBudget):
class SwrListBudget(ListBudget): class SwrListBudget(ListBudget):
""" """"""
"""
def set_budget_key(self): def set_budget_key(self):
self.budgetkey = "VOLUMETRIC SURFACE WATER BUDGET FOR ENTIRE MODEL" self.budgetkey = "VOLUMETRIC SURFACE WATER BUDGET FOR ENTIRE MODEL"

12
flopy/utils/modpathfile.py
View File

@ -88,7 +88,7 @@ class PathlineFile:
def _build_index(self): def _build_index(self):
""" """
Set position of the start of the pathline data. Set position of the start of the pathline data.
""" """
self.skiprows = 0 self.skiprows = 0
self.file = open(self.fname, "r") self.file = open(self.fname, "r")
@ -123,7 +123,7 @@ class PathlineFile:
def _get_dtypes(self): def _get_dtypes(self):
""" """
Build numpy dtype for the MODPATH 6 pathline file. Build numpy dtype for the MODPATH 6 pathline file.
""" """
if self.version == 3 or self.version == 5: if self.version == 3 or self.version == 5:
dtype = np.dtype( dtype = np.dtype(
@ -706,7 +706,7 @@ class EndpointFile:
def _build_index(self): def _build_index(self):
""" """
Set position of the start of the pathline data. Set position of the start of the pathline data.
""" """
self.skiprows = 0 self.skiprows = 0
self.file = open(self.fname, "r") self.file = open(self.fname, "r")
@ -750,7 +750,7 @@ class EndpointFile:
def _get_dtypes(self): def _get_dtypes(self):
""" """
Build numpy dtype for the MODPATH 6 endpoint file. Build numpy dtype for the MODPATH 6 endpoint file.
""" """
if self.version == 3 or self.version == 5: if self.version == 3 or self.version == 5:
dtype = self._get_mp35_dtype() dtype = self._get_mp35_dtype()
@ -1252,7 +1252,7 @@ class TimeseriesFile:
def _build_index(self): def _build_index(self):
""" """
Set position of the start of the timeseries data. Set position of the start of the timeseries data.
""" """
compact = False compact = False
self.skiprows = 0 self.skiprows = 0
@ -1295,7 +1295,7 @@ class TimeseriesFile:
def _get_dtypes(self): def _get_dtypes(self):
""" """
Build numpy dtype for the MODPATH 6 timeseries file. Build numpy dtype for the MODPATH 6 timeseries file.
""" """
if self.version == 3 or self.version == 5: if self.version == 3 or self.version == 5:
if self.compact: if self.compact:

3
flopy/utils/reference.py
View File

@ -972,8 +972,7 @@ class SpatialReference(object):
return yedge return yedge
def write_gridSpec(self, filename): def write_gridSpec(self, filename):
""" write a PEST-style grid specification file """write a PEST-style grid specification file"""
"""
f = open(filename, "w") f = open(filename, "w")
f.write( f.write(
"{0:10d} {1:10d}\n".format(self.delc.shape[0], self.delr.shape[0]) "{0:10d} {1:10d}\n".format(self.delc.shape[0], self.delr.shape[0])

6
flopy/utils/util_list.py
View File

@ -140,7 +140,7 @@ class MfList(DataInterface, DataListInterface):
return export.utils.mflist_export(f, self, **kwargs) return export.utils.mflist_export(f, self, **kwargs)
def append(self, other): def append(self, other):
""" append the recarrays from one MfList to another """append the recarrays from one MfList to another
Parameters Parameters
---------- ----------
other: variable: an item that can be cast in to an MfList other: variable: an item that can be cast in to an MfList
@ -866,7 +866,7 @@ class MfList(DataInterface, DataListInterface):
def get_indices(self): def get_indices(self):
""" """
a helper function for plotting - get all unique indices a helper function for plotting - get all unique indices
""" """
names = self.dtype.names names = self.dtype.names
lnames = [] lnames = []
@ -1242,7 +1242,7 @@ class MfList(DataInterface, DataListInterface):
@staticmethod @staticmethod
def masked4D_arrays_to_stress_period_data(dtype, m4ds): def masked4D_arrays_to_stress_period_data(dtype, m4ds):
""" convert a dictionary of 4-dim masked arrays to """convert a dictionary of 4-dim masked arrays to
a stress_period_data style dict of recarray a stress_period_data style dict of recarray
Parameters Parameters
---------- ----------

8
flopy/utils/zonbud.py
View File

@ -2517,7 +2517,13 @@ class ZoneBudgetOutput(object):
data[col] = np.zeros((totim.size, zones.size), dtype=float) data[col] = np.zeros((totim.size, zones.size), dtype=float)
for i, time in enumerate(totim): for i, time in enumerate(totim):
tdf = df.loc[df.totim.isin([time,])] tdf = df.loc[
df.totim.isin(
[
time,
]
)
]
tdf = tdf.sort_values(by=["zone"]) tdf = tdf.sort_values(by=["zone"])
for col in df.columns: for col in df.columns:

369
release/make-release.py
View File

@ -10,21 +10,27 @@ from collections import OrderedDict
# update files and paths so that there are the same number of # update files and paths so that there are the same number of
# path and file entries in the paths and files list. Enter '.' # path and file entries in the paths and files list. Enter '.'
# as the path if the file is in the root repository directory # as the path if the file is in the root repository directory
paths = ['../flopy', '../', paths = ["../flopy", "../", "../docs", "../docs", "../", "../", "../docs"]
'../docs', '../docs', files = [
'../', '../', '../docs'] "version.py",
files = ['version.py', 'README.md', "README.md",
'USGS_release.md', 'PyPi_release.md', "USGS_release.md",
'code.json', 'DISCLAIMER.md', 'notebook_examples.md'] "PyPi_release.md",
"code.json",
"DISCLAIMER.md",
"notebook_examples.md",
]
# check that there are the same number of entries in files and paths # check that there are the same number of entries in files and paths
if len(paths) != len(files): if len(paths) != len(files):
msg = 'The number of entries in paths ' + \ msg = (
'({}) must equal '.format(len(paths)) + \ "The number of entries in paths "
'the number of entries in files ({})'.format(len(files)) + "({}) must equal ".format(len(paths))
+ "the number of entries in files ({})".format(len(files))
)
assert False, msg assert False, msg
pak = 'flopy' pak = "flopy"
# local import of package variables in flopy/version.py # local import of package variables in flopy/version.py
# imports author_dict # imports author_dict
@ -34,12 +40,12 @@ exec(open(os.path.join("..", "flopy", "version.py")).read())
authors = [] authors = []
for key in author_dict.keys(): for key in author_dict.keys():
t = key.split() t = key.split()
author = '{}'.format(t[-1]) author = "{}".format(t[-1])
for str in t[0:-1]: for str in t[0:-1]:
author += ' {}'.format(str) author += " {}".format(str)
authors.append(author) authors.append(author)
approved = '''Disclaimer approved = """Disclaimer
---------- ----------
This software has been approved for release by the U.S. Geological Survey This software has been approved for release by the U.S. Geological Survey
@ -51,9 +57,9 @@ shall the fact of release constitute any such warranty. Furthermore, the
software is released on condition that neither the USGS nor the U.S. Government software is released on condition that neither the USGS nor the U.S. Government
shall be held liable for any damages resulting from its authorized or shall be held liable for any damages resulting from its authorized or
unauthorized use. unauthorized use.
''' """
preliminary = '''Disclaimer preliminary = """Disclaimer
---------- ----------
This software is preliminary or provisional and is subject to revision. It is This software is preliminary or provisional and is subject to revision. It is
@ -64,14 +70,14 @@ functionality of the software and related material nor shall the fact of release
constitute any such warranty. The software is provided on the condition that constitute any such warranty. The software is provided on the condition that
neither the USGS nor the U.S. Government shall be held liable for any damages neither the USGS nor the U.S. Government shall be held liable for any damages
resulting from the authorized or unauthorized use of the software. resulting from the authorized or unauthorized use of the software.
''' """
def get_disclaimer(): def get_disclaimer():
# get current branch # get current branch
branch = get_branch() branch = get_branch()
if branch.lower().startswith('release') or 'master' in branch.lower(): if branch.lower().startswith("release") or "master" in branch.lower():
disclaimer = approved disclaimer = approved
is_approved = True is_approved = True
else: else:
@ -86,79 +92,79 @@ def get_branch():
# determine if branch defined on command line # determine if branch defined on command line
for argv in sys.argv: for argv in sys.argv:
if 'master' in argv: if "master" in argv:
branch = 'master' branch = "master"
elif 'develop' in argv.lower(): elif "develop" in argv.lower():
branch = 'develop' branch = "develop"
if branch is None: if branch is None:
try: try:
# determine current branch # determine current branch
b = subprocess.Popen(("git", "status"), b = subprocess.Popen(
stdout=subprocess.PIPE, ("git", "status"),
stderr=subprocess.STDOUT).communicate()[0] stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
).communicate()[0]
if isinstance(b, bytes): if isinstance(b, bytes):
b = b.decode('utf-8') b = b.decode("utf-8")
for line in b.splitlines(): for line in b.splitlines():
if 'On branch' in line: if "On branch" in line:
branch = line.replace('On branch ', '').rstrip() branch = line.replace("On branch ", "").rstrip()
except: except:
msg = 'Could not determine current branch. Is git installed?' msg = "Could not determine current branch. Is git installed?"
raise ValueError(msg) raise ValueError(msg)
return branch return branch
def get_version_str(v0, v1, v2): def get_version_str(v0, v1, v2):
version_type = ('{}'.format(v0), version_type = ("{}".format(v0), "{}".format(v1), "{}".format(v2))
'{}'.format(v1), version = ".".join(version_type)
'{}'.format(v2))
version = '.'.join(version_type)
return version return version
def get_tag(v0, v1, v2): def get_tag(v0, v1, v2):
tag_type = ('{}'.format(v0), tag_type = ("{}".format(v0), "{}".format(v1), "{}".format(v2))
'{}'.format(v1), tag = ".".join(tag_type)
'{}'.format(v2))
tag = '.'.join(tag_type)
return tag return tag
def get_software_citation(version, is_approved): def get_software_citation(version, is_approved):
now = datetime.datetime.now() now = datetime.datetime.now()
sb = '' sb = ""
if not is_approved: if not is_approved:
sb = ' &mdash; release candidate' sb = " &mdash; release candidate"
# format author names # format author names
line = '[' line = "["
for ipos, author in enumerate(authors): for ipos, author in enumerate(authors):
if ipos > 0: if ipos > 0:
line += ', ' line += ", "
if ipos == len(authors) - 1: if ipos == len(authors) - 1:
line += 'and ' line += "and "
sv = author.split() sv = author.split()
tauthor = '{}'.format(sv[0]) tauthor = "{}".format(sv[0])
if len(sv) < 3: if len(sv) < 3:
gname = sv[1] gname = sv[1]
if len(gname) > 1: if len(gname) > 1:
tauthor += ', {}'.format(gname) tauthor += ", {}".format(gname)
else: else:
tauthor += ', {}.'.format(gname[0]) tauthor += ", {}.".format(gname[0])
else: else:
tauthor += ', {}. {}.'.format(sv[1][0], sv[2][0]) tauthor += ", {}. {}.".format(sv[1][0], sv[2][0])
# add formatted author name to line # add formatted author name to line
line += tauthor line += tauthor
# add the rest of the citation # add the rest of the citation
line += ', {}, '.format(now.year) + \ line += (
'FloPy v{}{}: '.format(version, sb) + \ ", {}, ".format(now.year)
'U.S. Geological Survey Software Release, ' + \ + "FloPy v{}{}: ".format(version, sb)
'{}, '.format(now.strftime('%d %B %Y')) + \ + "U.S. Geological Survey Software Release, "
'http://dx.doi.org/10.5066/F7BK19FH]' + \ + "{}, ".format(now.strftime("%d %B %Y"))
'(http://dx.doi.org/10.5066/F7BK19FH)' + "http://dx.doi.org/10.5066/F7BK19FH]"
+ "(http://dx.doi.org/10.5066/F7BK19FH)"
)
return line return line
@ -171,44 +177,49 @@ def update_version():
vmajor = 0 vmajor = 0
vminor = 0 vminor = 0
vmicro = 0 vmicro = 0
lines = [line.rstrip('\n') for line in open(fpth, 'r')] lines = [line.rstrip("\n") for line in open(fpth, "r")]
for idx, line in enumerate(lines): for idx, line in enumerate(lines):
t = line.split() t = line.split()
if 'major =' in line: if "major =" in line:
vmajor = int(t[2]) vmajor = int(t[2])
elif 'minor =' in line: elif "minor =" in line:
vminor = int(t[2]) vminor = int(t[2])
elif 'micro =' in line: elif "micro =" in line:
vmicro = int(t[2]) vmicro = int(t[2])
elif '__version__' in line: elif "__version__" in line:
name_pos = idx + 1 name_pos = idx + 1
except: except:
msg = 'There was a problem updating the version file' msg = "There was a problem updating the version file"
raise IOError(msg) raise IOError(msg)
try: try:
# write new version file # write new version file
f = open(fpth, 'w') f = open(fpth, "w")
f.write('# {} version file automatically '.format(pak) + f.write(
'created using...{0}\n'.format(os.path.basename(__file__))) "# {} version file automatically ".format(pak)
f.write('# created on...' + + "created using...{0}\n".format(os.path.basename(__file__))
'{0}\n'.format( )
datetime.datetime.now().strftime("%B %d, %Y %H:%M:%S"))) f.write(
f.write('\n') "# created on..."
f.write('major = {}\n'.format(vmajor)) + "{0}\n".format(
f.write('minor = {}\n'.format(vminor)) datetime.datetime.now().strftime("%B %d, %Y %H:%M:%S")
f.write('micro = {}\n'.format(vmicro)) )
)
f.write("\n")
f.write("major = {}\n".format(vmajor))
f.write("minor = {}\n".format(vminor))
f.write("micro = {}\n".format(vmicro))
f.write("__version__ = '{:d}.{:d}.{:d}'.format(major, minor, micro)\n") f.write("__version__ = '{:d}.{:d}.{:d}'.format(major, minor, micro)\n")
# write the remainder of the version file # write the remainder of the version file
if name_pos is not None: if name_pos is not None:
for line in lines[name_pos:]: for line in lines[name_pos:]:
f.write('{}\n'.format(line)) f.write("{}\n".format(line))
f.close() f.close()
print('Successfully updated version.py') print("Successfully updated version.py")
except: except:
msg = 'There was a problem updating the version file' msg = "There was a problem updating the version file"
raise IOError(msg) raise IOError(msg)
# update README.md with new version information # update README.md with new version information
@ -235,24 +246,24 @@ def update_codejson(vmajor, vminor, vmicro):
version = get_tag(vmajor, vminor, vmicro) version = get_tag(vmajor, vminor, vmicro)
# load and modify json file # load and modify json file
with open(json_fname, 'r') as f: with open(json_fname, "r") as f:
data = json.load(f, object_pairs_hook=OrderedDict) data = json.load(f, object_pairs_hook=OrderedDict)
# modify the json file data # modify the json file data
now = datetime.datetime.now() now = datetime.datetime.now()
sdate = now.strftime('%Y-%m-%d') sdate = now.strftime("%Y-%m-%d")
data[0]['date']['metadataLastUpdated'] = sdate data[0]["date"]["metadataLastUpdated"] = sdate
if branch.lower().startswith('release') or 'master' in branch.lower(): if branch.lower().startswith("release") or "master" in branch.lower():
data[0]['version'] = version data[0]["version"] = version
data[0]['status'] = 'Production' data[0]["status"] = "Production"
else: else:
data[0]['version'] = version data[0]["version"] = version
data[0]['status'] = 'Release Candidate' data[0]["status"] = "Release Candidate"
# rewrite the json file # rewrite the json file
with open(json_fname, 'w') as f: with open(json_fname, "w") as f:
json.dump(data, f, indent=4) json.dump(data, f, indent=4)
f.write('\n') f.write("\n")
return return
@ -263,54 +274,61 @@ def update_readme_markdown(vmajor, vminor, vmicro):
# define branch # define branch
if is_approved: if is_approved:
branch = 'master' branch = "master"
else: else:
branch = 'develop' branch = "develop"
# create version # create version
version = get_tag(vmajor, vminor, vmicro) version = get_tag(vmajor, vminor, vmicro)
# read README.md into memory # read README.md into memory
fpth = os.path.join(paths[1], files[1]) fpth = os.path.join(paths[1], files[1])
with open(fpth, 'r') as file: with open(fpth, "r") as file:
lines = [line.rstrip() for line in file] lines = [line.rstrip() for line in file]
# rewrite README.md # rewrite README.md
terminate = False terminate = False
f = open(fpth, 'w') f = open(fpth, "w")
for line in lines: for line in lines:
if '### Version ' in line: if "### Version " in line:
line = '### Version {}'.format(version) line = "### Version {}".format(version)
if not is_approved: if not is_approved:
line += ' &mdash; release candidate' line += " &mdash; release candidate"
elif '[Build Status]' in line: elif "[Build Status]" in line:
line = '[![Build Status](https://travis-ci.org/modflowpy/' + \ line = (
'flopy.svg?branch={})]'.format(branch) + \ "[![Build Status](https://travis-ci.org/modflowpy/"
'(https://travis-ci.org/modflowpy/flopy)' + "flopy.svg?branch={})]".format(branch)
elif '[Coverage Status]' in line: + "(https://travis-ci.org/modflowpy/flopy)"
line = '[![Coverage Status](https://coveralls.io/repos/github/' + \ )
'modflowpy/flopy/badge.svg?branch={})]'.format(branch) + \ elif "[Coverage Status]" in line:
'(https://coveralls.io/github/modflowpy/' + \ line = (
'flopy?branch={})'.format(branch) "[![Coverage Status](https://coveralls.io/repos/github/"
elif '[Binder]' in line: + "modflowpy/flopy/badge.svg?branch={})]".format(branch)
+ "(https://coveralls.io/github/modflowpy/"
+ "flopy?branch={})".format(branch)
)
elif "[Binder]" in line:
# [![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/modflowpy/flopy.git/develop) # [![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/modflowpy/flopy.git/develop)
line = '[![Binder](https://mybinder.org/badge_logo.svg)]' + \ line = (
'(https://mybinder.org/v2/gh/modflowpy/flopy.git/' + \ "[![Binder](https://mybinder.org/badge_logo.svg)]"
'{}'.format(branch) + ')' + "(https://mybinder.org/v2/gh/modflowpy/flopy.git/"
elif 'http://dx.doi.org/10.5066/F7BK19FH' in line: + "{}".format(branch)
+ ")"
)
elif "http://dx.doi.org/10.5066/F7BK19FH" in line:
line = get_software_citation(version, is_approved) line = get_software_citation(version, is_approved)
elif 'Disclaimer' in line: elif "Disclaimer" in line:
line = disclaimer line = disclaimer
terminate = True terminate = True
f.write('{}\n'.format(line)) f.write("{}\n".format(line))
if terminate: if terminate:
break break
f.close() f.close()
# write disclaimer markdown file # write disclaimer markdown file
fpth = os.path.join(paths[0], 'DISCLAIMER.md') fpth = os.path.join(paths[0], "DISCLAIMER.md")
f = open(fpth, 'w') f = open(fpth, "w")
f.write(disclaimer) f.write(disclaimer)
f.close() f.close()
@ -323,25 +341,28 @@ def update_notebook_examples_markdown():
# define branch # define branch
if is_approved: if is_approved:
branch = 'master' branch = "master"
else: else:
branch = 'develop' branch = "develop"
# read notebook_examples.md into memory # read notebook_examples.md into memory
fpth = os.path.join(paths[6], files[6]) fpth = os.path.join(paths[6], files[6])
with open(fpth, 'r') as file: with open(fpth, "r") as file:
lines = [line.rstrip() for line in file] lines = [line.rstrip() for line in file]
# rewrite notebook_examples.md # rewrite notebook_examples.md
terminate = False terminate = False
f = open(fpth, 'w') f = open(fpth, "w")
for line in lines: for line in lines:
if '[Binder]' in line: if "[Binder]" in line:
# [![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/modflowpy/flopy.git/develop) # [![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/modflowpy/flopy.git/develop)
line = '[![Binder](https://mybinder.org/badge_logo.svg)]' + \ line = (
'(https://mybinder.org/v2/gh/modflowpy/flopy.git/' + \ "[![Binder](https://mybinder.org/badge_logo.svg)]"
'{}'.format(branch) + ')' + "(https://mybinder.org/v2/gh/modflowpy/flopy.git/"
f.write('{}\n'.format(line)) + "{}".format(branch)
+ ")"
)
f.write("{}\n".format(line))
f.close() f.close()
@ -354,108 +375,110 @@ def update_USGSmarkdown(vmajor, vminor, vmicro):
# set repo branch string based on is_approved # set repo branch string based on is_approved
if is_approved: if is_approved:
repo_str = 'master' repo_str = "master"
else: else:
repo_str = 'develop' repo_str = "develop"
# create version # create version
version = get_tag(vmajor, vminor, vmicro) version = get_tag(vmajor, vminor, vmicro)
# read README.md into memory # read README.md into memory
fpth = os.path.join(paths[1], files[1]) fpth = os.path.join(paths[1], files[1])
with open(fpth, 'r') as file: with open(fpth, "r") as file:
lines = [line.rstrip() for line in file] lines = [line.rstrip() for line in file]
# write USGS_release.md # write USGS_release.md
fpth = os.path.join(paths[2], files[2]) fpth = os.path.join(paths[2], files[2])
f = open(fpth, 'w') f = open(fpth, "w")
# write PyPi_release.md # write PyPi_release.md
fpth = os.path.join(paths[3], files[3]) fpth = os.path.join(paths[3], files[3])
f2 = open(fpth, 'w') f2 = open(fpth, "w")
# date and branch information # date and branch information
now = datetime.datetime.now() now = datetime.datetime.now()
sdate = now.strftime("%m/%d/%Y") sdate = now.strftime("%m/%d/%Y")
# write header information # write header information
f.write('---\n') f.write("---\n")
f.write('title: FloPy Release Notes\n') f.write("title: FloPy Release Notes\n")
f.write('author:\n') f.write("author:\n")
for author in authors: for author in authors:
sv = author.split() sv = author.split()
tauthor = '{}'.format(sv[1]) tauthor = "{}".format(sv[1])
if len(sv) > 2: if len(sv) > 2:
tauthor += ' {}.'.format(sv[2][0]) tauthor += " {}.".format(sv[2][0])
tauthor += ' {}'.format(sv[0]) tauthor += " {}".format(sv[0])
f.write(' - {}\n'.format(tauthor)) f.write(" - {}\n".format(tauthor))
f.write('header-includes:\n') f.write("header-includes:\n")
f.write(' - \\usepackage{fancyhdr}\n') f.write(" - \\usepackage{fancyhdr}\n")
f.write(' - \\usepackage{lastpage}\n') f.write(" - \\usepackage{lastpage}\n")
f.write(' - \\pagestyle{fancy}\n') f.write(" - \\pagestyle{fancy}\n")
f.write(' - \\fancyhf{{}}\n') f.write(" - \\fancyhf{{}}\n")
f.write(' - \\fancyhead[LE, LO, RE, RO]{}\n') f.write(" - \\fancyhead[LE, LO, RE, RO]{}\n")
f.write(' - \\fancyhead[CE, CO]{FloPy Release Notes}\n') f.write(" - \\fancyhead[CE, CO]{FloPy Release Notes}\n")
f.write(' - \\fancyfoot[LE, RO]{{FloPy version {}}}\n'.format(version)) f.write(" - \\fancyfoot[LE, RO]{{FloPy version {}}}\n".format(version))
f.write(' - \\fancyfoot[CO, CE]{\\thepage\\ of \\pageref{LastPage}}\n') f.write(" - \\fancyfoot[CO, CE]{\\thepage\\ of \\pageref{LastPage}}\n")
f.write(' - \\fancyfoot[RE, LO]{{{}}}\n'.format(sdate)) f.write(" - \\fancyfoot[RE, LO]{{{}}}\n".format(sdate))
f.write('geometry: margin=0.75in\n') f.write("geometry: margin=0.75in\n")
f.write('---\n\n') f.write("---\n\n")
# write select information from README.md # write select information from README.md
writeline = False writeline = False
for line in lines: for line in lines:
if line == 'Introduction': if line == "Introduction":
writeline = True writeline = True
elif line == 'Installation': elif line == "Installation":
writeline = False writeline = False
elif line == 'Documentation': elif line == "Documentation":
writeline = True writeline = True
elif line == 'Getting Started': elif line == "Getting Started":
writeline = False writeline = False
elif line == 'Contributing': elif line == "Contributing":
writeline = True writeline = True
elif line == 'How to Cite': elif line == "How to Cite":
writeline = True writeline = True
elif 'http://dx.doi.org/10.5066/F7BK19FH' in line: elif "http://dx.doi.org/10.5066/F7BK19FH" in line:
writeline = True writeline = True
line = get_software_citation(version, is_approved) line = get_software_citation(version, is_approved)
elif line == 'MODFLOW Resources': elif line == "MODFLOW Resources":
writeline = False writeline = False
elif 'Installing the latest FloPy release candidate' in line: elif "Installing the latest FloPy release candidate" in line:
writeline = False writeline = False
elif line == 'Disclaimer': elif line == "Disclaimer":
writeline = True writeline = True
elif '[MODFLOW 6](docs/mf6.md)' in line: elif "[MODFLOW 6](docs/mf6.md)" in line:
line = line.replace('[MODFLOW 6](docs/mf6.md)', 'MODFLOW 6') line = line.replace("[MODFLOW 6](docs/mf6.md)", "MODFLOW 6")
if writeline: if writeline:
# USGS_release.md # USGS_release.md
f.write('{}\n'.format(line)) f.write("{}\n".format(line))
# PyPi_release.md # PyPi_release.md
line = line.replace('***', '*') line = line.replace("***", "*")
line = line.replace('##### ', '') line = line.replace("##### ", "")
if 'CONTRIBUTING.md' in line: if "CONTRIBUTING.md" in line:
rstr = 'https://github.com/modflowpy/flopy/blob/' + \ rstr = (
'{}/CONTRIBUTING.md'.format(repo_str) "https://github.com/modflowpy/flopy/blob/"
line = line.replace('CONTRIBUTING.md', rstr) + "{}/CONTRIBUTING.md".format(repo_str)
f2.write('{}\n'.format(line)) )
line = line.replace("CONTRIBUTING.md", rstr)
f2.write("{}\n".format(line))
# write installation information # write installation information
cweb = 'https://water.usgs.gov/ogw/flopy/flopy-{}.zip'.format(version) cweb = "https://water.usgs.gov/ogw/flopy/flopy-{}.zip".format(version)
line = '' line = ""
line += 'Installation\n' line += "Installation\n"
line += '-----------------------------------------------\n' line += "-----------------------------------------------\n"
line += 'To install FloPy version {} '.format(version) line += "To install FloPy version {} ".format(version)
line += 'from the USGS FloPy website:\n' line += "from the USGS FloPy website:\n"
line += '```\n' line += "```\n"
line += 'pip install {}\n'.format(cweb) line += "pip install {}\n".format(cweb)
line += '```\n\n' line += "```\n\n"
line += 'To update to FloPy version {} '.format(version) line += "To update to FloPy version {} ".format(version)
line += 'from the USGS FloPy website:\n' line += "from the USGS FloPy website:\n"
line += '```\n' line += "```\n"
line += 'pip install {} --upgrade\n'.format(cweb) line += "pip install {} --upgrade\n".format(cweb)
line += '```\n' line += "```\n"
# write installation instructions of USGS_release.md # write installation instructions of USGS_release.md
f.write(line) f.write(line)
@ -471,4 +494,4 @@ def update_USGSmarkdown(vmajor, vminor, vmicro):
if __name__ == "__main__": if __name__ == "__main__":
update_version() update_version()
get_software_citation('3.1.1', True) get_software_citation("3.1.1", True)

126
release/update-version_changes.py
View File

@ -4,6 +4,7 @@ from subprocess import Popen, PIPE, STDOUT
PY3 = sys.version_info[0] >= 3 PY3 = sys.version_info[0] >= 3
def process_Popen(cmdlist, verbose=False): def process_Popen(cmdlist, verbose=False):
"""Generic function to initialize a Popen process. """Generic function to initialize a Popen process.
@ -34,152 +35,161 @@ def process_Popen(cmdlist, verbose=False):
# catch non-zero return code # catch non-zero return code
if process.returncode != 0: if process.returncode != 0:
errmsg = '{} failed\n'.format(' '.join(process.args)) + \ errmsg = "{} failed\n".format(
'\tstatus code {}\n'.format(process.returncode) " ".join(process.args)
) + "\tstatus code {}\n".format(process.returncode)
raise ChildProcessError(errmsg) raise ChildProcessError(errmsg)
return stdout return stdout
def get_version(): def get_version():
major = 0 major = 0
minor = 0 minor = 0
micro = 0 micro = 0
# read existing version file # read existing version file
fpth = os.path.join('..', 'flopy', 'version.py') fpth = os.path.join("..", "flopy", "version.py")
f = open(fpth, 'r') f = open(fpth, "r")
lines = [line.rstrip('\n') for line in open(fpth, 'r')] lines = [line.rstrip("\n") for line in open(fpth, "r")]
for idx, line in enumerate(lines): for idx, line in enumerate(lines):
t = line.split() t = line.split()
if 'major =' in line: if "major =" in line:
major = int(t[2]) major = int(t[2])
elif 'minor =' in line: elif "minor =" in line:
minor = int(t[2]) minor = int(t[2])
elif 'micro =' in line: elif "micro =" in line:
micro = int(t[2]) micro = int(t[2])
f.close() f.close()
return '{:d}.{:d}.{:d}'.format(major, minor, micro) return "{:d}.{:d}.{:d}".format(major, minor, micro)
def get_branch(): def get_branch():
branch = None branch = None
# determine if branch defined on command line # determine if branch defined on command line
for argv in sys.argv: for argv in sys.argv:
if '--master' in argv: if "--master" in argv:
branch = 'master' branch = "master"
elif '--develop' in argv.lower(): elif "--develop" in argv.lower():
branch = 'develop' branch = "develop"
# use git to determine branch # use git to determine branch
if branch is None: if branch is None:
cmdlist = ("git", "status") cmdlist = ("git", "status")
stdout = process_Popen(cmdlist) stdout = process_Popen(cmdlist)
for line in stdout.splitlines(): for line in stdout.splitlines():
if 'On branch' in line: if "On branch" in line:
branch = line.replace('On branch ', '').rstrip() branch = line.replace("On branch ", "").rstrip()
break break
return branch return branch
def get_last_tag_date(): def get_last_tag_date():
cmdlist = ("git", "log", "--tags", "--no-walk", "--date=iso-local", cmdlist = (
"--pretty='%cd %D'") "git",
"log",
"--tags",
"--no-walk",
"--date=iso-local",
"--pretty='%cd %D'",
)
stdout = process_Popen(cmdlist) stdout = process_Popen(cmdlist)
line = stdout.splitlines()[0] line = stdout.splitlines()[0]
ipos = line.find("tag") ipos = line.find("tag")
if ipos > -1: if ipos > -1:
tag_date = line[1:20] tag_date = line[1:20]
tag = line[ipos+4:].split(',')[0].strip() tag = line[ipos + 4 :].split(",")[0].strip()
return tag, tag_date return tag, tag_date
def get_hash_dict(branch): def get_hash_dict(branch):
tag, tag_date = get_last_tag_date() tag, tag_date = get_last_tag_date()
# get hash and # get hash and
fmt = '--pretty="%H"' fmt = '--pretty="%H"'
since = '--since="{}"'.format(tag_date) since = '--since="{}"'.format(tag_date)
hash_dict = {'fix': {}, 'feat': {}} hash_dict = {"fix": {}, "feat": {}}
cmdlist = ("git", "log", branch, fmt, since) cmdlist = ("git", "log", branch, fmt, since)
stdout = process_Popen(cmdlist) stdout = process_Popen(cmdlist)
fix_dict = {} fix_dict = {}
feat_dict = {} feat_dict = {}
fix_tags = ['fix', 'bug'] fix_tags = ["fix", "bug"]
feat_tags = ['feat'] feat_tags = ["feat"]
# parse stdout # parse stdout
for line in stdout.splitlines(): for line in stdout.splitlines():
hash = line.split()[0].replace('"', '') hash = line.split()[0].replace('"', "")
url = 'https://github.com/modflowpy/flopy/commit/{}'.format(hash) url = "https://github.com/modflowpy/flopy/commit/{}".format(hash)
fmt = '--pretty="%s."' fmt = '--pretty="%s."'
cmdlist = ("git", "log", fmt, "-n1", hash) cmdlist = ("git", "log", fmt, "-n1", hash)
subject = process_Popen(cmdlist).strip().replace('"', '') subject = process_Popen(cmdlist).strip().replace('"', "")
fmt = '--pretty="Committed by %aN on %cd."' fmt = '--pretty="Committed by %aN on %cd."'
cmdlist = ("git", "log", "--date=short", fmt, "-n1", hash) cmdlist = ("git", "log", "--date=short", fmt, "-n1", hash)
cdate = process_Popen(cmdlist).strip().replace('"', '') cdate = process_Popen(cmdlist).strip().replace('"', "")
ipos = subject.find(':') ipos = subject.find(":")
key = None key = None
if ipos > -1: if ipos > -1:
type = subject[0:ipos] type = subject[0:ipos]
subject = subject.replace(type + ':', '').strip().capitalize() subject = subject.replace(type + ":", "").strip().capitalize()
for tag in fix_tags: for tag in fix_tags:
if type.lower().startswith(tag): if type.lower().startswith(tag):
key = 'fix' key = "fix"
type = key + type[3:] type = key + type[3:]
break break
if key is None: if key is None:
for tag in feat_tags: for tag in feat_tags:
if type.lower().startswith(tag): if type.lower().startswith(tag):
key = 'feat' key = "feat"
break break
else: else:
type = None type = None
slower = subject.lower() slower = subject.lower()
for tag in fix_tags: for tag in fix_tags:
if slower.startswith(tag): if slower.startswith(tag):
key = 'fix' key = "fix"
type = 'fix()' type = "fix()"
break break
if key is None: if key is None:
for tag in feat_tags: for tag in feat_tags:
if slower.startswith(tag): if slower.startswith(tag):
key = 'feat' key = "feat"
type = 'feat()' type = "feat()"
break break
if key is not None: if key is not None:
message = '[{}]({}): '.format(type, url) message = "[{}]({}): ".format(type, url)
message += subject + ' ' + cdate message += subject + " " + cdate
if key == 'fix': if key == "fix":
fix_dict[hash] = message fix_dict[hash] = message
elif key == 'feat': elif key == "feat":
feat_dict[hash] = message feat_dict[hash] = message
# add dictionaries to the hash dictionary # add dictionaries to the hash dictionary
hash_dict['fix'] = fix_dict hash_dict["fix"] = fix_dict
hash_dict['feat'] = feat_dict hash_dict["feat"] = feat_dict
return hash_dict return hash_dict
def create_md(hash_dict): def create_md(hash_dict):
# get current version information # get current version information
version = get_version() version = get_version()
tag = '### Version' tag = "### Version"
version_text = '{} {}'.format(tag, version) version_text = "{} {}".format(tag, version)
# #
# read the lines in the existing version_changes.md # read the lines in the existing version_changes.md
fpth = os.path.join('..', 'docs', 'version_changes.md') fpth = os.path.join("..", "docs", "version_changes.md")
with open(fpth) as f: with open(fpth) as f:
lines = f.read().splitlines() lines = f.read().splitlines()
# rewrite the existing version_changes.md # rewrite the existing version_changes.md
f = open(fpth, 'w') f = open(fpth, "w")
write_line = True write_line = True
write_update = True write_update = True
@ -192,35 +202,35 @@ def create_md(hash_dict):
# write the changes for the latest comment # write the changes for the latest comment
if write_update: if write_update:
write_update = False write_update = False
f.write('{}\n\n'.format(version_text)) f.write("{}\n\n".format(version_text))
write_version_changes(f, hash_dict) write_version_changes(f, hash_dict)
if write_line: if write_line:
f.write('{}\n'.format(line)) f.write("{}\n".format(line))
f.close() f.close()
def write_version_changes(f, hash_dict): def write_version_changes(f, hash_dict):
# features # features
istart = True istart = True
for key, val in hash_dict['feat'].items(): for key, val in hash_dict["feat"].items():
if istart: if istart:
istart = False istart = False
f.write('* New features:\n\n') f.write("* New features:\n\n")
f.write(4*' ' + '* ' + val + '\n') f.write(4 * " " + "* " + val + "\n")
if not istart: if not istart:
f.write('\n\n') f.write("\n\n")
# bug fixes # bug fixes
istart = True istart = True
for key, val in hash_dict['fix'].items(): for key, val in hash_dict["fix"].items():
if istart: if istart:
istart = False istart = False
f.write('* Bug fixes:\n\n') f.write("* Bug fixes:\n\n")
f.write(4 * ' ' + '* ' + val + '\n') f.write(4 * " " + "* " + val + "\n")
if not istart: if not istart:
f.write('\n') f.write("\n")
return return
@ -234,4 +244,4 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
main() main()

61
setup.py
View File

@ -4,7 +4,7 @@ from setuptools import setup
# ensure minimum version of Python is running # ensure minimum version of Python is running
if sys.version_info[0:2] < (3, 5): if sys.version_info[0:2] < (3, 5):
raise RuntimeError('Flopy requires Python >= 3.5') raise RuntimeError("Flopy requires Python >= 3.5")
# local import of package variables in flopy/version.py # local import of package variables in flopy/version.py
# imports __version__, __pakname__, __author__, __author_email__ # imports __version__, __pakname__, __author__, __author_email__
@ -13,27 +13,42 @@ exec(open(os.path.join("flopy", "version.py")).read())
try: try:
import pypandoc import pypandoc
fpth = os.path.join('docs', 'PyPi_release.md') fpth = os.path.join("docs", "PyPi_release.md")
long_description = pypandoc.convert(fpth, 'rst') long_description = pypandoc.convert(fpth, "rst")
except ImportError: except ImportError:
long_description = '' long_description = ""
setup(name=__pakname__, setup(
description='FloPy is a Python package to create, run, and ' + name=__pakname__,
'post-process MODFLOW-based models.', description="FloPy is a Python package to create, run, and "
long_description=long_description, + "post-process MODFLOW-based models.",
author=__author__, long_description=long_description,
author_email=__author_email__, author=__author__,
url='https://github.com/modflowpy/flopy/', author_email=__author_email__,
license='CC0', url="https://github.com/modflowpy/flopy/",
platforms='Windows, Mac OS-X, Linux', license="CC0",
install_requires=['numpy'], platforms="Windows, Mac OS-X, Linux",
packages=['flopy', 'flopy.modflow', 'flopy.modflowlgr', 'flopy.modpath', install_requires=["numpy"],
'flopy.mt3d', 'flopy.seawat', 'flopy.utils', 'flopy.plot', packages=[
'flopy.pest', 'flopy.export', 'flopy.discretization', "flopy",
'flopy.mf6', 'flopy.mf6.coordinates', 'flopy.mf6.data', "flopy.modflow",
'flopy.mf6.modflow', 'flopy.mf6.utils'], "flopy.modflowlgr",
include_package_data=True, # includes files listed in MANIFEST.in "flopy.modpath",
# use this version ID if .svn data cannot be found "flopy.mt3d",
version=__version__, "flopy.seawat",
classifiers=['Topic :: Scientific/Engineering :: Hydrology']) "flopy.utils",
"flopy.plot",
"flopy.pest",
"flopy.export",
"flopy.discretization",
"flopy.mf6",
"flopy.mf6.coordinates",
"flopy.mf6.data",
"flopy.mf6.modflow",
"flopy.mf6.utils",
],
include_package_data=True, # includes files listed in MANIFEST.in
# use this version ID if .svn data cannot be found
version=__version__,
classifiers=["Topic :: Scientific/Engineering :: Hydrology"],
)