Mercurial > repos > public > sbplib_julia

comparison src/Grids/tensor_grid.jl @ 1437:e3a80ef08d09

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Implement size(g,d) for EquidistantGrid and TensorGrid
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 22 Nov 2023 17:34:35 +0100
parents e82240df974d
children 66ad232e45a1
comparison
equal deleted inserted replaced
1421:69c9e6eae686 1437:e3a80ef08d09
44 44
45 Base.IteratorSize(::Type{<:TensorGrid{<:Any, D}}) where D = Base.HasShape{D}() 45 Base.IteratorSize(::Type{<:TensorGrid{<:Any, D}}) where D = Base.HasShape{D}()
46 Base.eltype(::Type{<:TensorGrid{T}}) where T = T 46 Base.eltype(::Type{<:TensorGrid{T}}) where T = T
47 Base.length(g::TensorGrid) = sum(length, g.grids) 47 Base.length(g::TensorGrid) = sum(length, g.grids)
48 Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...) 48 Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...)
49 Base.size(g::TensorGrid, d) = size(g)[d]
49 50
50 51
51 refine(g::TensorGrid, r::Int) = mapreduce(g->refine(g,r), TensorGrid, g.grids) 52 refine(g::TensorGrid, r::Int) = mapreduce(g->refine(g,r), TensorGrid, g.grids)
52 coarsen(g::TensorGrid, r::Int) = mapreduce(g->coarsen(g,r), TensorGrid, g.grids) 53 coarsen(g::TensorGrid, r::Int) = mapreduce(g->coarsen(g,r), TensorGrid, g.grids)
53 54