Changeset 1504 in lmdz_wrf

Timestamp:

Apr 10, 2017, 3:50:08 PM (8 years ago)

Author:

lfita

Message:

Adding:

`check_timestep': Function to check if a time-units are 'timestep' based. If it's the case, transform them

File:

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

trunk/tools/generic_tools.py

@@ -55 +55 @@
 # chainSnum_num: Function to pass a `ChainStrNum' string to a number
 # changedate360: Class to change a date  on a 360 days/yr (or 12 30-days months) calendar
+# check_timestep: Function to check if a time-units are 'timestep' based. If it's the case, transform them
 # coincident_CFtimes: Function to make coincident times for two different sets of CFtimes
 # coldec_hex: Function to pas a decimal ([r,g,b]; [0.,1.]) color to hexadecimal (#[RR][GG][BB], 00-64, 0A-FF)
@@ -6275 +6276 @@
     return Nvals
 
+def check_timestep(otval):
+    """ Function to check if a time-units are 'timestep' based. If it's the case, transform them
+      otval = time variable object
+    """
+    fname = 'check_timestep'
+
+    varatrs = otval.ncattrs()
+
+    if not gen.searchInfile(varatrs, 'units'):
+        print errormsg
+        print '  ' + main+'.'+fname + ": object variable without 'units' attribute!!"
+        print '    attributes found:', varatrs
+        quit(-1)
+
+    tunits = varattrs.units
+    if tunits.split(' ')[0] == 'timesteps':
+        print warnmsg
+        print '  ' + fname + ": time variable with 'timesteps' as units !!"
+        print '    modifying time values'
+
+        dimvn = otval.dimensions[0]
+
+        # looking for units of time-step as 'clever' combination with dimension name?
+        Tunitssecs = ['second', 'minute', 'hour', 'day']
+        foundCFunits = False
+        Tunitsguess = []
+        for Tunit in Tunitssecs:
+            attrn = Tunit[0:1]
+            if gen.searchInlist(varatrs,dimvn+'_'+attrn):
+                CFtu = Tunit + 's'
+                foundCFunits = True
+                Tunitsguess.append(dimvn+'_'+attrn)
+                break
+            attrn = Tunit[0:4]
+            if gen.searchInlist(varatrs,dimvn+'_'+attrn):
+                CFtu = Tunit + 's'
+                foundCFunits = True
+                Tunitsguess.append(dimvn+'_'+attrn)
+                break
+            attrn = Tunit
+            if gen.searchInlist(varatrs,dimvn+'_'+attrn):
+                CFtu = Tunit + 's'
+                foundCFunits = True
+                Tunitsguess.append(dimvn+'_'+attrn)
+                break
+            attrn = Tunit + 's'
+            if gen.searchInlist(varatrs,dimvn+'_'+attrn):
+                CFtu = Tunit + 's'
+                foundCFunits = True
+                Tunitsguess.append(dimvn+'_'+attrn)
+                break
+
+        if not foundCFunits:
+            print errormsg
+            print '  ' + fname + ": time-variable with units as 'timesteps' does " + \
+              'not have a prepared way:' + Tunitsguess + 'to find its units !!'
+            print "    variables' attributes:", varatrs
+            quit(-1)
+
+        CFtv = otval.getncattr(dimvn+'_'+attrn)
+        newTS = [CFtu] + tunits[1:-1]
+        newTunits = ' '.join(newTS)
+        newTvals = otval[:]*CFtv
+
+    return newTunits, newTvals
+
+
 def coincident_CFtimes(tvalB, tunitA, tunitB):
     """ Function to make coincident times for two different sets of CFtimes