Context Navigation

source: trunk/UTIL/PYTHON/gcm.py @ 373

Last change on this file since 373 was 372, checked in by acolaitis, 14 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: 14.5 KB

Rev	Line
[349]	1	#!/usr/bin/env python
	2
[361]	3	### T. Navarro + A. Spiga + A. Colaitis
[349]	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
[361]	12	from zrecast_wrapper import call_zrecast
[349]	13	from netCDF4 import Dataset
	14	from myplot import getlschar, separatenames, readslices, adjust_length
	15	from os import system
	16	from planetoplot import planetoplot
[361]	17
[349]	18	import numpy as np
	19
	20	#############################
	21	### Get options and variables
	22	parser = OptionParser()
	23	parser.add_option('-f', '--file', action='append',dest='namefile', type="string", default=None, help='[NEEDED] name of WRF file (append). Plot files separated by comas in the same figure')
[351]	24	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]	25	parser.add_option('-p', '--proj', action='store',dest='proj', type="string", default=None, help='projection')
	26	parser.add_option('-b', '--back', action='store',dest='back', type="string", default=None, help='background image (def: None)')
	27	parser.add_option('-t', '--target', action='store',dest='target', type="string", default=None, help='destination folder')
	28	parser.add_option('-s', '--stride', action='store',dest='stride', type="int", default=3, help='stride vectors (def=3)')
	29	parser.add_option('-v', '--var', action='append',dest='var', type="string", default=None, help='variable color-shaded (append)')
[350]	30	parser.add_option('-n', '--num', action='store',dest='numplot', type="int", default=None, help='plot number (def=2)(<0: plot LT -numplot)')
[349]	31	parser.add_option('-i', '--interp', action='store',dest='interp', type="int", default=None, help='interpolation (2: p, 3: z-amr, 4:z-als)')
	32	parser.add_option('-c', '--color', action='store',dest='colorb', type="string", default="def", help='change colormap (nobar: no colorbar)')
	33	parser.add_option('-x', '--no-vect',action='store_false',dest='winds', default=True, help='no wind vectors')
	34	parser.add_option('-m', '--min', action='append',dest='vmin', type="float", default=None, help='bounding minimum value (append)')
	35	parser.add_option('-M', '--max', action='append',dest='vmax', type="float", default=None, help='bounding maximum value (append)')
	36	parser.add_option('-T', '--tiled', action='store_true',dest='tile', default=False, help='draw a tiled plot (no blank zone)')
	37	parser.add_option('-z', '--zoom', action='store',dest='zoom', type="float", default=None, help='zoom factor in %')
	38	parser.add_option('-N', '--no-api', action='store_true',dest='nocall', default=False, help='do not recreate api file')
	39	parser.add_option('-d', '--display',action='store_false',dest='display', default=True, help='do not pop up created images')
	40	parser.add_option('-e', '--itstep', action='store',dest='itstep', type="int", default=None, help='stride time (def=4)')
	41	parser.add_option('-H', '--hole', action='store_true',dest='hole', default=False, help='holes above max and below min')
	42	parser.add_option('-S', '--save', action='store',dest='save', type="string", default="gui", help='save mode (png,eps,svg,pdf or gui)(def=gui)')
	43	parser.add_option('-a', '--anomaly',action='store_true',dest='anomaly', default=False, help='compute and plot relative anomaly in %')
	44	parser.add_option('-w', '--with', action='store',dest='var2', type="string", default=None, help='variable contoured')
	45	parser.add_option('--div', action='store',dest='ndiv', type="int", default=10, help='number of divisions in colorbar (def: 10)')
	46	parser.add_option('-F', '--first', action='store',dest='first', type="int", default=1, help='first subscript to plot (def: 1)')
	47	parser.add_option('--mult', action='store',dest='mult', type="float", default=1., help='a multiplicative factor to plotted field')
	48	parser.add_option('--title', action='store',dest='zetitle', type="string", default="fill",help='customize the whole title')
	49	#parser.add_option('-V', action='store', dest='comb', type="float", default=None, help='a defined combination of variables')
	50
[359]	51	parser.add_option('-O','--output', action='store',dest='output', type="string", default=None, help='Output file name')
	52	parser.add_option('--res', action='store',dest='res', type="float", default=200., help='Resolution for png outputs. --save png must be specified. (def=200.)')
	53	parser.add_option('--operation', action='store',dest='operation', type="string", default=None, help='Basic operation to perform between input file and a specified reference file. Available: addition "+", soustraction "-". Operation is done as: file1 +or- fileref. The reference file must be specified using -fref option')
	54	parser.add_option('--fref', action='store',dest='fref', type="string", default=None, help='Reference namefile for the --operation option.')
	55	parser.add_option('--mope', action='store',dest='vminope', type="float", default=0., help='bounding minimum value for inter-file operation')
	56	parser.add_option('--Mope', action='store',dest='vmaxope', type="float", default=0., help='bounding maximum value for inter-file operation')
	57	parser.add_option('--titleref', action='store',dest='titref', type="string", default="fill", help='title for the reference file. Default: title of figure (1)')
[369]	58	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).')
	59	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)')
	60	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).')
	61	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)')
[363]	62	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)')
[372]	63	parser.add_option('--logy', action='store_true',dest='logy', default=False, help='Set y-axis to logarithmic.')
[359]	64
[349]	65	############# T.N. changes
	66	#parser.add_option('-o','--operation',action='store',dest='operation',type="string", default=None ,help='matrix of operations between files (for now see code, sorry)')
	67	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')
	68	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')
	69	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 ?
	70	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 ?
	71
	72	(opt,args) = parser.parse_args()
	73	if opt.namefile is None:
	74	print "I want to eat one file at least ! Use winds.py -f name_of_my_file. Or type winds.py -h"
	75	exit()
	76	if opt.var is None and opt.anomaly is True:
	77	print "Cannot ask to compute anomaly if no variable is set"
	78	exit()
	79	print "Options:", opt
	80
[361]	81	# listvar = ''
	82	# if opt.var is None:
	83	# zerange = [-999999]
	84	# else:
	85	# zelen = len(opt.var)
	86	# zerange = range(zelen)
	87	# #if zelen == 1: listvar = opt.var[0] + ','
	88	# #else :
	89	# for jjj in zerange: listvar += opt.var[jjj] + ','
	90	# listvar = listvar[0:len(listvar)-1]
	91	# vmintab = adjust_length (opt.vmin, zelen)
	92	# vmaxtab = adjust_length (opt.vmax, zelen)
[349]	93
[359]	94
	95	################################
	96	### General check
	97
	98	if opt.fref is not None:
	99	if opt.operation is None:
	100	print "you must specify an operation when using a reference file"
	101	exit()
	102	if opt.operation is not None:
	103	if opt.fref is None:
	104	print "you must specifiy a reference file when using inter-file operations"
	105	exit()
	106
[362]	107	interpref=False
[361]	108	if opt.fref is not None:
	109	if opt.operation is not None:
	110	if opt.interp is not None:
	111	interpref=True
	112
[359]	113	################################
	114
	115
[349]	116	print "namefile, length", opt.namefile, len(opt.namefile)
	117
	118	zeslat = readslices(opt.slat)
	119	zeslon = readslices(opt.slon)
	120	zesvert = readslices(opt.svert)
	121	zestime = readslices(opt.stime)
	122	print "slat,zeslat", opt.slat, zeslat
	123	print "slon,zeslon", opt.slon, zeslon
	124	print "svert,zesvert", opt.svert, zesvert
	125	print "stime,zestime", opt.stime, zestime
	126
	127	for i in range(len(opt.namefile)):
	128	for j in range(len(opt.var)):
	129
	130	zenamefiles = separatenames(opt.namefile[i])
	131	print "zenamefiles", zenamefiles
	132
	133	if opt.vmin is not None : zevmin = opt.vmin[min(i,len(opt.vmin)-1)]
	134	else: zevmin = None
	135	if opt.vmax is not None : zevmax = opt.vmax[min(i,len(opt.vmax)-1)]
	136	else: zevmax = None
	137	print "vmin, zevmin", opt.vmin, zevmin
	138	print "vmax, zevmax", opt.vmax, zevmax
	139
	140	zevar = separatenames(opt.var[j])
[361]	141	zevars = zevar[j]
[349]	142	zevar = zevar[0]
	143	print "var, zevar", opt.var, zevar
	144
	145	#checkcoherence(len(zenamefiles),len(opt.slat),len(opt.slon),len(opt.stime))
	146
	147	zefile = zenamefiles[0]
	148
[351]	149	#zelevel = opt.nvert
[349]	150	stralt = None
[357]	151	#[lschar,zehour,zehourin] = getlschar ( zefile ) ## getlschar from wrfout (or simply return "" if another file)
	152	[lschar,zehour,zehourin] = ["",0,0] ## dummy
[349]	153
[351]	154	inputnvert = separatenames(opt.nvert)
	155	if np.array(inputnvert).size == 1:
	156	zelevel = float(inputnvert[0])
	157	ze_interp_levels = [-9999.]
	158	else:
	159	zelevel = -99.
	160	ze_interp_levels = np.linspace(float(inputnvert[0]),float(inputnvert[1]),float(inputnvert[2]))
	161	print 'level: ', zelevel
	162	print 'interp_levels: ',ze_interp_levels
	163
[349]	164	#####################################################
[351]	165	### Call Fortran routines for vertical interpolations --> zrecast for GCM ?
[361]	166	# if opt.interp is not None:
	167	# if zelevel == 0. and opt.interp == 4: zelevel = 0.010
	168	# ### winds or no winds
	169	# if opt.winds : zefields = 'uvmet'
	170	# else : zefields = ''
	171	# ### var or no var
	172	# #if opt.var is None : pass
	173	# if zefields == '' : zefields = listvar
	174	# else : zefields = zefields + "," + listvar
	175	# if opt.var2 is not None : zefields = zefields + "," + opt.var2
	176	# print zefields
	177	# zefile = api_onelevel ( path_to_input = '', \
	178	# input_name = zefile, \
	179	# fields = zefields, \
	180	# interp_method = opt.interp, \
	181	# interp_level = ze_interp_levels, \
	182	# onelevel = zelevel, \
	183	# nocall = opt.nocall )
	184	# print zefile
	185	# zelevel = 0 ## so that zelevel could play again the role of nvert
	186
	187	# A.C. #####################################################
	188	### Call Fortran routines for vertical interpolations --> zrecast
	189
[349]	190	if opt.interp is not None:
[361]	191	interpolated_files=""
	192	interpolated_files=call_zrecast(interp_mode=opt.interp,\
	193	input_name=zenamefiles,\
	194	fields=zevars)
[349]	195
[361]	196	zenamefiles=interpolated_files
	197	if interpref:
	198	interpolated_ref=""
	199	interpolated_ref=call_zrecast(interp_mode=opt.interp,\
	200	input_name=[opt.fref],\
	201	fields=zevars)
[349]	202
[361]	203	reffile=interpolated_ref[0]
	204	else:
	205	reffile=opt.fref
[369]	206	# Divers ####################################################
	207
	208	zexaxis=[opt.xmin,opt.xmax]
	209	zeyaxis=[opt.ymin,opt.ymax]
[361]	210
[349]	211	#############
	212	### Main call
[351]	213	name = planetoplot (zenamefiles,level=int(zelevel),vertmode=opt.interp,\
[349]	214	proj=opt.proj,back=opt.back,target=opt.target,stride=opt.stride,var=zevar,\
	215	numplot=opt.numplot,colorb=opt.colorb,winds=opt.winds,\
	216	addchar=lschar,interv=[zehour,zehourin],vmin=zevmin,vmax=zevmax,\
	217	tile=opt.tile,zoom=opt.zoom,display=opt.display,\
	218	itstep=opt.itstep,hole=opt.hole,save=opt.save,\
	219	anomaly=opt.anomaly,var2=opt.var2,ndiv=opt.ndiv,first=opt.first,\
	220	mult=opt.mult,zetitle=opt.zetitle,\
[359]	221	slon=zeslon,slat=zeslat,svert=zesvert,stime=zestime,\
	222	outputname=opt.output,resolution=opt.res,\
[363]	223	ope=opt.operation,fileref=reffile,minop=opt.vminope,maxop=opt.vmaxope,titleref=opt.titref,\
[372]	224	invert_y=opt.inverty,xaxis=zexaxis,yaxis=zeyaxis,ylog=opt.logy)
[349]	225	print 'Done: '+name
	226	system("rm -f to_be_erased")
	227
	228	#########################################################
	229	### Generate a .sh file with the used command saved in it
	230	command = ""
	231	for arg in sys.argv: command = command + arg + ' '
	232	name = 'pycommand'
	233	f = open(name+'.sh', 'w')
	234	f.write(command)

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

Download in other formats: