62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
import logging
|
|
import subprocess
|
|
import tempfile
|
|
|
|
import numpy as np
|
|
|
|
|
|
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):
|
|
with tempfile.TemporaryFile(mode="w+t") as tmpfile:
|
|
log.info(f"Replacing \\s with \\n in '{path}'")
|
|
subprocess.run(("sed", r"s/\s\+/\n/g;/^$/d", path), stdout=tmpfile, text=True)
|
|
log.info(f"Loading '{path}'")
|
|
tmpfile.seek(0)
|
|
a = np.asarray(tmpfile.readlines(), dtype=float)
|
|
log.debug(f"path={a}")
|
|
return a
|
|
|
|
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
|
|
|
|
def read_scalar(self, path, const=False):
|
|
if const:
|
|
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))
|
|
|
|
def read_const(self, path):
|
|
return self.read_scalar(path, const=True)
|
|
|
|
def read_vector(self, path):
|
|
return self.read_nohead(path).reshape((self._n_t, 2, self._n_x))
|
|
|
|
def read_scalar_lay(self, path):
|
|
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
|