add mapper_pixels_zeroed_indices

Author: Jammy2211

autoarray/preloads.py

@@ -24,6 +24,7 @@ def __init__(
         source_pixel_zeroed_indices: np.ndarray = None,
         image_plane_mesh_grid_list: np.ndarray = None,
         linear_light_profile_blurred_mapping_matrix=None,
+        mapper_pixels_zeroed_indices: np.ndarray = None,
     ):
         """
         Stores preloaded arrays and matrices used during pixelized linear inversions, improving both performance
@@ -113,6 +114,12 @@ def __init__(
                 else:
                     self.image_plane_mesh_grid_list.append(None)
 
+        if mapper_pixels_zeroed_indices is not None:
+
+            self.mapper_pixels_zeroed_indices = np.array(
+                mapper_pixels_zeroed_indices
+            )
+
         if linear_light_profile_blurred_mapping_matrix is not None:
 
             self.linear_light_profile_blurred_mapping_matrix = np.array(

autoarray/structures/mesh/delaunay_2d.py

@@ -14,7 +14,7 @@
 from autoarray.inversion.pixelization.mesh import mesh_numba_util
 
 
-def scipy_delaunay(points_np, query_points_np, source_pixel_zeroed_indices):
+def scipy_delaunay(points_np, query_points_np, mapper_pixel_zeroed_indices):
     """Compute Delaunay simplices (simplices_padded) and Voronoi areas in one call."""
 
     max_simplices = 2 * points_np.shape[0]
@@ -57,7 +57,7 @@ def scipy_delaunay(points_np, query_points_np, source_pixel_zeroed_indices):
     areas[areas > max_area] = max_area
 
     # ---------- Set zeroed source pixels to max area ----------
-    areas[source_pixel_zeroed_indices] = max_area
+    areas[mapper_pixel_zeroed_indices] = max_area
 
     split_point_areas = 0.5 * np.sqrt(areas)
 
@@ -80,7 +80,7 @@ def scipy_delaunay(points_np, query_points_np, source_pixel_zeroed_indices):
     return points, simplices_padded, mappings, split_points, splitted_mappings
 
 
-def jax_delaunay(points, query_points, source_pixel_zeroed_indices):
+def jax_delaunay(points, query_points, mapper_pixel_zeroed_indices):
     import jax
     import jax.numpy as jnp
 
@@ -107,7 +107,7 @@ def jax_delaunay(points, query_points, source_pixel_zeroed_indices):
         ),
         points,
         query_points,
-        source_pixel_zeroed_indices
+        mapper_pixel_zeroed_indices
     )
 
 
@@ -457,15 +457,15 @@ def delaunay(self) -> "scipy.spatial.Delaunay":
             points, simplices, mappings, split_points, splitted_mappings = jax_delaunay(
                 points=self.mesh_grid_xy,
                 query_points=self._source_plane_data_grid_over_sampled,
-                source_pixel_zeroed_indices=self.preloads.source_pixel_zeroed_indices
+                mapper_pixel_zeroed_indices=self.preloads.mapper_pixel_zeroed_indices
             )
 
         else:
 
             points, simplices, mappings, split_points, splitted_mappings = scipy_delaunay(
                 points_np=self.mesh_grid_xy,
                 query_points_np=self._source_plane_data_grid_over_sampled,
-                source_pixel_zeroed_indices=self.preloads.source_pixel_zeroed_indices,
+                mapper_pixel_zeroed_indices=self.preloads.mapper_pixel_zeroed_indices,
             )
 
         return DelaunayInterface(