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

Timestamp:

Nov 27, 2018, 7:49:55 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `draw_river_pattern: Function to plot rivers' patterns from ORCHIDEE's routing scheme file ('routing.nc' and 'river_desc.nc')
• Fixing other issues

File:

• trunk/tools/drawing_tools.py (modified) (6 diffs)

trunk/tools/drawing_tools.py

@@ -189 +189 @@
 # plot_lines: Function to plot a collection of lines
 # plot_multiWindRose: Function to plot multi wind roses (from where the dinw blows)
+# plot_river_pattern: Function to plot ORCHIDEE's pattern or rivers (following trip 
+#    variable from 'routing.nc' and river description following 'river_desc.nc')
 # plot_SkewT: Function to plot a SkewT-logP diagram using matplotlib's API example 
 # plot_stations_map: Function to plot a map with a series of stations
@@ -1822 +1824 @@
 def legend_values(legstring, char):
     """ Function to determine the values for the legend
-      legstring= char separated list of [locleg][char][fontsize]: 
+      legstring= [char] separated list of [locleg][char][fontsize][char][ncol]: 
           [locleg]: location of the legend (0, automatic)
             1: 'upper right', 2: 'upper left', 3: 'lower left', 4: 'lower right',
@@ -7584 +7586 @@
   stn, stlon, stlat, drawcountry, drawbasinid, colordescid, coloriver, colornoriver, \
   colorst, stsize, graphtit, mapv, kfig, figname):
-    """ Function to plot vectors
+    """ Function to plot ORCHIDEE's pattern of one river (following trip variable from 
+        'routing.nc' and river description following 'river_desc.nc')
       xvals= values for the x-axis
       yvals= values for the y-axis
@@ -7590 +7593 @@
       veccolor= basin id
       descid= description id
-, desclon, desclat,
+      desclon= longitude of description
+      desclat= latitude of description
+      stn= name of the stations
+      stlon= longitude of the stations
+      stlat= latitude of the stations
       mapv= map characteristics: [proj],[res]
         see full documentation: http://matplotlib.org/basemap/
@@ -7604 +7611 @@
       drawcountry= whether country lines should be plotted or not
       drawbasinid= whether basins id should be plotted or not
+      colordescid= color map for the id of the rivers
+      coloriver= color for the non-selected rivers in the plot
+      colornoriver= color for the non-river paths values
+      colorst= color for the stations
+      stsize= color for the size of the labels of the stations
       graphtit= title of the graph ('|', for spaces)
       kfig= kind of figure
@@ -7794 +7806 @@
 
     return
+
+def plot_multi_river_pattern(xvals, yvals, fvals, veccolor, descids, desclon,        \
+  desclat, stn, stlon, stlat, drawcountry, drawbasinid, colordescid, coloriver,      \
+  colornoriver, pathwidth, colorst, stsize, graphtit, mapv, kfig, riverns):
+    """ Function to plot ORCHIDEE's pattern or multiple rivers (following trip variable from 
+        'routing.nc' and river description following 'river_desc.nc')
+      xvals= values for the x-axis
+      yvals= values for the y-axis
+      fvals= values for the flow (1-8: N, NE, E, ..., NW; 97: sub-basin; 98: small to sea; 99: large to sea)
+      veccolor= basin id
+      descids= dictionary with the description of each river
+      desclon= longitude of description
+      desclat= latitude of description
+      stn= name of the stations
+      stlon= longitude of the stations
+      stlat= latitude of the stations
+      mapv= map characteristics: [proj],[res]
+        see full documentation: http://matplotlib.org/basemap/
+        [proj]: projection
+          * 'cyl', cilindric
+          * 'lcc', lambert conformal
+        [res]: resolution:
+          * 'c', crude
+          * 'l', low
+          * 'i', intermediate
+          * 'h', high
+          * 'f', full
+      drawcountry= whether country lines should be plotted or not
+      drawbasinid= whether basins id should be plotted or not
+      colordescid= color map for the id of the rivers
+      coloriver= color for the non-selected rivers in the plot
+      colornoriver= color for the non-river paths values
+      pathwidth= width of the line with the river paths
+      colorst= color for the stations
+      stsize= color for the size of the labels of the stations
+      graphtit= title of the graph ('|', for spaces)
+      kfig= kind of figure
+      riverns= list with the name of the rivers to be plotted
+    """
+    fname = 'plot_multi_river_pattern'
+
+    figname = 'multi_river_pattern'
+
+    dx=xvals.shape[1]
+    dy=xvals.shape[0]
+    xmin = np.min(xvals)
+    xmax = np.max(xvals)
+    ymin = np.min(yvals)
+    ymax = np.max(yvals)
+
+# flow direction
+
+    plt.rc('text', usetex=True)
+
+    ddx = np.abs(xvals[dy/2+1,dx/2] - xvals[dy/2,dx/2])
+    ddy = np.abs(yvals[dy/2,dx/2+1] - yvals[dy/2,dx/2])
+    fontcharac = {'family': 'serif', 'weight': 'normal', 'size': 3}
+
+    xlabpos = []
+    ylabpos = []
+    labels = []
+    labcol = []
+    flow = []
+    flowvals = []
+    xtrack = []
+    ytrack = []
+    colortrack = []
+    trackid = []
+    stnlL = []
+    stlonlL = []
+    stlatlL = []
+
+# Setting up colors for each label
+#   From: http://stackoverflow.com/questions/15235630/matplotlib-pick-up-one-color-associated-to-one-value-in-a-colorbar
+    my_cmap = plt.cm.get_cmap(colordescid)
+    vcolmin = np.min(descids.values())
+    vcolmax = np.max(descids.values())
+
+    norm = mpl.colors.Normalize(vcolmin, vcolmax)
+
+# Vector angles
+    lengthtrac = np.float(xvals[0,1] - xvals[0,0])
+    lengthtrac2 = lengthtrac*np.sqrt(2.)
+
+    print '  ' + fname + ': getting colors for the river-paths ...'
+    for j in range(0,dy):
+        for i in range(0,dx):
+            if veccolor[j,i] != '--':
+#                ibeg = xvals[j,i]-lengthtrac/2.
+#                jbeg = yvals[j,i]-lengthtrac/2.
+                ibeg = xvals[j,i]
+                jbeg = yvals[j,i]
+                labels.append(int(veccolor[j,i]))
+                flowvals.append(fvals[j,i])
+                angle = (fvals[j,i] - 1)*np.pi/4
+                if gen.searchInlist([2,4,6,8], fvals[j,i]):
+                    iend = ibeg + lengthtrac2*np.sin(angle)
+                    jend = jbeg + lengthtrac2*np.cos(angle)
+                elif gen.searchInlist([1,3,5,7], fvals[j,i]):
+                    iend = ibeg + lengthtrac*np.sin(angle)
+                    jend = jbeg + lengthtrac*np.cos(angle)
+                else:
+                    ibeg = xvals[j,i]
+                    jbeg = yvals[j,i]
+                    iend = None
+                    jend = None
+
+                xlabpos.append(ibeg)
+                ylabpos.append(jbeg)
+                xtrack.append(ibeg)
+                xtrack.append(iend)
+                xtrack.append(None)
+                ytrack.append(jbeg)
+                ytrack.append(jend)
+                ytrack.append(None)
+                if len(desclon.shape) == 2:
+                    difflonlat = np.sqrt((desclon-xvals[j,i])**2 + (desclat-yvals[j,i])**2)
+                    mindiffLl = np.min(difflonlat)
+                    ilatlon = gen.index_mat(difflonlat, mindiffLl)
+                else:
+                    ilatlon = np.zeros((2), dtype=int)
+                    difflon = np.abs(desclon - xvals[j,i])
+                    mindiffl = np.min(difflon)
+                    ilatlon[1] = gen.index_vec(difflon,mindiffl)
+                    difflat = np.abs(desclat - yvals[j,i])
+                    mindiffL = np.min(difflat)
+                    ilatlon[0] = gen.index_vec(difflat,mindiffL)
+
+                if not gen.searchInlist(descids.values(), veccolor[j,i]):
+                    labcol.append(colornoriver)
+                    colortrack.append(colornoriver)
+#                        print 'Lluis: veccol:', veccolor[j,i], 'mindiffl', mindiffl,'mindiffL:',mindiffL
+                else:
+#                        print 'Lluis right! mindiffl', mindiffl,'mindiffL:',mindiffL
+                    labcol.append(coloriver)
+                    colortrack.append(my_cmap(norm(veccolor[j,i])))
+
+                trackid.append(veccolor[j,i])
+
+    totvals = len(flowvals)
+
+    if not mapv is None:
+        lon00 = np.where(xvals[:] < 0., 360. + xvals[:], xvals[:])
+        lat00 = yvals[:]
+
+        map_proj=mapv.split(',')[0]
+        map_res=mapv.split(',')[1]
+
+        nlon = np.min(xvals)
+        xlon = np.max(xvals)
+        nlat = np.min(yvals)
+        xlat = np.max(yvals)
+
+        lon2 = xvals[dy/2,dx/2]
+        lat2 = yvals[dy/2,dx/2]
+
+        print 'lon2:', lon2, 'lat2:', lat2, 'SW pt:', nlon, ',', nlat, 'NE pt:',     \
+          xlon, ',', xlat
+
+        if map_proj == 'cyl':
+            m = Basemap(projection=map_proj, llcrnrlon=nlon, llcrnrlat=nlat,         \
+              urcrnrlon=xlon, urcrnrlat= xlat, resolution=map_res)
+        elif map_proj == 'lcc':
+            m = Basemap(projection=map_proj, lat_0=lat2, lon_0=lon2, llcrnrlon=nlon, \
+              llcrnrlat=nlat, urcrnrlon=xlon, urcrnrlat= xlat, resolution=map_res)
+        else:
+            print errormsg
+            print '  ' + fname + ": projection '" + map_proj + "' not ready!!"
+            print '    projections available: cyl, lcc'
+            quit(-1)
+
+        m.drawcoastlines()
+        drawcountry = True
+        if drawcountry: m.drawcountries()
+
+        meridians = gen.pretty_int(nlon,xlon,5)
+        m.drawmeridians(meridians,labels=[True,False,False,True],color="black")
+
+        parallels = gen.pretty_int(nlat,xlat,5)
+        m.drawparallels(parallels,labels=[False,True,True,False],color="black")
+
+    Npaths = len(xlabpos)
+    print '  ' + fname + ' plotting ', Npaths ,'river-paths ...'
+    j = 0
+    for i in range(Npaths):
+        gen.percendone(i,Npaths,5.,'  ' + fname + ': drawn')
+        plt.plot(xtrack[j:j+2], ytrack[j:j+2], color=colortrack[i],                  \
+          linewidth=pathwidth)
+        j = j + 3
+
+# Sea-flow
+    for i in range(len(xlabpos)):
+        if flowvals[i] == 97:
+            plt.plot(xlabpos[i], ylabpos[i], 'x', color=colortrack[i],               \
+              markersize=lengthtrac2)
+        elif flowvals[i] == 98:
+            plt.plot(xlabpos[i], ylabpos[i], '*', color=colortrack[i],               \
+              markersize=lengthtrac2)
+        elif flowvals[i] == 99:
+            plt.plot(xlabpos[i], ylabpos[i], 'h', color=colortrack[i],               \
+              markersize=lengthtrac2)
+
+    plt.xlim(xmin,xmax)
+    plt.ylim(ymin,ymax)
+
+    if drawbasinid:
+        print '  ' + fname + ' drawing basins ids ...'
+        idlabelled = []
+        for i in range(len(xlabpos)):
+            if not gen.searchInlist(idlabelled, labels[i]):
+                plt.text(xlabpos[i]+0.05*lengthtrac, ylabpos[i]+0.05*lengthtrac,     \
+                  labels[i], color=labcol[i], fontdict=fontcharac)
+                idlabelled.append(labels[i])
+                
+#            plt.text(xlabpos[i]+0.1*lengthtrac, ylabpos[i]+0.1*lengthtrac, trackid[i],   \
+#              color=colortrack[i], fontdict=fontcharac)
+#            plt.text(xlabpos[i]+0.1*lengthtrac, ylabpos[i]+0.1*lengthtrac, str(flowvals[i]),   \
+#              color=colortrack[i], fontdict=fontcharac)
+
+    if stn is not None:
+        for ist in range(len(stlon)):
+            if stlon[ist] > xmin and stlon[ist] < xmax and stlat[ist] > ymin and     \
+              stlat[ist] < ymax:
+                stname = '*'
+                for ic in range(len(stn[ist,:])):
+                    stname  = stname + stn[ist,ic]
+#            plt.annotate(stname, xy=(stlon[ist],stlat[ist]), color=colorst, )
+                plt.text(stlon[ist], stlat[ist], stname, color=colorst, fontsize=stsize)
+
+#    cbar = plt.colorbar()
+    plt.title(graphtit.replace('&','\&'))
+
+    output_kind(kfig, figname, True)
+
+    return
+
 
 def var_3desc(ovar):