Add individual number input and key control

IPI_bombyx.py

@@ -135,7 +135,7 @@ if __name__ == '__main__':
     parser.add_argument("--channel", type=int, default=0, help="Sound channel to be analysed. Indices start from 0.")
     parser.add_argument("--nb_ind", type=int, default=1, help="Maximum number of individual for each track. -1 for no limit")
     parser.add_argument("--erase", action='store_true', help="If out file exist and this option is not given,"
-                                                             " the computation will be halted")
+                                                             " the previous output will be read and appended")
     parser.add_argument("--wd", type=str, default='/nfs/NASDELL/SABIOD/SITE/BOMBYX/', help='Path to root dir'
                                                                                            ' containing Bombyx files')
     parser.add_argument("--done_file", type=str,

ipi_extract.py

@@ -259,8 +259,11 @@ class Callback(object):
         self.cursor = None
         self.f_cursor = None
         self.reset_b = None
+        self.r_button = None
+        self.ind_b = None
         self.spec_b = None
         self.nfft = 128
+        self.ind_select = False
 
     def shift_left(self, event):
         self.p = max(0, self.p - FSSR*13)
@@ -347,6 +350,10 @@ class Callback(object):
                 else:
                     self.view_data[self.curr][3][1].set_xdata((row['ipi_ceps_man'], row['ipi_ceps_man']))
                     self.view_data[self.curr][3][1].set_visible(True)
+            self.ind_select = False
+            self.ind_b.color = '0.85'
+            self.ind_b.howercolor = '0.95'
+            self.ind_b.label.set_text(f'Current individual:\n{self.df[mpos/FSSR]["ind_number"]}')
             click = self.song[max(int(mpos*self.sr/FSSR-10e-3*self.sr),0):int(mpos*self.sr/FSSR+10e-3*self.sr)]
             if len(click) != 2*int(10e-3*self.sr):
                 np.pad(click, (0, 2*int(10e-3*self.sr) - len(click)), mode='constant')
@@ -401,6 +408,30 @@ class Callback(object):
         self.reset_b.label.set_c('rgb'[self.curr])
         plt.draw()
 
+    def toggle_ind(self, event):
+        if not len(self.offset):
+            return
+        self.ind_select = not self.ind_select
+        self.ind_b.color = 'limegreen' if self.ind_select else '0.85'
+        self.ind_b.hovercolor = 'lime' if self.ind_select else '0.95'
+        if self.ind_select:
+            self.df[self.offset[self.curr_ind[self.curr], 0]]['ind_number'] = ''
+        self.ind_b.label.set_text(f'Current individual:\n')
+        plt.draw()
+
+    def key_pressed(self, event):
+        if self.ind_select:
+            row = self.df[self.offset[self.curr_ind[self.curr],0]]
+            row['ind_number'] = row['ind_number'] + event.key
+            self.ind_b.label.set_text(f'Current individual:\n{row["ind_number"]}')
+            plt.draw()
+        else:
+            if event.key in '012':
+                self.change_curr(event.key)
+                self.r_button.set_active(int(event.key))
+                pass
+
+
     def play(self, event):
         sound = (norm(self.song_resample[self.p:self.p+FSSR*20])*(2**15-1)).astype(np.int16)
         try:
@@ -452,7 +483,6 @@ class Callback(object):
             self.spec_b['minus_res'].label.set_text(f'{self.nfft//2}\nbins')
         plt.draw()
 
-
     def _set_label(self, ind=None, dic=None):
         if ind is None:
             ind = self.curr
@@ -489,6 +519,7 @@ class Callback(object):
             self.sr = sr
             sr_update = True
         self.change_curr('0')  # reset current view to 0
+        self.r_button.set_active(0)
         self.offset = np.zeros((0, 2))
         self.scat.set_offsets(self.offset)
         self.scat.set_color([[0, 0, 0, 1]])
@@ -506,6 +537,10 @@ class Callback(object):
         for i in range(3):
             self._set_label(i, EMLN)
             self._set_visible(i)
+        self.ind_select = False
+        self.ind_b.color = '0.85'
+        self.ind_b.howercolor = '0.95'
+        self.ind_b.label.set_text(f'Current individual:\nnan')
         plt.draw()
 
 
@@ -523,7 +558,7 @@ def init(in_path, channel, low=2e3, high=20e3):
     full_sig.set_ylim(-lim, lim)
     full_sig.set_yticks([])
     callback.f_cursor = Cursor(full_sig, horizOn=False, useblit=True, c='r')
-    cid = fig.canvas.mpl_connect('button_release_event', callback.on_clicked)
+    cid1 = fig.canvas.mpl_connect('button_release_event', callback.on_clicked)
 
     b_left_ax = plt.subplot(gs[:2, 0])
     b_right_ax = plt.subplot(gs[:2, -1])
@@ -538,6 +573,10 @@ def init(in_path, channel, low=2e3, high=20e3):
     r_button.on_clicked(callback.change_curr)
     for i, c in enumerate('rgb'):
         r_button.labels[i].set_c(c)
+    callback.r_button = r_button
+    for v in "home,back,forward,pan,zoom,save,quit,grid,yscale,xscale,all_axes".split(','):
+        plt.rcParams[f'keymap.{v}'] = [] # disable default shortcut but fullsreen
+    cid2 = fig.canvas.mpl_connect('key_press_event', callback.key_pressed)
     # c_button_ax = plt.subplot(gs[10,3:6])
     # c_button = CheckButtons(c_button_ax, [f'Save {i}' for i in range(3)], [False for i in range(3)])
 
@@ -573,6 +612,11 @@ def init(in_path, channel, low=2e3, high=20e3):
     reset_b.on_clicked(callback.reset_curr)
     callback.reset_b = reset_b
 
+    ind_b_ax = plt.subplot(gs[-1,6:8])
+    ind_b = Button(ind_b_ax, 'Current individual:\nnan')
+    ind_b.on_clicked(callback.toggle_ind)
+    callback.ind_b = ind_b
+
     freq_p_b_ax = plt.subplot(gs[2, 0])
     freq_m_b_ax = plt.subplot(gs[3, 0])
     freq_res_p_b_ax = plt.subplot(gs[4, 0])
@@ -641,7 +685,7 @@ def init(in_path, channel, low=2e3, high=20e3):
     callback.view_data = data_view
     return {'callback': callback, 'fig': fig, 'gridspec': gs, 'buttons':
         {'b_left': b_left, 'b_right': b_right, 'play_b': play_b, 'resize_b': resize_b, 'r_button': r_button,
-         'fs_click': cid, 'reset_b': reset_b ,
+        'ind_b': ind_b, 'fs_click': cid1, 'key_pressed': cid2, 'reset_b': reset_b ,
          'spec_button': spec_button}}  # Needed to keep the callbacks alive