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

Timestamp:

Aug 15, 2016, 2:37:52 PM (8 years ago)

Author:

lfita

Message:

Adding `[chdimname]': whether west_east, south_north dimensions should change names to 'lon','lat' to 'WRF_CFlonlat_creation'

File:

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

trunk/tools/nc_var_tools.py

@@ -10056 +10056 @@
     """ Function to add a CF-convention longitude/latitude variables in a WRF file
     WRF_CFlon_creation(values, ncfile, varn)
-      [values]= [lonname],[latname] names of the variables to add (standard names 
-        'longitude', 'latitude')
+      [values]= [lonname],[latname],[chdimname] 
+        [lonname],[latname]: names of the variables to add (standard names 'longitude', 'latitude')
+        [chdimname]: whether west_east, south_north dimensions should change names to 'lon','lat'
       [ncfile]= WRF file to add the CF-convention time variable
       [varn]= [lonname],[latname] name of the variables to use
@@ -10070 +10071 @@
     cflonn = values.split(',')[0]
     cflatn = values.split(',')[1]
+    chdimn = gen.Str_Bool(values.split(',')[2])
 
     wlonn = varn.split(',')[0]
@@ -10137 +10139 @@
 
     # Creation of CF-dimensions
-    if not wrfnc.dimensions.has_key('lon'):
-        newdim = wrfnc.createDimension('lon', dimx)
-    if not wrfnc.dimensions.has_key('lat'):
-        newdim = wrfnc.createDimension('lat', dimy)
+    if chdimn:
+        wrfnc.renameDimension('west_east', 'lon')
+        wrfnc.renameDimension('south_north', 'lat')
+    else:
+        if not wrfnc.dimensions.has_key('lon'):
+            newdim = wrfnc.createDimension('lon', dimx)
+        if not wrfnc.dimensions.has_key('lat'):
+            newdim = wrfnc.createDimension('lat', dimy)
 
     if wrfnc.variables.has_key(cflonn):
@@ -16258 +16264 @@
     # Creation of pressure variable dimension
     newvar = onewnc.createVariable('pres', 'f8', ('pres'))
-    newvar[:] = interplevs * 100.
+    newvar[:] = interplevs
     basicvardef(newvar, 'pressure', 'Pressure', 'Pa')
     attr = set_attribute(newvar, 'positive', 'down')
@@ -16478 +16484 @@
   
     # CFing lon,lat
-    WRF_CFlonlat_creation('lon,lat', ncfile, wrfvl+','+wrfvL)
+    WRF_CFlonlat_creation('lon,lat,true', ncfile, wrfvl+','+wrfvL)
 
     ncf = NetCDFFile(ncfile,'a')