def turbine_drag_function():
from firedrake_fluids.shallow_water import ShallowWater
sw = ShallowWater(path=os.path.join(cwd, "bump.swml"))
array = sw.get_turbine_array()
bump = assemble(array.turbine_drag() * dx)
sw = ShallowWater(path=os.path.join(cwd, "tophat.swml"))
array = sw.get_turbine_array()
tophat = assemble(array.turbine_drag() * dx)
return bump, tophat
def swe_mms_p2p1():
from firedrake_fluids.shallow_water import ShallowWater
configs = ["MMS_A", "MMS_B", "MMS_C"]
ux_norms = []
uy_norms = []
h_norms = []
for c in configs:
sw = ShallowWater(path=os.path.join(cwd, c + ".swml"))
solution = sw.run()
ux_old = solution.split()[0][0]
uy_old = solution.split()[0][1]
h_old = solution.split()[1]
fs_exact = FunctionSpace(sw.mesh, "CG", 3)
ux_exact = project(Expression("cos(x[0])*sin(x[1])"), fs_exact)
uy_exact = project(Expression("sin(x[0]*x[0]) + cos(x[1])"), fs_exact)
h_exact = project(Expression("sin(x[0])*sin(x[1])"), fs_exact)
h_norms.append(
sqrt(assemble(dot(h_old - h_exact, h_old - h_exact) * dx)))
ux_norms.append(
sqrt(assemble(dot(ux_old - ux_exact, ux_old - ux_exact) * dx)))
uy_norms.append(
sqrt(assemble(dot(uy_old - uy_exact, uy_old - uy_exact) * dx)))
return (h_norms, ux_norms, uy_norms)
def swe_tidal_flow_regular_bed():
from firedrake_fluids.shallow_water import ShallowWater
from firedrake_adjoint import project
sw = ShallowWater(
path=os.path.join(cwd, "swe_tidal_flow_regular_bed.swml"))
solution = sw.run()
mesh = sw.mesh
# Get the coordinates of each node
coordinates = mesh.coordinates
# For projecting to the same space as the coordinate field
vfs = VectorFunctionSpace(mesh, "CG", 1)
fs = FunctionSpace(mesh, "CG", 1)
u = solution.split()[0]
h = solution.split()[1]
u = project(u, vfs, annotate=False)
h = project(h, fs, annotate=False)
print coordinates.vector().array()
print u.vector().array()
print h.vector().array()
# Use [0::2] to select the even-numbered components of the vector's array (i.e. the x-component only).
return coordinates.vector().array()[0::2], u.vector().array(
)[0::2], h.vector().array()
def swe_steady_state(): from firedrake_fluids.shallow_water import ShallowWater sw = ShallowWater(path=os.path.join(cwd, "swe_steady_state.swml")) solution = sw.run() u_old = solution.split()[0] h_old = solution.split()[1] return u_old.vector().array(), h_old.vector().array()
def checkpoint():
from firedrake_fluids.shallow_water import ShallowWater
# Initialise the simulation from scratch.
sw = ShallowWater(path=os.path.join(cwd, "checkpoint.swml"))
solution = sw.run(checkpoint=None)
ux_initial = solution.split()[0]
h_initial = solution.split()[1]
# Initialise the simulation using the data in checkpoint.npy for the initial conditions.
sw = ShallowWater(path=os.path.join(cwd, "checkpoint.swml"))
solution = sw.run(checkpoint=os.path.join(cwd, "checkpoint.npy"))
ux_initial_from_checkpoint = solution.split()[0]
h_initial_from_checkpoint = solution.split()[1]
return ux_initial.vector().array(), h_initial.vector().array(
), ux_initial_from_checkpoint.vector().array(
), h_initial_from_checkpoint.vector().array()
def swe_standing_wave():
from firedrake_fluids.shallow_water import ShallowWater
sw = ShallowWater(path=os.path.join(cwd, "swe_standing_wave.swml"))
solution = sw.run()
h = solution.split()[-1]
return h.vector().array()
def swe_dam_break_1d():
from firedrake_fluids.shallow_water import ShallowWater
sw = ShallowWater(path=os.path.join(cwd, "swe_dam_break_1d.swml"))
solution = sw.run()
h = solution.split()[1]
u = solution.split()[0]
return h.vector().array(), u.vector().array()
def swe_wave_speed_1d():
from firedrake_fluids.shallow_water import ShallowWater
sw = ShallowWater(path=os.path.join(cwd, "swe_wave_speed_1d.swml"))
solution = sw.run()
h_old = solution.split()[-1]
return h_old.vector().array()
