Changeset 549 in lmdz_wrf for trunk/tools

Timestamp:

Jul 1, 2015, 11:03:15 AM (9 years ago)

Author:

lfita

Message:

Adding in `draw_points':
[kindfigure]: kind of figure

'legend': only points in the map with the legend with the names
'labelled',[txtsize],[txtcol]: points with the names and size, color of text

Location:

trunk/tools

Files:

• drawing.py (modified) (7 diffs)
• drawing_tools.py (modified) (3 diffs)

trunk/tools/drawing.py

@@ -22 +22 @@
 ## e.g. # drawing.py -o draw_lines_time -f wrfout_d01_1980-03-01_00:00:00_Time_B0-E48-I1_south_north_B15-E15-I1_west_east_B15-E15-I1.nc -S 'time;y;time ([DD]${[HH]}$);file1;tas;evolution;time|hours!since!1949-12-01_00:00:00|exct,12,h|%d$^{%H}$;pdf' -v T2
 ## e.g. # drawing.py -o draw_barbs -f ERAI_pl199501_131-132.nc -S 'X|lon|lon|-1,Y|lat|lat|-1,Z|lev|lev|4,T|time|time|0:auto,auto,auto:wind,ms-1:cyl,c:ERA-Interim|winds|at|1000|hPa|on|1996|January|1st|at|00|UTC:pdf:ERAI_pl199501_131-132' -v var131,var132
-## e.g. # python ~/etudes/WRF_LMDZ/svn/LMDZ_WRF/tools/drawing.py -o draw_points -S 'tslist.dat,#,3,2,1:SuperStorm|sfc|stations:cyl,i:auto:None:0:png:stations_loc' -f $HOME'/etudes/domains/WRFdynamicoSENS_SuperStorm/geo_em.d02.nc,XLONG_M,XLAT_M,HGT_M,Time|0,height,0.,3000.,terrain,m'
-
+## e.g. # ~/etudes/WRF_LMDZ/svn/LMDZ_WRF/tools/drawing.py -o draw_points -S 'tslist.dat,#,3,2,1:SuperStorm|sfc|stations:cyl,i:legend:auto:None:0:png:stations_loc' -f $HOME'/etudes/domains/WRFdynamicoSENS_SuperStorm/geo_em.d02.nc,XLONG_M,XLAT_M,HGT_M,Time|0,height,0.,3000.,terrain,m'
+## e.g. # drawing.py -o draw_points -S 'tslist.dat,#,3,2,1:SuperStorm|sfc|stations:cyl,i:labelled,8,black:auto:None:0:png:stations_loc' -f $HOME'/etudes/domains/WRFdynamicoSENS_SuperStorm/geo_em_west_east_B25-E180-I1_south_north_B160-E262-I1.nc,XLONG_M,XLAT_M,HGT_M,Time|0,height,0.,1500.,terrain,m'
 main = 'drawing.py'
 
@@ -2186 +2186 @@
 def draw_points(filen, values):
     """ Function to plot a series of points
-      [values]= [ptasciifile]:[gtit]:[mapvalues]:[pointcolor]:[pointlabels]:[locleg]:
-        [figurek]:[figuren]
+      [values]= [ptasciifile]:[gtit]:[mapvalues]:[kindfigure]:[pointcolor]:[pointlabels]:
+        [locleg]:[figureko]:[figuren]
         [ptasciifile]:[file],[comchar],[collon],[collat],[lab]
           [file]: column ASCII file with the location of the points
@@ -2206 +2206 @@
              * 'h', high
              * 'f', full
+        [kindfigure]: kind of figure
+          'legend': only points in the map with the legend with the names
+          'labelled',[txtsize],[txtcol]: points with the names and size, color of text
         [pointcolor]: color for the points ('auto' for "red")
         [pointlabels]: ',' of labels [only used if [collab]='None'] ('None' for no labels)
@@ -2212 +2215 @@
           5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
           9: 'upper center', 10: 'center'
-        [figurek]: kind of figure
+        [figureko]: kind of the output file (pdf, png, ...)
         [figuren]: name of the figure
       [filen]= [ncfile],[lonvarn],[latvarn][,[varn],[dimvals],[vargn],[min],[max],[cbar],[varu]]
@@ -2235 +2238 @@
         quit()
 
-    expectargs = '[ptasciifile]:[gtit]:[mapvalues]:[pointcolor]:[pointlabels]:' +    \
-      '[locleg]:[figurek]:[figuren]'
+    expectargs = '[ptasciifile]:[gtit]:[mapvalues]:[kindfigure]:[pointcolor]:' +     \
+      '[pointlabels]:[locleg]:[figurek]:[figuren]'
  
     drw.check_arguments(fname,len(expectargs.split(':')),values,':',                 \
@@ -2244 +2247 @@
     gtit = values.split(':')[1]
     mapvalues = values.split(':')[2]
-    pointcolor = values.split(':')[3]
-    pointlabels = values.split(':')[4]
-    locleg = int(values.split(':')[5])
-    figurek = values.split(':')[6]
-    figuren = values.split(':')[7]
+    kindfigure = values.split(':')[3]
+    pointcolor = values.split(':')[4]
+    pointlabels = values.split(':')[5]
+    locleg = int(values.split(':')[6])
+    figureko = values.split(':')[7]
+    figuren = values.split(':')[8]
 
 # Getting station locations
@@ -2336 +2340 @@
 
     drw.plot_points(xptval, yptval, lonv, latv, varextra, varextrav, gtit, mapvalues,\
-      pointcolor, ptlabels, locleg, figurek, figuren)
+      kindfigure, pointcolor, ptlabels, locleg, figureko, figuren)
 
     onc.close()

trunk/tools/drawing_tools.py

@@ -2537 +2537 @@
 #quit()
 
-def plot_points(xval, yval, vlon, vlat, extravals, extrapar, vtit, mapv, color,      \
+def plot_points(xval, yval, vlon, vlat, extravals, extrapar, vtit, mapv, figk, color,\
   labels, lloc, kfig, figname):
     """ plotting points
@@ -2557 +2557 @@
           * 'h', high
           * 'f', full
+      figK= kind of figure
+        'legend': only points in the map with the legend with the names
+        'labelled',[txtsize],[txtcol]: points with the names and size, color of text
       color= color for the points/labels ('auto', for "red")
       labels= list of labels for the points (None, no labels)
@@ -2637 +2640 @@
     if labels is not None:
         for iv in range(len(xval)):
-#            plt.annotate(labels[iv], xy=(xval[iv],yval[iv]), xycoords='data')
-
             if np.mod(iv,7) == 0: N7pts = N7pts + 1
 #            print iv,xval[iv],yval[iv],labels[iv],ptkinds[N7pts]
-
             plt.plot(xval[iv],yval[iv], ptkinds[N7pts],label=labels[iv])
-        plt.legend(loc=lloc)
+
+        if figk[0:8] == 'labelled':
+            txtsize=int(figk.split(',')[1])
+            txtcol=figk.split(',')[2]
+            for iv in range(len(xval)):
+                plt.annotate(labels[iv], xy=(xval[iv],yval[iv]), xycoords='data',    \
+                  fontsize=txtsize, color=txtcol)
+        elif figk == 'legend':
+            plt.legend(loc=lloc)
 
     else: