Merge pull request #3484 from CliMA/zs/add_zg_diagnostics

add geopotential height diagnostics
CliMA · Dec 16, 2024 · 0b0103f · 0b0103f
2 parents 3cc8b13 + 000e0cb
commit 0b0103f
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/diagnostics/core_diagnostics.jl b/src/diagnostics/core_diagnostics.jl
@@ -210,6 +210,23 @@ add_diagnostic_variable!(
     end,
 )
 
+###
+# Geopotential height (3d)
+###
+add_diagnostic_variable!(
+    short_name = "zg",
+    long_name = "Geopotential Height",
+    standard_name = "geopotential_height",
+    units = "m",
+    compute! = (out, state, cache, time) -> begin
+        if isnothing(out)
+            return cache.core.ᶜΦ ./ CAP.grav(cache.params)
+        else
+            out .= cache.core.ᶜΦ ./ CAP.grav(cache.params)
+        end
+    end,
+)
+
 ###
 # Cloud fraction (3d)
 ###

diff --git a/src/diagnostics/default_diagnostics.jl b/src/diagnostics/default_diagnostics.jl
@@ -141,6 +141,7 @@ function core_default_diagnostics(output_writer, duration, start_date)
         "thetaa",
         "ha",
         "pfull",
+        "zg",
         "rhoa",
         "ua",
         "va",
-Original file line number
+Diff line change
@@ Expand Up @@
             "thetaa",
             "ha",
             "pfull",
+            "zg",
             "rhoa",
             "ua",
             "va",
@@ Expand Down @@