Base.length dispatch for ArrayPartition

dehann · web-flow · commit 93d84eddd116 · 2022-07-15T11:49:38.000-07:00
This only works for partition elements that themselves have a length-on-type function, e.g. SVector or SMatrix.
diff --git a/src/array_partition.jl b/src/array_partition.jl
@@ -503,3 +503,5 @@ function Base.convert(::Type{ArrayPartition{T,S}}, A::ArrayPartition{<:Any,<:NTu
         ntuple((@inline i -> convert(S.parameters[i], A.x[i])), Val(N))
     )
 end
+            
+Base.length(::Type{<:ArrayPartition{F,T}}) where {F,T <: Tuple} = T.parameters .|> length |> sum

Original file line number	Diff line number	Diff line change
`@@ -503,3 +503,5 @@ function Base.convert(::Type{ArrayPartition{T,S}}, A::ArrayPartition{<:Any,<:NTu`
`503`	`503`	`ntuple((@inline i -> convert(S.parameters[i], A.x[i])), Val(N))`
`504`	`504`	`)`
`505`	`505`	`end`
	`506`	`+`
	`507`	`+Base.length(::Type{<:ArrayPartition{F,T}}) where {F,T <: Tuple} = T.parameters .\|> length \|> sum`