Skip to content

Commit

Permalink
added base_state and fixed tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Uwe Hernandez Acosta committed Jun 19, 2024
1 parent a40938b commit da639a5
Show file tree
Hide file tree
Showing 4 changed files with 167 additions and 22 deletions.
5 changes: 5 additions & 0 deletions src/QEDcore.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ module QEDcore

import Base: *
import StaticArrays: similar_type
import QEDbase: base_state

# lorentz vectors
export SLorentzVector, MLorentzVector
Expand All @@ -28,6 +29,9 @@ export IncomingFermionSpinor,
export SpinorU, SpinorUbar, SpinorV, SpinorVbar
export @valid_spinor_input

# particle base states
export base_state

using QEDbase: QEDbase
using DocStringExtensions
using StaticArrays
Expand All @@ -42,5 +46,6 @@ include("algebraic_objects/gamma_matrices.jl")

include("particles/particle_types.jl")
include("particles/spinors.jl")
include("particles/states.jl")

end
136 changes: 136 additions & 0 deletions src/particles/states.jl
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,136 @@
function _booster_fermion(mom::QEDbase.AbstractFourMomentum, mass::Real)
return (slashed(mom) + mass * one(DiracMatrix)) / (sqrt(abs(QEDbase.getT(mom)) + mass))
end

function _booster_antifermion(mom::QEDbase.AbstractFourMomentum, mass::Real)
return (mass * one(DiracMatrix) - slashed(mom)) / (sqrt(abs(QEDbase.getT(mom)) + mass))
end

function base_state(
particle::Fermion,
::QEDbase.Incoming,
mom::QEDbase.AbstractFourMomentum,
spin::QEDbase.AbstractDefiniteSpin,
)
booster = _booster_fermion(mom, QEDbase.mass(particle))
return BiSpinor(booster[:, QEDbase._spin_index(spin)])
end

function base_state(
particle::Fermion,
::QEDbase.Incoming,
mom::QEDbase.AbstractFourMomentum,
spin::QEDbase.AllSpin,
)
booster = _booster_fermion(mom, QEDbase.mass(particle))
return SVector(BiSpinor(booster[:, 1]), BiSpinor(booster[:, 2]))
end

function base_state(
particle::AntiFermion,
::QEDbase.Incoming,
mom::QEDbase.AbstractFourMomentum,
spin::QEDbase.AbstractDefiniteSpin,
)
booster = _booster_antifermion(mom, QEDbase.mass(particle))
return AdjointBiSpinor(BiSpinor(booster[:, QEDbase._spin_index(spin) + 2])) * GAMMA[1]
end

function base_state(
particle::AntiFermion,
::QEDbase.Incoming,
mom::QEDbase.AbstractFourMomentum,
spin::QEDbase.AllSpin,
)
booster = _booster_antifermion(mom, QEDbase.mass(particle))
return SVector(
AdjointBiSpinor(BiSpinor(booster[:, 3])) * GAMMA[1],
AdjointBiSpinor(BiSpinor(booster[:, 4])) * GAMMA[1],
)
end

function base_state(
particle::Fermion,
::QEDbase.Outgoing,
mom::QEDbase.AbstractFourMomentum,
spin::QEDbase.AbstractDefiniteSpin,
)
booster = _booster_fermion(mom, QEDbase.mass(particle))
return AdjointBiSpinor(BiSpinor(booster[:, QEDbase._spin_index(spin)])) * GAMMA[1]
end

function base_state(
particle::Fermion,
::QEDbase.Outgoing,
mom::QEDbase.AbstractFourMomentum,
spin::QEDbase.AllSpin,
)
booster = _booster_fermion(mom, QEDbase.mass(particle))
return SVector(
AdjointBiSpinor(BiSpinor(booster[:, 1])) * GAMMA[1],
AdjointBiSpinor(BiSpinor(booster[:, 2])) * GAMMA[1],
)
end

function base_state(
particle::AntiFermion,
::QEDbase.Outgoing,
mom::QEDbase.AbstractFourMomentum,
spin::QEDbase.AbstractDefiniteSpin,
)
booster = _booster_antifermion(mom, QEDbase.mass(particle))
return BiSpinor(booster[:, QEDbase._spin_index(spin) + 2])
end

function base_state(
particle::AntiFermion,
::QEDbase.Outgoing,
mom::QEDbase.AbstractFourMomentum,
spin::QEDbase.AllSpin,
)
booster = _booster_antifermion(mom, QEDbase.mass(particle))
return SVector(BiSpinor(booster[:, 3]), BiSpinor(booster[:, 4]))
end

function _photon_state(pol::QEDbase.AllPolarization, mom::QEDbase.AbstractFourMomentum)
cth = QEDbase.getCosTheta(mom)
sth = sqrt(1 - cth^2)
cos_phi = QEDbase.getCosPhi(mom)
sin_phi = QEDbase.getSinPhi(mom)
return SVector(
SLorentzVector{Float64}(0.0, cth * cos_phi, cth * sin_phi, -sth),
SLorentzVector{Float64}(0.0, -sin_phi, cos_phi, 0.0),
)
end

function _photon_state(pol::QEDbase.PolarizationX, mom::QEDbase.AbstractFourMomentum)
cth = QEDbase.getCosTheta(mom)
sth = sqrt(1 - cth^2)
cos_phi = QEDbase.getCosPhi(mom)
sin_phi = QEDbase.getSinPhi(mom)
return SLorentzVector{Float64}(0.0, cth * cos_phi, cth * sin_phi, -sth)
end

function _photon_state(pol::QEDbase.PolarizationY, mom::QEDbase.AbstractFourMomentum)
cos_phi = QEDbase.getCosPhi(mom)
sin_phi = QEDbase.getSinPhi(mom)
return SLorentzVector{Float64}(0.0, -sin_phi, cos_phi, 0.0)
end

@inline function base_state(
particle::Photon,
::QEDbase.ParticleDirection,
mom::QEDbase.AbstractFourMomentum,
pol::QEDbase.AllPolarization,
)
return _photon_state(pol, mom)
end

@inline function base_state(
particle::Photon,
::QEDbase.ParticleDirection,
mom::QEDbase.AbstractFourMomentum,
pol::QEDbase.AbstractPolarization,
)
return _photon_state(pol, mom)
end
44 changes: 22 additions & 22 deletions test/particles/base_states.jl → test/particles/states.jl
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
using QEDcore
using QEDbase: QEDbase
using QEDcore
using StaticArrays
using Random

Expand Down Expand Up @@ -52,22 +52,22 @@ test_broadcast(x::QEDbase.AbstractSpinOrPolarization) = x
particle_mass = QEDbase.mass(p())
groundtruth_states = FERMION_STATES_GROUNDTRUTH_FACTORY[(d, p)](mom, particle_mass)
groundtruth_tuple = SVector(groundtruth_states(1), groundtruth_states(2))
@test QEDbase.base_state(p(), d(), mom, QEDbase.AllSpin()) == groundtruth_tuple
@test QEDbase.base_state(p(), d(), mom, QEDbase.SpinUp()) == groundtruth_tuple[1]
@test QEDbase.base_state(p(), d(), mom, QEDbase.SpinDown()) == groundtruth_tuple[2]
@test base_state(p(), d(), mom, QEDbase.AllSpin()) == groundtruth_tuple
@test base_state(p(), d(), mom, QEDbase.SpinUp()) == groundtruth_tuple[1]
@test base_state(p(), d(), mom, QEDbase.SpinDown()) == groundtruth_tuple[2]

@test QEDbase._as_svec(QEDbase.base_state(p(), d(), mom, QEDbase.AllSpin())) isa
@test QEDbase._as_svec(base_state(p(), d(), mom, QEDbase.AllSpin())) isa
SVector
@test QEDbase._as_svec(QEDbase.base_state(p(), d(), mom, QEDbase.SpinUp())) isa
@test QEDbase._as_svec(base_state(p(), d(), mom, QEDbase.SpinUp())) isa
SVector
@test QEDbase._as_svec(QEDbase.base_state(p(), d(), mom, QEDbase.SpinDown())) isa
@test QEDbase._as_svec(base_state(p(), d(), mom, QEDbase.SpinDown())) isa
SVector

@test QEDbase._as_svec(QEDbase.base_state(p(), d(), mom, QEDbase.AllSpin())) ==
@test QEDbase._as_svec(base_state(p(), d(), mom, QEDbase.AllSpin())) ==
groundtruth_tuple
@test QEDbase._as_svec(QEDbase.base_state(p(), d(), mom, QEDbase.SpinUp()))[1] ==
@test QEDbase._as_svec(base_state(p(), d(), mom, QEDbase.SpinUp()))[1] ==
groundtruth_tuple[1]
@test QEDbase._as_svec(QEDbase.base_state(p(), d(), mom, QEDbase.SpinDown()))[1] ==
@test QEDbase._as_svec(base_state(p(), d(), mom, QEDbase.SpinDown()))[1] ==
groundtruth_tuple[2]
end
end
Expand All @@ -87,7 +87,7 @@ end
#@testset "$x $y $z" for (x,y,z) in Iterators.product(X_arr,Y_arr,Z_arr)

mom = SFourMomentum(_cartesian_coordinates(om, om, cth, phi))
both_photon_states = QEDbase.base_state(
both_photon_states = base_state(
Photon(), D(), mom, QEDbase.AllPolarization()
)

Expand All @@ -105,36 +105,36 @@ end
)

# test the single polarization states
@test QEDbase.base_state(Photon(), D(), mom, QEDbase.PolarizationX()) ==
@test base_state(Photon(), D(), mom, QEDbase.PolarizationX()) ==
both_photon_states[1]
@test QEDbase.base_state(Photon(), D(), mom, QEDbase.PolarizationY()) ==
@test base_state(Photon(), D(), mom, QEDbase.PolarizationY()) ==
both_photon_states[2]
@test QEDbase.base_state(Photon(), D(), mom, QEDbase.PolX()) ==
@test base_state(Photon(), D(), mom, QEDbase.PolX()) ==
both_photon_states[1]
@test QEDbase.base_state(Photon(), D(), mom, QEDbase.PolY()) ==
@test base_state(Photon(), D(), mom, QEDbase.PolY()) ==
both_photon_states[2]

@test QEDbase._as_svec(
QEDbase.base_state(Photon(), D(), mom, QEDbase.PolX())
base_state(Photon(), D(), mom, QEDbase.PolX())
) isa SVector
@test QEDbase._as_svec(
QEDbase.base_state(Photon(), D(), mom, QEDbase.PolY())
base_state(Photon(), D(), mom, QEDbase.PolY())
) isa SVector
@test QEDbase._as_svec(
QEDbase.base_state(Photon(), D(), mom, QEDbase.AllPol())
base_state(Photon(), D(), mom, QEDbase.AllPol())
) isa SVector

@test QEDbase._as_svec(
QEDbase.base_state(Photon(), D(), mom, QEDbase.PolX())
base_state(Photon(), D(), mom, QEDbase.PolX())
)[1] == both_photon_states[1]
@test QEDbase._as_svec(
QEDbase.base_state(Photon(), D(), mom, QEDbase.PolY())
base_state(Photon(), D(), mom, QEDbase.PolY())
)[1] == both_photon_states[2]
@test QEDbase._as_svec(
QEDbase.base_state(Photon(), D(), mom, QEDbase.AllPol())
base_state(Photon(), D(), mom, QEDbase.AllPol())
)[1] == both_photon_states[1]
@test QEDbase._as_svec(
QEDbase.base_state(Photon(), D(), mom, QEDbase.AllPol())
base_state(Photon(), D(), mom, QEDbase.AllPol())
)[2] == both_photon_states[2]
end
end
Expand Down
4 changes: 4 additions & 0 deletions test/runtests.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,10 @@ begin
include("particles/types.jl")
end

@time @safetestset "particle types" begin
include("particles/states.jl")
end

@time @safetestset "particle spinors" begin
include("particles/spinors.jl")
end
Expand Down

0 comments on commit da639a5

Please sign in to comment.