Fix #98

Author: YingboMa

src/vector_of_array.jl

@@ -39,13 +39,26 @@ DiffEqArray(vec::AbstractVector{VT},ts::AbstractVector) where {T, N, VT<:Abstrac
 @inline Base.getindex(VA::AbstractVectorOfArray{T, N}, I::AbstractArray{Int}) where {T, N} = VectorOfArray(VA.u[I])
 @inline Base.getindex(VA::AbstractDiffEqArray{T, N}, I::AbstractArray{Int}) where {T, N} = DiffEqArray(VA.u[I],VA.t[I])
 @inline Base.getindex(VA::AbstractVectorOfArray{T, N}, i::Int,::Colon) where {T, N} = [VA.u[j][i] for j in 1:length(VA)]
+Base.@propagate_inbounds function Base.getindex(VA::AbstractVectorOfArray{T,N}, ii::CartesianIndex) where {T, N}
+    ti = Tuple(ii)
+    i = first(ti)
+    jj = CartesianIndex(Base.tail(ti))
+    return VA.u[i][jj]
+end
 @inline Base.setindex!(VA::AbstractVectorOfArray{T, N}, v, I::Int) where {T, N} = VA.u[I] = v
 @inline Base.setindex!(VA::AbstractVectorOfArray{T, N}, v, I::Colon) where {T, N} = VA.u[I] = v
 @inline Base.setindex!(VA::AbstractVectorOfArray{T, N}, v, I::AbstractArray{Int}) where {T, N} = VA.u[I] = v
 @inline function Base.setindex!(VA::AbstractVectorOfArray{T, N}, v, i::Int,::Colon) where {T, N}
   for j in 1:length(VA)
     VA.u[j][i] = v[j]
   end
+  return v
+end
+Base.@propagate_inbounds function Base.setindex!(VA::AbstractVectorOfArray{T,N}, x, ii::CartesianIndex) where {T, N}
+    ti = Tuple(ii)
+    i = first(ti)
+    jj = CartesianIndex(Base.tail(ti))
+    return VA.u[i][jj] = x
 end
 
 # Interface for the two dimensional indexing, a more standard AbstractArray interface

test/basic_indexing.jl

@@ -80,3 +80,8 @@ a = testva .+ rand(3,3)
 recs = [rand(2,2) for i in 1:5]
 testva = VectorOfArray(recs)
 @test Array(testva) isa Array{Float64,3}
+
+v = VectorOfArray([zeros(20), zeros(10,10), zeros(3,3,3)])
+v[CartesianIndex((3, 2, 3, 2))] = 1
+@test v[CartesianIndex((3, 2, 3, 2))] == 1
+@test v.u[3][2, 3, 2] == 1