view datafile.py @ 69:7386f2888508

Make function more configurable
author Daniel O'Connor <doconnor@gsoft.com.au>
date Wed, 04 Aug 2021 16:07:44 +0930
parents 91b476ebc0f2
children
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 as e:
                try:
                    self.opts[key] = float(value)
                except exceptions.ValueError as e:
                    self.opts[key] = value

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