incremental progress

Author: mjreno

docs/examples/image/quickstart.png

@@ -1,11 +1,13 @@
 from pathlib import Path
 
+import flopy
 import imod.mf6
 import matplotlib.pyplot as plt
 import numpy as np
 
 from flopy4.mf6.gwf import Chd, Dis, Gwf, Ic, Npf, Oc
 from flopy4.mf6.ims import Ims
+from flopy4.mf6.interface.flopy3 import Flopy3Model
 from flopy4.mf6.simulation import Simulation
 from flopy4.mf6.tdis import Tdis
 
@@ -30,6 +32,9 @@
     save_budget={"*": "all"},
 )
 
+# model interface
+gwfif = Flopy3Model(gwf)
+
 # check CHD
 assert chd.data["head"][0, 0] == 1.0
 assert chd.data.head.sel(per=0)[99] == 0.0
@@ -42,29 +47,26 @@
 assert oc.data["save_head"][0] == "all"
 assert oc.data.save_head.sel(per=0) == "all"
 
-# PLOT
-# create budget reader
+# set paths
 bpth = Path("./quickstart_data/mymodel.bud")
 grbpth = Path("./quickstart_data/mymodel.dis.grb")
-
-# set specific discharge
-spdis = imod.mf6.open_cbc(bpth, grbpth, merge_to_dataset=True)
-
-# create head reader
 hpth = Path("./quickstart_data/mymodel.hds")
+
+# set data
+bobj = flopy.utils.CellBudgetFile(bpth, precision="double")
+spdis = bobj.get_data(text="DATA-SPDIS")[0]
+# spdis = imod.mf6.open_cbc(bpth, grbpth, merge_to_dataset=True)
 heads = imod.mf6.open_hds(hpth, grbpth)
-sq = heads.squeeze()
+
+# discharge vectors
+qx, qy, qz = flopy.utils.postprocessing.get_specific_discharge(spdis, gwfif)
+
+# plot
 fig, ax = plt.subplots()
-ax.tick_params()
-ax.set_xticks(np.arange(0, 11, 2), minor=False)
-ax.set_xticks(np.arange(1, 10, 2), minor=True)
-ax.set_yticks(np.arange(0, 11, 2), minor=False)
-ax.set_yticks(np.arange(1, 10, 2), minor=True)
-ax.grid(which="both", color="white")
-sq.plot.imshow(ax=ax)
-sq.plot.contour(ax=ax, levels=[0.2, 0.4, 0.6, 0.8], linewidths=3.0)
-spdis.squeeze().plot.quiver(
-    x="x", y="y", u="npf-qx", v="npf-qy", ax=ax, color="white"
-)
+pmv = flopy.plot.PlotMapView(model=gwfif, ax=ax)
+pmv.plot_array(heads[0][0])
+pmv.plot_grid(colors="white")
+pmv.contour_array(heads[0][0], levels=[0.2, 0.4, 0.6, 0.8], linewidths=3.0)
+pmv.plot_vector(qx, qy, color="white")
 qs_pth = Path("./image/quickstart.png")
 fig.savefig(qs_pth)

docs/examples/quickstart.py

@@ -59,6 +59,9 @@ class Ims(Solution):
     inner_dvclose: Optional[float] = field(
         default=None, metadata={"block": "linear"}
     )
+    inner_hclose: Optional[float] = field(
+        default=None, metadata={"block": "linear"}
+    )
     inner_rclose: Optional[float] = field(
         default=None, metadata={"block": "linear"}
     )