Changeset 2468 in lmdz_wrf for trunk

Timestamp:

Apr 24, 2019, 5:54:43 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `CFfile_fixTime': Operation to fix a netCDF file with a time axis folowing CF-conventions, but with a wrong set-up following a set of available operations

File:

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

trunk/tools/nc_var_tools.py

@@ -34 +34 @@
 # CDO_toCF: Function to pass a CDO output file to CF-conventions
 # CFfile_creation: Operation to create a file folowing CF-conventions 
+# CFfile_fixTime: Operation to fix a netCDF file with a time axis folowing CF-conventions, but 
+#   with a wrong set-up following a set of available operations
 # CFmorzization: Function to provide a CF-compilation version of a variable within a file
 # CFvars: Function to adapt a given variable and file following CF-standards
@@ -31504 +31506 @@
 def CFfile_fixTime(values, ncfile, variable):
     """ Operation to fix a netCDF file with a time axis folowing CF-conventions, but
-        with a wrong set-up
+        with a wrong set-up following a set of available operations
       values= [operations]
         [operations]: ';' list of names of operations to perform
+          'setCalendar': re-set the calendar to 'gregorian' or alos know as 'standard'
           'setRefDate',[YYYYMMDDHHMMSS]: re-set the reference date to [YYYYMMDDHHMMSS]
-          'setCalendar': re-set the calendar to gregorian or standard
+          'setTunits',[tunits]: re-set the temporal units. Accepted ones: 'weeks', 
+            'days',  'hours', 'minutes', 'seconds'
       ncfile= name of the file to fix its time-axis
       variable= name of the variable time 
     """
     fname = 'CFfile_fixTime'
+    availop = ['setCalendar', "setRefDate,[YYYYMMDDHHMMSS]", "setTunits,[tunits]"]
+    availTunits = ['weeks', 'days', 'hours', 'minutes', 'seconds']
 
     if values == 'h':
@@ -31523 +31529 @@
     gen.check_arguments(fname,values,expectargs,':')
 
-    operations = values.split(':')[0]
+    # In case of multiple values
+    #operations = values.split(':')[0].split(';')
+    operations = values.split(';')
 
     onc = NetCDFFile(ncfile,'a')
-    if not onc.variables.has_key(variable):
-
+    varinfile(onc, ncfile, errormsg, '', variable)
 
     otime = onc.variables[variable]
-
-    for op in operations:
-
-
-
+    timevs = otime[:]
+    timeattrs = otime.ncattrs()
+    if not gen.searchInlist(timeattrs,'units'):
+        print errormsg
+        print '  ' + fname + ": time variable '" + variable + "' without units !!"
+        print '    a CF compilant time axis, must have units in format: [Tunits] ' + \
+          'since [YYYY]-[MM]-[DD] [HH]:[MM]:[SS]'
+        quit(-1)
+    tunits = otime.getncattr('units')
+    ltunits = tunits.split(' ')
+    if len(ltunits) < 3 or not gen.searchInlist(ltunits,'since'):
+        print errormsg
+        print '  ' + fname + ": wrong time units '" + tunits + "' !!"
+        print '    a CF compilant time axis, must have units in format: [Tunits] ' + \
+          'since [YYYY]-[MM]-[DD] [HH]:[MM]:[SS]'
+        quit(-1)
+    if not gen.searchInlist(timeattrs,'calendar'):
+        print warnmsg
+        print '  ' + fname + ": time variable '" + variable + "' without calendar !!"
+        print '    a CF compilant time axis, must have units with a calendar'
+        print "    imposing 'gregorian'"
+        origcal = 'gregorian'
+    else:
+        origcal = otime.getncattr('calendar')
+
+    for opn in operations:
+        if opn.count(',') != 0:
+            lopn = opn.split(',')
+        else:
+            lopn = [opn]
+
+        # Fixing calendar
+        if lopn[0] == 'setCalendar':
+            refop = 'setCalendar'
+            Nopreq = len(refop.split(','))
+            if len(lopn) != Nopreq:
+                print errormsg
+                print '  ' + fname + ": wrong number of terms for operation '" +     \
+                  lopn[0] + "' !!"
+                print '    it requires:', Nopreq, 'and', len(lopn), 'were passed !'
+                print "    requested terms for the operation: '" + refpop + "'"
+                quit(-1)
+            print '    ' + fname + ": fixing calendar to 'gregorian'..."
+
+            newtimv = gen.change_CFcalendar(timevs, tunits, origcal)
+            newtunits = tunits + ''
+
+        # Fixing reference date
+        # python datetime only accepts dates >= 1900-Jan-1
+        elif lopn[0] == 'setRefDate':
+            refop = 'setRefDate,[YYYYMMDDHHMMSS]'
+            Nopreq = len(refop.split(','))
+            if len(lopn) != Nopreq:
+                print errormsg
+                print '  ' + fname + ": wrong number of terms for operation '" +     \
+                  lopn[0] + "' !!"
+                print '    it requires:', Nopreq, 'and', len(lopn), 'were passed !'
+                print "    requested terms for the operation: '" + refpop + "'"
+                quit(-1)
+            if len(lopn[1]) != len('YYYYMMDDHHMMSS'):
+                print errormsg
+                print '  ' + fname + ": wrong new reference date for operation '" +  \
+                  lopn[0] + "' !!"
+                print '    it requires:', len('YYYYMMDDHHMMSS'), 'charactgers and',  \
+                  len(lopn[1]), 'were passed !'
+                print "     expected date format is 'YYYYMMDDHHMMSS'"
+                quit(-1)
+
+            newRdate = gen.datetimeStr_conversion(lopn[1],'YmdHMS','Y-m-d H:M:S')
+            print '    ' + fname + ": fixing Reference date to '" + newRdate + "'..."
+
+            newtimv, newtunits = gen.change_CFRefdate(timevs, tunits, newRdate)
+
+        # Fixing units
+        elif lopn[0] == 'setTunits':
+            refop = 'setTunits,[Tunits]'
+            Nopreq = len(refop.split(','))
+            if len(lopn) != Nopreq:
+                print errormsg
+                print '  ' + fname + ": wrong number of terms for operation '" +     \
+                  lopn[0] + "' !!"
+                print '    it requires:', Nopreq, 'and', len(lopn), 'were passed !'
+                print "    requested terms for the operation: '" + refpop + "'"
+                quit(-1)
+            if not gen.searchInlist(availTunits,lopn[1]):
+                print errormsg
+                print '  ' + fname + ": wrong new time-units '" + lopn[1] + "' !!"
+                print '    avaialable ones:', availTunits
+                quit(-1)
+
+            print '    ' + fname + ": fixing temporal units to '" + lopn[1] + "'..."
+            newtimv, newtunits = gen.change_CFTunits(timevs, tunits, lopn[1])
+
+        else:
+            print errormsg
+            print '  ' + fname + ": operation '" + lopn[0] + "' not ready !!"
+            print '    available ones:', availop
+            quit(-1)
+        timevs = newtimv*1.
+        tunits = newtunits + ''
+
+    # Changing values in file
+    otime[:] = timevs
+    otime.setncattr('units', tunits)
+
+    onc.sync()
+
+    # Adding PyNCplot transformation
+    add_global_PyNCplot(onc, 'nc_var_tools', fname, '1.0')
+    
+    onc.sync()
+
+    onc.close()
+    print fname + ": successfull correction of time-axis of file '" + ncfile + "' !!"
+
+    return
+
+#timesv = [56649.5, 56680, 56710.5, 56741, 56771.5, 56802.5, 56833, 56863.5, 56894, 56924.5]
+#dimt = len(timesv)
+#values='lon|360,lat|180,time|None@' + str(dimt) + ':global'
+#varsS='tas#time;lat;lon#f#None,time#time#f8#units|days!since!1850-01-01!00:00:00|S'
+#CFfile_creation(values, 'CFtest.nc', varsS)
+#fname = 'timev1850-2005.dat'
+#of = open(fname, 'w')
+#for tv in timesv:
+#    of.write(str(tv) + '\n')
+#of.close()
+#setvar_asciivalues(fname, 'CFtest.nc', 'time')
+#values="setCalendar;setRefDate,19491201000000;setTunits,hours"
+#varaddattr('calendar|noleap', 'CFtest.nc', 'time')
+#CFfile_fixTime(values, 'CFtest.nc', 'time')
 
 #quit()