rubydragon/FeynmanDiagrams.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Small changes

Browse Source
This commit is contained in:
AntonReinhard
2024-07-09 18:33:37 +02:00
parent e8bc26b0c0
commit f6084065e5
3 changed files with 26 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/diagrams/diagrams.jl
View File

@@ -105,6 +105,9 @@ function normalize(virtual_particle::VirtualParticle{P,S,IN_T,OUT_T}) where {P,S
end
function _momentum_contribution(proc::AbstractProcessDefinition, dir::ParticleDirection, species::AbstractParticleType, index::Int)
I = number_incoming_particles(proc)
O = number_outgoing_particles(proc)
# get index of n-th "dir species" particle in proc
particles_seen = 0
c = 0
@@ -114,9 +117,11 @@ function _momentum_contribution(proc::AbstractProcessDefinition, dir::ParticleDi
particles_seen += 1
end
if particles_seen == index
return (ntuple(x -> is_incoming(dir) && x == c, number_incoming_particles(proc)), ntuple(x -> is_outgoing(dir) && x == c, number_outgoing_particles(proc)))
return (((is_incoming(dir) && x == c for x in 1:I)...,), ((is_outgoing(dir) && x == c for x in 1:O)...,))
end
end
@assert false "tried to get momentum contribution of $dir $species $index but it does not exist in $proc"
end
function _momentum_contribution(proc::AbstractProcessDefinition, diagram::FeynmanDiagram{N,E,U,T,M,FM}, n::Int) where {N,E,U,T,M,FM}