Wave tweaks, up font size to 9

Author: kburns

atmospheric_waves/plot_waves.py

@@ -60,8 +60,8 @@
 # Figures
 fig, ax = plt.subplots(nrows=2, sharex=True, figsize=(3.4, 2.8))
 fig_eig, ax_eig = plt.subplots(nrows=2, sharex=True, figsize=(3.4, 2.8))
-fig.subplots_adjust(left=0.11, bottom=0.12, right=0.97, top=0.97, hspace=0.06)
-fig_eig.subplots_adjust(left=0.11, bottom=0.12, right=0.97, top=0.97, hspace=0.06)
+fig.subplots_adjust(left=0.12, bottom=0.12, right=0.97, top=0.97, hspace=0.06)
+fig_eig.subplots_adjust(left=0.12, bottom=0.12, right=0.97, top=0.97, hspace=0.06)
 
 # Loop over kx
 for i, k1 in enumerate(ks):
@@ -78,14 +78,17 @@
     acoustic = (np.abs(ω) > ω_upper[i])
 
     # Plot frequencies
-    ax[0].plot(k[acoustic], np.abs(ω[acoustic]), color=c_acoustic, **freq_props)
-    ax[0].plot(k[f_mode], np.abs(ω[f_mode]), color=c_f_mode, **freq_props)
-    ax[0].plot(k[gwaves], np.abs(ω[gwaves]), color=c_gravity, **freq_props)
     if i == 0:
+        ax[0].plot(k[acoustic], np.abs(ω[acoustic]), color=c_acoustic, label='ac', **freq_props)
+        ax[0].plot(k[f_mode], np.abs(ω[f_mode]), color=c_f_mode, label='f', **freq_props)
+        ax[0].plot(k[gwaves], np.abs(ω[gwaves]), color=c_gravity, label='ac', **freq_props)
         ax[1].plot(k[gwaves], np.abs(1/ω[gwaves]), color=c_gravity, label='gw', **freq_props)
         ax[1].plot(k[f_mode], np.abs(1/ω[f_mode]), color=c_f_mode, label='f', **freq_props)
         ax[1].plot(k[acoustic], np.abs(1/ω[acoustic]), color=c_acoustic, label='ac', **freq_props)
     else:
+        ax[0].plot(k[acoustic], np.abs(ω[acoustic]), color=c_acoustic, **freq_props)
+        ax[0].plot(k[f_mode], np.abs(ω[f_mode]), color=c_f_mode, **freq_props)
+        ax[0].plot(k[gwaves], np.abs(ω[gwaves]), color=c_gravity, **freq_props)
         ax[1].plot(k[acoustic], np.abs(1/ω[acoustic]), color=c_acoustic,**freq_props)
         ax[1].plot(k[f_mode], np.abs(1/ω[f_mode]), color=c_f_mode, **freq_props)
         ax[1].plot(k[gwaves], np.abs(1/ω[gwaves]), color=c_gravity, **freq_props)
@@ -104,7 +107,7 @@
         # Mode properties
         colors = ['C1', 'C2', 'C4', 'C5']
         marker = 'o'
-        msc = 4
+        msc = 6
         mec = "none"
         print("   ω/N, N/ω")
         mode_props = dict(marker=marker, mec=mec, markersize=msc, alpha=0.5, zorder=3)
@@ -126,8 +129,8 @@
 
 # Frequency lines
 line_props = {'ls': 'solid', 'lw': 1, 'alpha': 0.5, 'zorder': 1}
-ax[0].plot(ks, ω_lower, color=c_gravity, **line_props)
-ax[0].plot(ks, ω_upper, color=c_acoustic, **line_props)
+ax[0].plot(ks, ω_lower, color=c_gravity, label=r'$\omega_-$', **line_props)
+ax[0].plot(ks, ω_upper, color=c_acoustic, label=r'$\omega_+$', **line_props)
 #ax[0].axhline(y=1, color='black', **line_props)
 ax[1].plot(ks, 1/ω_lower, color=c_gravity, label=r'$\omega_-$', **line_props)
 #ax[1].axhline(y=1, color='black', label=r'$N$', **line_props)
@@ -150,8 +153,8 @@
 
 ax_eig[0].set_ylabel(r'$\sqrt{\rho}w$')
 ax_eig[1].set_ylabel(r'$\sqrt{\rho}w$')
-ax_eig[1].text(0, 0.65, 'gravity waves\n'+r'($\omega \leq \omega_-$)', verticalalignment='center', multialignment='center', fontsize=8)
-ax_eig[0].text(0, 0.65, 'acoustic waves\n'+r'($\omega > \omega_+$)', verticalalignment='center', multialignment='center', fontsize=8)
+ax_eig[1].text(0, 0.65, 'gravity waves\n'+r'($\omega \leq \omega_-$)', verticalalignment='center', multialignment='center', fontsize=9)
+ax_eig[0].text(0, 0.65, 'acoustic waves\n'+r'($\omega > \omega_+$)', verticalalignment='center', multialignment='center', fontsize=9)
 ax_eig[1].set_xlabel(r'height $z$')
 ax_eig[0].set_ylim(-1.3, 1.3)
 ax_eig[1].set_ylim(-1.3, 1.3)
@@ -160,23 +163,23 @@
 ax_eig[1].yaxis.set_label_coords(-0.06, 0.5)
 
 # Legends
-legend = ax[1].legend(ncol=2, frameon=False, fontsize=6)
+legend = ax[1].legend(ncol=2, loc='upper left', frameon=False, fontsize=7, handlelength=1.5)
 legend.get_frame().set_linewidth(0.0)
 
-legend = ax_eig[1].legend(frameon=False, title=r'period $N/\omega$', loc='lower left', ncol=2, fontsize=6)
-plt.setp(legend.get_title(),fontsize=8)
+legend = ax_eig[1].legend(frameon=False, title=r'period $N/\omega$', loc='lower left', ncol=2, fontsize=7)
+plt.setp(legend.get_title(),fontsize=9)
 for line,text in zip(legend.get_lines(), legend.get_texts()):
     text.set_color(line.get_color())
-legend = ax_eig[0].legend(frameon=False, title=r'frequency $\omega/N$', loc='lower left', ncol=2, fontsize=6)
-plt.setp(legend.get_title(),fontsize=8)
+legend = ax_eig[0].legend(frameon=False, title=r'frequency $\omega/N$', loc='lower left', ncol=2, fontsize=7)
+plt.setp(legend.get_title(),fontsize=9)
 for line,text in zip(legend.get_lines(), legend.get_texts()):
     text.set_color(line.get_color())
 
 # Save
-ax[0].text(-0.06, 1.0, "(a)", transform=ax[0].transAxes, fontsize=8, va='top', ha='right')
-ax[1].text(-0.06, 1.0, "(b)", transform=ax[1].transAxes, fontsize=8, va='top', ha='right')
-ax_eig[0].text(-0.06, 1.0, "(a)", transform=ax_eig[0].transAxes, fontsize=8, va='top', ha='right')
-ax_eig[1].text(-0.06, 1.0, "(b)", transform=ax_eig[1].transAxes, fontsize=8, va='top', ha='right')
+ax[0].text(-0.06, 1.0, "(a)", transform=ax[0].transAxes, fontsize=9, va='top', ha='right')
+ax[1].text(-0.06, 1.0, "(b)", transform=ax[1].transAxes, fontsize=9, va='top', ha='right')
+ax_eig[0].text(-0.06, 1.0, "(a)", transform=ax_eig[0].transAxes, fontsize=9, va='top', ha='right')
+ax_eig[1].text(-0.06, 1.0, "(b)", transform=ax_eig[1].transAxes, fontsize=9, va='top', ha='right')
 
 fig.savefig('fig_waves_spectrum.pdf')
 fig_eig.savefig('fig_waves_eigenfunctions.pdf')

atmospheric_waves/solve_atmosphere.py

@@ -209,19 +209,19 @@
 
 # Plot structure
 fig, (ax1, ax2) = plt.subplots(nrows=2, sharex=True, figsize=(3.4, 2.8))
-fig.subplots_adjust(left=0.14, bottom=0.12, right=0.86, top=0.97, hspace=0.06)
+fig.subplots_adjust(left=0.12, bottom=0.12, right=0.86, top=0.97, hspace=0.06)
 
 ax1.plot(z, ln_T['g'], label=r'$\ln \,T$', c='C0')
 ax1.plot(z_phot, ln_T['g'][i_tau_23], marker='.', color='black')
 ax1.set_ylabel(r"$\ln \,T$")
 axr = ax1.twinx()
 axr.plot(z, ln_rho['g'], label=r'$\ln\,\rho$', linestyle='dashed', c='C1')
 axr.plot(z, ln_rho['g']+ln_T['g'], label=r'$\ln \,P$', linestyle='dashed', c='C2')
-axr.set_ylabel(r"$\ln \,\rho, \ln \,P$")
+axr.set_ylabel(r"$\ln \,\rho, \; \ln \,P$")
 ax1.set_xlim([0, Lz])
 lines1, labels1 = ax1.get_legend_handles_labels()
 lines2, labels2 = axr.get_legend_handles_labels()
-ax1.legend(lines1+lines2, labels1+labels2, loc='lower left', frameon=False, fontsize=8, handlelength=1.8)
+ax1.legend(lines1+lines2, labels1+labels2, loc='lower left', frameon=False, fontsize=9, handlelength=1.8)
 
 #ax.plot(z_diag, diagnostics['s_Cp']['g'], color='black', label=r'$s/c_P$')
 #ax.set_ylabel(r"$s/c_P$")
@@ -239,14 +239,15 @@
 ax2.set_ylabel(r'$\omega/N_{\mathrm{max}}$')
 ax2.set_xlabel(r'height $z$')
 lines1, labels1 = ax2.get_legend_handles_labels()
-legend=ax2.legend(lines1, labels1, loc='center left', frameon=False, ncol=1, fontsize=8, handlelength=1.8)
+legend=ax2.legend(lines1, labels1, loc='center left', frameon=False, ncol=1, fontsize=9, handlelength=1.8)
 legend.get_frame().set_linewidth(0.0)
 
 ax2.xaxis.set_label_coords(0.5, -0.15)
-ax1.yaxis.set_label_coords(-0.12, 0.5)
-ax2.yaxis.set_label_coords(-0.12, 0.5)
-ax1.text(-0.12, 1.0, "(a)", transform=ax1.transAxes, fontsize=8, va='top', ha='right')
-ax2.text(-0.12, 1.0, "(b)", transform=ax2.transAxes, fontsize=8, va='top', ha='right')
+ax1.yaxis.set_label_coords(-0.08, 0.5)
+axr.yaxis.set_label_coords(1.1, 0.55)
+ax2.yaxis.set_label_coords(-0.08, 0.5)
+ax1.text(-0.08, 1.0, "(a)", transform=ax1.transAxes, fontsize=9, va='top', ha='right')
+ax2.text(-0.08, 1.0, "(b)", transform=ax2.transAxes, fontsize=9, va='top', ha='right')
 fig.savefig('fig_waves_atmosphere.pdf')
 
 # Plot T-P

methods_paper.mplstyle

@@ -126,7 +126,7 @@ font.weight         : normal
 # settings for axes and ticks. Special text sizes can be defined
 # relative to font.size, using the following values: xx-small, x-small,
 # small, medium, large, x-large, xx-large, larger, or smaller
-font.size           : 8
+font.size           : 9
 font.serif          : Times New Roman #Bitstream Vera Serif, New Century Schoolbook, Century Schoolbook L, Utopia, ITC Bookman, Bookman, Nimbus Roman No9 L, Times New Roman, Times, Palatino, Charter, serif
 #font.serif : Times New Roman
 #font.sans-serif     : Roboto
@@ -209,9 +209,9 @@ mathtext.fontset : stix # Should be 'cm' (Computer Modern), 'stix',
 #axes.edgecolor      : black   # axes edge color
 axes.linewidth      : 1.0     # edge linewidth
 #axes.grid           : False   # display grid or not
-axes.titlesize      : 10   # fontsize of the axes title
+axes.titlesize      : 9   # fontsize of the axes title
 axes.titleweight    : normal
-axes.labelsize      : 8  # fontsize of the x any y labels
+axes.labelsize      : 9  # fontsize of the x any y labels
 axes.labelweight    : normal  # weight of the x and y labels
 #axes.labelcolor     : black
 #axes.axisbelow      : False   # whether axis gridlines and ticks are below
@@ -254,7 +254,7 @@ ytick.minor.width    : 1    # minor tick width in points
 #ytick.major.pad      : 4      # distance to major tick label in points
 #ytick.minor.pad      : 4      # distance to the minor tick label in points
 #ytick.color          : k      # color of the tick labels
-ytick.labelsize      : 8 # fontsize of the tick labels
+ytick.labelsize      : 9 # fontsize of the tick labels
 ytick.direction      : in     # direction: in or out
 
 
@@ -268,7 +268,7 @@ ytick.direction      : in     # direction: in or out
                                # legend, else a rectangle
 #legend.isaxes        : True
 #legend.numpoints     : 2      # the number of points in the legend line
-legend.fontsize      : 6
+legend.fontsize      : 7
 #legend.pad           : 0.0    # deprecated; the fractional whitespace inside the legend border
 #legend.borderpad     : 0.5    # border whitespace in fontsize units
 #legend.markerscale   : 1.0    # the relative size of legend markers vs. original

quasigeostrophic_flow/plot_slice.py

@@ -50,14 +50,15 @@
 x = np.array(f['scales/x/4'])
 y = np.array(f['scales/y/4'])
 vort = np.array(f['tasks/vorticity-top'][0,:,:,0])
+b = np.array(f['tasks/buoyancy-top'][0,:,:,0])
 PV   = np.array(f['tasks/PV-top'][0,:,:,0])
 
 f.close()
 
 xm, ym = plot_tools.quad_mesh(x, y)
 
-data = [vort,PV]
-label = [r'$\omega$',r'$PV$']
+data = [b,PV]
+label = [r'$b$',r'$PV$']
 cmaps = ['RdBu_r','PuOr']
 
 c_im = []
@@ -66,7 +67,7 @@
   max = 0.8*np.max(np.abs(data[i]))
   c_im.append(slice_axes[i].pcolormesh(xm, ym, data[i].T, vmin=-max, vmax=max, cmap=cmaps[i]))
 
-  slice_axes[i].axis([-40,40,-20,20])
+  #slice_axes[i].axis([-40,40,-20,20])
   if i == 0:
     plt.setp(slice_axes[i].get_xticklabels(), visible=False)
     slice_axes[i].xaxis.set_major_locator(MaxNLocator(nbins=5))
@@ -75,7 +76,7 @@
     slice_axes[i].set_xlabel(r'$x$')
     slice_axes[i].xaxis.set_major_locator(MaxNLocator(nbins=5))
     slice_axes[i].yaxis.set_major_locator(MaxNLocator(nbins=5,prune='upper'))
-  
+
   slice_axes[i].set_ylabel(r'$y$')
 
   cbar.append(fig.colorbar(c_im[i], cax=cbar_axes[i], orientation='vertical', ticks=MaxNLocator(nbins=5)))