handle boundary case

Author: Oisin-M

hat/station_mapping/station_mapping.py

@@ -73,10 +73,11 @@ def conduct_mapping(
                 best_error_1d_index = np.nanargmin(error)
                 min_index = np.unravel_index(best_error_1d_index, shape)
                 best_error = error[best_error_1d_index]
-                # 1d equivalent to [self.max_search_distance, self.max_search_distance]
-                closest_1d_index = (
-                    self.max_search_distance * (2 * self.max_search_distance + 1) + self.max_search_distance
-                )
+
+                center_offset_x = closest_idx - searchbox_min_x
+                center_offset_y = closest_idy - searchbox_min_y
+                subset_width = searchbox_max_y - searchbox_min_y
+                closest_1d_index = center_offset_x * subset_width + center_offset_y
 
                 closest_error = error[closest_1d_index]
 
@@ -92,9 +93,10 @@ def conduct_mapping(
                     indy = closest_idy
                     best_error = closest_error
             except ValueError:
-                closest_1d_index = (
-                    self.max_search_distance * (2 * self.max_search_distance + 1) + self.max_search_distance
-                )
+                center_offset_x = closest_idx - searchbox_min_x
+                center_offset_y = closest_idy - searchbox_min_y
+                subset_width = searchbox_max_y - searchbox_min_y
+                closest_1d_index = center_offset_x * subset_width + center_offset_y
 
                 closest_error = error[closest_1d_index]
                 indx = closest_idx

pyproject.toml

@@ -72,7 +72,7 @@ dependencies = [
 [project.scripts]
     hat-extract-timeseries = "hat.cli:extractor_cli"
     hat-hydrostats = "hat.cli:stat_calc_cli"
-    hat-stations-mapping = "hat.cli:mapper_cli"
+    hat-station-mapping = "hat.cli:mapper_cli"
 
 # Linting settings
 [tool.ruff]