Merge pull request #220 from dehann/patch-1

Base.length dispatch for ArrayPartition (StaticArrays)

Author: ChrisRackauckas

src/array_partition.jl

@@ -503,3 +503,11 @@ function Base.convert(::Type{ArrayPartition{T,S}}, A::ArrayPartition{<:Any,<:NTu
         ntuple((@inline i -> convert(S.parameters[i], A.x[i])), Val(N))
     )
 end
+            
+@generated function Base.length(::Type{<:ArrayPartition{F,T}}) where {F,N,T <: NTuple{N, StaticArraysCore.StaticArray}} 
+    sum_expr = Expr(:call, :+)
+    for param in T.parameters
+        push!(sum_expr.args, :(length($param)))
+    end
+    return sum_expr
+end

test/partitions_and_static_arrays.jl

@@ -8,3 +8,6 @@ p = ArrayPartition((zeros(Float32, 2), zeros(SMatrix{2,2, Int64},2), zeros(SVect
 
 p2 = similar(p)
 @test typeof(p2)==typeof(p)
+
+p3 = ArrayPartition(SA[1.0, 2.0], MMatrix{2,2}([3.0 4.0; 3.0 5.0]))
+@test (@inferred length(typeof(p3))) == 6