From b1a5771d3ad0bb756940888eb39c9db7164f4e29 Mon Sep 17 00:00:00 2001
From: fmheu <fhchl@users.noreply.github.com>
Date: Wed, 27 Feb 2019 14:49:51 +0000
Subject: [PATCH] fix: labels work properly in plots

---
 response.py | 32 ++++++++++++++++++++++++++------
 1 file changed, 26 insertions(+), 6 deletions(-)

diff --git a/response.py b/response.py
index 93a8a37..c030206 100644
--- a/response.py
+++ b/response.py
@@ -384,6 +384,9 @@ def plot_magnitude(
         if dblim is not None:
             ax.set_ylim(dblim)
 
+        if label is not None:
+            ax.legend()
+
         return fig
 
     def plot_phase(
@@ -421,7 +424,7 @@ def plot_phase(
             np.unwrap(np.angle(freq_plotready)) if unwrap else np.angle(freq_plotready)
         )
 
-        ax.semilogx(self.freqs, phase, **plot_kw)
+        ax.semilogx(self.freqs, phase, label=label, **plot_kw)
         ax.set_xlabel("Frequency [Hz]")
         ax.set_ylabel("Phase [rad]")
         ax.set_title("Phase response")
@@ -433,10 +436,20 @@ def plot_phase(
         if ylim:
             ax.set_ylim(ylim)
 
+        if label is not None:
+            ax.legend()
+
         return fig
 
     def plot_time(
-        self, use_ax=None, slce=None, tlim=None, ylim=None, plot_kw={}, **fig_kw
+        self,
+        use_ax=None,
+        slce=None,
+        tlim=None,
+        ylim=None,
+        label=None,
+        plot_kw={},
+        **fig_kw,
     ):
         """Plot time response."""
         if use_ax is None:
@@ -458,16 +471,20 @@ def plot_time(
             (self.nt, -1)
         )
 
-        ax.plot(self.times, time_plotready, **plot_kw)
+        ax.plot(self.times, time_plotready, label=label, **plot_kw)
         ax.set_xlabel("Time [s]")
         ax.set_ylabel("")
         ax.set_title("Time response")
         ax.grid(True)
+
         if tlim:
             ax.set_xlim(tlim)
         if ylim:
             ax.set_ylim(ylim)
 
+        if label is not None:
+            ax.legend()
+
         return fig
 
     def plot_group_delay(
@@ -505,17 +522,22 @@ def plot_group_delay(
         # TODO: use scipy.signal.group_delay here as below has problem at larger delays
         grpd = -np.gradient(np.unwrap(np.angle(freq_plotready)), df, axis=0)
 
-        ax.semilogx(self.freqs, grpd, **plot_kw)
+        ax.semilogx(self.freqs, grpd, label=label, **plot_kw)
         ax.set_xlabel("Frequency [Hz]")
         ax.set_ylabel("Delay [s]")
         ax.set_title("Group Delay")
         ax.grid(True)
+
         if flim is None:
             flim = (10, self.fs / 2)
         ax.set_xlim(flim)
+
         if ylim:
             ax.set_ylim(ylim)
 
+        if label is not None:
+            ax.legend()
+
         return fig
 
     def plot_power_in_bands(
@@ -669,8 +691,6 @@ def timecrop(self, start, end):
         else:
             _, i_end = find_nearest(self.times, end)
 
-        print(i_start, i_end)
-
         h = self.in_time[..., i_start:i_end]
 
         new_response = self.from_time(self.fs, h)