update figsaver and plot utils

python/energy_flow_proper/figsaver.py

@@ -150,6 +150,7 @@ MPL_RC = {
     "xtick.labelsize": 5,
     "ytick.labelsize": 5,
     "figure.constrained_layout.use": True,
+    # "text.latex.preamble": r"\usepackage{mathtools}",
 }
 
 MPL_RC_POSTER = {

python/energy_flow_proper/plot_utils.py

@@ -748,3 +748,16 @@ def plot_multi_energy_overview(
     ax_sys.legend(
         # ncol=2,
     )
+
+
+@wrap_plot
+def plot_ρ(model, i, j, ax=None, **kwargs):
+    with aux.get_data(model) as data:
+        return plot_with_σ(
+            model.t,
+            EnsembleValue(
+                (data.rho_t_accum.mean[:, i, j], data.rho_t_accum.ensemble_std[:, i, j])
+            ),
+            ax=ax,
+            **kwargs,
+        )