Add support to SA[1;;1]

Author: N5N3

src/initializers.jl

@@ -59,3 +59,27 @@ end
 @inline Base.typed_hvcat(sa::Type{SA}, rows::Dims, xs::Number...)            = _SA_typed_hvcat(sa, rows, xs)
 @inline Base.typed_hvcat(sa::Type{SA{T}}, rows::Dims, xs::Number...) where T = _SA_typed_hvcat(sa, rows, xs)
 
+if VERSION >= v"1.7"
+@generated function reorder(x::NTuple{M,Any}, ::Val{N1}, ::Val{N2}) where {M,N1,N2}
+    a = Base.PermutedDimsArray(reshape(1:M, N1, N2, :), (2, 1, 3))
+    args = (:(x[$i]) for i in a)
+    return :(tuple($(args...)))
+end
+    
+@inline function _SA_typed_hvncat(sa, dimsshape, row_first, xs)
+    msize = Size(dimsshape)
+    xs′ = row_first ? reorder(xs, Val(msize[1]), Val(msize[2])) : xs
+    x = similar_type(sa, msize)(xs′)
+end
+
+@inline Base.typed_hvncat(sa::Type{SA}, dimsshape::Dims, row_first::Bool, xs::Number...) = _SA_typed_hvncat(sa, dimsshape, row_first, xs)
+@inline Base.typed_hvncat(sa::Type{SA{T}}, dimsshape::Dims, row_first::Bool, xs::Number...) where T = _SA_typed_hvncat(sa, dimsshape, row_first, xs)
+
+@inline function _SA_typed_hvncat(sa, ::Val{dim}, xs) where {dim}
+    msize = Size(ntuple(_->1, Val(dim))..., length(xs))
+    x = similar_type(sa, msize)(xs)
+end
+
+@inline Base.typed_hvncat(sa::Type{SA}, dim::Int, xs::Number...) = _SA_typed_hvncat(sa, Val(dim-1), xs)
+@inline Base.typed_hvncat(sa::Type{SA{T}}, dim::Int, xs::Number...) where T = _SA_typed_hvncat(sa, Val(dim-1), xs)
+end

test/initializers.jl

@@ -42,6 +42,17 @@ SA_test_hvcat(x,T) = SA{T}[1 x x;
 @test_inlined SA[1;2]
 @test_inlined SA_test_hvcat(3)
 
+SA_test_hvncat1(x) = SA[1 x;x 2;;;x 2;1 x]
+SA_test_hvncat2(x) = SA[1;x;;x;2;;;x;2;;1;x]
+if VERSION >= v"1.7.0"
+    @test SA[1;;;2] === SArray{Tuple{1,1,2}}(1,2)
+    @test SA[1;2;;1;2] === SMatrix{2,2}(1,2,1,2)
+    @test SA[1 2;1 2 ;;; 1 2;1 2] === SArray{Tuple{2,2,2}}(Tuple([1 2;1 2 ;;; 1 2;1 2]))
+    @test_inlined SA_test_hvncat1(3)
+    @test_inlined SA_test_hvncat2(2)
+    @test_throws ArgumentError SA[1;2;;3]
+end
+
 # https://github.com/JuliaArrays/StaticArrays.jl/pull/685
 @test Union{}[] isa Vector{Union{}}
 @test Base.typed_vcat(Union{}) isa Vector{Union{}}