named properties for staticarrays

Author: aplavin

ext/StructArraysStaticArraysCoreExt.jl

@@ -5,7 +5,7 @@ using StructArrays: StructArrayStyle, createinstance, replace_structarray, isnon
 
 using Base.Broadcast: Broadcasted
 
-using StaticArraysCore: StaticArray, FieldArray, tuple_prod
+using StaticArraysCore: StaticArray, FieldArray, tuple_prod, SVector, MVector
 
 """
     StructArrays.staticschema(::Type{<:StaticArray{S, T}}) where {S, T}
@@ -74,4 +74,13 @@ end
     end
 end
 
+@inline function StructArrays.component(s::StructArray{<:Union{SVector,MVector}}, key::Symbol)
+    i = key == :x ? 1 : 
+        key == :y ? 2 : 
+        key == :z ? 3 : 
+        key == :w ? 4 : 
+        throw(ArgumentError("invalid key $key"))
+    StructArrays.component(s, i)
+end
+
 end # module

test/runtests.jl

@@ -1381,6 +1381,12 @@ end
         @test StructArrays.components(x) == ([1.0,2.0], [2.0,3.0])
         @test x .+ y == StructArray([StaticVectorType{2}(Float64[2*i+1;2*i+3]) for i = 1:2])
     end
+    for StaticVectorType = [SVector, MVector]
+        x = @inferred StructArray([StaticVectorType{2}(Float64[i;i+1]) for i = 1:2])
+        # numbered and named property access:
+        @test x.:1 == [1.0,2.0]
+        @test x.y == [2.0,3.0]
+    end
     # test broadcast + components for general arrays
     for StaticArrayType = [SArray, MArray, SizedArray]
         x = @inferred StructArray([StaticArrayType{Tuple{1,2}}(ones(1,2) .+ i) for i = 0:1])