diff src/Grids/equidistant_grid.jl @ 1437:e3a80ef08d09

Implement size(g,d) for EquidistantGrid and TensorGrid
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 22 Nov 2023 17:34:35 +0100
parents adbdbf1c3480
children 05eb8ba149e2
line wrap: on
line diff
--- a/src/Grids/equidistant_grid.jl	Fri Aug 25 08:49:07 2023 +0200
+++ b/src/Grids/equidistant_grid.jl	Wed Nov 22 17:34:35 2023 +0100
@@ -29,6 +29,7 @@
 Base.IteratorSize(::Type{<:EquidistantGrid}) = Base.HasShape{1}()
 Base.length(g::EquidistantGrid) = length(g.points)
 Base.size(g::EquidistantGrid) = size(g.points)
+Base.size(g::EquidistantGrid, d) = size(g.points)[d]
 
 
 """