From a1ddab1ce84c0eeae2651cc89b8e6670727d13b5 Mon Sep 17 00:00:00 2001
From: "matthias@quintern.xyz" <matthias.quintern@posteo.de>
Date: Sun, 9 Mar 2025 22:01:31 +0100
Subject: [PATCH] refactor

---
 src/algorithms/dmrg.cpp | 14 ++++++++------
 src/algorithms/dmrg.hpp |  5 +++--
 src/algorithms/tebd.cpp |  2 +-
 src/algorithms/tebd.hpp |  2 +-
 4 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/src/algorithms/dmrg.cpp b/src/algorithms/dmrg.cpp
index a1a7597..277a71c 100644
--- a/src/algorithms/dmrg.cpp
+++ b/src/algorithms/dmrg.cpp
@@ -1,6 +1,6 @@
 #include "dmrg.hpp"
 
-#include "lanczos.hpp"
+#include "algorithm.hpp"
 #include "type.hpp"
 #include "util.hpp"
 
@@ -70,14 +70,16 @@ namespace dmrg {
     }
 
 	template<typename T>
-    void DMRGEngine<T>::sweep(MPS<T>& psi) {
+    void DMRGEngine<T>::sweep(MPS<T>& psi, postUpdateFunction_t<T> postUpdateFunc) {
         // Sweep left to right
         for (unsigned i = 0; i < psi.getL() - 2; ++i) {
             update_bond(i, psi);
+            postUpdateFunc(psi, i);
         }
         // Sweep right to left
         for (unsigned i = psi.getL() - 2; i > 0; --i) {
             update_bond(i, psi);
+            postUpdateFunc(psi, i);
         }
     }
 
@@ -273,16 +275,16 @@ namespace dmrg {
     }
 
     template<typename T>
-    void run(MPS<T>& psi, const std::vector<std::shared_ptr<e::Tensor<T, 4>>>& H_mpos, unsigned chiMax, T eps, unsigned nSweeps, postUpdateFunction_t<T> postUpdateFunc) {
+    void run(MPS<T>& psi, const std::vector<std::shared_ptr<e::Tensor<T, 4>>>& H_mpos, unsigned chiMax, T eps, unsigned nSweeps, postUpdateFunction_t<T> postSweepFunc, postUpdateFunction_t<T> postUpdateFunc) {
         auto engine = DMRGEngine(psi, H_mpos, chiMax, eps);
         for (unsigned i = 0; i < nSweeps; i++) {
-            engine.sweep(psi);
-            if (postUpdateFunc(psi, i)) break;
+            engine.sweep(psi, postUpdateFunc);
+            if (postSweepFunc(psi, i)) break;
         }
     }
 
     template class dmrg::DMRGEngine<dtype>;
-	template void run<dtype>(MPS<dtype>& psi, const std::vector<std::shared_ptr<e::Tensor<dtype, 4>>>& H_mpos, unsigned chiMax, dtype eps, unsigned nSweeps, postUpdateFunction_t<dtype> postUpdateFunc);
+	template void run<dtype>(MPS<dtype>& psi, const std::vector<std::shared_ptr<e::Tensor<dtype, 4>>>& H_mpos, unsigned chiMax, dtype eps, unsigned nSweeps, postUpdateFunction_t<dtype>, postUpdateFunction_t<dtype>);
 
 }  // namespace dmrg
 
diff --git a/src/algorithms/dmrg.hpp b/src/algorithms/dmrg.hpp
index 59d675b..36b89bc 100644
--- a/src/algorithms/dmrg.hpp
+++ b/src/algorithms/dmrg.hpp
@@ -41,12 +41,13 @@ namespace dmrg {
                    unsigned chiMax = 100, T eps = 1e-12);
 
         void initialize(MPS<T>& psi); 
-        void sweep(MPS<T>& psi);
+        /// Post update function is called after each bond update, its only meant for emscripten_sleep!
+        void sweep(MPS<T>& psi, postUpdateFunction_t<T> postUpdateFunc=postUpdateNoop<T>);
         void update_bond(int i, MPS<T>& psi);
         void update_RP(int i, MPS<T>& psi); 
         void update_LP(int i, MPS<T>& psi);
     };
 
     template<typename T>
-    void run(MPS<T>& psi, const std::vector<std::shared_ptr<e::Tensor<T, 4>>>& H_mpos, unsigned chiMax, T eps, unsigned nSweeps, postUpdateFunction_t<T> postUpdateFunc);
+    void run(MPS<T>& psi, const std::vector<std::shared_ptr<e::Tensor<T, 4>>>& H_mpos, unsigned chiMax, T eps, unsigned nSweeps, postUpdateFunction_t<T> postSweepFunc, postUpdateFunction_t<T> postUpdateFunc=postUpdateNoop<T>);
 }
diff --git a/src/algorithms/tebd.cpp b/src/algorithms/tebd.cpp
index e1a2971..30a1f87 100644
--- a/src/algorithms/tebd.cpp
+++ b/src/algorithms/tebd.cpp
@@ -29,7 +29,7 @@ std::vector<std::shared_ptr<OutType>> forEachUnique(std::function<OutType(const
 namespace tebd {
     template<typename T>
     e::Tensor<T, 4> getTimeEvolutionMatrix(const e::MatrixX<T>& bond, T dt, unsigned localDim) {
-        e::MatrixX<T> U = (-bond * dt).exp();
+        e::MatrixX<T> U = (bond * (Imag*dt)).exp();
         // std::cout << "U = \n" << U << "\n\n";
         return bondMatrix2Tensor<T>(U, localDim);
     }
diff --git a/src/algorithms/tebd.hpp b/src/algorithms/tebd.hpp
index 500918d..b8152ab 100644
--- a/src/algorithms/tebd.hpp
+++ b/src/algorithms/tebd.hpp
@@ -20,7 +20,7 @@ namespace tebd {
     /**
      * @brief Apply time evolution on a bond
      * @details
-     *  \f[U = \exp{ B \, \partial t} \f]
+     *  \f[U = \exp{ B \, i \partial t} \f]
      */
     template<typename T>
     e::Tensor<T, 4> getTimeEvolutionMatrix(const e::MatrixX<T>& bond, T dt, unsigned localDim);