display2d center_cut minor improvement

epileptor/display_2d.py

@@ -191,20 +191,22 @@ def display_state_at_points(dt, state, data_filename, point_index_list, line_lis
 
 
 def center_cut_screenshot(dt, t_dt, state, data_filename):
-    fig = plt.gcf()
+    fig = plt.figure(figsize=(5, 5))
     ax = plt.gca()
     ax.axis('off')
     ax.imshow(state.values[t_dt, :, :], cmap='jet')
-    rect = patches.Rectangle((40, 0), 5, 79, linewidth=2, edgecolor='#CD5C5C', clip_on=False, fill=False)
+    rect = patches.Rectangle((40, 0), 5, 79, linewidth=4, edgecolor='#FF0000', clip_on=False, fill=False)
     ax.add_patch(rect)
 
     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)
+    ax.text(-12, 67, '1mm', color='000', fontsize=18, rotation=90)
 
     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)
+    ax.text(0, 89, '1mm', color='000', fontsize=18)
+    # ax.set_title(r'$V$', fontdict=dict(fontsize=26, fontweight='bold'), y=-0.2, clip_on=False)
+    ax.text(-10, -5, 'A', color='000', fontsize=32, fontweight='bold')
 
     title = 'cut_H_{}_at{}s_{}'.format(state.name, util.dt_to_sec(t_dt, dt), data_filename)
     fig_fname = '{}.{}'.format(title, img_ext)
@@ -232,9 +234,11 @@ def center_cut_t_evolution(dt, state, data_filename):
     ax.add_patch(rect)
     ax.text(160, 115, '0.1s', color='000', fontsize=10)
 
-    rect = patches.Rectangle((-35, 65), 15, 15, clip_on=False, facecolor='000')
+    rect = patches.Rectangle((-50, 1), 35, 77, linewidth=2, edgecolor='#FF0000', clip_on=False, fill=False)
+    # rect = patches.Rectangle((-35, 65), 15, 15, clip_on=False, facecolor='000')
     ax.add_patch(rect)
-    ax.text(-100, 65, '1mm', color='000', fontsize=10, rotation=90)
+    # ax.text(-100, 65, '1mm', color='000', fontsize=10, rotation=90)
+    ax.text(-80, -15, 'B', color='000', fontsize=16, fontweight='bold')
 
     cbar_ax = fig.add_axes([0.5, 0.25, .4, .1], xticks=[], yticks=[], frameon=False)
     cbar = plt.colorbar(
@@ -271,9 +275,10 @@ if 'points' in points_data and len(points_data['points'][0]) > 1:
     ]
 
 # plain videos
-plain_names = plain_data['names']
-for name in plain_names:
+for name in plain_data.files:
     state = State(name, plain_data[name])
+    if state.name == 'V':
+        center_cut_t_evolution(dt, state, data_filename)
     make_animation_video(dt, state, data_filename)
     display_t_points(state, data_filename)
     if point_list is not None: