allow for default list of segs for a template

Author: akhanf

spimquant/config/snakebids.yml

@@ -275,6 +275,11 @@ templates:
       YoPro: https://zenodo.org/records/18749025/files/tpl-ABAv3_level-5_stain-YoPro_SPIM.nii.gz
       Iba1: https://zenodo.org/records/18749025/files/tpl-ABAv3_level-5_stain-Iba1_SPIM.nii.gz
       Abeta: https://zenodo.org/records/18749025/files/tpl-ABAv3_level-5_stain-Abeta_SPIM.nii.gz
+    default_segs:
+      - all
+      - coarse
+      - mid
+      - fine
     atlases:
       all: 
         dseg: https://zenodo.org/records/18906782/files/tpl-ABAv3_seg-all_dseg.nii.gz

spimquant/workflow/Snakefile

@@ -55,10 +55,13 @@ stains = get_stains_all_subjects()
 
 stain_for_reg = None
 
+template = config["templates"][config["template"]]
+
+
 # first, check if there are any SPIM templates defined
 # for the stains we have
 use_spim_template = False
-spim_templates = config["templates"][config["template"]].get("spim_templates", None)
+spim_templates = template.get("spim_templates", None)
 
 if spim_templates is not None:
     for stain in spim_templates.keys():
@@ -103,36 +106,35 @@ else:
 
 
 # atlas segmentations to use
-all_atlas_segs = config["templates"][config["template"]]["atlases"].keys()
 
 if config["atlas_segs"] is None:
-    atlas_segs = all_atlas_segs
+    atlas_segs = template.get("default_segs", template["atlases"].keys())
 else:
     atlas_segs = []
     for seg in config["atlas_segs"]:
-        if seg not in all_atlas_segs:
+        if seg not in template["atlases"]:
             raise ValueError(
-                f"Chosen segmentation {seg} was not found in the template {config['template']}"
+                f"Chosen segmentation {seg} was not found in the template {template}"
             )
         else:
             atlas_segs.append(seg)
 
 # atlas segmentations to use for patches (defaults to roi22)
 patch_atlas_segs = []
 for seg in config["patch_atlas_segs"]:
-    if seg not in all_atlas_segs:
+    if seg not in template["atlases"]:
         raise ValueError(
-            f"Chosen patch segmentation {seg} was not found in the template {config['template']}"
+            f"Chosen patch segmentation {seg} was not found in the template {template}"
         )
     else:
         patch_atlas_segs.append(seg)
 
 # atlas segmentations to use for Imaris crops (defaults to roi22)
 crop_atlas_segs = []
 for seg in config["crop_atlas_segs"]:
-    if seg not in all_atlas_segs:
+    if seg not in template["atlases"]:
         raise ValueError(
-            f"Chosen crop segmentation {seg} was not found in the template {config['template']}"
+            f"Chosen crop segmentation {seg} was not found in the template {template}"
         )
     else:
         crop_atlas_segs.append(seg)