Fix bug in DSS of level Fields

NEWS.md

@@ -4,6 +4,9 @@ ClimaCore.jl Release Notes
 main
 -------
 
+ - Fixed bug in DSS for level fields. PR [2194](https://github.com/CliMA/ClimaCore.jl/pull/2194).
+   Closed [2193](https://github.com/CliMA/ClimaCore.jl/issues/2193).
+
 v0.14.25
 -------
 

src/Grids/extruded.jl

@@ -131,8 +131,9 @@ vertical_topology(grid::ExtrudedFiniteDifferenceGrid) =
 
 # Since ∂z/∂ξ₃ and r are continuous across element boundaries, we can reuse
 # the horizontal weights instead of calling compute_dss_weights on the
-# extruded local geometry.
-dss_weights(grid::AbstractExtrudedFiniteDifferenceGrid, ::Staggering) =
+# extruded local geometry. If we ever need to use extruded weights, this method
+# will need to distinguish between weights on centers and weights on faces.
+dss_weights(grid::AbstractExtrudedFiniteDifferenceGrid, _) =
     dss_weights(grid.horizontal_grid, nothing)
 
 local_geometry_data(grid::AbstractExtrudedFiniteDifferenceGrid, ::CellCenter) =

src/Grids/level.jl

@@ -16,10 +16,10 @@ level(
 
 topology(levelgrid::LevelGrid) = topology(levelgrid.full_grid)
 
-dss_weights(levelgrid::LevelGrid{<:Any, Int}, ::Nothing) =
-    level(dss_weights(levelgrid.full_grid, CellCenter()), levelgrid.level)
-dss_weights(levelgrid::LevelGrid{<:Any, PlusHalf{Int}}, ::Nothing) =
-    level(dss_weights(levelgrid.full_grid, CellFace()), levelgrid.level + half)
+# The DSS weights for extruded spaces are currently the same as the weights for
+# horizontal spaces. If we ever need to use extruded weights, this method will
+# need to extract the weights at a particular level.
+dss_weights(levelgrid::LevelGrid, _) = dss_weights(levelgrid.full_grid, nothing)
 
 local_geometry_type(::Type{LevelGrid{G, L}}) where {G, L} =
     local_geometry_type(G)

test/Spaces/unit_dss.jl

@@ -138,7 +138,10 @@ end
             space_kwargs = (; context, topography, deep, autodiff_metric)
             center_space =
                 TU.CenterExtrudedFiniteDifferenceSpace(FT; space_kwargs...)
-            for space in (center_space, Spaces.face_space(center_space))
+            face_space = Spaces.face_space(center_space)
+            level_space1 = Spaces.level(center_space, 2)
+            level_space2 = Spaces.level(face_space, 1 + Spaces.half)
+            for space in (center_space, face_space, level_space1, level_space2)
                 test_dss_conservation(space)
             end
         end