Changeset 1896 in lmdz_wrf for trunk

Timestamp:

Apr 6, 2018, 9:08:28 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `WRFsetup': Function to check the set-up of a WRF model namelist or from any other model

Location:

trunk/tools

Files:

• generic.py (modified) (4 diffs)
• generic_tools.py (modified) (2 diffs)

trunk/tools/generic.py

@@ -39 +39 @@
 # variables_values: Function to provide values to plot the different variables values from ASCII file
 # wdismean: Function to compute the mean value weighted to its 4 distances
+# WRFsetup: Function to check the set-up of a WRF model namelist or from any other model
+
 # Character to split passed values
-
 cS = ','
-# Character to split serie of values
+# Character to split a serie of values
 cV = '@'
 # Character for spaces
@@ -54 +55 @@
   'rmNOnum', 'running_mean',                                                         \
   'significant_decomposition', 'squared_radial',                                     \
-  'table_tex_file', 'unitsDate', 'variables_values', 'wdismean']
+  'table_tex_file', 'unitsDate', 'variables_values', 'wdismean', 'WRFsetup']
 
 hundredvals = '0'
@@ -81 +82 @@
 ## e.g. # generic.py -o variables_values -S 'hus'
 ## e.g. # generic.py -o wdismean -S 0.005@0.005,0.@1.@2.@3.
+## e.g. # python /home/lluis/PyNCplot/generic.py -o WRFsetup -S ../UBA/namelist.input,basic,textabrow
 
 operationnames = "'" + gen.numVector_String(operations, "', '") + "'"
@@ -402 +404 @@
         print gen.wdismean(vals0, vals1)
 
+elif oper == 'WRFsetup':
+    Nvals = 3
+    vals = opts.values.split(cS)
+    if vals[0] == 'h':
+        print gen.WRFsetup.__doc__
+        quit(-1)
+    else:
+        if len(vals) != Nvals:
+            print errormsg
+            print '  ' + main + ": operation '" + oper + "' requires", Nvals, 'and', \
+              len(vals), ' has passed!!'
+            print gen.WRFsetup.__doc__
+            quit(-1)
+        vals0 = vals[0]
+        vals1 = vals[1]
+        vals2 = vals[2]
+        
+        print gen.WRFsetup(vals0, vals1, vals2)
+

trunk/tools/generic_tools.py

@@ -153 +153 @@
 # vals_around: Function to provide the 3x3 values around a given j,i point
 # xtrm_nx: Function to provide the extreme (a percentage above min,max) of a series of values
+# WRFsetup: Function to check the set-up of a WRF model namelist or from any other model
 
 def searchInlist(listname, nameFind):
@@ -12817 +12818 @@
 #print CFtime_freq(521438., 521439., 'hours since 1950-01-01_00:00:00')
 
+def WRFsetup(namelistn, params='basic', output='twopoints@'):
+    """ Function to check the set-up of a WRF model namelist or from any other model
+      namelistn= file name of the 'namelist.input'
+      params= parameters to check
+        'basic': a pre-determined list of basic parameters (default)
+        'specific',[listparams]: a given ':' separated list of parameters to check
+      output= kind of format of the output
+        'textabcol': a LaTeX-table column formated
+        'textabrow': a LaTeX-table row formated
+        'twopoints@': as ':' separated list of values [attrn]@[attrv]:... (default)
+      WRFsetup('namelist.input')
+    """
+    fname = 'WRFsetup'
+    availparams = ['basic', 'specific,[listparams]']
+    availoutput = ['textabcol', 'textabrow', 'twopoints@']
+
+    if params == 'basic':
+        lookparams = {                                                               \
+          'time_control': ['start_year', 'start_month', 'start_day', 'start_hour',   \
+          'start_minute', 'start_second', 'end_year', 'end_month', 'end_day',        \
+          'end_hour', 'end_minute', 'end_second', 'history_interval',                \
+          'frames_per_outfile', 'restart', 'restart_interval_h'],                    \
+          'domains': ['time_step', 'max_dom', 'e_we', 'e_sn', 'e_vert',              \
+          'p_top_requested', 'dx', 'dy', 'parent_grid_ratio',                        \
+          'parent_time_step_ratio', 'feedback'],                                     \
+          'physics': ['mp_physics', 'ra_lw_physics', 'ra_sw_physics', 'radt',        \
+          'sf_sfclay_physics', 'sf_surface_physics', 'bl_pbl_physics', 'bldt',       \
+          'cu_physics', 'cudt', 'shcu_physics', 'sst_update'],                       \
+          'fdda': [],                                                                \
+          'dynamics': ['w_damping', 'diff_opt', 'km_opt', 'diff_6th_opt',            \
+          'diff_6th_factor', 'damp_opt', 'zdamp', 'dampcoef', 'khdif', 'kvdif',      \
+          'moist_adv_opt', 'scalar_adv_opt'],                                        \
+          'bdy_control': ['spec_bdy_width', 'relax_zone'],                           \
+          'IFexist': ['eta_levels']}
+
+        secnames = ['time_control', 'domains', 'physics', 'fdda', 'dynamics',        \
+          'bdy_control']
+
+    elif params[0:8] == 'specific':
+        secnames = ['specific']
+        lparns = params.split(',')[1].split(':')
+        lpns = []
+        for lp in lparns: lpns.append(lp)
+        lookparams = {'specific': lpns}
+    else:
+        print errormsg
+        print '  ' + fname +  ": parameters to check '" + params + "' not ready !!"
+        print '    available ones:', availparams
+        quit(-1)
+
+    onf = open(namelistn, 'r')
+    fparams = {}
+    pvals = ''
+    for line in onf:
+        if line[0:1] != '#' and len(line) > 1 and line.find('&') == -1 and           \
+          line.find('/') == -1:
+            if line.find('=') != -1:
+                if len(pvals) != 0:
+                    fparams[paramn] = pvals
+                ilvals = line.replace('\n', '').replace('\t','').split('=')
+                paramn = ilvals[0].replace(' ','')
+                pvals = ilvals[1].replace(' ','')
+            else:
+                ilvals = line.replace('\n', '').replace('\t','').replace(' ', '')
+                pvals = pvals + ilvals
+
+    outl = ''
+    ipar = 0
+    for secn in secnames:
+        print secn + "&"
+        psec = lookparams[secn]
+        for ps in psec:
+            if not fparams.has_key(ps):
+                print errormsg
+                print '  ' + fname + ": provided namelist file '" + namelistn +      \
+                  "' has not parameter '" + ps + "' !!"
+                quit(-1)
+
+            # output
+            if output == 'textabcol':
+                if ipar == 0: outl = ps.replace('_', '\\_') + ' & ' +                \
+                  fparams[ps].replace('_', '\\_') + ' \\\\'
+                else: outl = outl + '\n' + ps.replace('_', '\\_') +' & '+            \
+                  fparams[ps].replace('_', '\\_') + ' \\\\'
+            elif output == 'textabrow':
+                if ipar == 0: 
+                    outl1 = '{\\bfseries{' + ps.replace('_', '\\_') + '}}'
+                    outl2 = fparams[ps].replace('_', '\\_')
+                else: 
+                    outl1 = outl1 + ' & {\\bfseries{' + ps.replace('_', '\\_') + '}}'
+                    outl2 = outl2 + ' & ' + fparams[ps].replace('_', '\\_')
+            elif output == 'twopoints@':
+                if ipar == 0: outl = ps + '@' + fparams[ps]
+                else: outl = outl + ':' + ps + '@' + fparams[ps]
+            else:
+                print errormsg
+                print '  ' + fname + ": kind of output '" + output + "' not ready !!"
+                print '    available ones:', availoutput
+                quit(-1)
+            ipar=ipar+1
+
+    # Additional parameters (if they exist)
+    if params == 'basic':
+        psec = lookparams['IFexist']
+        for ps in psec:
+            if fparams.has_key(ps):
+                if output == 'textabcol':
+                    outl = outl + '\n' + ps.replace('_', '\\_') +' & '+              \
+                      fparams[ps].replace('_', '\\_') + ' \\\\'
+                elif output == 'textabrow':
+                    outl1 = outl1 + ' & {\\bfseries{' + ps.replace('_', '\\_') + '}}'
+                    outl2 = outl2 + ' & ' + fparams[ps].replace('_', '\\_')
+                elif output == 'twopoints@':
+                    outl = outl + ':' + ps + '@' + fparams[ps]
+
+    if output == 'textabrow':
+        outl = outl1 + ' \\\\\n' + outl2
+
+    return outl
+
+#print WRFsetup('/home/lluis/estudios/RELAMPAGO/SimCoor/UBA/namelist.input', 'specific,start_year:end_year', 'textabcol')
+
 #quit()
 
 
+