From 2cdf0c4b9865f025e52938fb9b772cc79319db09 Mon Sep 17 00:00:00 2001 From: Trygve Laugstøl Date: Tue, 14 May 2019 22:19:15 +0200 Subject: ee: Moving old ltspice code to its own file. --- src/ee/__init__.py | 144 +-------------------------------------- src/ee/ltspice.py | 138 +++++++++++++++++++++++++++++++++++++ src/ee/tools/read_ltspice_raw.py | 4 +- 3 files changed, 143 insertions(+), 143 deletions(-) create mode 100644 src/ee/ltspice.py (limited to 'src') diff --git a/src/ee/__init__.py b/src/ee/__init__.py index 8a128f6..93c73f7 100644 --- a/src/ee/__init__.py +++ b/src/ee/__init__.py @@ -1,13 +1,13 @@ -import math -import numpy as np import re from functools import total_ordering + +import math + from ee.formatting import eng_str __all__ = [ "EeException", "EeVal", - "read_ltspice_raw" ] @@ -17,7 +17,6 @@ class EeException(Exception): @total_ordering class EeVal(object): - from decimal import Decimal units = ['F', 'Ohm', '\u2126', # Ohm symbol 'H'] @@ -104,140 +103,3 @@ class EeVal(object): def __float__(self): return self._value * math.pow(10, self._exp) - -class LtSpiceRaw(object): - def __init__(self, variables, values_first, values_rest): - self.variables = variables - self.values_first = values_first - self.values_rest = values_rest - - def get_values(self, variable): - return self.values[variable.idx] - - def get_variable(self, idx=None, expression=None): - if idx is not None: - return self.variables[idx] - if expression is not None: - v = [v for v in self.variables if v.expression == expression] - if len(v) != 1: - raise Exception('Unknown variable: ' + str(v)) - return v[0] - raise Exception('idx or expression must be given') - - def to_pandas(self): - import pandas - data = [] - for (i, v) in enumerate(self.values_first): - data.append(pandas.Series(v, dtype='float64')) - - data = {} - for (i, v) in enumerate(self.values_first): - data[self.variables[i].expression] = v - - return pandas.DataFrame(data=data) - - -class LtSpiceVariable(object): - def __init__(self, idx, expression, kind): - self.idx = idx - self.expression = expression - self.kind = kind - - -class LtSpiceReader(object): - def __init__(self, f): - self.f = f - self.mode = 'header' - self.headers = {} - self.variables = [] - self.values = [] - - def read_header(self, line): - sep = line.find(':') - key = line[0:sep] - value = line[sep + 1:].strip() - - # print("key='{}', value='{}'".format(key, str(value))) - - if key == 'Binary': - self.set_binary_mode() - elif key == 'Variables': - self.no_variables = int(self.headers['No. Variables']) - self.mode = 'variables' - else: - self.headers[key] = value - - def read_variable(self, line): - parts = line.split('\t') - idx = int(parts[1]) - expression = parts[2] - kind = parts[3] - self.variables.append(LtSpiceVariable(idx, expression, kind)) - - if len(self.variables) == self.no_variables: - self.mode = 'header' - - def set_binary_mode(self): - self.mode = 'binary' - - def read_binary(self): - pos = self.f.tell() - no_points = int(self.headers['No. Points']) - no_variables = int(self.headers['No. Variables']) - - # print("no_points={}, no_variables={}".format(no_points, no_variables)) - - if True: - self.values = np.zeros((no_variables, no_points), dtype='float32') - - for p in range(no_points): - self.values[0][p] = np.fromfile(self.f, count=1, dtype='float64') - row = np.fromfile(self.f, count=no_variables - 1, dtype='float32'); - for v in range(0, len(row)): - self.values[v + 1][p] = row[v] - else: - self.values = np.zeros((no_points, no_variables), dtype='float64') - - for p in range(no_points): - self.values[p][0] = np.fromfile(self.f, count=1, dtype='float64') - row = np.fromfile(self.f, count=no_variables - 1, dtype='float32'); - for v in range(0, len(row)): - self.values[p][v + 1] = row[v] - - def read(self): - while True: - line = self.f.readline(); - - if len(line) == 0: - break - - self.f.read(1) # The data is utf16 encoded so read the extra null byte - - # if this wasn't the last line .readline() includes the newline - if line[-1] == '\n' or line[-1] == 0x0a: - line = line[:-1] - - # print("len(line)={}, line={}".format(len(line), str(line))) - line = line.decode(encoding="utf-16") - - if self.mode == 'header': - self.read_header(line) - elif self.mode == 'variables': - self.read_variable(line) - - # The binary data can't be read with readline() - if self.mode == 'binary': - # self.f.read() - self.read_binary() - break - - if len(self.variables) == 0: - raise Exception("Something didn't quite work when parsing file, no variables found") - - return LtSpiceRaw(self.variables, self.values, []) - - -def read_ltspice_raw(filename): - with open(filename, mode="rb") as f: - r = LtSpiceReader(f) - return r.read() diff --git a/src/ee/ltspice.py b/src/ee/ltspice.py new file mode 100644 index 0000000..3e47bf0 --- /dev/null +++ b/src/ee/ltspice.py @@ -0,0 +1,138 @@ +import numpy as np + +class LtSpiceRaw(object): + def __init__(self, variables, values_first, values_rest): + self.variables = variables + self.values_first = values_first + self.values_rest = values_rest + + def get_values(self, variable): + return self.values[variable.idx] + + def get_variable(self, idx=None, expression=None): + if idx is not None: + return self.variables[idx] + if expression is not None: + v = [v for v in self.variables if v.expression == expression] + if len(v) != 1: + raise Exception('Unknown variable: ' + str(v)) + return v[0] + raise Exception('idx or expression must be given') + + def to_pandas(self): + import pandas + data = [] + for (i, v) in enumerate(self.values_first): + data.append(pandas.Series(v, dtype='float64')) + + data = {} + for (i, v) in enumerate(self.values_first): + data[self.variables[i].expression] = v + + return pandas.DataFrame(data=data) + + +class LtSpiceVariable(object): + def __init__(self, idx, expression, kind): + self.idx = idx + self.expression = expression + self.kind = kind + + +class LtSpiceReader(object): + def __init__(self, f): + self.f = f + self.mode = 'header' + self.headers = {} + self.variables = [] + self.values = [] + + def read_header(self, line): + sep = line.find(':') + key = line[0:sep] + value = line[sep + 1:].strip() + + # print("key='{}', value='{}'".format(key, str(value))) + + if key == 'Binary': + self.set_binary_mode() + elif key == 'Variables': + self.no_variables = int(self.headers['No. Variables']) + self.mode = 'variables' + else: + self.headers[key] = value + + def read_variable(self, line): + parts = line.split('\t') + idx = int(parts[1]) + expression = parts[2] + kind = parts[3] + self.variables.append(LtSpiceVariable(idx, expression, kind)) + + if len(self.variables) == self.no_variables: + self.mode = 'header' + + def set_binary_mode(self): + self.mode = 'binary' + + def read_binary(self): + pos = self.f.tell() + no_points = int(self.headers['No. Points']) + no_variables = int(self.headers['No. Variables']) + + # print("no_points={}, no_variables={}".format(no_points, no_variables)) + + if True: + self.values = np.zeros((no_variables, no_points), dtype='float32') + + for p in range(no_points): + self.values[0][p] = np.fromfile(self.f, count=1, dtype='float64') + row = np.fromfile(self.f, count=no_variables - 1, dtype='float32'); + for v in range(0, len(row)): + self.values[v + 1][p] = row[v] + else: + self.values = np.zeros((no_points, no_variables), dtype='float64') + + for p in range(no_points): + self.values[p][0] = np.fromfile(self.f, count=1, dtype='float64') + row = np.fromfile(self.f, count=no_variables - 1, dtype='float32'); + for v in range(0, len(row)): + self.values[p][v + 1] = row[v] + + def read(self): + while True: + line = self.f.readline(); + + if len(line) == 0: + break + + self.f.read(1) # The data is utf16 encoded so read the extra null byte + + # if this wasn't the last line .readline() includes the newline + if line[-1] == '\n' or line[-1] == 0x0a: + line = line[:-1] + + # print("len(line)={}, line={}".format(len(line), str(line))) + line = line.decode(encoding="utf-16") + + if self.mode == 'header': + self.read_header(line) + elif self.mode == 'variables': + self.read_variable(line) + + # The binary data can't be read with readline() + if self.mode == 'binary': + # self.f.read() + self.read_binary() + break + + if len(self.variables) == 0: + raise Exception("Something didn't quite work when parsing file, no variables found") + + return LtSpiceRaw(self.variables, self.values, []) + + +def read_ltspice_raw(filename): + with open(filename, mode="rb") as f: + r = LtSpiceReader(f) + return r.read() diff --git a/src/ee/tools/read_ltspice_raw.py b/src/ee/tools/read_ltspice_raw.py index f04abb0..c5cd71b 100644 --- a/src/ee/tools/read_ltspice_raw.py +++ b/src/ee/tools/read_ltspice_raw.py @@ -1,7 +1,7 @@ -import ee +import ee.ltspice import sys -raw = ee.read_ltspice_raw(sys.argv[1]) +raw = ee.ltspice.read_ltspice_raw(sys.argv[1]) # print("Variables:") # for i, v in enumerate(raw.variables): -- cgit v1.2.3