view datafile.py @ 54:42621291eb9b

Improve usage message.
author Daniel O'Connor <doconnor@gsoft.com.au>
date Thu, 02 May 2013 17:55:20 +0930
parents 3d2306e39700
children 91b476ebc0f2
line wrap: on
line source

import exceptions
import numpy

class DataFile(object):
    def __init__(self, fname):
        f = file(fname)
        self.opts = {}
        for line in f:
            key, value = line.strip().split(' ', 1)
            if key == "XDATA":
                self.freqs = numpy.fromstring(value, sep = ', ')
                continue
            
            if key == "YDATA":
                self.powers = numpy.fromstring(value, sep = ', ')
                continue
            
            try:
                self.opts[key] = int(value)
            except exceptions.ValueError, e:
                try:
                    self.opts[key] = float(value)
                except exceptions.ValueError, e:
                    self.opts[key] = value

    def __getitem__(self, k):
        return self.opts[k]