fix issue

CliMA · Jan 19, 2025 · 00b4525 · 00b4525
1 parent 234bc98
commit 00b4525
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/src/Operators/interpolation_operators.jl b/src/Operators/interpolation_operators.jl
@@ -83,14 +83,23 @@ using Oceananigans.Grids: Flat
 
 using Oceananigans.Grids: XFlatGrid, YFlatGrid, ZFlatGrid
 
-@inline ℑxᶜᵃᵃ(i, j, k, grid::XFlatGrid, u) = @inbounds u[i, j, k]
-@inline ℑxᶠᵃᵃ(i, j, k, grid::XFlatGrid, c) = @inbounds c[i, j, k]
+@inline ℑxᶜᵃᵃ(i, j, k, ::XFlatGrid, u) = @inbounds u[i, j, k]
+@inline ℑxᶠᵃᵃ(i, j, k, ::XFlatGrid, c) = @inbounds c[i, j, k]
 
-@inline ℑyᵃᶜᵃ(i, j, k, grid::YFlatGrid, w) = @inbounds w[i, j, k]
-@inline ℑyᵃᶠᵃ(i, j, k, grid::YFlatGrid, c) = @inbounds c[i, j, k]
+@inline ℑyᵃᶜᵃ(i, j, k, ::YFlatGrid, v) = @inbounds v[i, j, k]
+@inline ℑyᵃᶠᵃ(i, j, k, ::YFlatGrid, c) = @inbounds c[i, j, k]
 
-@inline ℑzᵃᵃᶜ(i, j, k, grid::ZFlatGrid, w) = @inbounds w[i, j, k]
-@inline ℑzᵃᵃᶠ(i, j, k, grid::ZFlatGrid, c) = @inbounds c[i, j, k]
+@inline ℑzᵃᵃᶜ(i, j, k, ::ZFlatGrid, w) = @inbounds w[i, j, k]
+@inline ℑzᵃᵃᶠ(i, j, k, ::ZFlatGrid, c) = @inbounds c[i, j, k]
+
+@inline ℑxᶜᵃᵃ(i, j, k, ::XFlatGrid, u::Number) = u
+@inline ℑxᶠᵃᵃ(i, j, k, ::XFlatGrid, c::Number) = c
+
+@inline ℑyᵃᶜᵃ(i, j, k, ::YFlatGrid, v::Number) = v
+@inline ℑyᵃᶠᵃ(i, j, k, ::YFlatGrid, c::Number) = c
+
+@inline ℑzᵃᵃᶜ(i, j, k, ::ZFlatGrid, w::Number) = w
+@inline ℑzᵃᵃᶠ(i, j, k, ::ZFlatGrid, c::Number) = c
 
 @inline ℑxᶜᵃᵃ(i, j, k, grid::XFlatGrid, f::F, args...) where {F<:Function} = f(i, j, k, grid, args...)
 @inline ℑxᶠᵃᵃ(i, j, k, grid::XFlatGrid, f::F, args...) where {F<:Function} = f(i, j, k, grid, args...)