added utils for flattening of momenta to tests

Author: Uwe Hernandez Acosta

src/phase_spaces.jl

@@ -19,6 +19,19 @@ struct PhasespaceDefinition{CS<:AbstractCoordinateSystem,F<:AbstractFrameOfRefer
     frame::F
 end
 
+"""
+_generate_momenta(
+    proc::AbstractProcessDefinition,
+    model::AbstractModelDefinition,
+    in_phase_space_def::AbstractPhasespaceDefinition,
+    in_phase_space::AbstractVector{T},
+    out_phase_space_def::AbstractPhasespaceDefinition,
+    out_phase_space::AbstractVector{T},
+) where {T<:Real}
+"""
+function _generate_momenta end
+
+
 # abstract type for generic phase spaces
 #
 # Currently, elements can be either four-momenta, or real numbers,

test/utils/utils.jl

@@ -6,3 +6,31 @@ include("test_implementations.jl")
 _any_fail(x...) = true
 _any_fail(::TestProcess, ::TestModel) = false
 _any_fail(::TestProcess, ::TestModel, ::TestPhasespaceDef, ::TestPhasespaceDef) = false
+
+# unrolls all elements of a list of four-momenta into vector of coordinates
+function _unroll_moms(ps_moms::AbstractVector{T}) where {T<:QEDbase.AbstractFourMomentum}
+     collect(Iterators.flatten(ps_moms))
+end
+
+function _unroll_moms(ps_moms::AbstractMatrix{T}) where {T<:QEDbase.AbstractFourMomentum}
+end
+
+
+# collect components of four-momenta from a vector of coordinates
+function __furl_moms(ps_coords::AbstractVector{T}) where {T<:Real}
+    SFourMomentum.(eachcol(reshape(ps_coords,4,:))) 
+end
+
+function _furl_moms(ps_coords::AbstractVector{T}) where {T<:Real}
+    @assert length(ps_coords)%4==0
+    __furl_moms(ps_coords)
+end
+
+function _furl_moms(ps_coords::AbstractMatrix{T}) where {T<:Real}
+    @assert size(ps_coords,1)%4==0
+    res = Matrix{SFourMomentum}(undef,Int(size(ps_coords,1)//4),size(ps_coords,2))
+    for i in 1:size(ps_coords,2)
+        res[:,i] .= __furl_moms(view(ps_coords,:,i))
+    end
+    return res
+end