@@ -57,14 +57,14 @@ function ITensors.dag(ψ::InfiniteCanonicalMPS, args...; kwargs...)
5757 return fmap (x -> dag (x, args... ; kwargs... ), ψ)
5858end
5959
60- ITensors . siteinds (f:: typeof (only), ψ:: InfiniteCanonicalMPS ) = siteinds (f, ψ. AL)
60+ ITensorMPS . siteinds (f:: typeof (only), ψ:: InfiniteCanonicalMPS ) = siteinds (f, ψ. AL)
6161
6262getcell (i:: Index ) = ITensorInfiniteMPS. getcell (tags (i))
6363getsite (i:: Index ) = getsite (tags (i))
6464
6565# Before, for a two site Hamiltonian, findsites(ψ, H[3]) would return [1, 2]
6666# Appeared unsafe for index purpose
67- function ITensors . findfirstsiteind (ψ:: InfiniteMPS , i:: Index )
67+ function ITensorMPS . findfirstsiteind (ψ:: InfiniteMPS , i:: Index )
6868 c = ITensorInfiniteMPS. getcell (i)
6969 n1 = getsite (i)
7070 # if translator(ψ) == translatecelltags
@@ -95,28 +95,28 @@ function ITensors.findfirstsiteind(ψ::InfiniteMPS, i::Index)
9595 # return 0
9696 # end
9797end
98- function ITensors . findfirstsiteind (ψ:: InfiniteCanonicalMPS , i:: Index )
99- return ITensors . findfirstsiteind (ψ. AL, i)
98+ function ITensorMPS . findfirstsiteind (ψ:: InfiniteCanonicalMPS , i:: Index )
99+ return ITensorMPS . findfirstsiteind (ψ. AL, i)
100100end
101101
102- function ITensors . findsites (ψ:: InfiniteCanonicalMPS , is:: Union{<:Tuple,<:Vector} )
102+ function ITensorMPS . findsites (ψ:: InfiniteCanonicalMPS , is:: Union{<:Tuple,<:Vector} )
103103 return sort ([ITensors. findfirstsiteind (ψ, i) for i in is])
104104end
105- function ITensors . findsites (ψ:: InfiniteMPS , is:: Union{<:Tuple,<:Vector} )
105+ function ITensorMPS . findsites (ψ:: InfiniteMPS , is:: Union{<:Tuple,<:Vector} )
106106 return sort ([ITensors. findfirstsiteind (ψ, i) for i in is])
107107end
108- function ITensors . findsites (ψ:: InfiniteMPS , T:: MPO )
108+ function ITensorMPS . findsites (ψ:: InfiniteMPS , T:: MPO )
109109 s = [noprime (filterinds (T[x]; plev= 1 )[1 ]) for x in 1 : length (T)]
110110 return sort ([ITensors. findfirstsiteind (ψ, i) for i in s])
111111end
112- ITensors . findsites (ψ:: InfiniteCanonicalMPS , T:: MPO ) = findsites (ψ. AL, T)
112+ ITensorMPS . findsites (ψ:: InfiniteCanonicalMPS , T:: MPO ) = findsites (ψ. AL, T)
113113
114114# Kept for historical reason
115- function ITensors . findsites (ψ:: InfiniteMPS , T:: ITensor )
115+ function ITensorMPS . findsites (ψ:: InfiniteMPS , T:: ITensor )
116116 s = filterinds (T; plev= 0 )
117117 return sort ([ITensors. findfirstsiteind (ψ, i) for i in s])
118118end
119- ITensors . findsites (ψ:: InfiniteCanonicalMPS , T:: ITensor ) = findsites (ψ. AL, T)
119+ ITensorMPS . findsites (ψ:: InfiniteCanonicalMPS , T:: ITensor ) = findsites (ψ. AL, T)
120120
121121# For now, only represents nearest neighbor interactions
122122# on a linear chain
@@ -193,20 +193,20 @@ function nrange(h::MPO; ncell=1)
193193 return ns[end ] - ns[1 ] + 1
194194end
195195
196- ITensors . findsites (h:: InfiniteSum ) = [findsites (h, n) for n in 1 : nsites (h)]
197- ITensors . findsites (h:: InfiniteSum , n:: Int64 ) = findsites (h. data[n]; ncell= nsites (h))
198- # ITensors .findsites(h::InfiniteSum, is::Union{<:Tuple,<:Vector}) = [findsites(h.data[n], is) for n in 1:nsites(h)]
199- # ITensors .findsites(h::InfiniteSum, i::Index) = [findsites(h.data[n], i) for n in 1:nsites(h)]
196+ ITensorMPS . findsites (h:: InfiniteSum ) = [findsites (h, n) for n in 1 : nsites (h)]
197+ ITensorMPS . findsites (h:: InfiniteSum , n:: Int64 ) = findsites (h. data[n]; ncell= nsites (h))
198+ # ITensorMPS .findsites(h::InfiniteSum, is::Union{<:Tuple,<:Vector}) = [findsites(h.data[n], is) for n in 1:nsites(h)]
199+ # ITensorMPS .findsites(h::InfiniteSum, i::Index) = [findsites(h.data[n], i) for n in 1:nsites(h)]
200200# TODO improve the findsites routines for Infinite Sum
201201
202- function ITensors . findfirstsiteind (i:: Index , ncell:: Int64 )
202+ function ITensorMPS . findfirstsiteind (i:: Index , ncell:: Int64 )
203203 c = ITensorInfiniteMPS. getcell (i)
204204 n1 = getsite (i)
205205 return (c - 1 ) * ncell + n1
206206end
207- function ITensors . findsites (h:: MPO ; ncell:: Int64 = 1 )
207+ function ITensorMPS . findsites (h:: MPO ; ncell:: Int64 = 1 )
208208 s = [filterinds (h[x]; plev= 1 )[1 ] for x in 1 : length (h)]
209- return sort ([ITensors . findfirstsiteind (i, ncell) for i in s])
209+ return sort ([ITensorMPS . findfirstsiteind (i, ncell) for i in s])
210210end
211211
212212# Kept for historical reasons
@@ -223,12 +223,12 @@ function nrange(ψ::InfiniteCanonicalMPS, h::MPO)
223223 return ns[end ] - ns[1 ] + 1
224224end
225225
226- function ITensors . findfirstsiteind (h:: ITensor , i:: Index , ncell:: Int64 )
226+ function ITensorMPS . findfirstsiteind (h:: ITensor , i:: Index , ncell:: Int64 )
227227 c = getcell (i)
228228 n1 = getsite (i)
229229 return (c - 1 ) * ncell + n1
230230end
231- function ITensors . findsites (h:: ITensor ; ncell:: Int64 = 1 )
231+ function ITensorMPS . findsites (h:: ITensor ; ncell:: Int64 = 1 )
232232 s = filterinds (h; plev= 0 )
233233 return sort ([ITensors. findfirstsiteind (h, i, ncell) for i in s])
234234end
0 commit comments