Source code for mantidimaging.core.operations.circular_mask.circular_mask

# Copyright (C) 2024 ISIS Rutherford Appleton Laboratory UKRI
# SPDX - License - Identifier: GPL-3.0-or-later
from __future__ import annotations

from functools import partial
from typing import TYPE_CHECKING, Any

import numpy as np
import tomopy
from mantidimaging.core.parallel import shared as ps
from mantidimaging.core.operations.base_filter import BaseFilter
from mantidimaging.gui.utility.qt_helpers import Type

    from import ImageStack

[docs] class CircularMaskFilter(BaseFilter): """Masks a circular area around the center of the image, by setting it to a specified value. Intended to be used on: Reconstructed slices When: To remove reconstruction artifacts on the outer edge of the image. Caution: Ensure that the radius does not mask data from the sample. """ filter_name = "Circular Mask" link_histograms = True
[docs] @staticmethod def filter_func(data: ImageStack, circular_mask_ratio=0.95, circular_mask_value=0., progress=None) -> ImageStack: """ :param data: Input data as a 3D numpy.ndarray :param circular_mask_ratio: The ratio to the full image. The ratio must be 0 < ratio < 1 :param circular_mask_value: The value that all pixels in the mask will be set to. :return: The processed 3D numpy.ndarray """ if not circular_mask_ratio or not circular_mask_ratio < 1: raise ValueError(f'circular_mask_ratio must be > 0 and < 1. Value provided was {circular_mask_ratio}') params = {'circular_mask_ratio': circular_mask_ratio, 'circular_mask_value': circular_mask_value} ps.run_compute_func(CircularMaskFilter.compute_function, len(, [data.shared_array], params, progress) return data
[docs] @staticmethod def compute_function(i: int, arrays: list[np.ndarray], params: dict[str, Any]): array = arrays[i] if array.ndim == 2: array = np.expand_dims(array, axis=0) tomopy.circ_mask(array, axis=0, ratio=params['circular_mask_ratio'], val=params['circular_mask_value'])
[docs] @staticmethod def register_gui(form, on_change, view): from mantidimaging.gui.utility import add_property_to_form _, radius_field = add_property_to_form('Radius', Type.FLOAT, 0.95, (0.01, 0.99), form=form, on_change=on_change, tooltip="Radius [0, 1] of image that should be left untouched.") _, value_field = add_property_to_form('Set to value', Type.FLOAT, 0, (-10000, 10000), form=form, on_change=on_change, tooltip="The value of the mask.") return {'radius_field': radius_field, 'value_field': value_field}
[docs] @staticmethod def execute_wrapper(radius_field=None, value_field=None): return partial(CircularMaskFilter.filter_func, circular_mask_ratio=radius_field.value(), circular_mask_value=value_field.value())