Changeset 1694 in lmdz_wrf

Timestamp:

Dec 7, 2017, 8:09:46 PM (7 years ago)

Author:

lfita

Message:

Removing WRF-related actions

File:

• trunk/tools/get_stations.py (modified) (7 diffs)

trunk/tools/get_stations.py

@@ -21 +21 @@
 
 # Gneric variables prepared to be computed
-Cvars = ['C_td']
+Cvars = ['C_td', 'C_wd', 'C_ws']
 
 # WRF specific diagnostics
@@ -28 +28 @@
 
 # Variables not to check their existence inside file
-NONcheckingvars = Cvars + WRFvars
+NONcheckingvars = Cvars + WRFvars + ['None']
 
 def creation_sfcstation_file(filen, lv, Lv, hv, lab, tunits, sfcvars, dimt, ifilen):
@@ -362 +362 @@
         quit(-1)
 
-    # Except for temporal variables, we don't want dimensions with time-axis 
-    #  (assuming fixed)
-    if varn == 'WRFtime':
-        varvalues, CFtu= ncvar.compute_WRFtime(onc.variables['Times'], ReferenceDate,\
-          UnitsTime)
-        dt = varvalues.shape[0]
-        dimvarvalues[varn] = varvalues
-    else:
-        ovar = onc.variables[varn]
-        slicevar = {}
-        if not gen.searchInlist(axesvars['T'], varn):
-            for dn in ovar.dimensions:
-                if not gen.searchInlist(axesdims['T'], dn):
-                    slicevar[dn] = -1
-                else:
-                    slicevar[dn] = 0
+    if not gen.searchInlist(NONcheckingvars, varn):
+        # Except for temporal variables, we don't want dimensions with time-axis 
+        #  (assuming fixed)
+        if varn == 'WRFtime':
+            varvalues, CFtu = ncvar.compute_WRFtime(onc.variables['Times'],          \
+              ReferenceDate, UnitsTime)
+            dt = varvalues.shape[0]
+            dimvarvalues[varn] = varvalues
         else:
-            slicevar[dn] = -1
-            dt = len(onc.dimensions[dn])
-        slicevar, vardims = ncvar.SliceVarDict(ovar, slicevar)
-        dimvarvalues[varn] = ovar[tuple(slicevar)]
+            ovar = onc.variables[varn]
+            slicevar = {}
+            if not gen.searchInlist(axesvars['T'], varn):
+                for dn in ovar.dimensions:
+                    if not gen.searchInlist(axesdims['T'], dn):
+                        slicevar[dn] = -1
+                    else:
+                        slicevar[dn] = 0
+            else:
+                slicevar[dn] = -1
+                dt = len(onc.dimensions[dn])
+            slicevar, vardims = ncvar.SliceVarDict(ovar, slicevar)
+            dimvarvalues[varn] = ovar[tuple(slicevar)]
 
     if dimn == Pressdimref: dz = len(onc.dimensions[dimn])
+
+# Looking for 2D 'lon', 'lat' related variables
+Xvarvals = dimvarvalues[dimvariables['lon']]
+Yvarvals = dimvarvalues[dimvariables['lat']]
+if len(Xvarvals.shape) == 1:
+    newXvarvals, newYvarvals = np.meshgrid(Xvarvals, Yvarvals) 
+    dimvarvalues[dimvariables['lon']] = newXvarvals
+    dimvarvalues[dimvariables['lat']] = newYvarvals
 
 # Retrieving surface data
@@ -427 +436 @@
             getdims = ogetvar.dimensions
         else:
-            getdims = ['Time', 'south_north', 'west_east']
-            getvdims = ['WRFtime', 'XLAT', 'XLONG']
+            getdims = list(onc.variables[sfcrefvar].dimensions)
+            getvdims = [dimvariables[getdims[0]], dimvariables[getdims[1]],          \
+              dimvariables[getdims[2]]]
 
         # Looking for the X,Y axis for location of station within file
@@ -566 +576 @@
             getdims = ogetvar.dimensions
         else:
-            getdims = ['Time', 'bottom_top', 'south_north', 'west_east']
-            getvdims = ['WRFtime', 'WRFp', 'XLAT', 'XLONG']
+            getdims = list(onc.variables[sndrefvar].dimensions)
+            getvdims = [dimvariables[getdims[0]], dimvariables[getdims[1]],          \
+              dimvariables[getdims[2]], dimvariables[getdims[3]]]
 
         # Looking for the X,Y axis for location of station within file
@@ -613 +624 @@
         
         # Writting information to file
-        if sfcv == sfcrefvar:
+        if sndv == sndrefvar:
             ojvar = onewnc.variables['jpoint']
             ojvar[:] = minji[0]
             ojvar = onewnc.variables['ipoint']
             ojvar[:] = minji[1]
-            ohvar = onewnc.variables['fheight']
-            ohvar[:] = dimvarvalues[dimvariables['H']][minji[0],minji[1]]
+            if dimvariables['H'] != 'None':
+                ohvar = onewnc.variables['fheight']
+                ohvar[:] = dimvarvalues[dimvariables['H']][minji[0],minji[1]]
             onewnc.sync()
 
@@ -655 +667 @@
         # Writting data and slicing
         onewvar = onewnc.variables[sndv]
+        print 'Lluis sndv:', sndv, ' shapes onewvar:', onewvar.shape, 'slice:', slicevar
         onewvar[:] = ogetvar[tuple(slicevar)]    
         onewnc.sync()