/
drain_return_example.py
101 lines (86 loc) · 2.17 KB
/
drain_return_example.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# ---
# jupyter:
# jupytext:
# text_representation:
# extension: .py
# format_name: light
# format_version: '1.5'
# jupytext_version: 1.14.5
# kernelspec:
# display_name: Python 3 (ipykernel)
# language: python
# name: python3
# metadata:
# section: mf2005
# ---
# # Flopy Drain Return (DRT) capabilities
import os
# +
import sys
from pprint import pformat
from tempfile import TemporaryDirectory
import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np
import flopy
print(sys.version)
print(f"numpy version: {np.__version__}")
print(f"matplotlib version: {mpl.__version__}")
print(f"flopy version: {flopy.__version__}")
# +
# temporary directory
temp_dir = TemporaryDirectory()
modelpth = temp_dir.name
# creat the model package
m = flopy.modflow.Modflow(
"drt_test",
model_ws=modelpth,
exe_name="mfnwt",
version="mfnwt",
)
d = flopy.modflow.ModflowDis(
m,
nlay=1,
nrow=10,
ncol=10,
nper=1,
perlen=1,
top=10,
botm=0,
steady=True,
)
b = flopy.modflow.ModflowBas(m, strt=10, ibound=1)
u = flopy.modflow.ModflowUpw(m, hk=10)
n = flopy.modflow.ModflowNwt(m)
o = flopy.modflow.ModflowOc(m)
# -
# create the drt package
spd = []
for i in range(m.nrow):
spd.append([0, i, m.ncol - 1, 5.0, 50.0, 1, 1, 1, 1.0])
d = flopy.modflow.ModflowDrt(m, stress_period_data={0: spd})
# run the drt model
m.write_input()
success, buff = m.run_model(silent=True, report=True)
assert success, pformat(buff)
# plot heads for the drt model
hds = flopy.utils.HeadFile(os.path.join(m.model_ws, f"{m.name}.hds"))
hds.plot(colorbar=True)
# remove the drt package and create a standard drain file
spd = []
for i in range(m.nrow):
spd.append([0, i, m.ncol - 1, 5.0, 1.0])
m.remove_package("DRT")
d = flopy.modflow.ModflowDrn(m, stress_period_data={0: spd})
# run the drain model
m.write_input()
success, buff = m.run_model(silent=True, report=True)
assert success, pformat(buff)
# plot the heads for the model with the drain
hds = flopy.utils.HeadFile(os.path.join(m.model_ws, f"{m.name}.hds"))
hds.plot(colorbar=True)
try:
# ignore PermissionError on Windows
temp_dir.cleanup()
except:
pass