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.genfromtxt(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 ) ] log.debug(f"Final size: {bathy.shape}") log.info(f"Saving subdomain to 'bathy'") np.save(bathy_out, bathy)