pytme 0.2.1__cp311-cp311-macosx_14_0_arm64.whl → 0.2.2__cp311-cp311-macosx_14_0_arm64.whl

tme/matching_exhaustive.py

@@ -4,1300 +4,57 @@
 
     Author: Valentin Maurer <valentin.maurer@embl-hamburg.de>
 """
-import os
 import sys
 import warnings
-from itertools import product
-from typing import Callable, Tuple, Dict
+import traceback
 from functools import wraps
+from itertools import product
+from typing import Callable, Tuple, Dict, Optional
+
 from joblib import Parallel, delayed
 from multiprocessing.managers import SharedMemoryManager
 
-import numpy as np
-from scipy.ndimage import laplace
-
-from .analyzer import MaxScoreOverRotations
-from .matching_utils import (
-    handle_traceback,
-    split_numpy_array_slices,
-    conditional_execute,
-)
-from .matching_memory import MatchingMemoryUsage, MATCHING_MEMORY_REGISTRY
+from .backends import backend as be
 from .preprocessing import Compose
-from .matching_data import MatchingData
-from .backends import backend
-from .types import NDArray, CallbackClass
-
-os.environ["MKL_NUM_THREADS"] = "1"
-os.environ["OMP_NUM_THREADS"] = "1"
-os.environ["PYFFTW_NUM_THREADS"] = "1"
-os.environ["OPENBLAS_NUM_THREADS"] = "1"
-
-
-def _run_inner(backend_name, backend_args, **kwargs):
-    from tme.backends import backend
+from .matching_utils import split_shape
+from .matching_scores import MATCHING_EXHAUSTIVE_REGISTER
+from .types import CallbackClass, MatchingData
+from .memory import MatchingMemoryUsage, MATCHING_MEMORY_REGISTRY
 
-    backend.change_backend(backend_name, **backend_args)
-    return scan(**kwargs)
 
-
-def normalize_under_mask(template: NDArray, mask: NDArray, mask_intensity) -> None:
+def _handle_traceback(last_type, last_value, last_traceback):
     """
-    Standardizes the values in in template by subtracting the mean and dividing by the
-    standard deviation based on the elements in mask. Subsequently, the template is
-    multiplied by the mask.
+    Handle sys.exc_info().
 
     Parameters
     ----------
-    template : NDArray
-        The data array to be normalized. This array is modified in-place.
-    mask : NDArray
-        A boolean array of the same shape as `template`. True values indicate the positions in `template`
-        to consider for normalization.
-    mask_intensity : float
-        Mask intensity used to compute expectations.
-
-    References
-    ----------
-    .. [1]  T. Hrabe, Y. Chen, S. Pfeffer, L. Kuhn Cuellar, A.-V. Mangold,
-            and F. Förster, J. Struct. Biol. 178, 177 (2012).
-
-    Returns
-    -------
-    None
-        This function modifies `template` in-place and does not return any value.
-    """
-    masked_mean = backend.sum(backend.multiply(template, mask))
-    masked_mean = backend.divide(masked_mean, mask_intensity)
-    masked_std = backend.sum(backend.multiply(backend.square(template), mask))
-    masked_std = backend.subtract(
-        masked_std / mask_intensity, backend.square(masked_mean)
-    )
-    masked_std = backend.sqrt(backend.maximum(masked_std, 0))
-
-    backend.subtract(template, masked_mean, out=template)
-    backend.divide(template, masked_std, out=template)
-    backend.multiply(template, mask, out=template)
-    return None
-
-
-def _normalize_under_mask_overflow_safe(
-    template: NDArray, mask: NDArray, mask_intensity
-) -> None:
-    _template = backend.astype(template, backend._overflow_safe_dtype)
-    _mask = backend.astype(mask, backend._overflow_safe_dtype)
-    normalize_under_mask(template=_template, mask=_mask, mask_intensity=mask_intensity)
-    template[:] = backend.astype(_template, template.dtype)
-    return None
-
-
-def apply_filter(ft_template, template_filter):
-    # This is an approximation to applying the mask, irfftn, normalize, rfftn
-    std_before = backend.std(ft_template)
-    backend.multiply(ft_template, template_filter, out=ft_template)
-    backend.multiply(
-        ft_template, std_before / backend.std(ft_template), out=ft_template
-    )
-
-
-def cc_setup(
-    rfftn: Callable,
-    irfftn: Callable,
-    template: NDArray,
-    target: NDArray,
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    shared_memory_handler: Callable,
-    callback_class: Callable,
-    callback_class_args: Dict,
-    **kwargs,
-) -> Dict:
-    """
-    Setup to compute the cross-correlation between a target f and template g
-    defined as:
-
-    .. math::
-
-        \\mathcal{F}^{-1}(\\mathcal{F}(f) \\cdot \\mathcal{F}(g)^*)
-
-
-    See Also
-    --------
-    :py:meth:`corr_scoring`
-    :py:class:`tme.matching_optimization.CrossCorrelation`
-    """
-    real_dtype, complex_dtype = backend._float_dtype, backend._complex_dtype
-    target_pad = backend.topleft_pad(target, fast_shape)
-    target_pad_ft = backend.preallocate_array(fast_ft_shape, complex_dtype)
-
-    rfftn(target_pad, target_pad_ft)
-
-    target_ft_out = backend.arr_to_sharedarr(
-        arr=target_pad_ft, shared_memory_handler=shared_memory_handler
-    )
-
-    template_out = backend.arr_to_sharedarr(
-        arr=template, shared_memory_handler=shared_memory_handler
-    )
-    inv_denominator_buffer = backend.arr_to_sharedarr(
-        arr=backend.preallocate_array(1, real_dtype) + 1,
-        shared_memory_handler=shared_memory_handler,
-    )
-    numerator_buffer = backend.arr_to_sharedarr(
-        arr=backend.preallocate_array(1, real_dtype),
-        shared_memory_handler=shared_memory_handler,
-    )
-
-    target_ft_tuple = (target_ft_out, fast_ft_shape, complex_dtype)
-    template_tuple = (template_out, template.shape, template.dtype)
-
-    inv_denominator_tuple = (inv_denominator_buffer, (1,), real_dtype)
-    numerator_tuple = (numerator_buffer, (1,), real_dtype)
-
-    ret = {
-        "template": template_tuple,
-        "ft_target": target_ft_tuple,
-        "inv_denominator": inv_denominator_tuple,
-        "numerator": numerator_tuple,
-        "targetshape": target.shape,
-        "templateshape": template.shape,
-        "fast_shape": fast_shape,
-        "fast_ft_shape": fast_ft_shape,
-        "callback_class": callback_class,
-        "callback_class_args": callback_class_args,
-        "use_memmap": kwargs.get("use_memmap", False),
-        "temp_dir": kwargs.get("temp_dir", None),
-    }
-
-    return ret
-
-
-def lcc_setup(**kwargs) -> Dict:
-    """
-    Setup to compute the cross-correlation between a laplace transformed target f
-    and laplace transformed template g defined as:
-
-    .. math::
-
-        \\mathcal{F}^{-1}(\\mathcal{F}(\\nabla^{2}f) \\cdot \\mathcal{F}(\\nabla^{2} g)^*)
-
-
-    See Also
-    --------
-    :py:meth:`corr_scoring`
-    :py:class:`tme.matching_optimization.LaplaceCrossCorrelation`
-    """
-    kwargs["target"] = laplace(kwargs["target"], mode="wrap")
-    kwargs["template"] = laplace(kwargs["template"], mode="wrap")
-    return cc_setup(**kwargs)
-
-
-def corr_setup(
-    rfftn: Callable,
-    irfftn: Callable,
-    template: NDArray,
-    template_mask: NDArray,
-    target: NDArray,
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    shared_memory_handler: Callable,
-    callback_class: Callable,
-    callback_class_args: Dict,
-    **kwargs,
-) -> Dict:
-    """
-    Setup to compute a normalized cross-correlation between a target f and a template g.
-
-    .. math::
-
-        \\frac{CC(f,g) - \\overline{g} \\cdot CC(f, m)}
-        {(CC(f^2, m) - \\frac{CC(f, m)^2}{N_g}) \\cdot \\sigma_{g}}
-
-    Where:
-
-    .. math::
-
-        CC(f,g) = \\mathcal{F}^{-1}(\\mathcal{F}(f) \\cdot \\mathcal{F}(g)^*)
-
-    and m is a mask with the same dimension as the template filled with ones.
-
-    References
-    ----------
-    .. [1]  J. P. Lewis, "Fast Normalized Cross-Correlation", Industrial Light
-            and Magic.
-
-    See Also
-    --------
-    :py:meth:`corr_scoring`
-    :py:class:`tme.matching_optimization.NormalizedCrossCorrelation`.
-    """
-    real_dtype, complex_dtype = backend._float_dtype, backend._complex_dtype
-    target_pad = backend.topleft_pad(target, fast_shape)
-
-    # The exact composition of the denominator is debatable
-    # scikit-image match_template multiplies the running sum of the target
-    # with a scaling factor derived from the template. This is probably appropriate
-    # in pattern matching situations where the template exists in the target
-    window_template = backend.topleft_pad(template_mask, fast_shape)
-    ft_window_template = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    rfftn(window_template, ft_window_template)
-    window_template = None
-
-    # Target and squared target window sums
-    ft_target = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    ft_target2 = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    denominator = backend.preallocate_array(fast_shape, real_dtype)
-    target_window_sum = backend.preallocate_array(fast_shape, real_dtype)
-    rfftn(target_pad, ft_target)
-
-    rfftn(backend.square(target_pad), ft_target2)
-    backend.multiply(ft_target2, ft_window_template, out=ft_target2)
-    irfftn(ft_target2, denominator)
-
-    backend.multiply(ft_target, ft_window_template, out=ft_window_template)
-    irfftn(ft_window_template, target_window_sum)
-
-    target_pad, ft_target2, ft_window_template = None, None, None
-
-    # Normalizing constants
-    n_observations = backend.sum(template_mask)
-    template_mean = backend.sum(backend.multiply(template, template_mask))
-    template_mean = backend.divide(template_mean, n_observations)
-    template_ssd = backend.sum(
-        backend.square(
-            backend.multiply(backend.subtract(template, template_mean), template_mask)
-        )
-    )
-    template_volume = np.prod(tuple(int(x) for x in template.shape))
-    backend.multiply(template, template_mask, out=template)
-
-    # Final numerator is score - numerator
-    numerator = backend.multiply(target_window_sum, template_mean)
-
-    # Compute denominator
-    backend.multiply(target_window_sum, target_window_sum, out=target_window_sum)
-    backend.divide(target_window_sum, template_volume, out=target_window_sum)
-
-    backend.subtract(denominator, target_window_sum, out=denominator)
-    backend.multiply(denominator, template_ssd, out=denominator)
-    backend.maximum(denominator, 0, out=denominator)
-    backend.sqrt(denominator, out=denominator)
-    target_window_sum = None
-
-    # Invert denominator to compute final score as product
-    denominator_mask = denominator > backend.eps(denominator.dtype)
-    inv_denominator = backend.preallocate_array(fast_shape, real_dtype)
-    inv_denominator[denominator_mask] = 1 / denominator[denominator_mask]
-
-    # Convert arrays used in subsequent fitting to SharedMemory objects
-    template_buffer = backend.arr_to_sharedarr(
-        arr=template, shared_memory_handler=shared_memory_handler
-    )
-    target_ft_buffer = backend.arr_to_sharedarr(
-        arr=ft_target, shared_memory_handler=shared_memory_handler
-    )
-    inv_denominator_buffer = backend.arr_to_sharedarr(
-        arr=inv_denominator, shared_memory_handler=shared_memory_handler
-    )
-    numerator_buffer = backend.arr_to_sharedarr(
-        arr=numerator, shared_memory_handler=shared_memory_handler
-    )
-
-    template_tuple = (template_buffer, template.shape, template.dtype)
-    target_ft_tuple = (target_ft_buffer, fast_ft_shape, complex_dtype)
-
-    inv_denominator_tuple = (inv_denominator_buffer, fast_shape, real_dtype)
-    numerator_tuple = (numerator_buffer, fast_shape, real_dtype)
-
-    ft_target, inv_denominator, numerator = None, None, None
-
-    ret = {
-        "template": template_tuple,
-        "ft_target": target_ft_tuple,
-        "inv_denominator": inv_denominator_tuple,
-        "numerator": numerator_tuple,
-        "targetshape": target.shape,
-        "templateshape": template.shape,
-        "fast_shape": fast_shape,
-        "fast_ft_shape": fast_ft_shape,
-        "callback_class": callback_class,
-        "callback_class_args": callback_class_args,
-        "template_mean": kwargs.get("template_mean", template_mean),
-    }
-
-    return ret
-
-
-def cam_setup(**kwargs):
-    """
-    Setup to compute a normalized cross-correlation between a target f and a template g
-    over their means. In practice this can be expressed like the cross-correlation
-    CORR defined in :py:meth:`corr_scoring`, so that:
-
-    Notes
-    -----
-
-    .. math::
-
-        \\text{CORR}(f-\\overline{f}, g - \\overline{g})
-
-    Where
-
-    .. math::
-
-        \\overline{f}, \\overline{g}
-
-    are the mean of f and g respectively.
-
-    References
-    ----------
-    .. [1]  J. P. Lewis, "Fast Normalized Cross-Correlation", Industrial Light
-            and Magic.
-
-    See Also
-    --------
-    :py:meth:`corr_scoring`.
-    :py:class:`tme.matching_optimization.NormalizedCrossCorrelationMean`.
-    """
-    kwargs["template"] = kwargs["template"] - kwargs["template"].mean()
-    return corr_setup(**kwargs)
-
-
-def flc_setup(
-    rfftn: Callable,
-    irfftn: Callable,
-    template: NDArray,
-    template_mask: NDArray,
-    target: NDArray,
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    shared_memory_handler: Callable,
-    callback_class: Callable,
-    callback_class_args: Dict,
-    **kwargs,
-) -> Dict:
-    """
-    Setup to compute a normalized cross-correlation score of a target f a template g
-    and a mask m:
-
-    .. math::
-
-        \\frac{CC(f, \\frac{g*m - \\overline{g*m}}{\\sigma_{g*m}})}
-        {N_m * \\sqrt{
-            \\frac{CC(f^2, m)}{N_m} - (\\frac{CC(f, m)}{N_m})^2}
-        }
-
-    Where:
-
-    .. math::
-
-        CC(f,g) = \\mathcal{F}^{-1}(\\mathcal{F}(f) \\cdot \\mathcal{F}(g)^*)
+    last_type : type
+        The type of the last exception.
+    last_value :
+        The value of the last exception.
+    last_traceback : traceback
+        Traceback call stack at the point where the Exception occured.
 
-    and Nm is the number of voxels within the template mask m.
-
-    References
-    ----------
-    .. [1]  W. Wan, S. Khavnekar, J. Wagner, P. Erdmann, and W. Baumeister
-            Microsc. Microanal. 26, 2516 (2020)
-    .. [2]  T. Hrabe, Y. Chen, S. Pfeffer, L. Kuhn Cuellar, A.-V. Mangold,
-            and F. Förster, J. Struct. Biol. 178, 177 (2012).
-
-    See Also
-    --------
-    :py:meth:`flc_scoring`
-    """
-    target_pad = backend.topleft_pad(target, fast_shape)
-
-    # Target and squared target window sums
-    ft_target = backend.preallocate_array(fast_ft_shape, backend._complex_dtype)
-    ft_target2 = backend.preallocate_array(fast_ft_shape, backend._complex_dtype)
-    rfftn(target_pad, ft_target)
-    backend.square(target_pad, out=target_pad)
-    rfftn(target_pad, ft_target2)
-
-    # Convert arrays used in subsequent fitting to SharedMemory objects
-    ft_target = backend.arr_to_sharedarr(
-        arr=ft_target, shared_memory_handler=shared_memory_handler
-    )
-    ft_target2 = backend.arr_to_sharedarr(
-        arr=ft_target2, shared_memory_handler=shared_memory_handler
-    )
-
-    normalize_under_mask(
-        template=template, mask=template_mask, mask_intensity=backend.sum(template_mask)
-    )
-
-    template_buffer = backend.arr_to_sharedarr(
-        arr=template, shared_memory_handler=shared_memory_handler
-    )
-    template_mask_buffer = backend.arr_to_sharedarr(
-        arr=template_mask, shared_memory_handler=shared_memory_handler
-    )
-
-    template_tuple = (template_buffer, template.shape, template.dtype)
-    template_mask_tuple = (
-        template_mask_buffer,
-        template_mask.shape,
-        template_mask.dtype,
-    )
-
-    target_ft_tuple = (ft_target, fast_ft_shape, backend._complex_dtype)
-    target_ft2_tuple = (ft_target2, fast_ft_shape, backend._complex_dtype)
-
-    ret = {
-        "template": template_tuple,
-        "template_mask": template_mask_tuple,
-        "ft_target": target_ft_tuple,
-        "ft_target2": target_ft2_tuple,
-        "targetshape": target.shape,
-        "templateshape": template.shape,
-        "fast_shape": fast_shape,
-        "fast_ft_shape": fast_ft_shape,
-        "callback_class": callback_class,
-        "callback_class_args": callback_class_args,
-    }
-
-    return ret
-
-
-def flcSphericalMask_setup(
-    rfftn: Callable,
-    irfftn: Callable,
-    template: NDArray,
-    template_mask: NDArray,
-    target: NDArray,
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    shared_memory_handler: Callable,
-    callback_class: Callable,
-    callback_class_args: Dict,
-    **kwargs,
-) -> Dict:
-    """
-    Like :py:meth:`flc_setup` but for rotation invariant masks. In such cases
-    the score can be computed quicker by not computing the fourier transforms
-    of the mask for each rotation.
-
-    .. math::
-
-        \\frac{CC(f, \\frac{g*m - \\overline{g*m}}{\\sigma_{g*m}})}
-        {N_m * \\sqrt{
-            \\frac{CC(f^2, m)}{N_m} - (\\frac{CC(f, m)}{N_m})^2}
-        }
-
-    Where:
-
-    .. math::
-
-        CC(f,g) = \\mathcal{F}^{-1}(\\mathcal{F}(f) \\cdot \\mathcal{F}(g)^*)
-
-    and Nm is the number of voxels within the template mask m.
-
-    References
-    ----------
-    .. [1]  W. Wan, S. Khavnekar, J. Wagner, P. Erdmann, and W. Baumeister
-            Microsc. Microanal. 26, 2516 (2020)
-    .. [2]  T. Hrabe, Y. Chen, S. Pfeffer, L. Kuhn Cuellar, A.-V. Mangold,
-            and F. Förster, J. Struct. Biol. 178, 177 (2012).
-
-    See Also
-    --------
-    :py:meth:`corr_scoring`
-    """
-    real_dtype, complex_dtype = backend._float_dtype, backend._complex_dtype
-    target_pad = backend.topleft_pad(target, fast_shape)
-
-    # Target and squared target window sums
-    ft_target = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    ft_template_mask = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    ft_temp = backend.preallocate_array(fast_ft_shape, complex_dtype)
-
-    temp = backend.preallocate_array(fast_shape, real_dtype)
-    temp2 = backend.preallocate_array(fast_shape, real_dtype)
-    numerator = backend.preallocate_array(1, real_dtype)
-
-    n_observations, norm_func = backend.sum(template_mask), normalize_under_mask
-    if backend.datatype_bytes(template_mask.dtype) == 2:
-        norm_func = _normalize_under_mask_overflow_safe
-        n_observations = backend.sum(
-            backend.astype(template_mask, backend._overflow_safe_dtype)
-        )
-
-    template_mask_pad = backend.topleft_pad(template_mask, fast_shape)
-    rfftn(template_mask_pad, ft_template_mask)
-
-    # Denominator E(X^2) - E(X)^2
-    rfftn(backend.square(target_pad), ft_target)
-    backend.multiply(ft_target, ft_template_mask, out=ft_temp)
-    irfftn(ft_temp, temp2)
-    backend.divide(temp2, n_observations, out=temp2)
-
-    rfftn(target_pad, ft_target)
-    backend.multiply(ft_target, ft_template_mask, out=ft_temp)
-    irfftn(ft_temp, temp)
-    backend.divide(temp, n_observations, out=temp)
-    backend.square(temp, out=temp)
-
-    backend.subtract(temp2, temp, out=temp)
-
-    backend.maximum(temp, 0.0, out=temp)
-    backend.sqrt(temp, out=temp)
-    backend.multiply(temp, n_observations, out=temp)
-
-    backend.fill(temp2, 0)
-    nonzero_indices = temp > backend.eps(real_dtype)
-    temp2[nonzero_indices] = 1 / temp[nonzero_indices]
-
-    norm_func(template=template, mask=template_mask, mask_intensity=n_observations)
-
-    template_buffer = backend.arr_to_sharedarr(
-        arr=template, shared_memory_handler=shared_memory_handler
-    )
-    template_mask_buffer = backend.arr_to_sharedarr(
-        arr=template_mask, shared_memory_handler=shared_memory_handler
-    )
-    target_ft_buffer = backend.arr_to_sharedarr(
-        arr=ft_target, shared_memory_handler=shared_memory_handler
-    )
-    inv_denominator_buffer = backend.arr_to_sharedarr(
-        arr=temp2, shared_memory_handler=shared_memory_handler
-    )
-    numerator_buffer = backend.arr_to_sharedarr(
-        arr=numerator, shared_memory_handler=shared_memory_handler
-    )
-
-    template_tuple = (template_buffer, template.shape, template.dtype)
-    template_mask_tuple = (template_mask_buffer, template.shape, template_mask.dtype)
-    target_ft_tuple = (target_ft_buffer, fast_ft_shape, complex_dtype)
-
-    inv_denominator_tuple = (inv_denominator_buffer, fast_shape, real_dtype)
-    numerator_tuple = (numerator_buffer, (1,), real_dtype)
-
-    ret = {
-        "template": template_tuple,
-        "template_mask": template_mask_tuple,
-        "ft_target": target_ft_tuple,
-        "inv_denominator": inv_denominator_tuple,
-        "numerator": numerator_tuple,
-        "targetshape": target.shape,
-        "templateshape": template.shape,
-        "fast_shape": fast_shape,
-        "fast_ft_shape": fast_ft_shape,
-        "callback_class": callback_class,
-        "callback_class_args": callback_class_args,
-    }
-
-    return ret
-
-
-def mcc_setup(
-    rfftn: Callable,
-    irfftn: Callable,
-    template: NDArray,
-    template_mask: NDArray,
-    target: NDArray,
-    target_mask: NDArray,
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    shared_memory_handler: Callable,
-    callback_class: Callable,
-    callback_class_args: Dict,
-    **kwargs,
-) -> Dict:
-    """
-    Setup to compute a normalized cross-correlation score with masks
-    for both template and target.
-
-    .. math::
-
-        \\frac{
-               CC(f, g) - \\frac{CC(f, m) \\cdot CC(t, g)}{CC(t, m)}
-            }{
-            \\sqrt{
-                (CC(f ^ 2, m) - \\frac{CC(f, m) ^ 2}{CC(t, m)}) \\cdot
-                (CC(t, g^2) - \\frac{CC(t, g) ^ 2}{CC(t, m)})
-                }
-            }
-
-    Where:
-
-    .. math::
-
-        CC(f,g) = \\mathcal{F}^{-1}(\\mathcal{F}(f) \\cdot \\mathcal{F}(g)^*)
-
-
-    References
-    ----------
-    .. [1]  Masked FFT registration, Dirk Padfield, CVPR 2010 conference
-
-    See Also
-    --------
-    :py:meth:`mcc_scoring`
-    :py:class:`tme.matching_optimization.MaskedCrossCorrelation`
-    """
-    real_dtype, complex_dtype = backend._float_dtype, backend._complex_dtype
-    target = backend.multiply(target, target_mask > 0, out=target)
-
-    target_pad = backend.topleft_pad(target, fast_shape)
-    target_mask_pad = backend.topleft_pad(target_mask, fast_shape)
-
-    target_ft = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    rfftn(target_pad, target_ft)
-    target_ft_buffer = backend.arr_to_sharedarr(
-        arr=target_ft, shared_memory_handler=shared_memory_handler
-    )
-
-    target_ft2 = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    rfftn(backend.square(target_pad), target_ft2)
-    target_ft2_buffer = backend.arr_to_sharedarr(
-        arr=target_ft2, shared_memory_handler=shared_memory_handler
-    )
-
-    target_mask_ft = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    rfftn(target_mask_pad, target_mask_ft)
-    target_mask_ft_buffer = backend.arr_to_sharedarr(
-        arr=target_mask_ft, shared_memory_handler=shared_memory_handler
-    )
-
-    template_buffer = backend.arr_to_sharedarr(
-        arr=template, shared_memory_handler=shared_memory_handler
-    )
-    template_mask_buffer = backend.arr_to_sharedarr(
-        arr=template_mask, shared_memory_handler=shared_memory_handler
-    )
-
-    template_tuple = (template_buffer, template.shape, template.dtype)
-    template_mask_tuple = (template_mask_buffer, template.shape, template_mask.dtype)
-
-    target_ft_tuple = (target_ft_buffer, fast_ft_shape, complex_dtype)
-    target_ft2_tuple = (target_ft2_buffer, fast_ft_shape, complex_dtype)
-    target_mask_ft_tuple = (target_mask_ft_buffer, fast_ft_shape, complex_dtype)
-
-    ret = {
-        "template": template_tuple,
-        "template_mask": template_mask_tuple,
-        "ft_target": target_ft_tuple,
-        "ft_target2": target_ft2_tuple,
-        "ft_target_mask": target_mask_ft_tuple,
-        "targetshape": target.shape,
-        "templateshape": template.shape,
-        "fast_shape": fast_shape,
-        "fast_ft_shape": fast_ft_shape,
-        "callback_class": callback_class,
-        "callback_class_args": callback_class_args,
-    }
-
-    return ret
-
-
-def corr_scoring(
-    template: Tuple[type, Tuple[int], type],
-    ft_target: Tuple[type, Tuple[int], type],
-    inv_denominator: Tuple[type, Tuple[int], type],
-    numerator: Tuple[type, Tuple[int], type],
-    template_filter: Tuple[type, Tuple[int], type],
-    targetshape: Tuple[int],
-    templateshape: Tuple[int],
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    rotations: NDArray,
-    callback_class: CallbackClass,
-    callback_class_args: Dict,
-    interpolation_order: int,
-    convolution_mode: str = "full",
-    **kwargs,
-) -> CallbackClass:
-    """
-    Calculates a normalized cross-correlation between a target f and a template g.
-
-    .. math::
-
-        (CC(f,g) - numerator) \\cdot inv\\_denominator
-
-    Parameters
-    ----------
-    template : Tuple[type, Tuple[int], type]
-        Tuple containing a pointer to the template data, its shape, and its datatype.
-    ft_target : Tuple[type, Tuple[int], type]
-        Tuple containing a pointer to the fourier tranform of the target,
-        its shape, and its datatype.
-    inv_denominator : Tuple[type, Tuple[int], type]
-        Tuple containing a pointer to the inverse denominator data, its shape, and its
-        datatype.
-    numerator : Tuple[type, Tuple[int], type]
-        Tuple containing a pointer to the numerator data, its shape, and its datatype.
-    fast_shape : Tuple[int]
-        The shape for fast Fourier transform.
-    fast_ft_shape : Tuple[int]
-        The shape for fast Fourier transform of the target.
-    rotations : NDArray
-        Array containing the rotation matrices to be applied on the template.
-    real_dtype : type
-        Data type for the real part of the array.
-    complex_dtype : type
-        Data type for the complex part of the array.
-    callback_class : CallbackClass
-        A callable class or function for processing the results after each
-        rotation.
-    callback_class_args : Dict
-        Dictionary of arguments to be passed to the callback class if it's
-        instantiable.
-    interpolation_order : int
-        The order of interpolation to be used while rotating the template.
-    **kwargs :
-        Additional arguments to be passed to the function.
-
-    Returns
-    -------
-    CallbackClass
-        If callback_class was provided an instance of callback_class otherwise None.
-
-    See Also
-    --------
-    :py:meth:`cc_setup`
-    :py:meth:`corr_setup`
-    :py:meth:`cam_setup`
-    :py:meth:`flcSphericalMask_setup`
-    """
-    real_dtype, complex_dtype = backend._float_dtype, backend._complex_dtype
-
-    callback = callback_class
-    if callback_class is not None and isinstance(callback_class, type):
-        callback = callback_class(**callback_class_args)
-
-    template_buffer, template_shape, template_dtype = template
-    template = backend.sharedarr_to_arr(template_buffer, template_shape, template_dtype)
-    ft_target = backend.sharedarr_to_arr(*ft_target)
-    inv_denominator = backend.sharedarr_to_arr(*inv_denominator)
-    numerator = backend.sharedarr_to_arr(*numerator)
-    template_filter = backend.sharedarr_to_arr(*template_filter)
-
-    norm_func = normalize_under_mask
-    norm_template, template_mask, mask_sum = False, 1, 1
-    if "template_mask" in kwargs:
-        norm_template = True
-        template_mask = backend.sharedarr_to_arr(*kwargs["template_mask"])
-        mask_sum = backend.sum(template_mask)
-        if backend.datatype_bytes(template_mask.dtype) == 2:
-            norm_func = _normalize_under_mask_overflow_safe
-            mask_sum = backend.sum(
-                backend.astype(template_mask, backend._overflow_safe_dtype)
-            )
-
-    norm_template = conditional_execute(norm_func, norm_template)
-    norm_numerator = (backend.sum(numerator) != 0) & (backend.size(numerator) != 1)
-    norm_func_numerator = conditional_execute(backend.subtract, norm_numerator)
-
-    norm_denominator = (backend.sum(inv_denominator) != 1) & (
-        backend.size(inv_denominator) != 1
-    )
-    norm_func_denominator = conditional_execute(backend.multiply, norm_denominator)
-    callback_func = conditional_execute(callback, callback_class is not None)
-    template_filter_func = conditional_execute(
-        apply_filter, backend.size(template_filter) != 1
-    )
-
-    arr = backend.preallocate_array(fast_shape, real_dtype)
-    ft_temp = backend.preallocate_array(fast_ft_shape, complex_dtype)
-
-    rfftn, irfftn = backend.build_fft(
-        fast_shape=fast_shape,
-        fast_ft_shape=fast_ft_shape,
-        real_dtype=real_dtype,
-        complex_dtype=complex_dtype,
-        fftargs=kwargs.get("fftargs", {}),
-        temp_real=arr,
-        temp_fft=ft_temp,
-    )
-
-    unpadded_slice = tuple(slice(0, stop) for stop in template.shape)
-    for index in range(rotations.shape[0]):
-        rotation = rotations[index]
-        backend.fill(arr, 0)
-        backend.rotate_array(
-            arr=template,
-            rotation_matrix=rotation,
-            out=arr,
-            use_geometric_center=True,
-            order=interpolation_order,
-        )
-        norm_template(arr[unpadded_slice], template_mask, mask_sum)
-
-        rfftn(arr, ft_temp)
-        template_filter_func(ft_template=ft_temp, template_filter=template_filter)
-        backend.multiply(ft_target, ft_temp, out=ft_temp)
-        irfftn(ft_temp, arr)
-
-        norm_func_numerator(arr, numerator, out=arr)
-        norm_func_denominator(arr, inv_denominator, out=arr)
-
-        callback_func(
-            arr,
-            rotation_matrix=rotation,
-            rotation_index=index,
-            **callback_class_args,
-        )
-
-    return callback
-
-
-def flc_scoring(
-    template: Tuple[type, Tuple[int], type],
-    template_mask: Tuple[type, Tuple[int], type],
-    ft_target: Tuple[type, Tuple[int], type],
-    ft_target2: Tuple[type, Tuple[int], type],
-    template_filter: Tuple[type, Tuple[int], type],
-    targetshape: Tuple[int],
-    templateshape: Tuple[int],
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    rotations: NDArray,
-    callback_class: CallbackClass,
-    callback_class_args: Dict,
-    interpolation_order: int,
-    **kwargs,
-) -> CallbackClass:
-    """
-    Computes a normalized cross-correlation score of a target f a template g
-    and a mask m:
-
-    .. math::
-
-        \\frac{CC(f, \\frac{g*m - \\overline{g*m}}{\\sigma_{g*m}})}
-        {N_m * \\sqrt{
-            \\frac{CC(f^2, m)}{N_m} - (\\frac{CC(f, m)}{N_m})^2}
-        }
-
-    Where:
-
-    .. math::
-
-        CC(f,g) = \\mathcal{F}^{-1}(\\mathcal{F}(f) \\cdot \\mathcal{F}(g)^*)
-
-    and Nm is the number of voxels within the template mask m.
-
-    References
-    ----------
-    .. [1]  W. Wan, S. Khavnekar, J. Wagner, P. Erdmann, and W. Baumeister
-            Microsc. Microanal. 26, 2516 (2020)
-    .. [2]  T. Hrabe, Y. Chen, S. Pfeffer, L. Kuhn Cuellar, A.-V. Mangold,
-            and F. Förster, J. Struct. Biol. 178, 177 (2012).
-    """
-    real_dtype, complex_dtype = backend._float_dtype, backend._complex_dtype
-
-    callback = callback_class
-    if callback_class is not None and isinstance(callback_class, type):
-        callback = callback_class(**callback_class_args)
-
-    template = backend.sharedarr_to_arr(*template)
-    template_mask = backend.sharedarr_to_arr(*template_mask)
-    ft_target = backend.sharedarr_to_arr(*ft_target)
-    ft_target2 = backend.sharedarr_to_arr(*ft_target2)
-    template_filter = backend.sharedarr_to_arr(*template_filter)
-
-    arr = backend.preallocate_array(fast_shape, real_dtype)
-    temp = backend.preallocate_array(fast_shape, real_dtype)
-    temp2 = backend.preallocate_array(fast_shape, real_dtype)
-
-    ft_temp = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    ft_denom = backend.preallocate_array(fast_ft_shape, complex_dtype)
-
-    rfftn, irfftn = backend.build_fft(
-        fast_shape=fast_shape,
-        fast_ft_shape=fast_ft_shape,
-        real_dtype=real_dtype,
-        complex_dtype=complex_dtype,
-        fftargs=kwargs.get("fftargs", {}),
-        temp_real=arr,
-        temp_fft=ft_temp,
-    )
-    eps = backend.eps(real_dtype)
-    template_filter_func = conditional_execute(
-        apply_filter, backend.size(template_filter) != 1
-    )
-    callback_func = conditional_execute(callback, callback_class is not None)
-
-    unpadded_slice = tuple(slice(0, stop) for stop in template.shape)
-    for index in range(rotations.shape[0]):
-        rotation = rotations[index]
-        backend.fill(arr, 0)
-        backend.fill(temp, 0)
-        backend.rotate_array(
-            arr=template,
-            arr_mask=template_mask,
-            rotation_matrix=rotation,
-            out=arr,
-            out_mask=temp,
-            use_geometric_center=True,
-            order=interpolation_order,
-        )
-        # Given the amount of FFTs, might aswell normalize properly
-        n_observations = backend.sum(temp)
-
-        normalize_under_mask(
-            template=arr[unpadded_slice],
-            mask=temp[unpadded_slice],
-            mask_intensity=n_observations,
-        )
-
-        rfftn(temp, ft_temp)
-
-        backend.multiply(ft_target, ft_temp, out=ft_denom)
-        irfftn(ft_denom, temp)
-        backend.divide(temp, n_observations, out=temp)
-        backend.square(temp, out=temp)
-
-        backend.multiply(ft_target2, ft_temp, out=ft_denom)
-        irfftn(ft_denom, temp2)
-        backend.divide(temp2, n_observations, out=temp2)
-
-        backend.subtract(temp2, temp, out=temp)
-        backend.maximum(temp, 0.0, out=temp)
-        backend.sqrt(temp, out=temp)
-        backend.multiply(temp, n_observations, out=temp)
-
-        rfftn(arr, ft_temp)
-        template_filter_func(ft_template=ft_temp, template_filter=template_filter)
-        backend.multiply(ft_target, ft_temp, out=ft_temp)
-        irfftn(ft_temp, arr)
-
-        tol = eps
-        # tol = 1e3 * eps * backend.max(backend.abs(temp))
-        nonzero_indices = temp > tol
-        backend.fill(temp2, 0)
-        temp2[nonzero_indices] = arr[nonzero_indices] / temp[nonzero_indices]
-
-        callback_func(
-            temp2,
-            rotation_matrix=rotation,
-            rotation_index=index,
-            **callback_class_args,
-        )
-
-    return callback
-
-
-def flc_scoring2(
-    template: Tuple[type, Tuple[int], type],
-    template_mask: Tuple[type, Tuple[int], type],
-    ft_target: Tuple[type, Tuple[int], type],
-    ft_target2: Tuple[type, Tuple[int], type],
-    template_filter: Tuple[type, Tuple[int], type],
-    targetshape: Tuple[int],
-    templateshape: Tuple[int],
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    rotations: NDArray,
-    callback_class: CallbackClass,
-    callback_class_args: Dict,
-    interpolation_order: int,
-    **kwargs,
-) -> CallbackClass:
-    """
-    Computes a normalized cross-correlation score of a target f a template g
-    and a mask m:
-
-    .. math::
-
-        \\frac{CC(f, \\frac{g*m - \\overline{g*m}}{\\sigma_{g*m}})}
-        {N_m * \\sqrt{
-            \\frac{CC(f^2, m)}{N_m} - (\\frac{CC(f, m)}{N_m})^2}
-        }
-
-    Where:
-
-    .. math::
-
-        CC(f,g) = \\mathcal{F}^{-1}(\\mathcal{F}(f) \\cdot \\mathcal{F}(g)^*)
-
-    and Nm is the number of voxels within the template mask m.
-
-    References
-    ----------
-    .. [1]  W. Wan, S. Khavnekar, J. Wagner, P. Erdmann, and W. Baumeister
-            Microsc. Microanal. 26, 2516 (2020)
-    .. [2]  T. Hrabe, Y. Chen, S. Pfeffer, L. Kuhn Cuellar, A.-V. Mangold,
-            and F. Förster, J. Struct. Biol. 178, 177 (2012).
-    """
-    real_dtype, complex_dtype = backend._float_dtype, backend._complex_dtype
-    callback = callback_class
-    if callback_class is not None and isinstance(callback_class, type):
-        callback = callback_class(**callback_class_args)
-
-    # Retrieve objects from shared memory
-    template = backend.sharedarr_to_arr(*template)
-    template_mask = backend.sharedarr_to_arr(*template_mask)
-    ft_target = backend.sharedarr_to_arr(*ft_target)
-    ft_target2 = backend.sharedarr_to_arr(*ft_target2)
-    template_filter = backend.sharedarr_to_arr(*template_filter)
-
-    arr = backend.preallocate_array(fast_shape, real_dtype)
-    temp = backend.preallocate_array(fast_shape, real_dtype)
-    temp2 = backend.preallocate_array(fast_shape, real_dtype)
-
-    ft_temp = backend.preallocate_array(fast_ft_shape, complex_dtype)
-    ft_denom = backend.preallocate_array(fast_ft_shape, complex_dtype)
-
-    eps = backend.eps(real_dtype)
-    template_filter_func = conditional_execute(
-        apply_filter, backend.size(template_filter) != 1
-    )
-    callback_func = conditional_execute(callback, callback_class is not None)
-
-    squeeze_axis = tuple(i for i, x in enumerate(template.shape) if x == 1)
-    squeeze = tuple(
-        slice(0, stop) if i not in squeeze_axis else 0
-        for i, stop in enumerate(template.shape)
-    )
-    squeeze_fast = tuple(
-        slice(0, stop) if i not in squeeze_axis else 0
-        for i, stop in enumerate(fast_shape)
-    )
-    squeeze_fast_ft = tuple(
-        slice(0, stop) if i not in squeeze_axis else 0
-        for i, stop in enumerate(fast_ft_shape)
-    )
-
-    rfftn, irfftn = backend.build_fft(
-        fast_shape=temp[squeeze_fast].shape,
-        fast_ft_shape=fast_ft_shape,
-        real_dtype=real_dtype,
-        complex_dtype=complex_dtype,
-        fftargs=kwargs.get("fftargs", {}),
-        inverse_fast_shape=fast_shape,
-        temp_real=arr[squeeze_fast],
-        temp_fft=ft_temp,
-    )
-    for index in range(rotations.shape[0]):
-        rotation = rotations[index]
-        backend.fill(arr, 0)
-        backend.fill(temp, 0)
-        backend.rotate_array(
-            arr=template[squeeze],
-            arr_mask=template_mask[squeeze],
-            rotation_matrix=rotation,
-            out=arr[squeeze],
-            out_mask=temp[squeeze],
-            use_geometric_center=True,
-            order=interpolation_order,
-        )
-        # Given the amount of FFTs, might aswell normalize properly
-        n_observations = backend.sum(temp)
-
-        normalize_under_mask(
-            template=arr[squeeze],
-            mask=temp[squeeze],
-            mask_intensity=n_observations,
-        )
-        rfftn(temp[squeeze_fast], ft_temp[squeeze_fast_ft])
-
-        backend.multiply(ft_target, ft_temp[squeeze_fast_ft], out=ft_denom)
-        irfftn(ft_denom, temp)
-        backend.divide(temp, n_observations, out=temp)
-        backend.square(temp, out=temp)
-
-        backend.multiply(ft_target2, ft_temp[squeeze_fast_ft], out=ft_denom)
-        irfftn(ft_denom, temp2)
-        backend.divide(temp2, n_observations, out=temp2)
-
-        backend.subtract(temp2, temp, out=temp)
-        backend.maximum(temp, 0.0, out=temp)
-        backend.sqrt(temp, out=temp)
-        backend.multiply(temp, n_observations, out=temp)
-
-        rfftn(arr[squeeze_fast], ft_temp[squeeze_fast_ft])
-        template_filter_func(ft_template=ft_temp, template_filter=template_filter)
-        backend.multiply(ft_target, ft_temp[squeeze_fast_ft], out=ft_denom)
-        irfftn(ft_denom, arr)
-
-        nonzero_indices = temp > eps
-        backend.fill(temp2, 0)
-        temp2[nonzero_indices] = arr[nonzero_indices] / temp[nonzero_indices]
-
-        callback_func(
-            temp2,
-            rotation_matrix=rotation,
-            rotation_index=index,
-            **callback_class_args,
-        )
-    return callback
-
-
-def mcc_scoring(
-    template: Tuple[type, Tuple[int], type],
-    template_mask: Tuple[type, Tuple[int], type],
-    ft_target: Tuple[type, Tuple[int], type],
-    ft_target2: Tuple[type, Tuple[int], type],
-    ft_target_mask: Tuple[type, Tuple[int], type],
-    template_filter: Tuple[type, Tuple[int], type],
-    targetshape: Tuple[int],
-    templateshape: Tuple[int],
-    fast_shape: Tuple[int],
-    fast_ft_shape: Tuple[int],
-    rotations: NDArray,
-    callback_class: CallbackClass,
-    callback_class_args: type,
-    interpolation_order: int,
-    overlap_ratio: float = 0.3,
-    **kwargs,
-) -> CallbackClass:
-    """
-    Computes a cross-correlation score with masks for both template and target.
-
-    .. math::
-
-        \\frac{
-               CC(f, g) - \\frac{CC(f, m) \\cdot CC(t, g)}{CC(t, m)}
-            }{
-            \\sqrt{
-                (CC(f ^ 2, m) - \\frac{CC(f, m) ^ 2}{CC(t, m)}) \\cdot
-                (CC(t, g^2) - \\frac{CC(t, g) ^ 2}{CC(t, m)})
-                }
-            }
-
-    Where:
-
-    .. math::
-
-        CC(f,g) = \\mathcal{F}^{-1}(\\mathcal{F}(f) \\cdot \\mathcal{F}(g)^*)
-
-
-    References
-    ----------
-    .. [1]  Masked FFT registration, Dirk Padfield, CVPR 2010 conference
-    .. [2]  https://scikit-image.org/docs/stable/api/skimage.registration.html
-
-    See Also
-    --------
-    :py:class:`tme.matching_optimization.MaskedCrossCorrelation`
+    Raises
+    ------
+    Exception
+        Re-raises the last exception.
     """
-    real_dtype, complex_dtype = backend._float_dtype, backend._complex_dtype
-    callback = callback_class
-    if callback_class is not None and isinstance(callback_class, type):
-        callback = callback_class(**callback_class_args)
-
-    # Retrieve objects from shared memory
-    template_buffer, template_shape, template_dtype = template
-    template = backend.sharedarr_to_arr(*template)
-    target_ft = backend.sharedarr_to_arr(*ft_target)
-    target_ft2 = backend.sharedarr_to_arr(*ft_target2)
-    template_mask = backend.sharedarr_to_arr(*template_mask)
-    target_mask_ft = backend.sharedarr_to_arr(*ft_target_mask)
-    template_filter = backend.sharedarr_to_arr(*template_filter)
-
-    axes = tuple(range(template.ndim))
-    eps = backend.eps(real_dtype)
-
-    # Allocate score and process specific arrays
-    template_rot = backend.preallocate_array(fast_shape, real_dtype)
-    mask_overlap = backend.preallocate_array(fast_shape, real_dtype)
-    numerator = backend.preallocate_array(fast_shape, real_dtype)
-
-    temp = backend.preallocate_array(fast_shape, real_dtype)
-    temp2 = backend.preallocate_array(fast_shape, real_dtype)
-    temp3 = backend.preallocate_array(fast_shape, real_dtype)
-    temp_ft = backend.preallocate_array(fast_ft_shape, complex_dtype)
-
-    rfftn, irfftn = backend.build_fft(
-        fast_shape=fast_shape,
-        fast_ft_shape=fast_ft_shape,
-        real_dtype=real_dtype,
-        complex_dtype=complex_dtype,
-        fftargs=kwargs.get("fftargs", {}),
-        temp_real=numerator,
-        temp_fft=temp_ft,
-    )
-
-    template_filter_func = conditional_execute(
-        apply_filter, backend.size(template_filter) != 1
-    )
-    callback_func = conditional_execute(callback, callback_class is not None)
-
-    # Calculate scores across all rotations
-    for index in range(rotations.shape[0]):
-        rotation = rotations[index]
-        backend.fill(template_rot, 0)
-        backend.fill(temp, 0)
-
-        backend.rotate_array(
-            arr=template,
-            arr_mask=template_mask,
-            rotation_matrix=rotation,
-            out=template_rot,
-            out_mask=temp,
-            use_geometric_center=True,
-            order=interpolation_order,
-        )
-
-        backend.multiply(template_rot, temp > 0, out=template_rot)
-
-        # template_rot_ft
-        rfftn(template_rot, temp_ft)
-        template_filter_func(ft_template=temp_ft, template_filter=template_filter)
-        irfftn(target_mask_ft * temp_ft, temp2)
-        irfftn(target_ft * temp_ft, numerator)
-
-        # temp template_mask_rot | temp_ft template_mask_rot_ft
-        # Calculate overlap of masks at every point in the convolution.
-        # Locations with high overlap should not be taken into account.
-        rfftn(temp, temp_ft)
-        irfftn(temp_ft * target_mask_ft, mask_overlap)
-        mask_overlap[:] = np.round(mask_overlap)
-        mask_overlap[:] = np.maximum(mask_overlap, eps)
-        irfftn(temp_ft * target_ft, temp)
-
-        backend.subtract(
-            numerator,
-            backend.divide(backend.multiply(temp, temp2), mask_overlap),
-            out=numerator,
-        )
-
-        # temp_3 = fixed_denom
-        backend.multiply(temp_ft, target_ft2, out=temp_ft)
-        irfftn(temp_ft, temp3)
-        backend.subtract(
-            temp3, backend.divide(backend.square(temp), mask_overlap), out=temp3
-        )
-        backend.maximum(temp3, 0.0, out=temp3)
+    if last_type is None:
+        return None
+    traceback.print_tb(last_traceback)
+    raise Exception(last_value)
 
-        # temp = moving_denom
-        rfftn(backend.square(template_rot), temp_ft)
-        backend.multiply(target_mask_ft, temp_ft, out=temp_ft)
-        irfftn(temp_ft, temp)
 
-        backend.subtract(
-            temp, backend.divide(backend.square(temp2), mask_overlap), out=temp
-        )
-        backend.maximum(temp, 0.0, out=temp)
-
-        # temp_2 = denom
-        backend.multiply(temp3, temp, out=temp)
-        backend.sqrt(temp, temp2)
-
-        # Pixels where `denom` is very small will introduce large
-        # numbers after division. To get around this problem,
-        # we zero-out problematic pixels.
-        tol = 1e3 * eps * backend.max(backend.abs(temp2), axis=axes, keepdims=True)
-        nonzero_indices = temp2 > tol
-
-        backend.fill(temp, 0)
-        temp[nonzero_indices] = numerator[nonzero_indices] / temp2[nonzero_indices]
-        backend.clip(temp, a_min=-1, a_max=1, out=temp)
-
-        # Apply overlap ratio threshold
-        number_px_threshold = overlap_ratio * backend.max(
-            mask_overlap, axis=axes, keepdims=True
-        )
-        temp[mask_overlap < number_px_threshold] = 0.0
+def _wrap_backend(func):
+    @wraps(func)
+    def wrapper(*args, backend_name: str, backend_args: Dict, **kwargs):
+        from tme.backends import backend as be
 
-        callback_func(
-            temp,
-            rotation_matrix=rotation,
-            rotation_index=index,
-            **callback_class_args,
-        )
+        be.change_backend(backend_name, **backend_args)
+        return func(*args, **kwargs)
 
-    return callback
+    return wrapper
 
 
 def _setup_template_filter_apply_target_filter(
@@ -1306,42 +63,47 @@ def _setup_template_filter_apply_target_filter(
     irfftn: Callable,
     fast_shape: Tuple[int],
     fast_ft_shape: Tuple[int],
+    pad_template_filter: bool = True,
 ):
     filter_template = isinstance(matching_data.template_filter, Compose)
     filter_target = isinstance(matching_data.target_filter, Compose)
 
-    template_filter = backend.full(shape=(1,), fill_value=1, dtype=backend._float_dtype)
+    template_filter = be.full(shape=(1,), fill_value=1, dtype=be._float_dtype)
 
     if not filter_template and not filter_target:
         return template_filter
 
-    target_temp = backend.astype(
-        backend.topleft_pad(matching_data.target, fast_shape), backend._float_dtype
-    )
-    target_temp_ft = backend.preallocate_array(fast_ft_shape, backend._complex_dtype)
+    target_temp = be.topleft_pad(matching_data.target, fast_shape)
+    target_temp_ft = be.zeros(fast_ft_shape, be._complex_dtype)
 
-    filter_shape = backend.multiply(fast_ft_shape, 1 - matching_data._batch_mask)
-    filter_shape[filter_shape == 0] = 1
-    fast_shape = backend.multiply(fast_shape, 1 - matching_data._batch_mask)
-    fast_shape = fast_shape[fast_shape != 0]
+    inv_mask = be.subtract(1, be.to_backend_array(matching_data._batch_mask))
+    filter_shape = be.multiply(be.to_backend_array(fast_ft_shape), inv_mask)
+    filter_shape = tuple(int(x) if x != 0 else 1 for x in filter_shape)
 
-    fast_shape = tuple(int(x) for x in fast_shape)
-    filter_shape = tuple(int(x) for x in filter_shape)
+    fast_shape = be.multiply(be.to_backend_array(fast_shape), inv_mask)
+    fast_shape = tuple(int(x) for x in fast_shape if x != 0)
+
+    target_temp_ft = rfftn(target_temp, target_temp_ft)
+    if filter_template:
+        # TODO: Pad to fast shapes and adapt _setup_template_filtering accordingly
+        template_fast_shape, template_filter_shape = fast_shape, filter_shape
+        if not pad_template_filter:
+            template_fast_shape = tuple(int(x) for x in matching_data._template.shape)
+            template_filter_shape = [
+                int(x) for x in matching_data._output_template_shape
+            ]
+            template_filter_shape[-1] = template_filter_shape[-1] // 2 + 1
 
-    rfftn(target_temp, target_temp_ft)
-    if isinstance(matching_data.template_filter, Compose):
         template_filter = matching_data.template_filter(
-            shape=fast_shape,
+            shape=template_fast_shape,
             return_real_fourier=True,
             shape_is_real_fourier=False,
             data_rfft=target_temp_ft,
             batch_dimension=matching_data._target_dims,
-        )
-        template_filter = template_filter["data"]
-        template_filter[tuple(0 for _ in range(template_filter.ndim))] = 0
-        template_filter = backend.reshape(template_filter, filter_shape)
+        )["data"]
+        template_filter = be.reshape(template_filter, template_filter_shape)
 
-    if isinstance(matching_data.target_filter, Compose):
+    if filter_target:
         target_filter = matching_data.target_filter(
             shape=fast_shape,
             return_real_fourier=True,
@@ -1349,15 +111,12 @@ def _setup_template_filter_apply_target_filter(
             data_rfft=target_temp_ft,
             weight_type=None,
             batch_dimension=matching_data._target_dims,
-        )
-        target_filter = target_filter["data"]
-        target_filter = backend.reshape(target_filter, filter_shape)
-        backend.multiply(target_temp_ft, target_filter, out=target_temp_ft)
+        )["data"]
+        target_filter = be.reshape(target_filter, filter_shape)
+        target_temp_ft = be.multiply(target_temp_ft, target_filter, out=target_temp_ft)
 
-        irfftn(target_temp_ft, target_temp)
-        matching_data._target = backend.topleft_pad(
-            target_temp, matching_data.target.shape
-        )
+        target_temp = irfftn(target_temp_ft, target_temp)
+        matching_data._target = be.topleft_pad(target_temp, matching_data.target.shape)
 
     return template_filter
 
@@ -1371,13 +130,14 @@ def device_memory_handler(func: Callable):
         last_type, last_value, last_traceback = sys.exc_info()
         try:
             with SharedMemoryManager() as smh:
-                with backend.set_device(kwargs.get("gpu_index", 0)):
+                gpu_index = kwargs.pop("gpu_index") if "gpu_index" in kwargs else 0
+                with be.set_device(gpu_index):
                     return_value = func(shared_memory_handler=smh, *args, **kwargs)
         except Exception as e:
             print(e)
             last_type, last_value, last_traceback = sys.exc_info()
         finally:
-            handle_traceback(last_type, last_value, last_traceback)
+            _handle_traceback(last_type, last_value, last_traceback)
         return return_value
 
     return inner_function
@@ -1391,18 +151,20 @@ def scan(
     n_jobs: int = 4,
     callback_class: CallbackClass = None,
     callback_class_args: Dict = {},
-    fftargs: Dict = {},
     pad_fourier: bool = True,
+    pad_template_filter: bool = True,
     interpolation_order: int = 3,
     jobs_per_callback_class: int = 8,
-    **kwargs,
-) -> Tuple:
+    shared_memory_handler=None,
+) -> Optional[Tuple]:
     """
-    Perform template matching.
+    Run template matching.
+
+    .. warning:: ``matching_data`` might be altered or destroyed during computation.
 
     Parameters
     ----------
-    matching_data : MatchingData
+    matching_data : :py:class:`tme.matching_data.MatchingData`
         Template matching data.
     matching_setup : Callable
         Function pointer to setup function.
@@ -1414,21 +176,21 @@ def scan(
         Analyzer class pointer to operate on computed scores.
     callback_class_args : dict, optional
         Arguments passed to the callback_class. Default is an empty dictionary.
-    fftargs : dict, optional
-        Arguments for the FFT operations. Default is an empty dictionary.
     pad_fourier: bool, optional
         Whether to pad target and template to the full convolution shape.
+    pad_template_filter: bool, optional
+        Whether to pad potential template filters to the full convolution shape.
     interpolation_order : int, optional
         Order of spline interpolation for rotations.
     jobs_per_callback_class : int, optional
         How many jobs should be processed by a single callback_class instance,
         if one is provided.
-    **kwargs : various
-        Additional keyword arguments.
+    shared_memory_handler : type, optional
+        Manager for shared memory objects, None by default.
 
     Returns
     -------
-    Tuple
+    Optional[Tuple]
         The merged results from callback_class if provided otherwise None.
 
     Examples
@@ -1450,159 +212,95 @@ def scan(
 
     """
     matching_data.to_backend()
-    shape_diff = backend.subtract(
-        matching_data._output_target_shape, matching_data._output_template_shape
-    )
-    shape_diff = backend.multiply(shape_diff, 1 - matching_data._batch_mask)
-
-    if backend.sum(shape_diff < 0) and not pad_fourier:
-        warnings.warn(
-            "Target is larger than template and Fourier padding is turned off."
-            " This can lead to shifted results. You can swap template and target,"
-            " zero-pad the target or turn off template centering."
-        )
     fast_shape, fast_ft_shape, fourier_shift = matching_data.fourier_padding(
         pad_fourier=pad_fourier
     )
 
-    callback_class_args["fourier_shift"] = fourier_shift
-    rfftn, irfftn = backend.build_fft(
+    rfftn, irfftn = be.build_fft(
         fast_shape=fast_shape,
         fast_ft_shape=fast_ft_shape,
-        real_dtype=backend._float_dtype,
-        complex_dtype=backend._complex_dtype,
-        fftargs=fftargs,
+        real_dtype=be._float_dtype,
+        complex_dtype=be._complex_dtype,
     )
-
     template_filter = _setup_template_filter_apply_target_filter(
         matching_data=matching_data,
         rfftn=rfftn,
         irfftn=irfftn,
         fast_shape=fast_shape,
         fast_ft_shape=fast_ft_shape,
+        pad_template_filter=pad_template_filter,
     )
+    template_filter = be.astype(be.to_backend_array(template_filter), be._float_dtype)
 
     setup = matching_setup(
         rfftn=rfftn,
         irfftn=irfftn,
         template=matching_data.template,
+        template_filter=template_filter,
         template_mask=matching_data.template_mask,
         target=matching_data.target,
         target_mask=matching_data.target_mask,
         fast_shape=fast_shape,
         fast_ft_shape=fast_ft_shape,
-        callback_class=callback_class,
-        callback_class_args=callback_class_args,
-        **kwargs,
+        shared_memory_handler=shared_memory_handler,
     )
     rfftn, irfftn = None, None
-
-    template_filter = backend.to_backend_array(template_filter)
-    template_filter = backend.astype(template_filter, backend._float_dtype)
-    template_filter_buffer = backend.arr_to_sharedarr(
-        arr=template_filter,
-        shared_memory_handler=kwargs.get("shared_memory_handler", None),
-    )
-    setup["template_filter"] = (
-        template_filter_buffer,
-        template_filter.shape,
-        template_filter.dtype,
-    )
-
-    callback_class_args["translation_offset"] = backend.astype(
-        matching_data._translation_offset, int
-    )
-    callback_class_args["thread_safe"] = n_jobs > 1
-    callback_class_args["gpu_index"] = kwargs.get("gpu_index", -1)
-
-    n_callback_classes = max(n_jobs // jobs_per_callback_class, 1)
-    callback_class = setup.pop("callback_class", callback_class)
-    callback_class_args = setup.pop("callback_class_args", callback_class_args)
-    callback_classes = [callback_class for _ in range(n_callback_classes)]
-
-    convolution_mode = "same"
-    if backend.sum(backend.to_backend_array(matching_data._target_pad)) > 0:
-        convolution_mode = "valid"
-
-    callback_class_args["fourier_shift"] = fourier_shift
-    callback_class_args["convolution_mode"] = convolution_mode
-    callback_class_args["targetshape"] = setup["targetshape"]
-    callback_class_args["templateshape"] = setup["templateshape"]
-
-    if callback_class == MaxScoreOverRotations:
-        callback_classes = [
-            class_name(
-                score_space_shape=fast_shape,
-                score_space_dtype=backend._float_dtype,
-                shared_memory_handler=kwargs.get("shared_memory_handler", None),
-                rotation_space_dtype=backend._int_dtype,
-                **callback_class_args,
-            )
-            for class_name in callback_classes
-        ]
-
-    matching_data._target, matching_data._template = None, None
-    matching_data._target_mask, matching_data._template_mask = None, None
-
-    setup["fftargs"] = fftargs.copy()
-    convolution_mode = "same"
-    if backend.sum(backend.to_backend_array(matching_data._target_pad)) > 0:
-        convolution_mode = "valid"
-    setup["convolution_mode"] = convolution_mode
     setup["interpolation_order"] = interpolation_order
-    rotation_list = matching_data._split_rotations_on_jobs(n_jobs)
-
-    backend.free_cache()
-
-    def _run_scoring(backend_name, backend_args, rotations, **kwargs):
-        from tme.backends import backend
+    setup["template_filter"] = be.to_sharedarr(template_filter, shared_memory_handler)
+
+    offset = be.to_backend_array(matching_data._translation_offset)
+    convmode = "valid" if getattr(matching_data, "_is_padded", False) else "same"
+    default_callback_args = {
+        "offset": be.astype(offset, be._int_dtype),
+        "thread_safe": n_jobs > 1,
+        "fourier_shift": fourier_shift,
+        "convolution_mode": convmode,
+        "targetshape": matching_data.target.shape,
+        "templateshape": matching_data.template.shape,
+        "fast_shape": fast_shape,
+        "indices": getattr(matching_data, "indices", None),
+        "shared_memory_handler": shared_memory_handler,
+        "only_unique_rotations": True,
+    }
+    default_callback_args.update(callback_class_args)
 
-        backend.change_backend(backend_name, **backend_args)
-        return matching_score(rotations=rotations, **kwargs)
+    matching_data._free_data()
+    be.free_cache()
 
+    # For highly parallel jobs, blocking in certain analyzers becomes a bottleneck
+    if getattr(callback_class, "shared", True):
+        jobs_per_callback_class = 1
+    n_callback_classes = max(n_jobs // jobs_per_callback_class, 1)
+    callback_classes = [
+        callback_class(
+            shape=fast_shape,
+            **default_callback_args,
+        )
+        if callback_class is not None
+        else None
+        for _ in range(n_callback_classes)
+    ]
     callbacks = Parallel(n_jobs=n_jobs)(
-        delayed(_run_scoring)(
-            backend_name=backend._backend_name,
-            backend_args=backend._backend_args,
+        delayed(_wrap_backend(matching_score))(
+            backend_name=be._backend_name,
+            backend_args=be._backend_args,
             rotations=rotation,
-            callback_class=callback_classes[index % n_callback_classes],
-            callback_class_args=callback_class_args,
+            callback=callback_classes[index % n_callback_classes],
             **setup,
         )
-        for index, rotation in enumerate(rotation_list)
+        for index, rotation in enumerate(matching_data._split_rotations_on_jobs(n_jobs))
     )
 
-    callbacks = callbacks[0:n_callback_classes]
     callbacks = [
-        tuple(
-            callback._postprocess(
-                fourier_shift=fourier_shift,
-                convolution_mode=convolution_mode,
-                targetshape=setup["targetshape"],
-                templateshape=setup["templateshape"],
-                shared_memory_handler=kwargs.get("shared_memory_handler", None),
-            )
-        )
-        if hasattr(callback, "_postprocess")
-        else tuple(callback)
+        tuple(callback._postprocess(**default_callback_args))
         for callback in callbacks
         if callback is not None
     ]
-    backend.free_cache()
+    be.free_cache()
 
-    merged_callback = None
     if callback_class is not None:
-        score_indices = None
-        if hasattr(matching_data, "indices"):
-            score_indices = matching_data.indices
-        merged_callback = callback_class.merge(
-            callbacks,
-            **callback_class_args,
-            score_indices=score_indices,
-            inner_merge=True,
-        )
-
-    return merged_callback
+        return callback_class.merge(callbacks, **default_callback_args)
+    return None
 
 
 def scan_subsets(
@@ -1616,10 +314,12 @@ def scan_subsets(
     template_splits: Dict = {},
     pad_target_edges: bool = False,
     pad_fourier: bool = True,
+    pad_template_filter: bool = True,
     interpolation_order: int = 3,
     jobs_per_callback_class: int = 8,
-    **kwargs,
-) -> Tuple:
+    backend_name: str = None,
+    backend_args: Dict = {},
+) -> Optional[Tuple]:
     """
     Wrapper around :py:meth:`scan` that supports matching on splits
     of ``matching_data``.
@@ -1651,21 +351,17 @@ def scan_subsets(
         along each axis.
     pad_fourier: bool, optional
         Whether to pad target and template to the full convolution shape.
+    pad_template_filter: bool, optional
+        Whether to pad potential template filters to the full convolution shape.
     interpolation_order : int, optional
         Order of spline interpolation for rotations.
     jobs_per_callback_class : int, optional
         How many jobs should be processed by a single callback_class instance,
         if ones is provided.
-    **kwargs : various
-        Additional arguments.
-
-    Notes
-    -----
-        Objects in matching_data might be destroyed during computation.
 
     Returns
     -------
-    Tuple
+    Optional[Tuple]
         The merged results from callback_class if provided otherwise None.
 
     Examples
@@ -1720,73 +416,59 @@ def scan_subsets(
     >>>    target_splits = target_splits,
     >>> )
 
-    The retuned ``results`` tuple contains the output of the chosen analyzer.
+    The ``results`` tuple contains the output of the chosen analyzer.
 
     See Also
     --------
     :py:meth:`tme.matching_utils.compute_parallelization_schedule`
     """
-    target_splits = split_numpy_array_slices(
-        matching_data._target.shape, splits=target_splits
-    )
+    template_splits = split_shape(matching_data._template.shape, splits=template_splits)
+    target_splits = split_shape(matching_data._target.shape, splits=target_splits)
     if (len(target_splits) > 1) and not pad_target_edges:
         warnings.warn(
             "Target splitting without padding target edges leads to unreliable "
             "similarity estimates around the split border."
         )
-
-    template_splits = split_numpy_array_slices(
-        matching_data._template.shape, splits=template_splits
-    )
-    target_pad = matching_data.target_padding(pad_target=pad_target_edges)
+    splits = tuple(product(target_splits, template_splits))
 
     outer_jobs, inner_jobs = job_schedule
-    results = Parallel(n_jobs=outer_jobs)(
-        delayed(_run_inner)(
-            backend_name=backend._backend_name,
-            backend_args=backend._backend_args,
-            matching_data=matching_data.subset_by_slice(
-                target_slice=target_split,
-                target_pad=target_pad,
-                template_slice=template_split,
-            ),
-            matching_score=matching_score,
-            matching_setup=matching_setup,
-            n_jobs=inner_jobs,
+    target_pad = matching_data.target_padding(pad_target=pad_target_edges)
+    if hasattr(be, "scan"):
+        corr_scoring = MATCHING_EXHAUSTIVE_REGISTER.get("CORR", (None, None))[1]
+        results = be.scan(
+            matching_data=matching_data,
+            splits=splits,
+            n_jobs=outer_jobs,
+            rotate_mask=matching_score != corr_scoring,
             callback_class=callback_class,
-            callback_class_args=callback_class_args,
-            interpolation_order=interpolation_order,
-            pad_fourier=pad_fourier,
-            gpu_index=index % outer_jobs,
-            **kwargs,
         )
-        for index, (target_split, template_split) in enumerate(
-            product(target_splits, template_splits)
+    else:
+        results = Parallel(n_jobs=outer_jobs)(
+            delayed(_wrap_backend(scan))(
+                backend_name=be._backend_name,
+                backend_args=be._backend_args,
+                matching_data=matching_data.subset_by_slice(
+                    target_slice=target_split,
+                    target_pad=target_pad,
+                    template_slice=template_split,
+                ),
+                matching_score=matching_score,
+                matching_setup=matching_setup,
+                n_jobs=inner_jobs,
+                callback_class=callback_class,
+                callback_class_args=callback_class_args,
+                interpolation_order=interpolation_order,
+                pad_fourier=pad_fourier,
+                gpu_index=index % outer_jobs,
+                pad_template_filter=pad_template_filter,
+            )
+            for index, (target_split, template_split) in enumerate(splits)
         )
-    )
-
-    matching_data._target, matching_data._template = None, None
-    matching_data._target_mask, matching_data._template_mask = None, None
 
-    candidates = None
+    matching_data._free_data()
     if callback_class is not None:
-        candidates = callback_class.merge(
-            results, **callback_class_args, inner_merge=False
-        )
-
-    return candidates
-
-
-MATCHING_EXHAUSTIVE_REGISTER = {
-    "CC": (cc_setup, corr_scoring),
-    "LCC": (lcc_setup, corr_scoring),
-    "CORR": (corr_setup, corr_scoring),
-    "CAM": (cam_setup, corr_scoring),
-    "FLCSphericalMask": (flcSphericalMask_setup, corr_scoring),
-    "FLC": (flc_setup, flc_scoring),
-    "FLC2": (flc_setup, flc_scoring2),
-    "MCC": (mcc_setup, mcc_scoring),
-}
+        return callback_class.merge(results, **callback_class_args)
+    return None
 
 
 def register_matching_exhaustive(
@@ -1803,20 +485,17 @@ def register_matching_exhaustive(
     matching : str
         Name of the matching method.
     matching_setup : Callable
-        The setup function associated with the name.
+        Corresponding setup function.
     matching_scoring : Callable
-        The scoring function associated with the name.
+        Corresponing scoring function.
     memory_class : MatchingMemoryUsage
-        The custom memory estimation class extending
-        :py:class:`tme.matching_memory.MatchingMemoryUsage`.
+        Child of :py:class:`tme.memory.MatchingMemoryUsage`.
 
     Raises
     ------
     ValueError
-        If a function with the name ``matching`` already exists in the registry.
-    ValueError
-        If ``memory_class`` is not a subclass of
-        :py:class:`tme.matching_memory.MatchingMemoryUsage`.
+        If a function with the name ``matching`` already exists in the registry, or
+        if ``memory_class`` is no child of :py:class:`tme.memory.MatchingMemoryUsage`.
     """
 
     if matching in MATCHING_EXHAUSTIVE_REGISTER: