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

Add support for multiple timeseries

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-03-31 10:59:30 +02:00
parent b3af9fd66f
commit 24d7407dfb
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
2 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
data/config.ini
View file

@ -5,7 +5,7 @@ hires=bathyhires.dat
hstru=Hstru.dat
poro=Poro.dat
psize=Psize.dat
raw_ts=201702281700.raw
raw_ts=201702281700.raw,201702281730.raw
hires_step=0.5
[out]

19
data/processing/ts.py
View file

@ -21,22 +21,25 @@ config.read(args.config)
inp_root = pathlib.Path(config.get("inp", "root"))
out_root = pathlib.Path(config.get("out", "root"))
inp_ts = inp_root.joinpath(config.get("inp", "raw_ts"))
out_ts = out_root.joinpath("ts.dat")
raw_ts = np.loadtxt(
inp_ts,
dtype=[("state", int), ("z", float), ("y", float), ("x", float)],
delimiter=",",
max_rows=2304,
)
raw_ts = []
for tsi in config.get("inp", "raw_ts").split(","):
raw_ts.append(np.loadtxt(
inp_root.joinpath(tsi),
dtype=[("state", int), ("z", float), ("y", float), ("x", float)],
delimiter=",",
max_rows=2304,
))
n = len(raw_ts)
raw_ts = np.concatenate(raw_ts)
log.debug(f"{raw_ts=}")
if (errs := np.count_nonzero(raw_ts["state"])) != 0:
log.warning(f"{errs} transmission errors!")
log.debug(f"{dict(zip(*np.unique(raw_ts['state'], return_counts=True)))}")
t = np.linspace(0, 30 * 60, 2305)[:-1]
t = np.linspace(0, 30 * 60 * n, 2304*n+1)[:-1]
log.debug(f"{t=}")
log.info(f"Saving timeseries to '{out_ts}'")