Changeset 1897 in lmdz_wrf

Timestamp:

Apr 7, 2018, 2:20:33 PM (7 years ago)

Author:

lfita

Message:

Converting 'WRFsetup' to a multiple namelist table output function

Location:

trunk/tools

Files:

• generic.py (modified) (3 diffs)
• generic_tools.py (modified) (4 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
+# WRFsetup: Function to check the set-up of a series of model namelist
 
 # Character to split passed values
@@ -82 +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
+## e.g. # generic.py -o WRFsetup -S '/home/lluis/estudios/RELAMPAGO/SimCoor/UBA/namelist.input@/home/lluis/estudios/RELAMPAGO/SimCoor/SMN/namelist.input@/home/lluis/estudios/RELAMPAGO/SimCoor/NOA-IERSD/namelist.input@/home/lluis/estudios/RELAMPAGO/SimCoor/UBAmili/namelist.input,basic,textabrow'
 
 operationnames = "'" + gen.numVector_String(operations, "', '") + "'"
@@ -417 +417 @@
             print gen.WRFsetup.__doc__
             quit(-1)
-        vals0 = vals[0]
+        vals0 = vals[0].replace('@', ',')
         vals1 = vals[1]
         vals2 = vals[2]

trunk/tools/generic_tools.py

@@ -114 +114 @@
 # multi_index_string: Function to provide the indeces of every repetition of a group of characters within a string
 # Nomasked: Function to bring back a given array wthout the masked values reducing a given dimension
+# Ntchar: Function to repeat a number of times a given tring
 # num_chainSnum: Function to pass a value to a `ChainStrNum' number
 # num_ordinal: Function to provide the ordinal of a given number, language, format and gender
@@ -166 +167 @@
         return True
     return False
+
+def Ntchar(char,N):
+    """ Function to repeat a number of times a given tring
+      char= string to repeat
+      N= quantity of repetitions
+    >>> Ntchar('c', 4)
+    cccc
+    """
+    fname = 'Ntchar'
+
+    newstr = ''
+    for it in range(N): newstr = newstr + char
+
+    return newstr
 
 def values_line(line, splitv, chars):
@@ -12818 +12833 @@
 #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'
+def WRFsetup(namelistns, params='WRFbasic', output='twopoints@'):
+    """ Function to check the set-up of a series of model namelist
+      namelistns= ',' separated list of namelist files
       params= parameters to check
-        'basic': a pre-determined list of basic parameters (default)
+        'WRFbasic': a pre-determined WRF 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)
+        'textabcol': a LaTeX-table column formated for each model 
+        'textabrow': a LaTeX-table row formated for each model
+        'twopoints@': as ':' separated list of values [attrn]@[attrv]:... (default) for each
+          model with ';' for the following model
       WRFsetup('namelist.input')
     """
@@ -12868 +12884 @@
         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(' ','')
+    # namelist files
+    nmlfs = str_list(namelistns, ',')
+    Nfiles = len(nmlfs)
+
+    # Number of total parameters to look for
+    Npars = 0
+    for kn in lookparams.keys():
+        Npars = Npars + len(lookparams[kn])
+
+    allfparams = {}
+    for namelistn in nmlfs:
+        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('\r', '').replace('\t','').split('=')
+                    paramn = ilvals[0].replace(' ','')
+                    pvals = ilvals[1].replace(' ','')
+                else:
+                    ilvals = line.replace('\n', '').replace('\r', '').replace('\t','').replace(' ', '')
+                    pvals = pvals + ilvals
+        allfparams[namelistn] = fparams
+
+    # Output
+    ##
+    outl = ''
+    # LaTeX table with a column per file
+    if output == 'textabcol':
+        ipar = 0
+        outl = '\\begin{tabular}{' + Ntchar('c',Nfiles+1) + '}\n' +                  \
+         '{\\bfseries{filename}}'
+        for fn in nmlfs:
+            outl = outl + ' & {\\bfseries{' + fn.replace('_', '\\_') + '}}'
+        outl = outl + ' \\\\ \hline\n'
+        for secn in secnames:
+            psec = lookparams[secn]
+            ipar = 0
+            for ps in psec:
+                ifile = 0
+                outl = outl + '{\\bfseries{' + ps + '}}'
+                for fn in nmlfs:
+                    fparams = allfparams[fn]
+                    if not fparams.has_key(ps):
+                        #print infmsg
+                        #print '  ' + fname + ": provided namelist file '" + fn +     \
+                        #  "' has not parameter '" + ps + "' !!"
+                        val = '-'
+                    else:
+                        val = fparams[ps].replace('_', '\\_')
+                    outl = outl + ' & ' + val
+                    ifile = ifile + 1
+                outl = outl + ' \\\\ \n'
+                ipar = ipar + 1
+        outl = outl + '\\end{tabular}\n'
+
+    # LaTeX table with a row per file
+    elif output == 'textabrow':
+        ifile = 0
+        outl = '\\begin{tabular}{' + Ntchar('c',Npars+1) + '}\n' +                   \
+           '{\\bfseries{filename}}'
+        for secn in secnames:
+            psec = lookparams[secn]
+            for ps in psec:
+                outl = outl + ' & {\\bfseries{' + ps.replace('_', '\\_') + '}}'
+        outl = outl + ' \\\\ \\hline\n'
+
+        for fn in nmlfs:
+            fparams = allfparams[fn]
+            ipar = 0
+            outl = outl + fn.replace('_', '\\_')
+            for secn in secnames:
+                print secn + "&"
+                psec = lookparams[secn]
+                for ps in psec:
+                    if not fparams.has_key(ps):
+                        #print infmsg
+                        #print '  ' + fname + ": provided namelist file '" + fn +     \
+                        #  "' has not parameter '" + ps + "' !!"
+                        val = '-'
+                    else:
+                        val = fparams[ps].replace('_', '\\_')
+                    outl = outl + ' & ' + val
+            outl = outl + '\\\\ \n'
+        outl = outl + '\\end{tabular}\n'
+
+    # ASCII line
+    elif output == 'twopoints@':
+        ifile = 0
+        for fn in nmlfs:
+            fparams = allfparams[fn]
+            ipar = 0
+            if ifile == 0:
+                outl = fn
             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
+                outl = outl + ';' + fn
+
+            for secn in secnames:
+                print secn + "&"
+                psec = lookparams[secn]
+                for ps in psec:
+                    if not fparams.has_key(ps):
+                        print infmsg
+                        print '  ' + fname + ": provided namelist file '" + fn +     \
+                          "' has not parameter '" + ps + "' !!"
+                        val = '-'
+                    else:
+                        val = fparams[ps]
+
+                    if ipar == 0: outl = ps + '@' + val
+                    else: outl = outl + ':' + ps + '@' + val
+
+                    ipar = ipar + 1
+            ifile = ifile + 1
+    else:
+        print errormsg
+        print '  ' + fname + ": kind of output '" + output + "' not ready !!"
+        print '    available ones:', availoutput
+        quit(-1)
 
     return outl
 
-#print WRFsetup('/home/lluis/estudios/RELAMPAGO/SimCoor/UBA/namelist.input', 'specific,start_year:end_year', 'textabcol')
+#print WRFsetup('/home/lluis/estudios/RELAMPAGO/SimCoor/UBA/namelist.input,/home/lluis/estudios/RELAMPAGO/SimCoor/SMN/namelist.input,/home/lluis/estudios/RELAMPAGO/SimCoor/NOA-IERSD/namelist.input,/home/lluis/estudios/RELAMPAGO/SimCoor/UBAmili/namelist.input', 'basic', 'textabcol')
 
 #quit()