Implement inds for getting physical and virtual indices subsets

bsc-quantic · Aug 14, 2024 · 10b5f10 · 10b5f10
1 parent 9467998
commit 10b5f10
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/Quantum.jl b/src/Quantum.jl
@@ -1,3 +1,5 @@
+using KeywordDispatch
+
 """
     AbstractQuantum
 
@@ -174,6 +176,15 @@ function Base.show(io::IO, q::Quantum)
 end
 
 @kwmethod inds(tn::AbstractQuantum; at) = Quantum(tn).sites[at]
+@kwmethod function inds(tn::AbstractQuantum; set)
+    if set === :physical
+        return values(Quantum(tn).sites)
+    elseif set === :virtual
+        return setdiff(inds(tn), values(Quantum(tn).sites))
+    else
+        return inds(TensorNetwork(tn); set)
+    end
+end
 
 """
     adjoint(q::Quantum)