Changeset 2402 in lmdz_wrf for trunk

Timestamp:

Mar 11, 2019, 10:16:24 PM (6 years ago)

Author:

lfita

Message:

Trying to fix 'agg' in `temporal_stats'

File:

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

trunk/tools/nc_var_tools.py

@@ -26116 +26116 @@
             'hour': statistics within full hours [00:00 - 59:59]
             'minute': statistics within full minutes [00 - 59]
+            'agg'[tunit]: aggregation of multiple values to the same period [tunit]
+              [tunit]: period to aggregate: year, month, day, hour, minute
           [amount]: quantity of periods to group
           [statn]: statistics to compute
@@ -26134 +26136 @@
     Lperiodn = {'year': 'yearly', 'month': 'monthly', 'day': 'daily',                \
       'LTday': 'local-time daily', 'hour': 'hourly', 'minute': 'minutely',           \
-      'aggseason': 'seasonal aggregation'}
+      'agg': 'aggregation'}
     Lstatn = {'min': 'minimum', 'max': 'maximum', 'mean': 'mean',                    \
       'mean2': 'quadratic mean', 'std': 'standard deviation',                        \
       'percentiles': 'percentiles'}
     availcftunits = ['days', 'hours', 'minutes', 'seconds']
+    availagg = ['year', 'month', 'day', 'hour', 'minute', 'second']
 
     if values == 'h':
@@ -26284 +26287 @@
           time_desc)
         statslices[iTst] = [Ntimeslice, timeslice]
+
+        print 'Lluis slices period:', period, 'amount:', amount, ' ________'
+        for it in range(Ntimeslice):
+            print it, ':', timeslice[it]
  
     # Creation of output file
@@ -26354 +26361 @@
           'statistics:', statn, " ..."
 
-        # Adding period dimension and variables
-        if not gen.searchInlist(onewnc.dimensions, periodn+'_time'):
+        if not onewnc.dimensions.has_key(periodn+'_time'):
             newdim = onewnc.createDimension(periodn+'_time', Ntslc)
-            newvart = onewnc.createVariable(periodn+'_time', 'f8',                   \
+            newvart = onewnc.createVariable(periodn+'_time', 'f8',               \
               (periodn+'_time'))
-            basicvardef(newvart, periodn+'_time', 'time for ' + periodn, timeu)
-            newvart.setncattr('calendar', timec)
-            newvart.setncattr('bounds', periodn+'_time_bounds')
-            newvarb = onewnc.createVariable(periodn+'_time_bounds', 'f8',            \
-              (periodn+'_time', 'bounds'))
-            basicvardef(newvarb, periodn+'_time_bounds', 'time boundaries for '+     \
-              periodn, timeu)
-            newvarb.setncattr('calendar', timec)
-            for it in range(Ntslc):
-                timeslcev = tslcs[it]
-                islc = timevc[timeslcev[0]]
-                eslc = timevc[timeslcev[1]]
+            # Adding period dimension and variables
+            if periodn[0:3] != 'agg':
+                basicvardef(newvart, periodn+'_time', 'time for ' + periodn, timeu)
+                newvart.setncattr('calendar', timec)
+                newvart.setncattr('bounds', periodn+'_time_bounds')
+                newvarb = onewnc.createVariable(periodn+'_time_bounds', 'f8',        \
+                  (periodn+'_time', 'bounds'))
+                basicvardef(newvarb, periodn+'_time_bounds', 'time boundaries for '+ \
+                  periodn, timeu)
+                newvarb.setncattr('calendar', timec)
+                for it in range(Ntslc):
+                    timeslcev = tslcs[it]
+                    islc = timevc[timeslcev[0]]
+                    eslc = timevc[timeslcev[1]]
     
-                newvart[it] = (eslc + islc) / 2.
-                newvarb[it,:] = [islc, eslc]
-            onewnc.sync()
+                    newvart[it] = (eslc + islc) / 2.
+                    newvarb[it,:] = [islc, eslc]
+                onewnc.sync()
+            else:
+                # Time units lose any sense, thus redefining them
+                PerN = periodn[3:len(periodn)]
+                if PerN == 'year':
+                    dtsec = 365 * 24 * 3600
+                elif PerN == 'month':
+                    dtsec = 365 * 24 * 3600
+                elif PerN == 'day':
+                    dtsec = 24 * 3600
+                elif PerN == 'hour':
+                    dtsec = 3600
+                elif PerN == 'minute':
+                    dtsec = 60
+                elif PerN == 'second':
+                    dtsec = 1
+                else:
+                    print errormsg
+                    print '  ' + fname + ": aggregation over '" + PerN + "' not " +  \
+                      "ready !!"
+                    print '    available ones:', availagg
+                    quit(-1)
+ 
+                # Getting slices in axis-time units
+                ut = timeu.split(' ')[0]
+                if ut == 'year': dtsec / (365 * 24 * 3600.)
+                elif ut == 'month': dtsec / (365 * 24 * 3600.)
+                elif ut == 'day': dtsec / (24 * 3600.)
+                elif ut == 'hour': dtsec / (3600.)
+                elif ut == 'minute': dtsec / (60.)
+                elif ut == 'second': dtsec / 1. 
+
+                timeslcev = tslcs[0]
+                firstval = timevc[timeslcev[0]]
+                basicvardef(newvart, periodn+'_time', 'time for ' + periodn, timeu)
+                newvart.setncattr('calendar', timec)
+                newvart.setncattr('bounds', periodn+'_time_bounds')
+                newvarb = onewnc.createVariable(periodn+'_time_bounds', 'f8',        \
+                  (periodn+'_time', 'bounds'))
+                basicvardef(newvarb, periodn+'_time_bounds', 'time boundaries' +     \
+                  ' for ' + periodn, timeu)
+                newvarb.setncattr('calendar', timec)
+                for it in range(Ntslc-1):
+                    timeslcev = tslcs[it]
+                    Nvalsslc = len(timeslcev)
+                    print 'timeslcev:', tslcs[it], 'Nvals:', Nvalsslc
+                    islc = newvart[it]+dtsec/2
+                    timeslcev = tslcs[it+1]
+                    eslc = newvart[it]-dtsec/2
+                    print it, ' Lluis islc:', islc, 'eslc:', eslc
+                    newvart[it] = dtsec
+                    newvarb[it,:] = [islc, eslc]
+                onewnc.sync()
 
         for varn in varnames:
@@ -26439 +26499 @@
             if statn == 'std': varunits = '(' + varunits + ')**2)'
             CFvarattr = gen.variables_values(varn)
-            Lname = Lstatn[statn] + ' ' + Lperiodn[periodn] + ' ' +                  \
-              CFvarattr[4].replace('|',' ')
+            if periodn[0:3] != 'agg':
+                Lname = Lstatn[statn] + ' ' + Lperiodn[periodn] + ' ' +              \
+                  CFvarattr[4].replace('|',' ')
+            else:
+                PerN = periodn[3:len(periodn)+1]
+                Lname = Lstatn[statn] + ' ' + Lperiodn[PerN] + ' ' +Lperiodn['agg']+ \
+                  ' ' + CFvarattr[4].replace('|',' ')
+
             add_varattrs(onc,onewnc,[varn],[newvarn])
             basicvardef(newvarv, newvarn, Lname, varunits)