Changeset 2079 in lmdz_wrf

Timestamp:

Aug 10, 2018, 9:25:59 PM (7 years ago)

Author:

lfita

Message:

Starting to add draw_WRFeta_levels

Location:

trunk/tools

Files:

• drawing.py (modified) (1 diff)
• drawing_tools.py (modified) (1 diff)

trunk/tools/drawing.py

@@ -10298 +10298 @@
     return
 
+def draw_WRFeta_levels(filenames, values):
+    """ Function to plot vertical levels from WRF wrfinput_d[nn] file
+      filenames= ',' list of name of files to use
+      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
+        [legvals]=[loclegend]|[fonstsize] values for the legend
+          [locleg]: location of the legend (0, automatic)
+            1: 'upper right', 2: 'upper left', 3: 'lower left', 4: 'lower right',
+            5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
+            9: 'upper center', 10: 'center'
+          [fontsize]: font size for the legend ('auto' for 12)
+        [imgtit]= title of the image ('!' for spaces)
+        [imgkind]= kind of file output of the image (ps, pns, pdf, ...)
+        [close]= whether figure should be closed or not
+    """
+    fname = 'draw_WRFeta_levels'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print draw_WRFeta_levels.__doc__
+        quit()
+
+    expectargs = '[labs]:[colors]:[markers]:[legvals]:[imgtit]:[imgkind]:[close]'
+    drw.check_arguments(fname,values,expectargs,':')
+
+    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(',')
+    imgtit = values.split(':')[4].replace('!', ' ')
+    imgkind = values.split(':')[5]
+    close = gen.str_Bool(values.split(':')[6])
+    
+    files = filenames.split(',')
+    allhgtsea = []
+    allxhgt = []
+    allhgtxhgt = []
+    alldhgtsea = []
+    alldhgtxhgt = []
+    alletaw = []
+    alletau = []
+
+    for filen in files:
+        if not os.path.isfile(filen):
+            print errormsg
+            print '  ' + fname + ": file '" + filen + "' does not exist !!"
+            quit(-1)
+
+        onc = NetCDFFile(filen, 'r')
+        ophb = onc.variables['PHB']
+        olandseamask = onc.variables['LANDMASK']
+        ohgt = onc.variables['HGT']
+        oznw = onc.variables['ZNW']
+        oznu = onc.variables['ZNU']
+
+        dimx = ophb.shape[3]
+        dimy = ophb.shape[2]
+        dimz = ophb.shape[1]
+        dimt = ophb.shape[0]
+
+        for j in range(10,dimy):
+            for i in range(10,dimx):
+                if olandseamask[0,j,i] == 0:
+                    hgtssea = ophb[0,:,j,i]/9.8
+                    break
+
+        maxhgt = np.max(ohgt[0,:,:])
+        ijxhgt = gen.index_mat(ohgt[0,:,:], maxhgt)
+        hgtsxhgt = ophb[0,:, ijxhgt[0], ijxhgt[1]]/9.8
+
+        allhgtsea.append(hgtssea)
+        allxhgt.append(maxhgt)
+        allhgtxhgt.append(hgtsxhgt)
+        alldhgtsea.append(hgtssea[1:dimz]-hgtssea[0:dimz-1])
+        alldhgtxhgt.append(hgtsxhgt[1:dimz]-hgtsxhgt[0:dimz-1])
+        alletaw.append(oznw[0,:])
+        alletau.append(oznu[0,:])
+
+        onc.close()
+
+    # Legend
+    locleg, legfontsize = drw.legend_values(legvals,'|')
+
+    plt.plot_draw_WRFeta_levels(allhgtsea, allxhgt, allhgtxhgt, alldhgtsea,          \
+      alldhgtxhgt, alletaw, alletau, labs, colors, markers, [locleg, legfontsize],   \
+      imgtit, imgkind, close)
+
+    return
+
 #quit()
 

trunk/tools/drawing_tools.py

@@ -12742 +12742 @@
     return
 
+def plot_draw_WRFeta_levels(ahgtsea, axhgt, ahgtxhgt, adhgtsea, adhgtxhgt, aetaw,    \
+     imgtit, labels, cols, marks, legvs, figtitle, kfig, close):
+    """ Function to plot different sets of WRF eta-levels
+      ahgtsea: all hegights of a point above sea
+      axhgt: all maximum hegights
+      ahgtxhgt: all hegights of the point above mximum height
+      adhgtsea: all relative heights of a point above sea
+      adhgtxhgt: all relative heights of the point above maximum topo
+      aetaw: all eta-full values
+      aetau: all eta-half values
+      labels: labels
+      cols: colors
+      marks: markers
+      legvs: [legloc, legfs] legend values
+      figtitle: title of figure
+      kfig: kind of file of graphical output
+      close: whether figure should be closed
+    """
+    fname = 'plot_draw_WRFeta_levels'
+
+
+    plt.rc('text', usetex=True)
+
+    fig = plt.subplots(2)
+
+    plt.subplot(1,1,2)
+    # Absolute heights
+    Nlabs = len(labels)
+    for ilab in range(Nlabs):
+        hgtsea = ahgtsea[ilab]
+        hgtxhgt = ahgtxhgt[ilab]
+        znw = aetaw[ilab]
+
+        plt.plot(znw, hgtssea, '-', marker=marks[ilab], linewidth=1.5, markersize=4, \
+          label='$hgt_{sea}^{'+labels[ilab]+'}$', color=cols[ilab])
+        if ilab == 0:
+            plt.plot(znw, hgtsxhgt,  '-.', marker=marks[ilab], linewidth=1.5,        \
+              markersize=4, label='$hgt_{hgtmax}^{'+labels[ilab]+'}$',               \
+              color=colors[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('absolute height ($m$)', color='black')
+    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)
+    # Relative heights
+    Nlabs = len(labels)
+    for ilab in range(Nlabs):
+        hgtsea = adhgtsea[ilab]
+        hgtxhgt = adhgtxhgt[ilab]
+        znw = aetau[ilab]
+
+        plt.plot(znw, hgtssea, '-', marker=marks[ilab], linewidth=1.5, markersize=4, \
+          label='$\delta hgt_{sea}^{'+labels[ilab]+'}$', color=cols[ilab])
+        if ilab == 0:
+            plt.plot(znw, hgtsxhgt,  '-.', marker=marks[ilab], linewidth=1.5,        \
+              markersize=4, label='$\delta hgt_{hgtmax}^{'+labels[ilab]+'}$',        \
+              color=colors[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'
+    output_kind(kfig, figname, close)
+
+    return
+