view datafile.py @ 65:29bcef559283

Add initiateClear and a few others and use it to clear buffers on setup.
author Daniel O'Connor <doconnor@gsoft.com.au>
date Tue, 19 Jan 2021 16:06:29 +1030
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]