def main(cmdLineArgs,stream=False):
numpy.seterr(divide='ignore', invalid='ignore', over='ignore') # To avoid warnings
if not os.path.exists(cmdLineArgs.gridspec): raise ValueError('Specified gridspec directory/tar file does not exist.')
if os.path.isdir(cmdLineArgs.gridspec):
x = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['x'][::2,::2]
xcenter = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['x'][1::2,1::2]
y = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['y'][::2,::2]
ycenter = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['y'][1::2,1::2]
msk = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_mask.nc').variables['mask'][:]
area = msk*netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['area'][:,:].reshape([msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
depth = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_topog.nc').variables['depth'][:]
elif os.path.isfile(cmdLineArgs.gridspec):
x = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','x')[::2,::2]
xcenter = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','x')[1::2,1::2]
y = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','y')[::2,::2]
ycenter = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','y')[1::2,1::2]
msk = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_mask.nc','mask')[:]
area = msk*m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','area')[:,:].reshape([msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
depth = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_topog.nc','depth')[:]
else:
raise ValueError('Unable to extract grid information from gridspec directory/tar file.')
Sobs = netCDF4.Dataset( cmdLineArgs.woa ).variables['salt']
if len(Sobs.shape)==3: Sobs = Sobs[0]
else: Sobs = Sobs[:,0].mean(axis=0)
rootGroup = netCDF4.MFDataset( cmdLineArgs.infile )
if 'salt' in rootGroup.variables: varName = 'salt'
elif 'so' in rootGroup.variables: varName = 'so'
else: raise Exception('Could not find "salt" or "so" in file "%s"'%(cmdLineArgs.infile))
if rootGroup.variables[varName].shape[0]>1: Smod = rootGroup.variables[varName][:,0].mean(axis=0)
else: Smod = rootGroup.variables[varName][0,0]
if cmdLineArgs.suptitle != '': suptitle = cmdLineArgs.suptitle + ' ' + cmdLineArgs.label
else: suptitle = rootGroup.title + ' ' + cmdLineArgs.label
imgbufs = []
ci=m6plot.pmCI(0.125,2.25,.25)
if stream is True: img = io.BytesIO()
else: img = cmdLineArgs.outdir+'/SSS_bias_WOA05.png'
m6plot.xyplot( Smod - Sobs , x, y, area=area,
suptitle=suptitle, title='SSS bias (w.r.t. WOA\'05) [ppt]',
clim=ci, colormap='dunnePM', centerlabels=True, extend='both',
save=img)
if stream is True: imgbufs.append(img)
m6plot.xycompare( Smod, Sobs , x, y, area=area,
suptitle=suptitle,
title1='SSS [ppt]',
title2='WOA\'05 SSS [ppt]',
clim=m6plot.linCI(20,30,10, 31,39,.5), colormap='dunneRainbow', extend='both',
dlim=ci, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/SSS_bias_WOA05.3_panel.png')
if stream is True:
return imgbufs
def main(cmdLineArgs,stream=None):
numpy.seterr(divide='ignore', invalid='ignore', over='ignore') # To avoid warnings
if not os.path.exists(cmdLineArgs.gridspec): raise ValueError('Specified gridspec directory/tar file does not exist.')
if os.path.isdir(cmdLineArgs.gridspec):
x = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['x'][::2,::2]
xcenter = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['x'][1::2,1::2]
y = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['y'][::2,::2]
ycenter = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['y'][1::2,1::2]
msk = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_mask.nc').variables['mask'][:]
area = msk*netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_hgrid.nc').variables['area'][:,:].reshape([msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
depth = netCDF4.Dataset(cmdLineArgs.gridspec+'/ocean_topog.nc').variables['depth'][:]
elif os.path.isfile(cmdLineArgs.gridspec):
x = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','x')[::2,::2]
xcenter = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','x')[1::2,1::2]
y = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','y')[::2,::2]
ycenter = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','y')[1::2,1::2]
msk = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_mask.nc','mask')[:]
area = msk*m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_hgrid.nc','area')[:,:].reshape([msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
depth = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,'ocean_topog.nc','depth')[:]
else:
raise ValueError('Unable to extract grid information from gridspec directory/tar file.')
Sobs = netCDF4.Dataset( cmdLineArgs.woa ).variables['salt']
if len(Sobs.shape)==3: Sobs = Sobs[0]
else: Sobs = Sobs[:,0].mean(axis=0)
rootGroup = netCDF4.MFDataset( cmdLineArgs.annual_file )
if 'salt' in rootGroup.variables: varName = 'salt'
elif 'so' in rootGroup.variables: varName = 'so'
else: raise Exception('Could not find "salt" or "so" in file "%s"'%(cmdLineArgs.annual_file))
if rootGroup.variables[varName].shape[0]>1: Smod = rootGroup.variables[varName][:,0].mean(axis=0)
else: Smod = rootGroup.variables[varName][0,0]
if cmdLineArgs.suptitle != '': suptitle = cmdLineArgs.suptitle + ' ' + cmdLineArgs.label
else: suptitle = rootGroup.title + ' ' + cmdLineArgs.label
ci=m6plot.pmCI(0.125,2.25,.25)
if stream is None: stream = cmdLineArgs.outdir+'/SSS_bias_WOA05.png'
m6plot.xyplot( Smod - Sobs , x, y, area=area,
suptitle=suptitle, title='SSS bias (w.r.t. WOA\'05) [ppt]',
clim=ci, colormap='dunnePM', centerlabels=True, extend='both',
save=stream)
m6plot.xycompare( Smod, Sobs , x, y, area=area,
suptitle=suptitle,
title1='SSS [ppt]',
title2='WOA\'05 SSS [ppt]',
clim=m6plot.linCI(20,30,10, 31,39,.5), colormap='dunneRainbow', extend='both',
dlim=ci, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/SSS_bias_WOA05.3_panel.png')
elif 'so' in rootGroupRef.variables: varName='so'
else: raise Exception('Could not find "salt" or "so" in file "%s"'%(cmdLineArgs.ref))
if rootGroupRef.variables[varName].shape[0]>1: Sref = rootGroupRef.variables[varName][:,0].mean(axis=0)
else: Sref = rootGroupRef.variables[varName][0,0]
rootGroup = netCDF4.Dataset( cmdLineArgs.annual_file )
if 'salt' in rootGroup.variables: varName='salt'
elif 'so' in rootGroup.variables: varName='so'
else: raise Exception('Could not find "salt" or "so" in file "%s"'%(cmdLineArgs.annual_file))
if len(rootGroup.variables[varName].shape)==4: Smod = rootGroup.variables[varName][:,0].mean(axis=0)
else: Smod = rootGroup.variables[varName][0]
if len(cmdLineArgs.label1): title1 = cmdLineArgs.label1
else: title1 = rootGroup.title
if len(cmdLineArgs.label2): title2 = cmdLineArgs.label2
else: title2 = rootGroupRef.title
ci=m6plot.pmCI(0.025,0.525,.05)
m6plot.xyplot( Smod - Sref , x, y, area=area,
suptitle=title1+' - '+title2,
title='SSS difference [ppt] '+cmdLineArgs.label,
clim=ci, colormap='dunnePM', centerlabels=True, extend='both',
save=cmdLineArgs.outdir+'/SSS_difference.png')
m6plot.xycompare( Smod, Sref , x, y, area=area,
suptitle='SSS difference [ppt] '+cmdLineArgs.label,
title1=title1, title2=title2,
clim=m6plot.linCI(20,30,10, 31,39,.5), colormap='dunneRainbow', extend='max',
dlim=ci, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/SSS_difference.3_panel.png')
msk = numpy.ma.array(msk, mask=(msk==0))
Tobs = netCDF4.Dataset( cmdLineArgs.woa )
if 'temp' in Tobs.variables: Tobs = Tobs.variables['temp']
elif 'ptemp' in Tobs.variables: Tobs = Tobs.variables['ptemp']
else: raise Exception('Could not find "temp" or "ptemp" in file "%s"'%(cmdLineArgs.woa))
if len(Tobs.shape)==3: Tobs = Tobs[0]
else: Tobs = Tobs[:,0].mean(axis=0)
rootGroup = netCDF4.Dataset( cmdLineArgs.annual_file )
if 'temp' in rootGroup.variables: varName = 'temp'
elif 'ptemp' in rootGroup.variables: varName = 'ptemp'
elif 'thetao' in rootGroup.variables: varName = 'thetao'
else: raise Exception('Could not find "temp", "ptemp" or "thetao" in file "%s"'%(cmdLineArgs.annual_file))
if rootGroup.variables[varName].shape[0]>1: salt = rootGroup.variables[varName][:,0].mean(axis=0)
else: Tmod = rootGroup.variables[varName][0,0]
ci=m6plot.pmCI(0.25,4.5,.5)
m6plot.xyplot( Tmod - Tobs , x, y, area=area,
suptitle=rootGroup.title+' '+cmdLineArgs.label, title='SST bias (w.r.t. WOA\'05) [$\degree$C]',
clim=ci, colormap='dunnePM', centerlabels=True, extend='both',
save=cmdLineArgs.outdir+'/SST_bias_WOA05.png')
m6plot.xycompare( Tmod, Tobs , x, y, area=area,
suptitle=rootGroup.title+' '+cmdLineArgs.label,
title1='SST [$\degree$C]',
title2='WOA\'05 SST [$\degree$C]',
clim=m6plot.linCI(-2,29,.5), colormap='dunneRainbow', extend='max',
dlim=ci, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/SST_bias_WOA05.3_panel.png')
ci = m6plot.pmCI(0.025, 0.525, .05)
m6plot.xyplot(Smod - Sref,
x,
y,
area=area,
suptitle=title1 + ' - ' + title2,
title='SSS difference [ppt] ' + cmdLineArgs.label,
clim=ci,
colormap='dunnePM',
centerlabels=True,
extend='both',
save=cmdLineArgs.outdir + '/SSS_difference.png')
m6plot.xycompare(Smod,
Sref,
x,
y,
area=area,
suptitle='SSS difference [ppt] ' + cmdLineArgs.label,
title1=title1,
title2=title2,
clim=m6plot.linCI(20, 30, 10, 31, 39, .5),
colormap='dunneRainbow',
extend='max',
dlim=ci,
dcolormap='dunnePM',
dextend='both',
centerdlabels=True,
save=cmdLineArgs.outdir + '/SSS_difference.3_panel.png')
area=area,
suptitle=title1 + ' - ' + title2,
title=r'%i-%im depth-average $\theta$ difference [$\degree$C]' %
(uDepth, lDepth) + cmdLineArgs.label,
clim=ci,
colormap='dunnePM',
centerlabels=True,
extend='both',
save=cmdLineArgs.outdir + '/T_%i-%im_zave_difference.png' %
(uDepth, lDepth))
m6plot.xycompare(
tPlot,
tRefPlot,
x,
y,
area=area,
suptitle=r'%i-%im depth-average $\theta$ difference [$\degree$C]' %
(uDepth, lDepth) + cmdLineArgs.label,
title1=title1,
title2=title2,
clim=m6plot.linCI(-2, 29, .5),
colormap='dunneRainbow',
extend='max',
dlim=ci,
dcolormap='dunnePM',
dextend='both',
centerdlabels=True,
save=cmdLineArgs.outdir + '/T_%i-%im_zave_difference.3_panel.png' %
(uDepth, lDepth))
def main(cmdLineArgs, stream=False):
numpy.seterr(divide='ignore', invalid='ignore',
over='ignore') # To avoid warnings
if not os.path.exists(cmdLineArgs.gridspec):
raise ValueError(
'Specified gridspec directory/tar file does not exist.')
if os.path.isdir(cmdLineArgs.gridspec):
x = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['x'][::2, ::2]
xcenter = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['x'][1::2, 1::2]
y = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['y'][::2, ::2]
ycenter = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['y'][1::2, 1::2]
msk = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_mask.nc').variables['mask'][:]
area = msk * netCDF4.Dataset(
cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['area'][:, :].reshape(
[msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
depth = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_topog.nc').variables['depth'][:]
elif os.path.isfile(cmdLineArgs.gridspec):
x = m6toolbox.readNCFromTar(cmdLineArgs.gridspec, 'ocean_hgrid.nc',
'x')[::2, ::2]
xcenter = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,
'ocean_hgrid.nc', 'x')[1::2, 1::2]
y = m6toolbox.readNCFromTar(cmdLineArgs.gridspec, 'ocean_hgrid.nc',
'y')[::2, ::2]
ycenter = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,
'ocean_hgrid.nc', 'y')[1::2, 1::2]
msk = m6toolbox.readNCFromTar(cmdLineArgs.gridspec, 'ocean_mask.nc',
'mask')[:]
area = msk * m6toolbox.readNCFromTar(
cmdLineArgs.gridspec, 'ocean_hgrid.nc', 'area')[:, :].reshape(
[msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
depth = m6toolbox.readNCFromTar(cmdLineArgs.gridspec, 'ocean_topog.nc',
'depth')[:]
else:
raise ValueError(
'Unable to extract grid information from gridspec directory/tar file.'
)
# open dataset
rootGroup = netCDF4.MFDataset(cmdLineArgs.infile)
# gather months from input dataset
tvar = rootGroup.variables['time']
times = [
netCDF4.num2date(i, tvar.units, calendar=tvar.calendar.lower())
for i in tvar[:]
]
idx = list(set([i.month - 1 for i in times]))
month_label = [i.strftime('%b') for i in times]
month_label = month_label[:len(idx)]
month_label = str.join(', ', month_label)
# read sst from model
if 'sst' in rootGroup.variables: varName = 'sst'
elif 'tos' in rootGroup.variables: varName = 'tos'
else:
raise Exception('Could not find "sst", "ptemp" or "tos" in file "%s"' %
(cmdLineArgs.infile))
if rootGroup.variables[varName].shape[0] > 1:
Tmod = rootGroup.variables[varName][:].mean(axis=0)
else:
Tmod = rootGroup.variables[varName][0]
# read sst from obs
Tobs = netCDF4.Dataset(cmdLineArgs.woa_monthly)
if 'temp' in Tobs.variables: Tobs = Tobs.variables['temp']
elif 'ptemp' in Tobs.variables: Tobs = Tobs.variables['ptemp']
else:
raise Exception('Could not find "temp" or "ptemp" in file "%s"' %
(cmdLineArgs.woa_monthly))
if len(Tobs.shape) == 3: Tobs = Tobs[0]
else: Tobs = Tobs[idx, 0].mean(axis=0)
# create title for plot
if cmdLineArgs.suptitle != '':
suptitle = cmdLineArgs.suptitle + ' ' + cmdLineArgs.label
else:
suptitle = rootGroup.title + ' ' + cmdLineArgs.label
# invoke m6plot
imgbufs = []
ci = m6plot.pmCI(0.25, 4.5, .5)
if stream is True: objOut = io.BytesIO()
else: objOut = cmdLineArgs.outdir + '/SST_bias_WOA05.png'
m6plot.xyplot(Tmod - Tobs,
x,
y,
area=area,
suptitle=suptitle,
title=month_label +
' SST bias (w.r.t. WOA\'05) [$\degree$C]',
clim=ci,
colormap='dunnePM',
centerlabels=True,
extend='both',
save=objOut)
if stream is True: imgbufs.append(objOut)
m6plot.xycompare(Tmod,
Tobs,
x,
y,
area=area,
suptitle=suptitle,
title1=month_label + ' SST [$\degree$C]',
title2='WOA\'05 ' + month_label + ' SST [$\degree$C]',
clim=m6plot.linCI(-2, 29, .5),
colormap='dunneRainbow',
extend='max',
dlim=ci,
dcolormap='dunnePM',
dextend='both',
centerdlabels=True,
save=cmdLineArgs.outdir + '/SST_bias_WOA05.3_panel.png')
if stream is True:
return imgbufs
for k in range(T.shape[0]):
zTop = numpy.minimum( z[k], -upperDepth )
zBot = numpy.maximum( z[k+1], -D )
dh = numpy.maximum( zTop -zBot, 0. )
#dh = numpy.minimum( z[k]-z[k+1], D-H)
H = H + dh
HT = HT + dh*T[k]
return HT/(H+1.e-20)
if len(cmdLineArgs.label1): title1 = cmdLineArgs.label1
else: title1 = rootGroup.title
if len(cmdLineArgs.label2): title2 = cmdLineArgs.label2
else: title2 = rootGroupRef.title
ci=m6plot.pmCI(0.05,1.,.1)
tPlot = depthAverageT(Tmod, Zmod, depth, lDepth, uDepth)
tRefPlot = depthAverageT(Tref, Zref, depth, lDepth, uDepth)
m6plot.xyplot( tPlot - tRefPlot , x, y, area=area,
suptitle=rootGroup.title+' '+cmdLineArgs.label,
title=r'%i-%im depth-average $\theta$ difference [$\degree$C]'%(uDepth,lDepth),
clim=ci, colormap='dunnePM', centerlabels=True, extend='both',
save=cmdLineArgs.outdir+'/T_%i-%im_zave_difference.png'%(uDepth,lDepth))
m6plot.xycompare( tPlot, tRefPlot , x, y, area=area,
suptitle=r'%i-%im depth-average $\theta$ difference [$\degree$C]'%(uDepth,lDepth)+cmdLineArgs.label,
title1=title1, title2=title2,
clim=m6plot.linCI(-2,29,.5), colormap='dunneRainbow', extend='max',
dlim=ci, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/T_%i-%im_zave_difference.3_panel.png'%(uDepth,lDepth))
def main(cmdLineArgs, stream=None):
numpy.seterr(divide='ignore', invalid='ignore',
over='ignore') # To avoid warnings
if not os.path.exists(cmdLineArgs.gridspec):
raise ValueError(
'Specified gridspec directory/tar file does not exist.')
if os.path.isdir(cmdLineArgs.gridspec):
x = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['x'][::2, ::2]
xcenter = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['x'][1::2, 1::2]
y = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['y'][::2, ::2]
ycenter = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['y'][1::2, 1::2]
msk = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_mask.nc').variables['mask'][:]
area = msk * netCDF4.Dataset(
cmdLineArgs.gridspec +
'/ocean_hgrid.nc').variables['area'][:, :].reshape(
[msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
depth = netCDF4.Dataset(cmdLineArgs.gridspec +
'/ocean_topog.nc').variables['depth'][:]
elif os.path.isfile(cmdLineArgs.gridspec):
x = m6toolbox.readNCFromTar(cmdLineArgs.gridspec, 'ocean_hgrid.nc',
'x')[::2, ::2]
xcenter = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,
'ocean_hgrid.nc', 'x')[1::2, 1::2]
y = m6toolbox.readNCFromTar(cmdLineArgs.gridspec, 'ocean_hgrid.nc',
'y')[::2, ::2]
ycenter = m6toolbox.readNCFromTar(cmdLineArgs.gridspec,
'ocean_hgrid.nc', 'y')[1::2, 1::2]
msk = m6toolbox.readNCFromTar(cmdLineArgs.gridspec, 'ocean_mask.nc',
'mask')[:]
area = msk * m6toolbox.readNCFromTar(
cmdLineArgs.gridspec, 'ocean_hgrid.nc', 'area')[:, :].reshape(
[msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
depth = m6toolbox.readNCFromTar(cmdLineArgs.gridspec, 'ocean_topog.nc',
'depth')[:]
else:
raise ValueError(
'Unable to extract grid information from gridspec directory/tar file.'
)
Tobs = netCDF4.Dataset(cmdLineArgs.woa)
if 'temp' in Tobs.variables: Tobs = Tobs.variables['temp']
elif 'ptemp' in Tobs.variables: Tobs = Tobs.variables['ptemp']
else:
raise Exception('Could not find "temp" or "ptemp" in file "%s"' %
(cmdLineArgs.woa))
if len(Tobs.shape) == 3: Tobs = Tobs[0]
else: Tobs = Tobs[:, 0].mean(axis=0)
rootGroup = netCDF4.MFDataset(cmdLineArgs.annual_file)
if 'temp' in rootGroup.variables: varName = 'temp'
elif 'ptemp' in rootGroup.variables: varName = 'ptemp'
elif 'thetao' in rootGroup.variables: varName = 'thetao'
else:
raise Exception(
'Could not find "temp", "ptemp" or "thetao" in file "%s"' %
(cmdLineArgs.annual_file))
if rootGroup.variables[varName].shape[0] > 1:
Tmod = rootGroup.variables[varName][:, 0].mean(axis=0)
else:
Tmod = rootGroup.variables[varName][0, 0]
if cmdLineArgs.suptitle != '':
suptitle = cmdLineArgs.suptitle + ' ' + cmdLineArgs.label
else:
suptitle = rootGroup.title + ' ' + cmdLineArgs.label
ci = m6plot.pmCI(0.25, 4.5, .5)
if stream is None: stream = cmdLineArgs.outdir + '/SST_bias_WOA05.png'
m6plot.xyplot(Tmod - Tobs,
x,
y,
area=area,
suptitle=suptitle,
title='SST bias (w.r.t. WOA\'05) [$\degree$C]',
clim=ci,
colormap='dunnePM',
centerlabels=True,
extend='both',
save=stream)
m6plot.xycompare(Tmod,
Tobs,
x,
y,
area=area,
suptitle=suptitle,
title1='SST [$\degree$C]',
title2='WOA\'05 SST [$\degree$C]',
clim=m6plot.linCI(-2, 29, .5),
colormap='dunneRainbow',
extend='max',
dlim=ci,
dcolormap='dunnePM',
dextend='both',
centerdlabels=True,
save=cmdLineArgs.outdir + '/SST_bias_WOA05.3_panel.png')
def depthAverageT(T, z, d, lowerDepth, upperDepth):
D = numpy.minimum(d, lowerDepth)
H = numpy.zeros(T.shape[1:])
HT = numpy.zeros(T.shape[1:])
for k in range(T.shape[0]):
zTop = numpy.minimum( z[k], -upperDepth )
zBot = numpy.maximum( z[k+1], -D )
dh = numpy.maximum( zTop -zBot, 0. )
#dh = numpy.minimum( z[k]-z[k+1], D-H)
H = H + dh
HT = HT + dh*T[k]
return HT/(H+1.e-20)
ci=m6plot.pmCI(0.25,4.5,.5)
tPlot = depthAverageT(Tmod, Zmod, depth, lDepth, uDepth)
tObsPlot = depthAverageT(Tobs, Zobs, depth, lDepth, uDepth)
m6plot.xyplot( tPlot - tObsPlot , x, y, area=area,
suptitle=rootGroup.title+' '+cmdLineArgs.label,
title=r'''%i-%im depth-average $\theta$ bias (w.r.t. WOA'05) [$\degree$C]'''%(uDepth,lDepth),
clim=ci, colormap='dunnePM', centerlabels=True, extend='both',
save=cmdLineArgs.outdir+'/T_%i-%im_zave_bias_WOA05.png'%(uDepth,lDepth))
m6plot.xycompare( tPlot, tObsPlot , x, y, area=area,
suptitle=rootGroup.title+' '+cmdLineArgs.label,
title1=r'%i-%im depth-average $\theta$ [$\degree$C]'%(uDepth,lDepth),
title2=r'''WOA'05 $\theta$ [$\degree$C]''',
clim=m6plot.linCI(-2,29,.5), colormap='dunneRainbow', extend='max',
dlim=ci, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/T_%i-%im_zave_bias_WOA05.3_panel.png'%(uDepth,lDepth))
ciMin = m6plot.linCI(0,95,5)
ciMax = m6plot.linCI(0,680,20)
diMin = m6plot.pmCI(2.5,32.5,5)
diMax = m6plot.pmCI(0,20,5, 40,100,20)
diMax = numpy.array([-250,-200,-150,-100,-50,-20,-10,-3,3,10,20,50,100,150,200,250])
m6plot.xyplot( MLDmod.min(axis=0) - MLDref.min(axis=0), x, y, area=area,
suptitle=title1+' - '+title2,
title='MLD difference [m] '+cmdLineArgs.label,
clim=diMin, colormap='dunnePM', centerlabels=True, extend='both',
save=cmdLineArgs.outdir+'/MLD_minimum_difference.png')
m6plot.xycompare( MLDmod.min(axis=0), MLDref.min(axis=0), x, y, area=area,
suptitle='MLD difference [m] '+cmdLineArgs.label,
title1=title1, title2=title2,
clim=ciMin, colormap='dunneRainbow', extend='max',
dlim=diMin, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/MLD_minimum_difference.3_panel.png')
m6plot.xyplot( MLDmod.max(axis=0) - MLDref.max(axis=0), x, y, area=area,
suptitle=title1+' - '+title2,
title='MLD difference [m] '+cmdLineArgs.label,
clim=diMax, colormap='dunnePM', centerlabels=False, extend='both',
save=cmdLineArgs.outdir+'/MLD_maximum_difference.png')
m6plot.xycompare( MLDmod.max(axis=0), MLDref.max(axis=0), x, y, area=area,
suptitle='MLD difference [m] '+cmdLineArgs.label,
title1=title1, title2=title2,
clim=ciMax, colormap='dunneRainbow', extend='max',
dlim=diMax, dcolormap='dunnePM', dextend='both', centerdlabels=False,
save=cmdLineArgs.outdir+'/MLD_maximum_difference.3_panel.png')
y,
area=area,
suptitle=rootGroup.title + ' ' + cmdLineArgs.label,
title=r'''%i-%im depth-average $\theta$ bias (w.r.t. WOA'05) [$\degree$C]'''
% (uDepth, lDepth),
clim=ci,
colormap='dunnePM',
centerlabels=True,
extend='both',
save=cmdLineArgs.outdir + '/T_%i-%im_zave_bias_WOA05.png' %
(uDepth, lDepth))
m6plot.xycompare(
tPlot,
tObsPlot,
x,
y,
area=area,
suptitle=rootGroup.title + ' ' + cmdLineArgs.label,
title1=r'%i-%im depth-average $\theta$ [$\degree$C]' % (uDepth, lDepth),
title2=r'''WOA'05 $\theta$ [$\degree$C]''',
clim=m6plot.linCI(-2, 29, .5),
colormap='dunneRainbow',
extend='max',
dlim=ci,
dcolormap='dunnePM',
dextend='both',
centerdlabels=True,
save=cmdLineArgs.outdir + '/T_%i-%im_zave_bias_WOA05.3_panel.png' %
(uDepth, lDepth))
else: raise Exception('Could not find "temp", "ptemp" or "thetao" in file "%s"'%(cmdLineArgs.ref))
if rootGroupRef.variables[varName].shape[0]>1: Tref = rootGroupRef.variables[varName][:,0].mean(axis=0)
else: Tref = rootGroupRef.variables[varName][0,0]
rootGroup = netCDF4.Dataset( cmdLineArgs.annual_file )
if 'temp' in rootGroup.variables: varName='temp'
elif 'ptemp' in rootGroup.variables: varName='ptemp'
elif 'thetao' in rootGroup.variables: varName='thetao'
else: raise Exception('Could not find "temp", "ptemp" or "thetao" in file "%s"'%(cmdLineArgs.annual_file))
if len(rootGroup.variables[varName].shape)==4: Tmod = rootGroup.variables[varName][:,0].mean(axis=0)
else: Tmod = rootGroup.variables[varName][0]
if len(cmdLineArgs.label1): title1 = cmdLineArgs.label1
else: title1 = rootGroup.title
if len(cmdLineArgs.label2): title2 = cmdLineArgs.label2
else: title2 = rootGroupRef.title
ci=m6plot.pmCI(0.05,1.05,.1)
m6plot.xyplot( Tmod - Tref , x, y, area=area,
suptitle=title1+' - '+title2,
title='SST difference [$\degree$C] '+cmdLineArgs.label,
clim=ci, colormap='dunnePM', centerlabels=True, extend='both',
save=cmdLineArgs.outdir+'/SST_difference.png')
m6plot.xycompare( Tmod, Tref , x, y, area=area,
suptitle='SST difference [$\degree$C] '+cmdLineArgs.label,
title1=title1, title2=title2,
clim=m6plot.linCI(-2,29,.5), colormap='dunneRainbow', extend='max',
dlim=ci, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/SST_difference.3_panel.png')
x = netCDF4.Dataset(cmdLineArgs.gridspecdir+'/ocean_hgrid.nc').variables['x'][::2,::2]
y = netCDF4.Dataset(cmdLineArgs.gridspecdir+'/ocean_hgrid.nc').variables['y'][::2,::2]
msk = netCDF4.Dataset(cmdLineArgs.gridspecdir+'/ocean_mask.nc').variables['mask'][:]
area = msk*netCDF4.Dataset(cmdLineArgs.gridspecdir+'/ocean_hgrid.nc').variables['area'][:,:].reshape([msk.shape[0], 2, msk.shape[1], 2]).sum(axis=-3).sum(axis=-1)
msk = numpy.ma.array(msk, mask=(msk==0))
Sobs = netCDF4.Dataset( cmdLineArgs.woa ).variables['salt']
if len(Sobs.shape)==3: Sobs = Sobs[0]
else: Sobs = Sobs[:,0].mean(axis=0)
rootGroup = netCDF4.Dataset( cmdLineArgs.annual_file )
if 'salt' in rootGroup.variables: varName = 'salt'
elif 'so' in rootGroup.variables: varName = 'so'
else: raise Exception('Could not find "salt" or "so" in file "%s"'%(cmdLineArgs.annual_file))
if rootGroup.variables[varName].shape[0]>1: salt = rootGroup.variables[varName][:,0].mean(axis=0)
else: Smod = rootGroup.variables[varName][0,0]
ci=m6plot.pmCI(0.125,2.25,.25)
m6plot.xyplot( Smod - Sobs , x, y, area=area,
suptitle=rootGroup.title+' '+cmdLineArgs.label, title='SSS bias (w.r.t. WOA\'05) [ppt]',
clim=ci, colormap='dunnePM', centerlabels=True, extend='both',
save=cmdLineArgs.outdir+'/SSS_bias_WOA05.png')
m6plot.xycompare( Smod, Sobs , x, y, area=area,
suptitle=rootGroup.title+' '+cmdLineArgs.label,
title1='SSS [ppt]',
title2='WOA\'05 SSS [ppt]',
clim=m6plot.linCI(20,30,10, 31,39,.5), colormap='dunneRainbow', extend='both',
dlim=ci, dcolormap='dunnePM', dextend='both', centerdlabels=True,
save=cmdLineArgs.outdir+'/SSS_bias_WOA05.3_panel.png')
ci = m6plot.pmCI(0.25, 4.5, .5)
m6plot.xyplot(Tmod - Tobs,
x,
y,
area=area,
suptitle=rootGroup.title + ' ' + cmdLineArgs.label,
title='SST bias (w.r.t. WOA\'05) [$\degree$C]',
clim=ci,
colormap='dunnePM',
centerlabels=True,
extend='both',
save=cmdLineArgs.outdir + '/SST_bias_WOA05.png')
m6plot.xycompare(Tmod,
Tobs,
x,
y,
area=area,
suptitle=rootGroup.title + ' ' + cmdLineArgs.label,
title1='SST [$\degree$C]',
title2='WOA\'05 SST [$\degree$C]',
clim=m6plot.linCI(-2, 29, .5),
colormap='dunneRainbow',
extend='max',
dlim=ci,
dcolormap='dunnePM',
dextend='both',
centerdlabels=True,
save=cmdLineArgs.outdir + '/SST_bias_WOA05.3_panel.png')
