Skip to content

Commit 7863a9b

Browse files
committed
format
1 parent 8f2d090 commit 7863a9b

File tree

3 files changed

+31
-27
lines changed

3 files changed

+31
-27
lines changed

src/algorithms/contractions/ctmrg_contractions.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1320,7 +1320,7 @@ function renormalize_west_edge(
13201320

13211321
corner[χS DEt DEb; χNW DNt DNb] :=
13221322
PEket[χS χNW DNt DEt; DSb DWb d] * conj(bra(A)[d; DNb DEb DSb DWb])
1323-
1323+
13241324
edge[χS DEt DEb; χN] := corner[χS DEt DEb; χNW DNt DNb] * P_bottom[χNW DNt DNb; χN]
13251325
end
13261326
end

src/algorithms/ctmrg/sequential.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,9 @@ for a specific `coordinate` (where `dir=WEST` is already implied in the `:sequen
8181
"""
8282
function sequential_projectors(col::Int, network, env::CTMRGEnv, alg::ProjectorAlgorithm)
8383
coordinates = eachcoordinate(env)[:, col]
84-
T_dst = Base.promote_op(sequential_projectors, NTuple{3,Int}, typeof(network), typeof(env), typeof(alg))
84+
T_dst = Base.promote_op(
85+
sequential_projectors, NTuple{3,Int}, typeof(network), typeof(env), typeof(alg)
86+
)
8587
proj_and_info = similar(coordinates, T_dst)
8688
proj_and_info′::typeof(proj_and_info) = dtmap!!(proj_and_info, coordinates) do (r, c)
8789
trscheme = truncation_scheme(alg, env.edges[WEST, _prev(r, size(env, 2)), c])

src/algorithms/ctmrg/simultaneous.jl

Lines changed: 27 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ function ctmrg_iteration(network, env::CTMRGEnv, alg::SimultaneousCTMRG)
4444
enlarged_corners′ = similar(coordinates, T_corners)
4545
enlarged_corners::typeof(enlarged_corners′) =
4646
dtmap!!(enlarged_corners′, eachcoordinate(network, 1:4)) do idx
47-
return TensorMap(EnlargedCorner(network, env, idx))
48-
end # expand environment
47+
return TensorMap(EnlargedCorner(network, env, idx))
48+
end # expand environment
4949
projectors, info = simultaneous_projectors(enlarged_corners, env, alg.projector_alg) # compute projectors on all coordinates
5050
env′ = renormalize_simultaneously(enlarged_corners, projectors, network, env) # renormalize enlarged corners
5151
return env′, info
@@ -90,14 +90,11 @@ function simultaneous_projectors(
9090
proj_and_info::typeof(proj_and_info′) =
9191
dtmap!!(proj_and_info′, coordinates) do coordinate
9292
return simultaneous_projectors(coordinate, enlarged_corners, env, alg)
93-
end
93+
end
9494
return _split_proj_and_info(proj_and_info)
9595
end
9696
function simultaneous_projectors(
97-
coordinate,
98-
enlarged_corners::Array{E,3},
99-
env,
100-
alg::HalfInfiniteProjector,
97+
coordinate, enlarged_corners::Array{E,3}, env, alg::HalfInfiniteProjector
10198
) where {E}
10299
coordinate′ = _next_coordinate(coordinate, size(env)[2:3]...)
103100
trscheme = truncation_scheme(alg, env.edges[coordinate[1], coordinate′[2:3]...])
@@ -106,10 +103,7 @@ function simultaneous_projectors(
106103
return compute_projector(ec, coordinate, alg′)
107104
end
108105
function simultaneous_projectors(
109-
coordinate,
110-
enlarged_corners::Array{E,3},
111-
env,
112-
alg::FullInfiniteProjector,
106+
coordinate, enlarged_corners::Array{E,3}, env, alg::FullInfiniteProjector
113107
) where {E}
114108
coordinate′ = _next_coordinate(coordinate, size(env)[2:3]...)
115109
trscheme = truncation_scheme(alg, env.edges[coordinate[1], coordinate′[2:3]...])
@@ -139,21 +133,29 @@ function renormalize_simultaneously(enlarged_corners, projectors, network, env)
139133
corners_edges′ = similar(coordinates, T_CE)
140134
corners_edges::typeof(corners_edges′) =
141135
dtmap!!(corners_edges′, coordinates) do (dir, r, c)
142-
if dir == NORTH
143-
corner = renormalize_northwest_corner((r, c), enlarged_corners, P_left, P_right)
144-
edge = renormalize_north_edge((r, c), env, P_left, P_right, network)
145-
elseif dir == EAST
146-
corner = renormalize_northeast_corner((r, c), enlarged_corners, P_left, P_right)
147-
edge = renormalize_east_edge((r, c), env, P_left, P_right, network)
148-
elseif dir == SOUTH
149-
corner = renormalize_southeast_corner((r, c), enlarged_corners, P_left, P_right)
150-
edge = renormalize_south_edge((r, c), env, P_left, P_right, network)
151-
elseif dir == WEST
152-
corner = renormalize_southwest_corner((r, c), enlarged_corners, P_left, P_right)
153-
edge = renormalize_west_edge((r, c), env, P_left, P_right, network)
136+
if dir == NORTH
137+
corner = renormalize_northwest_corner(
138+
(r, c), enlarged_corners, P_left, P_right
139+
)
140+
edge = renormalize_north_edge((r, c), env, P_left, P_right, network)
141+
elseif dir == EAST
142+
corner = renormalize_northeast_corner(
143+
(r, c), enlarged_corners, P_left, P_right
144+
)
145+
edge = renormalize_east_edge((r, c), env, P_left, P_right, network)
146+
elseif dir == SOUTH
147+
corner = renormalize_southeast_corner(
148+
(r, c), enlarged_corners, P_left, P_right
149+
)
150+
edge = renormalize_south_edge((r, c), env, P_left, P_right, network)
151+
elseif dir == WEST
152+
corner = renormalize_southwest_corner(
153+
(r, c), enlarged_corners, P_left, P_right
154+
)
155+
edge = renormalize_west_edge((r, c), env, P_left, P_right, network)
156+
end
157+
return corner / norm(corner), edge / norm(edge)
154158
end
155-
return corner / norm(corner), edge / norm(edge)
156-
end
157159

158160
return CTMRGEnv(map(first, corners_edges), map(last, corners_edges))
159161
end

0 commit comments

Comments
 (0)