Changeset 2081 in lmdz_wrf

Timestamp:

Aug 11, 2018, 5:33:45 AM (6 years ago)

Author:

lfita

Message:

Final version of `draw_WRFeta_levels'

Location:

trunk/tools

Files:

• drawing.py (modified) (4 diffs)
• drawing_tools.py (modified) (5 diffs)

trunk/tools/drawing.py

@@ -65 +65 @@
 ## e.g. # drawing.py -o draw_multiWindRose -f '/media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out/sounding_uava_10868.nc@pres|-1;time|1@ua,va#/media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out/simout_vars_sndpt_10868_38lev.nc@bottom_top|-1;time|0@ua,va#/media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out/simout_vars_sndpt_10868_50lev.nc@bottom_top|-1;time|0@ua,va#/media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out/simout_vars_sndpt_10868_50lev_assigned.nc@bottom_top|-1;time|0@ua,va#/media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out/simout_vars_sndpt_10868_80lev.nc@bottom_top|-1;time|0@ua,va#/media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out/simout_vars_sndpt_10868_120lev.nc@bottom_top|-1;time|0@ua,va#/media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/select/out/simout_vars_sndpt_10868_NOaerosol.nc@bottom_top|-1;time|0@ua,va' -S 'linepoint;multicol;pres;-;auto;spectral;auto:obs,38lev,50lev,50leva,80lev,120lev:WindRose!obs!sim!comparison!on!2012/10/23!00!UTC:png:auto:True'
 ## e.g. # drawing.py -o draw_stations_map -f stations.inf -S 'snd|x|#FFAAAA|2|8@sfc|x|#AAFFAA|2|8:cyl,l:1.,20.,35.,49.:FPS!Alps!stations:pdf:true'
+## e.g. # drawing.py -o draw_WRFeta_levels -f 120lev/simin_vars.nc,80lev/simin_vars.nc,50lev/simin_vars.nc,50lev_assigned/simin_vars.nc,38lev/simin_vars.nc -f '120lev,80lev,50lev,50leva,38lev:auto:auto:0|10:FPS!Alps!vert!levels:pdf:true'
+
 
 #######
@@ -10304 +10306 @@
       values = [labs]:[colors]:[markers]:[legvals]:[imgtit]:[imgkind]:[close]
         [labs]= ',' list of labels for the plot (LaTeX like)
-        [colors]= ',' list of colors for the lines
-        [markers]= '@' list of markers for the lines
+        [colors]= ',' list of colors for the lines ('auto' for automatic)
+        [markers]= '@' list of markers for the lines ('auto' for automatic)
         [legvals]=[loclegend]|[fonstsize] values for the legend
           [locleg]: location of the legend (0, automatic)
@@ -10327 +10329 @@
 
     labs = values.split(':')[0].split(',')
-    colors = gen.auto_val_list(values.split(':')[1], ',', colorsauto)
-    markers = gen.auto_val_list(values.split(':')[2], '@', pointkindsauto)
-    legvals = values.split(':')[3].split(',')
+    colors = gen.auto_val_list(values.split(':')[1].split(','), len(labs),           \
+      drw.colorsauto)
+    markers = gen.auto_val_list(values.split(':')[2].split('@'), len(labs),          \
+      drw.pointkindsauto)
+    legvals = values.split(':')[3]
     imgtit = values.split(':')[4].replace('!', ' ')
     imgkind = values.split(':')[5]
-    close = gen.str_Bool(values.split(':')[6])
+    close = gen.Str_Bool(values.split(':')[6])
     
     files = filenames.split(',')
@@ -10384 +10388 @@
     locleg, legfontsize = drw.legend_values(legvals,'|')
 
-    plt.plot_WRFeta_levels(allhgtsea, allxhgt, allhgtxhgt, alldhgtsea, alldhgtxhgt,  \
+    drw.plot_WRFeta_levels(allhgtsea, allxhgt, allhgtxhgt, alldhgtsea, alldhgtxhgt,  \
       alletaw, alletau, labs, colors, markers, [locleg, legfontsize], imgtit,        \
       imgkind, close)
 
     return
+#foldn='/media/lluis/ExtDiskC_ext3/DATA/estudios/FPS_Alps/additional/IOP/sims/'
+#fils=foldn + '120lev/simin_vars.nc,' + foldn + '80lev/simin_vars.nc,' + foldn +      \
+#  '50lev/simin_vars.nc,' + foldn + '50lev_assigned/simin_vars.nc,' + foldn +         \
+#  '38lev/simin_vars.nc'
+#vals='120lev,80lev,50lev,50leva,38lev:auto:auto:0|8:FPS!Alps!vertical!levels:pdf:true'
+#print fils
+#draw_WRFeta_levels(fils, vals)
 
 #quit()

trunk/tools/drawing_tools.py

@@ -12743 +12743 @@
     return
 
-def plot_WRFeta_levels(ahgtsea, axhgt, ahgtxhgt, adhgtsea, adhgtxhgt, aetaw,    \
-     imgtit, labels, cols, marks, legvs, figtitle, kfig, close):
+def plot_WRFeta_levels(ahgtsea, axhgt, ahgtxhgt, adhgtsea, adhgtxhgt, aetaw, aetau,  \
+     labels, cols, marks, legvs, figtitle, kfig, close):
     """ Function to plot different sets of WRF eta-levels
       ahgtsea: all hegights of a point above sea
@@ -12761 +12761 @@
       close: whether figure should be closed
     """
-    fname = 'plot_draw_WRFeta_levels'
-
+    fname = 'plot_WRFeta_levels'
+
+    titleloc = (0.5,1.015)
 
     plt.rc('text', usetex=True)
 
-    fig = plt.subplots(2)
-
-    plt.subplot(1,1,2)
+    fig, ax = plt.subplots(4)
+
+    plt.subplot(2,2,1)
     # Absolute heights
     Nlabs = len(labels)
     for ilab in range(Nlabs):
-        hgtsea = ahgtsea[ilab]
-        hgtxhgt = ahgtxhgt[ilab]
+        hgtssea = ahgtsea[ilab]
+        hgtsxhgt = ahgtxhgt[ilab]
         znw = aetaw[ilab]
 
@@ -12781 +12782 @@
             plt.plot(znw, hgtsxhgt,  '-.', marker=marks[ilab], linewidth=1.5,        \
               markersize=4, label='$hgt_{hgtmax}^{'+labels[ilab]+'}$',               \
-              color=colors[ilab])
+              color=cols[ilab])
         else:
             plt.plot(znw, hgtsxhgt,  '-.', marker=marks[ilab], linewidth=1.5,        \
@@ -12790 +12791 @@
     plt.yscale('log')
 
-    titleloc = (0.5,1.075)
-
-    plt.title('Absoulte Heights from sea \& highest grid point',position=titleloc)
-    plt.legend(loc=0, prop={'size':10})
-
-    plt.title(gen.latex_text(figtitle))
-
-    plt.subplot(2,1,2)
+    plt.title('Absolute Heights',position=titleloc)
+    plt.legend(loc=legvs[0], prop={'size':legvs[1]})
+
+    plt.subplot(2,2,2)
+    # Absolute heights & relative
+    Nlabs = len(labels)
+    for ilab in range(Nlabs):
+        hgtssea = ahgtsea[ilab]
+        hgtsxhgt = ahgtxhgt[ilab]
+        dhgtssea = adhgtsea[ilab]
+        dhgtsxhgt = adhgtxhgt[ilab]
+        dz = len(hgtssea)
+
+        plt.plot(hgtssea[1:dz], dhgtssea, '-', marker=marks[ilab], linewidth=1.5,    \
+          markersize=4, label='$hgt_{sea}^{'+labels[ilab]+'}$', color=cols[ilab])
+#        if ilab == 0:
+#            plt.plot(hgtsxhgt[1:dz], dhgtsxhgt, '-.', marker=marks[ilab],            \
+#              linewidth=1.5, markersize=4, label='$hgt_{hgtmax}^{'+labels[ilab]+'}$',\
+#              color=cols[ilab])
+#        else:
+#            plt.plot(hgtsxhgt[1:dz], dhgtsxhgt, '-.', marker=marks[ilab],            \
+#              linewidth=1.5, markersize=4, label=None, color=cols[ilab])
+
+    plt.xlabel('aboslute height ($m$)', color='k')
+    plt.ylabel('$\delta height$ ($m$)', color='black')
+    plt.xscale('log')
+    plt.yscale('log')
+
+    plt.title('Absolute \& relative heights',position=titleloc)
+    #plt.legend(loc=0, prop={'size':10})
+
+    plt.subplot(2,2,3)
     # Relative heights
     Nlabs = len(labels)
     for ilab in range(Nlabs):
-        hgtsea = adhgtsea[ilab]
-        hgtxhgt = adhgtxhgt[ilab]
+        hgtssea = adhgtsea[ilab]
+        hgtsxhgt = adhgtxhgt[ilab]
         znw = aetau[ilab]
 
@@ -12810 +12835 @@
             plt.plot(znw, hgtsxhgt,  '-.', marker=marks[ilab], linewidth=1.5,        \
               markersize=4, label='$\delta hgt_{hgtmax}^{'+labels[ilab]+'}$',        \
-              color=colors[ilab])
+              color=cols[ilab])
         else:
             plt.plot(znw, hgtsxhgt,  '-.', marker=marks[ilab], linewidth=1.5,        \
               markersize=4, label=None, color=cols[ilab])
-
     plt.xlabel('$\eta$-level', color='k')
     plt.ylabel('$\delta height$ ($m$)', color='black')
     plt.yscale('log')
 
-    titleloc = (0.5,1.075)
-
-    plt.title('Relative Heights from sea \& highest grid point',position=titleloc)
-    plt.legend(loc=0, prop={'size':10})
-
-    plt.title(gen.latex_text(figtitle))
-
-    fig.suptitle(imgtit)
-
-    figname = 'stations_map'
+    plt.title('Relative Heights',position=titleloc)
+    #plt.legend(loc=0, prop={'size':10})
+
+    plt.subplot(2,2,4)
+    # Absoulte heights & grid-level
+    Nlabs = len(labels)
+    for ilab in range(Nlabs):
+        hgtssea = ahgtsea[ilab]
+        hgtsxhgt = ahgtxhgt[ilab]
+        znw = aetau[ilab]
+        dz = len(hgtssea)
+        zi = np.arange(1.,dz+1.)*100./dz
+
+        plt.plot(zi, hgtssea, '-', marker=marks[ilab], linewidth=1.5, markersize=0,  \
+          label='$\delta hgt_{sea}^{'+labels[ilab]+'}$', color=cols[ilab])
+        if ilab == 0:
+            plt.plot(zi, hgtsxhgt,  '-.', marker=marks[ilab], linewidth=1.5,         \
+              markersize=0, label='$\delta hgt_{hgtmax}^{' + labels[ilab] + '}$',    \
+              color=cols[ilab])
+        else:
+            plt.plot(zi, hgtsxhgt,  '-.', marker=marks[ilab], linewidth=1.5,         \
+              markersize=0, label=None, color=cols[ilab])
+    plt.grid()
+    plt.xlabel('\% z-grid point', color='k')
+    plt.ylabel('height ($m$)', color='black')
+    #plt.xscale('log')
+    plt.yscale('log')
+
+    plt.title('grid point',position=titleloc)
+    #plt.legend(loc=0, prop={'size':10})
+
+    fig.suptitle(gen.latex_text(figtitle))
+    plt.subplots_adjust(wspace=0.3, hspace=0.4)
+
+    figname = 'WRFeta_levels'
     output_kind(kfig, figname, close)