From f9e2c6aea1cf0f9de836f019acc5324d09e74474 Mon Sep 17 00:00:00 2001
From: Jim Wittig <jwittig@ucar.edu>
Date: Thu, 16 May 2024 15:32:53 -0600
Subject: [PATCH] Add stream_function and velocity_potential fields to the diag
 struct.

---
 src/core_atmosphere/Registry.xml | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/core_atmosphere/Registry.xml b/src/core_atmosphere/Registry.xml
index 4dbab8d9dc..f075bcdb87 100644
--- a/src/core_atmosphere/Registry.xml
+++ b/src/core_atmosphere/Registry.xml
@@ -1605,6 +1605,14 @@
                 <var name="vorticity" type="real" dimensions="nVertLevels nVertices Time" units="s^{-1}"
                      description="Relative vorticity at vertices"/>
 
+                <var name="stream_function" type="real" dimensions="nVertLevels nCells Time" units="m^2 s^{-1}"
+                     description="Stream function"
+                     packages="jedi_da"/>
+
+                <var name="velocity_potential" type="real" dimensions="nVertLevels nCells Time" units="m^2 s^{-1}"
+                     description="Velocity potential"
+                     packages="jedi_da"/>
+
                 <var name="pv_edge" type="real" dimensions="nVertLevels nEdges Time" units="s^{-1}"
                      description="absolute vertical vorticity averaged to the cell edge from the vertices"/>