nsubsystems move (#44)

QuantumSavory · Sep 25, 2023 · d33a453 · d33a453
1 parent 763be70
commit d33a453
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 10 deletions.
diff --git a/Project.toml b/Project.toml
@@ -37,9 +37,9 @@ Makie = "0.19"
 NetworkLayout = "0.4.4"
 PrecompileTools = "1"
 QuantumClifford = "0.8"
-QuantumInterface = "0.3.0"
+QuantumInterface = "0.3.3"
 QuantumOptics = "1.0.5"
-QuantumOpticsBase = "0.4.16"
+QuantumOpticsBase = "0.4.17"
 QuantumSymbolics = "0.2.4"
 Reexport = "1.2.2"
 ResumableFunctions = "0.6.1"

diff --git a/src/QuantumSavory.jl b/src/QuantumSavory.jl
@@ -8,6 +8,7 @@ using Graphs
 
 using QuantumInterface: basis, tensor, ⊗, apply!, traceout!,
     AbstractOperator, AbstractKet, AbstractSuperOperator, Basis, SpinBasis
+import QuantumInterface: nsubsystems
 
 export apply!, traceout!, removebackref!
 export project_traceout! #TODO should move to QuantumInterface

diff --git a/src/backends/quantumoptics/quantumoptics.jl b/src/backends/quantumoptics/quantumoptics.jl
@@ -5,17 +5,10 @@ import QuantumOpticsBase: GenericBasis, CompositeBasis,
     projector, identityoperator, embed, dm, expect, ptrace, spre, spost
 import QuantumOptics
 import QuantumOptics: timeevolution
+import QuantumInterface: nsubsystems
 
 const QOR = QuantumOpticsRepr()
 
-nsubsystems(s::Ket) = nsubsystems(s.basis)
-function nsubsystems(s::Operator)
-    @assert s.basis_l == s.basis_r
-    nsubsystems(s.basis_l)
-end
-nsubsystems(b::CompositeBasis) = length(b.bases)
-nsubsystems(b::Basis) = 1
-
 subsystemcompose(states::Ket...) = tensor(states...)
 subsystemcompose(ops::Operator...) = tensor(ops...)
 subsystemcompose(k::Ket, op::Operator) = tensor(dm(k),op) # TODO this should be more general