From f67479535890a71804062a7e4e11b7a74cdcff8f Mon Sep 17 00:00:00 2001
From: Kazuyoshi Yoshimi <k-yoshimi@issp.u-tokyo.ac.jp>
Date: Thu, 18 Apr 2024 01:53:36 +0900
Subject: [PATCH] update te

---
 src/readdef.c | 61 +++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 45 insertions(+), 16 deletions(-)

diff --git a/src/readdef.c b/src/readdef.c
index c01ac62e..1980d41e 100644
--- a/src/readdef.c
+++ b/src/readdef.c
@@ -1861,24 +1861,53 @@ int ReadDefFileIdxPara(
             X->NTEInterAll[idx] = icnt_interall;
             X->NTEInterAllDiagonal[idx] = icnt_diagonal;
             X->NTEInterAllOffDiagonal[idx] = icnt_interall - icnt_diagonal;
+
             //Diagonal -> OffDiagonal -> search pair -> hermite
-            if (GetDiagonalInterAll(X->TEInterAll[idx], X->ParaTEInterAll[idx], X->NTEInterAll[idx], X->TEInterAllDiagonal[idx], X->ParaTEInterAllDiagonal[idx],
-                    X->TEInterAllOffDiagonal[idx], X->ParaTEInterAllOffDiagonal[idx], X->TEChemi[idx], X->SpinTEChemi[idx], X->ParaTEChemi[idx], &X->NTEChemi[idx], X->iCalcModel) != 0)
-            {
-              fclose(fp);
-              return (-1);
-            }
+//            if (GetDiagonalInterAll(X->TEInterAll[idx], X->ParaTEInterAll[idx], X->NTEInterAll[idx], X->TEInterAllDiagonal[idx], X->ParaTEInterAllDiagonal[idx],
+//                    X->TEInterAllOffDiagonal[idx], X->ParaTEInterAllOffDiagonal[idx], X->TEChemi[idx], X->SpinTEChemi[idx], X->ParaTEChemi[idx], &X->NTEChemi[idx], X->iCalcModel) != 0)
+//            {
+//              fclose(fp);
+//              return (-1);
+//            }
+//
+//            if(CheckInterAllHermite(
+//                    X->TEInterAll[idx], X->ParaTEInterAll[idx],
+//                    X->TEInterAllOffDiagonal[idx], X->ParaTEInterAllOffDiagonal[idx],
+//                    X->NTEInterAllOffDiagonal[idx], X->iCalcModel
+//            )!=0) {
+//              fprintf(stdoutMPI, "%s", cErrNonHermiteInterAllForAll);
+//              fclose(fp);
+//              return (-1);
+//            }
+//            idx++;
+//          }
+
+              if (GetDiagonalInterAll_simple(X->TEInterAll[idx], X->ParaTEInterAll[idx], X->NTEInterAll[idx], X->TEInterAllDiagonal[idx], X->ParaTEInterAllDiagonal[idx],
+                                      X->TEInterAllOffDiagonal[idx], X->ParaTEInterAllOffDiagonal[idx], X->TEChemi[idx], X->SpinTEChemi[idx], X->ParaTEChemi[idx], &X->NTEChemi[idx], X->iCalcModel) != 0)
+              {
+                  fclose(fp);
+                  return (-1);
+              }
 
-            if(CheckInterAllHermite(
-                    X->TEInterAll[idx], X->ParaTEInterAll[idx],
-                    X->TEInterAllOffDiagonal[idx], X->ParaTEInterAllOffDiagonal[idx],
-                    X->NTEInterAllOffDiagonal[idx], X->iCalcModel
-            )!=0) {
-              fprintf(stdoutMPI, "%s", cErrNonHermiteInterAllForAll);
-              fclose(fp);
-              return (-1);
-            }
-            idx++;
+              if(CheckInterAllHermite_simple(
+                      X->TEInterAll[idx], X->ParaTEInterAll[idx],
+                      X->TEInterAllOffDiagonal[idx], X->ParaTEInterAllOffDiagonal[idx],
+                      X->NTEInterAllOffDiagonal[idx], X->iCalcModel
+              )!=0) {
+                  fprintf(stdoutMPI, "%s", cErrNonHermiteInterAllForAll);
+                  fclose(fp);
+                  return (-1);
+              }
+
+                if(ArrangeInterAllOffDiagonal(
+                        X->NTEInterAllOffDiagonal[idx],
+                        X->TEInterAllOffDiagonal[idx], X->ParaTEInterAllOffDiagonal[idx],
+                        X->iCalcModel
+                )!=0){
+                    fclose(fp);
+                    return(-1);
+                }
+              idx++;
           }
 
           if(idx!=X->NTETimeSteps){