Network solvers redesign

Project.toml

@@ -1,12 +1,14 @@
 name = "ITensorNetworks"
 uuid = "2919e153-833c-4bdc-8836-1ea460a35fc7"
 authors = ["Matthew Fishman <mfishman@flatironinstitute.org>, Joseph Tindall <jtindall@flatironinstitute.org> and contributors"]
-version = "0.13.16"
+version = "0.14.0"
 
 [deps]
 AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
+Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
 Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
+ConstructionBase = "187b0558-2788-49d3-abe0-74a17ed4e7c9"
 DataGraphs = "b5a273c3-7e6c-41f6-98bd-8d7f1525a36a"
 DataStructures = "864edb3b-99cc-5e75-8d2d-829cb0a9cfe8"
 Dictionaries = "85a47980-9c8c-11e8-2b9f-f7ca1fa99fb4"
@@ -50,9 +52,11 @@ ITensorNetworksTensorOperationsExt = "TensorOperations"
 
 [compat]
 AbstractTrees = "0.4.4"
+Accessors = "0.1.42"
 Adapt = "4"
 Combinatorics = "1"
 Compat = "3, 4"
+ConstructionBase = "1.6.0"
 DataGraphs = "0.2.3"
 DataStructures = "0.18, 0.19"
 Dictionaries = "0.4"

docs/Project.toml

@@ -5,5 +5,5 @@ Literate = "98b081ad-f1c9-55d3-8b20-4c87d4299306"
 
 [compat]
 Documenter = "1.10.0"
-ITensorNetworks = "0.13.0"
+ITensorNetworks = "0.14.0"
 Literate = "2.20.1"

examples/Project.toml

@@ -2,4 +2,4 @@
 ITensorNetworks = "2919e153-833c-4bdc-8836-1ea460a35fc7"
 
 [compat]
-ITensorNetworks = "0.13.2"
+ITensorNetworks = "0.14.0"

src/ITensorNetworks.jl

@@ -34,11 +34,7 @@ include("formnetworks/quadraticformnetwork.jl")
 include("gauging.jl")
 include("utils.jl")
 include("update_observer.jl")
-include("solvers/local_solvers/eigsolve.jl")
-include("solvers/local_solvers/exponentiate.jl")
-include("solvers/local_solvers/dmrg_x.jl")
-include("solvers/local_solvers/contract.jl")
-include("solvers/local_solvers/linsolve.jl")
+
 include("treetensornetworks/abstracttreetensornetwork.jl")
 include("treetensornetworks/treetensornetwork.jl")
 include("treetensornetworks/opsum_to_ttn/matelem.jl")
@@ -48,18 +44,29 @@ include("treetensornetworks/projttns/abstractprojttn.jl")
 include("treetensornetworks/projttns/projttn.jl")
 include("treetensornetworks/projttns/projttnsum.jl")
 include("treetensornetworks/projttns/projouterprodttn.jl")
-include("solvers/solver_utils.jl")
-include("solvers/defaults.jl")
-include("solvers/insert/insert.jl")
-include("solvers/extract/extract.jl")
-include("solvers/alternating_update/alternating_update.jl")
-include("solvers/alternating_update/region_update.jl")
-include("solvers/tdvp.jl")
-include("solvers/dmrg.jl")
-include("solvers/dmrg_x.jl")
-include("solvers/contract.jl")
-include("solvers/linsolve.jl")
-include("solvers/sweep_plans/sweep_plans.jl")
+
+include("solvers/local_solvers/eigsolve.jl")
+include("solvers/local_solvers/exponentiate.jl")
+include("solvers/local_solvers/runge_kutta.jl")
+include("solvers/truncation_parameters.jl")
+include("solvers/iterators.jl")
+include("solvers/adapters.jl")
+include("solvers/sweep_solve.jl")
+include("solvers/region_plans/dfs_plans.jl")
+include("solvers/region_plans/euler_tour.jl")
+include("solvers/region_plans/euler_plans.jl")
+include("solvers/region_plans/tdvp_region_plans.jl")
+include("solvers/extract.jl")
+include("solvers/insert.jl")
+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")
+
 include("apply.jl")
 include("inner.jl")
 include("normalize.jl")

src/itensornetwork.jl

@@ -1,6 +1,6 @@
 using DataGraphs: DataGraphs, DataGraph
 using Dictionaries: Indices, dictionary
-using ITensors: ITensors, ITensor, op, state
+using ITensors: ITensors, ITensor, op
 using .ITensorsExtensions: trivial_space
 using NamedGraphs: NamedGraphs, NamedEdge, NamedGraph, vertextype
 
@@ -180,7 +180,7 @@ function generic_state(x::Op, inds::NamedTuple)
 end
 function generic_state(s::AbstractString, inds::NamedTuple)
   # TODO: Figure out what to do if there is more than one site.
-  site_tensors = [state(s, only(inds.siteinds))]
+  site_tensors = [ITensors.state(s, only(inds.siteinds))]
   link_tensors = [[onehot(i => 1) for i in inds.linkinds[e]] for e in keys(inds.linkinds)]
   return contract(reduce(vcat, link_tensors; init=site_tensors))
 end

src/solvers/abstract_problem.jl

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

src/solvers/adapters.jl

@@ -0,0 +1,32 @@
+
+#
+# TupleRegionIterator
+#
+# Adapts outputs to be (region, region_kwargs) tuples
+#
+# More generic design? maybe just assuming RegionIterator
+# or its outputs implement some interface function that
+# generates each tuple?
+#
+
+mutable struct TupleRegionIterator{RegionIter}
+  region_iterator::RegionIter
+end
+
+region_iterator(T::TupleRegionIterator) = T.region_iterator
+
+function Base.iterate(T::TupleRegionIterator, which=1)
+  state = iterate(region_iterator(T), which)
+  isnothing(state) && return nothing
+  (current_region, region_kwargs) = current_region_plan(region_iterator(T))
+  return (current_region, region_kwargs), last(state)
+end
+
+"""
+  region_tuples(R::RegionIterator)
+
+The `region_tuples` adapter converts a RegionIterator into an 
+iterator which outputs a tuple of the form (current_region, current_region_kwargs)
+at each step.
+"""
+region_tuples(R::RegionIterator) = TupleRegionIterator(R)

src/solvers/align_indices.jl

@@ -0,0 +1,16 @@
+
+function align_indices(tn)
+  si = siteinds(tn)
+  ptn = copy(tn)
+  for v in vertices(tn)
+    is = inds(tn[v])
+    ls = setdiff(is, si[v])
+    isempty(ls) && continue
+    new_is = [first(ls), si[v]...]
+    if length(ls) >= 2
+      new_is = vcat(new_is, ls[2:end])
+    end
+    ptn[v] = permute(tn[v], new_is)
+  end
+  return ptn
+end