Fix @reindex! when virtual indices intersect

bsc-quantic · Aug 2, 2024 · b7a3b11 · b7a3b11
1 parent 5995710
commit b7a3b11
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/Quantum.jl b/src/Quantum.jl
@@ -256,6 +256,8 @@ function reindex!(a::Quantum, ioa, b::Quantum, iob)
         return b
     end
 
+    resetindex_mapping = resetindex!(Val(:return_mapping), TensorNetwork(b); init=ninds(TensorNetwork(a)))
+    replacements = merge!(resetindex_mapping, Dict(replacements))
     replace!(TensorNetwork(b), replacements...)
 
     for site in sitesb

diff --git a/src/TensorNetwork.jl b/src/TensorNetwork.jl
@@ -212,9 +212,13 @@ See also: [`ntensors`](@ref)
 """
 ninds(tn::TensorNetwork) = length(tn.indexmap)
 
-function resetindex!(tn::TensorNetwork)
-    gen = IndexCounter()
-    mapping = Dict{Symbol,Symbol}([i => nextindex!(gen) for i in inds(tn)])
+function resetindex!(::Val{:return_mapping}, tn::TensorNetwork; init::Int=1)
+    gen = IndexCounter(init)
+    return Dict{Symbol,Symbol}([i => nextindex!(gen) for i in inds(tn)])
+end
+
+function resetindex!(tn::TensorNetwork; init::Int=1)
+    mapping = resetindex!(Val(:return_mapping), tn; init=init)
     return replace!(tn, mapping)
 end