Define AbstractProblem. Work on saving max truncerror

Author: emstoudenmire

src/ITensorNetworks.jl

@@ -62,6 +62,7 @@ include("solvers/subspace/subspace.jl")
 include("solvers/subspace/densitymatrix.jl")
 include("solvers/align_indices.jl")
 include("solvers/operator_map.jl")
+include("solvers/abstract_problem.jl")
 include("solvers/eigsolve.jl")
 include("solvers/applyexp.jl")
 include("solvers/fitting.jl")

src/solvers/abstract_problem.jl

@@ -0,0 +1,4 @@
+
+abstract type AbstractProblem end
+
+set_truncation_info(P::AbstractProblem, args...; kws...) = P

src/solvers/applyexp.jl

@@ -1,6 +1,6 @@
 using Printf: @printf
 
-@kwdef mutable struct ApplyExpProblem{State}
+@kwdef mutable struct ApplyExpProblem{State} <: AbstractProblem
   operator
   state::State
   current_exponent::Number = 0.0

src/solvers/eigsolve.jl

@@ -1,22 +1,37 @@
 using Printf: @printf
+using ITensors: truncerror
 
-@kwdef mutable struct EigsolveProblem{State,Operator}
+@kwdef mutable struct EigsolveProblem{State,Operator} <: AbstractProblem
   operator::Operator
   state::State
   eigenvalue::Number = Inf
+  max_truncerr::Real = 0.0
 end
 
 eigenvalue(E::EigsolveProblem) = E.eigenvalue
 state(E::EigsolveProblem) = E.state
 operator(E::EigsolveProblem) = E.operator
+max_truncerr(E::EigsolveProblem) = E.max_truncerr
 
 function set_operator(E::EigsolveProblem, operator)
-  EigsolveProblem(operator, E.state, E.eigenvalue)
+  EigsolveProblem(operator, E.state, E.eigenvalue, E.max_truncerr)
 end
 function set_eigenvalue(E::EigsolveProblem, eigenvalue)
-  EigsolveProblem(E.operator, E.state, eigenvalue)
+  EigsolveProblem(E.operator, E.state, eigenvalue, E.max_truncerr)
+end
+function set_state(E::EigsolveProblem, state)
+  EigsolveProblem(E.operator, state, E.eigenvalue, E.max_truncerr)
+end
+function set_max_truncerr(E::EigsolveProblem, truncerr)
+  EigsolveProblem(E.operator, E.state, E.eigenvalue, truncerr)
+end
+
+function set_truncation_info(E::EigsolveProblem; spectrum=nothing)
+  if !isnothing(spectrum)
+    E = set_max_truncerr(E, max(max_truncerr(E), truncerror(spectrum)))
+  end
+  return E
 end
-set_state(E::EigsolveProblem, state) = EigsolveProblem(E.operator, state, E.eigenvalue)
 
 function update(
   prob::EigsolveProblem,
@@ -43,8 +58,9 @@ function sweep_printer(problem::EigsolveProblem; outputlevel, sweep, nsweeps, kw
     else
       @printf("After sweep %d/%d ", sweep, nsweeps)
     end
-    @printf("eigenvalue=%.12f ", eigenvalue(problem))
-    @printf("maxlinkdim=%d", maxlinkdim(state(problem)))
+    @printf("eigenvalue=%.12f", eigenvalue(problem))
+    @printf(" maxlinkdim=%d", maxlinkdim(state(problem)))
+    @printf(" max truncerr=%d", max_truncerr(problem))
     println()
     flush(stdout)
   end

src/solvers/fitting.jl

@@ -4,7 +4,8 @@ using NamedGraphs.PartitionedGraphs: partitionedges
 using Printf: @printf
 using ConstructionBase: setproperties
 
-@kwdef mutable struct FittingProblem{State<:AbstractBeliefPropagationCache}
+@kwdef mutable struct FittingProblem{State<:AbstractBeliefPropagationCache} <:
+                      AbstractProblem
   state::State
   ket_graph::AbstractNamedGraph
   overlap::Number = 0

src/solvers/insert.jl

@@ -8,6 +8,7 @@ function insert(
   set_orthogonal_region=true,
   sweep,
   trunc=(;),
+  outputlevel=0,
   kws...,
 )
   trunc = truncation_parameters(sweep; trunc...)
@@ -19,14 +20,16 @@ function insert(
     e = edgetype(psi)(first(region), last(region))
     indsTe = inds(psi[first(region)])
     tags = ITensors.tags(psi, e)
-    U, C, _ = factorize(local_tensor, indsTe; tags, trunc...)
+    U, C, spectrum = factorize(local_tensor, indsTe; tags, trunc...)
     @preserve_graph psi[first(region)] = U
+    problem = set_truncation_info(problem; spectrum)
   else
     error("Region of length $(length(region)) not currently supported")
   end
   v = last(region)
   @preserve_graph psi[v] = C
   psi = set_orthogonal_region ? set_ortho_region(psi, [v]) : psi
   normalize && @preserve_graph psi[v] = psi[v] / norm(psi[v])
-  return set_state(problem, psi)
+  problem = set_state(problem, psi)
+  return problem
 end