check linearisation of coriolis is required before adding it

firedrakeproject · Jan 24, 2023 · dc3ce62 · dc3ce62
1 parent 391de30
commit dc3ce62
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/gusto/equations.py b/gusto/equations.py
@@ -668,11 +668,12 @@ def __init__(self, domain, parameters, fexpr=None, bexpr=None,
                     subject(prognostic(f*inner(u, w)*dx, "u"), self.X)
                 ), domain.perp)
             # Add linearisation
-            linear_coriolis = perp(
-                coriolis(
-                    subject(prognostic(f*inner(u_trial, w)*dx, "u"), self.X)
-                ), domain.perp)
-            coriolis_form = linearisation(coriolis_form, linear_coriolis)
+            if self.linearisation_map(coriolis_form.terms[0]):
+                linear_coriolis = perp(
+                    coriolis(
+                        subject(prognostic(f*inner(u_trial, w)*dx, "u"), self.X)
+                    ), domain.perp)
+                coriolis_form = linearisation(coriolis_form, linear_coriolis)
             residual += coriolis_form
 
         if bexpr is not None: