structarrays

Author: cscherrer

Project.toml

@@ -29,6 +29,7 @@ Static = "aedffcd0-7271-4cad-89d0-dc628f76c6d3"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
+StructArrays = "09ab397b-f2b6-538f-b94a-2f83cf4a842a"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Tricks = "410a4b4d-49e4-4fbc-ab6d-cb71b17b3775"
 
@@ -65,6 +66,7 @@ Static = "0.8, 1"
 StaticArrays = "1.5"
 Statistics = "1"
 StatsBase = "0.34"
+StructArrays = "0.7.0"
 Test = "1"
 Tricks = "0.1"
 julia = "1.10"

src/parameterized.jl

@@ -67,6 +67,33 @@ function ConstructionBase.setproperties(
     return constructorof(P)(merge(params(d), nt))
 end
 
+###############################################################################
+# StructArrays
+
+# TODO: Can this be in an extension?
+
+
+import StructArrays
+
+function StructArrays.staticschema(::Type{P}) where {N,P<:ParameterizedMeasure{N}}
+    # Get the types from the NamedTuple parameter
+    par_type = fieldtype(P, :par)
+    # Use proper type construction for the component types
+    types = Tuple{(fieldtype(par_type, n) for n in N)...}
+    # Return NamedTuple type with parameter names
+    return NamedTuple{N,types}
+end
+
+function StructArrays.component(m::ParameterizedMeasure, key::Symbol)
+    # Extract component from the parameters NamedTuple
+    return getfield(getfield(m, :par), key)
+end
+
+function StructArrays.createinstance(::Type{P}, args...) where {N,P<:ParameterizedMeasure{N}}
+    # Create a new ParameterizedMeasure instance from components
+    return constructorof(P)(NamedTuple{N}(args))
+end
+
 ###############################################################################
 # params