Context Navigation

source: trunk/UTIL/PYTHON/meso.py @ 372

Last change on this file since 372 was 372, checked in by acolaitis, 13 years ago
PYTHON: small commit to add possibility to do semilog plot along the y axis of 2D contour plots.
Property svn:executable set to ``*
File size: 11.7 KB

Rev	Line
[349]	1	#!/usr/bin/env python
	2
	3	### A. Spiga
	4
	5	###########################################################################################
	6	###########################################################################################
	7	### What is below relate to running the file as a command line executable (very convenient)
	8	if __name__ == "__main__":
	9	import sys
	10	from optparse import OptionParser ### to be replaced by argparse
	11	from api_wrapper import api_onelevel
	12	from netCDF4 import Dataset
	13	from myplot import getlschar, separatenames, readslices, adjust_length
	14	from os import system
	15	from planetoplot import planetoplot
	16	import numpy as np
	17
	18	#############################
	19	### Get options and variables
	20	parser = OptionParser()
	21	parser.add_option('-f', '--file', action='append',dest='namefile', type="string", default=None, help='[NEEDED] name of WRF file (append)')
[351]	22	parser.add_option('-l', '--level', action='store',dest='nvert', type="string", default="0", help='level or start,stop,step (def=0)(-i 2: p,mbar)(-i 3,4: z,km)')
[349]	23	parser.add_option('-p', '--proj', action='store',dest='proj', type="string", default=None, help='projection')
	24	parser.add_option('-b', '--back', action='store',dest='back', type="string", default=None, help='background image (def: None)')
	25	parser.add_option('-t', '--target', action='store',dest='target', type="string", default=None, help='destination folder')
	26	parser.add_option('-s', '--stride', action='store',dest='stride', type="int", default=3, help='stride vectors (def=3)')
	27	parser.add_option('-v', '--var', action='append',dest='var', type="string", default=None, help='variable color-shaded (append)')
[350]	28	parser.add_option('-n', '--num', action='store',dest='numplot', type="int", default=None, help='plot number (def=2)(<0: plot LT -numplot)')
[349]	29	parser.add_option('-i', '--interp', action='store',dest='interp', type="int", default=None, help='interpolation (2: p, 3: z-amr, 4:z-als)')
	30	parser.add_option('-c', '--color', action='store',dest='colorb', type="string", default="def", help='change colormap (nobar: no colorbar)')
	31	parser.add_option('-x', '--no-vect',action='store_false',dest='winds', default=True, help='no wind vectors')
	32	parser.add_option('-m', '--min', action='append',dest='vmin', type="float", default=None, help='bounding minimum value (append)')
	33	parser.add_option('-M', '--max', action='append',dest='vmax', type="float", default=None, help='bounding maximum value (append)')
	34	parser.add_option('-T', '--tiled', action='store_true',dest='tile', default=False, help='draw a tiled plot (no blank zone)')
	35	parser.add_option('-z', '--zoom', action='store',dest='zoom', type="float", default=None, help='zoom factor in %')
	36	parser.add_option('-N', '--no-api', action='store_true',dest='nocall', default=False, help='do not recreate api file')
	37	parser.add_option('-d', '--display',action='store_false',dest='display', default=True, help='do not pop up created images')
	38	parser.add_option('-e', '--itstep', action='store',dest='itstep', type="int", default=None, help='stride time (def=4)')
	39	parser.add_option('-H', '--hole', action='store_true',dest='hole', default=False, help='holes above max and below min')
	40	parser.add_option('-S', '--save', action='store',dest='save', type="string", default="gui", help='save mode (png,eps,svg,pdf or gui)(def=gui)')
	41	parser.add_option('-a', '--anomaly',action='store_true',dest='anomaly', default=False, help='compute and plot relative anomaly in %')
	42	parser.add_option('-w', '--with', action='store',dest='var2', type="string", default=None, help='variable contoured')
	43	parser.add_option('--div', action='store',dest='ndiv', type="int", default=10, help='number of divisions in colorbar (def: 10)')
	44	parser.add_option('-F', '--first', action='store',dest='first', type="int", default=1, help='first subscript to plot (def: 1)')
	45	parser.add_option('--mult', action='store',dest='mult', type="float", default=1., help='a multiplicative factor to plotted field')
	46	parser.add_option('--title', action='store',dest='zetitle', type="string", default="fill",help='customize the whole title')
[363]	47	parser.add_option('--inverty', action='store_true',dest='inverty', default=False, help='Force matplotlib.pyplot to revert the y-axis. Can be of use when plotting data along pressure levels. (pyplot naturally tend to plot the z-axis with increasing values)')
[369]	48	parser.add_option('--xmax', action='store',dest='xmax', type="float", default=None, help='Maximum value for the y-axis in contour-plots. Default is max(xaxis).')
	49	parser.add_option('--ymax', action='store',dest='ymax', type="float", default=None, help='Maximum value for the y-axis in contour-plots Default is max(yaxis)')
	50	parser.add_option('--xmin', action='store',dest='xmin', type="float", default=None, help='Minimum value for the y-axis in contour-plots. Default is min(xaxis).')
	51	parser.add_option('--ymin', action='store',dest='ymin', type="float", default=None, help='Minimum value for the y-axis in contour-plots Default is min(yaxis)')
[372]	52	parser.add_option('--logy', action='store_true',dest='logy', default=False, help='Set y-axis to logarithmic.')
[349]	53	#parser.add_option('-V', action='store', dest='comb', type="float", default=None, help='a defined combination of variables')
	54
	55	############# T.N. changes
	56	#parser.add_option('-o','--operation',action='store',dest='operation',type="string", default=None ,help='matrix of operations between files (for now see code, sorry)')
	57	parser.add_option('--lat', action='append',dest='slat',type="string", default=None, help='slices along latitude. One value, or two values separated by comas for averaging')
	58	parser.add_option('--lon', action='append',dest='slon', type="string", default=None, help='slices along longitude. One value, or two values separated by comas for averaging')
	59	parser.add_option('--vert', action='append',dest='svert',type="string", default=None, help='slices along vertical axis. One value, or two values separated by comas for averaging') # quelles coordonnees ?
	60	parser.add_option('--time', action='append',dest='stime',type="string", default=None, help='slices along time. One value, or two values separated by comas for averaging') # quelles coordonnees ?
	61
[359]	62	############# A.C. changes
	63	parser.add_option('-O','--output', action='store' ,dest='output',type="string", default=None, help='Output file name')
	64	parser.add_option('--res', action='store' ,dest='res',type="float", default=200., help='Resolution for png outputs. --save png must be specified. (def=200.)')
[349]	65
	66	(opt,args) = parser.parse_args()
	67	if opt.namefile is None:
	68	print "I want to eat one file at least ! Use winds.py -f name_of_my_file. Or type winds.py -h"
	69	exit()
	70	if opt.var is None and opt.anomaly is True:
	71	print "Cannot ask to compute anomaly if no variable is set"
	72	exit()
	73	print "Options:", opt
	74
	75	listvar = ''
	76	if opt.var is None:
	77	zerange = [-999999]
	78	else:
	79	zelen = len(opt.var)
	80	zerange = range(zelen)
	81	#if zelen == 1: listvar = opt.var[0] + ','
	82	#else :
	83	for jjj in zerange: listvar += opt.var[jjj] + ','
	84	listvar = listvar[0:len(listvar)-1]
	85	vmintab = adjust_length (opt.vmin, zelen)
	86	vmaxtab = adjust_length (opt.vmax, zelen)
	87
	88	print "namefile, length", opt.namefile, len(opt.namefile)
	89
	90	zeslat = readslices(opt.slat)
	91	zeslon = readslices(opt.slon)
	92	zesvert = readslices(opt.svert)
	93	zestime = readslices(opt.stime)
	94	print "slat,zeslat", opt.slat, zeslat
	95	print "slon,zeslon", opt.slon, zeslon
	96	print "svert,zesvert", opt.svert, zesvert
	97	print "stime,zestime", opt.stime, zestime
	98
	99	for i in range(len(opt.namefile)):
	100
	101	zefile = opt.namefile[i]
	102	print zefile
[351]	103	#zelevel = opt.nvert
[349]	104	stralt = None
	105	[lschar,zehour,zehourin] = getlschar ( zefile ) ## getlschar from wrfout (or simply return "" if another file)
	106
[351]	107	inputnvert = separatenames(opt.nvert)
	108	if np.array(inputnvert).size == 1:
	109	zelevel = float(inputnvert[0])
	110	ze_interp_levels = [-9999.]
	111	else:
	112	zelevel = -99.
	113	ze_interp_levels = np.linspace(float(inputnvert[0]),float(inputnvert[1]),float(inputnvert[2]))
	114	print 'level: ', zelevel
	115	print 'interp_levels: ',ze_interp_levels
	116
[349]	117	#####################################################
[351]	118	### Call Fortran routines for vertical interpolations
[349]	119	if opt.interp is not None:
[351]	120	if zelevel == 0. and opt.interp == 4: zelevel = 0.010
[349]	121	### winds or no winds
	122	if opt.winds : zefields = 'uvmet'
	123	else : zefields = ''
	124	### var or no var
	125	#if opt.var is None : pass
	126	if zefields == '' : zefields = listvar
	127	else : zefields = zefields + "," + listvar
	128	if opt.var2 is not None : zefields = zefields + "," + opt.var2
	129	print zefields
	130	zefile = api_onelevel ( path_to_input = '', \
	131	input_name = zefile, \
	132	fields = zefields, \
	133	interp_method = opt.interp, \
[351]	134	interp_level = ze_interp_levels, \
[349]	135	onelevel = zelevel, \
	136	nocall = opt.nocall )
	137	print zefile
	138	zelevel = 0 ## so that zelevel could play again the role of nvert
	139
	140	if opt.var is None: zerange = [-999999]
	141	else: zerange = range(zelen)
	142	for jjj in zerange:
	143	if jjj == -999999:
	144	argvar = None
	145	zevmin = None
	146	zevmax = None
	147	else:
	148	argvar = opt.var[jjj]
	149	if vmintab[jjj] != -999999: zevmin = vmintab[jjj]
	150	else: zevmin = None
	151	if vmaxtab[jjj] != -999999: zevmax = vmaxtab[jjj]
	152	else: zevmax = None
[369]	153
	154
	155	zexaxis=[opt.xmin,opt.xmax]
	156	zeyaxis=[opt.ymin,opt.ymax]
	157
	158
[349]	159	#############
	160	### Main call
[351]	161	name = planetoplot (zefile,level=int(zelevel),vertmode=opt.interp,\
[349]	162	proj=opt.proj,back=opt.back,target=opt.target,stride=opt.stride,var=argvar,\
	163	numplot=opt.numplot,colorb=opt.colorb,winds=opt.winds,\
	164	addchar=lschar,interv=[zehour,zehourin],vmin=zevmin,vmax=zevmax,\
	165	tile=opt.tile,zoom=opt.zoom,display=opt.display,\
	166	itstep=opt.itstep,hole=opt.hole,save=opt.save,\
	167	anomaly=opt.anomaly,var2=opt.var2,ndiv=opt.ndiv,first=opt.first,\
	168	mult=opt.mult,zetitle=opt.zetitle,\
[359]	169	slon=zeslon,slat=zeslat,svert=zesvert,stime=zestime,\
[369]	170	outputname=opt.output,resolution=opt.res,\
[372]	171	invert_y=opt.inverty,xaxis=zexaxis,yaxis=zeyaxis,ylog=opt.logy)
[349]	172	print 'Done: '+name
	173	system("rm -f to_be_erased")
	174
	175	#########################################################
	176	### Generate a .sh file with the used command saved in it
	177	command = ""
	178	for arg in sys.argv: command = command + arg + ' '
	179	f = open(name+'.sh', 'w')
	180	f.write(command)

Note: See TracBrowser for help on using the repository browser.

Download in other formats: