Changeset 1454 in lmdz_wrf for trunk/tools/generic_tools.py

Timestamp:

Feb 20, 2017, 4:44:18 PM (8 years ago)

Author:

lfita

Message:

Improving `draw_vals_trajectories'

File:

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

trunk/tools/generic_tools.py

@@ -62 +62 @@
 #   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
+# 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
 # dictionary_key: Function to provide the first key in a dictionay with a given value
 # dictionary_key_list: Function to provide the first key in a dictionary of lists with a given value
@@ -123 +125 @@
 # timestep_conform: Function to provide the time-step in seconds which conforms 1 temporal unit and the resultant number
 #   of time-steps for a whole period is multiple of a given number
+# timefmt_timelab: Function to transform from C-like time format to general one
 # unitsdsDate: Function to know how many units of time are from a given pair of dates
 # vals_around: Function to provide the 3x3 values around a given j,i point
@@ -2557 +2560 @@
 def timeStr_time(StringDate):
   """ Function to transform a string date ([HH]:[MI]:[SS] format) to a time object
-  >>> datetimeStr_datetime('04:32:54')
+  >>> timeStr_datetime('04:32:54')
   04:32:54
   """
@@ -2569 +2572 @@
 
   return newtime
+
+def DateTimeStr_date(StringDate):
+    """ Function to transform a string date-time ([YYYY][MM][DD][HH][MI][SS] format) to a date object
+    >>> DateTimeStr_date('19760217083205')
+    1976-02-17 08:32:05
+    """
+    import datetime as dt
+
+    yr = int(StringDate[0:4])
+    mo = int(StringDate[4:6])
+    da = int(StringDate[6:8])
+    ho = int(StringDate[8:10])
+    mi = int(StringDate[10:12])
+    se = int(StringDate[12:14])
+
+    if yr == 0:
+      print warnmsg
+      print '     ' + fname + ': 0 reference year!! changing to 1'
+      yr = 1
+    newdate = dt.datetime(yr, mo, da, ho, mi, se)
+
+    return newdate
 
 def timeref_datetime(refd, timeval, tu):
@@ -11314 +11339 @@
     return uwindtestv, vwindtestv, twindtestv
 
+def timefmt_timelab(timefmt):
+    """ Function to transform from C-like time format to general one
+      timefmt= C-like instruction for time
+      %y as [YY], %Y as [YYYY], %m as [MM], %d as [DD], %h as [HH], %M as [MI], %S as [SS]
+    >>> timefmt_timelab('$%d^{%H}$')
+    $[DD]^{[HH]}$
+    """
+    fname = 'timefmt_timelab'
+
+    # Dictionary wtth the equivalencies
+    dictchanges = {'%y': '[YY]', '%Y': '[YYYY]', '%m': '[MM]', '%d': '[DD]',         \
+      '%H': '[HH]', '%M': '[MI]', '%S': '[SS]'}
+    timelab = timefmt
+    for chg in dictchanges.keys():
+        newv = dictchanges[chg]
+        timelab = timelab.replace(chg, newv)
+
+    if timelab.find('%') != -1:
+        print warnsmg
+        print '  '+fname+": unsuccesfull substitution of all C-like values from '" + \
+          timefmt + "' !!"
+
+    return timelab
+
 #quit()