junifer 0.0.5.dev110__py3-none-any.whl → 0.0.5.dev145__py3-none-any.whl

junifer/__init__.py

@@ -20,3 +20,20 @@ from . import (
     onthefly,
 )
 from ._version import __version__
+
+
+__all__ = [
+    "api",
+    "configs",
+    "data",
+    "datagrabber",
+    "datareader",
+    "markers",
+    "pipeline",
+    "preprocess",
+    "stats",
+    "storage",
+    "utils",
+    "external",
+    "onthefly",
+]

junifer/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.0.5.dev110'
-__version_tuple__ = version_tuple = (0, 0, 5, 'dev110')
+__version__ = version = '0.0.5.dev145'
+__version_tuple__ = version_tuple = (0, 0, 5, 'dev145')

junifer/api/__init__.py

@@ -7,3 +7,6 @@
 from . import decorators
 from .cli import cli
 from .functions import collect, queue, run
+
+
+__all__ = ["decorators", "cli", "collect", "queue", "run"]

junifer/api/decorators.py

@@ -10,6 +10,15 @@ from typing import Type
 from ..pipeline.registry import register
 
 
+__all__ = [
+    "register_datagrabber",
+    "register_datareader",
+    "register_preprocessor",
+    "register_marker",
+    "register_storage",
+]
+
+
 def register_datagrabber(klass: Type) -> Type:
     """Register DataGrabber.
 

junifer/api/functions.py

@@ -23,6 +23,9 @@ from .queue_context import GnuParallelLocalAdapter, HTCondorAdapter
 from .utils import yaml
 
 
+__all__ = ["run", "collect", "queue", "reset", "list_elements"]
+
+
 def _get_datagrabber(datagrabber_config: Dict) -> BaseDataGrabber:
     """Get DataGrabber.
 
@@ -92,7 +95,7 @@ def run(
     datagrabber : dict
         DataGrabber to use. Must have a key ``kind`` with the kind of
         DataGrabber to use. All other keys are passed to the DataGrabber
-        init function.
+        constructor.
     markers : list of dict
         List of markers to extract. Each marker is a dict with at least two
         keys: ``name`` and ``kind``. The ``name`` key is used to name the
@@ -102,11 +105,11 @@ def run(
     storage : dict
         Storage to use. Must have a key ``kind`` with the kind of
         storage to use. All other keys are passed to the storage
-        init function.
+        constructor.
     preprocessors : list of dict, optional
         List of preprocessors to use. Each preprocessor is a dict with at
         least a key ``kind`` specifying the preprocessor to use. All other keys
-        are passed to the preprocessor init function (default None).
+        are passed to the preprocessor constructor (default None).
     elements : str or tuple or list of str or tuple, optional
         Element(s) to process. Will be used to index the DataGrabber
         (default None).
@@ -188,7 +191,7 @@ def collect(storage: Dict) -> None:
     storage : dict
         Storage to use. Must have a key ``kind`` with the kind of
         storage to use. All other keys are passed to the storage
-        init function.
+        constructor.
 
     """
     storage_params = storage.copy()

junifer/api/parser.py

@@ -14,6 +14,9 @@ from ..utils.logging import logger, raise_error
 from .utils import yaml
 
 
+__all__ = ["parse_yaml"]
+
+
 def parse_yaml(filepath: Union[str, Path]) -> Dict:
     """Parse YAML.
 

junifer/api/queue_context/__init__.py

@@ -6,3 +6,6 @@
 from .queue_context_adapter import QueueContextAdapter
 from .htcondor_adapter import HTCondorAdapter
 from .gnu_parallel_local_adapter import GnuParallelLocalAdapter
+
+
+__all__ = ["QueueContextAdapter", "HTCondorAdapter", "GnuParallelLocalAdapter"]

junifer/api/res/afni/run_afni_docker.sh

@@ -31,7 +31,7 @@ if [ -d "${var}" ]; then
 done
 
 echo "Docker args: ${docker_args[*]}" >&2
-echo "Corrected args for afni: ${corrected_args[*]}" >&2
+echo "Corrected args for AFNI: ${corrected_args[*]}" >&2
 
 cwd=$(pwd)
 cmd="docker run --rm ${docker_args[*]} -v ${cwd}:${cwd} -w ${cwd} afni/afni_make_build ${corrected_args[*]}"

junifer/api/res/ants/run_ants_docker.sh

@@ -31,7 +31,7 @@ if [ -d "${var}" ]; then
 done
 
 echo "Docker args: ${docker_args[*]}" >&2
-echo "Corrected args for fsl: ${corrected_args[*]}" >&2
+echo "Corrected args for ANTs: ${corrected_args[*]}" >&2
 
 cwd=$(pwd)
 cmd="docker run --rm ${docker_args[*]} -v ${cwd}:${cwd} -w ${cwd} antsx/ants ${corrected_args[*]}"

junifer/api/res/fsl/run_fsl_docker.sh

@@ -31,7 +31,7 @@ if [ -d "${var}" ]; then
 done
 
 echo "Docker args: ${docker_args[*]}" >&2
-echo "Corrected args for fsl: ${corrected_args[*]}" >&2
+echo "Corrected args for FSL: ${corrected_args[*]}" >&2
 
 cwd=$(pwd)
 cmd="docker run --rm ${docker_args[*]} -v ${cwd}:${cwd} -w ${cwd} brainlife/fsl ${corrected_args[*]}"

junifer/api/tests/test_api_utils.py

@@ -45,6 +45,7 @@ def test_get_dependency_information_short() -> None:
         "httpx",
         "tqdm",
         "templateflow",
+        "lapy",
         "looseversion",
     ]
 
@@ -74,6 +75,7 @@ def test_get_dependency_information_long() -> None:
         "httpx",
         "tqdm",
         "templateflow",
+        "lapy",
     ]
     for key in dependency_list:
         assert key in dependency_information_keys

junifer/configs/juseless/__init__.py

@@ -4,3 +4,6 @@
 # License: AGPL
 
 from . import datagrabbers
+
+
+__all__ = ["datagrabbers"]

junifer/configs/juseless/datagrabbers/__init__.py

@@ -10,3 +10,12 @@ from .camcan_vbm import JuselessDataladCamCANVBM
 from .ixi_vbm import JuselessDataladIXIVBM
 from .ucla import JuselessUCLA
 from .ukb_vbm import JuselessDataladUKBVBM
+
+
+__all__ = [
+    "JuselessDataladAOMICID1000VBM",
+    "JuselessDataladCamCANVBM",
+    "JuselessDataladIXIVBM",
+    "JuselessUCLA",
+    "JuselessDataladUKBVBM",
+]

junifer/configs/juseless/datagrabbers/aomic_id1000_vbm.py

@@ -11,6 +11,9 @@ from ....api.decorators import register_datagrabber
 from ....datagrabber import PatternDataladDataGrabber
 
 
+__all__ = ["JuselessDataladAOMICID1000VBM"]
+
+
 @register_datagrabber
 class JuselessDataladAOMICID1000VBM(PatternDataladDataGrabber):
     """Concrete implementation for Juseless AOMIC ID1000 VBM data fetching.

junifer/configs/juseless/datagrabbers/camcan_vbm.py

@@ -12,6 +12,9 @@ from ....api.decorators import register_datagrabber
 from ....datagrabber import PatternDataladDataGrabber
 
 
+__all__ = ["JuselessDataladCamCANVBM"]
+
+
 @register_datagrabber
 class JuselessDataladCamCANVBM(PatternDataladDataGrabber):
     """Concrete implementation for Juseless CamCAN VBM data fetching.

junifer/configs/juseless/datagrabbers/ixi_vbm.py

@@ -13,6 +13,9 @@ from ....datagrabber import PatternDataladDataGrabber
 from ....utils import raise_error
 
 
+__all__ = ["JuselessDataladIXIVBM"]
+
+
 @register_datagrabber
 class JuselessDataladIXIVBM(PatternDataladDataGrabber):
     """Concrete implementation for Juseless IXI VBM data fetching.

junifer/configs/juseless/datagrabbers/ucla.py

@@ -12,6 +12,9 @@ from ....datagrabber import PatternDataGrabber
 from ....utils import raise_error
 
 
+__all__ = ["JuselessUCLA"]
+
+
 @register_datagrabber
 class JuselessUCLA(PatternDataGrabber):
     """Concrete implementation for Juseless UCLA data fetching.

junifer/configs/juseless/datagrabbers/ukb_vbm.py

@@ -12,6 +12,9 @@ from ....api.decorators import register_datagrabber
 from ....datagrabber import PatternDataladDataGrabber
 
 
+__all__ = ["JuselessDataladUKBVBM"]
+
+
 @register_datagrabber
 class JuselessDataladUKBVBM(PatternDataladDataGrabber):
     """Concrete implementation for Juseless UKB VBM data fetching.

junifer/data/__init__.py

@@ -28,3 +28,23 @@ from .masks import (
 from .template_spaces import get_template, get_xfm
 
 from . import utils
+
+
+__all__ = [
+    "list_coordinates",
+    "load_coordinates",
+    "register_coordinates",
+    "get_coordinates",
+    "list_parcellations",
+    "load_parcellation",
+    "register_parcellation",
+    "merge_parcellations",
+    "get_parcellation",
+    "list_masks",
+    "load_mask",
+    "register_mask",
+    "get_mask",
+    "get_template",
+    "get_xfm",
+    "utils",
+]

junifer/data/coordinates.py

@@ -16,6 +16,14 @@ from ..pipeline import WorkDirManager
 from ..utils import logger, raise_error, run_ext_cmd
 
 
+__all__ = [
+    "register_coordinates",
+    "list_coordinates",
+    "get_coordinates",
+    "load_coordinates",
+]
+
+
 # Path to the VOIs
 _vois_path = Path(__file__).parent / "VOIs"
 

junifer/data/masks.py

@@ -35,6 +35,16 @@ from .utils import closest_resolution
 if TYPE_CHECKING:
     from nibabel import Nifti1Image
 
+
+__all__ = [
+    "compute_brain_mask",
+    "register_mask",
+    "list_masks",
+    "get_mask",
+    "load_mask",
+]
+
+
 # Path to the masks
 _masks_path = Path(__file__).parent / "masks"
 

junifer/data/parcellations.py

@@ -30,6 +30,15 @@ if TYPE_CHECKING:
     from nibabel import Nifti1Image
 
 
+__all__ = [
+    "register_parcellation",
+    "list_parcellations",
+    "get_parcellation",
+    "load_parcellation",
+    "merge_parcellations",
+]
+
+
 # A dictionary containing all supported parcellations and their respective
 # valid parameters.
 

junifer/data/template_spaces.py

@@ -15,6 +15,9 @@ from ..utils import logger, raise_error
 from .utils import closest_resolution
 
 
+__all__ = ["get_xfm", "get_template"]
+
+
 def get_xfm(
     src: str, dst: str, xfms_dir: Union[str, Path, None] = None
 ) -> Path:  # pragma: no cover

junifer/data/tests/test_masks.py

@@ -350,7 +350,7 @@ def test_get_mask_errors() -> None:
         with pytest.raises(ValueError, match=r"callable params"):
             get_mask(masks={"GM_prob0.2": {"param": 1}}, target_data=vbm_gm)
 
-        # Pass only parametesr to the intersection function
+        # Pass only parameters to the intersection function
         with pytest.raises(
             ValueError, match=r" At least one mask is required."
         ):

junifer/data/utils.py

@@ -7,6 +7,9 @@ import numpy as np
 from ..utils.logging import logger
 
 
+__all__ = ["closest_resolution"]
+
+
 def closest_resolution(
     resolution: Optional[Union[float, int]],
     valid_resolution: Union[List[float], List[int], np.ndarray],

junifer/datagrabber/__init__.py

@@ -16,3 +16,18 @@ from .aomic import DataladAOMICID1000, DataladAOMICPIOP1, DataladAOMICPIOP2
 from .hcp1200 import HCP1200, DataladHCP1200
 from .multiple import MultipleDataGrabber
 from .dmcc13_benchmark import DMCC13Benchmark
+
+
+__all__ = [
+    "BaseDataGrabber",
+    "DataladDataGrabber",
+    "PatternDataGrabber",
+    "PatternDataladDataGrabber",
+    "DataladAOMICID1000",
+    "DataladAOMICPIOP1",
+    "DataladAOMICPIOP2",
+    "HCP1200",
+    "DataladHCP1200",
+    "MultipleDataGrabber",
+    "DMCC13Benchmark",
+]

junifer/datagrabber/aomic/__init__.py

@@ -7,3 +7,6 @@
 from .id1000 import DataladAOMICID1000
 from .piop1 import DataladAOMICPIOP1
 from .piop2 import DataladAOMICPIOP2
+
+
+__all__ = ["DataladAOMICID1000", "DataladAOMICPIOP1", "DataladAOMICPIOP2"]

junifer/datagrabber/aomic/id1000.py

@@ -14,6 +14,9 @@ from ...api.decorators import register_datagrabber
 from ..pattern_datalad import PatternDataladDataGrabber
 
 
+__all__ = ["DataladAOMICID1000"]
+
+
 @register_datagrabber
 class DataladAOMICID1000(PatternDataladDataGrabber):
     """Concrete implementation for datalad-based data fetching of AOMIC ID1000.

junifer/datagrabber/aomic/piop1.py

@@ -16,6 +16,9 @@ from ...utils import raise_error
 from ..pattern_datalad import PatternDataladDataGrabber
 
 
+__all__ = ["DataladAOMICPIOP1"]
+
+
 @register_datagrabber
 class DataladAOMICPIOP1(PatternDataladDataGrabber):
     """Concrete implementation for pattern-based data fetching of AOMIC PIOP1.

junifer/datagrabber/aomic/piop2.py

@@ -16,6 +16,9 @@ from ...utils import raise_error
 from ..pattern_datalad import PatternDataladDataGrabber
 
 
+__all__ = ["DataladAOMICPIOP2"]
+
+
 @register_datagrabber
 class DataladAOMICPIOP2(PatternDataladDataGrabber):
     """Concrete implementation for pattern-based data fetching of AOMIC PIOP2.

junifer/datagrabber/base.py

@@ -14,6 +14,9 @@ from ..utils import logger, raise_error
 from .utils import validate_types
 
 
+__all__ = ["BaseDataGrabber"]
+
+
 class BaseDataGrabber(ABC, UpdateMetaMixin):
     """Abstract base class for DataGrabber.
 

junifer/datagrabber/datalad_base.py

@@ -21,6 +21,9 @@ from ..utils import logger, raise_error, warn_with_log
 from .base import BaseDataGrabber
 
 
+__all__ = ["DataladDataGrabber"]
+
+
 class DataladDataGrabber(BaseDataGrabber):
     """Abstract base class for datalad-based data fetching.
 

junifer/datagrabber/hcp1200/__init__.py

@@ -5,3 +5,6 @@
 
 from .hcp1200 import HCP1200
 from .datalad_hcp1200 import DataladHCP1200
+
+
+__all__ = ["HCP1200", "DataladHCP1200"]

junifer/datagrabber/hcp1200/datalad_hcp1200.py

@@ -14,6 +14,9 @@ from ...api.decorators import register_datagrabber
 from .hcp1200 import HCP1200
 
 
+__all__ = ["DataladHCP1200"]
+
+
 @register_datagrabber
 class DataladHCP1200(DataladDataGrabber, HCP1200):
     """Concrete implementation for datalad-based data fetching of HCP1200.

junifer/datagrabber/hcp1200/hcp1200.py

@@ -14,6 +14,9 @@ from ..pattern import PatternDataGrabber
 from ..utils import raise_error
 
 
+__all__ = ["HCP1200"]
+
+
 @register_datagrabber
 class HCP1200(PatternDataGrabber):
     """Concrete implementation for pattern-based data fetching of HCP1200.

junifer/datagrabber/multiple.py

@@ -11,6 +11,9 @@ from ..utils import raise_error
 from .base import BaseDataGrabber
 
 
+__all__ = ["MultipleDataGrabber"]
+
+
 class MultipleDataGrabber(BaseDataGrabber):
     """Concrete implementation for multi sourced data fetching.
 

junifer/datagrabber/pattern.py

@@ -6,6 +6,7 @@
 # License: AGPL
 
 import re
+from copy import deepcopy
 from pathlib import Path
 from typing import Dict, List, Optional, Tuple, Union
 
@@ -17,6 +18,9 @@ from .base import BaseDataGrabber
 from .utils import validate_patterns, validate_replacements
 
 
+__all__ = ["PatternDataGrabber"]
+
+
 # Accepted formats for confounds specification
 _CONFOUNDS_FORMATS = ("fmriprep", "adhoc")
 
@@ -364,7 +368,7 @@ class PatternDataGrabber(BaseDataGrabber):
             # Data type dictionary
             t_pattern = self.patterns[t_type]
             # Copy data type dictionary in output
-            out[t_type] = t_pattern.copy()
+            out[t_type] = deepcopy(t_pattern)
             # Iterate to check for nested "types" like mask
             for k, v in t_pattern.items():
                 # Resolve pattern for base data type

junifer/datagrabber/pattern_datalad.py

@@ -12,6 +12,9 @@ from .datalad_base import DataladDataGrabber
 from .pattern import PatternDataGrabber
 
 
+__all__ = ["PatternDataladDataGrabber"]
+
+
 @register_datagrabber
 class PatternDataladDataGrabber(DataladDataGrabber, PatternDataGrabber):
     """Concrete implementation for pattern and datalad based data fetching.

junifer/datagrabber/utils.py

@@ -9,6 +9,9 @@ from typing import Dict, List
 from ..utils import logger, raise_error
 
 
+__all__ = ["validate_types", "validate_replacements", "validate_patterns"]
+
+
 # Define schema for pattern-based datagrabber's patterns
 PATTERNS_SCHEMA = {
     "T1w": {

junifer/datareader/__init__.py

@@ -6,3 +6,6 @@
 # License: AGPL
 
 from .default import DefaultDataReader
+
+
+__all__ = ["DefaultDataReader"]

junifer/datareader/default.py

@@ -15,6 +15,9 @@ from ..pipeline import PipelineStepMixin, UpdateMetaMixin
 from ..utils.logging import logger, warn_with_log
 
 
+__all__ = ["DefaultDataReader"]
+
+
 # Map each file extension to a type
 _extensions = {
     ".nii": "NIFTI",

junifer/external/BrainPrint/brainprint/__init__.py

@@ -0,0 +1,4 @@
+from ._version import __version__  # noqa: F401
+from .brainprint import compute_asymmetry, compute_brainprint  # noqa: F401
+from .surfaces import surf_to_vtk  # noqa: F401
+from .utils._config import sys_info  # noqa: F401

junifer/external/BrainPrint/brainprint/_version.py

@@ -0,0 +1,3 @@
+"""Version number."""
+
+__version__ = "0.4.0"

junifer/external/BrainPrint/brainprint/asymmetry.py

@@ -0,0 +1,91 @@
+"""
+Contains asymmetry estimation functionality.
+"""
+from typing import Dict
+
+import numpy as np
+from lapy import shapedna
+
+
+def compute_asymmetry(
+    eigenvalues, distance: str = "euc", skip_cortex: bool = False
+) -> Dict[str, float]:
+    """
+    Compute lateral shape distances from BrainPrint analysis results.
+
+    Parameters
+    ----------
+    eigenvalues : _type_
+        BrainPrint analysis results.
+    distance : str, optional
+        ShapeDNA distance, by default "euc".
+    skip_cortex : bool, optional
+        Whether to skip white matter and pial surfaces, by default False.
+
+    Returns
+    -------
+    Dict[str, float]
+        {left_label}_{right_label}, distance.
+    """
+    # Define structures
+
+    # combined and individual aseg labels:
+    # - Left  Striatum: left  Caudate + Putamen + Accumbens
+    # - Right Striatum: right Caudate + Putamen + Accumbens
+    # - CorpusCallosum: 5 subregions combined
+    # - Cerebellum: brainstem + (left+right) cerebellum WM and GM
+    # - Ventricles: (left+right) lat.vent + inf.lat.vent + choroidplexus + 3rdVent + CSF
+    # - Lateral-Ventricle: lat.vent + inf.lat.vent + choroidplexus
+    # - 3rd-Ventricle: 3rd-Ventricle + CSF
+
+    structures_left_right = [
+        ("Left-Striatum", "Right-Striatum"),
+        ("Left-Lateral-Ventricle", "Right-Lateral-Ventricle"),
+        (
+            "Left-Cerebellum-White-Matter",
+            "Right-Cerebellum-White-Matter",
+        ),
+        ("Left-Cerebellum-Cortex", "Right-Cerebellum-Cortex"),
+        ("Left-Thalamus-Proper", "Right-Thalamus-Proper"),
+        ("Left-Caudate", "Right-Caudate"),
+        ("Left-Putamen", "Right-Putamen"),
+        ("Left-Pallidum", "Right-Pallidum"),
+        ("Left-Hippocampus", "Right-Hippocampus"),
+        ("Left-Amygdala", "Right-Amygdala"),
+        ("Left-Accumbens-area", "Right-Accumbens-area"),
+        ("Left-VentralDC", "Right-VentralDC"),
+    ]
+
+    cortex_2d_left_right = [
+        ("lh-white-2d", "rh-white-2d"),
+        ("lh-pial-2d", "rh-pial-2d"),
+    ]
+
+    structures = structures_left_right
+    if not skip_cortex:
+        structures += cortex_2d_left_right
+
+    distances = dict()
+    for left_label, right_label in structures:
+        left_eigenvalues, right_eigenvalues = (
+            eigenvalues[left_label][2:],
+            eigenvalues[right_label][2:],
+        )
+        has_nan = np.isnan(left_eigenvalues).any() or np.isnan(right_eigenvalues).any()
+        key = f"{left_label}_{right_label}"
+        if has_nan:
+            message = (
+                "NaNs found for {left_label} or {right_label}, "
+                "skipping asymmetry computation...".format(
+                    left_label=left_label, right_label=right_label
+                )
+            )
+            print(message)
+            distances[key] = np.nan
+        else:
+            distances[key] = shapedna.compute_distance(
+                left_eigenvalues,
+                right_eigenvalues,
+                dist=distance,
+            )
+    return distances