Add momentum budgets

.github/workflows/ci.yml

@@ -64,6 +64,31 @@ jobs:
         env:
           TEST_GROUP: "tracer_diagnostics"
 
+  u_momentum_diagnostics:
+    name: U-momentum diagnostics - Julia ${{ matrix.version }} - ${{ matrix.os }} - ${{ matrix.arch }}
+    runs-on: ${{ matrix.os }}
+    timeout-minutes: 120
+    strategy:
+      fail-fast: false
+      matrix:
+        version:
+          - '1.10'
+        os:
+          - ubuntu-latest
+        arch:
+          - x64
+    steps:
+      - uses: actions/checkout@v4
+      - uses: julia-actions/setup-julia@v2
+        with:
+          version: ${{ matrix.version }}
+          arch: ${{ matrix.arch }}
+      - uses: julia-actions/cache@v2
+      - uses: julia-actions/julia-buildpkg@v1
+      - uses: julia-actions/julia-runtest@v1
+        env:
+          TEST_GROUP: "u_momentum_diagnostics"
+
   ke_diagnostics:
     name: KE diagnostics - Julia ${{ matrix.version }} - ${{ matrix.os }} - ${{ matrix.arch }}
     runs-on: ${{ matrix.os }}

src/Oceanostics.jl

@@ -5,13 +5,19 @@ using Oceananigans.AbstractOperations: KernelFunctionOperation
 const CustomKFO{F} = KernelFunctionOperation{<:Any, <:Any, <:Any, <:Any, <:Any, F}
 
 #+++ Module export
-export TracerEquation, KineticEnergyEquation, TurbulentKineticEnergyEquation, TracerVarianceEquation, PotentialEnergyEquation
+export TracerEquation, KineticEnergyEquation, TurbulentKineticEnergyEquation, TracerVarianceEquation, PotentialEnergyEquation, UMomentumEquation
 #---
 
 #+++ TracerEquation exports
 export TracerAdvection, TracerDiffusion, TracerImmersedDiffusion, TracerTotalDiffusion, TracerForcing
 #---
 
+#+++ UMomentumEquation exports
+export UAdvection, UBuoyancyAcceleration, UCoriolisAcceleration, UPressureGradient,
+       UViscousDissipation, UImmersedViscousDissipation, UTotalViscousDissipation,
+       UStokesShear, UStokesTendency, UForcing, UTotalTendency
+#---
+
 #+++ TracerVarianceEquation exports
 export TracerVarianceTendency, TracerVarianceDissipationRate, TracerVarianceDiffusion
 #---
@@ -150,14 +156,15 @@ using Oceananigans.TurbulenceClosures: νᶜᶜᶜ
 #---
 
 include("TracerEquation.jl")
+include("UMomentumEquation.jl")
 include("TracerVarianceEquation.jl")
 include("KineticEnergyEquation.jl")
 include("TurbulentKineticEnergyEquation.jl")
 include("PotentialEnergyEquation.jl")
 include("FlowDiagnostics.jl")
 include("ProgressMessengers/ProgressMessengers.jl")
 
-using .TracerEquation, .TracerVarianceEquation, .KineticEnergyEquation, .TurbulentKineticEnergyEquation, .PotentialEnergyEquation
+using .TracerEquation, .UMomentumEquation, .TracerVarianceEquation, .KineticEnergyEquation, .TurbulentKineticEnergyEquation, .PotentialEnergyEquation
 using .FlowDiagnostics
 using .ProgressMessengers