Merge branch 'main' of https://github.com/underworldcode/underworld3-documentation

Author: lmoresi

Jupyterbook/Notebooks/Examples-Annulus-Stokes/Ex_Stokes_Annulus_Benchmark_Kramer.py

@@ -24,14 +24,11 @@
 # os.environ["SYMPY_USE_CACHE"]="no"
 
 # options.getAll()
-# +
-import meshio
-
+# -
 meshball = uw.meshing.Annulus(
     radiusOuter=1.0, radiusInner=0.5, cellSize=0.2, refinement=1, qdegree=3
 )
 x, y = meshball.X
-# -
 
 
 v_soln = uw.discretisation.MeshVariable("U", meshball, meshball.dim, degree=2)

Jupyterbook/Notebooks/Examples-Annulus-Stokes/Ex_Stokes_Annulus_Benchmark_Thieulot.py

@@ -0,0 +1,203 @@
+# ---
+# jupyter:
+#   jupytext:
+#     formats: py:light
+#     text_representation:
+#       extension: .py
+#       format_name: light
+#       format_version: '1.5'
+#       jupytext_version: 1.16.1
+#   kernelspec:
+#     display_name: Python 3 (ipykernel)
+#     language: python
+#     name: python3
+# ---
+
+# # Darcy flow (1d) using xy coordinates to define permeability distribution
+
+# to fix trame issue
+import nest_asyncio
+nest_asyncio.apply()
+
+# + editable=true slideshow={"slide_type": ""}
+from petsc4py import PETSc
+import underworld3 as uw
+import numpy as np
+import sympy
+from sympy import Piecewise, ceiling, Abs
+import matplotlib.pyplot as plt
+# %matplotlib inline
+
+options = PETSc.Options()
+
+# + editable=true slideshow={"slide_type": ""}
+# vis tools
+if uw.mpi.size == 1:
+    import pyvista as pv
+    import underworld3.visualisation as vis
+
+# + editable=true slideshow={"slide_type": ""}
+minX, maxX = -1.0, 0.0
+minY, maxY = -1.0, 0.0
+
+mesh = uw.meshing.UnstructuredSimplexBox(minCoords=(minX, minY), maxCoords=(maxX, maxY), 
+                                         cellSize=1/25, qdegree=3)
+
+# x and y coordinates
+x = mesh.N.x
+y = mesh.N.y
+
+# +
+# Create Darcy Solver
+darcy = uw.systems.SteadyStateDarcy(mesh)
+
+p_soln = darcy.Unknowns.u
+v_soln = darcy.v
+
+# Needs to be smaller than the contrast in properties
+darcy.petsc_options["snes_rtol"] = 1.0e-6  
+
+darcy.constitutive_model = uw.constitutive_models.DarcyFlowModel
+# -
+
+
+p_soln_0 = p_soln.clone("P_no_g", r"{p_\textrm{(no g)}}")
+v_soln_0 = v_soln.clone("V_no_g", r"{v_\textrm{(no g)}}")
+
+
+# + editable=true slideshow={"slide_type": ""}
+def plot_P_V(_mesh, _p_soln, _v_soln):
+    '''
+    Plot pressure and velcity streamlines
+    '''
+    pvmesh = vis.mesh_to_pv_mesh(_mesh)
+    pvmesh.point_data["P"] = vis.scalar_fn_to_pv_points(pvmesh, _p_soln.sym)
+    pvmesh.point_data["V"] = vis.vector_fn_to_pv_points(pvmesh, _v_soln.sym)
+
+    velocity_points = vis.meshVariable_to_pv_cloud(_v_soln)
+    velocity_points.point_data["V"] = vis.vector_fn_to_pv_points(velocity_points, _v_soln.sym)
+
+    # point sources at cell centres
+    points = np.zeros((_mesh._centroids.shape[0], 3))
+    points[:, 0] = _mesh._centroids[:, 0]
+    points[:, 1] = _mesh._centroids[:, 1]
+    point_cloud = pv.PolyData(points[::3])
+
+    pvstream = pvmesh.streamlines_from_source(point_cloud, vectors="V", integrator_type=45, 
+                                              integration_direction="both", max_steps=1000,
+                                              max_time=0.1, initial_step_length=0.001, 
+                                              max_step_length=0.01,)
+
+    pl = pv.Plotter(window_size=(750, 750))
+    pl.add_mesh(pvmesh, cmap="coolwarm", edge_color="Black", show_edges=True, scalars="P",
+                use_transparency=False, opacity=1.0,)
+    pl.add_mesh(pvstream, line_width=1.0)
+    # pl.add_arrows(velocity_points.points, velocity_points.point_data["V"], mag=0.005, opacity=0.75)
+
+    pl.show(cpos="xy")
+
+
+# +
+# set up two materials
+interfaceY = -0.25
+k1 = 1.0
+k2 = 1.0e-4
+
+# Groundwater pressure boundary condition on the bottom wall
+max_pressure = 0.5
+
+# The piecewise version
+kFunc = Piecewise((k1, y >= interfaceY), (k2, y < interfaceY), (1.0, True))
+
+# A smooth version
+darcy.constitutive_model.Parameters.permeability = kFunc
+darcy.constitutive_model.Parameters.s = sympy.Matrix([0, 0]).T
+darcy.f = 0.0
+
+# set up boundary conditions
+darcy.add_dirichlet_bc(0.0, "Top")
+darcy.add_dirichlet_bc(-1.0 * minY * max_pressure, "Bottom")
+# + editable=true slideshow={"slide_type": ""}
+# darcy solve without gravity
+darcy.solve()
+
+
+# + editable=true slideshow={"slide_type": ""}
+# plotting soln without gravity
+plot_P_V(mesh, p_soln, v_soln)
+
+# + editable=true slideshow={"slide_type": ""}
+# copying solution
+with mesh.access(p_soln_0, v_soln_0):
+    p_soln_0.data[...] = p_soln.data[...]
+    v_soln_0.data[...] = v_soln.data[...]
+# -
+
+# now switch on gravity
+darcy.constitutive_model.Parameters.s = sympy.Matrix([0, -1]).T
+darcy.solve()
+# + editable=true slideshow={"slide_type": ""}
+# plotting soln with gravity
+plot_P_V(mesh, p_soln, v_soln)
+
+# +
+# set up interpolation coordinates
+ycoords = np.linspace(minY + 0.001 * (maxY - minY), maxY - 0.001 * (maxY - minY), 100)
+xcoords = np.full_like(ycoords, -0.5)
+xy_coords = np.column_stack([xcoords, ycoords])
+
+pressure_interp = uw.function.evaluate(p_soln.sym[0], xy_coords)
+pressure_interp_0 = uw.function.evaluate(p_soln_0.sym[0], xy_coords)
+
+
+# + editable=true slideshow={"slide_type": ""}
+La = -1.0 * interfaceY
+Lb = 1.0 + interfaceY
+dP = max_pressure
+
+S = 1
+Pa = (dP / Lb - S + k1 / k2 * S) / (1.0 / Lb + k1 / k2 / La)
+pressure_analytic = np.piecewise(
+    ycoords,
+    [ycoords >= -La, ycoords < -La],
+    [
+        lambda ycoords: -Pa * ycoords / La,
+        lambda ycoords: Pa + (dP - Pa) * (-ycoords - La) / Lb,
+    ],
+)
+
+y = sympy.symbols('y')
+P1 = -Pa * y / La
+P2 = Pa + (dP - Pa) * (-y - La) / Lb
+print(P1)
+print(P2)
+
+S = 0
+Pa = (dP / Lb - S + k1 / k2 * S) / (1.0 / Lb + k1 / k2 / La)
+pressure_analytic_noG = np.piecewise(
+    ycoords,
+    [ycoords >= -La, ycoords < -La],
+    [
+        lambda ycoords: -Pa * ycoords / La,
+        lambda ycoords: Pa + (dP - Pa) * (-ycoords - La) / Lb,
+    ],
+)
+
+y = sympy.symbols('y')
+P1 = -Pa * y / La
+P2 = Pa + (dP - Pa) * (-y - La) / Lb
+print(P1)
+print(P2)
+
+# + editable=true slideshow={"slide_type": ""}
+# plotting numerical and analytical solution
+fig = plt.figure()
+ax1 = fig.add_subplot(111, xlabel="Pressure", ylabel="Depth")
+ax1.plot(pressure_interp, ycoords, linewidth=3, label="Numerical solution")
+ax1.plot(pressure_interp_0, ycoords, linewidth=3, label="Numerical solution (no G)")
+ax1.plot(pressure_analytic, ycoords, linewidth=3, linestyle="--", label="Analytic solution")
+ax1.plot(pressure_analytic_noG, ycoords, linewidth=3, linestyle="--", label="Analytic (no gravity)")
+ax1.grid("on")
+ax1.legend()
+# + editable=true slideshow={"slide_type": ""}
+