Changeset 2177 in lmdz_wrf for trunk/tools

Timestamp:

Oct 11, 2018, 1:50:33 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `advance_date': Function to advance a date object a certain given units of time

File:

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

trunk/tools/generic_tools.py

@@ -57 +57 @@
 
 ####### Content
+# advance_date: Function to advance a date object a certain given units of time
 # advance_matDate: Function to advance matrix-date with a matrix-increment
 # angle_DegMinSec: Function to transform an angle to Degrees Minutes Seconds
@@ -14371 +14372 @@
 #    etime = tv[timeslc[1]]
 #    print islc, ':', itime, '-', etime, '<>', tvals, '=', tstat[islc]
-   
+
 def Latin_Greek(char):
     """ Function to pass an ASCII latin character to its equivalent greek
@@ -14452 +14453 @@
     return vals
 
+def advance_date(date, Tunit, NTunits):
+    """ Function to advance a date a certain given units of time
+      date: datetime object to advance
+      Tunit: units of time (one from 'years', 'months', 'days', 'minutes', 'hours', 
+        'seconds')
+      NTunits: number of [Tunit] to advance
+    >>> advance_date(dt.datetime(1976,02,17,8,30,0), 'seconds', 25)
+    1976-02-17 08:30:25
+    >>> advance_date(dt.datetime(1976,02,17,8,30,0), 'seconds', 75)
+    1976-02-17 08:31:15
+    >>> advance_date(dt.datetime(1976,02,17,8,30,0), 'seconds', 2500)
+    1976-02-17 09:11:40
+    >>> advance_date(dt.datetime(1976,02,17,8,30,0), 'seconds', 3600*20)
+    1976-02-18 04:30:00
+    >>> advance_date(dt.datetime(1976,02,17,8,30,0), 'seconds', 24*3600*100)
+    1976-05-27 08:30:00
+    >>> advance_date(dt.datetime(1976,02,17,8,30,0), 'seconds', 24*3600*1000)
+    1978-11-13 08:30:00
+    """
+    import datetime as dt
+    fname = 'advance_date'
+
+    availtunits = ['years', 'months', 'days', 'minutes', 'hours', 'seconds']
+
+    mdate = [date.year, date.month, date.day, date.hour, date.minute, date.second]
+
+    if Tunit == 'years':
+        idate = 0
+    elif Tunit == 'months':
+        idate = 1
+    elif Tunit == 'days':
+        idate = 2
+    elif Tunit == 'hours':
+        idate = 3
+    elif Tunit == 'minutes':
+        idate = 4
+    elif Tunit == 'seconds':
+        idate = 5
+    else:
+        print errormsg
+        print '  ' + fname + ": time units '" + uints + "' not ready !!"
+        print '    available ones:', availtunits
+        quit(-1)
+
+    thresholds = [100000, 13, days_month(mdate[0],mdate[1])+1, 24, 60, 60]
+    mins = [-100000., 1., 1., 0., 0., 0.]
+
+
+    ndate = mdate + []
+    ndate[idate] = ndate[idate] + NTunits
+
+    for idd in range(5,0,-1):
+        if ndate[idd] >= thresholds[idd]:
+            Nthres = int(ndate[idd]/thresholds[idd])
+
+            if idd == 2 and Nthres > 1:
+                TOTthres = thresholds[idd] - 1
+                year = ndate[0]
+                mon = ndate[1]
+                while TOTthres < ndate[idd]:
+                    if mon + 1 > 12:
+                        mon = 1
+                        year = year + 1
+                    else:
+                        mon = mon + 1
+                    dmon = days_month(year,mon)
+                    if TOTthres + dmon > ndate[idd]: break
+                    else:
+                        TOTthres = TOTthres + dmon
+                ndate[idd] = ndate[idd] - TOTthres
+                ndate[1] = mon
+                ndate[0] = year
+            else:
+                ndate[idd] = int(np.max([ndate[idd] - thresholds[idd]*Nthres*1.,     \
+                  mins[idd]]))
+                ndate[idd-1] = ndate[idd-1] + Nthres
+
+    newdate = dt.datetime(year=ndate[0], month=ndate[1], day=ndate[2],               \
+      hour=ndate[3], minute=ndate[4], second=ndate[5])
+
+    return newdate
+
+def generate_CFtimes(datei, datef, units, freq, dateref='19491201000000'):
+    """ Function to generate CFtimes for a given period, frequency and units
+      datei: initial date of the period ([YYYY][MM][DD][HH][MI][SS] format)
+      datef: final date of the period ([YYYY][MM][DD][HH][MI][SS] format)
+      units: units of time (one from 'years', 'months', 'days', 'minutes', 'hours', 
+        'seconds')
+      Tfreq: frequency of time-steps in number of [units] time
+      dateref: reference date ([YYYY][MM][DD][HH][MI][SS] format)
+    """
+    import datetime as dt
+    fname = 'generate_CFtimes'
+
+    availtunits = ['years', 'months', 'days', 'minutes', 'hours', 'seconds']
+
+    idate = dt.datetime.strptime(datei, '%Y%m%d%H%M%S')
+    edate = dt.datetime.strptime(datef, '%Y%m%d%H%M%S')
+    refdate = dt.datetime.strptime(dateref, '%Y%m%d%H%M%S')
+    refdateS = refdate.strftime("%Y-%m-%d %H:%M:%S")
+
+    delta = idate - refdate
+    # Python version depending
+    if searchInlist(dir(delta), 'total_seconds'):
+        diffT = delta.total_seconds()
+    else:
+        diffT = delta.days*24*3600. + delta.seconds
+
+    cftimes = [diffT]
+
+    date = idate
+    while date < edate:
+        date = advance_date(date, units, freq)
+
+        delta = date - refdate
+        # Python version depending
+        if searchInlist(dir(delta), 'total_seconds'):
+            diffT = delta.total_seconds()
+        else:
+            diffT = delta.days*24*3600. + delta.seconds
+        cftimes.append(diffT)
+        print 'Lluis:', date, ':', diffT, 'frq:', freq
+    
+    delta = edate - refdate
+    # Python version depending
+    if searchInlist(dir(delta), 'total_seconds'):
+        diffT = delta.total_seconds()
+    else:
+        diffT = delta.days*24*3600. + delta.seconds
+
+    cftimes.append(diffT)
+    timeu = units + ' since ' + refdateS
+
+    return cftimes, timeu
+
+print generate_CFtimes('20021201000000', '20030228000000', 'days', 1)
+
 #quit()