From 94cfdd7f66a1fc56edb24406df185e906782dada Mon Sep 17 00:00:00 2001
From: Fabian Joswig <fabian.joswig@ed.ac.uk>
Date: Mon, 27 Sep 2021 16:54:45 +0100
Subject: [PATCH] plateau feature implemented in Corr.show

---
 pyerrors/correlators.py | 9 ++++++++-
 pyerrors/pyerrors.py    | 6 +++---
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/pyerrors/correlators.py b/pyerrors/correlators.py
index c7e18563..9d43faa3 100644
--- a/pyerrors/correlators.py
+++ b/pyerrors/correlators.py
@@ -286,7 +286,7 @@ class Corr:
     #quick and dirty plotting function to view Correlator inside Jupyter
     #If one would not want to import pyplot, this could easily be replaced by a call to pe.plot_corrs
     #This might be a bit more flexible later
-    def show(self, x_range=None, comp=None, logscale=False, save=None):
+    def show(self, x_range=None, comp=None, logscale=False, plateau=None, save=None):
         """Plots the correlator, uses tag as label if available.
 
         Parameters
@@ -322,6 +322,13 @@ class Corr:
             else:
                 raise Exception('comp must be a correlator or a list of correlators.')
 
+        if plateau:
+            if isinstance(plateau, Obs):
+                ax1.axhline(y=plateau.value, linewidth=2, color=plt.rcParams['text.color'], alpha=0.6, marker=',', ls='--', label='Plateau')
+                ax1.axhspan(plateau.value - plateau.dvalue, plateau.value + plateau.dvalue, alpha=0.25, color=plt.rcParams['text.color'], ls='-')
+            else:
+                raise Exception('plateau must be an Obs')
+
         ax1.set_xlabel(r'$x_0 / a$')
         ax1.set_xlim([x_range[0] - 0.5, x_range[1] + 0.5])
 
diff --git a/pyerrors/pyerrors.py b/pyerrors/pyerrors.py
index f9bbb946..5803eab1 100644
--- a/pyerrors/pyerrors.py
+++ b/pyerrors/pyerrors.py
@@ -356,7 +356,7 @@ class Obs:
             plt.ylabel('tauint')
             length = int(len(self.e_n_tauint[e_name]))
             plt.errorbar(np.arange(length), self.e_n_tauint[e_name][:], yerr=self.e_n_dtauint[e_name][:], linewidth=1, capsize=2)
-            plt.axvline(x=self.e_windowsize[e_name], color='r', alpha=0.25, marker=',')
+            plt.axvline(x=self.e_windowsize[e_name], color='r', alpha=0.25, marker=',', ls='--')
             if self.tau_exp[e_name] > 0:
                 base = self.e_n_tauint[e_name][self.e_windowsize[e_name]]
                 x_help = np.arange(2 * self.tau_exp[e_name])
@@ -1180,11 +1180,11 @@ def plot_corrs(observables, **kwargs):
     if 'plateau' in kwargs:
         plateau = kwargs.get('plateau')
         if isinstance(plateau, Obs):
-            plt.axhline(y=plateau.value, linewidth=2, color='k', alpha=0.6, label='Plateau', marker=',')
+            plt.axhline(y=plateau.value, linewidth=2, color='k', alpha=0.6, label='Plateau', marker=',', ls='--')
             plt.axhspan(plateau.value - plateau.dvalue, plateau.value + plateau.dvalue, alpha=0.25, color='k')
         elif isinstance(plateau, list):
             for i in range(len(plateau)):
-                plt.axhline(y=plateau[i].value, linewidth=2, color='C' + str(i), alpha=0.6, label='Plateau' + str(i + 1), marker=',')
+                plt.axhline(y=plateau[i].value, linewidth=2, color='C' + str(i), alpha=0.6, label='Plateau' + str(i + 1), marker=',', ls='--')
                 plt.axhspan(plateau[i].value - plateau[i].dvalue, plateau[i].value + plateau[i].dvalue,
                             color='C' + str(i), alpha=0.25)
         else: