From d810a46803d8a4dbd8e755c49f4175caa9746dc6 Mon Sep 17 00:00:00 2001
From: ferrari <maxence.ferrari@gmail.com>
Date: Tue, 29 Nov 2022 16:07:38 +0100
Subject: [PATCH] Fix visual glitch

---
 ipi_extract.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/ipi_extract.py b/ipi_extract.py
index 04ab46f..22491af 100644
--- a/ipi_extract.py
+++ b/ipi_extract.py
@@ -311,11 +311,7 @@ class Callback(object):
             self.curr_vert[self.curr] = 0
             self.cursor[self.curr].linev[0].set_linestyle('--')
             self.cursor[self.curr].linev[self.cursor[self.curr].num_cur].set_linestyle('--')
-            for i in range(4):
-                if i < 2:
-                    ax_group[i].set_visible(False)
-                else:
-                    ax_group[i].set_visible(False)
+            ax_group.set_visible(False)
         else:
             self.curr_ind[self.curr] = np.argmax(mpos / FSSR == self.offset[:, 0])
             c = np.array([0, 0, 0, 0.])
@@ -537,6 +533,10 @@ class Callback(object):
         self.df[self.offset[self.curr_ind[self.curr], 0]] = EMLN.copy()
         self._set_label()
         self._set_visible()
+        self.curr_vert[self.curr] = 0
+        self.cursor[self.curr].linev[0].set_linestyle('--')
+        self.cursor[self.curr].linev[self.cursor[self.curr].num_cur].set_linestyle('--')
+        self.view_data[self.curr].set_visible(False)
         plt.draw()
 
     def reset(self, song, sr, song_resample):
-- 
GitLab