tbasis figure updated

Author: ccluri

figures/kCSD_properties/figure_Tbasis.py

@@ -18,7 +18,7 @@
 
 def set_axis(ax, letter=None):
     ax.text(
-        -0.05,
+        -0.15,
         1.05,
         letter,
         fontsize=18,
@@ -31,19 +31,26 @@ def make_subplot(ax, true_csd, est_csd, estm_x, title=None, ele_pos=None,
                  xlabel=False, ylabel=False, letter='', t_max=None):
 
     x = np.linspace(0, 1, 100)
-    ax.plot(x, true_csd, ':', label='TrueCSD')
-    ax.plot(estm_x, est_csd, '--', label='kCSD')
-    ax.plot(ele_pos, np.zeros(len(ele_pos)), 'ko', label='Electrodes')
-    ax.legend(fontsize=10)
+    l1 = ax.plot(x, true_csd, label='True CSD', lw=2.)
+    l2 = ax.plot(estm_x, est_csd, label='kCSD', lw=2.)
+    s1 = ax.scatter(ele_pos, np.zeros(len(ele_pos)), 13, 'k', label='Electrodes')
+    #ax.legend(fontsize=10)
     ax.set_xlim([0, 1])
     if xlabel:
-        ax.set_xlabel('Depth (mm)', fontsize=15)
+        ax.set_xlabel('Depth ($mm$)')
     if ylabel:
-        ax.set_ylabel('CSD (mA/mm)', fontsize=15)
+        ax.set_ylabel('CSD ($mA/mm$)')
     if title is not None:
-        ax.set_title(title, fontsize=15)
+        ax.set_title(title)
+    if np.max(est_csd) < 1.2:
+        ax.set_ylim(-0.2, 1.2)
+    elif np.max(est_csd) > 500:
+        ax.set_yticks([-5000, 0, 5000])
     ax.set_xticks([0, 0.5, 1])
     set_axis(ax, letter=letter)
+    # ax.legend(frameon=False, loc='upper center', ncol=3)
+    ax.spines['right'].set_visible(False)
+    ax.spines['top'].set_visible(False)
     return ax
 
 
@@ -58,10 +65,10 @@ def generate_figure(R, MU, N_SRC, TRUE_CSD_XLIMS, TOTAL_ELE, SAVE_PATH,
                                                             ELE_LIMS,
                                                             noise=noise)
 
-    fig = plt.figure(figsize=(12, 12))
+    fig = plt.figure(figsize=(15, 12))
     widths = [1, 1, 1]
     heights = [1, 1, 1]
-    gs = gridspec.GridSpec(3, 3, height_ratios=heights, width_ratios=widths)
+    gs = gridspec.GridSpec(3, 3, height_ratios=heights, width_ratios=widths, hspace=0.45, wspace=0.3)
 
     ax = fig.add_subplot(gs[0, 0])
     xmin = 0
@@ -71,7 +78,7 @@ def generate_figure(R, MU, N_SRC, TRUE_CSD_XLIMS, TOTAL_ELE, SAVE_PATH,
                             sigma=0.3, gdx=0.01, ext_x=ext_x, xmin=xmin,
                             xmax=xmax, method=method, Rs=Rs, lambdas=lambdas)
     make_subplot(ax, true_csd, obj.values('CSD'), obj.estm_x, ele_pos=ele_pos,
-                 title='Basis lims = [0, 1]', xlabel=False, ylabel=True,
+                 title='Basis limits = [0, 1]', xlabel=False, ylabel=True,
                  letter='A')
 
     ax = fig.add_subplot(gs[0, 1])
@@ -82,7 +89,7 @@ def generate_figure(R, MU, N_SRC, TRUE_CSD_XLIMS, TOTAL_ELE, SAVE_PATH,
                             sigma=0.3, gdx=0.01, ext_x=ext_x, xmin=xmin,
                             xmax=xmax, method=method, Rs=Rs, lambdas=lambdas)
     make_subplot(ax, true_csd, obj.values('CSD'), obj.estm_x, ele_pos=ele_pos,
-                 title='Basis lims = [0, 0.5]', xlabel=False, ylabel=False,
+                 title='Basis limits = [0, 0.5]', xlabel=False, ylabel=False,
                  letter='B')
 
     ax = fig.add_subplot(gs[0, 2])
@@ -93,7 +100,7 @@ def generate_figure(R, MU, N_SRC, TRUE_CSD_XLIMS, TOTAL_ELE, SAVE_PATH,
                             sigma=0.3, gdx=0.01, ext_x=ext_x, xmin=xmin,
                             xmax=xmax, method=method, Rs=Rs, lambdas=lambdas)
     make_subplot(ax, true_csd, obj.values('CSD'), obj.estm_x, ele_pos=ele_pos,
-                 title='Basis lims = [0.5, 1]', xlabel=False, ylabel=False,
+                 title='Basis limits = [0.5, 1]', xlabel=False, ylabel=False,
                  letter='C')
 
     ELE_LIMS = [0, 0.5]
@@ -164,10 +171,12 @@ def generate_figure(R, MU, N_SRC, TRUE_CSD_XLIMS, TOTAL_ELE, SAVE_PATH,
     obj = tb.modified_bases(val, pots, ele_pos, N_SRC, h=0.25,
                             sigma=0.3, gdx=0.01, ext_x=ext_x, xmin=xmin,
                             xmax=xmax, method=method, Rs=Rs, lambdas=lambdas)
-    make_subplot(ax, true_csd, obj.values('CSD'), obj.estm_x, ele_pos=ele_pos,
-                 title=None, xlabel=True, ylabel=False, letter='I')
-
-    plt.tight_layout()
+    ax = make_subplot(ax, true_csd, obj.values('CSD'), obj.estm_x, ele_pos=ele_pos,
+                      title=None, xlabel=True, ylabel=False, letter='I')
+    handles, labels = ax.get_legend_handles_labels()
+    fig.legend(handles, labels, loc='lower center', ncol=3, frameon=False)
+    
+    #plt.tight_layout()
     fig.savefig(os.path.join(SAVE_PATH, 'targeted_basis_' + method +
                              '_noise_' + str(noise) + '.png'), dpi=300)
     plt.show()

figures/kCSD_properties/figure_properties.py

@@ -10,7 +10,7 @@
     'axes.labelsize': 15,
     'axes.titlesize': 20,
     'axes.titlepad' : 30,
-    'legend.fontsize': 12,
+    'legend.fontsize': 15,
     # 'figure.subplot.wspace': 0.4,
     # 'figure.subplot.hspace': 0.4,
     # 'figure.subplot.left': 0.1,