Browse Source

display2d center_cut minor improvement

vogdb 5 years ago
parent
commit
fdb8c87eff
1 changed files with 7 additions and 7 deletions
  1. 7 7
      epileptor/display_2d.py

+ 7 - 7
epileptor/display_2d.py

@@ -198,11 +198,11 @@ def center_cut_screenshot(dt, t_dt, state, data_filename):
     rect = patches.Rectangle((40, 0), 5, 79, linewidth=2, edgecolor='#CD5C5C', clip_on=False, fill=False)
     ax.add_patch(rect)
 
-    rect = patches.Rectangle((-5, 63), 2, 16, clip_on=False, facecolor='000')
+    rect = patches.Rectangle((-5, 63), 2, 14, clip_on=False, facecolor='000')
     ax.add_patch(rect)
     ax.text(-10, 69, '1mm', color='000', fontsize=14, rotation=90)
 
-    rect = patches.Rectangle((0, 81), 16, 2, clip_on=False, facecolor='000')
+    rect = patches.Rectangle((0, 81), 14, 2, clip_on=False, facecolor='000')
     ax.add_patch(rect)
     ax.text(3, 87, '1mm', color='000', fontsize=14)
 
@@ -217,10 +217,8 @@ def center_cut_t_evolution(dt, state, data_filename):
     values = state.values
     # Those values should be set manually. Now they are set for 2019-04-04_19.16
     # 32800 - 33100, 51380 - 52000
-    # dt_begin = int(63300 / dt)
-    # dt_end = int(63500 / dt)
-    dt_begin = int(0 / dt)
-    dt_end = int(400 / dt)
+    dt_begin = int(63300 / dt)
+    dt_end = int(63700 / dt)
     _, y_nh, x_nh = values.shape
     x = x_nh // 2
     center_cut = values[dt_begin:dt_end, :, x:(x + 5)]
@@ -245,7 +243,9 @@ def center_cut_t_evolution(dt, state, data_filename):
     )
     ticks = [int(np.min(center_cut)), int(np.max(center_cut))]
     cbar.set_ticks(ticks)
-    cbar.set_ticklabels([str(tick) + 'mV' for tick in ticks])
+    cbar_labels = [str(tick) for tick in ticks]
+    cbar_labels[-1] += 'mV'
+    cbar.set_ticklabels(cbar_labels)
 
     title = 'cut_{}_at{}s_{}'.format(state.name, int(dt_begin / 1000), data_filename)
     fig_fname = '{}.{}'.format(title, img_ext)