Changeset 2159 in lmdz_wrf for trunk/tools

Timestamp:

Oct 4, 2018, 3:24:44 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `# cyclevar: Function to provide the given index of a cycle variable. A cycle variable is a given structure with a series of values, once the maximum is overpassed it restarts from the beginning (e.g.: 12 months of a year)

File:

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

trunk/tools/generic_tools.py

@@ -89 +89 @@
 #   i,j-quads within which the curve lays (-1, no value for the given quad [2x2 points around]) and provide 
 #   the weights of the quad to perform a distance-weighted mean at the given curve point
+# cyclevar: Function to provide the given index of a cycle variable. A cycle variable
+#   is a given structure with a series of values, once the maximum is overpassed
+#   it restarts from the beginning (e.g.: 12 months of a year)
 # DateTimeStr_date: Function to transform a string date-time ([YYYY][MM][DD][HH][MI][SS] format) to a date object
 # datetimeStr_datetime: Function to transform a string date-time ([YYYY]-[MM]-[DD]_[HH]:[MI]:[SS] format) to a date object
@@ -3001 +3004 @@
     fname = 'CFtimes_datetime'
 
+    availtunits = ['weeks', 'days', 'hours', 'minutes', 'seconds', 'miliseconds']
+
     times = ncfile.variables[tname]
     timevals = times[:]
@@ -3083 +3088 @@
         else:
               print errormsg
-              print '    CFtimes_datetime: time units "' + tunits + '" not ready!!!!'
+              print fname + '    : time units "' + tunits + '" not ready!!!!'
+              print '    avaialable ones:', availtunits
               quit(-1)
     else:
@@ -3142 +3148 @@
         else:
               print errormsg
-              print '    CFtimes_datetime: time units "' + tunits + '" not ready!!!!'
+              print '    ' + fname + ': time units "' + tunits + '" not ready!!!!'
               quit(-1)
     
@@ -3163 +3169 @@
     import datetime as dt
     fname = 'CFtimesvar_datetime'
+
+    availtunits = ['weeks', 'days', 'hours', 'minutes', 'seconds', 'miliseconds']
 
     txtunits = units.split(' ')
@@ -3232 +3240 @@
         else:
               print errormsg
-              print '    CFtimes_datetime: time units "' + tunits + '" not ready!!!!'
+              print '    ' + fname + ': time units "' + tunits + '" not ready!!!!'
+              print '    available ones:', availtunits
               quit(-1)
     else:
@@ -3291 +3300 @@
         else:
               print errormsg
-              print '    CFtimes_datetime: time units "' + tunits + '" not ready!!!!'
+              print '    ' + fname + ': time units "' + tunits + '" not ready!!!!'
+              print '    available ones:', availtunits
+
               quit(-1)
     
@@ -14027 +14038 @@
 #print WRF_percenlevels(38)
 
+def cyclevar(cycvar,ind):
+    """ Function to provide the given index of a cycle variable. A cycle variable
+        is a given structure with a series of values, once the maximum is overpassed
+        it restarts from the beginning (e.g.: 12 months of a year)
+      cycvar: array of values
+      ind: index to provide the value from
+    >>> cyclevar(np.arange(12)+1, 3)
+    4
+    >>> cyclevar(np.arange(12)+1, 14)
+    3
+    """
+    fname = 'cyclevar'
+
+    dcyc = cycvar.shape[0]
+    Ncyc = int(ind / dcyc)
+    rescyc = ind - Ncyc*dcyc
+
+    return cycvar[rescyc]
+
 def time_slices(tv, tu, cal, per, amount):
     """ Function to return temporal slices of a series of times for a given amount 
@@ -14044 +14074 @@
     fname = 'time_slices'
 
+    availper = ['year', 'month', 'day', 'hour', 'minute']
+
     mattimes = CFtimesvar_datetime(tv, tu, cal)
-    print 'CF times _______'
+    print 'mat times _______'
     print mattimes
 
@@ -14061 +14093 @@
     if per == 'year':
         imat = 0
+        # Getting unique values, since there is no cycle year periods
+        timatvals = list(mattimes[:,imat])
+        setvar = set(timatvals)
+        cyctimes = list(setvar)
+        cyctimes.sort()
+        Ncyctimes = len(cyctimes)
+        cyctimes.append(cyctimes[Ncyctimes-1]+1)
+        Ncyctimes = Ncyctimes + 1
+
+        print 'cyctimes:', cyctimes
+
+        iit = 0
+        itt = 0
+        ip = cyctimes[itt]
+        ep = cyctimes[itt+amount]
+        for it in range(dimt):
+            # timeslice: [ini_slice, end_slice, freq_slice]
+            timeslice = [iit,iit,1]
+            ttv = mattimes[it,imat]
+            timeslice[1] = it
+            print it, ttv, 'ip:', ip, 'ep:', ep, '<>', (ttv >= ip and ttv < ep)
+            if not (ttv >= ip and ttv < ep):
+                slices.append(timeslice)
+                iit = it + 0
+                itt = itt + amount
+                ip = cyctimes[itt]
+                ep = cyctimes[np.min([Ncyctimes-1,itt+amount])]
+
+    elif per == 'month':
+        imat = 1
+
+        # cycle for months
+        Ncyctimes = 12
+        cyctimes = np.arange(Ncyctimes)+1
+        
+        print 'cyctimes:', cyctimes
+
+        iit = 0
+        itt = 0
+        ip = cyctimes[itt]
+        ep = cyctimes[itt+amount]
+        for it in range(dimt):
+            # timeslice: [ini_slice, end_slice, freq_slice]
+            timeslice = [iit,iit,1]
+            ttv = mattimes[it,imat]
+            timeslice[1] = it
+            print it, ttv, 'ip:', ip, 'ep:', ep, '<>', (ttv >= ip and ttv < ep)
+            if not (ttv >= ip and ttv < ep):
+                slices.append(timeslice)
+                iit = it + 0
+                itt = itt + amount
+                ip = cyclevar(cyctimes,itt)
+                ep = cyclevar(cyctimes,itt+amount)
+
+    elif per == 'day':
+        imat = 2
         # Getting unique values
         years = list(mattimes[:,imat])
         setvar = set(years)
         utimes = list(setvar)
+        utimes.sort()
         Nutimes = len(utimes)
 
         # introducing amounts
-        print utimes, ':', Nutimes
-        atimes = np.arange(utimes[0], utimes[Nutimes-1], amount)
+        print '  unique ', Nutimes, 'times:', utimes, '<>', utimes[0], utimes[Nutimes-1]+amount, amount
+        atimes = np.arange(utimes[0], utimes[Nutimes-1]+amount, amount)
         Ntimes = len(atimes)
+        print '  amount ', Ntimes, 'periods:', atimes
         
         itt = 0
@@ -14083 +14173 @@
                     timeslice[1] = it
                 else:
+                    timeslice[1] = it
                     slices.append(timeslice)
                     itt = it + 0
                     break
+    else:
+        print errormsg
+        print '  ' + fname + ": period '" + per + "' not ready !!"
+        print '    available ones:', availper
+        quit(-1)
 
     Nslices = len(slices)
     for isl in range(Nslices):
-        print atimes[isl], atimes[isl+1], ':', slices[isl]
-
-    return slices
+        print isl, cyctimes[isl], cyctimes[isl+amount+1], ':', slices[isl]
+
+    return slices, Nslices
 
 tv = []
+values = []
+totT = 365.
+dT = 100.
 for it in range(0,10):
-    tv.append(it*365)
-    tv.append(it*365+180.)
-    tv.append(it*365+360.)
-
+    for itt in range(3):
+        tv.append(it*totT+dT*itt)
+        values.append(it*1.)
+
+vals = np.array(values)
+itdim = 0
 tu = 'days since 1949-12-01 00:00:00'
 per = 'year'
@@ -14104 +14205 @@
 amount = 1
 
-time_slices(tv, tu, calend, per, amount)
-
+tslc, Ntslc = time_slices(tv, tu, calend, per, amount)
+
+statn = 'mean'
+tstat = np.ones((Ntslc), dtype=np.float)
+for islc in range(Ntslc):
+    timeslcv = tslc[islc]
+    timeslc = [slice(timeslcv[0],timeslcv[1],timeslcv[2])]
+    tvals = vals[tuple(timeslc)]
+    if statn == 'min':
+        tstat[islc] = np.min(tvals, axis=itdim)
+    elif statn == 'max':
+        tstat[islc] = np.max(tvals, axis=itdim)
+    elif statn == 'mean':
+        tstat[islc] = np.mean(tvals, axis=itdim)
+    elif statn == 'mean2':
+        tstat[islc] = np.mean(tvals**2, axis=itdim)
+    elif statn == 'std':
+        tstat[islc] = np.std(tvals, axis=itdim)
+
+    timeslc = tslc[islc]
+    itime = tv[timeslc[0]]
+    etime = tv[timeslc[1]]
+    print islc, ':', itime, '-', etime, '<>', tvals, '=', tstat[islc]
+
+   
 #quit()