From 9824ea029dab8327db62235c69af6d1e30cc1834 Mon Sep 17 00:00:00 2001
From: Valentin Boettcher <hiro@protagon.space>
Date: Mon, 25 Jul 2022 11:43:49 +0200
Subject: [PATCH] add common code to 11

---
 .../11_new_ho_comparison/common.py            | 23 +++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 python/energy_flow_proper/11_new_ho_comparison/common.py

diff --git a/python/energy_flow_proper/11_new_ho_comparison/common.py b/python/energy_flow_proper/11_new_ho_comparison/common.py
new file mode 100644
index 0000000..2203433
--- /dev/null
+++ b/python/energy_flow_proper/11_new_ho_comparison/common.py
@@ -0,0 +1,23 @@
+from plot_utils import *
+
+
+@wrap_plot
+def compare_models_with_analytic(ax=None):
+    for params, flow, ex_flow, keys in zip(
+        multi_params, flow_hops, exact_flows, model_keys
+    ):
+        consistency = (-1 * flow).consistency(ex_flow)
+        pu.plot_with_σ(
+            params.IntP.t,
+            -1 * flow,
+            bath=0,
+            ax=ax,
+            label=rf"$α(0)={params.SysP.g[0].sum().real:.2f}$ $ω_c={keys['wc']}$ ${consistency}\%$",
+        )
+        ax.plot(params.IntP.t, ex_flow, linestyle="dotted", color="black")
+
+        ax.set_xlabel("$τ$")
+        ax.set_ylabel("$-J$")
+
+    # ax.plot(multi_params[-1].IntP.t, flow_τ, label="Analytic", linestyle="dotted", color="black")
+    ax.legend()