Changeset 2246 in lmdz_wrf for trunk/tools/nc_var_tools.py

Timestamp:

Nov 26, 2018, 2:44:21 PM (6 years ago)

Author:

lfita

Message:

Fixing issues in:

• `WRF_CFtime_creation': getting the right variable name and check for repetition of name of dimension 'time'

File:

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

trunk/tools/nc_var_tools.py

@@ -2150 +2150 @@
       print '  ' + fname + ": File '" + ncfile + "' does not have variable: '" +     \
         varn + "' !!"
+      varns = list(ncvars)
+      varns.sort
+      print '    available ones:', varns
       ncf.close()
       quit(-1)
@@ -11072 +11075 @@
     if not wrfnc.dimensions.has_key('time'):
         newdim = wrfnc.createDimension('time', None)
+        tdimn='time'
+    else:
+        for idim in range(10): 
+            tS = str(idim)
+            if not wrfnc.dimensions.has_key('time'+tS):
+                print warnmsg
+                print '  ' + fname + ": file already has dimension 'time' " +        \
+                  "adding 'time" + tS + "' instead !!"
+                newdim = wrfnc.createDimension('time'+tS, None)
+                tdimn='time'+tS
+                break
+
 
     if wrfnc.variables.has_key(varn):
         newvar = wrfnc.variables[varn]
     else:
-        newvar = wrfnc.createVariable('time','f8',('time'))
+        newvar = wrfnc.createVariable(varn,'f8',(tdimn))
 
     newattr = basicvardef(newvar, 'time','time',tunits)
@@ -26988 +27003 @@
 #same_deltasign(values, '/home/lluis/PY/wrfout_d01_1995-01-01_00:00:00', 'T2,Q2,XLAT,XLONG')
 
-def compute_Koeppen_Geiger_clims(values, ncfile, variable):
+def compute_Koeppen_Geiger_clims(values):
     """ Function to compute the Kooeppen-Geiger climatological classification
-      values= [slicedims]
-        [slicedims]: '|' separated list of [dimname]:[valdim] with he values to the 
-          dimensions with which perform a slice of the variables
-          [valdim]: 
-            * [integer]: which value of the dimension
-            * -1: all along the dimension
-            * -9: last value of the dimension
-            * [beg]@[end]@[freq] slice from [beg] to [end] every [freq]
-            * NOTE, no dim name all the dimension size
-      ncfile= netCDF file to use
-      variable: ',' list of variables ('all' for all variables)
+      values= [tasvalues]:[prvalues]
+        [tasvalues]: '|' separated list of [ncfilen]|[varn]|[timevarn] for tas (2m temperature)
+          [ncfilen]: name of the netCDF file with tas values
+          [varn]: name of the variable tas 
+          [timevarn]: name of the variable time ('None' for not present)
+        [prvalues]: '|' separated list of [ncfilen]|[varn]|[timevarn] for pr (precipitation)
+          [ncfilen]: name of the netCDF file with pr values
+          [varn]: name of the variable pr
+          [timevarn]: name of the variable time ('None' for not present)
     """
-    fname = 'same_deltasign'
+    fname = 'compute_Koeppen_Geiger_clims'
 
     if values == 'h':
@@ -27009 +27022 @@
         quit()
 
-    expectargs = '[slicedims]'
-    gen.check_arguments(fname, values, expectargs, ',')
-
-    slicedims = values.split(',')[0]
+    expectargs = '[tasvalues]:[prvalues]'
+    gen.check_arguments(fname, values, expectargs, ':')
+
+    tasvalues = values.split(':')[0].split('|')
+    prvalues = values.split(':')[1].split('|')
+
+    print "  uploading 'tas' values ..."
+    if not os.path.isfile(tasvalues[0]):
+        print errormsg
+        print '  ' + fname + ": file with tas values '" + tasvalues[0] + "' does " + \
+          "not exist !!"
+        quit(-1)
+    onctas = NetCDFFile(tasvalues[0], 'r')
+    if not onctas.variables.has_key(tasvalues[1]):
+        print errormsg
+        print '  '+fname+": tas file has not tas values as '" + tasvalues[1] + "' !!"
+        quit(-1)
+    otas = onctas.variables[tasvalues[1]]
+    if not len(otas.shape) == 3:
+        print errormsg
+        print '  ' + fname + ": tas must be rank 3 and it is: '" + otas.shape + "' !!"
+        quit(-1)
+    tas = otas[:]
+
+    if tasvalues[2] != 'None':
+        if not onctas.variables.has_key(tasvalues[2]):
+            print errormsg
+            print '  '+fname+": tas file has not time values as '" +tasvalues[2] + "' !!"
+            quit(-1)
+        otimetas = onctas.variables[tasvalues[2]]
+        timetas = otimetas[:]
+    else:
+        timetas = None
+
+    print "  uploading 'pr' values ..."
+    if not os.path.isfile(prvalues[0]):
+        print errormsg
+        print '  ' + fname + ": file with pr values '" + tasvalues[0] + "' does " +  \
+          "not exist !!"
+        quit(-1)
+    oncpr = NetCDFFile(prvalues[0], 'r')
+    if not oncpr.variables.has_key(prvalues[1]):
+        print errormsg
+        print '  ' + fname + ": pr file has not pr values as '" + prvalues[1] + "' !!"
+        quit(-1)
+    opr = oncpr.variables[prvalues[1]]
+    pr = opr[:]
+
+    if prvalues[2] != 'None': 
+        if not oncpr.variables.has_key(prvalues[2]):
+            print errormsg
+            print '  ' +fname+ ": pr file has not time values as '" + prvalues[2] + "' !!"
+            quit(-1)
+        otimepr = oncpr.variables[prvalues[2]]
+        timepr = otimepr[:]
+    else:
+        timepr = None
+
+    # Checking dimensions conssitency
+    gen.same_shape(tas,pr)
+
+    if timetas is not None and timepr is not None:
+        if timetas.shape[0] != timepr.shape[0]:
+            print errormsg2
+            print '  '+fname+": length 'timetas':",timetas.shape[0]," and 'timepr':",\
+              timepr.shape[0]," do not coincide !!"
+            quit(-1)
+        timeu = timetas.units
+        timec = timetas.calendar
+        tastdesc = gen.temporal_desc(timetas, timeu, timec)
+
+    if timetas is None:
+        if tas.shape[0] != timepr.shape[0]:
+            print errormsg
+            print '  ' + fname + ": length time-dimension from 'tas':", tas.shape[0],\
+              " and 'timepr':", timepr.shape[0], " do not coincide !!"
+            quit(-1)
+        timeu = timepr.units
+        timec = timepr.calendar
+        tastdesc = gen.temporal_desc(timepr, timeu, timec)
+
+    if timepr is None:
+        if pr.shape[0] != timetas.shape[0]:
+            print errormsg
+            print '  ' + fname + ": length time-dimension from 'pr':", pr.shape[0],  \
+              " and 'timetas':", timetas.shape[0], " do not coincide !!"
+            quit(-1)
+        timeu = timetas.units
+        timec = timetas.calendar
+        tastdesc = gen.temporal_desc(timetas, timeu, timec)
+
+    # Starting statistics
+    tasn = np.min(tas)
+    tasx = np.max(tas)
+
+    prn = np.min(pr)
+    prx = np.max(pr)
+
+
+    return