Changeset 2440 in lmdz_wrf

Timestamp:

Apr 16, 2019, 4:34:27 PM (6 years ago)

Author:

lfita

Message:

Imposing gregorian calendar for non-standard calendars in `get_timee'

File:

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

trunk/tools/nc_var_tools.py

@@ -25468 +25468 @@
 def get_time(values, ncfile, variable):
     """ Function to provide the closest time-step to a given time
-      values= [timevalue];[units]
+      values= [timevalue];[units];[calendar]
         [timevalue]: CF value of time to look for
         [units]: CF units of time [Tunit] since [date] ('!' for spaces)
+        [calenadar]: CF calendar used by the data
       ncfile= file from which provide the information
       variable= [timevarn] name of the variable with the CF-times ('WRFtime' for WRF
@@ -25482 +25483 @@
         quit()
 
-    expectargs = '[timevalue];[units]'
+    expectargs = '[timevalue];[units];[calendar]'
     gen.check_arguments(fname, values, expectargs, ';')
 
     timevalue = np.float(values.split(';')[0])
     uTunits = values.split(';')[1].replace('!',' ')
+    calendar = np.float(values.split(';')[2])
 
     onc = NetCDFFile(ncfile, 'r')
@@ -25500 +25502 @@
         otime = onc.variables['Times']
         timev0, timeu = compute_WRFtime(otime[:])
+        timec = 'gregorian'
     else:
         otime = onc.variables[variable]
         timev0 = otime[:]
         timeu = otime.units
-
+        timea = otime.ncattrs()
+        if gen.searchInlist(timea, 'calendar'):
+            timec = otime.calendar
+        else:
+            print warnmsg
+            print '  ' + fname + ": time variable '" + variable + "' without " +     \
+              " 'calendar' attribute !!" 
+            print "    assuming 'gregorian'"
+            timec = 'gregorian'
     onc.close()
 
+    # Checking calendar consistency
+    if calendar != timec:
+        print errmsg
+        print '  ' + fname + ": different calendars found !!" 
+        print "    input calendar: '" + calendar + "' calendar in data '" +timec + "'"
+        quit(-1)
+    
+    if calendar == 'noleap' or calendar == '365d':
+        # Adding leap days to avoid dephase (thus, imposing 'gregorian' calendar)
+        gregtime = gen.impose_gregorian(timev0, uTunits, calendar)
+
     # Referring to the same units
-    reftvals = gen.coincident_CFtimes(timev0, uTunits, timeu)
+    reftvals = gen.coincident_CFtimes(gregtime, uTunits, timeu)
 
     # looking for minimum distance