Number of spins or pols

src/QEDbase.jl

@@ -62,6 +62,7 @@ export AbstractDefinitePolarization, AbstractIndefinitePolarization
 export PolarizationX, PolX, PolarizationY, PolY, AllPolarization, AllPol
 export AbstractDefiniteSpin, AbstractIndefiniteSpin
 export SpinUp, SpinDown, AllSpin
+export number_of_spin_pol
 
 using StaticArrays
 using LinearAlgebra

src/particles/particle_spin_pol.jl

@@ -177,3 +177,15 @@ y-polarized
 struct PolarizationY <: AbstractDefinitePolarization end
 const PolY = PolarizationY
 Base.show(io::IO, ::MIME"text/plain", ::PolY) = print(io, "y-polarized")
+
+"""
+    number_of_spin_pol(spin_or_pol)
+
+Return the number of spins or polarizations respresented by `spin_or_pol`, e.g. `number_of_spin_pol(SpinUp()) == 1`.
+
+"""
+function number_of_spin_pol end
+number_of_spin_pol(::AbstractDefinitePolarization) = 1
+number_of_spin_pol(::AbstractDefiniteSpin) = 1
+number_of_spin_pol(::AbstractIndefinitePolarization) = 2
+number_of_spin_pol(::AbstractIndefiniteSpin) = 2

test/particles.jl

@@ -52,6 +52,19 @@ test_broadcast(x::AbstractSpinOrPolarization) = x
     end
 end
 
+@testset "number of spins or pols" begin
+    @testset "single" begin
+        @testset "$spin_or_pol" for spin_or_pol in (SpinUp(), SpinDown(), PolX(), PolY())
+            @test number_of_spin_pol(spin_or_pol) == 1
+        end
+    end
+    @testset "multiple" begin
+        @testset "$spin_or_pol" for spin_or_pol in (AllSpin(), AllPol())
+            @test number_of_spin_pol(spin_or_pol) == 2
+        end
+    end
+end
+
 @testset "fermion likes" begin
     @testset "fermion" begin
         struct TestFermion <: Fermion end