diff src/Grids/tensor_grid.jl @ 1448:0322c181a1cd feature/grids/curvilinear

Merge default
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 22 Nov 2023 17:53:31 +0100
parents 433245fd33c0 c0c5e0620b83
children d9d9ab18cdfc
line wrap: on
line diff
--- a/src/Grids/tensor_grid.jl	Thu Sep 07 16:51:46 2023 +0200
+++ b/src/Grids/tensor_grid.jl	Wed Nov 22 17:53:31 2023 +0100
@@ -47,6 +47,7 @@
 Base.IteratorSize(::Type{<:TensorGrid{<:Any, D}}) where D = Base.HasShape{D}()
 Base.length(g::TensorGrid) = prod(length, g.grids)
 Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...)
+Base.size(g::TensorGrid, d) = size(g)[d]
 
 
 refine(g::TensorGrid, r::Int) = mapreduce(g->refine(g,r), TensorGrid, g.grids)