Revisions based on V1 equations

Author: sbrus89

components/omega/doc/Makefile

@@ -6,7 +6,7 @@
 SPHINXOPTS    ?=
 SPHINXBUILD   ?= sphinx-build
 SOURCEDIR     = .
-BUILDDIR      = /global/cfs/cdirs/e3sm/www/sbrus/pgrad
+BUILDDIR      = _build
 
 # Put it first so that "make" without argument is like "make help".
 help:

components/omega/doc/design/OmegaV1GoverningEqns.md

@@ -1,3 +1,4 @@
+(omega-design-gonverning-equations-omega1)=
 # Omega V1: Governing Equations
 
 <!--
@@ -726,6 +727,7 @@ In the tracer equation, we note that surface fluxes (e.g. latent heat fluxes) wi
 
 Omega will only predict the layer average normal velocity, so we drop the bold face on the $u$ terms except for the product of primes, which is specified in the next section.
 
+(omega-v1-momentum-eq)=
 $$
 \frac{\partial u_{e,k}}{\partial t}
 & + \left[ {\bf k} \cdot \nabla \times u_{e,k} +f_v\right]_e\left(u_{e,k}^{\perp}\right) + \left[\nabla K\right]_e  \\

components/omega/doc/design/PGrad.md

@@ -5,7 +5,7 @@
 ## 1 Overview
 The pressure gradient will be responsible for computing the horizontal gradients of both the pressure and geopotential terms for the non-Boussinesq primitive equations implemented in Omega.
 In the non-Boussinesq model, the conserved quantity is mass rather than volume.
-In Omega the prognostic variable $\tilde{h}$ is a pseudo thickness, rather than geometric thickness in m as in a Boussinesq model.
+In Omega the prognostic variable $\tilde{h}$ is a pseudo thickness, rather than geometric thickness as in a Boussinesq model.
  Some non-Boussinesq models are written in pressure coordinates (e.g. [de Szoeke and Samelson 2002](https://journals.ametsoc.org/view/journals/phoc/32/7/1520-0485_2002_032_2194_tdbtba_2.0.co_2.xml).
  However, Omega is written in general vertical coordinates and can reference either pressure $p$ or distance $z$ in the vertical.
 In a pure pressure coordinate the pressure gradient term disappears (since the pressure does not vary along lines of constant pressure), just as how the geopotential term disappears in a pure z coordinate model.
@@ -18,6 +18,7 @@ This means that Omega will need to compute both the pressure and geopotential gr
 
 The pressure gradient will compute the horizontal gradients of both the pressure and geopotential to support tilted pressure coordinates in the non-Boussinesq model.
 This will allow for the use of a $p^\star$ coordinate, which functions similarly to the $z^\star$ in the Boussinesq MPAS-Ocean model.
+Note that we use the "$p^\star$" to refer to the vertical coordinate, in Omega it will be expressed in terms of the pseudo-height, $\tilde{z}$, as opposed to pressure directly.
 
 ### 2.2 Requirement: Initial support for a simple centered pressure gradient
 
@@ -34,7 +35,7 @@ In later versions of Omega, the pressure gradient will need to be able to includ
 These tidal forcings include both the tidal potential and the self attraction and loading terms.
 Additionally, other long-term changes to the geoid can be included in the geopotential.
 
-### 2.5 Disired: Pressure gradient for barotropic mode
+### 2.5 Desired: Pressure gradient for barotropic mode
 
 For split barotropic-baroclinic timestepping, the pressure gradient should provide the bottom pressure gradient tendency in the barotropic mode.
 This will be added in a future version when split time stepping is implemented.
@@ -43,22 +44,23 @@ This will be added in a future version when split time stepping is implemented.
 
 ## 3 Algorithmic Formulation
 ### 3.1 Centered Pressure Gradient
-In the layered non-Boussinesq [momentum equation](OmegaV1GoverningEqns.md#discrete-momentum) solved in Omega, the pressure gradient tendency term for edge $e$ and level $k$, $T^p_{e,k}$, includes the gradient of the pressure and the gradient of the geopotential,
+In the layered non-Boussinesq  {ref}`momentum equation <omega-v1-momentum-eq>` solved in Omega, the pressure gradient tendency term for edge $e$ and level $k$, $T^p_{e,k}$, includes the gradient of the geopotential, the gradient of a term involving pressure, and two terms evaluated at the cell interface:
 
 $$
-T^p_{e,k} = -\left[ \alpha_{i,k} \right]_e \nabla p_{i,k} - \nabla \Phi_{i,k},
+T^p_{e,k} = - \left(\nabla \Phi \right)_{e,k} - \frac{1}{\left[\tilde{h}_k\right]_e} \nabla \left( \tilde{h}_k \alpha_k p_k \right) + \frac{1}{\left[\tilde{h}_k\right]_e} \left\{ \left[ \alpha p \nabla \tilde{z}\right]_{e,k}^\text{top} -  \left[ \alpha p \nabla \tilde{z}\right]_{e,k+1}^\text{top} \right\}.
 $$
 
-where the second term is necessary to account for tilted layers that occur when using a general vertical coordinate.
-In this equation, $\alpha_{i,k}$ is the specific volume for cell $i$ at the mid-point of level $k$, $p_{i,k}$ is the pressure, and $\Phi_{i,k}$ is the geopotential.
+The geopotential and interface terms are necessary to account for tilted layers that occur when using a general vertical coordinate, where the gradient operator is taken along layers.
+In this equation, $\alpha_{i,k}$ specific volume, $p_{i,k}$ is the pressure, and $\Phi_{i,k}$ is the geopotential.
+These three quantities are evaluated at the mid-point of level $k$ of cell $i$ in the first two terms and at the cell interfaces in the third term along with the interface psudo-height, \tilde{z}.
 The discrete gradient operator at an edge is:
 
 $$
  \nabla {(\cdot)} = \frac{1}{d_e} \sum_{i\in CE(e)} -n_{e,i} (\cdot)_i
 $$
 
 where $d_e$ is the distance between cell centers, $CE(e)$ are the cells on edge $e$, and $n_{e,i}$ is the sign of the edge normal with respect to cell $i$.
-The horizontal averaging operator is:
+The (cell-to-edge) horizontal averaging operator is:
 
 $$
  [\cdot]_e = \frac{1}{2}\sum_{i\in CE(e)} (\cdot)_i
@@ -67,53 +69,60 @@ $$
 Therefore, the centered pressure gradient will be calculated as:
 
 $$
- T^p_{e,k} = \frac{1}{d_e} \left( [\alpha_{i,k}]_e \sum_{i \in CE(e)} n_{e,i} p_{k,i} + \sum_{i\in CE(e)} n_{e,i} \Phi_{k,i}\right),
-$$
-
-$$
- = \frac{1}{d_e} \left(  \sum_{i \in CE(e)} n_{e,i} \left( [\alpha_{i,k}]_ep_{k,i} + \Phi_{k,i} \right) \right),
-$$
-
-### 3.2 Barotropic Pressure Gradient
-
-When split baroclinic-barotropic time stepping is implemented in the future, the barotropic pressure gradient will be calculated by the pressure gradient class.
-The barotropic pressure gradient is found by depth integrating the pressure gradient.
-The pressure is
-
-$$
-p(z) = p_b - g \int^z_{-h} \rho dz^\prime,
-$$
-
-where $p_b$ is the bottom pressure.
-The bottom pressure is the sum of the atmospheric surface pressure, $p_s$, and the pressure contribution of the water column:
-
-$$
-p(z) &=  p_s + g\int_{-h}^\eta \rho dz - g \int^z_{-h} \rho dz^\prime, \\
-     &=  p_s + g\rho_0\widetilde{H} - g \int^z_{-h} \rho dz^\prime,
-$$
-
-where the total water column pseudo height is expressed by
-
-$$
-\widetilde{H} = \int_{-h}^\eta \frac{\rho}{\rho_0} dz.
-$$
-
-$\widetilde{H}$ is the prognositc variable in the barotropic continuity equation.
-The vertical integral of the pressure gradient is
-
-$$
-\frac{1}{\rho_0\widetilde{H}}\int^\eta_{-h} \nabla p dz &= \frac{1}{\rho_0\widetilde{H}}\int^\eta_{-h} \nabla \left( p_s + g\rho_0 \widetilde{H} - g \int^z_{-h} \rho dz^\prime \right) dz, \\
-                           &= \frac{H}{\rho_0\widetilde{H}}\nabla p_s + \frac{gH}{\widetilde{H}}\nabla \widetilde{H} - \frac{g}{\rho_0\widetilde{H}} \int_{-h}^\eta \left( \nabla \int_{-h}^z \rho dz^\prime\right) dz,
-$$
-
-where the height of the water column is represented by $H$.
-The $1/\rho_0\widetilde{H}$ factor comes vertically integrating the material derivative and expressing the resulting barotropic momentum equation in non-conservative form.
-
-Therefore, the barotorpic pressure gradient term is discretized as:
-
-$$
-\overline{T}_e^p = g\left[ \frac{H_i}{\widetilde{H}_i} \right]_e\sum_{i \in CE(e)} n_{e,i}\widetilde{H}_e
-$$
+ T^p_{e,k} = \frac{1}{d_e}\sum_{i\in CE(e)} n_{e,i} \Phi_{k,i} + \frac{2}{d_e\sum_{i \in CE(e)}\tilde{h}_{k,i}}\left(\sum_{i \in CE(e)} n_{e,i} \tilde{h}_{k,i}\alpha_{k,i}p_{k,i}\right. \\ \left. - \frac{1}{2} \sum_{i\in CE(e)} \alpha_{i,k-1/2} p_{i,k-1/2}\sum_{i\in CE(e)} n_{e,i}\tilde{z}_{i,k-1/2} \right.\\ \left.+ \frac{1}{2} \sum_{i\in CE(e)} \alpha_{i,k+1/2} p_{i,k+1/2}\sum_{i\in CE(e)} n_{e,i}\tilde{z}_{i,k+1/2}\right),
+$$
+
+
+### 3.2 High-order Pressure Gradient
+The high order pressure gradient will be based on the full volume integral
+
+$$
+- \int_A \int_{\tilde{z}_k^{\text{bot}}}^{\tilde{z}_k^{\text{top}}}  \nabla \left<\Phi\right>  d\tilde{z}  dA
+ + \int_{\partial A} \left( \int_{\tilde{z}_k^{\text{bot}}}^{\tilde{z}_k^{\text{top}}} \rho_0 \left(\left< \alpha \right> \left<p \right> + \left<\alpha^\prime p^\prime\right> \right) \, d\tilde{z} \right) dl \\
+ + \int_A \rho_0 \left[ \left< \alpha \right> \left<p \nabla \tilde{z}_k^{\text{top}} \right> + \left<\alpha^\prime \left(p \nabla \tilde{z}_k^{\text{top}}\right)^\prime\right> \right]_{\tilde{z} = \tilde{z}_k^{\text{top}}} \, dA \\
+ - \int_A \rho_0 \left[ \left< \alpha \right> \left<p \nabla \tilde{z}_k^{\text{bot}} \right> + \left<\alpha^\prime \left(p \nabla \tilde{z}_k^{\text{bot}}\right)^\prime\right> \right]_{\tilde{z} = \tilde{z}_k^{\text{bot}}} \, dA.
+$$
+
+%### 3.3 Barotropic Pressure Gradient
+%
+%When split baroclinic-barotropic time stepping is implemented in the future, the barotropic pressure gradient will be calculated by the pressure gradient class.
+%The barotropic pressure gradient is found by depth integrating the pressure gradient.
+%The pressure is
+%
+%$$
+%p(z) = p_b - g \int^z_{-h} \rho dz^\prime,
+%$$
+%
+%where $p_b$ is the bottom pressure.
+%The bottom pressure is the sum of the atmospheric surface pressure, $p_s$, and the pressure contribution of the water column:
+%
+%$$
+%p(z) &=  p_s + g\int_{-h}^\eta \rho dz - g \int^z_{-h} \rho dz^\prime, \\
+%     &=  p_s + g\rho_0\widetilde{H} - g \int^z_{-h} \rho dz^\prime,
+%$$
+%
+%where the total water column pseudo height is expressed by
+%
+%$$
+%\widetilde{H} = \int_{-h}^\eta \frac{\rho}{\rho_0} dz.
+%$$
+%
+%$\widetilde{H}$ is the prognositc variable in the barotropic continuity equation.
+%The vertical integral of the pressure gradient is
+%
+%$$
+%\frac{1}{\rho_0\widetilde{H}}\int^\eta_{-h} \nabla p dz &= \frac{1}{\rho_0\widetilde{H}}\int^\eta_{-h} \nabla \left( p_s + g\rho_0 \widetilde{H} - g \int^z_{-h} \rho dz^\prime \right) dz, \\
+%                           &= \frac{H}{\rho_0\widetilde{H}}\nabla p_s + \frac{gH}{\widetilde{H}}\nabla \widetilde{H} - \frac{g}{\rho_0\widetilde{H}} \int_{-h}^\eta \left( \nabla \int_{-h}^z \rho dz^\prime\right) dz,
+%$$
+%
+%where the height of the water column is represented by $H$.
+%The $1/\rho_0\widetilde{H}$ factor comes vertically integrating the material derivative and expressing the resulting barotropic momentum equation in non-conservative form.
+%
+%Therefore, the barotorpic pressure gradient term is discretized as:
+%
+%$$
+%\overline{T}_e^p = g\left[ \frac{H_i}{\widetilde{H}_i} \right]_e\sum_{i \in CE(e)} n_{e,i}\widetilde{H}_e
+%$$
 
 ## 4 Design
 
@@ -126,10 +135,11 @@ class PressureGrad{
     private:
         std::unique_ptr<PressureGrad> OmegaPressureGrad;
         PressureGradCentered CenteredPGrad;
-        PressureGradHighOrder HighOrderPGrad; // To be implemented later
+        PressureGradHighOrder HighOrderPGrad1; // To be implemented later
+        PressureGradHighOrder HighOrderPGrad2; // Multiple high order options are likely in the future
         PressureGradType PressureGradChoice;
         I4 NVertLevels;
-        I4 NChuncks;
+        I4 NChunks;
         Array2DI4 CellsOnEdge;
         Array1DReal DvEdge;
         Array1DReal EdgeSignOnCell;
@@ -144,7 +154,8 @@ An `enum class` will be used to specify options for the pressure gradient used f
 ```c++
 enum class PressureGradType{
    Centered,
-   HighOrder
+   HighOrder1,
+   highOrder2
 }
 ```
 The functions to compute the centered and high order pressure gradient terms will be implemented as functors and the pressure gradient class will have private instances of these classes.
@@ -164,12 +175,12 @@ class PressureGrad{
 
 The constructor will be responsible for storing any static mesh information as private variables and handling the selection of the user-specified pressure gradient option.
 ```c++
-PressureGrad::PressureGrad(const HorzMesh *Mesh, int NVertLevels, Config *Options);
+PressureGrad::PressureGrad(const HorzMesh *Mesh, const VertCoord *VCoord, Config *Options);
 ```
 
 #### 4.2.2 Initialization
 
-The init method will create the pressure gradient and return an error code:
+The init method will create the default pressure gradient and return an error code:
 ```c++
 int PressureGrad::init();
 ```
@@ -193,17 +204,25 @@ void PressureGrad::computePressureGrad(const Array2DReal &Tend,
                                        const Array2DReal &SpecVol) {
 OMEGA_SCOPE(LocCenteredPGrad, CenteredPGrad)
 OMEGA_SCOPE(LocHighOrderPGrad, HighOrderPGrad)
+const Array1DI4 &MinLyrEdgeBot   = VCoord->MinLayerEdgeBot;
+const Array1DI4 &MaxLyrEdgeTop   = VCoord->MaxLayerEdgeTop;
     if (PressureGradChoice == PressureGradType::Centered){
-       parallelFor("pgrad-centered", {NCellsAll, NChunks},
-       KOKKOS_LAMBDA(int ICell, int KChunk) {
-          LocCenteredPGrad(Tend, Pressure, Geopotential, SpecVol);
+       parallelForOuter(
+          {NEdgesAll}, KOKKOS_LAMBDA(int IEdge, const TeamMember &Team) {
+             const int NChunks = computeNChunks(MinLyrEdgeBottom, MaxLyrEdgeTop, IEdge);
+             parallelForInner(Team, NChunks, [=](const int KChunk) {
+                LocCenteredPGrad(Tend, IEdge, KChunk, Pressure, Geopotential, SpecVol);
+             });
        });
     }
     else if (PressureGradChoice == PressureGradType::HighOrder){
-       parallelFor("pgrad-highorder", {NCellsAll, NChunks},
-       KOKKOS_LAMBDA(int ICell, int KChunk) {
-          LocHighOrderPGrad(Tend, Pressure, Geopotential, SpecVol);
-        });
+       parallelForOuter(
+          {NEdgesAll}, KOKKOS_LAMBDA(int IEdge, const TeamMember &Team) {
+             const int NChunks = computeNChunks(MinLyrEdgeBottom, MaxLyrEdgeTop, IEdge);
+             parallelForInner(Team, NChunks, [=](const int KChunk) {
+                LocHighOrderPGrad(Tend, IEdge, KChunk, Pressure, Geopotential, SpecVol);
+             });
+       });
     }
 ```
 
@@ -218,7 +237,7 @@ void PressureGrad::clear();
 ## Verification and Testing
 
 ### Test: Spatial convergence to exact solution
-For a given analytical $v$,  $h$, and $b$, the spatial convergence of the pressure gradient can be assessed by computing errors on progressively finer meshes.
+For a given analytical $v$, $h$, and $b$, the spatial convergence of the pressure gradient can be assessed by computing errors on progressively finer meshes.
 
 ### Test: Baroclinic gyre
 The baroclinic gyre test case will test the pressure gradient term in the full non-Boussinesq equations.