Changeset 780 in lmdz_wrf for trunk/tools

Timestamp:

May 27, 2016, 5:19:37 PM (9 years ago)

Author:

lfita

Message:

Adding `ijlonlat': Function to provide the imin,jmin imax,jmax of a lon,lat box

File:

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

trunk/tools/generic_tools.py

@@ -19 +19 @@
 
 ####### Content
+# ijlonlat: Function to provide the imin,jmin imax,jmax of a lon,lat box
 # PolyArea: Function to compute the area of the polygon following 'Shoelace formula'
 
@@ -6245 +6246 @@
     return 0.5*np.abs(np.dot(x,np.roll(y,1))-np.dot(y,np.roll(x,1)))
 
+def ijlonlat(lon, lat, nlon, xlon, nlat, xlat):
+    """ Function to provide the imin,jmin imax,jmax of a lon,lat box
+      lon= 2D matrix with the longitudes
+      lat= 2D matrix with the latitudes
+      nlon, xlon= minimun, maximum of the longitudes for the lon,lat box
+      nlat, xlat= minimun, maximum of the latitudes for the lon,lat box
+    """
+    fname = 'ijlonlat'
+
+    if nlon < np.min(lon):
+        print errormsg
+        print '  ' + fname + ': minimum longitude:',nlon,'too small!!'
+        print '    smallest possible:',np.min(lon)
+        quit(-1)
+    if xlon > np.max(lon):
+        print errormsg
+        print '  ' + fname + ': maximum longitude:',xlon,'too large!!'
+        print '    largest possible:',np.max(lon)
+        quit(-1)
+    if nlat < np.min(lat):
+        print errormsg
+        print '  ' + fname + ': minimum laitude:',nlat,'too small!!'
+        print '    smallest possible:',np.min(lat)
+        quit(-1)
+    if xlat > np.max(lat):
+        print errormsg
+        print '  ' + fname + ': maximum latitude:',xlat,'too large!!'
+        print '    largest possible:',np.max(lat)
+        quit(-1)
+
+    nlonv = np.abs(lon - nlon)
+    xlonv = np.abs(lon - xlon)
+    nlatv = np.abs(lat - nlat)
+    xlatv = np.abs(lat - xlat)
+
+    ni0 = index_mat(nlonv,np.min(nlonv))[1]
+    xi0 = index_mat(xlonv,np.min(xlonv))[1] + 1
+    nj0 = index_mat(nlatv,np.min(nlatv))[0]
+    xj0 = index_mat(xlatv,np.min(xlatv))[0] + 1
+
+    if ni0 > xi0:
+        ni = xi0
+        xi = ni0
+    else:
+        ni = ni0
+        xi = xi0
+
+    if nj0 > xj0:
+        nj = xj0
+        xj = nj0
+    else:
+        nj = nj0
+        xj = xj0
+
+    return ni,xi,nj,xj
+