Changeset 2079 in lmdz_wrf for trunk/tools/drawing_tools.py

Timestamp:

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

Author:

lfita

Message:

Starting to add draw_WRFeta_levels

File:

• trunk/tools/drawing_tools.py (modified) (1 diff)

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
+