pytme 0.1.7__cp311-cp311-macosx_14_0_arm64.whl → 0.1.8__cp311-cp311-macosx_14_0_arm64.whl

{pytme-0.1.7.data → pytme-0.1.8.data}/scripts/match_template.py

@@ -539,15 +539,14 @@ def main():
         opening_axis, tilt_axis = [int(x) for x in args.wedge_axes.split(",")]
 
         if args.tilt_step is not None:
-            tilt_angles = np.arange(
-                -tilt_start, tilt_stop + args.tilt_step, args.tilt_step
-            )
-            angles = np.zeros((template.data.ndim, tilt_angles.size))
-            angles[tilt_axis, :] = tilt_angles
-            template_filter["wedge_mask"] = {
-                "tilt_angles": angles,
+            template_filter["step_wedge_mask"] = {
+                "start_tilt": tilt_start,
+                "stop_tilt": tilt_stop,
+                "tilt_step": args.tilt_step,
                 "sigma": args.wedge_smooth,
-                "opening_axes": opening_axis,
+                "opening_axis": opening_axis,
+                "tilt_axis": tilt_axis,
+                "omit_negative_frequencies": True,
             }
         else:
             template_filter["continuous_wedge_mask"] = {
@@ -557,6 +556,7 @@ def main():
                 "opening_axis": opening_axis,
                 "infinite_plane": True,
                 "sigma": args.wedge_smooth,
+                "omit_negative_frequencies": True,
             }
 
     if template_mask is None:
@@ -593,7 +593,9 @@ def main():
     print("\n" + "-" * 80)
 
     if args.scramble_phases:
-        template.data = scramble_phases(template.data, noise_proportion=1.0)
+        template.data = scramble_phases(
+            template.data, noise_proportion=1.0, normalize_power=True
+        )
 
     available_memory = backend.get_available_memory()
     if args.use_gpu:
@@ -671,7 +673,7 @@ def main():
         exit(-1)
 
     analyzer_args = {
-        "score_threshold": 0.0,
+        "score_threshold": 0,
         "number_of_peaks": 1000,
         "convolution_mode": "valid",
         "use_memmap": args.use_memmap,

{pytme-0.1.7.data → pytme-0.1.8.data}/scripts/postprocess.py

@@ -15,6 +15,7 @@ from dataclasses import dataclass
 
 import numpy as np
 from scipy.spatial.transform import Rotation
+from numpy.typing import NDArray
 
 from tme import Density, Structure
 from tme.analyzer import (
@@ -28,6 +29,7 @@ from tme.matching_utils import (
     load_pickle,
     euler_to_rotationmatrix,
     euler_from_rotationmatrix,
+    centered_mask,
 )
 
 PEAK_CALLERS = {
@@ -75,6 +77,13 @@ def parse_args():
         help="Minimum distance from target boundaries. Ignored when --orientations "
         "is provided.",
     )
+    parser.add_argument(
+        "--mask_edges",
+        action="store_true",
+        default=False,
+        help="Whether to mask edges of the input score array according to the template shape."
+        "Uses twice the value of --min_boundary_distance if boht are provided.",
+    )
     parser.add_argument(
         "--wedge_mask",
         type=str,
@@ -403,11 +412,25 @@ class Orientations:
         return translation, rotation, score, detail
 
 
+def load_template(filepath: str, sampling_rate: NDArray) -> "Density":
+    try:
+        template = Density.from_file(filepath)
+        template, _ = template.centered(0)
+        center_of_mass = template.center_of_mass(template.data)
+    except ValueError:
+        template = Structure.from_file(filepath)
+        center_of_mass = template.center_of_mass()[::-1]
+        template = Density.from_structure(template, sampling_rate=sampling_rate)
+
+    return template, center_of_mass
+
+
 def main():
     args = parse_args()
     data = load_pickle(args.input_file)
 
     meta = data[-1]
+    target_origin, _, sampling_rate, cli_args = meta
 
     if args.orientations is not None:
         orientations = Orientations.from_file(
@@ -419,6 +442,17 @@ def main():
         # Output is MaxScoreOverRotations
         if data[0].ndim == data[2].ndim:
             scores, offset, rotation_array, rotation_mapping, meta = data
+            if args.mask_edges:
+                template, center_of_mass = load_template(
+                    cli_args.template, sampling_rate=sampling_rate
+                )
+                if not cli_args.no_centering:
+                    template, *_ = template.centered(0)
+                mask_size = template.shape
+                if args.min_boundary_distance > 0:
+                    mask_size = 2 * args.min_boundary_distance
+                scores = centered_mask(scores, np.subtract(scores.shape, mask_size) + 1)
+
             peak_caller = PEAK_CALLERS[args.peak_caller](
                 number_of_peaks=args.number_of_peaks,
                 min_distance=args.min_distance,
@@ -458,19 +492,11 @@ def main():
         orientations.to_file(filename=f"{args.output_prefix}.tsv", file_format="text")
         exit(0)
 
-    target_origin, _, sampling_rate, cli_args = meta
-
-    template_is_density, index = True, 0
     _, template_extension = splitext(cli_args.template)
-    try:
-        template = Density.from_file(cli_args.template)
-        template, _ = template.centered(0)
-        center_of_mass = template.center_of_mass(template.data)
-    except ValueError:
-        template_is_density = False
-        template = Structure.from_file(cli_args.template)
-        center_of_mass = template.center_of_mass()[::-1]
-        template = Density.from_structure(template, sampling_rate=sampling_rate)
+    template, center_of_mass = load_template(
+        filepath=cli_args.template, sampling_rate=sampling_rate
+    )
+    template_is_density, index = isinstance(template, Density), 0
 
     if args.output_format == "relion":
         new_shape = np.add(template.shape, np.mod(template.shape, 2))

{pytme-0.1.7.data → pytme-0.1.8.data}/scripts/preprocessor_gui.py

@@ -29,17 +29,17 @@ preprocessor = Preprocessor()
 SLIDER_MIN, SLIDER_MAX = 0, 25
 
 
-def gaussian_filter(template, sigma: float, **kwargs: dict):
+def gaussian_filter(template: NDArray, sigma: float, **kwargs: dict) -> NDArray:
     return preprocessor.gaussian_filter(template=template, sigma=sigma, **kwargs)
 
 
 def bandpass_filter(
-    template,
+    template: NDArray,
     minimum_frequency: float,
     maximum_frequency: float,
     gaussian_sigma: float,
     **kwargs: dict,
-):
+) -> NDArray:
     return preprocessor.bandpass_filter(
         template=template,
         minimum_frequency=minimum_frequency,
@@ -51,8 +51,8 @@ def bandpass_filter(
 
 
 def difference_of_gaussian_filter(
-    template, sigmas: Tuple[float, float], **kwargs: dict
-):
+    template: NDArray, sigmas: Tuple[float, float], **kwargs: dict
+) -> NDArray:
     low_sigma, high_sigma = sigmas
     return preprocessor.difference_of_gaussian_filter(
         template=template, low_sigma=low_sigma, high_sigma=high_sigma, **kwargs
@@ -60,7 +60,7 @@ def difference_of_gaussian_filter(
 
 
 def edge_gaussian_filter(
-    template,
+    template: NDArray,
     sigma: float,
     edge_algorithm: Annotated[
         str,
@@ -68,7 +68,7 @@ def edge_gaussian_filter(
     ],
     reverse: bool = False,
     **kwargs: dict,
-):
+) -> NDArray:
     return preprocessor.edge_gaussian_filter(
         template=template,
         sigma=sigma,
@@ -78,13 +78,13 @@ def edge_gaussian_filter(
 
 
 def local_gaussian_filter(
-    template,
+    template: NDArray,
     lbd: float,
     sigma_range: Tuple[float, float],
     gaussian_sigma: float,
     reverse: bool = False,
     **kwargs: dict,
-):
+) -> NDArray:
     return preprocessor.local_gaussian_filter(
         template=template,
         lbd=lbd,
@@ -94,22 +94,30 @@ def local_gaussian_filter(
 
 
 def ntree(
-    template,
+    template: NDArray,
     sigma_range: Tuple[float, float],
     **kwargs: dict,
-):
+) -> NDArray:
     return preprocessor.ntree_filter(template=template, sigma_range=sigma_range)
 
 
 def mean(
-    template,
+    template: NDArray,
     width: int,
     **kwargs: dict,
-):
+) -> NDArray:
     return preprocessor.mean_filter(template=template, width=width)
 
 
-def resolution_sphere(template: NDArray, cutoff_angstrom: float, highpass : bool = False, sampling_rate = None):
+def resolution_sphere(
+    template: NDArray,
+    cutoff_angstrom: float,
+    highpass: bool = False,
+    sampling_rate=None,
+) -> NDArray:
+    if cutoff_angstrom == 0:
+        return template
+
     cutoff_frequency = np.max(2 * sampling_rate / cutoff_angstrom)
 
     min_freq, max_freq = 0, cutoff_frequency
@@ -123,34 +131,39 @@ def resolution_sphere(template: NDArray, cutoff_angstrom: float, highpass : bool
         omit_negative_frequencies=False,
     )
 
-    mask = np.fft.ifftshift(mask)
     template_ft = np.fft.fftn(template)
-    np.multiply(template_ft, mask, out = template_ft)
-
+    np.multiply(template_ft, mask, out=template_ft)
     return np.fft.ifftn(template_ft).real
 
 
-def resolution_gaussian(template : NDArray, cutoff_angstrom : float, highpass : bool = False,
-    sampling_rate = None):
+def resolution_gaussian(
+    template: NDArray,
+    cutoff_angstrom: float,
+    highpass: bool = False,
+    sampling_rate=None,
+) -> NDArray:
+    if cutoff_angstrom == 0:
+        return template
+
     grid = preprocessor.fftfreqn(
-        shape = template.shape, sampling_rate = sampling_rate / sampling_rate.max()
+        shape=template.shape, sampling_rate=sampling_rate / sampling_rate.max()
     )
 
     sigma_fourier = np.divide(
-        np.max(2 * sampling_rate / cutoff_angstrom),
-        np.sqrt(2 * np.log(2))
+        np.max(2 * sampling_rate / cutoff_angstrom), np.sqrt(2 * np.log(2))
     )
 
-    mask = np.exp(-grid ** 2 / (2 * sigma_fourier ** 2))
+    mask = np.exp(-(grid**2) / (2 * sigma_fourier**2))
     if highpass:
         mask = 1 - mask
 
     mask = np.fft.ifftshift(mask)
-    template_ft = np.fft.fftn(template)
-    np.multiply(template_ft, mask, out = template_ft)
 
+    template_ft = np.fft.fftn(template)
+    np.multiply(template_ft, mask, out=template_ft)
     return np.fft.ifftn(template_ft).real
 
+
 def wedge(
     template: NDArray,
     tilt_start: float,
@@ -162,7 +175,7 @@ def wedge(
     omit_negative_frequencies: bool = True,
     extrude_plane: bool = True,
     infinite_plane: bool = True,
-):
+) -> NDArray:
     template_ft = np.fft.rfftn(template)
 
     if tilt_step <= 0:
@@ -181,15 +194,14 @@ def wedge(
         template = np.real(np.fft.irfftn(template_ft))
         return template
 
-    tilt_angles = np.arange(-tilt_start, tilt_stop, tilt_step)
-    angles = np.zeros((template.ndim, tilt_angles.size))
-    angles[tilt_axis, :] = tilt_angles
-
-    wedge_mask = preprocessor.wedge_mask(
-        tilt_angles=angles,
+    wedge_mask = preprocessor.step_wedge_mask(
+        start_tilt=tilt_start,
+        stop_tilt=tilt_stop,
+        tilt_axis=tilt_axis,
+        tilt_step=tilt_step,
+        opening_axis=opening_axis,
         shape=template.shape,
         sigma=gaussian_sigma,
-        opening_axes=opening_axis,
         omit_negative_frequencies=omit_negative_frequencies,
     )
     np.multiply(template_ft, wedge_mask, out=template_ft)
@@ -197,7 +209,7 @@ def wedge(
     return template
 
 
-def compute_power_spectrum(template: NDArray):
+def compute_power_spectrum(template: NDArray) -> NDArray:
     return np.fft.fftshift(np.log(np.abs(np.fft.fftn(template))))
 
 
@@ -262,8 +274,8 @@ WRAPPED_FUNCTIONS = {
     "mean_filter": mean,
     "wedge_filter": wedge,
     "power_spectrum": compute_power_spectrum,
-    "resolution_gaussian" : resolution_gaussian,
-    "resolution_sphere" : resolution_sphere,
+    "resolution_gaussian": resolution_gaussian,
+    "resolution_sphere": resolution_sphere,
 }
 
 EXCLUDED_FUNCTIONS = [
@@ -405,7 +417,7 @@ class FilterWidget(widgets.Container):
 
 def sphere_mask(
     template: NDArray, center_x: float, center_y: float, center_z: float, radius: float
-):
+) -> NDArray:
     return create_mask(
         mask_type="ellipse",
         shape=template.shape,
@@ -422,7 +434,7 @@ def ellipsod_mask(
     radius_x: float,
     radius_y: float,
     radius_z: float,
-):
+) -> NDArray:
     return create_mask(
         mask_type="ellipse",
         shape=template.shape,
@@ -439,7 +451,7 @@ def box_mask(
     height_x: int,
     height_y: int,
     height_z: int,
-):
+) -> NDArray:
     return create_mask(
         mask_type="box",
         shape=template.shape,
@@ -457,7 +469,7 @@ def tube_mask(
     inner_radius: float,
     outer_radius: float,
     height: int,
-):
+) -> NDArray:
     return create_mask(
         mask_type="tube",
         shape=template.shape,
@@ -480,7 +492,7 @@ def wedge_mask(
     omit_negative_frequencies: bool = False,
     extrude_plane: bool = True,
     infinite_plane: bool = True,
-):
+) -> NDArray:
     if tilt_step <= 0:
         wedge_mask = preprocessor.continuous_wedge_mask(
             start_tilt=tilt_start,
@@ -496,25 +508,24 @@ def wedge_mask(
         wedge_mask = np.fft.fftshift(wedge_mask)
         return wedge_mask
 
-    tilt_angles = np.arange(-tilt_start, tilt_stop + tilt_step, tilt_step)
-    angles = np.zeros((template.ndim, tilt_angles.size))
-
-    angles[tilt_axis, :] = tilt_angles
-
-    wedge_mask = preprocessor.wedge_mask(
-        tilt_angles=angles,
+    wedge_mask = preprocessor.step_wedge_mask(
+        start_tilt=tilt_start,
+        stop_tilt=tilt_stop,
+        tilt_axis=tilt_axis,
+        tilt_step=tilt_step,
+        opening_axis=opening_axis,
         shape=template.shape,
         sigma=gaussian_sigma,
-        opening_axes=opening_axis,
         omit_negative_frequencies=omit_negative_frequencies,
     )
+
     wedge_mask = np.fft.fftshift(wedge_mask)
     return wedge_mask
 
 
 def threshold_mask(
     template: NDArray, standard_deviation: float = 5.0, invert: bool = False
-):
+) -> NDArray:
     template_mean = template.mean()
     template_deviation = standard_deviation * template.std()
     upper = template_mean + template_deviation
@@ -526,6 +537,15 @@ def threshold_mask(
     return mask
 
 
+def lowpass_mask(template: NDArray, sigma: float = 1.0):
+    template = template / template.max()
+    template = (template > np.exp(-2)) * 128.0
+    template = preprocessor.gaussian_filter(template=template, sigma=sigma)
+    mask = template > np.exp(-2)
+
+    return mask
+
+
 class MaskWidget(widgets.Container):
     def __init__(self, viewer):
         super().__init__(layout="vertical")
@@ -543,6 +563,7 @@ class MaskWidget(widgets.Container):
             "Box": box_mask,
             "Wedge": wedge_mask,
             "Threshold": threshold_mask,
+            "Lowpass": lowpass_mask,
         }
 
         self.method_dropdown = widgets.ComboBox(
@@ -606,6 +627,7 @@ class MaskWidget(widgets.Container):
             arr=active_layer.data,
             rotation_matrix=rotation_matrix,
             use_geometric_center=False,
+            order=1,
         )
         eps = np.finfo(rotated_data.dtype).eps
         rotated_data[rotated_data < eps] = 0
@@ -636,10 +658,10 @@ class MaskWidget(widgets.Container):
             dict(zip(["height_x", "height_y", "height_z"], coordinates_heights))
         )
 
-        defaults["radius"] = np.min(coordinate_radius)
+        defaults["radius"] = np.max(coordinate_radius)
         defaults["inner_radius"] = np.min(coordinate_radius)
         defaults["outer_radius"] = np.max(coordinate_radius)
-        defaults["height"] = defaults["radius"]
+        defaults["height"] = np.max(coordinates_heights)
 
         for widget in self.action_widgets:
             if widget.name in defaults:

{pytme-0.1.7.dist-info → pytme-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytme
-Version: 0.1.7
+Version: 0.1.8
 Summary: Python Template Matching Engine
 Author: Valentin Maurer
 Author-email: Valentin Maurer <valentin.maurer@embl-hamburg.de>
@@ -31,7 +31,7 @@ A software for template matching on electron microscopy data.
 
 ### Installation
 
-There are three ways to get PyTME up and running:
+There are three ways to get pyTME up and running:
 
 1. **Install from Source:**
 ```bash
@@ -44,7 +44,7 @@ pip install pytme
 ```
 
 3. **Docker Container:**
-Build and use the PyTME container from Docker. This assumes a linux/amd64 platform.
+Build and use the pyTME container from Docker. This assumes a linux/amd64 platform.
 ```bash
 docker build \
 	-t pytme \
@@ -58,12 +58,14 @@ docker build \
 
 ### Publication
 
-The pyTME publication is available on [bioRxiv](https://www.biorxiv.org/content/10.1101/2023.10.23.563472v1).
+The pyTME publication is available on [SoftwareX](https://www.sciencedirect.com/science/article/pii/S2352711024000074).
 
 ```
-@article{Maurer2023.10.23.563472,
-	author = {Valentin J. Maurer and Marc Siggel and Jan Kosinski},
-	journal = {bioRxiv},
+@article{Maurer:2024aa,
+	author = {Maurer, Valentin J. and Siggel, Marc and Kosinski, Jan},
+	journal = {SoftwareX},
+	pages = {101636},
 	title = {PyTME (Python Template Matching Engine): A fast, flexible, and multi-purpose template matching library for cryogenic electron microscopy data},
-	year = {2023}}
+	volume = {25},
+	year = {2024}}
 ```

{pytme-0.1.7.dist-info → pytme-0.1.8.dist-info}/RECORD

@@ -1,34 +1,34 @@
-pytme-0.1.7.data/scripts/estimate_ram_usage.py,sha256=R1NDpFajcF-MonJ4a43SfDlA-nxBYwK7D2quzCdsVFM,2767
-pytme-0.1.7.data/scripts/match_template.py,sha256=5AmxJi4ycQGqR5g19tFkPKWU3gTUdGaGfkq5LFFxx2Q,25706
-pytme-0.1.7.data/scripts/postprocess.py,sha256=3qyV66b8FS2yRp8TizuFin39HVgcpH8nibhEuV_Js0E,20883
-pytme-0.1.7.data/scripts/preprocess.py,sha256=zog-l2Je-GeouJ6SnamOMuHgTn7fFPiGnO5X03y5qSY,2527
-pytme-0.1.7.data/scripts/preprocessor_gui.py,sha256=iYP_7neFx3ui1ujvgIDTivgDeX0JNzpBuA0hwv4iedc,28472
+pytme-0.1.8.data/scripts/estimate_ram_usage.py,sha256=R1NDpFajcF-MonJ4a43SfDlA-nxBYwK7D2quzCdsVFM,2767
+pytme-0.1.8.data/scripts/match_template.py,sha256=moMpH8mLk42ZkW0vXzd_TMCWsTUKXohxuWrqHLh8lks,25743
+pytme-0.1.8.data/scripts/postprocess.py,sha256=MKyIgsXWRlO6cU8qGqe_tSSKIF2pNTRSuaDoyXZ22G8,21938
+pytme-0.1.8.data/scripts/preprocess.py,sha256=zog-l2Je-GeouJ6SnamOMuHgTn7fFPiGnO5X03y5qSY,2527
+pytme-0.1.8.data/scripts/preprocessor_gui.py,sha256=AKT_ovrdbPiOJ_ampRK3s0jslUKwdVQRF6XUuJ-6GDo,29000
 scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 scripts/estimate_ram_usage.py,sha256=rN7haobnHg3YcgGJIp81FNiCzy8-saJGeEurQlmQmNQ,2768
-scripts/match_template.py,sha256=7t34tNCiXtv6fSutZZ4IHNXBfVDIltWvQMHqmihliDU,25707
-scripts/postprocess.py,sha256=SzMj9uH8Yykwy8XU3dCKFCCS5kliW-mTfnGse6Q7oUA,20884
+scripts/match_template.py,sha256=dEvw9KDqgR7vZcIVn5Bld6FcRt_rp7nblybVuJhLYbs,25744
+scripts/postprocess.py,sha256=NDTo9qbnAzhxo_tbnR8oWlDrsflUGu-Sj4ULqOPix1U,21939
 scripts/preprocess.py,sha256=ebJVLxbRlB6TI5YHNr0VavZ4lmaRdf8QVafyiDhh_oU,2528
-scripts/preprocessor_gui.py,sha256=UZOezE1C_aavT8yvydi8eud664wI2X8dnfWITVxbXDg,28473
+scripts/preprocessor_gui.py,sha256=E0F6quZnu32yZsBle38KDel5naHpC3Aci0uMCyo8DzY,29001
 tme/__init__.py,sha256=NiPnYfrwb8e04ETubqwd3yzCGBliLBXRvCylD-T59jY,212
-tme/__version__.py,sha256=YpKDcdV7CqL8n45u267wKtyloM13FSVbOdrqgNZnSLM,22
+tme/__version__.py,sha256=C69ADlbQREQlR15trneyA2sk8x0-oH4rDAX5fsv19_U,22
 tme/analyzer.py,sha256=Yh1o4kpjlRSq6ki5KiRsipF3tBwtSrL3KYfEm87NNOc,39188
-tme/density.py,sha256=JCkh8DZPsTVeq6H7f6ly39kSXyn2Epl-wsKqyM2lrRE,83917
+tme/density.py,sha256=AwHhi5sHvOzXeXCnYHB_Mpw1kCG0x0cBg8NmwDalwKA,84326
 tme/extensions.cpython-311-darwin.so,sha256=vSNHvFMev0jvKCEwE4lzqgg9520Ucxf27gy3qqixYns,395056
 tme/helpers.py,sha256=TMtBuJoZk6q4M_rfkr8yPpGzJD74ycqyN3yXMuU9mr4,23625
-tme/matching_data.py,sha256=Ch_sgUzMRdGLkMyTlwQKxnTVK5og-7KkweRoSKOtD5Y,21150
-tme/matching_exhaustive.py,sha256=zWn3YJoOBgq-DORoafIrcXsXGx19brADiEQX1Bdl-to,52231
+tme/matching_data.py,sha256=M6Iq4dnIlLzTvnHLrjC9uoeyNEDAsCdCwI9Pga-k9pE,22140
+tme/matching_exhaustive.py,sha256=b3I2zzclEnCXYgjQ0l8C2O2UbH9pF1WKFRyDRLsrvTs,53862
 tme/matching_memory.py,sha256=GB71b7m-h0fF1A5HCUlAYQh62-Aq2lHgIoRGPYEzZrU,11245
 tme/matching_optimization.py,sha256=oo9HJV65Ji1oL2JumxgmtrHSMUQsKrpDhdOO5o-fAdM,37388
-tme/matching_utils.py,sha256=iYsvkIwM7I1n9SIXKqhkuA5Q5QU3t7bDDZVleUbv6TM,36391
+tme/matching_utils.py,sha256=-cs3Nq8d9snaZ0mlgkSqdEc0c9A1TkX1sz93YY7TE_k,37227
 tme/parser.py,sha256=tA9ABeV95cZn8lJCukVUaocQ9RguR6ZZzQsMXf_-ud0,13887
-tme/preprocessor.py,sha256=_5j5gDzmMqjB3c4zN9OSY-T1XDhzPbRB3A_eOCeJ92Y,46596
+tme/preprocessor.py,sha256=UkWKBR9Z7FU-xUZ0ER_4c6lgLW9-MGmm-YN5Cs2mWD0,50631
 tme/structure.py,sha256=AKGZML8bnsqZOpAOQADU1XDXZ-ITPhr3Dc0udQUMP6E,54134
 tme/types.py,sha256=2Tyh_xnMLxIWYb3aJDAUb6GWpaL6gcYMUm2YNbJlAPI,295
 tme/backends/__init__.py,sha256=xB2GBUFRskppvEs6S74VH0Pi-nXnIvu9_QFhESlcl3Y,4366
 tme/backends/cupy_backend.py,sha256=CRwGqtycawTIfSfEDWpCQh4mcG03wad9mEODqzy75yQ,10648
 tme/backends/matching_backend.py,sha256=E3cMXnMEazYJUr9RP5Q5rMEAf3vbkiOwzWrx5amt_nI,29311
 tme/backends/mlx_backend.py,sha256=MrwICZpUiAcpZXON70r4SH-KsWxfhq1PdHUe80WbT-k,8467
-tme/backends/npfftw_backend.py,sha256=vIgO82t7a3RNTf6AYHopB7fTt0GhxrfT4T5IJsPjAD0,26441
+tme/backends/npfftw_backend.py,sha256=VsoKwMmusUxW81b2YSa6KO0_pfUw-51fdv9O6BkN0cQ,26200
 tme/backends/pytorch_backend.py,sha256=lKv_7q0gDrCfbwHR38tfj5ixpD1lfCpqpfqS1KnOqdU,17978
 tme/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tme/data/c48n309.npy,sha256=NwH64mOEbm3tStq5c98o81fY1vMOoq4nvXDAh7Z7iZg,296768
@@ -53,9 +53,9 @@ tme/data/c600v.npy,sha256=JqSu3ALoL1A9iguehc0YGUMFPsh2fprHHp76VXeFXIw,2528
 tme/data/c600vc.npy,sha256=Yht-GFXDSjjGvsjFBvyxxEZAI-ODADPd5gEgFNZQVTA,14528
 tme/data/metadata.yaml,sha256=fAgX-mEzB0QMHTEtYDG4cSMbJhYxBbDJH3sdvJvL7a8,750
 tme/data/quat_to_numpy.py,sha256=-gkDZb10fKBxwfYrSLCUWvMB76TzZWELCeKsYProwws,1333
-pytme-0.1.7.dist-info/LICENSE,sha256=K1IUNSVAz8BXbpH5EA8y5FpaHdvFXnAF2zeK95Lr2bY,18467
-pytme-0.1.7.dist-info/METADATA,sha256=oav-9lg-OtEyMjPwgCmTPJNcKldIzNMMuYZcZMAIdNk,2081
-pytme-0.1.7.dist-info/WHEEL,sha256=-A_a4qtPtE7THspi2apYwdQWhv5IW90gRqlNhh7cwR4,110
-pytme-0.1.7.dist-info/entry_points.txt,sha256=ff3LQL3FCWfCYOwFiP9zatm7laUbnwCkuPELkQVyUO4,241
-pytme-0.1.7.dist-info/top_level.txt,sha256=J8FUkazOb2fZ0n_KexnqCGyNOtie2bwisFSUBiM5-0w,12
-pytme-0.1.7.dist-info/RECORD,,
+pytme-0.1.8.dist-info/LICENSE,sha256=K1IUNSVAz8BXbpH5EA8y5FpaHdvFXnAF2zeK95Lr2bY,18467
+pytme-0.1.8.dist-info/METADATA,sha256=zG0llsYOJsC3hU1UCPwepti-FJg1O8VPn1ovO-j4MNE,2121
+pytme-0.1.8.dist-info/WHEEL,sha256=-A_a4qtPtE7THspi2apYwdQWhv5IW90gRqlNhh7cwR4,110
+pytme-0.1.8.dist-info/entry_points.txt,sha256=ff3LQL3FCWfCYOwFiP9zatm7laUbnwCkuPELkQVyUO4,241
+pytme-0.1.8.dist-info/top_level.txt,sha256=J8FUkazOb2fZ0n_KexnqCGyNOtie2bwisFSUBiM5-0w,12
+pytme-0.1.8.dist-info/RECORD,,

scripts/match_template.py

@@ -539,15 +539,14 @@ def main():
         opening_axis, tilt_axis = [int(x) for x in args.wedge_axes.split(",")]
 
         if args.tilt_step is not None:
-            tilt_angles = np.arange(
-                -tilt_start, tilt_stop + args.tilt_step, args.tilt_step
-            )
-            angles = np.zeros((template.data.ndim, tilt_angles.size))
-            angles[tilt_axis, :] = tilt_angles
-            template_filter["wedge_mask"] = {
-                "tilt_angles": angles,
+            template_filter["step_wedge_mask"] = {
+                "start_tilt": tilt_start,
+                "stop_tilt": tilt_stop,
+                "tilt_step": args.tilt_step,
                 "sigma": args.wedge_smooth,
-                "opening_axes": opening_axis,
+                "opening_axis": opening_axis,
+                "tilt_axis": tilt_axis,
+                "omit_negative_frequencies": True,
             }
         else:
             template_filter["continuous_wedge_mask"] = {
@@ -557,6 +556,7 @@ def main():
                 "opening_axis": opening_axis,
                 "infinite_plane": True,
                 "sigma": args.wedge_smooth,
+                "omit_negative_frequencies": True,
             }
 
     if template_mask is None:
@@ -593,7 +593,9 @@ def main():
     print("\n" + "-" * 80)
 
     if args.scramble_phases:
-        template.data = scramble_phases(template.data, noise_proportion=1.0)
+        template.data = scramble_phases(
+            template.data, noise_proportion=1.0, normalize_power=True
+        )
 
     available_memory = backend.get_available_memory()
     if args.use_gpu:
@@ -671,7 +673,7 @@ def main():
         exit(-1)
 
     analyzer_args = {
-        "score_threshold": 0.0,
+        "score_threshold": 0,
         "number_of_peaks": 1000,
         "convolution_mode": "valid",
         "use_memmap": args.use_memmap,