Changeset 2603 in lmdz_wrf

Timestamp:

Jun 14, 2019, 5:37:52 PM (6 years ago)

Author:

lfita

Message:

Adding label characteristics in `drawpolygon_map'

File:

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

trunk/tools/drawing_tools.py

@@ -14449 +14449 @@
     return mapv
 
-def drawpolygon_map(vertices, mape, polylab, Nint, linev):
+def drawpolygon_map(vertices, mape, polylabv, Nint, linev):
     """ Function to plot a polygon into a map
       vertices: Matrix of vertices of the polygon [[lon1,lat1], ..., [lonM,latM]]
       mape: basemap map environtment where to draw the polygon
-      polylab: label of the polygon (already LaTeX freindly)
+      polylabv: [polylab, color, size, face, angle, pos]: values for the label of 
+          the polygon
+        polylab: label of the polygon (already LaTeX freindly, None for no label)
+        color: color of the characters
+        size: size of the font
+        face: weight of the letters
+          'normal': normal face
+          'bold': bold face
+        angle: angle of roation
+        pos: position as:
+          'c': center
+          'vert',Nvert: at coordinates of vertex number Nvert
+          'overt',xoffset,yoffset: at coordinates of vertex number Nvert aplying an
+            offset at each direction
       Nint: number of intermediate grid points to use in order to have lon, lat 
         following polygon borders
@@ -14459 +14472 @@
     """
     fname = 'drawpolygon_map'
+    availpolylabpos = ['c', "'vert',Nvert", "'osw',xoffset,yoffset"]
 
     Lvert = vertices.shape[0]
@@ -14480 +14494 @@
             if iv == 0 and ij == 0:
                 plt.plot([pptii[ij],pptii[ij+1]],[pptjj[ij],pptjj[ij+1]], linev[2],  \
-                  color=linev[0], linewidth=linev[1], label=polylab)
+                  color=linev[0], linewidth=linev[1], label=polylabv[0])
             else:
                 plt.plot([pptii[ij],pptii[ij+1]],[pptjj[ij],pptjj[ij+1]], linev[2],  \
@@ -14503 +14517 @@
           color=linev[0], linewidth=linev[1])
 
+    # Label location
+    polylab = polylabv[0]
+    if polylab is not None:
+        polylabpos = polylabv[5].split(',')
+        if polylabpos[0] == 'c':
+            xlabp = np.sum(vertices[:,0])
+            ylabp = np.sum(vertices[:,1])
+        elif polylabpos[0] == 'vert':
+            iv = int(polylabpos[1])
+            if iv > Lvert:
+                print errormsg
+                print '  ' + fname + ": provided vertex ", iv, 'is too large for ' + \
+                  'polygon with ', Lvert, 'vertices !!'
+                quit(-1)
+            xlabp = vertices[iv,0]
+            ylabp = vertices[iv,1]
+        elif polylabpos[0] == 'overt':
+            iv = int(polylabpos[1])
+            if iv > Lvert:
+                print errormsg
+                print '  ' + fname + ": provided vertex ", iv, 'is too large for ' + \
+                  'polygon with ', Lvert, 'vertices !!'
+                quit(-1)
+            xlabp = vertices[iv,0] + np.float(polylabpos[2])
+            ylabp = vertices[iv,1] + np.float(polylabpos[3])
+        else:
+            print errormsg
+            print '  ' + fname + ": position of the label '" + polylabpos[0] +       \
+              "' not available !!"
+            print '    available ones:', availpolylabpos
+            quit(-1)
+
+        if polylabv[3] == 'normal':        
+            plt.annotate(polylab, xy=(xlabp,ylabp), xycoords='data',                 \
+              color=polylabv[1], fontsize=polylabv[2], rotation=polylabv[4])
+        else:
+            plt.annotate(polylab, xy=(xlabp,ylabp), xycoords='data',                 \
+              color=polylabv[1], fontsize=polylabv[2], weight=polylabv[3],           \
+              rotation=polylabv[4])
+
+        print 'Lluis ' + fname + ': polylab', polylab, 'xy pos', xlabp, ylabp
+ 
     return