LFS wtf

data/processing/subdomain.py

@@ -1,3 +1,53 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:cee15595a8dac4e684da41c7b1cb9b80bd091a6879407ad516114d46addd0af0
-size 1435
+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)