Changeset 64cf353 in flex_extract.git for python/plot_retrieved.py

Timestamp:

Feb 8, 2018, 9:54:05 PM (6 years ago)

Author:

Anne Philipp <bscannephilipp@…>

Branches:

master, ctbto, dev

Children:

02c8c50

Parents:

6180177

Message:

pep8 changes + documentation added + minor code style changes

File:

• python/plot_retrieved.py (modified) (15 diffs)

python/plot_retrieved.py

@@ -1 +1 @@
 #!/usr/bin/env python
 # This software is licensed under the terms of the Apache Licence Version 2.0
-# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0. 
-# 
+# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
+#
 # Functionality provided: Simple tool for creating maps and time series of retrieved fields.
-# 
-# Requirements: 
+#
+# Requirements:
 # in addition to a standard python 2.6 or 2.7 installation the following packages need to be installed
 # ECMWF WebMARS, gribAPI with python enabled, emoslib, ecaccess web toolkit, all available from https://software.ecmwf.int/
@@ -19 +19 @@
 if localpythonpath not in sys.path:
     sys.path.append(localpythonpath)
-    
+
 from matplotlib.pylab import *
 import matplotlib.patches as mpatches
@@ -43 +43 @@
     c.levels=asarray(c.levels,dtype='int')
     c.area=asarray(c.area)
-    
+
     index_keys=["date","time","step"]
     indexfile=c.inputdir+"/date_time_stepRange.idx"
@@ -67 +67 @@
 
         index_vals.append(key_vals)
-    
+
     fdict=dict()
     fmeta=dict()
@@ -101 +101 @@
                 fdatetimestep=fdatetime+datetime.timedelta(hours=step)
                 if len(fstamp)==0:
-                    fstamp[key].append(fdatetimestamp)    
+                    fstamp[key].append(fdatetimestamp)
                     fmeta[key].append((paramId,parameterName,gtype,fdatetime,gtime,step,level))
                     fdict[key].append(flipud(reshape(grib_get_values(gid),[gdict['Nj'],gdict['Ni']])))
@@ -118 +118 @@
                         fmeta[key].append((paramId,parameterName,gtype,fdatetime,gtime,step,level))
                         fdict[key].append(flipud(reshape(grib_get_values(gid),[gdict['Nj'],gdict['Ni']])))
-                    
+
 
             grib_release(gid)
             gid = grib_new_from_index(iid)
-            
+
     for k in fdict.keys():
         fml=fmeta[k]
         fdl=fdict[k]
-        
+
         for fd,fm in zip(fdl,fml):
             ftitle=fm[1]+' {} '.format(fm[-1])+datetime.datetime.strftime(fm[3],'%Y%m%d%H')+' '+stats(fd)
             pname='_'.join(fm[1].split())+'_{}_'.format(fm[-1])+datetime.datetime.strftime(fm[3],'%Y%m%d%H')+'.{:0>3}'.format(fm[5])
             plotmap(fd, fm,gdict,ftitle,pname+'.eps')
-        
+
     for k in fdict.keys():
         fml=fmeta[k]
@@ -143 +143 @@
             lat=-20
             lon=20
-            plottimeseries(fdl,fml,fsl,lat,lon, gdict, ftitle, pname+'.eps')   
-    
+            plottimeseries(fdl,fml,fsl,lat,lon, gdict, ftitle, pname+'.eps')
+
 def plottimeseries(flist,fmetalist,ftimestamps,lat,lon,gdict,ftitle,filename):
     t1=time.time()
@@ -158 +158 @@
     plt.close(f)
     print time.time()-t1,'s'
-    
+
 def plotmap(flist,fmetalist,gdict,ftitle,filename):
     t1=time.time()
     f=plt.figure(figsize=(12,6.7))
-    mbaxes = f.add_axes([0.05, 0.15, 0.8, 0.7]) 
+    mbaxes = f.add_axes([0.05, 0.15, 0.8, 0.7])
     m =Basemap(llcrnrlon=-180.,llcrnrlat=-90.,urcrnrlon=180,urcrnrlat=90.)
     #if bw==0 :
@@ -168 +168 @@
     #else:
         #fill_color=rgb(0.85,0.85,0.85)
-    
+
     lw=0.3
     m.drawmapboundary()
@@ -179 +179 @@
     xleft=gdict['longitudeOfFirstGridPointInDegrees']
     if xleft>180.0:
-        xleft-=360. 
+        xleft-=360.
     x=linspace(xleft,gdict['longitudeOfLastGridPointInDegrees'],gdict['Ni'])
     y=linspace(gdict['latitudeOfLastGridPointInDegrees'],gdict['latitudeOfFirstGridPointInDegrees'],gdict['Nj'])
@@ -186 +186 @@
     s=m.contourf(xx,yy, flist)
     title(ftitle,y=1.1)
-    cbaxes = f.add_axes([0.9, 0.2, 0.04, 0.6]) 
+    cbaxes = f.add_axes([0.9, 0.2, 0.04, 0.6])
     cb=colorbar(cax=cbaxes)
-    
+
     savefig(c.outputdir+'/'+filename)
     print 'created ',c.outputdir+'/'+filename
@@ -200 +200 @@
                             formatter_class=ArgumentDefaultsHelpFormatter)
 
-# the most important arguments  
+# the most important arguments
     parser.add_argument("--start_date", dest="start_date",
                         help="start date YYYYMMDD")
@@ -241 +241 @@
             print 'Try "'+sys.argv[0].split('/')[-1]+' -h" to print usage information'
             exit(1)
-    
-    if args.levelist:        
+
+    if args.levelist:
         c.levels=args.levelist.split('/')
     else:
@@ -250 +250 @@
     else:
         c.area='[0,0]'
-    
+
     c.paramIds=args.paramIds.split('/')
     if args.start_step:
@@ -269 +269 @@
     else:
         c.outputdir=c.inputdir
-    
+
         return args,c
 
 if __name__ == "__main__":
-    
+
     args,c=interpret_plotargs()
     plot_retrieved(args,c)