{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "e6689343-4eee-471c-9245-e70388fd2bc6",
   "metadata": {},
   "source": [
    "# Modeling a higher order head line-sink with variable head"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "337fc43e-8f4c-405e-9d7e-b097a1c2fded",
   "metadata": {},
   "outputs": [],
   "source": [
    "import matplotlib.pyplot as plt\n",
    "import numpy as np\n",
    "\n",
    "import timflow.transient as tft\n",
    "\n",
    "plt.rcParams[\"font.size\"] = 8.0\n",
    "plt.rcParams[\"figure.figsize\"] = (4, 3)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3c108473-8937-4b82-8151-106ca0b4331e",
   "metadata": {},
   "outputs": [],
   "source": [
    "ml = tft.ModelMaq(\n",
    "    kaq=10, z=[11, 10, 0], Saq=1e-4, c=1000, topboundary=\"semi\", tmin=0.01, tmax=10\n",
    ")\n",
    "hls = tft.RiverHo(\n",
    "    model=ml, x1=-1000, y1=-500, x2=1000, y2=500, tsandh=[(0, 2), (5, 4)], order=5\n",
    ")\n",
    "ml.solve()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "cf5fa918-ef1b-456b-85d3-2253d80b4591",
   "metadata": {},
   "outputs": [],
   "source": [
    "t = np.linspace(0.01, 10, 100)\n",
    "h = ml.head(x=-200, y=400, t=t)\n",
    "plt.plot(t, h[0])\n",
    "plt.grid()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "87a2e713-0597-4500-b7a3-2d98fdaaa28a",
   "metadata": {},
   "outputs": [],
   "source": [
    "ml.plots.contour(win=[-2000, 2000, -1000, 1000], ngr=50, t=3);"
   ]
  }
 ],
 "metadata": {
  "language_info": {
   "name": "python"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}