#!/usr/bin/env python # Last modified: Time-stamp: <2008-10-22 13:43:15 haines> """lsrb_adcp_plot_month""" # plot each month import os, sys, glob, re import datetime, time, dateutil, dateutil.tz import pycdf import numpy sys.path.append('/home/haines/nccoos/raw2proc') del(sys) os.environ["MPLCONFIGDIR"]="/home/haines/.matplotlib/" from pylab import figure, twinx, savefig, setp, getp, cm, colorbar from matplotlib.dates import DayLocator, HourLocator, MinuteLocator, DateFormatter, date2num, num2date import procutil print 'lsrb_adcp_plot_month ...' # yyyy_mm = '2008_05' # yyyy_mm = '2008_06' # ncFile1='/seacoos/data/nccoos/level1/lsrb/adcp/lsrb_adcp_2008_01.nc' # ncFile2='/seacoos/data/nccoos/level1/lsrb/adcp/lsrb_adcp_2008_02.nc' # ncFile1='/seacoos/data/nccoos/level1/lsrb/adcp/lsrb_adcp_'+prev_month.strftime('%Y_%m')+'.nc' # ncFile2='/seacoos/data/nccoos/level1/lsrb/adcp/lsrb_adcp_'+this_month.strftime('%Y_%m')+'.nc' # ncFile = '/seacoos/data/nccoos/level1/lsrb/adcp/lsrb_adcp_'+yyyy_mm+'.nc' proc_dir = '/seacoos/data/nccoos/level1/lsrb/adcp/' fns = glob.glob((os.path.join(proc_dir, '*.nc'))) fns.sort() for fn in fns: m=re.search('\d{4}_\d{2}', fn) yyyy_mm = m.group() prev_month, this_month, next_month = procutil.find_months(yyyy_mm) # load data print ' ... ... read: ' + fn nc = pycdf.CDFMF((fn,)) ncvars = nc.variables() # print ncvars es = nc.var('time')[:] units = nc.var('time').units dt = [procutil.es2dt(e) for e in es] # set timezone info to UTC (since data from level1 should be in UTC!!) dt = [e.replace(tzinfo=dateutil.tz.tzutc()) for e in dt] # return new datetime based on computer local dt_local = [e.astimezone(dateutil.tz.tzlocal()) for e in dt] dn = date2num(dt) z = nc.var('z')[:] wd = nc.var('wd')[:] wl = nc.var('wl')[:] u = nc.var('u')[:] v = nc.var('v')[:] nc.close() # range for pcolor plots cmin, cmax = (-0.5, 0.5) # last dt in data for labels dt1 = dt[-1] dt2 = dt_local[-1] diff = abs(dt1 - dt2) if diff.days>0: last_dt_str = dt1.strftime("%H:%M %Z on %b %d, %Y") + ' (' + dt2.strftime("%H:%M %Z, %b %d") + ')' else: last_dt_str = dt1.strftime("%H:%M %Z") + ' (' + dt2.strftime("%H:%M %Z") + ')' \ + dt2.strftime(" on %b %d, %Y") fig = figure(figsize=(10, 8)) fig.subplots_adjust(left=0.1, bottom=0.1, right=0.9, top=0.9, wspace=0.1, hspace=0.1) ####################################### # Plot month ####################################### ax = fig.add_subplot(4,1,1) axs = [ax] # use masked array to hide NaN's on plot um = numpy.ma.masked_where(numpy.isnan(u), u) pc = ax.pcolor(dn, z, um.T, vmin=cmin, vmax=cmax) pc.set_label('True Eastward Current (m s-1)') ax.text(0.025, 0.1, pc.get_label(), fontsize="small", transform=ax.transAxes) # setup colorbar axes instance. l,b,w,h = ax.get_position() cax = fig.add_axes([l, b+h+0.04, 0.25*w, 0.03]) cb = colorbar(pc, cax=cax, orientation='horizontal') # draw colorbar cb.set_label('Current Velocity (m s-1)') cb.ax.xaxis.set_label_position('top') cb.ax.set_xticks([0.1, 0.3, 0.5, 0.7, 0.9]) cb.ax.set_xticklabels([-0.4, -0.2, 0, 0.2, 0.4]) # ax.plot returns a list of lines, so unpack tuple l1, = ax.plot_date(dt, wl, fmt='k-') l1.set_label('Water Level') ax.set_ylabel('Depth (m)') ax.set_ylim(-26.,2.) # ax.set_xlim(dt[0], dt[-1]) # first to last regardless of what ax.set_xlim(date2num(this_month), date2num(next_month-datetime.timedelta(seconds=1))) ax.xaxis.set_major_locator( DayLocator(range(2,32,2)) ) ax.xaxis.set_minor_locator( HourLocator(range(0,25,12)) ) ax.set_xticklabels([]) # this only moves the label not the tick labels ax.xaxis.set_label_position('top') ax.set_xlabel('LSRB ADCP -- ' + yyyy_mm) # right-hand side scale ax2 = twinx(ax) ax2.yaxis.tick_right() # convert (lhs) meters to (rhs) feet feet = [procutil.meters2feet(val) for val in ax.get_ylim()] ax2.set_ylim(feet) ax2.set_ylabel('Depth (ft)') # legend ls1 = l1.get_label() leg = ax.legend((l1,), (ls1,), loc='upper left') ltext = leg.get_texts() # all the text.Text instance in the legend llines = leg.get_lines() # all the lines.Line2D instance in the legend frame = leg.get_frame() # the patch.Rectangle instance surrounding the legend frame.set_facecolor('0.80') # set the frame face color to light gray frame.set_alpha(0.5) # set alpha low to see through setp(ltext, fontsize='small') # the legend text fontsize setp(llines, linewidth=1.5) # the legend linewidth # leg.draw_frame(False) # don't draw the legend frame ####################################### # ax = fig.add_subplot(4,1,2) axs.append(ax) # use masked array to hide NaN's on plot vm = numpy.ma.masked_where(numpy.isnan(v), v) pc = ax.pcolor(dn, z, vm.T, vmin=cmin, vmax=cmax) pc.set_label('True Northward Current (m s-1)') ax.text(0.025, 0.1, pc.get_label(), fontsize="small", transform=ax.transAxes) # ax.plot returns a list of lines, so unpack tuple l1, = ax.plot_date(dt, wl, fmt='k-') l1.set_label('Water Level') ax.set_ylabel('Depth (m)') ax.set_ylim(-26.,2) # ax.set_xlim(date2num(dt[0]), date2num(dt[-1])) ax.set_xlim(date2num(this_month), date2num(next_month-datetime.timedelta(seconds=1))) ax.xaxis.set_major_locator( DayLocator(range(2,32,2)) ) ax.xaxis.set_minor_locator( HourLocator(range(0,25,12)) ) ax.xaxis.set_major_formatter( DateFormatter('%m/%d') ) ax.set_xlabel('LSRB ADCP -- ' + yyyy_mm) # right-hand side scale ax2 = twinx(ax) ax2.yaxis.tick_right() # convert (lhs) meters to (rhs) feet feet = [procutil.meters2feet(val) for val in ax.get_ylim()] ax2.set_ylim(feet) ax2.set_ylabel('Depth (ft)') # legend ls1 = l1.get_label() leg = ax.legend((l1,), (ls1,), loc='upper left') ltext = leg.get_texts() # all the text.Text instance in the legend llines = leg.get_lines() # all the lines.Line2D instance in the legend frame = leg.get_frame() # the patch.Rectangle instance surrounding the legend frame.set_facecolor('0.80') # set the frame face color to light gray frame.set_alpha(0.5) # set alpha low to see through setp(ltext, fontsize='small') # the legend text fontsize setp(llines, linewidth=1.5) # the legend linewidth # leg.draw_frame(False) # don't draw the legend frame # save figure ofn = '/home/haines/rayleigh/img/lsrb/lsrb_adcp_'+yyyy_mm+'.png' print '... ... write: %s' % (ofn,) savefig(ofn)