Changeset 2160 in lmdz_wrf

Timestamp:

Oct 4, 2018, 3:46:14 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `cyclevar_within': Function to tell if a given value is within a period 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) (6 diffs)

trunk/tools/generic_tools.py

@@ -92 +92 @@
 #   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)
+# cyclevar_within: Function to tell if a given value is within a period 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
@@ -14057 +14060 @@
     return cycvar[rescyc]
 
+def cyclevar_within(cycvar,bper, eper, val):
+    """ Function to tell if a given value is within a period 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, 6, 4)
+    True
+    >>> cyclevar(np.arange(12)+1, 12, 3, 2)
+    True
+    >>> cyclevar(np.arange(12)+1, 12, 3, 5)
+    False
+    """
+    fname = 'cyclevar_within'
+
+    dcyc = cycvar.shape[0]
+    # Indices of the beginning and end of the period
+    bind = index_vec(cycvar,bper)
+    eind = index_vec(cycvar,eper)
+
+    # Index of the value
+    vind = index_vec(cycvar,val)
+
+    if eind < bind: 
+        eind = eind + dcyc
+        vind = vind + dcyc
+
+    if vind >= bind and vind < eind: within = True
+    else: within = False
+
+    return within
+
 def time_slices(tv, tu, cal, per, amount):
     """ Function to return temporal slices of a series of times for a given amount 
@@ -14130 +14166 @@
         print 'cyctimes:', cyctimes
 
+        # First cycle time from first time
         iit = 0
-        itt = 0
+        itt = index_vec(cyctimes,mattimes[0,imat])
         ip = cyctimes[itt]
-        ep = cyctimes[itt+amount]
+        ep = cyclevar(cyctimes, itt+amount)
         for it in range(dimt):
             # timeslice: [ini_slice, end_slice, freq_slice]
@@ -14139 +14176 @@
             ttv = mattimes[it,imat]
             timeslice[1] = it
-            print it, ttv, 'ip:', ip, 'ep:', ep, '<>', (ttv >= ip and ttv < ep)
+            print it, '.', itt, '|', ttv, 'ip:', ip, 'ep:', ep, '<>', (ttv >= ip and ttv < ep)
             if not (ttv >= ip and ttv < ep):
                 slices.append(timeslice)
@@ -14191 +14228 @@
 tv = []
 values = []
-totT = 365.
-dT = 100.
+totT = 30.
+dT = 7.
 for it in range(0,10):
     for itt in range(3):
@@ -14201 +14238 @@
 itdim = 0
 tu = 'days since 1949-12-01 00:00:00'
-per = 'year'
+per = 'month'
 calend = 'standard'
 amount = 1