added potential temperatrue to output

exorad · Mar 11, 2024 · a5276c0 · a5276c0
1 parent 4b7a9ab
commit a5276c0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gcm_toolkit/exorad/utils.py b/gcm_toolkit/exorad/utils.py
@@ -79,7 +79,11 @@ def convert_winds_and_t(dsi, temp_dim, w_dim):
     """
     import cubedsphere as cs
     import cubedsphere.const as c
-
+
+    # save potential temperature
+    dsi[temp_dim + '_pot'] = dsi[temp_dim]
+
+    #calculate temperature
     kappa = dsi.attrs["R"] / dsi.attrs["cp"]
     dsi[temp_dim] = dsi[temp_dim] * (dsi[c.Z] / dsi.attrs["p_ref"]) ** kappa