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

44 lines
1.2 KiB
Python
Raw Normal View History

2022-04-07 10:49:17 +02:00
import argparse
import configparser
from datetime import datetime, time
import logging
import pathlib
import re
from multiprocessing.pool import ThreadPool
import numpy as np
import scipy.io as sio
from .read_mat import ReadSwash
2022-04-07 10:49:17 +02:00
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}'")
2022-04-07 11:50:49 +02:00
swr = ReadSwash(sws_out, inp)
swr.save("t")
swr.save("x")
swr.save("c", "botl", "Botlev")
swr.save("s", "watl", "Watlev")
swr.save("s", "press", "Press")
swr.save("v", "vel", "vel")
swr.save("sk", "zk", "zk")
swr.save("sk", "nhprsk", "Nprs_k")
swr.save("sk", "pressk", "Pres_k")
swr.save("sk", "vz", "w")
swr.save("vk", "velk", "vel_k")