internship/data/processing/subdomain.py

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"))
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)