Bathymetry processing (subdomain + 1d interpolation)

data/processing/projection.py

@@ -0,0 +1,82 @@
+import argparse
+import configparser
+import logging
+import pathlib
+
+import matplotlib.pyplot as plt
+import numpy as np
+from scipy import interpolate
+
+from .lambert import Lambert
+
+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")
+
+bathy_inp = pathlib.Path(config.get("bathy", "sub"))
+bathy_out = pathlib.Path(config.get("bathy", "out"))
+
+log.info(f"Loading bathymetry from {bathy_inp}")
+bathy_curvi = np.load(bathy_inp)
+
+projection = Lambert()
+bathy = np.stack(
+    (
+        *projection.cartesian(bathy_curvi[:, 0], bathy_curvi[:, 1]),
+        bathy_curvi[:, 2],
+    ),
+    axis=1
+)
+log.debug(f"Cartesian bathy: {bathy}")
+
+artha_curvi = np.array(
+    (config.getfloat("artha", "lon"), config.getfloat("artha", "lat"))
+)
+buoy_curvi = np.array(
+    (config.getfloat("buoy", "lon"), config.getfloat("buoy", "lat"))
+)
+
+artha = np.asarray(projection.cartesian(*artha_curvi))
+buoy = np.asarray(projection.cartesian(*buoy_curvi))
+
+
+def display():
+    x = np.linspace(bathy[:, 0].min(), bathy[:, 0].max())
+    y = np.linspace(bathy[:, 1].min(), bathy[:, 1].max())
+
+    X, Y = np.meshgrid(x, y)
+
+    Z = interpolate.griddata(
+        bathy[:, :2], bathy[:, 2], (X, Y), method="nearest"
+    )
+
+    fix, ax = plt.subplots()
+    ax.pcolormesh(X, Y, Z)
+    ax.scatter(*artha, c="k")
+    ax.scatter(*buoy, c="k")
+
+    ax.axis("equal")
+    return ax
+
+
+D = np.diff(np.stack((artha, buoy)), axis=0)
+x = np.arange(-150, np.sqrt((D**2).sum()) + 150)
+theta = np.angle(D.dot((1, 1j)))
+
+coords = artha + (x * np.stack((np.cos(theta), np.sin(theta)))).T
+
+z = interpolate.griddata(bathy[:,:2], bathy[:,2], coords)
+
+ax = display()
+ax.scatter(*coords.T, c="k", marker=".")
+
+fig_1d, ax_1d = plt.subplots()
+ax_1d.plot(x, z)
+plt.show(block=True)