Changeset 2374 in lmdz_wrf

Timestamp:

Mar 5, 2019, 2:19:03 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `drawpolygon_map': Function to plot a polygon into a map

File:

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

trunk/tools/drawing_tools.py

@@ -127 +127 @@
 # draw_MerPara: Function to add the parallels and meridians into a figure with a 
 #   panel of plots
+# drawpolygon_map: Function to plot a polygon into a map
 # figureMap_size: Function to determine the real size of a figure adjusted to the 
 #   ratio of the maps
@@ -14383 +14384 @@
 
     return mapv
+
+def drawpolygon_map(vertices, mape, polylab, 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)
+      Nint: number of intermediate grid points to use in order to have lon, lat 
+        following polygon borders
+      linev: list with the values for the line to be drawn [color, width, type]
+    """
+    fname = 'drawpolygon_map'
+
+    Lvert = vertices.shape[0]
+    for iv in range(Lvert-1):
+        pti = [vertices[iv,0], vertices[iv,1]]
+        pte = [vertices[iv+1,0], vertices[iv+1,1]]
+        ddx = (pte[0]-pti[0])/(Nint-1)
+        ddy = (pte[1]-pti[1])/(Nint-1)
+        if ddx != 0.:
+            ipti = np.arange(pti[0],pte[0]+ddx,ddx)
+        else:
+            ipti = np.ones((Nint), dtype=np.float)*pti[0]
+        if ddy != 0.:
+            iptj = np.arange(pti[1],pte[1]+ddy,ddy)
+        else:
+            iptj = np.ones((Nint), dtype=np.float)*pti[1]
+        pptii, pptjj = mape(ipti[0:Nint], iptj[0:Nint])
+        for ij in range(Nint-1):
+            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)
+            else:
+                plt.plot([pptii[ij],pptii[ij+1]],[pptjj[ij],pptjj[ij+1]], linev[2],  \
+                  color=linev[0], linewidth=linev[1])
+    pti = [vertices[Lvert-1,0], vertices[Lvert-1,1]]
+    pte = [vertices[0,0], vertices[0,1]]
+    ddx = (pte[0]-pti[0])/(Nint-1)
+    ddy = (pte[1]-pti[1])/(Nint-1)
+    if ddx != 0.:
+        ipti = np.arange(pti[0],pte[0]+ddx,ddx)
+    else:
+        ipti = np.ones((Nint), dtype=np.float)*pti[0]
+    if ddy != 0.:
+        iptj = np.arange(pti[1],pte[1]+ddy,ddy)
+    else:
+        iptj = np.ones((Nint), dtype=np.float)*pti[1]
+    pptii, pptjj = mape(ipti[0:Nint], iptj[0:Nint])
+    for ij in range(Nint-1):
+        plt.plot([pptii[ij],pptii[ij+1]],[pptjj[ij],pptjj[ij+1]], linev[2],          \
+          color=linev[0], linewidth=linev[1])
+
+    return