merge on ComponentVectors

Author: Qfl3x

src/ComponentArrays.jl

@@ -2,6 +2,7 @@ module ComponentArrays
 
 import ChainRulesCore
 import StaticArrayInterface, ArrayInterface, Functors
+import Base.merge
 
 using LinearAlgebra
 
@@ -22,7 +23,7 @@ include("axis.jl")
 export AbstractAxis, Axis, PartitionedAxis, ShapedAxis, ViewAxis, FlatAxis
 
 include("componentarray.jl")
-export ComponentArray, ComponentVector, ComponentMatrix, getaxes, getdata, valkeys
+export ComponentArray, ComponentVector, ComponentMatrix, getaxes, getdata, valkeys, merge
 
 include("componentindex.jl")
 export KeepIndex

src/componentarray.jl

@@ -336,3 +336,15 @@ julia> sum(prod(ca[k]) for k in valkeys(ca))
     return :($k)
 end
 valkeys(ca::ComponentVector) = valkeys(getaxes(ca)[1])
+
+function merge(cvec1::ComponentVector{T1}, cvec2::ComponentVector{T2}) where {T1, T2}
+    typed_dict = ComponentVector{promote_type(T1, T2)}(cvec1)
+    for key in valkeys(cvec2)
+        keyname = getval(key)
+        val = cvec2[key]
+        typed_dict = eval(:( ComponentArray($typed_dict, $keyname = $val) ))
+    end
+    typed_dict
+end
+
+merge(a::ComponentVector, b::ComponentVector, cs::ComponentVector) = merge(merge(a,b), cs...)