Changeset 367 in lmdz_wrf for trunk/tools/vertical_interpolation.py

Timestamp:

Mar 19, 2015, 11:26:25 AM (10 years ago)

Author:

lfita

Message:

Fixing a problem for the non-WRF variables

File:

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

trunk/tools/vertical_interpolation.py

@@ -313 +313 @@
             varln = 'geopotential height'
             varu = 'gpm'
+            varinterp = np.zeros((dimv[0], Nintvals, dimv[2], dimv[3]), dtype=np.float)
         elif var == 'WRFrh':
             print '    ' + main + ': computing relative humidity from WRF as ' +     \
@@ -329 +330 @@
             varln = 'relative humidity of the air'
             varu = '%'
+            varinterp = np.zeros((dimv[0], Nintvals, dimv[2], dimv[3]), dtype=np.float)
         elif var == 'WRFt':
             print '    ' + main + ': computing temperature from WRF as ' +           \
@@ -339 +341 @@
             varln = 'temperature of the air'
             varu = 'K'
-
+            varinterp = np.zeros((dimv[0], Nintvals, dimv[2], dimv[3]), dtype=np.float)
         else:
             print errormsg
@@ -361 +363 @@
 
 # Getting variable values:
-    if len(varobj.shape) == 4:
-        varinterp = np.zeros((dimv[0], Nintvals, dimv[2], dimv[3]), dtype=np.float)
-        newvar = newnc.createVariable(var, 'f4', ('time','z','y','x'))
-        varvals = varobj[:]
-    elif len(varobj.shape) <= 3 and len(varobj.shape) >= 1:
-        varpdimvs = []
-        varpdimns = []
-        varpslice = []
-        for vdim in vardims:
-            vardim = len(ncobj.dimensions[vdim])
-            if ncvar.searchInlist(ncdims,vdim):
-                if vdim == dimpresn['Z']:
-                    varpdimvs.append(Nintvals)
-                    varpdimns.append('z')
+        if len(varobj.shape) == 4:
+            varinterp = np.zeros((dimv[0], Nintvals, dimv[2], dimv[3]), dtype=np.float)
+            newvar = newnc.createVariable(var, 'f4', ('time','z','y','x'))
+            varvals = varobj[:]
+        elif len(varobj.shape) <= 3 and len(varobj.shape) >= 1:
+            varpdimvs = []
+            varpdimns = []
+            varpslice = []
+            for vdim in vardims:
+                vardim = len(ncobj.dimensions[vdim])
+                if ncvar.searchInlist(ncdims,vdim):
+                    if vdim == dimpresn['Z']:
+                        varpdimvs.append(Nintvals)
+                        varpdimns.append('z')
+                        varpslice.append(slice(0,vardim))
+                    else:
+                        varpdimvs.append(vardim)
+                        varpslice.append(slice(0,vardim))
+                        if dimpresn.has_key('X') and vdim == dimpresn['X']:
+                            varpdimns.append('x')
+                        elif dimpresn.has_key('Y') and vdim == dimpresn['Y']:
+                            varpdimns.append('y')
+                        elif dimpresn.has_key('T') and vdim == dimpresn['T']:
+                            varpdimns.append('time')
+                        else:
+                            print errormsg
+                            print '  ' + main + ": dimension variable '" + vdim +        \
+                              "' is in pressure but it is not found?"
+                            print '    pressure dimensions:', ncdims
+                            quit(-1)
+                else:
+# Dimension of the variable is not in the pressure variable
+                    varpdimvs.append(vardim)
+                    varpdimns.append(vdim)
                     varpslice.append(slice(0,vardim))
-                else:
-                    varpdimvs.append(vardim)
-                    varpslice.append(slice(0,vardim))
-                    if dimpresn.has_key('X') and vdim == dimpresn['X']:
-                        varpdimns.append('x')
-                    elif dimpresn.has_key('Y') and vdim == dimpresn['Y']:
-                        varpdimns.append('y')
-                    elif dimpresn.has_key('T') and vdim == dimpresn['T']:
-                        varpdimns.append('time')
-                    else:
-                        print errormsg
-                        print '  ' + main + ": dimension variable '" + vdim +        \
-                          "' is in pressure but it is not found?"
-                        print '    pressure dimensions:', ncdims
-                        quit(-1)
-            else:
-# Dimension of the variable is not in the pressure variable
-                varpdimvs.append(vardim)
-                varpdimns.append(vdim)
-                varpslice.append(slice(0,vardim))
-                if not newnc.dimensions.has_key(vdim) and not                        \
-                  ncvar.searchInlist(dimpresn,vdim):
-                    print '  ' + main + ": dimension '" + vdim + "' not in the " +   \
-                      'pressure variable adding it!'
-                    if ncobj.dimensions[vdim].isunlimited():
-                        newnc.createDimension(vdim, None)
-                    else:
-                        newnc.createDimension(vdim, vardim)
-
-        varinterp = np.zeros((varpdimvs), dtype=np.float)
-        newvar = newnc.createVariable(var, 'f4', tuple(varpdimns))
-
-        varvals = varobj[tuple(varpslice)]
-    else:
-        print errormsg
-        print '  ' + main + ': variable shape "', varvals.shape, '" not ready !!!!'
-        quit(-1)
+                    if not newnc.dimensions.has_key(vdim) and not                        \
+                      ncvar.searchInlist(dimpresn,vdim):
+                        print '  ' + main + ": dimension '" + vdim + "' not in the " +   \
+                          'pressure variable adding it!'
+                        if ncobj.dimensions[vdim].isunlimited():
+                            newnc.createDimension(vdim, None)
+                        else:
+                            newnc.createDimension(vdim, vardim)
+    
+            varinterp = np.zeros((varpdimvs), dtype=np.float)
+            newvar = newnc.createVariable(var, 'f4', tuple(varpdimns))
+    
+            varvals = varobj[tuple(varpslice)]
+        else:
+            print errormsg
+            print '  ' + main + ': variable shape "', varobj.shape, '" not ready !!!!'
+            quit(-1)
 
 #    print 'variable:',var,'shape:',varvals.shape,'len:',len(varvals.shape)