fix space for LevelGrid

CliMA · Oct 28, 2023 · 9e15a8f · 9e15a8f
1 parent d8f0afb
commit 9e15a8f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/Spaces/spectralelement.jl b/src/Spaces/spectralelement.jl
@@ -43,6 +43,10 @@ struct SpectralElementSpace1D{G} <: AbstractSpectralElementSpace
 end
 space(grid::Grids.SpectralElementGrid1D, ::Nothing) =
     SpectralElementSpace1D(grid)
+space(
+    grid::Grids.LevelGrid{<:Grids.ExtrudedFiniteDifferenceField2D},
+    ::Nothing,
+) = SpectralElementSpace1D(grid)
 grid(space::Spaces.SpectralElementSpace1D) = getfield(space, :grid)
 
 function SpectralElementSpace1D(
@@ -98,6 +102,11 @@ struct SpectralElementSpace2D{G} <: AbstractSpectralElementSpace
 end
 space(grid::Grids.SpectralElementGrid2D, ::Nothing) =
     SpectralElementSpace2D(grid)
+space(
+    grid::Grids.LevelGrid{<:Grids.ExtrudedFiniteDifferenceField3D},
+    ::Nothing,
+) = SpectralElementSpace2D(grid)
+
 grid(space::Spaces.SpectralElementSpace2D) = getfield(space, :grid)
 
 function SpectralElementSpace2D(