adds convenient vector-like syntax

Author: rokke

README.md

@@ -32,6 +32,12 @@ vA = VectorOfArray(a)
 vB = VectorOfArray(b)
 
 vA .* vB # Now all standard array stuff works!
+
+# you can also create it directly with a vector-like syntax:
+c = VA[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
+d = VA[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
+
+c .* d
 ```
 
 ### ArrayPartition
@@ -44,15 +50,15 @@ pB = ArrayPartition(b)
 
 pA .* pB # Now all standard array stuff works!
 
-# or do:
+# or using the vector syntax:
 x0 = rand(3, 3)
 v0 = rand(3, 3)
 a0 = rand(3, 3)
-u0 = ArrayPartition(x0, v0, a0)
-u0.x[1] == x0 # true
+u0 = AP[x0, v0, a0]
+u0.x[1] === x0 # true
 
 u0 .+= 1
-u0.x[2] == v0 # still true
+u0.x[2] === v0 # still true
 
 # do some calculations creating a new partitioned array
 unew = u0 * 10

src/RecursiveArrayTools.jl

@@ -133,14 +133,14 @@ module RecursiveArrayTools
     Base.convert(T::Type{<:GPUArraysCore.AnyGPUArray}, VA::AbstractVectorOfArray) = stack(VA.u)
     (T::Type{<:GPUArraysCore.AnyGPUArray})(VA::AbstractVectorOfArray) = T(Array(VA))
 
-    export VectorOfArray, DiffEqArray, AbstractVectorOfArray, AbstractDiffEqArray,
+    export VectorOfArray, VA, DiffEqArray, AbstractVectorOfArray, AbstractDiffEqArray,
         AllObserved, vecarr_to_vectors, tuples
 
     export recursivecopy, recursivecopy!, recursivefill!, vecvecapply, copyat_or_push!,
         vecvec_to_mat, recursive_one, recursive_mean, recursive_bottom_eltype,
         recursive_unitless_bottom_eltype, recursive_unitless_eltype
 
-    export ArrayPartition, NamedArrayPartition
+    export ArrayPartition, AP, NamedArrayPartition
 
     include("precompilation.jl")
 

src/array_partition.jl

@@ -722,3 +722,6 @@ end
 function Adapt.adapt_structure(to, ap::ArrayPartition)
     return ArrayPartition(map(x -> Adapt.adapt(to, x), ap.x)...)
 end
+
+struct AP end
+Base.getindex(::Type{AP}, xs...) = ArrayPartition(xs...)

src/vector_of_array.jl

@@ -1655,3 +1655,6 @@ end
 end
 unpack_args_voa(i, args::Tuple{Any}) = (unpack_voa(args[1], i),)
 unpack_args_voa(::Any, args::Tuple{}) = ()
+
+struct VA end
+Base.getindex(::Type{VA}, xs...) = VectorOfArray(collect(xs))