1
Fork 0
internship/swash/processing/read_swash.py

58 lines
1.5 KiB
Python
Raw Normal View History

2022-03-30 10:53:11 +02:00
import logging
import subprocess
import tempfile
2022-03-02 14:25:53 +01:00
import numpy as np
2022-03-30 10:53:11 +02:00
log = logging.getLogger("read_swash")
class ReadSwash:
def __init__(self):
self._n_x = None
self._n_t = None
self._t = None
self._x = None
@classmethod
def read_nohead(cls, path):
2022-03-30 10:53:11 +02:00
log.info(f"Replacing \\s with \\n in '{path}'")
2022-03-29 15:34:26 +02:00
subprocess.run(("sed", "-i", r"s/\s\+/\n/g", path))
2022-03-30 10:53:11 +02:00
log.info(f"Loading '{path}'")
2022-03-15 11:17:17 +01:00
return np.loadtxt(path)
def read_time(self, path):
self._t = np.unique(self.read_nohead(path))
self._n_t = self._t.size
return self.t
def read_x(self, path):
self._x = np.unique(self.read_nohead(path))
self._n_x = self._x.size
return self.x
2022-03-04 10:43:45 +01:00
def read_scalar(self, path, const=False):
if const:
2022-03-15 10:43:29 +01:00
return self.read_nohead(path).reshape((self._n_t, self._n_x))[0, :]
return self.read_nohead(path).reshape((self._n_t, self._n_x))
2022-03-15 11:37:59 +01:00
def read_const(self, path):
return self.read_scalar(path, const=True)
def read_vector(self, path):
2022-03-15 10:43:29 +01:00
return self.read_nohead(path).reshape((self._n_t, 2, self._n_x))
def read_scalar_lay(self, path):
2022-03-15 10:43:29 +01:00
return self.read_nohead(path).reshape((self._n_t, -1, self._n_x))
def read_vector_lay(self, path):
return self.read_nohead(path).reshape((self._n_t, -1, 2, self._n_x))
@property
def t(self):
return self._t
@property
def x(self):
return self._x