Move code to ITensors.ITensorMPS

Project.toml

@@ -1,21 +1,14 @@
 name = "ITensorTDVP"
 uuid = "25707e16-a4db-4a07-99d9-4d67b7af0342"
 authors = ["Matthew Fishman <mfishman@flatironinstitute.org> and contributors"]
-version = "0.2.1"
+version = "0.3"
 
 [deps]
 ITensors = "9136182c-28ba-11e9-034c-db9fb085ebd5"
 KrylovKit = "0b1a1467-8014-51b9-945f-bf0ae24f4b77"
-LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
-Observers = "338f10d5-c7f1-4033-a7d1-f9dec39bcaa0"
-Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
-TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
 
 [compat]
-ITensors = "0.3.58"
-KrylovKit = "0.6"
-Observers = "0.2"
-TimerOutputs = "0.5"
+ITensors = "0.3.60"
 julia = "1.6"
 
 [extras]

src/ITensorTDVP.jl

@@ -1,19 +1,51 @@
 module ITensorTDVP
-export TimeDependentSum, dmrg_x, linsolve, tdvp, to_vec
 
-include("defaults.jl")
-include("update_observer.jl")
+using ITensors: ITensors
+using ITensors.ITensorMPS: ITensorMPS, MPO, MPS
+using ITensors.NDTensors: @Algorithm_str
+using KrylovKit: KrylovKit
+
+function tdvp(operator, t::Number, initial_state::MPS; kwargs...)
+  return ITensorMPS.itensortdvp_tdvp(operator, t, initial_state; kwargs...)
+end
+function tdvp(t::Number, operator, initial_state::MPS; kwargs...)
+  return ITensorMPS.itensortdvp_tdvp(t, operator, initial_state; kwargs...)
+end
+function tdvp(operator, initial_state::MPS, t::Number; kwargs...)
+  return ITensorMPS.itensortdvp_tdvp(operator, initial_state, t; kwargs...)
+end
+function tdvp(solver, operator::MPO, t::Number, initial_state::MPS; kwargs...)
+  return ITensorMPS.itensortdvp_tdvp(solver, operator, t, initial_state; kwargs...)
+end
+function tdvp(solver, t::Number, operator::MPO, initial_state::MPS; kwargs...)
+  return ITensorMPS.itensortdvp_tdvp(solver, t, operator, initial_state; kwargs...)
+end
+function tdvp(solver, initial_state::MPS, t::Number, operator::MPO; kwargs...)
+  return ITensorMPS.itensortdvp_tdvp(solver, initial_state, t, operator; kwargs...)
+end
+function tdvp(solver, operators::Vector{MPO}, t::Number, initial_state::MPS; kwargs...)
+  return ITensorMPS.itensortdvp_tdvp(solver, operators, t, initial_state; kwargs...)
+end
+
+function KrylovKit.linsolve(
+  A::MPO, b::MPS, x0::MPS, a0::Number=false, a1::Number=true; kwargs...
+)
+  return ITensorMPS.itensortdvp_linsolve(A, b, x0, a0, a1; kwargs...)
+end
+
+function dmrg(operator, initial_state::MPS; kwargs...)
+  return ITensorMPS.itensortdvp_dmrg(operator, initial_state; kwargs...)
+end
+
+function dmrg_x(operator, initial_state::MPS; kwargs...)
+  return ITensorMPS.itensortdvp_dmrg_x(operator, initial_state; kwargs...)
+end
+
+function ITensors.contract(alg::Algorithm"fit", A::MPO, psi::MPS; kwargs...)
+  return ITensorMPS.itensortdvp_contract(alg, A, psi; kwargs...)
+end
+
 include("solver_utils.jl")
-include("tdvporder.jl")
-include("tdvpinfo.jl")
-include("sweep_update.jl")
-include("alternating_update.jl")
-include("tdvp.jl")
-include("dmrg.jl")
-include("dmrg_x.jl")
-include("projmpo_apply.jl")
-include("contract_mpo_mps.jl")
-include("projmps2.jl")
-include("projmpo_mps2.jl")
-include("linsolve.jl")
+export TimeDependentSum, to_vec
+
 end