From 87d1f512d7e584a3f7be70fd7d68d4d242c6f803 Mon Sep 17 00:00:00 2001
From: Valentin Bruch <valentin.bruch@rwth-aachen.de>
Date: Sat, 18 Feb 2023 11:09:55 +0100
Subject: [PATCH] option improved_initial_conditions also in gen_pulse_data.py

---
 final_plots.py    | 16 ++++++++++++++++
 gen_data.py       |  2 +-
 gen_pulse_data.py |  6 +++++-
 3 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/final_plots.py b/final_plots.py
index 95ce3b3..94533c7 100644
--- a/final_plots.py
+++ b/final_plots.py
@@ -401,6 +401,22 @@ def export_interp(
             results[f"gdc_{method}_{order}{suffix}_ispline_old"] = bisplev(y_arr, x_arr, idc_tck_old, dy=1)
         except:
             settings.logger.exception(f"in spline interpolation of Idc for {order}{suffix} {method} (old)")
+        try:
+            results[f"idc_{method}_{order}{suffix}_old"] = griddata(
+                    xy_data,
+                    g_data.dc_current,
+                    mesh,
+                    method="cubic")
+        except:
+            settings.logger.exception(f"in griddata interpolation of Idc for {order}{suffix} {method} (old)")
+        try:
+            results[f"iac_{method}_{order}{suffix}_old"] = griddata(
+                    xy_data,
+                    2*g_data.ac_current_abs,
+                    mesh,
+                    method="cubic")
+        except:
+            settings.logger.exception(f"in griddata interpolation of Iac for {order}{suffix} {method} (old)")
         if has_current:
             try:
                 results[f"idc_{method}_{order}{suffix}"] = griddata(
diff --git a/gen_data.py b/gen_data.py
index be69e3e..da9b9ad 100644
--- a/gen_data.py
+++ b/gen_data.py
@@ -244,7 +244,7 @@ python -m frtrg_kondo.gen_data \\
             type=int, default=0,
             help = "Set initial condition for gammaL to 0")
     method_group.add_argument("--improved_initial_conditions", metavar="bool",
-            type=int, default=0,
+            type=int, default=1,
             help = "Set initial condition for dgammaL/dE to linear response estimate")
     method_group.add_argument("--include_Ga", metavar="int", type=int,
             default=1, choices=(0,1),
diff --git a/gen_pulse_data.py b/gen_pulse_data.py
index 3fca9e9..54413d2 100644
--- a/gen_pulse_data.py
+++ b/gen_pulse_data.py
@@ -67,8 +67,12 @@ def main():
     method_group = parser.add_argument_group(title="Method")
     method_group.add_argument("--method", type=str, required=True, choices=('J', 'mu'),
             help = "J: include all time dependence in coupling by unitary transformation.\nmu: describe time dependence by Floquet matrix for chemical potentials.")
-    method_group.add_argument("--simplified_initial_conditions", metavar="bool", type=bool, default=False,
+    method_group.add_argument("--simplified_initial_conditions", metavar="bool",
+            type=int, default=0,
             help = "Set initial condition for gammaL to 0")
+    method_group.add_argument("--improved_initial_conditions", metavar="bool",
+            type=int, default=1,
+            help = "Set initial condition for dgammaL/dE to linear response estimate")
     method_group.add_argument("--include_Ga", metavar="bool", type=bool, default=False,
             help = "include vertex parameter Ga in RG equations")
     method_group.add_argument("--solve_integral_exactly", metavar="bool", type=bool, default=False,
-- 
GitLab