Changeset 399 for trunk/UTIL/PYTHON/myplot.py

Timestamp:

Nov 18, 2011, 8:43:59 PM (13 years ago)

Author:

aslmd

Message:

GRAPHICS: improved the selection of lon,lat with pp.py for mesoscale files

File:

• trunk/UTIL/PYTHON/myplot.py (modified) (3 diffs)

trunk/UTIL/PYTHON/myplot.py

@@ -2 +2 @@
 def errormess(text,printvar=None):
     print text
-    if printvar: print printvar 
+    if printvar is not None: print printvar 
     exit()
     return
@@ -829 +829 @@
   return zesaxis
 
+## Author: AS
+def bidimfind(lon2d,lat2d,vlon,vlat):
+   import numpy as np
+   if vlat is None:    array = (lon2d - vlon)**2
+   elif vlon is None:  array = (lat2d - vlat)**2
+   else:               array = (lon2d - vlon)**2 + (lat2d - vlat)**2
+   idy,idx = np.unravel_index( np.argmin(array), lon2d.shape )
+   if vlon is not None:
+       #print lon2d[idy,idx],vlon
+       if (np.abs(lon2d[idy,idx]-vlon)) > 5: errormess("longitude not found ",printvar=lon2d)
+   if vlat is not None:
+       #print lat2d[idy,idx],vlat
+       if (np.abs(lat2d[idy,idx]-vlat)) > 5: errormess("latitude not found ",printvar=lat2d)
+   return idx,idy
+
 ## Author: TN
-def  getsindex(saxis,index,axis):
+def getsindex(saxis,index,axis):
 # input  : all the desired slices and the good index
 # output : all indexes to be taken into account for reducing field
@@ -890 +905 @@
    print "CHECK: x.shape, y.shape", x.shape, y.shape
    if len(shape) == 1:
-      print shape[0]
       if shape[0] != len(x):
          print "WARNING HERE !!!"
          x = y
    elif len(shape) == 2:
-      print shape[1], len(y), shape[0], len(x)
       if shape[1] == len(y) and shape[0] == len(x) and shape[0] != shape[1]:
          what_I_plot = swapaxes(what_I_plot,0,1)
          print "INFO: swapaxes", what_I_plot.shape, shape
-         #x0 = x
-         #x = y
-         #y = x0
-   #print "define_axis", x, y
    return what_I_plot,x,y