root/sodar/trunk/sodar.py

#!/usr/bin/python
"""
Module to process sodar data.
"""

import optparse
from sodar.utils import openAnything, findMissing 
from sodar import arrayData as a
import pylab as p
import os

def plotSingle(filein, pathout):
    """
    Plot singe sodar raw data file.
    
    plotSingle(filein, pathout) -> rc
    
    filein - path to raw data file.
    fileout - path to plot images directory.
    rc - 0 if successful
    """
    
    walkList = findMissing.findMissing(source, destination)
    if not os.path.exists(pathout):
        os.makedirs(pathout, mode=0775)
        
        try:
            rawDataHandle = openAnything.openAnything(filein)
            rawDataString = rawDataHandle.read()
            rawDataHandle.close()
        except:
            raise IOError("Failure to read raw data.")
        
        rawDataObject = a.rawData.RawData(rawDataString)
        formattedDataObject = a.formattedData.FormattedData(rawDataObject)
        arrayDataObject = a.ArrayData(formattedDataObject)
        
        fig = p.figure()
        p.pcolor(arrayDataObject.uComponents.T)
        p.colorbar()
        fig.savefig(os.path.join(pathout, 'uComponents.png'))
        
        fig = p.figure()
        p.pcolor(arrayDataObject.vComponents.T)
        p.colorbar()
        fig.savefig(os.path.join(pathout, 'vComponents.png'))
        
        fig = p.figure()
        p.pcolor(arrayDataObject.wComponents.T)
        p.colorbar()
        fig.savefig(os.path.join(pathout, 'wComponents.png'))
        
        fig = p.figure()
        p.pcolor(arrayDataObject.echoStrengths.T)
        p.colorbar()
        fig.savefig(os.path.join(pathout, 'echoStrengths.png'))
        
        fig = p.figure()
        p.quiver(arrayDataObject.uComponents,
                 arrayDataObject.vComponents,
                 arrayDataObject.wComponents)
        p.colorbar()
        xmin,xmax,ymin,ymax = p.axis()
        dx,dy = xmax-xmin, ymax-ymin
        p.axis([xmin-0.1*dx, xmax+0.1*dx, ymin-0.1*dy, ymax+0.1*dy])
        p.title('Horizontal wind vectors: time vs. altitude')
        fig.savefig(os.path.join(pathout, 'quiver.png'))
    return 0

def plotAll(source, destination):
    """
    Plot all sodar raw data files.
    
    plotAll(source, destination) -> rc
    
    source - path to raw data directory in NCCOOS format.
    detination - path to plot images directory in NCCOOS format.
    rc - 0 if successful
    """
    
    walkList = findMissing.findMissing(source, destination)
    
    for filein, pathout in walkList:
        plotSingle(filein, pathout)
    return 0

def _main():
    """bin/python %prog [options] /path/to/raw/files/ /path/to/plots/dir/"""
    
    __description__ = 'Plot all sodar raw data files.'
    
    parser = optparse.OptionParser(usage=_main.__doc__,
                                   version='1.0',
                                   description=__description__)
    (values, args) = parser.parse_args()
    (source, destination) = tuple(args)
    plotAll(source, destination)
    return 0

if __name__ == "__main__":
    _main()