updates to notebooks

firedrakeproject · Dec 18, 2024 · 4505adb · 4505adb
1 parent 3ec956d
commit 4505adb
Show file tree

Hide file tree

Showing 2 changed files with 212 additions and 215 deletions.
diff --git a/jupyter_notebooks/Book_3_Lowest_Order_Rising_Bubble.ipynb b/jupyter_notebooks/Book_3_Lowest_Order_Rising_Bubble.ipynb
@@ -39,7 +39,7 @@
      "traceback": [
       "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
       "\u001b[0;31mModuleNotFoundError\u001b[0m                       Traceback (most recent call last)",
-      "Input \u001b[0;32mIn [1]\u001b[0m, in \u001b[0;36m<cell line: 1>\u001b[0;34m()\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mgusto\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m \u001b[38;5;241m*\u001b[39m\n\u001b[1;32m      2\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mfiredrake\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m (IntervalMesh, ExtrudedMesh,\n\u001b[1;32m      3\u001b[0m                        SpatialCoordinate, conditional, cos, pi, sqrt,\n\u001b[1;32m      4\u001b[0m                        TestFunction, dx, TrialFunction, Constant, Function,\n\u001b[1;32m      5\u001b[0m                        LinearVariationalProblem, LinearVariationalSolver,\n\u001b[1;32m      6\u001b[0m                        FunctionSpace, BrokenElement, VectorFunctionSpace)\n",
+      "Input \u001b[0;32mIn [1]\u001b[0m, in \u001b[0;36m<cell line: 1>\u001b[0;34m()\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mgusto\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m \u001b[38;5;241m*\u001b[39m\n\u001b[1;32m      2\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mfiredrake\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m (IntervalMesh, ExtrudedMesh,\n\u001b[1;32m      3\u001b[0m                        SpatialCoordinate, conditional, cos, pi, sqrt,\n\u001b[1;32m      4\u001b[0m                        TestFunction, dx, TrialFunction, Constant, Function,\n\u001b[1;32m      5\u001b[0m                        LinearVariationalProblem, LinearVariationalSolver)\n",
       "\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'gusto'"
      ]
     }
@@ -49,8 +49,7 @@
     "from firedrake import (IntervalMesh, ExtrudedMesh,\n",
     "                       SpatialCoordinate, conditional, cos, pi, sqrt,\n",
     "                       TestFunction, dx, TrialFunction, Constant, Function,\n",
-    "                       LinearVariationalProblem, LinearVariationalSolver,\n",
-    "                       FunctionSpace, BrokenElement, VectorFunctionSpace)"
+    "                       LinearVariationalProblem, LinearVariationalSolver)"
    ]
   },
   {
@@ -262,6 +261,8 @@
     "# define perturbation\n",
     "xc = domain_width / 2\n",
     "zc = 2000.\n",
+    "rc = 2000.\n",
+    "Tdash = 2.0\n",
     "r = sqrt((xz[0] - xc) ** 2 + (xz[1] - zc) ** 2)\n",
     "theta_pert = Function(Vt).interpolate(\n",
     "    conditional(r > rc, 0.0, Tdash * (cos(pi * r / (2.0 * rc))) ** 2)\n",