Changeset 931 in lmdz_wrf for trunk/tools

Timestamp:

Jun 23, 2016, 5:09:37 PM (8 years ago)

Author:

lfita

Message:

Adding `transform': Function to transform the values and the axes

File:

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

trunk/tools/drawing_tools.py

@@ -69 +69 @@
 # plot_lines: Function to plot a collection of lines
 # plot_ZQradii: Function to plot following radial averages only at exact grid poins
+# transform: Function to transform the values and the axes
 
 # From nc_var_tools.py
@@ -3976 +3977 @@
 #quit()
 
-def transform(vals, dxv, dyv, dxt, dyt, dxl, dyl, dxtit, dytit, trans):
+def transform(vals, trans, dxv= None, dyv= None, dxt= None, dyt= None, dxl= None,    \
+  dyl= None, dxtit= None, dytit= None):
     """ Function to transform the values and the axes
       vals= values to transform
+      trans= '|' separated list of operations of transformation
+        'transpose': Transpose matrix of values (x-->y, y-->x)
+        'flip@[x/y]': Flip the given axis
       d[x/y]v= original values for the [x/y]-axis
       d[x/y]t= original ticks for the [x/y]-axis
       d[x/y]l= original tick-labels for the [x/y]-axis
       d[x/y]tit= original titels for the [x/y]-axis
-      trans= '|' separated list of operations of transformation
-        'transpose': Transpose matrix of values (x-->y, y-->x)
-        'flip@[x/y]': Flip the given axis
+
+    >>> mat = np.arange(15).reshape(3,5)
+    >>> xvals = np.zeros((3,5), dtype=np.float)
+    >>> yvals = np.zeros((3,5), dtype=np.float)
+    >>> for i in range(5):
+    >>>     xvals[:,i] = -1.25 + i*0.5
+    >>> for j in range(3):
+    >>>     yvals[j,:] = 40.25 + j*0.5
+    >>> xticks = np.array([-1.25, -0.25, 0.25])
+    >>> yticks = np.array([40.25, 41.25])
+    >>> xticklab = ['1.25 W', '0.25 W', '0,25 E']
+    >>> yticklab = ['40.25 N', '41.25 N']
+    >>> xtit = 'Longitude'
+    >>> ytit = 'Latitude'
+    >>> nv, ndxv, ndyv, ndxt, ndyt, dxl, ndyl, ndxT, ndyT = transform(mat, 'transpose', \
+      xvals, yvals, xticks, yticks, xticklab, yticklab, xtit, ytit)
+    [[ 0  5 10]
+     [ 1  6 11]
+     [ 2  7 12]
+     [ 3  8 13]
+     [ 4  9 14]]
+    [[ 40.25  40.75  41.25]
+     [ 40.25  40.75  41.25]
+     [ 40.25  40.75  41.25]
+     [ 40.25  40.75  41.25]
+     [ 40.25  40.75  41.25]]
+    [[-1.25 -1.25 -1.25]
+     [-0.75 -0.75 -0.75]
+     [-0.25 -0.25 -0.25]
+     [ 0.25  0.25  0.25]
+     [ 0.75  0.75  0.75]] 
+    [ 40.25  41.25] 
+    [-1.25 -0.25  0.25] 
+    ['41.25 N', '40.25 N'] 
+    ['0,25 E', '0.25 W', '1.25 W'] 
+    Latitude 
+    Longitude
     """
     fname = 'transform'
 
-    return newvals, newdxv, newdyv
+    transforms = trans.split('|')
+    Ntransforms = len(transforms)
+
+    newvals = vals.copy()
+    dv = False
+    if dxv is not None:
+        newdxv = dxv.copy()
+        newdyv = dyv.copy()
+        dv = True
+    else:
+        newdxv = None
+        newdyv = None
+
+    dt = False
+    if dxt is not None:
+        newdxt = dxt.copy()
+        newdyt = dyt.copy()
+        dt = True
+    else:
+        newdxt = None
+        newdyt = None
+
+    dl = False
+    if dxt is not None:
+        newdxl = list(dxl)
+        newdyl = list(dyl)
+        dl = True
+    else:
+        newdxl = None
+        newdyl = None
+
+    dT = False
+    if dxt is not None:
+        newdxtit = str(dxtit)
+        newdytit = str(dytit)
+        dT = True
+    else:
+        newdxtit = None
+        newdytit = None
+
+    dx = newvals.shape[0]
+    dy = newvals.shape[1]
+
+    for transform in transforms:
+        if transform == 'transpose':
+            newvals = np.transpose(newvals)
+            if dv:
+                copy = newdxv.copy()
+                newdxv = np.transpose(newdyv)
+                newdyv = np.transpose(copy)
+            if dt:
+                copy = newdxt.copy()
+                newdxt = np.transpose(newdyt)
+                newdyt = np.transpose(copy)
+            if dl:
+                copy = list(newdxl)
+                newdxl = list(newdyl[::-1])
+                newdyl = list(copy[::-1])
+            if dT:
+                copy = str(newdxtit)
+                newdxtit = str(newdytit)
+                newdytit = str(copy)
+
+        elif transform[0:4] == 'flip':
+            flip = transform.split('@')[1]
+            if flip == 'x':
+                newvals = newvals[...,::-1]
+                if dv:
+                    newdxv = newdxv[...,::-1]
+                if dt:
+                    newdxt = newdxt[::-1]
+                if dl:
+                    newdxl = newdxl[::-1]
+            elif flip == 'y':
+                newvals = newvals[...,::-1,:]
+                if dv:
+                    newdyv = newdyv[...,::-1,:]
+                if dt:
+                    newdyt = newdyt[::-1]
+                if dl:
+                    newdyl = newdyl[::-1]
+            elif flip == 'z':
+                newvals = newvals[...,::-1,:,:]
+        else:
+            print errormsg
+            print '  '+ fname + ": transformation '" + transform + "' not ready!!"
+            quit(-1)
+
+    return newvals, newdxv, newdyv, newdxt, newdyt, newdxl, newdyl, newdxtit, newdytit
 
 def plot_2D_shadow_time(varsv,vnames,dimxv,dimyv,dimxu,dimyu,dimn,colorbar,vs,uts,   \
@@ -4722 +4849 @@
 # contour
 ##
+    print fname + '; Lluis: shapes x:', x.shape,'y:', y.shape, 'varcv:', varcv.shape
     contkind = ckind.split(',')[0]
     if contkind == 'cmap':
@@ -4731 +4859 @@
     elif  contkind == 'fixsigc':
         coln = ckind.split(',')[1]
-        cplot = plt.contour(x, y, varcv, levels=vc, colors=coln)
+        cplot = plt.contour(x, y, varcv.transpose(), levels=vc, colors=coln)
     else:
         print errormsg