Skip to content

Commit 093a7aa

Browse files
Jammy2211Jammy2211
authored andcommitted
Array2D uses zoomed_around_mask
1 parent 8dd8755 commit 093a7aa

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

autoarray/structures/arrays/uniform_2d.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from autoconf.fitsable import ndarray_via_fits_from, header_obj_from
88

99
from autoarray.mask.mask_2d import Mask2D
10+
from autoarray.mask.derive.zoom_2d import Zoom2D
1011
from autoarray.structures.abstract_structure import Structure
1112
from autoarray.structures.header import Header
1213
from autoarray.structures.arrays.uniform_1d import Array1D
@@ -463,12 +464,14 @@ def zoomed_around_mask(self, buffer: int = 1) -> "Array2D":
463464
The number pixels around the extracted array used as a buffer.
464465
"""
465466

467+
zoom = Zoom2D(mask=self.mask)
468+
466469
extracted_array_2d = array_2d_util.extracted_array_2d_from(
467470
array_2d=np.array(self.native),
468-
y0=self.mask.zoom_region[0] - buffer,
469-
y1=self.mask.zoom_region[1] + buffer,
470-
x0=self.mask.zoom_region[2] - buffer,
471-
x1=self.mask.zoom_region[3] + buffer,
471+
y0=zoom.region[0] - buffer,
472+
y1=zoom.region[1] + buffer,
473+
x0=zoom.region[2] - buffer,
474+
x1=zoom.region[3] + buffer,
472475
)
473476

474477
mask = Mask2D.all_false(

0 commit comments

Comments
 (0)