root/sodar/trunk/sodar/arrayData.py

#!/usr/bin/python
"""
Module to handle sodar data samples as arrays.
"""

__author__ = 'Chris Calloway'
__email__ = 'cbc@unc.edu'
__copyright__ = 'Copyright 2007 UNC-CH Department of Marine Science'
__license__ = 'GPL2'

import rawData
import formattedData
import numpy as n
import datetime

class ArrayData(object):
    
    """Class to handle Daily sodar file data as arrays."""
    
    def __init__(self, data):
        """Create arrays from formatted data."""
        super(ArrayData, self).__init__()
        self.timeInterval = data.timeInterval
        self.altInterval = data.altInterval
        self.numAltitudes = data.numAltitudes
        self.minAltitude = data.minAltitude
        self.maxAltitude = data.maxAltitude
        self._azimuth(data)
    
    def _azimuth(self, data):
        """Convert direction to polar azimuth in radians CW from North"""
        self.azimuth = n.array([[altitude['DIR']
                                 for altitude
                                 in sample['body']]
                                for sample in data])
        self.azimuth = n.pi * (self.azimuth / 180)


def _main():
    """Process as script from command line."""
    import urllib2
    try:
        rawDataHandle = urllib2.urlopen('http://nemo.isis.unc.edu/'\
                                        'data/nccoos/level0/dukeforest/sodar/'\
                                        'store/2007-06/20070601.dat')
        rawDataString = rawDataHandle.read()
    except:
        raise IOError("Failure to read raw test data")
    rawDataObject = rawData.RawData(rawDataString)
    formattedDataObject = formattedData.FormattedData(rawDataObject)
    arrayDataObject = ArrayData(formattedDataObject)
    print arrayDataObject.azimuth

if __name__ == "__main__":
    _main()