comparison src/Grids/tensor_grid.jl @ 1440:28bd0ea7ee92

Merge refactor/grids/tensor_grid_eltype
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 22 Nov 2023 17:49:04 +0100
parents 66ad232e45a1
children c0c5e0620b83
comparison
equal deleted inserted replaced
1437:e3a80ef08d09 1440:28bd0ea7ee92
41 Base.iterate(g::TensorGrid, state) = iterate(Iterators.product(g.grids...), state) |> _iterate_combine_coords 41 Base.iterate(g::TensorGrid, state) = iterate(Iterators.product(g.grids...), state) |> _iterate_combine_coords
42 _iterate_combine_coords(::Nothing) = nothing 42 _iterate_combine_coords(::Nothing) = nothing
43 _iterate_combine_coords((next,state)) = combine_coordinates(next...), state 43 _iterate_combine_coords((next,state)) = combine_coordinates(next...), state
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
47 Base.length(g::TensorGrid) = sum(length, g.grids) 46 Base.length(g::TensorGrid) = sum(length, g.grids)
48 Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...) 47 Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...)
49 Base.size(g::TensorGrid, d) = size(g)[d] 48 Base.size(g::TensorGrid, d) = size(g)[d]
50 49
51 50