format

Author: maartenvd

src/operators/windowhamiltonian.jl

@@ -8,33 +8,33 @@ Acts simalar as just a finite hamiltonian, but we 'remember' the boundary hamilt
 # todo - what is the required interface for abstractmpo?
 # support densempo windows?
 struct WindowMPOHamiltonian{O} <: AbstractMPO{O}
-    left_ham :: InfiniteMPOHamiltonian{O}
-    finite_ham :: FiniteMPOHamiltonian{O}
-    right_ham :: InfiniteMPOHamiltonian{O}
+    left_ham::InfiniteMPOHamiltonian{O}
+    finite_ham::FiniteMPOHamiltonian{O}
+    right_ham::InfiniteMPOHamiltonian{O}
 end
 
 #utility constructor
 function WindowMPOHamiltonian(ham::InfiniteMPOHamiltonian, interval::UnitRange)
-    
+
     # to make sure the interval corresponds with finite_ham, it is important that the unitcell of the left/right hamiltonians is circshifted correctly
-    left_edge = (interval.start-1) % length(ham)
-    left_ham = InfiniteMPOHamiltonian([ham[i] for i in (left_edge-length(ham)+1):left_edge])
-    right_edge = (interval.stop+1)%length(ham)
-    right_ham = InfiniteMPOHamiltonian([ham[i] for i in right_edge:(right_edge+length(ham)-1)])
+    left_edge = (interval.start - 1) % length(ham)
+    left_ham = InfiniteMPOHamiltonian([ham[i] for i in (left_edge - length(ham) + 1):left_edge])
+    right_edge = (interval.stop + 1) % length(ham)
+    right_ham = InfiniteMPOHamiltonian([ham[i] for i in right_edge:(right_edge + length(ham) - 1)])
 
-    finite_ham = FiniteMPOHamiltonian([ham[i] for i in  interval])
-    WindowMPOHamiltonian(left_ham, finite_ham, right_ham)
+    finite_ham = FiniteMPOHamiltonian([ham[i] for i in interval])
+    return WindowMPOHamiltonian(left_ham, finite_ham, right_ham)
 end
 
 
 Base.copy(h::WindowMPOHamiltonian) = WindowMPOHamiltonian(copy(h.left_ham), copy(h.finite_ham), copy(h.right_ham))
 
 # some basic linalg
 for fun in (:(Base.:+), :(Base.:-), :(Base.:*))
-    @eval $fun(a::WindowMPOHamiltonian,b::WindowMPOHamiltonian) = WindowMPOHamiltonian($fun(a.left_ham,b.left_ham),$fun(a.finite_ham,b.finite_ham),$fun(a.right_ham,b.right_ham))
+    @eval $fun(a::WindowMPOHamiltonian, b::WindowMPOHamiltonian) = WindowMPOHamiltonian($fun(a.left_ham, b.left_ham), $fun(a.finite_ham, b.finite_ham), $fun(a.right_ham, b.right_ham))
 end
 
 TensorKit.dot(
-        bra::WindowMPS, H::WindowMPOHamiltonian, ket::WindowMPS = bra,
-        envs = environments(bra, H, ket)
-        ) = dot(bra.window, H.finite_ham, ket.window,envs)
+    bra::WindowMPS, H::WindowMPOHamiltonian, ket::WindowMPS = bra,
+    envs = environments(bra, H, ket)
+) = dot(bra.window, H.finite_ham, ket.window, envs)

test/algorithms/dynamical_dmrg.jl

@@ -38,15 +38,15 @@ end
     N = 20
 
     H = transverse_field_ising(g = -4)
-    Ω = InfiniteMPS(ComplexSpace(2),ComplexSpace(20))
+    Ω = InfiniteMPS(ComplexSpace(2), ComplexSpace(20))
 
-    (Ω,_) = find_groundstate(Ω, H, VUMPS(verbosity = verbosity_conv))
-    XΩ = WindowMPS(Ω,N)
-    H_w = WindowMPOHamiltonian(H,1:N)
-    
+    (Ω, _) = find_groundstate(Ω, H, VUMPS(verbosity = verbosity_conv))
+    XΩ = WindowMPS(Ω, N)
+    H_w = WindowMPOHamiltonian(H, 1:N)
 
-    gs_en =  expectation_value(XΩ, H_w)
-    vals = range(gs_en - 1.0,gs_en+1.0, length=5)
+
+    gs_en = expectation_value(XΩ, H_w)
+    vals = range(gs_en - 1.0, gs_en + 1.0, length = 5)
     eta = 0.3im
     predicted = [1 / (v + eta - gs_en) for v in vals]
 

test/states/windowmps.jl

@@ -63,7 +63,7 @@ using TensorKit: ℙ
 
     e1 = expectation_value(window, (2, 3) => O)
 
-    w_ham = WindowMPOHamiltonian(ham,1:length(window))
+    w_ham = WindowMPOHamiltonian(ham, 1:length(window))
     window, envs, _ = find_groundstate(window, w_ham, DMRG(; verbosity = 0))
 
     e2 = expectation_value(window, (2, 3) => O)