Added Python script that converts csv files to cpp functions, small edits on docpage, svn properties setup.

git-svn-id: https://svn.osgeo.org/grass/grass/trunk@42745 15284696-431f-4ddb-bdfa-cd5b030d7da7

imagery/i.atcorr/create_iwave.py

@@ -0,0 +1,256 @@
+#!/usr/bin/python2.6
+"""
+Created on Sat Mar 27 11:35:32 2010
+
+Program to interpolate filter function to correct 
+step. Should be 2.5 nm
+Then output filter function in a format similar to
+what is needed in the Iwave.cpp file
+
+Needs numpy and scipy
+
+@author: daniel
+contact: daniel {dot} victoria {at} gmail {dot} com
+
+usage() explains how this is supposed to work
+Basically it needs a .csv file with spectral response for each
+band in a column. First column has to be wavelength (nm)
+First line (and only first) is a header with Wl, followed by band names
+file name is used for sensor name
+
+Self though programer warning: beware of fractured python bellow
+and fractured english also :)
+"""
+import os
+import sys
+import numpy as np
+from scipy import interpolate
+
+def usage():
+    """How to use this..."""
+    print "create_iwave.py <csv file>"
+    print "Generate filter function IWave.cpp template from csv file"
+    print "csv file must have wl response for each band in each column"
+    print "first line must be a header with wl followed by band names"
+    print "following lines will be the data."
+    print "If response is null, leave empty in csv file. Ex.:"
+    print "WL(nm),band 1,band 2,band 3,band 4"
+    print "455,0.93,,,"
+    print "485,0.94,0.00,,"
+    print "545,0.00,0.87,0.00,"
+    print "Program will interpolate filter function to 2.5 nm steps"
+    print "and output a cpp template file in the IWave format"
+
+def read_input(csvfile):
+    """
+    Func to read input file
+    return number of bands and array of values for each band
+    should be a .csv file with the values
+    of the filter function for each band in the sensor
+    one column for band
+    first line must have a header with sensor band name
+    first column is wavelength
+    """
+    infile = open(csvfile, 'r')
+    
+    # get number of bands and band names
+    bands = infile.readline().split(',')
+    bands.remove(bands[0])
+    bands[-1] = bands[-1].strip()
+    
+    infile.close()
+    
+    # create converter dictionary for import
+    # fix nodata or \n
+    conv = {}
+    for b in range(len(bands)):
+        conv[b+1] = lambda s: float(s or -99)
+    
+    
+    values = np.loadtxt(csvfile, delimiter=',', skiprows=1, converters = conv)
+    
+    return(bands, values)
+
+def interpolate_band(values):
+    """
+    Receive wavelength and response for one band
+    interpolate at 2.5 nm steps
+    return interpolated filter func
+    and min, max wl values
+    values must be numpy array with 2 columns
+    """
+    
+    # remove nodata (-99) lines in values array
+    # where response is nodata?
+    w = values[:,1] >= 0
+    response = values[w]
+    
+    # interpolating
+    f = interpolate.interp1d(response[:,0],response[:,1])
+    
+    filter_f = f(np.arange(response[0,0], response[-1,0], 2.5))
+    
+    return(filter_f, (response[0,0], response[-1,0]))
+
+def plot_filter(values):
+    """Plot wl response values and interpolated
+    filter function. This is just for checking...
+    value is a 2 column numpy array
+    function has to be used inside Spyder python environment
+    """
+    filter_f, limits = interpolate_band(values)
+    
+    # removing nodata
+    w = values[:,1] >= 0
+    response = values[w]
+    
+    plot(response[:,0],response[:,1], 'ro')
+    plot(arange(limits[0], limits[1], 2.5), filter_f)
+    
+    return
+
+def pretty_print(filter_f):
+    """
+    Create pretty string out of filter function
+    8 values per line, with spaces, commas and all the rest
+    """
+    nlines = int(len(filter_f)/8)
+    rest = len(filter_f) - nlines*8
+    pstring = '    '
+    
+    if rest == 0: # multiple of 8; special case
+        for l in range(nlines):
+            for i in range(8):
+                if l == nlines-1 and i == 7: # last value
+                    pstring = pstring+' %.2f };\n' % (filter_f[l*8+i])
+                else:
+                    pstring = pstring+' %.2f,' % (filter_f[l*8+i])
+            pstring = pstring+'\n    '
+    else:
+        for l in range(nlines):
+            for i in range(8):
+                pstring = pstring+'    %.2f,' % (filter_f[l*8+i])
+            pstring = pstring+'\n    '
+        # adding rest of values in filter_f
+        for i in range(rest):
+            if i == rest-1: # last value
+                pstring = pstring+'    %.2f };\n' % (filter_f[nlines*7+i])
+            else:
+                pstring = pstring+'    %.2f,' % (filter_f[nlines*7+i])
+    
+    return pstring
+
+def write_cpp(bands, values, sensor):
+    """
+    from bands, values and sensor name
+    create output file in cpp style
+    needs other functions: interpolate_bands, pretty_print
+    """
+    
+    # getting necessary data
+    # single or multiple bands?
+    if len(bands) == 1:
+        filter_f, limits = interpolate_band(values)
+    else:
+        filter_f = []
+        limits = []
+        for b in range(len(bands)):
+            fi, li = interpolate_band(values[:,[0,b+1]])
+            filter_f.append(fi)
+            limits.append(li)
+    
+    # writing...
+    outfile = open(sensor+"_cpp_template.txt", 'w')
+    outfile.write('/* Following filter function created using create_iwave.py */\n\n')
+    
+    if len(bands) == 1:
+        outfile.write('void IWave::%s()\n{\n\n' % (sensor))
+    else:
+        outfile.write('void IWave::%s(int iwa)\n{\n\n' % (sensor))
+        
+    # single band case
+    if len(bands) == 1:
+        outfile.write('    /* %s of %s */\n' % (bands[0], sensor))
+        outfile.write('    static const float sr[%i] = {\n' % (len(filter_f)))
+        filter_text = pretty_print(filter_f)
+        outfile.write(filter_text)
+        
+        # calculate wl slot for band start
+        # slots range from 250 to 4000 at 2.5 increments (total 1500)
+        s_start = int((limits[0] - 250)/2.5)
+        
+        outfile.write('\n')
+        outfile.write('    ffu.wlinf = %.4ff;\n' % (limits[0]/1000.0))
+        outfile.write('    ffu.wlsup = %.4ff;\n' % (limits[1]/1000.0))
+        outfile.write('    int i = 0;\n')
+        outfile.write('    for(i = 0; i < %i; i++)\tffu.s[i] = 0;\n' % (s_start))
+        outfile.write('    for(i = 0; i < %i; i++)\tffu.s[%i+i] = sr[i];\n' % (len(filter_f), s_start))
+        outfile.write('    for(i = %i; i < 1501; i++)\tffu.s[i] = 0;\n' % (s_start + len(filter_f)))
+        outfile.write('}\n')
+        
+    else: # more than 1 band
+        # writing bands
+        for b in range(len(bands)):
+            outfile.write('    /* %s of %s */\n' % (bands[b], sensor))
+            outfile.write('    static const float sr%i[%i] = {\n' % (b+1,len(filter_f[b])))
+            filter_text = pretty_print(filter_f[b])
+            outfile.write(filter_text+'\n')
+        
+        # writing band limits
+        for b in range(len(bands)):
+            inf = str([i[0] for i in limits])
+            sup = str([i[1] for i in limits])
+            
+            for j in zip(['[',']'],['{','}']):
+                inf = inf.replace(j[0],j[1])
+                sup = sup.replace(j[0],j[1])
+        
+        outfile.write('    static const float wli[%i] = %s;\n' % (len(bands), inf))
+        outfile.write('    static const float wls[%i] = %s;\n' % (len(bands), sup))
+        
+        outfile.write('\n')
+        outfile.write('    ffu.wlinf = (float)wli[iwa-1];\n')
+        outfile.write('    ffu.wlsup = (float)wls[iwa-1];\n\n')
+        
+        outfile.write('    int i;\n')
+        outfile.write('    for(i = 0; i < 1501; i++) ffu.s[i] = 0;\n\n')
+        
+        outfile.write('    switch(iwa)\n    {\n')
+        
+        # now start of case part...
+        for b in range(len(bands)):
+            s_start = int((limits[b][0] - 250)/2.5)
+            outfile.write('    case %i: for(i = 0; i < %i; i++)  ffu.s[%i+i] = sr%i[i];\n' % ((b+1), len(filter_f[b]), s_start, (b+1)))
+            outfile.write('        break;\n')
+        outfile.write('    }\n}\n')
+        
+    return
+
+def main():
+    """ control function """
+    
+    inputfile = sys.argv[1]
+    
+    # getting sensor name from full csv file name
+    sensor = os.path.splitext(os.path.basename(inputfile))[0]
+    
+    print "Getting sensor name from csv file: %s" % (sensor)
+    
+    # getting data from file
+    bands, values = read_input(inputfile)
+    
+    # writing file
+    write_cpp(bands, values, sensor)
+    
+    print "Filter function written to %s" % (sensor+"_cpp_template.txt")
+    print "Please check file for possible errors before inserting into IWave.cpp"
+    print "Don't forget to add necessary data to IWave.h"
+    
+    return
+
+if __name__ == '__main__':
+    if len(sys.argv) == 1:
+        usage()
+        sys.exit()
+    else:
+        main()

imagery/i.atcorr/i.atcorr.html

@@ -141,7 +141,7 @@ the ascendant node at equator</td>
 </tr>
 
 <tr>
-<td>10</td>
+<td>11</td>
 <td><b>avnir</b></td>
 <td>enter month,day,hh.ddd,long.,lat. *</td>
 </tr>