Merge pull request #197 from PartitionedArrays/jagged_array

Added AbstractJaggedArray type

Author: fverdugo

src/jagged_array.jl

@@ -45,6 +45,8 @@ Return `a.data`
 """
 getdata(a) = a.data
 
+abstract type AbstractJaggedArray{T} <: AbstractVector{T} end
+
 """
     struct GenericJaggedArray{V,A,B}
 
@@ -61,7 +63,7 @@ object.
 
 Given `a::GenericJaggedArray`, `V` is `typeof(view(a.data,a.ptrs[i]:(a.ptrs[i+1]-1)))`.
 """
-struct GenericJaggedArray{T,A,B} <: AbstractVector{T}
+struct GenericJaggedArray{T,A,B} <: AbstractJaggedArray{T}
   data::A
   ptrs::B
   @doc """
@@ -104,7 +106,7 @@ range `a.ptrs[i]:(a.ptrs[i+1]-1)`. The number of inner vectors (i.e. `length(a)`
 Given `a::JaggedArray`, `V` is `typeof(view(a.data,a.ptrs[i]:(a.ptrs[i+1]-1)))`.
 
 """
-struct JaggedArray{T,Ti} <: AbstractVector{SubArray{T,1,Vector{T},Tuple{UnitRange{Ti}},true}}
+struct JaggedArray{T,Ti} <: AbstractJaggedArray{SubArray{T,1,Vector{T},Tuple{UnitRange{Ti}},true}}
   data::Vector{T}
   ptrs::Vector{Ti}
   @doc """

src/mpi_array.jl

@@ -325,7 +325,7 @@ function gather_impl!(
     destination, ::Type{T}) where T <: AbstractVector
     Tv = eltype(snd.item)
     @assert rcv.comm === snd.comm
-    @assert isa(rcv.item,JaggedArray)
+    @assert isa(rcv.item,AbstractJaggedArray)
     @assert eltype(eltype(rcv.item)) == Tv
     comm = snd.comm
     if isa(destination,Integer)
@@ -389,7 +389,7 @@ function scatter_impl!(
     source,::Type{T}) where T<:AbstractVector
     @assert source !== :all "All to all not implemented"
     @assert rcv.comm === snd.comm
-    @assert isa(snd.item,JaggedArray)
+    @assert isa(snd.item,AbstractJaggedArray)
     @assert eltype(eltype(snd.item)) == eltype(rcv.item)
     comm = snd.comm
     root = source - 1
@@ -583,9 +583,9 @@ function exchange_impl!(
     @assert graph.rcv.comm === graph.rcv.comm
     @assert graph.rcv.comm === graph.snd.comm
     comm = graph.rcv.comm
-    data_snd = JaggedArray(snd.item)
+    data_snd = jagged_array(snd.item)
     data_rcv = rcv.item
-    @assert isa(data_rcv,JaggedArray)
+    @assert isa(data_rcv,AbstractJaggedArray)
     req_all = setup
     ireq = 0
     state = (snd,rcv)