[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) |
---|