Add optimized copyto! for ArrayOfSimilarArrays

Author: oschulz

src/array_of_similar_arrays.jl

@@ -165,6 +165,12 @@ function Base.similar(A::ArrayOfSimilarArrays{T,M,N}, ::Type{<:AbstractArray{U}}
 end
 
 
+function Base.copyto!(dest::ArrayOfSimilarArrays{T,M,N}, src::ArrayOfSimilarArrays{U,M,N}) where {T,M,N,U}
+    copyto!(dest.data, src.data)
+    dest
+end
+
+
 function Base.append!(dest::ArrayOfSimilarArrays{T,M,N}, src::ArrayOfSimilarArrays{U,M,N}) where {T,M,N,U}
     innersize(dest) != innersize(src) && throw(DimensionMismatch("Can't append, shape of element arrays of source and dest are not equal"))
     append!(dest.data, src.data)

test/array_of_similar_arrays.jl

@@ -111,6 +111,16 @@ using UnsafeArrays
         test_from_nested(VectorOfSimilarVectors, VectorOfSimilarVectors{Float64,Array{Float64,2}}, Val(1), Val(1))
     end
 
+
+    @testset "similar and copyto!" begin
+        A = ArrayOfSimilarArrays{Float64,1}(rand_flat_array(Val(1)))
+        @test (@inferred copyto!((@inferred similar(A)), A)) == A
+
+        A = ArrayOfSimilarArrays{Float64,2}(rand_flat_array(Val(5)))
+        @test (@inferred copyto!((@inferred similar(A)), A)) == A
+    end
+
+
     @testset "flatview" begin
         A = rand_nested_similar_arrays(Val(3), Val(2))
         B = ArrayOfSimilarArrays(A)