fix: (drop me) add some constructor base dispatches

Author: avik-pal

Project.toml

@@ -6,6 +6,7 @@ version = "0.95.14"
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+ConstructionBase = "187b0558-2788-49d3-abe0-74a17ed4e7c9"
 Crayons = "a8cc5b0e-0ffa-5ad4-8c14-923d3ee1735f"
 CubedSphere = "7445602f-e544-4518-8976-18f8e8ae6cdb"
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
@@ -52,6 +53,7 @@ OceananigansReactantExt = ["Reactant", "KernelAbstractions"]
 [compat]
 Adapt = "4.1.1"
 CUDA = "5"
+ConstructionBase = "1.5.8"
 Crayons = "4"
 CubedSphere = "0.2, 0.3"
 Dates = "1.9"

src/Advection/centered_reconstruction.jl

@@ -36,6 +36,13 @@ struct Centered{N, FT, XT, YT, ZT, CA} <: AbstractCenteredAdvectionScheme{N, FT}
     end
 end
 
+# XXX: move to extension
+using ConstructionBase: ConstructionBase
+
+function ConstructionBase.constructorof(::Type{<:Centered{N, FT}}) where {N, FT}
+    return Centered{N, FT}
+end
+
 function Centered(FT::DataType = Float64; grid = nothing, order = 2) 
 
     if !(grid isa Nothing) 

src/Advection/upwind_biased_reconstruction.jl

@@ -38,6 +38,13 @@ struct UpwindBiased{N, FT, XT, YT, ZT, CA, SI} <: AbstractUpwindBiasedAdvectionS
     end
 end
 
+# XXX: move to extension
+using ConstructionBase: ConstructionBase
+
+function ConstructionBase.constructorof(::Type{<:UpwindBiased{N, FT}}) where {N, FT}
+    return UpwindBiased{N, FT}
+end
+
 function UpwindBiased(FT::DataType = Float64; grid = nothing, order = 3) 
 
     if !(grid isa Nothing) 

src/Advection/vector_invariant_advection.jl

@@ -29,6 +29,13 @@ struct VectorInvariant{N, FT, M, Z, ZS, V, K, D, U} <: AbstractAdvectionScheme{N
     end
 end
 
+# XXX: move to extension
+using ConstructionBase: ConstructionBase
+
+function ConstructionBase.constructorof(::Type{<:VectorInvariant{N, FT, M}}) where {N, FT, M}
+    return VectorInvariant{N, FT, M}
+end
+
 """
     VectorInvariant(; vorticity_scheme = EnstrophyConserving(),
                       vorticity_stencil = VelocityStencil(),

src/Advection/weno_reconstruction.jl

@@ -40,6 +40,13 @@ struct WENO{N, FT, XT, YT, ZT, PP, CA, SI} <: AbstractUpwindBiasedAdvectionSchem
     end
 end
 
+# XXX: move to extension
+using ConstructionBase: ConstructionBase
+
+function ConstructionBase.constructorof(::Type{<:WENO{N, FT}}) where {N, FT}
+    return WENO{N, FT}
+end
+
 """
     WENO([FT=Float64;] 
          order = 5,

src/Fields/field.jl

@@ -32,6 +32,13 @@ struct Field{LX, LY, LZ, O, G, I, D, T, B, S, F} <: AbstractField{LX, LY, LZ, G,
     end
 end
 
+# XXX: move to extension
+using ConstructionBase: ConstructionBase
+
+function ConstructionBase.constructorof(::Type{<:Field{LX, LY, LZ}}) where {LX, LY, LZ}
+    return Field{LX, LY, LZ}
+end
+
 #####
 ##### Constructor utilities
 #####

src/Grids/latitude_longitude_grid.jl

@@ -65,6 +65,13 @@ struct LatitudeLongitudeGrid{FT, TX, TY, TZ, CZ, M, MY, FX, FY, VX, VY, Arch} <:
                                                          Azᶠᶜᵃ, Azᶜᶠᵃ, Azᶠᶠᵃ, Azᶜᶜᵃ, radius)
 end
 
+# XXX: move to extension
+using ConstructionBase: ConstructionBase
+
+function ConstructionBase.constructorof(::Type{<:LatitudeLongitudeGrid{FT, TX, TY, TZ}}) where {FT, TX, TY, TZ}
+    return LatitudeLongitudeGrid{TX, TY, TZ}
+end
+
 const LLG = LatitudeLongitudeGrid
 const XRegularLLG    = LatitudeLongitudeGrid{<:Any, <:Any, <:Any, <:Any, <:Any, <:Any, <:Any, <:Number}
 const YRegularLLG    = LatitudeLongitudeGrid{<:Any, <:Any, <:Any, <:Any, <:Any, <:Any, <:Any, <:Any, <:Number}

src/Grids/rectilinear_grid.jl

@@ -38,6 +38,13 @@ function RectilinearGrid{TX, TY, TZ}(arch::Arch, Nx, Ny, Nz, Hx, Hy, Hz,
                                                      Δyᵃᶠᵃ, Δyᵃᶜᵃ, yᵃᶠᵃ, yᵃᶜᵃ, z)
 end
 
+# XXX: move to extension
+using ConstructionBase: ConstructionBase
+
+function ConstructionBase.constructorof(::Type{<:RectilinearGrid{FT, TX, TY, TZ}}) where {FT, TX, TY, TZ}
+    return RectilinearGrid{TX, TY, TZ}
+end
+
 const RG = RectilinearGrid
 
 const XRegularRG   = RectilinearGrid{<:Any, <:Any, <:Any, <:Any, <:Any, <:Number}