Changeset 429 in lmdz_wrf for trunk

Timestamp:

May 26, 2015, 1:03:19 PM (10 years ago)

Author:

lfita

Message:

Creation of function 'lonlat2D' to provide 2D lon,lat matrices

File:

• trunk/tools/nc_var_tools.py (modified) (15 diffs)

trunk/tools/nc_var_tools.py

@@ -10847 +10847 @@
     return dist
 
+def lonlat2D(lon,lat):
+    """ Function to return lon, lat 2D matrices from any lon,lat matrix
+      lon= matrix with longitude values
+      lat= matrix with latitude values
+    """
+    fname = 'lonlat2D'
+
+    if len(lon.shape) != len(lat.shape):
+        print errormsg
+        print '  ' + fname + ': longitude values with shape:', lon.shape,            \
+          'is different that latitude values with shape:', lat.shape, '(dif. size) !!'
+        quit(-1)
+
+    if len(lon.shape) == 3:
+        lonvv = lon[0,:,:]
+        latvv = lat[0,:,:]
+    elif len(lon.shape) == 2:
+        lonvv = lon[:]
+        latvv = lat[:]
+    elif len(lon.shape) == 1:
+        lonlatv = np.meshgrid(lon[:],lat[:])
+        lonvv = lonlatv[0]
+        latvv = lonlatv[1]
+
+    return lonvv, latvv
+
 def compute_tevolboxtraj(values, ncfile, varn):
     """ Function to compute tevolboxtraj: temporal evolution at a given point along 
@@ -10878 +10904 @@
     check_arguments(fname,len(arguments.split(',')),arguments,',',8)
 
-
     trajfile = values.split(',')[0].split('@')[0]
     Tbeg = int(values.split(',')[0].split('@')[1])
@@ -10931 +10956 @@
             varns = [varn]
 
-    if len(lonobj.shape) == 3:
-        lonv = lonobj[0,:,:]
-        latv = latobj[0,:,:]
-    elif len(lonobj.shape) == 2:
-        lonv = lonobj[:]
-        latv = latobj[:]
+    lonv, latv = lonlat2D(lonobj[:],latobj[:])
 
     dimx = lonv.shape[1]
@@ -10942 +10962 @@
 
     timv = timobj[:]
-
 
 # Selecting accordingly a trajectory
@@ -11133 +11152 @@
     newdim = objofile.createDimension('time', None)
 
-    statnames = ['minbox', 'maxbox', 'meanbox', 'mean2box', 'stdevbox']
+    stsn = ['min', 'max', 'mean', 'mean2', 'stdev', 'ac']
     vstlname = ['minimum value within', 'maximum value within', 'mean value within', \
-      'squared mean value within', 'standard deviation value within']
-    cstatnames = ['mincircle','maxcircle','meancircle','mean2circle','stdevcircle']
+      'squared mean value within', 'standard deviation value within',                \
+      'accumulated value within']
+    Nsts = len(stsn)
+    statnames = []
+    cstatnames = []
+    for i in range(Nsts):
+        statnames.append(stsn[i] + 'box')
+        cstatnames.append(stsn[i] + 'circle')
 
 # Getting values
@@ -11200 +11225 @@
             rlatvals = np.ones(tuple([Ttraj,dimz,Nrad*2+1,Nrad*2+1]), dtype=np.float)
 
-            statvarvals = np.ones(tuple([Ttraj,dimz,6]), dtype=np.float)
-            rstatvarvals = np.ones(tuple([Ttraj,dimz,6]), dtype=np.float)
+            statvarvals = np.ones(tuple([Ttraj,dimz,Nsts]), dtype=np.float)
+            rstatvarvals = np.ones(tuple([Ttraj,dimz,Nsts]), dtype=np.float)
 
             for it in range(Ttraj):
@@ -11248 +11273 @@
                         statvarvals[it,iz,5] = np.sqrt(statvarvals[it,iz,4] -        \
                           statvarvals[it,iz,3]*statvarvals[it,iz,3])
+                        statvarvals[it,iz,6] = np.sum(varvalst[iz,:,:])
+
                 else:
                     slicev.append(slice(0,dimz))
@@ -11280 +11307 @@
                         statvarvals[it,iz,5] = np.sqrt(statvarvals[it,iz,4] -        \
                           statvarvals[it,iz,3]*statvarvals[it,iz,3])
+                        statvarvals[it,iz,6] = np.sum(varvalst[iz,:,:])
 
 # Circle values
@@ -11324 +11352 @@
                         rstatvarvals[it,iz,5] = np.sqrt(rstatvarvals[it,iz,4] -      \
                           rstatvarvals[it,iz,3]*rstatvarvals[it,iz,3])
+                        rstatvarvals[it,iz,6] = np.sum(varvalst[iz,:,:])
 
                 else:
@@ -11364 +11393 @@
                         rstatvarvals[it,iz,5] = np.sqrt(rstatvarvals[it,iz,4] -      \
                           rstatvarvals[it,iz,3]*rstatvarvals[it,iz,3])
+                        rstatvarvals[it,iz,6] = maskedvals[iz,:,:].sum()
 
 #            print 'statistics:',rstatvarvals[it,:] 
@@ -11447 +11477 @@
             rlatvals = np.ones(tuple([Ttraj,Nrad*2+1,Nrad*2+1]), dtype=np.float)
 
-            statvarvals = np.ones(tuple([Ttraj,6]), dtype=np.float)
-            rstatvarvals = np.ones(tuple([Ttraj,6]), dtype=np.float)
+            statvarvals = np.ones(tuple([Ttraj,Nsts]), dtype=np.float)
+            rstatvarvals = np.ones(tuple([Ttraj,Nsts]), dtype=np.float)
 
             for it in range(Ttraj):
@@ -11489 +11519 @@
                     statvarvals[it,5] = np.sqrt(statvarvals[it,4] -                   \
                       statvarvals[it,3]*statvarvals[it,3])
+                    statvarvals[it,6] = maskedvals.sum()
 
                     varvalst[tuple(slice2D)] = lonv[tuple(slicevnoT)]
@@ -11523 +11554 @@
                     statvarvals[it,5] = np.sqrt(statvarvals[it,4] -                  \
                       statvarvals[it,3]*statvarvals[it,3])
+                    statvarvals[it,6] = np.sum(varvalst)
 
                     varvalst = lonv[tuple(slicevnoT)]
@@ -11561 +11593 @@
                     rstatvarvals[it,5] = np.sqrt(rstatvarvals[it,4] -                \
                       rstatvarvals[it,3]*rstatvarvals[it,3])
+                    rstatvarvals[it,6] = maskedvals2.sum()
 
                 else:
@@ -11593 +11626 @@
                     rstatvarvals[it,5] = np.sqrt(rstatvarvals[it,4] -                \
                       rstatvarvals[it,3]*rstatvarvals[it,3])
+                    rstatvarvals[it,6] = maskedvals.sum()
 
 #            print 'statistics:',rstatvarvals[it,:]