m2cce/internship
m2cce/internship
1
Fork 0
Code Issues Activity

Remove old sws_npz scripts

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-04-08 12:43:53 +02:00
parent 86252c67d4
commit 49dec9b76e
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
2 changed files with 0 additions and 109 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

60
swash/processing/read_swash.py
View file

@ -1,60 +0,0 @@
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):
data = []
with path.open() as inp:
for line in inp:
data += line.split()
return np.asarray(data, dtype=float)
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

49
swash/processing/sws_npz.py
View file

@ -1,49 +0,0 @@
import argparse
import configparser
import logging
import pathlib
from multiprocessing.pool import ThreadPool
import numpy as np
from .read_swash import ReadSwash
parser = argparse.ArgumentParser(description="Convert swash output to numpy")
parser.add_argument("-v", "--verbose", action="count", default=0)
parser.add_argument("-c", "--config", default="config.ini")
args = parser.parse_args()
logging.basicConfig(level=max((10, 20 - 10 * args.verbose)))
log = logging.getLogger("sws_npz")
log.info("Starting sws -> npz converter")
config = configparser.ConfigParser()
config.read(args.config)
sws_out = pathlib.Path(config.get("swash", "out"))
inp = pathlib.Path(config.get("post", "inp"))
inp.mkdir(parents=True, exist_ok=True)
log.info(f"Reading swash output from '{sws_out}'")
rsws = ReadSwash()
np.save(inp.joinpath("tsec"), rsws.read_time(sws_out.joinpath("tsec.dat")))
np.save(inp.joinpath("xp"), rsws.read_x(sws_out.joinpath("xp.dat")))
var = {
# "dep": rsws.read_scalar,
"botl": rsws.read_const,
"watl": rsws.read_scalar,
# "pressk": rsws.read_scalar_lay,
# "nhprsk": rsws.read_scalar_lay,
# "zk": rsws.read_scalar_lay,
# "velk": rsws.read_vector_lay,
# "vz": rsws.read_scalar_lay,
"vel": rsws.read_vector,
}
for name, f in var.items():
log.info(f"Converting {name}")
np.save(
inp.joinpath(name),
f(sws_out.joinpath(name).with_suffix(".dat")),
)