From 0a8fbc143188ceaec2158867520c243ac7fbeb3a Mon Sep 17 00:00:00 2001
From: ferrari <maxence.ferrari@gmail.com>
Date: Thu, 4 Mar 2021 14:41:40 +0100
Subject: [PATCH] Fix reset

---
 ipi_extract.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ipi_extract.py b/ipi_extract.py
index 14a7f85..bfe28d3 100644
--- a/ipi_extract.py
+++ b/ipi_extract.py
@@ -441,16 +441,18 @@ class Callback(object):
     def reset(self, song, sr, song_resample):
         self.p = 0
         self.df = dict()
-        self.line.set_ydata(song_resample[:FSSR * 20])
         self.song = song
         self.song_resample = song_resample
+        self._shit()
         self.sr = sr
         self.curr = 0  # current view selected
         self.offset = np.zeros((0, 2))
         self.scat.set_offsets(self.offset)
+        self.scat.set_color([[0, 0, 0, 1]])
         self.curr_ind = 3 * [None]  # Ind of click for each plot
         self.curr_vert = 3 * [0]  # Current vertical line of sig/spec for each  plot
         for i in range(3):
+            self.view_data[i][0][0].set_ydata(np.zeros(int(20e-3 * sr)))
             self.view_data[i][1][0].set_clim(2000,2100)
         for i in range(3):
             self._set_label(i, EMLN)
-- 
GitLab