change preprocessing config

aramis-lab · Oct 23, 2024 · 311cbed · 311cbed
1 parent 580f249
commit 311cbed
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/clinicadl/caps_dataset/preprocessing/config.py b/clinicadl/caps_dataset/preprocessing/config.py
@@ -8,6 +8,7 @@
 from clinicadl.utils.enum import (
     DTIMeasure,
     DTISpace,
+    ImageModality,
     LinearModality,
     Preprocessing,
     SUVRReferenceRegions,
@@ -100,6 +101,9 @@ def bids_nii(self, reconstruction: Optional[str] = None) -> FileType:
         )
         return file_type
 
+    def caps_nii(self) -> tuple:
+        return (self.preprocessing, ImageModality.PET)
+
     def get_filetype(self) -> FileType:
         if self.use_uncropped_image:
             description = ""
@@ -124,6 +128,9 @@ def bids_nii(self, reconstruction: Optional[str] = None) -> FileType:
             description="Custom suffix",
         )
 
+    def caps_nii(self) -> tuple:
+        return (self.preprocessing, ImageModality.CUSTOM)
+
     def get_filetype(self) -> FileType:
         return self.bids_nii()
 
@@ -136,6 +143,9 @@ class DTIPreprocessingConfig(PreprocessingConfig):
     def bids_nii(self, reconstruction: Optional[str] = None) -> FileType:
         return FileType(pattern="dwi/sub-*_ses-*_dwi.nii*", description="DWI NIfTI")
 
+    def caps_nii(self) -> tuple:
+        return (self.preprocessing, ImageModality.DWI)
+
     def get_filetype(self) -> FileType:
         """Return the query dict required to capture DWI DTI images.