Allow overloading on axes for paddeddata (#266)

JuliaArrays · Jul 25, 2023 · 7d7a939 · 7d7a939 · dlfivefifty · Jul 25, 2023
1 parent 1ebb490
commit 7d7a939
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "LazyArrays"
 uuid = "5078a376-72f3-5289-bfd5-ec5146d43c02"
-version = "1.5.0"
+version = "1.5.1"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

diff --git a/src/padded.jl b/src/padded.jl
@@ -488,7 +488,8 @@ const PaddedVector{T,M} = PaddedArray{T,1,M}
 const PaddedMatrix{T,M} = PaddedArray{T,2,M}
 
 MemoryLayout(::Type{<:PaddedArray{T,N,M}}) where {T,N,M} = PaddedLayout{typeof(MemoryLayout(M))}()
-paddeddata(A::PaddedArray) = A.args[2]
+paddeddata(A::PaddedArray) = paddeddata_axes(axes(A), A)
+paddeddata_axes(_, A) = A.args[2]
 
 PaddedArray(A::AbstractArray{T,N}, n::Vararg{Integer,N}) where {T,N} = PaddedArray(A, map(oneto,n))
 PaddedArray(A::AbstractArray{T,N}, ax::NTuple{N,Any}) where {T,N} = ApplyArray{T,N}(setindex, Zeros{T,N}(ax), A, axes(A)...)