54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
import argparse
|
|
import configparser
|
|
import logging
|
|
import pathlib
|
|
|
|
import numpy as np
|
|
|
|
parser = argparse.ArgumentParser(description="Pre-process bathymetry")
|
|
parser.add_argument("-v", "--verbose", action="count", default=0)
|
|
args = parser.parse_args()
|
|
|
|
logging.basicConfig(level=max((10, 20 - 10 * args.verbose)))
|
|
log = logging.getLogger("bathy")
|
|
|
|
log.info("Starting bathymetry pre-processing")
|
|
config = configparser.ConfigParser()
|
|
config.read("config.ini")
|
|
|
|
artha = np.array(
|
|
(config.getfloat("artha", "lon"), config.getfloat("artha", "lat"))
|
|
)
|
|
buoy = np.array(
|
|
(config.getfloat("buoy", "lon"), config.getfloat("buoy", "lat"))
|
|
)
|
|
log.debug(f"artha: {artha}")
|
|
log.debug(f"buoy: {buoy}")
|
|
|
|
domain = np.stack((artha, buoy))
|
|
domain.sort(axis=0)
|
|
log.debug(f"domain: {domain}")
|
|
|
|
domain[0] -= 0.002
|
|
domain[1] += 0.002
|
|
log.debug(f"domain: {domain}")
|
|
|
|
inp_root = pathlib.Path(config.get("inp", "root"))
|
|
out_root = pathlib.Path(config.get("out", "root"))
|
|
out_root.mkdir(exist_ok=True)
|
|
bathy_inp = inp_root.joinpath(config.get("inp", "base"))
|
|
bathy_out = out_root.joinpath(config.get("out", "sub"))
|
|
log.info(f"Reading bathymetry from '{bathy_inp}'")
|
|
raw_bathy = np.loadtxt(bathy_inp)
|
|
log.debug(f"Initial size: {raw_bathy.shape}")
|
|
|
|
bathy = raw_bathy[
|
|
((raw_bathy[:, :2] > domain[0]) & (raw_bathy[:, :2] < domain[1])).all(
|
|
axis=1
|
|
)
|
|
]
|
|
del raw_bathy
|
|
log.debug(f"Final size: {bathy.shape}")
|
|
|
|
log.info(f"Saving subdomain to 'bathy'")
|
|
np.save(bathy_out, bathy)
|