Add circulation plot

Author: cbegeman

polaris/tasks/ocean/barotropic_channel/viz.py

@@ -48,16 +48,14 @@ def run(self):
         cell_mask = ds_init.maxLevelCell >= 1
         vertex_mask = ds_init.boundaryVertex == 0
 
-        vmax = 0.1
-        # vmax = np.max(np.abs(ds_init.normalVelocity.values))
-
         # Uncomment these lines to get 10 evenly spaced time slices
         # nt = ds_out.sizes['Time']
         # for t_index in np.arange(0, nt, int(np.floor(nt / 10))):
 
         # These indices correspond to the first and last time step
         for t_index in [0, -1]:
             ds = ds_out.isel(Time=t_index)
+            vmax = np.max(np.abs(ds.velocityZonal.values))
             plot_horiz_field(
                 ds_mesh,
                 ds['velocityZonal'],
@@ -87,3 +85,14 @@ def run(self):
                 cmap='cmo.balance',
                 field_mask=vertex_mask,
             )
+
+            vmax = np.max(np.abs(ds.circulation.values))
+            plot_horiz_field(
+                ds_mesh,
+                ds['circulation'],
+                f'circulation_t{t_index}.png',
+                vmin=-vmax,
+                vmax=vmax,
+                cmap='cmo.balance',
+                field_mask=vertex_mask,
+            )