petpal 0.5.10__tar.gz → 0.6.0__tar.gz

{petpal-0.5.10 → petpal-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: petpal
-Version: 0.5.10
+Version: 0.6.0
 Summary: PET-PAL (Positron Emission Tomography Processing and Analysis Library)
 Project-URL: Repository, https://github.com/PETPAL-WUSM/PETPAL.git
 Author-email: Noah Goldman <noahg@wustl.edu>, Bradley Judge <bjudge@wustl.edu>, Furqan Dar <dar@wustl.edu>, Kenan Oestreich <kenan.oestreich@wustl.edu>

{petpal-0.5.10 → petpal-0.6.0}/petpal/cli/cli_preproc.py

@@ -83,7 +83,12 @@ Examples:
    .. code-block:: bash
 
        petpal-preproc warp-pet-atlas -i /path/to/input_img.nii.gz -o petpal_reg-atlas.nii.gz --anatomical /path/to/anat.nii.gz --reference-atlas /path/to/atlas.nii.gz
- 
+
+    * Crop segmentation to PET FOV:
+
+    .. code-block:: bash
+
+        petpal-preproc seg-crop -i /path/to/input_img.nii.gz -o petpal_cropped_seg.nii.gz --segmentation /path/to/segmentation.nii.gz
 
 See Also:
     * :mod:`~petpal.preproc.image_operations_4d` - module used for operations on 4D images.
@@ -99,7 +104,8 @@ from ..preproc import (image_operations_4d,
                        motion_corr,
                        register,
                        regional_tac_extraction,
-                       standard_uptake_value)
+                       standard_uptake_value,
+                       segmentation_tools)
 
 
 _PREPROC_EXAMPLES_ = r"""
@@ -126,6 +132,10 @@ Examples:
     petpal-preproc warp-pet-atlas -i /path/to/input_img.nii.gz -o petpal_reg-atlas.nii.gz --anatomical /path/to/anat.nii.gz --reference-atlas /path/to/atlas.nii.gz
   - SUV:
     petpal-preproc suv -i /path/to/input_img.nii.gz -o petpal_suv.nii.gz --weight 75 --dose 250 --start-time 1200 --end-time 3600
+  - Crop segmentation image to PET FOV:
+    petpal-preproc seg-crop -i /path/to/input_img.nii.gz -o petpal_cropped_seg.nii.gz --segmentation /path/to/segmentation.nii.gz
+  - Add eroded white matter region to segmentation image:
+    petpal-preproc eroded-wm -i /path/to/input_segmentation.nii.gz -o petpal_seg_with_eroded_wm.nii.gz
 """
 
 
@@ -357,6 +367,19 @@ def _generate_args() -> argparse.ArgumentParser:
                             required=True,
                             help='End time for SUV calculation in seconds from scan start',
                             type=float)
+
+    parser_seg_crop = subparsers.add_parser('seg-crop',help='Crop segmentation image to PET FOV')
+    _add_common_args(parser_seg_crop)
+    parser_seg_crop.add_argument('-s',
+                                 '--segmentation',
+                                 required=True,
+                                 help='Path to segmentation image',
+                                 type=str)
+
+    parser_eroded_wm = subparsers.add_parser('eroded-wm',
+                                             help='Add eroded white matter region to segmentation image')
+    _add_common_args(parser_eroded_wm)
+
     return parser
 
 
@@ -454,6 +477,15 @@ def main():
                                       end_time=args.end_time,
                                       weight=args.weight,
                                       dose=args.dose)
+        case 'seg_crop':
+            input_img = ants.image_read(filename=args.input_img)
+            seg_img = ants.image_read(filename=args.segmentation)
+            seg_cropped = segmentation_tools.seg_crop_to_pet_fov(pet_img=input_img,
+                                                                 segmentation_img=seg_img)
+            ants.image_write(seg_cropped,args.out_img)
+        case 'eroded_wm':
+            segmentation_tools.eroded_wm_segmentation(input_segmentation_path=args.input_img,
+                                                      out_segmentation_path=args.out_img)
 
 if __name__ == "__main__":
     main()

{petpal-0.5.10 → petpal-0.6.0}/petpal/preproc/segmentation_tools.py

@@ -84,7 +84,9 @@ def segmentations_merge(segmentation_primary: np.ndarray,
             regions added.
     """
     for region in regions:
-        region_mask = (segmentation_secondary > region - 0.1) & (segmentation_secondary < region + 0.1)
+        condition_above = segmentation_secondary > region - 0.1
+        condition_below = segmentation_secondary < region + 0.1
+        region_mask = condition_above & condition_below
         segmentation_primary[region_mask] = region
     return segmentation_primary
 
@@ -229,8 +231,9 @@ def resample_segmentation(input_image_path: str,
     seg_image = nibabel.load(segmentation_image_path)
     pet_series = pet_image.get_fdata()
     image_first_frame = pet_series[:, :, :, 0]
+    to_vox_map_tuple = (image_first_frame.shape, pet_image.affine)
     seg_resampled = processing.resample_from_to(from_img=seg_image,
-                                                to_vox_map=(image_first_frame.shape, pet_image.affine),
+                                                to_vox_map=to_vox_map_tuple,
                                                 order=0)
     nibabel.save(seg_resampled, out_seg_path)
     if verbose:
@@ -238,23 +241,30 @@ def resample_segmentation(input_image_path: str,
 
 
 def vat_wm_ref_region(input_segmentation_path: str,
-                      out_segmentation_path: str):
+                      out_segmentation_path: str | None) -> ants.ANTsImage:
     """
     Generates the cortical white matter reference region described in O'Donnell
-    JL et al. (2024) PET Quantification of [18F]VAT in Human Brain and Its 
+    JL et al. (2024).
+     
+    Reference: O'Donnell JL et al. (2024). PET Quantification of [18F]VAT in Human Brain and Its 
     Test-Retest Reproducibility and Age Dependence. J Nucl Med. 2024 Jun 
     3;65(6):956-961. doi: 10.2967/jnumed.123.266860. PMID: 38604762; PMCID:
-    PMC11149597. Requires FreeSurfer segmentation with original label mappings.
+    PMC11149597.
+    
+    Requires FreeSurfer segmentation with original label mappings. 
 
     Args:
         input_segmentation_path (str): Path to segmentation on which white
             matter reference region is computed.
         out_segmentation_path (str): Path to which white matter reference
             region mask image is saved.
+
+    Returns:
+        wm_erode (ants.ANTsImage): Eroded white matter reference region mask image.
     """
     wm_regions = [2,41,251,252,253,254,255,77,3000,3001,3002,3003,3004,3005,
                   3006,3007,3008,3009,3010,3011,3012,3013,3014,3015,3016,3017,
-                  3018,3019,3020,3021,3022,3023,3024,3025,3026,3027,3018,3029,
+                  3018,3019,3020,3021,3022,3023,3024,3025,3026,3027,3028,3029,
                   3030,3031,3032,3033,3034,3035,4000,4001,4002,4003,4004,4005,
                   4006,4007,4008,4009,4010,4011,4012,4013,4014,4015,4016,4017,
                   4018,4019,4020,4021,4022,4023,4024,4025,4026,4027,4028,4029,
@@ -277,7 +287,45 @@ def vat_wm_ref_region(input_segmentation_path: str,
     wm_csf_eroded = ants.threshold_image(image=wm_csf_blurred, low_thresh=0.95, binary=True)
     wm_erode = ants.mask_image(image=wm_merged, mask=wm_csf_eroded)
 
-    ants.image_write(image=wm_erode, filename=out_segmentation_path)
+    if out_segmentation_path is not None:
+        ants.image_write(image=wm_erode, filename=out_segmentation_path)
+
+    return wm_erode
+
+
+def eroded_wm_segmentation(input_segmentation_path: str,
+                           out_segmentation_path: str | None,
+                           eroded_wm_region_mapping: int = 1) -> ants.ANTsImage:
+    """
+    Generates eroded white matter region on a segmentation image and merges it into the image,
+    saving result as a new segmentation image.
+
+    Requires FreeSurfer segmentation with original label mappings.
+    
+    Args:
+        input_segmentation_path (str): Path to input freesurfer segmentation, such as aparc+aseg or
+            wmparc.
+        out_segmentation_path (str): Path to output segmentation image with replaced values in
+            eroded whited matter region.
+        eroded_wm_region_mapping (int): Segmentation mapping for the eroded white matter region in
+            the output image. Default "1".
+
+    Returns:
+        seg_img (ants.ANTsImage): Input segmentation image with values in eroded white matter
+            replaced with `eroded_wm_region_mapping`.
+
+    See also:
+        :meth:`~petpal.preproc.segmentation_tools.vat_wm_ref_region` - function that generates the
+            eroded white matter region.
+    """
+    wm_erode = vat_wm_ref_region(input_segmentation_path=input_segmentation_path,
+                                 out_segmentation_path=None)
+    seg_img = ants.image_read(input_segmentation_path)
+    seg_img[wm_erode==1] = int(eroded_wm_region_mapping)
+    if out_segmentation_path is not None:
+        ants.image_write(image=seg_img, filename=out_segmentation_path)
+
+    return seg_img
 
 
 def vat_wm_region_merge(wmparc_segmentation_path: str,
@@ -500,7 +548,8 @@ def calc_vesselness_mask_from_quantiled_vesselness(input_image: ants.core.ANTsIm
                                                    morph_dil_radius: int = 0,
                                                    z_crop: int = 3) -> ants.core.ANTsImage:
     """
-    Generates a binary vesselness mask from a given vesselness image using quantile-based thresholding.
+    Generates a binary vesselness mask from a given vesselness image using quantile-based
+    thresholding.
 
     This function creates a binary mask by thresholding a vesselness image at a specified
     quantile of non-zero voxel values. Additionally, it allows for optional z-axis cropping

{petpal-0.5.10 → petpal-0.6.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "petpal"
-version = "0.5.10"
+version = "0.6.0"
 description = "PET-PAL (Positron Emission Tomography Processing and Analysis Library)"
 authors = [
     {name = "Noah Goldman", email = "noahg@wustl.edu"},