utility mps copy

Author: lkdvos

src/states/infinitemps.jl

@@ -216,6 +216,14 @@ Base.size(ψ::InfiniteMPS, args...) = size(ψ.AL, args...)
 Base.length(ψ::InfiniteMPS) = length(ψ.AL)
 Base.eltype(ψ::InfiniteMPS) = eltype(ψ.AL)
 Base.copy(ψ::InfiniteMPS) = InfiniteMPS(copy(ψ.AL), copy(ψ.AR), copy(ψ.C), copy(ψ.AC))
+function Base.copy!(ψ::InfiniteMPS, ϕ::InfiniteMPS)
+    copy!.(ψ.AL, ϕ.AL)
+    copy!.(ψ.AR, ϕ.AR)
+    copy!.(ψ.AC, ϕ.AC)
+    copy!.(ψ.C, ϕ.C)
+    return ψ
+end
+
 function Base.repeat(ψ::InfiniteMPS, i::Int)
     return InfiniteMPS(repeat(ψ.AL, i), repeat(ψ.AR, i), repeat(ψ.C, i), repeat(ψ.AC, i))
 end

src/states/multilinemps.jl

@@ -83,5 +83,8 @@ TensorKit.normalize!(a::MultilineMPS) = (normalize!.(parent(a)); return a)
 Base.convert(::Type{MultilineMPS}, st::InfiniteMPS) = Multiline([st])
 Base.convert(::Type{InfiniteMPS}, st::MultilineMPS) = only(st)
 Base.eltype(t::MultilineMPS) = eltype(t[1])
+Base.copy!(ψ::MultilineMPS, ϕ::MultilineMPS) = (copy!.(parent(ψ), parent(ϕ));
+                                                ψ)
+
 left_virtualspace(t::MultilineMPS, i::Int, j::Int) = left_virtualspace(t[i], j)
 right_virtualspace(t::MultilineMPS, i::Int, j::Int) = right_virtualspace(t[i], j)