Mercurial > repos > public > sbplib_julia
view src/Grids/grid.jl @ 1275:dcd8654ca33b refactor/grids
Remove getcomponent stubs since we might have a cleaner way to solve it with map or lazymap
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Sun, 26 Feb 2023 11:53:23 +0100 |
parents | 7fab13c07412 |
children | 1157f889bf50 |
line wrap: on
line source
""" Grid{T,D} The top level type for grids. Should implement # TBD: """ #TBD: Should it be an AbstractArray? See notes in grid_refactor.md # TODO: Document that grids should implement the interfaces for iteration and indexing. abstract type Grid{T,D} end Base.ndims(::Grid{T,D}) where {T,D} = D Base.eltype(::Type{<:Grid{T}}) where T = T target_manifold_dim(::Grid{T}) where T = _ncomponents(T) # TBD: Name of this function?! """ # TODO """ function refine end """ # TODO """ function coarsen end """ # TODO """ function boundary_identifiers end """ # TODO """ function boundary_grid end # TBD Can we implement a version here that accepts multiple ids and grouped boundaries? Maybe we need multiblock stuff? # TODO: Make sure that all grids implement all of the above. """ dims(grid::Grid) Enumerate the dimensions of the grid. """ dims(grid::Grid) = 1:ndims(grid) # TBD: Is this function needed? Where is it used? # TBD: New file grid_functions.jl? """ TODO: * Mention map(f,g) if you want a concrete array """ eval_on(g::Grid, f) = eval_on(g, f, Base.IteratorSize(g)) # TBD: Borde f vara först som i alla map, sum, och dylikt function eval_on(g::Grid, f, ::Base.HasShape) if hasmethod(f, (Any,)) return LazyTensors.LazyFunctionArray((I...)->f(g[I...]), size(g)) else return LazyTensors.LazyFunctionArray((I...)->f(g[I...]...), size(g)) end end # TODO: Explain how these are intended to be used _ncomponents(::Type{<:Number}) = 1 _ncomponents(T::Type{<:SVector}) = length(T) _component_type(T::Type{<:Number}) = T _component_type(T::Type{<:SVector}) = eltype(T)