Changeset 2171 in lmdz_wrf for trunk

Timestamp:

Oct 8, 2018, 3:28:58 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `mapFigure': Function to provide the environment for a map

File:

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

trunk/tools/drawing_tools.py

@@ -154 +154 @@
 # check_arguments:
 # Str_Bool:
+# mapFigure: Function to provide the environment for a map
 # movie_2D_shadow: Plot of a 2D field with shadow
 # multi_plot_2D_shadow: plot of multiple 2D shadowing variables sharing colorbar
@@ -13774 +13775 @@
     return
 
+def mapFigure(lon0, lat0, lonlatbox, mapvals, Nlon, Nlat):
+    """ Function to provide the environment for a map
+      lon0: 2D longitudes
+      lat0: 2D latitudes
+      lonlatbox: [SWlon, SWlat, NElon, NElat] box or None
+      mapvals: values for the map
+      Nlon: Quantity of meridians to plot
+      Nlat: Quantity of parallels to plot
+    """
+    fname = 'mapFigure'
+
+    lon, lat = gen.lonlat2D(lon0, lat0)
+
+    map_proj=mapvals.split(',')[0]
+    map_res=mapvals.split(',')[1]
+
+    if lonlatbox is not None:
+        nlon = lonlatbox[0]
+        nlat = lonlatbox[1]
+        xlon = lonlatbox[2]
+        xlat = lonlatbox[3]
+    else:
+        nlon = np.min(lon)
+        xlon = np.max(lon)
+        nlat = np.min(lat)
+        xlat = np.max(lat)
+
+    lon2 = (xlon+nlon)/2.
+    lat2 = (xlat+nlat)/2.
+
+    print 'lon2:', lon2, 'lat2:', lat2, 'SW pt:', nlon, ',', nlat, 'NE pt:',     \
+      xlon, ',', xlat
+
+    if map_proj == 'cyl':
+      mapenv = Basemap(projection=map_proj, llcrnrlon=nlon, llcrnrlat=nlat,          \
+        urcrnrlon=xlon, urcrnrlat= xlat, resolution=map_res)
+    elif map_proj == 'lcc':
+      mapenv = Basemap(projection=map_proj, lat_0=lat2, lon_0=lon2, llcrnrlon=nlon,  \
+        llcrnrlat=nlat, urcrnrlon=xlon, urcrnrlat= xlat, resolution=map_res)
+    else:
+      print gen.errormsg
+      print '  ' + fname + ": map projection '" + map_proj + "' not defined!!!"
+      print '    available: cyl, lcc'
+      quit(-1)
+
+    merids = gen.pretty_int(nlon, xlon, Nlon)
+    parals = gen.pretty_int(nlat, xlat, Nlat)
+
+    xmap,ymap = mapenv(lon,lat)
+
+    return mapenv, merids, parals, xmap, ymap
+