Changeset 565 in lmdz_wrf for trunk/tools

Timestamp:

Jul 6, 2015, 1:28:00 PM (10 years ago)

Author:

lfita

Message:

Adding `values_fortran_fmt', to retrieve values from a fortran FMT command
Adding `rmNOnum': Removing from a string all that characters which are not numbers

File:

• trunk/tools/nc_var_tools.py (modified) (1 diff)

trunk/tools/nc_var_tools.py

@@ -16307 +16307 @@
 #TimeSplitmat(fileobj, 'time', 'time', 'all', 'd,H')
 
-#quit()
+def rmNOnum(string):
+    """ Removing from a string all that characters which are not numbers
+    # From: http://stackoverflow.com/questions/4289331/python-extract-numbers-from-a-string
+    """
+    fname = 'rmNOnum'
+
+    newstr = str(re.findall("[-+]?\d+[\.]?\d*", string)[0])
+
+    return newstr
+
+def values_fortran_fmt(lin,fFMT):
+    """ Function to give back the values of an ASCII line according to its fortran printing format
+      lin= ASCII line
+      fFMT= list with the fortran FORMAT formats
+    forline = 'Natchitoches (RGNL)        1 11 0011  ( 31.733, -93.100) (  28, 157) ( 31.761, -93.113)   41.2 meters'
+    formats = ['A26', 'I2', 'I3', 'A6', 'A2', 'F7.3', 'A1', 'F8.3', 'A3', 'I4', 'A1', 'I4',
+      'A3', 'F7.3', 'A1', 'F8.3', 'A2', 'F6.1', 'A7']
+    >>> values_fortran_fmt(forline, formats)
+    ['Natchitoches (RGNL)        ', 1, 11, ' 0011  ', ' ( ', 31.733, ', ', -93.1, ') ( ', 28, ', ', 157, ')
+      ( ', 31.761, ', ', -93.113, ')  ', 41.2, ' meters']
+    """
+    fname = 'values_fortran_fmt'
+
+    afmts = ['A', 'D', 'F', 'I']
+
+    if lin == 'h':
+        print fname + '_____________________________________________________________'
+        print values_fortran_fmt.__doc__
+        quit()
+
+    fvalues = []
+    ichar=0
+    ival = 0
+    for ft in formats:
+        Lft = len(ft)
+        print ft, Lft
+      
+        if ft[0:1] == 'A' or ft[0:1] == 'a':
+            Lfmt = int(ft[1:Lft+1])
+            fvalues.append(lin[ichar:ichar+Lfmt+1])
+            ichar = ichar + Lfmt
+        elif ft[0:1] == 'F' or ft[0:1] == 'f':
+            if ft.find('.') != -1:
+                Lft = len(ft.split('.')[0])
+            Lfmt = int(ft[1:Lft])
+            fvalues.append(np.float(rmNOnum(lin[ichar:ichar+Lfmt+1])))
+            ichar = ichar + Lfmt
+        elif ft[0:1] == 'D' or ft[0:1] == 'd':
+            if ft.find('.') != -1:
+                Lft = len(ft.split('.')[0])
+            Lfmt = int(ft[1:Lft])
+            fvalues.append(np.float64(rmNOnum(lin[ichar:ichar+Lfmt+1])))
+            ichar = ichar + Lfmt
+        elif ft[0:1] == 'I' or ft[0:1] == 'i':
+            Lfmt = int(ft[1:Lft+1])
+            fvalues.append(int(rmNOnum(lin[ichar:ichar+Lfmt+1])))
+            ichar = ichar + Lfmt
+        elif ft.find('X') != -1 or ft.find('x') != -1:
+            print '    ' + fname + ': skipping space'
+            ichar = ichar + len(ft)
+        else:
+            print errormsg
+            print '  ' + fname + ": format '" + ft[0:1] + "' not ready!!"
+            print '    Available formats:',afmts
+            quit(-1)
+
+        print '    ', ichar, fvalues[ival-1]
+        ival = ival + 1
+
+    print fvalues
+
+    return fvalues
+
+forline = 'Natchitoches (RGNL)        1 11 0011  ( 31.733, -93.100) (  28, 157) ( 31.761, -93.113)   41.2 meters'
+formats = ['A26', 'I2', 'I3', 'A6', 'A2', 'F7.3', 'A1', 'F8.3', 'A3', 'I4', 'A1', 'I4', \
+  'A3', 'F7.3', 'A1', 'F8.3', 'A2', 'F6.1', 'A7']
+
+values_fortran_fmt(forline, formats)
+
+quit()
 
 """operation to make: