Add QEDprocess functionality

docs/Project.toml

@@ -1,3 +1,5 @@
 [deps]
 Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
+QEDbase = "10e22c08-3ccb-4172-bfcf-7d7aa3d04d93"
 QEDcore = "35dc0263-cb5f-4c33-a114-1d7f54ab753e"
+QEDprocesses = "46de9c38-1bb3-4547-a1ec-da24d767fdad"

docs/make.jl

@@ -1,7 +1,16 @@
+using Pkg
+
+# targeting the correct source code
+# this asumes the make.jl script is located in QEDcore.jl/docs
+project_path = Base.Filesystem.joinpath(Base.Filesystem.dirname(Base.source_path()), "..")
+Pkg.develop(; path=project_path)
+
+Pkg.add(; url="https://github.com/QEDjl-project/QEDbase.jl/", rev="dev")
+
 using QEDcore
 using Documenter
 
-DocMeta.setdocmeta!(QEDcore, :DocTestSetup, :(using QEDcore); recursive=true)
+# DocMeta.setdocmeta!(QEDcore, :DocTestSetup, :(using QEDcore); recursive=true)
 
 makedocs(;
     modules=[QEDcore],

src/QEDcore.jl

@@ -1,9 +1,5 @@
 module QEDcore
 
-import Base: *
-import StaticArrays: similar_type
-import QEDbase: base_state
-
 # lorentz vectors
 export SLorentzVector, MLorentzVector
 
@@ -32,6 +28,13 @@ export @valid_spinor_input
 # particle base states
 export base_state
 
+# phase space
+export SphericalCoordinateSystem
+export CenterOfMomentumFrame, ElectronRestFrame
+export PhasespaceDefinition
+export ParticleStateful, PhaseSpacePoint, InPhaseSpacePoint, OutPhaseSpacePoint
+export spin, polarization, particle_direction, particle_species, momentum, momenta, getindex
+
 using QEDbase: QEDbase
 using DocStringExtensions
 using StaticArrays
@@ -40,12 +43,19 @@ using SimpleTraits
 include("algebraic_objects/dirac_tensors/types.jl")
 include("algebraic_objects/dirac_tensors/multiplication.jl")
 
+include("phase_spaces/types.jl")
+include("phase_spaces/access.jl")
+include("phase_spaces/create.jl")
+include("phase_spaces/print.jl")
+include("phase_spaces/utility.jl")
+
 include("algebraic_objects/four_momentum.jl")
 include("algebraic_objects/lorentz_vector.jl")
 include("algebraic_objects/gamma_matrices.jl")
 
 include("particles/particle_types.jl")
-include("particles/spinors.jl")
+include("particles/propagators.jl")
 include("particles/states.jl")
+include("particles/spinors.jl")
 
 end

src/algebraic_objects/dirac_tensors/multiplication.jl

@@ -3,6 +3,9 @@
 # Concrete implementation of multiplication for Dirac Tensors
 #
 #######
+
+import Base: *
+
 """
 $(TYPEDSIGNATURES)
 

src/algebraic_objects/four_momentum.jl

@@ -5,6 +5,7 @@
 #######
 
 import QEDbase: getT, getX, getY, getZ, setT!, setX!, setY!, setZ!
+import StaticArrays: similar_type
 
 """
 $(TYPEDEF)

src/algebraic_objects/lorentz_vector.jl

@@ -5,6 +5,7 @@
 #######
 
 import QEDbase: getT, getX, getY, getZ, setT!, setX!, setY!, setZ!
+import StaticArrays: similar_type
 
 #######
 #

src/particles/particle_types.jl

@@ -81,7 +81,7 @@ is_anti_particle(::MajoranaFermion) = true
 Concrete type for *electrons* as a particle species. Mostly used for dispatch. 
 
 ```jldoctest
-julia> using QEDbase
+julia> using QEDcore
 
 julia> Electron()
 electron
@@ -104,7 +104,7 @@ Base.show(io::IO, ::Electron) = print(io, "electron")
 Concrete type for *positrons* as a particle species. Mostly used for dispatch. 
 
 ```jldoctest
-julia> using QEDbase
+julia> using QEDcore
 
 julia> Positron()
 positron
@@ -182,7 +182,7 @@ is_anti_particle(::MajoranaBoson) = true
 Concrete type for the *photons* as a particle species. Mostly used for dispatch. 
 
 ```jldoctest
-julia> using QEDbase
+julia> using QEDcore
 
 julia> Photon()
 photon

src/particles/propagators.jl

@@ -0,0 +1,29 @@
+import QEDbase: propagator
+
+function _scalar_propagator(K::QEDbase.AbstractFourMomentum, mass::Real)
+    return one(mass) / (K * K - mass^2)
+end
+
+function _scalar_propagator(K::QEDbase.AbstractFourMomentum)
+    return one(getT(K)) / (K * K)
+end
+
+function _fermion_propagator(P::QEDbase.AbstractFourMomentum, mass::Real)
+    return (slashed(P) + mass * one(DiracMatrix)) * _scalar_propagator(P, mass)
+end
+
+function _fermion_propagator(P::QEDbase.AbstractFourMomentum)
+    return (slashed(P)) * _scalar_propagator(P)
+end
+
+function QEDbase.propagator(particle_type::BosonLike, K::QEDbase.AbstractFourMomentum)
+    return _scalar_propagator(K, QEDbase.mass(particle_type))
+end
+
+function QEDbase.propagator(particle_type::Photon, K::QEDbase.AbstractFourMomentum)
+    return _scalar_propagator(K)
+end
+
+function QEDbase.propagator(particle_type::FermionLike, P::QEDbase.AbstractFourMomentum)
+    return _fermion_propagator(P, QEDbase.mass(particle_type))
+end

src/particles/states.jl

@@ -1,3 +1,5 @@
+import QEDbase: base_state
+
 function _booster_fermion(mom::QEDbase.AbstractFourMomentum, mass::Real)
     return (slashed(mom) + mass * one(DiracMatrix)) / (sqrt(abs(QEDbase.getT(mom)) + mass))
 end

src/phase_spaces/access.jl

@@ -0,0 +1,30 @@
+# accessor interface particle stateful
+QEDbase.particle_direction(part::ParticleStateful) = part.dir
+QEDbase.particle_species(part::ParticleStateful) = part.species
+QEDbase.momentum(part::ParticleStateful) = part.mom
+
+# accessor interface phase space point
+"""
+    Base.getindex(psp::PhaseSpacePoint, dir::Incoming, n::Int)
+
+Overload for the array indexing operator `[]`. Returns the nth incoming particle in this phase space point.
+"""
+function Base.getindex(psp::PhaseSpacePoint, ::QEDbase.Incoming, n::Int)
+    return psp.in_particles[n]
+end
+
+"""
+    Base.getindex(psp::PhaseSpacePoint, dir::Outgoing, n::Int)
+
+Overload for the array indexing operator `[]`. Returns the nth outgoing particle in this phase space point.
+"""
+function Base.getindex(psp::PhaseSpacePoint, ::QEDbase.Outgoing, n::Int)
+    return psp.out_particles[n]
+end
+
+QEDbase.process(psp::PhaseSpacePoint) = psp.proc
+QEDbase.model(psp::PhaseSpacePoint) = psp.model
+QEDbase.phase_space_definition(psp::PhaseSpacePoint) = psp.ps_def
+
+QEDbase.particles(psp::PhaseSpacePoint, ::QEDbase.Incoming) = psp.in_particles
+QEDbase.particles(psp::PhaseSpacePoint, ::QEDbase.Outgoing) = psp.out_particles

src/phase_spaces/create.jl

@@ -0,0 +1,158 @@
+# PSP constructors from particle statefuls
+
+"""
+    InPhaseSpacePoint(
+        proc::QEDbase.AbstractProcessDefinition, 
+        model::QEDbase.AbstractModelDefinition, 
+        ps_def::QEDbase.AbstractPhasespaceDefinition, 
+        in_ps::Tuple{ParticleStateful},
+    )
+
+    Construct a [`PhaseSpacePoint`](@ref) with only input particles from [`ParticleStateful`](@ref)s. The result will be `<: InPhaseSpacePoint` but **not** `<: OutPhaseSpacePoint`.
+"""
+function InPhaseSpacePoint(
+    proc::PROC, model::MODEL, ps_def::PSDEF, in_ps::IN_PARTICLES
+) where {
+    PROC<:QEDbase.AbstractProcessDefinition,
+    MODEL<:QEDbase.AbstractModelDefinition,
+    PSDEF<:QEDbase.AbstractPhasespaceDefinition,
+    IN_PARTICLES<:Tuple{Vararg{ParticleStateful}},
+}
+    return PhaseSpacePoint(proc, model, ps_def, in_ps, ())
+end
+
+"""
+    OutPhaseSpacePoint(
+        proc::QEDbase.AbstractProcessDefinition, 
+        model::QEDbase.AbstractModelDefinition, 
+        ps_def::QEDbase.AbstractPhasespaceDefinition, 
+        out_ps::Tuple{ParticleStateful},
+    )
+
+Construct a [`PhaseSpacePoint`](@ref) with only output particles from [`ParticleStateful`](@ref)s. The result will be `<: OutPhaseSpacePoint` but **not** `<: InPhaseSpacePoint`.
+"""
+function OutPhaseSpacePoint(
+    proc::PROC, model::MODEL, ps_def::PSDEF, out_ps::OUT_PARTICLES
+) where {
+    PROC<:QEDbase.AbstractProcessDefinition,
+    MODEL<:QEDbase.AbstractModelDefinition,
+    PSDEF<:QEDbase.AbstractPhasespaceDefinition,
+    OUT_PARTICLES<:Tuple{Vararg{ParticleStateful}},
+}
+    return PhaseSpacePoint(proc, model, ps_def, (), out_ps)
+end
+
+# PSP constructors from momenta
+
+"""
+    PhaseSpacePoint(
+        proc::QEDbase.AbstractProcessDefinition,
+        model::QEDbase.AbstractModelDefinition,
+        ps_def::QEDbase.AbstractPhasespaceDefinition,
+        in_momenta::NTuple{N,QEDbase.AbstractFourMomentum},
+        out_momenta::NTuple{M,QEDbase.AbstractFourMomentum},
+    )
+
+Construct the phase space point from given momenta of incoming and outgoing particles regarding a given process.
+"""
+function PhaseSpacePoint(
+    proc::QEDbase.AbstractProcessDefinition,
+    model::QEDbase.AbstractModelDefinition,
+    ps_def::QEDbase.AbstractPhasespaceDefinition,
+    in_momenta::NTuple{N,ELEMENT},
+    out_momenta::NTuple{M,ELEMENT},
+) where {N,M,ELEMENT<:QEDbase.AbstractFourMomentum}
+    in_particles = _build_particle_statefuls(proc, in_momenta, QEDbase.Incoming())
+    out_particles = _build_particle_statefuls(proc, out_momenta, QEDbase.Outgoing())
+
+    return PhaseSpacePoint(proc, model, ps_def, in_particles, out_particles)
+end
+
+"""
+    InPhaseSpacePoint(
+        proc::QEDbase.AbstractProcessDefinition,
+        model::QEDbase.AbstractModelDefinition,
+        ps_def::QEDbase.AbstractPhasespaceDefinition,
+        in_momenta::NTuple{N,QEDbase.AbstractFourMomentum},
+    )
+
+Construct a [`PhaseSpacePoint`](@ref) with only input particles from given momenta. The result will be `<: InPhaseSpacePoint` but **not** `<: OutPhaseSpacePoint`.
+"""
+function InPhaseSpacePoint(
+    proc::QEDbase.AbstractProcessDefinition,
+    model::QEDbase.AbstractModelDefinition,
+    ps_def::QEDbase.AbstractPhasespaceDefinition,
+    in_momenta::NTuple{N,ELEMENT},
+) where {N,ELEMENT<:QEDbase.AbstractFourMomentum}
+    in_particles = _build_particle_statefuls(proc, in_momenta, QEDbase.Incoming())
+
+    return PhaseSpacePoint(proc, model, ps_def, in_particles, ())
+end
+
+"""
+    OutPhaseSpacePoint(
+        proc::QEDbase.AbstractProcessDefinition,
+        model::QEDbase.AbstractModelDefinition,
+        ps_def::QEDbase.AbstractPhasespaceDefinition,
+        out_momenta::NTuple{N,QEDbase.AbstractFourMomentum},
+    )
+
+Construct a [`PhaseSpacePoint`](@ref) with only output particles from given momenta. The result will be `<: OutPhaseSpacePoint` but **not** `<: InPhaseSpacePoint`.
+"""
+function OutPhaseSpacePoint(
+    proc::QEDbase.AbstractProcessDefinition,
+    model::QEDbase.AbstractModelDefinition,
+    ps_def::QEDbase.AbstractPhasespaceDefinition,
+    out_momenta::NTuple{N,ELEMENT},
+) where {N,ELEMENT<:QEDbase.AbstractFourMomentum}
+    out_particles = _build_particle_statefuls(proc, out_momenta, QEDbase.Outgoing())
+
+    return PhaseSpacePoint(proc, model, ps_def, (), out_particles)
+end
+
+# PSP constructors from coordinates
+
+"""
+    PhaseSpacePoint(
+        proc::QEDbase.AbstractProcessDefinition,
+        model::QEDbase.AbstractModelDefinition,
+        ps_def::QEDbase.AbstractPhasespaceDefinition,
+        in_coords::NTuple{N,Real},
+        out_coords::NTuple{M,Real},
+    )
+
+Construct a [`PhaseSpacePoint`](@ref) from given coordinates by using the `QEDbase._generate_momenta` interface.
+"""
+function PhaseSpacePoint(
+    proc::QEDbase.AbstractProcessDefinition,
+    model::QEDbase.AbstractModelDefinition,
+    ps_def::QEDbase.AbstractPhasespaceDefinition,
+    in_coords::NTuple{N,Real},
+    out_coords::NTuple{M,Real},
+) where {N,M}
+    in_ps, out_ps = _generate_momenta(proc, model, ps_def, in_coords, out_coords)
+    return PhaseSpacePoint(proc, model, ps_def, in_ps, out_ps)
+end
+
+"""
+    InPhaseSpacePoint(
+        proc::QEDbase.AbstractProcessDefinition,
+        model::QEDbase.AbstractModelDefinition,
+        ps_def::QEDbase.AbstractPhasespaceDefinition,
+        in_coords::NTuple{N,Real},
+    )
+
+Construct a [`PhaseSpacePoint`](@ref) from given coordinates by using the `QEDbase._generate_momenta` interface. The result will be `<: InPhaseSpacePoint` but **not** `<: OutPhaseSpacePoint`.
+
+!!! note
+    A similar function for [`OutPhaseSpacePoint`](@ref) does not exist from coordinates, only a full [`PhaseSpacePoint`](@ref).
+"""
+function InPhaseSpacePoint(
+    proc::QEDbase.AbstractProcessDefinition,
+    model::QEDbase.AbstractModelDefinition,
+    ps_def::QEDbase.AbstractPhasespaceDefinition,
+    in_coords::NTuple{N,Real},
+) where {N}
+    in_ps = _generate_incoming_momenta(proc, model, ps_def, in_coords)
+    return InPhaseSpacePoint(proc, model, ps_def, in_ps)
+end