Special case flatten of iterators of static arrays

mschauer · mschauer · commit 3d55528042a1 · 2018-09-27T15:42:23.000+02:00
diff --git a/src/StaticArrays.jl b/src/StaticArrays.jl
@@ -114,6 +114,7 @@ include("svd.jl")
 include("lu.jl")
 include("qr.jl")
 include("deque.jl")
+include("flatten.jl")
 include("io.jl")
 
 include("FixedSizeArrays.jl")
diff --git a/src/flatten.jl b/src/flatten.jl
@@ -0,0 +1,6 @@
+# Special case flatten of iterators of static arrays.
+import Base.Iterators: flatten_iteratorsize, flatten_length
+flatten_iteratorsize(::Union{Base.HasShape, Base.HasLength}, ::Type{<:Union{SArray,MArray}}) = Base.HasLength()
+function flatten_length(f, T::Type{<:Union{SArray,MArray}})
+    length(T)*length(f.it)
+end
