Add function nestedmap2 and deepmap

Author: oschulz

src/ArraysOfArrays.jl

@@ -11,6 +11,7 @@ using Compat: axes
 using UnsafeArrays
 
 include("util.jl")
+include("functions.jl")
 include("array_of_similar_arrays.jl")
 include("vector_of_arrays.jl")
 

src/array_of_similar_arrays.jl

@@ -178,6 +178,20 @@ UnsafeArrays.unsafe_uview(A::ArrayOfSimilarArrays{T,M,N}) where {T,M,N} =
     ArrayOfSimilarArrays{T,M,N}(uview(A.data))
 
 
+function nestedmap2(f::Base.Callable, A::ArrayOfSimilarArrays{T,M,N}) where {T,M,N}
+    new_data = map(f, A.data)
+    U = eltype(new_data)
+    ArrayOfSimilarArrays{U,M,N}(new_data)
+end
+
+
+function deepmap(f::Base.Callable, A::ArrayOfSimilarArrays{T,M,N}) where {T,M,N}
+    new_data = deepmap(f, A.data)
+    U = eltype(new_data)
+    ArrayOfSimilarArrays{U,M,N}(new_data)
+end
+
+
 
 const VectorOfSimilarArrays{
     T, M, L,

src/functions.jl

@@ -0,0 +1,29 @@
+# This file is a part of ArraysOfArrays.jl, licensed under the MIT License (MIT).
+
+
+"""
+    nestedmap2(f::Base.Callable, A::AbstractArray{<:AbstractArray})
+
+Nested `map` at depth 2. Equivalent to `map(X -> map(f, X) A)`.
+"""
+function nestedmap2 end
+export nestedmap2
+
+nestedmap2(f::Base.Callable, A::AbstractArray{<:AbstractArray{T,M},N}) where {T,M,N} =
+    map(X -> map(f, X), A)
+
+
+
+"""
+    deepmap(f::Base.Callable, A::AbstractArray)
+    deepmap(f::Base.Callable, A::AbstractArray{<:AbstractArray{<:...}})
+
+Applies `map` at the deepest possible layer of nested arrays.
+"""
+function deepmap end
+export deepmap
+
+deepmap(f::Base.Callable, x::Any) = map(f, x)
+
+deepmap(f::Base.Callable, A::AbstractArray{<:AbstractArray}) =
+    map(X -> deepmap(f, X), A)

src/vector_of_arrays.jl

@@ -300,6 +300,18 @@ function UnsafeArrays.uview(A::VectorOfArrays)
 end
 
 
+function nestedmap2(f::Base.Callable, A::VectorOfArrays)
+    new_data = map(f, A.data)
+    VectorOfArrays(new_data, A.elem_ptr, A.kernel_size, simple_consistency_checks)
+end
+
+
+function deepmap(f::Base.Callable, A::VectorOfArrays)
+    new_data = deepmap(f, A.data)
+    VectorOfArrays(new_data, A.elem_ptr, A.kernel_size, simple_consistency_checks)
+end
+
+
 
 const VectorOfVectors{
     T,