{
"cells": [
{
"cell_type": "code",
"execution_count": 1,
"metadata": {
"hideCode": false,
"hidePrompt": false,
"tags": [
"remove-input"
]
},
"outputs": [],
"source": [
"from myst_nb import glue\n",
"%config InlineBackend.rc = {'figure.dpi': 300}\n",
"%config InlineBackend.figure_format = 'svg'\n",
"import matplotlib.pyplot as plt\n",
"import numpy as np\n",
"import sys\n",
"sys.path.append(\"../py_figures\")\n",
"import diagrams_2d"
]
},
{
"cell_type": "markdown",
"metadata": {
"hideCode": false,
"hidePrompt": false
},
"source": [
"(content:chap7.5:two_wells_uni_flow)=\n",
"# Two wells in uniform background flow"
]
},
{
"cell_type": "markdown",
"metadata": {
"hideCode": false,
"hidePrompt": false
},
"source": [
"The discharge potential and stream function for groundwater flow problems governed by Laplace's equation (i.e., steady flow and negligible areal infiltration) may be combined into a complex potential, because the potential and stream function fulfill the Cauchy-Riemann conditions ({eq}`eq_cauchyrieman6.0`) ({cite:alp}`strack1989groundwater`). The real part of the complex potential $\\Omega$ is the discharge potential and the imaginary part is the stream function\n",
"\n",
"$$\n",
"\\Omega = \\Phi + i\\Psi\n",
"$$(eq_omega7.5)\n",
"\n",
"In mathematical terms, this means that the stream function is the harmonic\n",
"conjugate of the discharge potential ({cite:alp}`strack1989groundwater`). The complex potential $\\Omega$ is a function of the complex coordinate $\\zeta=x+iy$, where $i$ is the imaginary unit ($i^2=-1$). The letter `j` is used for the imaginary unit in Python, but $i$ is used in the mathematical equations in this book. The negative derivative of the complex potential is the complex discharge\n",
"\n",
"$$\n",
"W(\\zeta)=-\\frac{\\text{d} \\Omega}{\\text{d}\\zeta} = Q_x-iQ_y\n",
"$$(eq_W7.5)\n",
"\n",
"Note that the imaginary part of the complex discharge is minus the component of the discharge vector in the $y$-direction.\n",
" The introduction of complex variables makes it easy to compute the discharge potential and stream function simultaneously and allows for the use of sophisticated tools, including conformal mapping, to solve groundwater flow problems. More information on the application of complex variables in solving groundwater flow problems can be found in, e.g., {cite:t}`PolubarinovaKochina1962`, {cite:t}`Harr1962`, {cite:t}`verruijt1970`, and {cite:t}`strack1989groundwater`.\n",
" The complex potential for uniform flow is\n",
"\n",
"$$\n",
"\\Omega = -W_u \\zeta\n",
"$$(eq_Omega27.5)\n",
"\n",
"where $W_u$ is the uniform complex discharge\n",
"\n",
"$$\n",
"W=W_u = U - iV\n",
"$$(eq_W_u7.5)\n",
"\n",
"The complex potential for a well is\n",
"\n",
"$$\n",
"\\Omega = \\frac{Q}{2\\pi}\\ln(\\zeta-\\zeta_w)\n",
"$$(eq_Omega37.5)\n",
"\n",
"where $Q$ is the discharge of the well and $\\zeta_w=x_w + iy_w$ is the complex coordinate of the well location. The validity of ({eq}`eq_Omega37.5`) is shown by writing $\\zeta-\\zeta_w$ in polar form\n",
"\n",
"$$\n",
"\\zeta - \\zeta_w = r\\text{e}^{i\\theta} \n",
"$$(eq_Zeta7.5)\n",
"\n",
"where $r$ is the distance between $\\zeta$ and $\\zeta_w$ and $\\theta$ is the angle that a vector pointing from $\\zeta_w$ to $\\zeta$ makes with the positive $x$-direction. Substitution of ({eq}`eq_Zeta7.5`) for $\\zeta-\\zeta_w$ in ({eq}`eq_Omega37.5`) gives\n",
"\n",
"$$\n",
"\\Omega = \\frac{Q}{2\\pi}\\ln(r\\text{e}^{i\\theta}) = \\frac{Q}{2\\pi}\\ln(r) + i\\frac{Q}{2\\pi}\\theta\n",
"$$(eq_Omega47.5)\n",
"\n",
"which means that the real part of $\\Omega$ is indeed equal to the discharge potential for a well ({eq}`eq_welln6.2`) and the imaginary part is equal to the stream function ({eq}`eq_psithiem6.2`). The corresponding complex discharge for a well is\n",
"\n",
"$$\n",
"W = -\\frac{\\text{d}\\Omega}{\\text{d}\\zeta}= -\\frac{Q}{2\\pi}\\frac{1}{\\zeta-\\zeta_w}\n",
"$$(eq_W27.5)\n",
"\n",
"The complex potential for $N$ wells in uniform background flow is obtained through superposition of ({eq}`eq_Omega27.5`) and ({eq}`eq_Omega37.5`) as\n",
"\n",
"$$\n",
"\\Omega = -W_u \\zeta + \\sum_{n=1}^N\\frac{Q_n}{2\\pi}\\ln(\\zeta-\\zeta_n) + C\n",
"$$\n",
"\n",
"where $Q_n$ is the discharge of well $n$, $\\zeta_n$ is the complex location of well $n$, and an arbitrary constant $C$ is added for completeness.\n",
" The complex potential is applied to determine the capture zones of two wells in a field of uniform flow. The complex potential is\n",
"\n",
"$$\n",
"\\Omega = -W_u \\zeta + \\frac{Q_0}{2\\pi}\\ln(\\zeta-\\zeta_0) + \\frac{Q_1}{2\\pi}\\ln(\\zeta-\\zeta_1) + C\n",
"$$(eq_zeta_C7.5)\n",
"\n",
"and the complex discharge function\n",
"\n",
"$$\n",
"W = W_u -\\frac{Q_0}{2\\pi}\\frac{1}{\\zeta-\\zeta_0} -\\frac{Q_1}{2\\pi}\\frac{1}{\\zeta-\\zeta_1}\n",
"$$(eq_comp_dis7.5)\n",
"\n",
"The stagnation points are found by setting $W=0$, which gives, after multiplication with $2\\pi(\\zeta-\\zeta_0)(\\zeta-\\zeta_1)$ and canceling terms\n",
"\n",
"$$\n",
"2\\pi W_u(\\zeta-\\zeta_0)(\\zeta-\\zeta_1) -Q_0(\\zeta-\\zeta_1) -Q_1(\\zeta-\\zeta_0) = 0\n",
"$$(eq_stag_points7.5)\n",
"\n",
"This is a second order polynomial in $\\zeta$, which may be rearranged to give\n",
"\n",
"$$\n",
"2\\pi W_u\\zeta^2 - (Q_0+Q_1+2\\pi W_u(\\zeta_0+\\zeta_1))\\zeta + Q_0\\zeta_1 + Q_1\\zeta_0 + 2\\pi W_u\\zeta_0\\zeta_1 = 0\n",
"$$(eq_rearranged_zeta7.5)\n",
"\n",
"The complex locations of the stagnation points may now be found with standard equations for the roots of a second order polynomial. Note that application of the complex potential makes it possible to compute the locations of the stagnation points quite easily even when the wells are at arbitrary locations and the uniform flow is in an arbitrary direction. The procedure may be extended to an arbitrary number of wells as shown in {cite:t}`bakker2025capture`.\n",
" As an example, the capture zone envelopes of the two wells of {numref}`fig_captwowells_7.2` are computed. After the stagnation points are computed, the capture zone envelopes can be plotted. The `streamplot` function is used to compute and plot the capture zone envelopes, because contouring the stream function results in a figure with two ugly branch cuts. Two pathlines are started from each stagnation point, one slightly to the left of a stagnation point and one slightly to the right of a stagnation point. The pathlines are integrated in both directions (the default in `streamplot`) to create the two boundaries of a capture zone envelope. "
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {},
"outputs": [],
"source": [
"# parameters\n",
"Wu = 1.2 + 0j # complex uniform flow, m^2/d\n",
"Q0, Q1 = 500, 500 # discharge of wellw, m^3/d\n",
"zeta0, zeta1 = 0 + 0j, 100 + 100j # complex locations of wells, m"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"(x,y) stagnation point 0: 41.96, 16.38\n",
"(x,y) stagnation point 1: 190.67, 83.62\n"
]
}
],
"source": [
"# solution\n",
"a = 2 * np.pi * Wu\n",
"b = -(Q0 + Q1 + 2 * np.pi * Wu * (zeta0 + zeta1))\n",
"c = Q0 * zeta1 + Q1 * zeta0 + 2 * np.pi * Wu * zeta0 * zeta1\n",
"zetas0 = (-b - np.sqrt(b ** 2 - 4 * a * c)) / (2 * a)\n",
"zetas1 = (-b + np.sqrt(b ** 2 - 4 * a * c)) / (2 * a)\n",
"xs0, ys0 = zetas0.real, zetas0.imag\n",
"xs1, ys1 = zetas1.real, zetas1.imag\n",
"print(f'(x,y) stagnation point 0: {xs0:.2f}, {ys0:.2f}')\n",
"print(f'(x,y) stagnation point 1: {xs1:.2f}, {ys1:.2f}')"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Alternatively, if you don't want to program the equation for the roots of a polynomial, you can use the `Polynomial` class of the `numpy.polynomial` package. The `Polynomial` class takes the coefficients of the polynomial as inputs (from low to high order), and the `roots` function returns the roots."
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"stagnation points: [ 41.96276198+16.37606515j 190.66635726+83.62393485j]\n"
]
}
],
"source": [
"from numpy.polynomial import Polynomial\n",
"poly = Polynomial([c, b, a])\n",
"print('stagnation points:', poly.roots())"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"Finally, functions for the complex potential and discharge function are defined and the capture zones are drawn. "
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {},
"outputs": [],
"source": [
"# functions for complex potential and complex discharge function\n",
"def omega(x, y):\n",
" zeta = x + y * 1j\n",
" om = -Wu * zeta + Q0 / (2 * np.pi) * np.log(zeta - zeta0) + Q1 / (2 * np.pi) * np.log(zeta - zeta1)\n",
" return om\n",
"\n",
"def wdis(x, y):\n",
" zeta = x + y * 1j\n",
" W = Wu - Q0 / (2 * np.pi * (zeta - zeta0)) - Q1 / (2 * np.pi * (zeta - zeta1))\n",
" return W"
]
},
{
"cell_type": "code",
"execution_count": 6,
"metadata": {
"tags": [
"remove-output"
]
},
"outputs": [
{
"data": {
"text/plain": [
"[]"
]
},
"execution_count": 6,
"metadata": {},
"output_type": "execute_result"
},
{
"data": {
"image/svg+xml": [
"\n",
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"# plot of head contours and capture zone envelopes\n",
"x, y = np.meshgrid(np.linspace(-800, 200, 100), np.linspace(-300, 400, 100))\n",
"om = omega(x, y)\n",
"w = wdis(x, y)\n",
"Qx, Qy = w.real, -w.imag\n",
"plt.subplot(111, aspect=1)\n",
"plt.contour(x, y, om.real, 10, colors='C0')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs0 + 1, ys0), (xs0 - 1, ys0)]), \n",
" density=100, color='C1')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs1 + 1, ys1), (xs1 - 1, ys1)]), \n",
" density=100, color='C2')\n",
"plt.plot([zeta0.real, zeta1.real], [zeta0.imag, zeta1.imag], 'k.')"
]
},
{
"cell_type": "code",
"execution_count": 7,
"metadata": {
"tags": [
"remove-input",
"remove-output"
]
},
"outputs": [
{
"data": {
"application/papermill.record/image/svg+xml": "\n\n\n",
"application/papermill.record/text/plain": ""
},
"metadata": {
"scrapbook": {
"mime_prefix": "application/papermill.record/",
"name": "fig_head_contours"
}
},
"output_type": "display_data"
},
{
"data": {
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
},
{
"data": {
"image/svg+xml": [
"\n",
"\n",
"\n"
],
"text/plain": [
""
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"# plot of head contours and capture zone envelopes\n",
"fig = plt.figure()\n",
"x, y = np.meshgrid(np.linspace(-866, 233, 100), np.linspace(-271, 410, 100))\n",
"om = omega(x, y)\n",
"w = wdis(x, y)\n",
"Qx, Qy = w.real, -w.imag\n",
"fig = plt.figure(figsize=diagrams_2d.figdef2)\n",
"plt.subplot(111, aspect=1)\n",
"plt.contour(x, y, om.real, 10, colors='C0')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs0 + 1, ys0)]), \n",
" density=100, color='C1', integration_direction='forward')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs0 + 1, ys0)]), \n",
" density=100, color='C1', integration_direction='backward')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs0 - 1, ys0)]), \n",
" density=100, color='C1', integration_direction='forward')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs0 - 1, ys0)]), \n",
" density=100, color='C1', integration_direction='backward')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs1 + 1, ys1)]), \n",
" density=100, color='C2', integration_direction='both')\n",
"# plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs1 + 1, ys1)]), \n",
"# density=100, color='C2', integration_direction='backward')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs1 - 1, ys1)]), \n",
" density=100, color='C2', integration_direction='forward')\n",
"plt.streamplot(x, y, Qx, Qy, start_points=np.array([(xs1 - 1, ys1)]), \n",
" density=100, color='C2', integration_direction='backward')\n",
"plt.plot([zeta0.real, zeta1.real], [zeta0.imag, zeta1.imag], 'k.')\n",
"plt.xlabel('x (m)')\n",
"plt.ylabel('y (m)')\n",
"glue(\"fig_head_contours\", fig, display=False)"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"```{glue:figure} fig_head_contours\n",
"---\n",
"name: fig_contour_head7.5\n",
"---\n",
"Capture zone envelopes for two wells in a uniform background flow for the same problem as {numref}`fig_captwowells_7.2`.\n",
"```"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"```{exercise}\n",
":label: ex_79_redo_74\n",
"\n",
"Repeat {numref}`ex_locstag2_7.1` but with complex variables. \n",
"\n",
"\n",
"```"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"```{exercise}\n",
":label: ex_79_redo_77\n",
"\n",
"Repeat {numref}`ex_minQ_7.3` but with complex variables. \n",
"\n",
"\n",
"```"
]
}
],
"metadata": {
"celltoolbar": "Tags",
"hide_code_all_hidden": false,
"kernelspec": {
"display_name": "Python 3 (ipykernel)",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.13.7"
},
"latex_envs": {
"LaTeX_envs_menu_present": true,
"autoclose": false,
"autocomplete": true,
"bibliofile": "biblio.bib",
"cite_by": "apalike",
"current_citInitial": 1,
"eqLabelWithNumbers": true,
"eqNumInitial": 1,
"hotkeys": {
"equation": "Ctrl-E",
"itemize": "Ctrl-I"
},
"labels_anchors": false,
"latex_user_defs": false,
"report_style_numbering": false,
"user_envs_cfg": false
}
},
"nbformat": 4,
"nbformat_minor": 4
}