Changeset 847 in lmdz_wrf

Timestamp:

Jun 16, 2016, 6:59:39 PM (9 years ago)

Author:

lfita

Message:

Adding:

• `model_characteristics': Function to provide major characterisitcs of a given model output
• `ModelChar?': Class object to determine major characterisitcs of a given model output

Location:

trunk/tools

Files:

• nc_var.py (modified) (3 diffs)
• nc_var_tools.py (modified) (6 diffs)

trunk/tools/nc_var.py

@@ -13 +13 @@
 ## e.g. # nc_var.py -o subbasin -f routing.nc -S Caceres,-57.75,-16.25,left
 ## e.g. # nc_var.py -o computevar_model -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S hurs
+## e.g. # nc_var.py -o model_characteristics -f ~/PY/wrfout_d01_2001-11-11_00:00:00 -S WRF
 
 from optparse import OptionParser
@@ -37 +38 @@
   'getvalues_lonlat', 'grattr',                                                      \
   'grmattr', 'igattrs', 'increaseDimvar', 'isgattrs', 'isvattrs', 'ivars', 'ivattrs',\
-  'maskvar',                                                                         \
+  'maskvar', 'model_characteristics',                                                 \
   'ncreplace', 'ncstepdiff', 'netcdf_concatenation', 'netcdf_fold_concatenation',    \
   'Partialmap_Entiremap', 'Partialmap_EntiremapFor', 'Partialmap_EntiremapForExact', \
@@ -203 +204 @@
 elif oper == 'maskvar':
     ncvar.maskvar(opts.values, opts.ncfile, opts.varname)
+elif oper == 'model_characteristics':
+    ncvar.model_characteristics(opts.values, opts.ncfile)
 elif oper == 'ncreplace':
     ncvar.ncreplace(opts.values, opts.ncfile, opts.varname)

trunk/tools/nc_var_tools.py

@@ -66 +66 @@
 # lonlatProj: Function to compute longitudes and latitudes for a given projection following subroutine calc_resolution_in from ORCHIDEE/src_global/module_InterpWeight
 # maskvar: Function to mask a variable using a mask. It is assumed that mask[...,dimM,dimJ,dimK] and var[...,dimM,dimJ,dimK] share the last dimensions
+# ModelCharacteristics: Class object to determine major characterisitcs of a given model output
 # ncreplace: Function to replace something from a netCDF file
 # ncstepdiff: Function to compute differencies between time-steps (deacumulate) a netCDF file
@@ -12647 +12648 @@
 
     if varn == 'global':
-        if not searchInlist(onc.ncattrs(), attrn):
+        if not gen.searchInlist(onc.ncattrs(), attrn):
             print errormsg
             print '  ' + fname + ": file '" + filen + "' does not have atribute '" + \
@@ -12655 +12656 @@
             attrv = onc.getncattr(attrn)
     else:
-        if not searchInlist(onc.variables, varn): 
+        if not gen.searchInlist(onc.variables, varn): 
             print errormsg
             print '  ' + fname + ": file '" + filen + "' does not have variable '" + \
@@ -12662 +12663 @@
         else:
             ovar = onc.variables[varn]
-            if not searchInlist(ovar.ncattrs(), attrn):
+            if not gen.searchInlist(ovar.ncattrs(), attrn):
                 print errormsg
                 print '  ' + fname + ": variable '" + varn + "' does not have " +    \
@@ -12673 +12674 @@
     onc.close()
 
-    print attrv
-
-    return
+    return attrv
 
 def DimsLoop(ovar,seldims):
@@ -15188 +15187 @@
     return
 
+class ModelChar(object):
+    """ Class object to determine major characterisitcs of a given model output
+      model= name of the model
+      ncfile= name of the file to use to check
+    """
+    def __init__(self,model,ncfile):
+        fname = 'ModelChar'
+        models = ['WRF', 'LMDZ', 'WRF_LMDZ']
+
+        if model == 'WRF' or model == 'WRF_LMDZ':
+            if model == 'WRF':
+                self.model = 'Weather Research and Forecasting'
+            else:
+                self.model= 'LMDZ physics coupled to Weather Research and Forecasting'
+
+            self.dimxn = 'west_east'
+            self.dimyn = 'south_north'
+            self.dimzn = 'bottom_top'
+            self.dimtn = 'Time'
+            self.vardxn = 'XLONG'
+            self.vardyn = 'XLAT'
+            self.vardzn = 'DN'
+            self.vardtn = 'Times'
+            if ncfile is not None and ncfile[0:2] == 'geo':
+                self.vardxn = 'XLONG_M'
+                self.vardyn = 'XLAT_M'
+
+            if ncfile is not None:
+                attrv = get_attribute('TITLE', ncfile, 'global')
+                self.version = attrv.split(' ')[4]
+
+        elif model == 'LMD':
+            self.model = 'LMDZ'
+            self.dimxn = 'lon'
+            self.dimyn = 'lat'
+            self.dimzn = 'presnivs'
+            self.dimtn = 'time_counter'
+            self.vardxn = 'lon'
+            self.vardyn = 'lat'
+            self.vardzn = 'pres'
+            self.vardtn = 'time_counter'
+
+            if ncfile is not None:
+                self.version = None
+
+def model_characteristics(values, ncfile):
+    """ Class object to determine major characterisitcs of a given model output
+      values= [model]
+        model: name of the model
+      ncfile= name of the file to use to check
+    """
+    fname = 'model_characteristics'
+
+    if values == 'h':
+        print fname + '_____________________________________________________________'
+        print model_characteristics.__doc__
+        quit()
+
+    model = values
+
+    if ncfile == 'None': ncfile = None
+
+    modchar = ModelChar(model,ncfile)
+    gen.printing_class(modchar)
+
+    return
+
 #quit()