add plot to quickstart

Author: mjreno

docs/examples/image/quickstart.png

@@ -103,9 +103,9 @@ def _pcolormesh(grid, da, ax):
 ax.coastlines()
 ax.stock_img()
 ax.set_extent([-5, 15, -4, 14], crs=ccrs.PlateCarree())
-glines = ax.gridlines(draw_labels=True)
-glines.top_labels = False
-glines.right_labels = False
+gln = ax.gridlines(draw_labels=True)
+gln.top_labels = False
+gln.right_labels = False
 ax.set_title("Head")
 qsprj_pth = os.path.join(QS_ROOT, "image", "qsprj.png")
 fig.savefig(qsprj_pth)

docs/examples/qsplot.py

@@ -1,4 +1,9 @@
+from pathlib import Path
+
+import flopy
+import matplotlib.pyplot as plt
 import numpy as np
+import xarray as xr
 
 from flopy4.mf6.gwf import Chd, Dis, Gwf, Ic, Npf, Oc
 from flopy4.mf6.ims import Ims
@@ -38,3 +43,51 @@
 # check OC
 assert oc.data["save_head"][0].item() == "all"
 assert oc.data["save_budget"][0].item() == "all"
+
+# PLOT
+# create budget reader
+bpth = Path("./quickstart_data/mymodel.bud")
+bobj = flopy.utils.CellBudgetFile(bpth, precision="double")
+
+# set specific discharge
+spdis = bobj.get_data(text="DATA-SPDIS")[0]
+
+# create head reader
+hpth = Path("./quickstart_data/mymodel.hds")
+hobj = flopy.utils.HeadFile(hpth, precision="double")
+
+# set heads
+heads = hobj.get_alldata()
+
+# create grid
+grbpth = Path("./quickstart_data/mymodel.dis.grb")
+grid = flopy.discretization.StructuredGrid.from_binary_grid_file(grbpth)
+
+# TODO: get_specific_discharge is dependent on flopy3 model
+# qx, qy, qz = flopy.utils.postprocessing.get_specific_discharge(spdis, gwf)
+
+# set discharge component arrays
+u = []
+v = []
+for r in spdis:
+    u.append(r[3])
+    v.append(r[4])
+uflow = np.array(u).reshape(grid.nrow, grid.ncol)
+vflow = np.array(v).reshape(grid.nrow, grid.ncol)
+
+# set data coordinate arrays
+xcrs = grid.xycenters[0]
+ycrs = grid.xycenters[1]
+
+# create qx, qy dataarrys
+qx = xr.DataArray(uflow, dims=("y", "x"), coords={"x": xcrs, "y": ycrs})
+qy = xr.DataArray(vflow, dims=("y", "x"), coords={"x": xcrs, "y": ycrs})
+
+fig, ax = plt.subplots()
+pmv = flopy.plot.PlotMapView(modelgrid=grid, 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, normalize=True, color="white")
+qs_pth = Path("./image/quickstart.png")
+fig.savefig(qs_pth)