import time import os import sys import cartopy.crs as ccrs import cartopy.feature as cfeature from cartopy.mpl.gridliner import LONGITUDE_FORMATTER, LATITUDE_FORMATTER import matplotlib import matplotlib.dates as mdate import matplotlib.pyplot as plt import matplotlib.ticker as mticker from matplotlib.dates import DateFormatter import numpy as np from pylab import * import matplotlib.gridspec as gridspec from matplotlib.colorbar import Colorbar matplotlib.use('Agg') class Plothovmoller: """ Contains methods for generating hovmoller diagrams of azimuthally-averaged rainfall rates from GPM/IMERG L3 data, by radial rings, compass quadrant, and/or shear quadrant. """ def __init__(self, xx, xi, yi, dpi, tfnt, cfnt, cols, ptime, invtime, doscp, userid, server): """ :param xx: 1D float array of x-axis radial values :param xi: x-dimension of plot in inches :param yi: y-dimension of plot in inches :param dpi: input dots per inch :param tfnt: title font size :param cfnt: colorbar font size :param cols: list of RGB colors from pickle file :param ptime: str - plot time axis either 'utc' or 'loc' (local) :param invtime: logical - Invert y-axis (T) or not (F) :param doscp: Logical flag for scp-ing file to remote server :param userid: User ID on remote server for scp-ing files :param server: name/address of remote server for scp-ing files """ self.xx = xx self.xinch = xi self.yinch = yi self.dpi = dpi self.tfont = tfnt self.cfont = cfnt self.colors = cols self.ptime = ptime self.invtime = invtime self.doscp = doscp self.userid = userid self.server = server def plot_hov_single(self, ytimes, array, png, title, clevs, ptype, remdir): """ Make single-panel hovmoller plots of IMERG hourly precip. Supports UTC or local time, and generates plots for both multi-day and diurnal mean results. :param ytimes: Dates/times along y-axis :param array: array of mean precip by radial bin :param png: image name for .png file :param title: text string for plot title :param clevs: contour levels for color-filled plots :param ptype: text string of plot type 'multiday' or 'diurnal' :param remdir: remote directory of server for scp-ing files """ yy = mdate.num2date(ytimes) fig, hovax = plt.subplots(1, 1, figsize=(self.xinch, self.yinch), dpi=self.dpi) cs = hovax.contourf(self.xx, yy, array, levels=clevs, colors=self.colors, extend='max') # Add title and color bar; adjust label font sizes # Vertical color bar [left, bottom, width, height] cbar_hovax = fig.add_axes([0.915, 0.11, 0.02, 0.77]) fig.colorbar(cs, cax=cbar_hovax, orientation='vertical', ticks=clevs) cbar_hovax.tick_params(labelsize=self.cfont) hovax.set_xlabel('Radial bin (km)', size=self.cfont, weight='bold') if ptype == 'multiday': fig.suptitle(title, weight='bold', fontsize=self.tfont, y=0.93) if self.ptime == 'utc': ytext = 'Date / UTC Time' else: ytext = 'Date / Local Time' hovax.set_ylabel(ytext, size=self.cfont, weight='bold') elif ptype == 'diurnal': fig.suptitle(title, weight='bold', fontsize=self.tfont, y=0.95) if self.ptime == 'utc': ytext = 'UTC Time (HHmm)' else: ytext = 'Local Time (HHmm)' hovax.set_ylabel(ytext, size=self.cfont, weight='bold') hovax.set_yticks(ytimes) hovax.set_yticklabels( [date.strftime("%H%M") for date in mdate.num2date(ytimes)]) if self.invtime: print('Using inverted time for y-axis') hovax.invert_yaxis() plt.savefig(png) if self.doscp: print(f'scp {png} to {self.server} at {remdir}/') cmd = f'scp -q {png} {self.userid}@{self.server}:{remdir}/' os.system(cmd) def plot_hov_quad(self, ytimes, array, png, title, clevs, ptype, remdir, shr=False): """ Makes 4-panel hovmoller plots of IMERG hourly precip by compass (NW/NE/SW/SE) or shear (downshear-left, downshear-right, upshear-left, upshear-right) quadrants. Supports UTC or local time, and generates plots for both multi-day and diurnal mean results. :param ytimes: Dates/times along y-axis :param array: array of mean precip by radial bin :param png: image name for .png file :param title: text string for plot title :param clevs: contour levels for color-filled plots :param ptype: text string of plot type 'multiday' or 'diurnal' :param remdir: remote directory of server for scp-ing files :param shr: Optional logical to indicate compass (False) or shear (True) relative quadrant plots """ yy = mdate.num2date(ytimes) fig = plt.figure(figsize=(11, 9), dpi=self.dpi) # NW/Downshear-left quad plot hovax = fig.add_subplot(2, 2, 1) hovax.contourf(self.xx, yy, array[0][:][:], levels=clevs, colors=self.colors, extend='max') if shr: hovax.set_xlabel('\n(a) Downshr-Left Quad Radial Bin (km)', size=self.cfont, weight='bold') else: hovax.set_xlabel('\n(a) NW Quadrant Radial Bin (km)', size=self.cfont, weight='bold') # hovax.axes.xaxis.set_visible(False) # hovax.yaxis.set_ticks_position('left') hovax.tick_params(labelleft = True, bottom = False, labelbottom = False) if self.invtime: print('Using inverted time for y-axis') hovax.invert_yaxis() ytext = date_form = None if ptype == 'multiday': date_form = DateFormatter("%m%d-%H") if self.ptime == 'utc': ytext = 'Date / UTC Time (MMdd-HH)' else: ytext = 'Date / Local Time (MMdd-HH)' elif ptype == 'diurnal': date_form = DateFormatter("%H%M") if self.ptime == 'utc': ytext = 'UTC Time (HHmm)' else: ytext = 'Local Time (HHmm)' hovax.yaxis.set_major_formatter(date_form) hovax.set_ylabel(ytext, size=self.cfont, weight='bold') # NE/Downshear-right quad plot hovax = fig.add_subplot(2, 2, 2) hovax.contourf(self.xx, yy, array[1][:][:], levels=clevs, colors=self.colors, extend='max') if shr: hovax.set_xlabel('\n(b) Downshr-Right Quad Radial Bin (km)', size=self.cfont, weight='bold') else: hovax.set_xlabel('\n(b) NE Quadrant Radial Bin (km)', size=self.cfont, weight='bold') # hovax.axes.xaxis.set_visible(False) # hovax.axes.yaxis.set_visible(False) hovax.tick_params(left= False, labelleft = False, bottom = False, labelbottom = False) # SW/Upshear-left quad plot hovax = fig.add_subplot(2, 2, 3) hovax.contourf(self.xx, yy, array[2][:][:], levels=clevs, colors=self.colors, extend='max') if shr: hovax.set_xlabel('(c) Upshr-Left Quad Radial Bin (km)', size=self.cfont, weight='bold') else: hovax.set_xlabel('(c) SW Quadrant Radial Bin (km)', size=self.cfont, weight='bold') # hovax.xaxis.set_ticks_position('bottom') # hovax.yaxis.set_ticks_position('left') hovax.tick_params(labelleft = True, labelbottom = True) if self.invtime: print('Using inverted time for y-axis') hovax.invert_yaxis() hovax.yaxis.set_major_formatter(date_form) hovax.set_ylabel(ytext, size=self.cfont, weight='bold') # SE/Upshear-right quad plot hovax = fig.add_subplot(2, 2, 4) csse = hovax.contourf(self.xx, yy, array[3][:][:], levels=clevs, colors=self.colors, extend='max') if shr: hovax.set_xlabel('(d) Upshr-Right Quad Radial Bin (km)', size=self.cfont, weight='bold') else: hovax.set_xlabel('(d) SE Quadrant Radial Bin (km)', size=self.cfont, weight='bold') # hovax.xaxis.set_ticks_position('bottom') # hovax.axes.yaxis.set_visible(False) hovax.tick_params(labelleft = False, labelbottom = True) # Add title and color bar; adjust label font sizes if ptype == 'multiday': fig.suptitle(title, weight='bold', fontsize=self.tfont, y=0.95) elif ptype == 'diurnal': fig.suptitle(title, weight='bold', fontsize=self.tfont, y=0.97) # Vertical color bar plt.subplots_adjust(left=0.09, right=0.91, bottom=0.06, top=0.91, wspace=0.07, hspace=0.14) cbar_hovax = plt.axes( [0.93, 0.08, 0.02, 0.82]) # [left, bottom, width, height] plt.colorbar(csse, cax=cbar_hovax, orientation='vertical', ticks=clevs) cbar_hovax.tick_params(labelsize=self.cfont) plt.savefig(png) if self.doscp: print(f'scp {png} to {self.server} at {remdir}/') cmd = f'scp -q {png} {self.userid}@{self.server}:{remdir}/' os.system(cmd) def plot_hovship_sgl(self, ytimes, array, mag, head, png, title, clevs, ptype, remdir): """ Make single-panel hovmoller+SHIPS shear plots of IMERG hourly precip. Supports only UTC or local time, and generates plots only for multi-day results. :param ytimes: Dates/times along y-axis :param array: array of mean precip by radial bin :param png: image name for .png file :param title: text string for plot title :param clevs: contour levels for color-filled plots :param ptype: text string of plot type 'multiday' or 'diurnal' :param remdir: remote directory of server for scp-ing files """ yy = mdate.num2date(ytimes) fig = plt.figure(figsize=(10, 10), dpi=self.dpi) gs = gridspec.GridSpec(10, 8) gs.update(left=0.12, right=0.90, bottom=0.05, top=0.95, wspace=0.1, hspace=0.1) cbar_axes = [0.92, 0.232, 0.02, 0.71] r = np.zeros(24) theta = np.zeros(24) plabels = ["", "45$^\circ$", "", "135$^\circ$", "180$^\circ$", "225$^\circ$", "270$^\circ$", "315$^\circ$"] ax = plt.subplot(gs[0:8, 0:8]) fill = ax.contourf(self.xx, yy, array, levels=clevs, colors=self.colors, extend='max') ax.set_xlabel('Radial bin (km)', size=self.cfont, weight='bold') if ptype == 'multiday': fig.suptitle(title, weight='bold', fontsize=self.tfont, y=0.98) if self.ptime == 'utc': ytext = 'Date / UTC Time' else: ytext = 'Date / Local Time' ax.set_ylabel(ytext, size=self.cfont, weight='bold') elif ptype == 'diurnal': print("Diurnal mean not supported in Hovmoller+SHIP shear plots.\n") print("Exiting script....\n") sys.exit() if self.invtime: print('Using inverted time for y-axis') ax.invert_yaxis() # New section to plot SHIPS shear polar plots beneath IMERG hovmollers. # Make [up to] 7-panels of shear hodographs # Reset to 7 if # days exceeds 7. num_days = int(len(ytimes)/24) if num_days > 7: num_days = 7 for c in range(num_days): axr = plt.subplot(gs[9, c], projection='polar') # Set polar plot in meteorological convention axr.set_theta_direction(-1) axr.set_theta_offset(np.pi / 2.0) tstart = 24 * c tstop = tstart + 24 r[0:24] = mag[tstart:tstop] theta[0:24] = head[tstart:tstop] # print(f'Day {c + 1}; r/mag \n {r}') # print(f'Day {c + 1}; theta/head: \n {theta}') theta = theta * np.pi / 180.0 for hh in range(0,24): colors = hh axr.scatter(theta[hh], r[hh], c=colors, s=2*r[hh], vmin=0, vmax=24, cmap='viridis', alpha=0.75) # axr.set_rmax(18) # axr.set_rticks([6, 12, 18]) axr.set_rmax(24) axr.set_rticks([8, 16, 24]) # Move radial labels away from plotted line axr.set_rlabel_position(135) axr.grid(True) # plt.title(f'Day {c+1}') plt.title(f'{str(yy[24*c+12])[5:10]}') if c == 0: axr.set_xticklabels(plabels) else: axr.set_xticklabels([]) # if c != 0: # axr.set_xticklabels([]) # Vertical color bar [left, bottom, width, height] cbar_hovax = fig.add_axes(cbar_axes) fig.colorbar(fill, cax=cbar_hovax, orientation='vertical', pad="3%", ticks=clevs) cbar_hovax.tick_params(labelsize=self.cfont) # Plot shear hodograph labels text = 'Daily SHIPS Shear Vectors (kt)' fig.text(0.27, 0.02, text, color='black', fontsize='x-large', fontweight='heavy', va='baseline') # Add legend for hodographs in lower-right space self._plot_shp_legend(fig) plt.savefig(png) if self.doscp: print(f'scp {png} to {self.server} at {remdir}/') cmd = f'scp -q {png} {self.userid}@{self.server}:{remdir}/' os.system(cmd) def plot_hovship_quad(self, ytimes, array, mag, head, png, title, clevs, ptype, remdir, shr=False): """ Makes 4-panel hovmoller plots of IMERG hourly precip by compass (NW/NE/SW/SE) or shear (downshear-left, downshear-right, upshear-left, upshear-right) quadrants. Supports only UTC or local time, and generates plots only for multi-day results. :param ytimes: Dates/times along y-axis :param array: array of mean precip by radial bin :param png: image name for .png file :param title: text string for plot title :param clevs: contour levels for color-filled plots :param ptype: text string of plot type 'multiday' or 'diurnal' :param remdir: remote directory of server for scp-ing files :param shr: Optional logical to indicate compass (False) or shear (True) relative quadrant plots """ ytext = date_form = None yy = mdate.num2date(ytimes) fig = plt.figure(figsize=(10, 10), dpi=self.dpi) gs = gridspec.GridSpec(11, 8) gs.update(left=0.11, right=0.90, bottom=0.05, top=0.95, wspace=0.25, hspace=0.0) cbar_axes = [0.92, 0.232, 0.02, 0.71] r = np.zeros(24) theta = np.zeros(24) plabels = ["", "45$^\circ$", "", "135$^\circ$", "180$^\circ$", "235$^\circ$", "270$^\circ$", "315$^\circ$"] # NW/Downshear-left quad plot ax1 = plt.subplot(gs[0:4, 0:4]) ax1.contourf(self.xx, yy, array[0][:][:], levels=clevs, colors=self.colors, extend='max') if shr: ax1.set_xlabel('(a) Downshr-Left Quad Radial Bin (km)', size=self.cfont, weight='bold') else: ax1.set_xlabel('(a) NW Quadrant Radial Bin (km)', size=self.cfont, weight='bold') if self.invtime: print('Using inverted time for NW/DSL quad y-axis') ax1.invert_yaxis() if ptype == 'multiday': fig.suptitle(title, weight='bold', fontsize=self.tfont, y=0.98) date_form = DateFormatter("%m%d-%H") if self.ptime == 'utc': ytext = 'Date / UTC Time (MMdd-HH)' else: ytext = 'Date / Local Time (MMdd-HH)' elif ptype == 'diurnal': print("Diurnal mean not supported in Hovmoller+SHIP shear plots.\n") print("Exiting script....\n") sys.exit() ax1.yaxis.set_major_formatter(date_form) ax1.set_ylabel(ytext, size=self.cfont, weight='bold') ax1.yaxis.set_ticks_position('left') ax1.xaxis.set_ticks_position('bottom') # NE/Downshear-right quad plot ax2 = plt.subplot(gs[0:4, 4:8]) ax2.contourf(self.xx, yy, array[1][:][:], levels=clevs, colors=self.colors, extend='max') if shr: ax2.set_xlabel('(b) Downshr-Right Quad Radial Bin (km)', size=self.cfont, weight='bold') else: ax2.set_xlabel('(b) NE Quadrant Radial Bin (km)', size=self.cfont, weight='bold') if self.invtime: print('Using inverted time for NE/DSR y-axis') ax2.invert_yaxis() ax2.axes.yaxis.set_visible(False) ax2.xaxis.set_ticks_position('bottom') # SW/Upshear-left quad plot ax3 = plt.subplot(gs[5:9, 0:4]) ax3.contourf(self.xx, yy, array[2][:][:], levels=clevs, colors=self.colors, extend='max') if shr: ax3.set_xlabel('(c) Upshr-Left Quad Radial Bin (km)', size=self.cfont, weight='bold') else: ax3.set_xlabel('(c) SW Quadrant Radial Bin (km)', size=self.cfont, weight='bold') if self.invtime: print('Using inverted time for SW/USL y-axis') ax3.invert_yaxis() ax3.yaxis.set_major_formatter(date_form) ax3.set_ylabel(ytext, size=self.cfont, weight='bold') ax3.yaxis.set_ticks_position('left') ax3.xaxis.set_ticks_position('bottom') # SE/Upshear-right quad plot ax4 = plt.subplot(gs[5:9, 4:8]) csse = ax4.contourf(self.xx, yy, array[3][:][:], levels=clevs, colors=self.colors, extend='max') if shr: ax4.set_xlabel('(d) Upshr-Right Quad Radial Bin (km)', size=self.cfont, weight='bold') else: ax4.set_xlabel('(d) SE Quadrant Radial Bin (km)', size=self.cfont, weight='bold') if self.invtime: print('Using inverted time for SE/USR y-axis') ax4.invert_yaxis() ax4.axes.yaxis.set_visible(False) ax4.xaxis.set_ticks_position('bottom') # New section to plot SHIPS shear polar plots beneath IMERG hovmollers. # Make [up to] 7-panels of shear hodographs # Reset to 7 if # days exceeds 7. num_days = int(len(ytimes)/24) if num_days > 7: num_days = 7 for c in range(num_days): axr = plt.subplot(gs[10, c], projection='polar') # Set polar plot in meteorological convention axr.set_theta_direction(-1) axr.set_theta_offset(np.pi / 2.0) tstart = 24 * c tstop = tstart + 24 r[0:24] = mag[tstart:tstop] theta[0:24] = head[tstart:tstop] # print(f'Day {c + 1}; r/mag \n {r}') # print(f'Day {c + 1}; theta/head: \n {theta}') theta = theta * np.pi / 180.0 for hh in range(0,24): colors = hh axr.scatter(theta[hh], r[hh], c=colors, s=2*r[hh], vmin=0, vmax=24, cmap='viridis', alpha=0.75) # axr.set_rmax(18) # axr.set_rticks([6, 12, 18]) axr.set_rmax(24) axr.set_rticks([8, 16, 24]) # Move radial labels away from plotted line axr.set_rlabel_position(135) axr.grid(True) # plt.title(f'Day {c+1}') plt.title(f'{str(yy[24*c+12])[5:10]}') if c == 0: axr.set_xticklabels(plabels) else: axr.set_xticklabels([]) # if c != 0: # axr.set_xticklabels([]) # Vertical color bar [left, bottom, width, height] cbar_hovax = fig.add_axes(cbar_axes) fig.colorbar(csse, cax=cbar_hovax, orientation='vertical', pad="3%", ticks=clevs) cbar_hovax.tick_params(labelsize=self.cfont) # Plot shear hodograph labels text = 'Daily SHIPS Shear Vectors (kt)' fig.text(0.27, 0.02, text, color='black', fontsize='x-large', fontweight='heavy', va='baseline') # Add legend for hodographs in lower-right space self._plot_shp_legend(fig) plt.savefig(png) if self.doscp: print(f'scp {png} to {self.server} at {remdir}/') cmd = f'scp -q {png} {self.userid}@{self.server}:{remdir}/' os.system(cmd) def plot_png(self, geo, proj, lons, lats, array, png, title, clevs, rings, remdir): """ Generates .png images of geographical plots, including radial rings. :param geo: geographical extent (LL/UR lon, LL/UR lat) :param proj: plotting projection :param lons: 2D array of longitude :param lats: 2D array of latitude :param array: 2D array of data (IMERG rainfall, in this case) :param png: image name for .png file :param title: text string for plot title :param clevs: contour levels for color-filled plots :param rings: 1D array of range rings around TC center to plot on image :param remdir: remote directory of server for scp-ing files """ plot_range_rings = True mapscale = '10m' # supports 10m, 50m, or 110m. print('\nIn plot_png:') print('--------------') geo_extent = geo clon = 0.5 * (geo_extent[0] + geo_extent[1]) clat = 0.5 * (geo_extent[2] + geo_extent[3]) print('clon: ', clon) print('clat: ', clat) print('geo_extent: ', geo_extent) print('geo_extent[0]: ', geo_extent[0]) print('geo_extent[1]: ', geo_extent[1]) print('geo_extent[2]: ', geo_extent[2]) print('geo_extent[3]: ', geo_extent[3]) print('pngFile: ', png) print('title: ', title) print('clevs: ', clevs) print('radius range rings: ', rings) cs = None fig, ax = plt.subplots(1, 1, figsize=(self.xinch, self.yinch), dpi=self.dpi, subplot_kw={'projection': ccrs.PlateCarree( central_longitude=clon)}) ax.set_extent(geo_extent, crs=ccrs.PlateCarree()) # Draw Lats, Lons, Coastlines, Countries, States, and Lakes using # Natural Earth (public domain map dataset http://naturalearthdata.com) coastlines = cfeature.NaturalEarthFeature( category='physical', name='coastline', scale=mapscale, edgecolor='black', facecolor='none') ax.add_feature(coastlines, linewidth=0.2, ) countries = cfeature.NaturalEarthFeature( category='cultural', name='admin_0_countries', scale=mapscale, edgecolor='black', facecolor='none') ax.add_feature(countries, linewidth=0.2, ) states = cfeature.NaturalEarthFeature( category='cultural', name='admin_1_states_provinces', scale=mapscale, edgecolor='black', facecolor='none') ax.add_feature(states, linewidth=0.1, ) lakes = cfeature.NaturalEarthFeature( category='physical', name='lakes', scale=mapscale, edgecolor='black', facecolor='none') ax.add_feature(lakes, linewidth=0.1, ) # Removes large white space between data plot and colorbar # plt.tight_layout() # Range Rings (optional) if plot_range_rings: # tissot wants rad_km in meters, not kilometers! # Define locations lat = 0 lon = 1 tc = [clat, clon] # Setup location list & append as needed locs = [] locs.append(tc) # Loop through locations and range ring radii for loc in locs: ax.tissot(lons=loc[lon], lats=loc[lat], rad_km=2, n_samples=150, facecolor='None', edgecolor='red', lw=1.5) for ring_m in rings: ax.tissot(lons=loc[lon], lats=loc[lat], rad_km=ring_m, n_samples=150, facecolor='None', edgecolor='red', lw=0.5) # Reserved block for labeling radial rings (code not working yet). # ax.clabel( # circles, # Best results when labelling line contours. # colors=['black'], # manual=False, # Automatic placement vs manual placement. # inline=True, # Cut the line where the label will be placed. # fmt=' {:.0f} '.format, # Labels as ints, with extra space. # ) # Draw Gridlines (Lats and Lons) if proj == 'PlateCarree': self._draw_grid_lines(ax, geo_extent) start_time = time.time() print('\n --Running contourf....') if proj == 'PlateCarree': cs = ax.contourf(lons, lats, array, levels=clevs, colors=self.colors, extend='max', transform=ccrs.PlateCarree()) elif proj == 'AzimuthalEquidistant': cs = ax.contourf(lons, lats, array, levels=clevs, colors=self.colors, extend='max', transform=ccrs.AzimuthalEquidistant()) elif proj == 'Mercator': cs = ax.contourf(lons, lats, array, levels=clevs, colors=self.colors, extend='max', transform=ccrs.Mercator()) stop_time = time.time() contour_time = stop_time - start_time print('Timing for contourf: ', contour_time, ' seconds') cs.cmap.set_under('white') cs.cmap.set_over('white') # Colorbar convention for cartopy (horizontal) # [left, bottom, width, height] # cbar_ax = fig.add_axes([0.10, 0.06, 0.83, 0.02]) # fig.colorbar(cs, cax=cbar_ax, orientation='horizontal', ticks=clevs) # Colorbar convention for cartopy (vertical) # [left, bottom, width, height] cbar_ax = fig.add_axes([0.915, 0.11, 0.02, 0.77]) fig.colorbar(cs, cax=cbar_ax, orientation='vertical', ticks=clevs, extendfrac=0) cbar_ax.tick_params(labelsize=self.cfont) fig.suptitle(title, weight='bold', fontsize=self.tfont, y=0.92) plt.savefig(png) if self.doscp: print(f'scp {png} to {self.server} at {remdir}/') cmd = f'scp -q {png} {self.userid}@{self.server}:{remdir}/' os.system(cmd) def _draw_grid_lines(self, ax, geo_extent): """ : Draws lat/lon grid lines in geographical plots. """ latlonfont = 10 gl = ax.gridlines(crs=ccrs.PlateCarree(), draw_labels=True, linewidth=0.4, color='0.6', alpha=0.3, linestyle='dashed') gl.xlabels_top = False # Determine longitude increment siz_lon = geo_extent[1] - geo_extent[0] if siz_lon <= 8: lon_inc = 1 elif siz_lon <= 16: lon_inc = 2 elif siz_lon <= 25: lon_inc = 5 else: lon_inc = 10 # Set starting lon to be multiple of lon_inc # (ensuring starting value is lower than LLlon) lon = int((geo_extent[0] - lon_inc / 2.) / lon_inc) * lon_inc lonlist = [lon] while lon <= geo_extent[1]: lon += lon_inc # Modify Longitude labels for Dateline crossing # (put pos. AND neg. 180, so gridlines will be drawn) if lon == 180: lonlist.append(180.0) lonlist.append(-180.0) elif lon > 180: lonlist.append(lon - 360.0) else: lonlist.append(lon) gl.xlabels_top = False gl.xlocator = mticker.FixedLocator(lonlist) gl.xformatter = LONGITUDE_FORMATTER gl.xlabel_style = {'color': 'black', 'size': latlonfont} # Determine latitude increment siz_lat = geo_extent[3] - geo_extent[2] if siz_lat <= 8: lat_inc = 1 elif siz_lat <= 16: lat_inc = 2 elif siz_lat <= 25: lat_inc = 5 else: lat_inc = 10 # Set starting lat to be multiple of lat_inc # (ensuring starting value is lower than LLlat) lat = int((geo_extent[2] - lat_inc / 2.) / lat_inc) * lat_inc latlist = [lat] while lat <= geo_extent[3]: lat += lat_inc latlist.append(lat) gl.ylabels_right = False gl.ylocator = mticker.FixedLocator(latlist) gl.yformatter = LATITUDE_FORMATTER gl.ylabel_style = {'color': 'black', 'size': latlonfont} def _plot_shp_legend(self, fig): """ : Draw SHIPs 6-hourly shear block legend in the hovshp plots. """ x1 = 0.91 y1 = 0.15 dy = 0.018 cmap = matplotlib.cm.get_cmap('viridis') col00 = cmap(0.0) col03 = cmap(3/24) col06 = cmap(6/24) col09 = cmap(9/24) col12 = cmap(12/24) col15 = cmap(15/24) col18 = cmap(18/24) col21 = cmap(21/24) fig.text(x1, y1+dy, 'Hour Key', color=col00, fontsize='large') fig.text(x1, y1, ' o = 00', color=col00, fontsize='large') fig.text(x1, y1-dy, ' o = 03', color=col03, fontsize='large') fig.text(x1, y1-2*dy, ' o = 06', color=col06, fontsize='large') fig.text(x1, y1-3*dy, ' o = 09', color=col09, fontsize='large') fig.text(x1, y1-4*dy, ' o = 12', color=col12, fontsize='large') fig.text(x1, y1-5*dy, ' o = 15', color=col15, fontsize='large') fig.text(x1, y1-6*dy, ' o = 18', color=col18, fontsize='large') fig.text(x1, y1-7*dy, ' o = 21', color=col21, fontsize='large')