Changeset 942 in lmdz_wrf for trunk/tools

Timestamp:

Jun 24, 2016, 11:12:47 AM (8 years ago)

Author:

lfita

Message:

Adding `CFmonthU_daysU': Function to transform from a CF date series with units as 'months since [DATE]' to 'days since [DATE]'

File:

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

trunk/tools/generic_tools.py

@@ -19 +19 @@
 
 ####### Content
+# CFmonthU_daysU: Function to transform from a CF date series with units as 'months since [DATE]' to 'days since [DATE]'
 # CFvar_DIAGvar: Function to provide which model diagnostic values can provide a CF-variable from ASCII file
 # CFvar_MODvar: Function to provide which model values can provide a CF-variable from ASCII file
@@ -7922 +7923 @@
         return varv
 
+def CFmonthU_daysU(datev,tunits):
+    """ Function to transform from a CF date series with units as 'months since [DATE]' to 'days since [DATE]'
+      datev= series of dates to transform
+      tunits= time units of the dates
+    >>> CFmonthU_daysU([1,2,3,4,5,6,17,18,], 'months since 1979-12-01 00:00:00')
+    array([  31.,   62.,   91.,  122.,  152.,  183.,  517.,  548.])
+    """
+    import datetime as dt
+    fname = 'CFmonthU_secondsU'
+
+    if tunits.find('month') == -1:
+        print errormsg
+        print '  ' + fname + ": times do not have 'months' as units !!"
+        print "    they have: '" + tunits + "'"
+        quit(-1)
+
+    secstunits = tunits.replace('_',' ').split(' ')
+    newtunits = 'seconds'
+    for itsec in range(1,len(secstunits)):
+        newtunits = newtunits + ' ' + secstunits[itsec]
+
+    yr = np.int(secstunits[2].split('-')[0])
+    mm = np.int(secstunits[2].split('-')[1])
+    dd = np.int(secstunits[2].split('-')[2])
+
+    if tunits.find(':') != -1:
+        hh = np.int(secstunits[3].split(':')[0])
+        mi = np.int(secstunits[3].split(':')[1])
+        ss = np.int(secstunits[3].split(':')[2])
+    else:
+        hh = 0
+        mi = 0
+        ss = 0
+
+    refT = dt.datetime(yr, mm, dd, hh, mi, ss)
+
+    refm = 0
+    newdatev = np.zeros((len(datev)), dtype=np.float)
+    for itv in range(len(datev)):
+        if mm + datev[itv] - refm > 12:
+            yr = yr + 1
+            refm = refm + 12
+
+        print itv, datev[itv], yr, refm, mm + datev[itv] - refm
+        idate = dt.datetime(yr, mm + datev[itv] - refm, dd, hh, mi, ss)
+        
+        diffT = idate - refT
+        if searchInlist(dir(dt.timedelta), 'total_seconds'):
+            newdatev[itv] = diffT.total_seconds()/(3600.*24.)
+        else:
+            newdatev[itv] = diffT.days() + diffT.seconds()/(3600.*24)
+
+    return newdatev, newtunits
+
+print CFmonthU_daysU([1,2,3,4,5,6,17,18,], 'months since 1979-12-01 00:00:00')
+
 #quit()