view datafile.py @ 46:a6e5c97b4767

Add noise test
author Daniel O'Connor <darius@dons.net.au>
date Thu, 20 Dec 2012 11:48:19 +1030
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]