Skip to content

Commit 5315bdf

Browse files
szabo137szabo137
authored
added generate_phase_space and respective tests (#54)
1 parent 10b50e6 commit 5315bdf

File tree

3 files changed

+106
-41
lines changed

3 files changed

+106
-41
lines changed

‎src/QEDprocesses.jl

+1
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ export AbstractFrameOfReference, CenterOfMomentumFrame, ElectronRestFrame
3232
export AbstractPhasespaceDefinition, PhasespaceDefinition
3333
export ParticleStateful, PhaseSpacePoint
3434
export spin, polarization, momentum, getindex
35+
export generate_phase_space
3536

3637
using QEDbase
3738

‎src/phase_spaces.jl

+40
Original file line numberDiff line numberDiff line change
@@ -259,3 +259,43 @@ Returns the momentum of the `n`th particle in the given [`PhaseSpacePoint`](@ref
259259
function momentum(psp::PhaseSpacePoint, dir::ParticleDirection, n::Int)
260260
return psp[dir, n].mom
261261
end
262+
263+
"""
264+
265+
generate_phase_space(
266+
proc::AbstractProcessDefinition,
267+
model::AbstractModelDefinition,
268+
ps_def::AbstractPhasespaceDefinition,
269+
in_ps::AbstractVector{ElType},
270+
out_ps::AbstractVector{ElType},
271+
) where {ElType<:QEDbase.AbstractFourMomentum}
272+
273+
Return the respective phase space point for given momenta of incoming and outgoing particles regarding a given process.
274+
"""
275+
function generate_phase_space(
276+
proc::AbstractProcessDefinition,
277+
model::AbstractModelDefinition,
278+
ps_def::AbstractPhasespaceDefinition,
279+
in_ps::AbstractVector{ElType},
280+
out_ps::AbstractVector{ElType},
281+
) where {ElType<:QEDbase.AbstractFourMomentum}
282+
in_particles = incoming_particles(proc)
283+
in_n = number_incoming_particles(proc)
284+
in_parts = SVector{in_n,ParticleStateful{SFourMomentum}}(
285+
collect(
286+
ParticleStateful(Incoming(), particle, mom) for
287+
(particle, mom) in zip(in_particles, in_ps)
288+
),
289+
)
290+
291+
out_particles = outgoing_particles(proc)
292+
out_n = number_outgoing_particles(proc)
293+
out_parts = SVector{out_n,ParticleStateful{SFourMomentum}}(
294+
collect(
295+
ParticleStateful(Outgoing(), particle, mom) for
296+
(particle, mom) in zip(out_particles, out_ps)
297+
),
298+
)
299+
300+
return PhaseSpacePoint(proc, model, ps_def, in_parts, out_parts)
301+
end

‎test/phase_spaces.jl

+65-41
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,15 @@ RNG = Random.MersenneTwister(727)
5454
end
5555

5656
@testset "Phasespace Point" begin
57-
in_el = ParticleStateful(Incoming(), Electron(), rand(RNG, SFourMomentum))
58-
in_ph = ParticleStateful(Incoming(), Photon(), rand(RNG, SFourMomentum))
59-
out_el = ParticleStateful(Outgoing(), Electron(), rand(RNG, SFourMomentum))
60-
out_ph = ParticleStateful(Outgoing(), Photon(), rand(RNG, SFourMomentum))
57+
in_el_mom = rand(RNG, SFourMomentum)
58+
in_ph_mom = rand(RNG, SFourMomentum)
59+
out_el_mom = rand(RNG, SFourMomentum)
60+
out_ph_mom = rand(RNG, SFourMomentum)
61+
62+
in_el = ParticleStateful(Incoming(), Electron(), in_el_mom)
63+
in_ph = ParticleStateful(Incoming(), Photon(), in_ph_mom)
64+
out_el = ParticleStateful(Outgoing(), Electron(), out_el_mom)
65+
out_ph = ParticleStateful(Outgoing(), Photon(), out_ph_mom)
6166

6267
in_particles_valid = SVector(in_el, in_ph)
6368
in_particles_invalid = SVector(in_el, out_ph)
@@ -76,58 +81,77 @@ end
7681
process, model, phasespace_def, in_particles_valid, out_particles_valid
7782
)
7883

79-
@test momentum(psp, Incoming(), 1) == in_el.mom
80-
@test momentum(psp, Incoming(), 2) == in_ph.mom
81-
@test momentum(psp, Outgoing(), 1) == out_el.mom
82-
@test momentum(psp, Outgoing(), 2) == out_ph.mom
84+
@testset "Accessor" begin
85+
@test momentum(psp, Incoming(), 1) == in_el.mom
86+
@test momentum(psp, Incoming(), 2) == in_ph.mom
87+
@test momentum(psp, Outgoing(), 1) == out_el.mom
88+
@test momentum(psp, Outgoing(), 2) == out_ph.mom
89+
90+
@test psp[Incoming(), 1] == in_el
91+
@test psp[Incoming(), 2] == in_ph
92+
@test psp[Outgoing(), 1] == out_el
93+
@test psp[Outgoing(), 2] == out_ph
94+
end
95+
96+
@testset "Error handling" begin
97+
if (VERSION >= v"1.8")
98+
# julia versions before 1.8 did not have support for regex matching in @test_throws
99+
@test_throws r"stateful particle (.*) is given as an incoming particle but is outgoing" PhaseSpacePoint(
100+
process, model, phasespace_def, in_particles_invalid, out_particles_valid
101+
)
102+
103+
@test_throws r"stateful particle (.*) is given as an outgoing particle but is incoming" PhaseSpacePoint(
104+
process, model, phasespace_def, in_particles_valid, out_particles_invalid
105+
)
106+
107+
@test_throws r"process given particle species \((.*)Electron\(\)\) does not match stateful particle species \((.*)Photon\(\)\)" PhaseSpacePoint(
108+
process, model, phasespace_def, SVector(in_ph, in_el), out_particles_valid
109+
)
110+
111+
@test_throws r"process given particle species \((.*)Electron\(\)\) does not match stateful particle species \((.*)Photon\(\)\)" PhaseSpacePoint(
112+
process, model, phasespace_def, in_particles_valid, SVector(out_ph, out_el)
113+
)
114+
end
115+
116+
@test_throws BoundsError momentum(psp, Incoming(), -1)
117+
@test_throws BoundsError momentum(psp, Outgoing(), -1)
118+
@test_throws BoundsError momentum(psp, Incoming(), 4)
119+
@test_throws BoundsError momentum(psp, Outgoing(), 4)
83120

84-
@test psp[Incoming(), 1] == in_el
85-
@test psp[Incoming(), 2] == in_ph
86-
@test psp[Outgoing(), 1] == out_el
87-
@test psp[Outgoing(), 2] == out_ph
121+
@test_throws BoundsError psp[Incoming(), -1]
122+
@test_throws BoundsError psp[Outgoing(), -1]
123+
@test_throws BoundsError psp[Incoming(), 4]
124+
@test_throws BoundsError psp[Outgoing(), 4]
88125

89-
if (VERSION >= v"1.8")
90-
# julia versions before 1.8 did not have support for regex matching in @test_throws
91-
@test_throws r"stateful particle (.*) is given as an incoming particle but is outgoing" PhaseSpacePoint(
126+
@test_throws InvalidInputError PhaseSpacePoint(
92127
process, model, phasespace_def, in_particles_invalid, out_particles_valid
93128
)
94129

95-
@test_throws r"stateful particle (.*) is given as an outgoing particle but is incoming" PhaseSpacePoint(
130+
@test_throws InvalidInputError PhaseSpacePoint(
96131
process, model, phasespace_def, in_particles_valid, out_particles_invalid
97132
)
98133

99-
@test_throws r"process given particle species \((.*)Electron\(\)\) does not match stateful particle species \((.*)Photon\(\)\)" PhaseSpacePoint(
134+
@test_throws InvalidInputError PhaseSpacePoint(
100135
process, model, phasespace_def, SVector(in_ph, in_el), out_particles_valid
101136
)
102137

103-
@test_throws r"process given particle species \((.*)Electron\(\)\) does not match stateful particle species \((.*)Photon\(\)\)" PhaseSpacePoint(
138+
@test_throws InvalidInputError PhaseSpacePoint(
104139
process, model, phasespace_def, in_particles_valid, SVector(out_ph, out_el)
105140
)
106141
end
107142

108-
@test_throws BoundsError momentum(psp, Incoming(), -1)
109-
@test_throws BoundsError momentum(psp, Outgoing(), -1)
110-
@test_throws BoundsError momentum(psp, Incoming(), 4)
111-
@test_throws BoundsError momentum(psp, Outgoing(), 4)
112-
113-
@test_throws BoundsError psp[Incoming(), -1]
114-
@test_throws BoundsError psp[Outgoing(), -1]
115-
@test_throws BoundsError psp[Incoming(), 4]
116-
@test_throws BoundsError psp[Outgoing(), 4]
117-
118-
@test_throws InvalidInputError PhaseSpacePoint(
119-
process, model, phasespace_def, in_particles_invalid, out_particles_valid
120-
)
121-
122-
@test_throws InvalidInputError PhaseSpacePoint(
123-
process, model, phasespace_def, in_particles_valid, out_particles_invalid
124-
)
143+
@testset "Generation" begin
144+
test_psp = generate_phase_space(
145+
process, model, phasespace_def, [in_el_mom, in_ph_mom], [out_el_mom, out_ph_mom]
146+
)
125147

126-
@test_throws InvalidInputError PhaseSpacePoint(
127-
process, model, phasespace_def, SVector(in_ph, in_el), out_particles_valid
128-
)
148+
@test test_psp.proc == process
149+
@test test_psp.model == model
150+
@test test_psp.ps_def == phasespace_def
129151

130-
@test_throws InvalidInputError PhaseSpacePoint(
131-
process, model, phasespace_def, in_particles_valid, SVector(out_ph, out_el)
132-
)
152+
@test test_psp[Incoming(), 1] == in_el
153+
@test test_psp[Incoming(), 2] == in_ph
154+
@test test_psp[Outgoing(), 1] == out_el
155+
@test test_psp[Outgoing(), 2] == out_ph
156+
end
133157
end

0 commit comments

Comments
 (0)