Add (possibly temporary) fix for tests

QEDjl-project · May 21, 2024 · db547ef · db547ef
1 parent b3a8536
commit db547ef
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 8 deletions.
diff --git a/src/cross_sections.jl b/src/cross_sections.jl
@@ -143,8 +143,8 @@ function unsafe_differential_cross_section(phase_space_point::PhaseSpacePoint)
         phase_space_point.proc,
         phase_space_point.model,
         phase_space_point.ps_def,
-        momentum.(phase_space_point.in_particles),
-        momentum.(phase_space_point.out_particles),
+        momenta(phase_space_point, Incoming()),
+        momenta(phase_space_point, Outgoing()),
     )
 end
 
@@ -280,8 +280,8 @@ function differential_cross_section(phase_space_point::PhaseSpacePoint)
         phase_space_point.proc,
         phase_space_point.model,
         phase_space_point.ps_def,
-        momentum.(phase_space_point.in_particles),
-        momentum.(phase_space_point.out_particles),
+        momenta(phase_space_point, Incoming()),
+        momenta(phase_space_point, Outgoing()),
     )
 end
 

diff --git a/src/phase_spaces.jl b/src/phase_spaces.jl
@@ -321,6 +321,14 @@ struct PhaseSpacePoint{
     end
 end
 
+function momenta(psp::PhaseSpacePoint, ::Incoming)
+    return QEDbase._as_svec(momentum.(psp.in_particles))
+end
+
+function momenta(psp::PhaseSpacePoint, ::Outgoing)
+    return QEDbase._as_svec(momentum.(psp.out_particles))
+end
+
 # recursion termination: success
 @inline _recursive_type_check(t::Tuple{}, p::Tuple{}, dir::ParticleDirection) = nothing
 

diff --git a/src/probabilities.jl b/src/probabilities.jl
@@ -92,8 +92,8 @@ function unsafe_differential_probability(phase_space_point::PhaseSpacePoint)
         phase_space_point.proc,
         phase_space_point.model,
         phase_space_point.ps_def,
-        momentum.(phase_space_point.in_particles),
-        momentum.(phase_space_point.out_particles),
+        momenta(phase_space_point, Incoming()),
+        momenta(phase_space_point, Outgoing()),
     )
 end
 
@@ -230,8 +230,8 @@ function differential_probability(phase_space_point::PhaseSpacePoint)
         phase_space_point.proc,
         phase_space_point.model,
         phase_space_point.ps_def,
-        momentum.(phase_space_point.in_particles),
-        momentum.(phase_space_point.out_particles),
+        momenta(phase_space_point, Incoming()),
+        momenta(phase_space_point, Outgoing()),
     )
 end