Specialize Base.unsafe_view for ArrayOfSimilarArrays

Author: oschulz

src/array_of_similar_arrays.jl

@@ -135,6 +135,12 @@ Base.@propagate_inbounds Base.getindex(A::ArrayOfSimilarArrays{T,M,N}, idxs::Var
 Base.@propagate_inbounds Base.setindex!(A::ArrayOfSimilarArrays{T,M,N}, x::AbstractArray{U,M}, idxs::Vararg{Integer,N}) where {T,M,N,U} =
     setindex!(A.data, x, _ncolons(Val{M}())..., idxs...)
 
+Base.@propagate_inbounds function Base.unsafe_view(A::ArrayOfSimilarArrays{T,M,N}, idxs::Vararg{Union{Real, AbstractArray},N}) where {T,M,N}
+    dataview = view(A.data, _ncolons(Val{M}())..., idxs...)
+    L = length(size(dataview))
+    N_view = L - M
+    ArrayOfSimilarArrays{T,M,N_view}(dataview)
+end
 
 
 @inline function Base.resize!(A::ArrayOfSimilarArrays{T,M,N}, dims::Vararg{Integer,N}) where {T,M,N}