Pre-allocate function to store vorticity (#180)

* Pre-allocate function to store vorticity * Project instead of interpolate vorticity
dynamicslab · Apr 7, 2024 · 7462583 · 7462583
1 parent be10c46
commit 7462583
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/examples/cylinder/pressure-probes.py b/examples/cylinder/pressure-probes.py
@@ -30,7 +30,7 @@
     Re=100,
     mesh=mesh,
     velocity_order=velocity_order,
-    observation_type="pressure_probes",
+    observation_type="vorticity_probes",
     probes=probes,
 )
 

diff --git a/hydrogym/firedrake/flow.py b/hydrogym/firedrake/flow.py
@@ -151,6 +151,8 @@ def initialize_state(self):
 
     self.split_solution()  # Break out and rename main solution
 
+    self._vorticity = fd.Function(self.pressure_space, name="vort")
+
   def set_state(self, q: fd.Function):
     """Set the current state fields
 
@@ -207,9 +209,8 @@ def vorticity(self, u: fd.Function = None) -> fd.Function:
         """
     if u is None:
       u = self.u
-    vort = fd.project(curl(u), self.pressure_space)
-    vort.rename("vort")
-    return vort
+    self._vorticity.project(curl(u))
+    return self._vorticity
 
   def function_spaces(self, mixed: bool = True):
     """Function spaces for velocity and pressure