use Galewsky.H0 in checkpointing script

firedrakeproject · Apr 23, 2024 · 9e491f9 · 9e491f9
1 parent 2327d81
commit 9e491f9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 14 deletions.
diff --git a/case_studies/shallow_water/checkpoints/galewsky_checkpoints.py b/case_studies/shallow_water/checkpoints/galewsky_checkpoints.py
@@ -4,7 +4,6 @@
 
 from utils import units
 from utils import mg
-from utils.misc import function_mean
 from utils.planets import earth
 import utils.shallow_water as swe
 from utils.shallow_water import galewsky
@@ -75,7 +74,7 @@
 w1 = fd.Function(W).assign(w_initial)
 
 # mean height
-H = function_mean(h_initial)
+H = galewsky.H0
 
 
 # shallow water equation forms

diff --git a/utils/misc.py b/utils/misc.py
@@ -33,18 +33,6 @@ def function_minimum(f):
     return fmin.data[0]
 
 
-def function_mean(f):
-    """
-    Find the volume averaged mean of a function.
-    """
-    mesh = f.function_space().mesh()
-    cells = fd.Function(fd.FunctionSpace(mesh, "DG", 0))
-    cells.assign(1)
-    area = fd.assemble(cells*fd.dx)
-    ftotal = fd.assemble(f*fd.dx)
-    return ftotal / area
-
-
 def curl0(u):
     """
     Curl function from y-cpt field to x-z field