Merge pull request #34 from mmikhasenko/feature-vcat

feature: vcat

Project.toml

@@ -1,7 +1,7 @@
 name = "ThreeBodyDecays"
 uuid = "e6563dab-9ca1-5843-bde3-2ccf38d63843"
 authors = ["Misha Mikhasenko <[email protected]>"]
-version = "0.11.2"
+version = "0.11.3"
 
 [deps]
 Parameters = "d96e819e-fc66-5662-9728-84c9c7592b0a"

src/ThreeBodyDecays.jl

@@ -7,7 +7,7 @@ using RecipesBase
 using Polynomials
 using PolynomialRoots
 
-import Base: getindex, iterate, length
+import Base: getindex, iterate, length, vcat
 
 
 export Kallen, sqrtKallenFact, Kibble,

src/model.jl

@@ -81,3 +81,23 @@ Computes squared amplitude summed over spin projections.
 unpolarized_intensity(model, σs; kw...) =
     sum(abs2, amplitude(model, σs, two_λs; kw...)
               for two_λs in itr(spins(model)))
+
+
+"""
+    Base.vcat(models::ThreeBodyDecay...)
+
+Concatenates multiple `ThreeBodyDecay` objects into a single `ThreeBodyDecay`.
+Argument is variable number of `ThreeBodyDecay` objects.
+
+# An example
+```julia
+extended_model = vcat(model[2], model[2:3], model)
+```
+"""
+function vcat(models::ThreeBodyDecay...)
+    names = vcat(getproperty.(models, :names)...)
+    couplings = vcat(getproperty.(models, :couplings)...)
+    chains = vcat(getproperty.(models, :chains)...)
+    ThreeBodyDecay(names .=> zip(couplings, chains))
+end
+

test/test_model.jl

@@ -83,6 +83,14 @@ end
     @test length(_model) == 1
 end
 
+
+@testset "Vertical Concat model" begin
+    _model = vcat(model[2], model[2:3], model)
+    @test _model isa ThreeBodyDecay
+    @test length(_model) == 1 + 2 + 3
+end
+
+
 struct MyDecayChain{T} <: AbstractDecayChain
     x::T
 end