add a trace dispatch for normalized objects

QuantumSavory · Oct 31, 2023 · 62d01f8 · 62d01f8
1 parent 3d058f8
commit 62d01f8
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/src/StatesZoo/StatesZoo.jl b/src/StatesZoo/StatesZoo.jl
@@ -8,7 +8,7 @@ import QuantumSymbolics: express_nolookup
 using LinearAlgebra
 import LinearAlgebra: tr
 
-export SingleRailMidSwapBellU, SingleRailMidSwapBellN, DualRailMidSwapBellU, DualRailMidSwapBellN, ZALMSpinPairU, ZALMSpinPairN
+export SingleRailMidSwapBellU, SingleRailMidSwapBellN, DualRailMidSwapBellU, DualRailMidSwapBellN, ZALMSpinPairU, ZALMSpinPairN, tr
 
 abstract type AbstractTwoQubitState <: QuantumSymbolics.AbstractTwoQubitOp end #For representing density matrices
 Base.show(io::IO, x::AbstractTwoQubitState) = print(io, "$(symbollabel(x))")

diff --git a/src/StatesZoo/single_dual_rail_midswap/single_dual_rail_midswap.jl b/src/StatesZoo/single_dual_rail_midswap/single_dual_rail_midswap.jl
@@ -187,4 +187,9 @@ end
 function express_nolookup(x::DualRailMidSwapBellN, ::QuantumOpticsRepr)
     data = midswap_dual_rail(x.eA, x.eB, x.gA, x.gB, x.Pd, x.Vis)
     return SparseOperator(_bspin⊗_bspin, Complex.(data/tr(data)))
-end
+end
+
+# Symbolic trace
+
+tr(::SingleRailMidSwapBellN) = 1
+tr(::DualRailMidSwapBellN) = 1
diff --git a/src/StatesZoo/zalm_pair/zalm_pair.jl b/src/StatesZoo/zalm_pair/zalm_pair.jl
@@ -672,4 +672,7 @@ end
 function express_nolookup(x::ZALMSpinPairN, ::QuantumOpticsRepr)
     data = cascaded_source_spin(x.Ns, x.gA, x.gB, x.eAm, x.eBm, x.eAs, x.eBs, x.eD, x.Pd, x.Pdo1, x.Pdo2, x.VisF)
     return SparseOperator(_bspin⊗_bspin, Complex.(data/tr(data)))
-end
+end
+
+## Symbolic trace 
+tr(::ZALMSpinPairN) = 1