junifer 0.0.6.dev3__py3-none-any.whl → 0.0.6.dev11__py3-none-any.whl

junifer/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.0.6.dev3'
-__version_tuple__ = version_tuple = (0, 0, 6, 'dev3')
+__version__ = version = '0.0.6.dev11'
+__version_tuple__ = version_tuple = (0, 0, 6, 'dev11')

junifer/onthefly/__init__.py

@@ -4,6 +4,7 @@
 # License: AGPL
 
 from .read_transform import read_transform
+from . import _brainprint as brainprint
 
 
-__all__ = ["read_transform"]
+__all__ = ["read_transform", "brainprint"]

junifer/onthefly/_brainprint.py

@@ -0,0 +1,141 @@
+"""Provide onthefly functions for BrainPrint post-analysis."""
+
+# Authors: Synchon Mandal <s.mandal@fz-juelich.de>
+# License: AGPL
+
+from typing import TYPE_CHECKING, Dict, Optional, Type
+
+import numpy as np
+import pandas as pd
+
+from ..utils import raise_error
+
+
+if TYPE_CHECKING:
+    from junifer.storage import BaseFeatureStorage
+
+
+__all__ = ["normalize", "reweight"]
+
+
+def normalize(
+    storage: Type["BaseFeatureStorage"],
+    features: Dict[str, Dict[str, Optional[str]]],
+    kind: str,
+) -> pd.DataFrame:
+    """Read stored brainprint data and normalize either surfaces or volumes.
+
+    Parameters
+    ----------
+    storage : storage-like
+        The storage class, for example, :class:`.HDF5FeatureStorage`.
+    features : dict, optional
+        The feature names or MD5 hashes to read as dict.
+        The dict should have the keys:
+
+        * ``"areas"`` (if ``kind="surface"``)
+        * ``"volumes"`` (if ``kind="volume"``)
+        * ``"eigenvalues"``
+
+        and the corresponding value for each of the keys is again
+        a dict with the keys:
+
+        * ``"feature_name"`` : str or None
+        * ``"feature_md5"`` : str or None
+
+        Either one of ``"feature_name"`` or ``"feature_md5"`` needs to be
+        not None for each first-level key, but both keys are mandatory.
+
+    kind : {"surface", "volume"}
+        The kind of normalization.
+
+    Returns
+    -------
+    pandas.DataFrame
+        The transformed feature as a ``pandas.DataFrame``.
+
+    Raises
+    ------
+    ValueError
+        If ``kind`` is invalid.
+
+    """
+    # Read storage
+    data_dict = {}
+    for k, v in features.items():
+        data_dict[k] = storage.read_df(**v)  # type: ignore
+
+    # Check and normalize
+    valid_kind = ["surface", "volume"]
+    normalized_df = None
+    if kind == "surface":
+        eigenvalues_df = data_dict["eigenvalues"]
+        areas_df = data_dict["areas"]
+        normalized_df = eigenvalues_df.combine(
+            areas_df, lambda left, right: left * right
+        )
+    elif kind == "volume":
+        eigenvalues_df = data_dict["eigenvalues"]
+        volumes_df = data_dict["volumes"]
+        normalized_df = eigenvalues_df.combine(
+            volumes_df, lambda left, right: left * right ** np.divide(2.0, 3.0)
+        )
+    else:
+        raise_error(
+            "Invalid value for `kind`, should be one of: " f"{valid_kind}"
+        )
+
+    return normalized_df
+
+
+def reweight(
+    storage: Type["BaseFeatureStorage"],
+    feature_name: Optional[str] = None,
+    feature_md5: Optional[str] = None,
+) -> pd.DataFrame:
+    """Read stored brainprint data and reweight eigenvalues.
+
+    Parameters
+    ----------
+    storage : storage-like
+        The storage class, for example, :class:`.HDF5FeatureStorage`.
+    feature_name : str, optional
+        Name of the feature to read (default None).
+    feature_md5 : str, optional
+        MD5 hash of the feature to read (default None).
+
+    Returns
+    -------
+    pandas.DataFrame
+        The transformed feature as a ``pandas.DataFrame``.
+
+    """
+    # Read storage
+    eigenvalues_df = storage.read_df(
+        feature_name=feature_name, feature_md5=feature_md5
+    )  # type: ignore
+
+    # Create data for operation
+    exploded_count_idx_df = (
+        eigenvalues_df.reset_index("eigenvalue")
+        .index.value_counts()
+        .apply(lambda x: np.arange(1, x + 1).astype(float))
+        .to_frame()
+        .explode("count")
+    )
+    idx_data = (
+        pd.concat(
+            [exploded_count_idx_df] * len(eigenvalues_df.columns), axis=1
+        )
+        .reset_index()
+        .drop("subject", axis=1, inplace=False)
+        .to_numpy()
+    )
+    idx_df = pd.DataFrame(
+        data=idx_data,
+        index=eigenvalues_df.index,
+        columns=eigenvalues_df.columns,
+    )
+
+    # Combine
+    return eigenvalues_df.combine(idx_df, lambda left, right: left / right)

junifer/utils/helpers.py

@@ -5,6 +5,7 @@
 
 import collections.abc
 import subprocess
+import sys
 from typing import Dict, List
 
 from .logging import logger, raise_error
@@ -45,13 +46,13 @@ def run_ext_cmd(name: str, cmd: List[str]) -> None:
     if process.returncode == 0:
         logger.info(
             f"{name} command succeeded with the following output:\n"
-            f"{process.stdout}"
+            f"{process.stdout.decode(sys.stdout.encoding)}"
         )
     else:
         raise_error(
             msg=(
                 f"{name} command failed with the following error:\n"
-                f"{process.stdout}"
+                f"{process.stdout.decode(sys.stdout.encoding)}"
             ),
             klass=RuntimeError,
         )

{junifer-0.0.6.dev3.dist-info → junifer-0.0.6.dev11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: junifer
-Version: 0.0.6.dev3
+Version: 0.0.6.dev11
 Summary: JUelich NeuroImaging FEature extractoR
 Author-email: Fede Raimondo <f.raimondo@fz-juelich.de>, Synchon Mandal <s.mandal@fz-juelich.de>
 Maintainer-email: Fede Raimondo <f.raimondo@fz-juelich.de>, Synchon Mandal <s.mandal@fz-juelich.de>

{junifer-0.0.6.dev3.dist-info → junifer-0.0.6.dev11.dist-info}/RECORD

@@ -1,5 +1,5 @@
 junifer/__init__.py,sha256=-T9XmiCCL0j3YLx-0Pph15sPfL5FlcBDscajjJ-V4sU,604
-junifer/_version.py,sha256=6OixPFZ1NORJup9F0nlIFHM0I_Gk0_4c-kcWUt3niik,424
+junifer/_version.py,sha256=mU31Rbo9393TgDvfpsOHnq8SiUzPPQ4wTt9IrlhivXo,426
 junifer/stats.py,sha256=BjQb2lfTGDP9l4UuQYmJFcJJNRfbJDGlNvC06SJaDDE,6237
 junifer/api/__init__.py,sha256=lwyIF0hPc7fICuSoddJfay0LPqlTRxHJ_xbtizgFYZA,312
 junifer/api/cli.py,sha256=53pews3mXkJ7DUDSkV51PbitYnuVAdQRkWG-gjO08Uw,16142
@@ -206,7 +206,8 @@ junifer/markers/tests/test_marker_utils.py,sha256=SR3ADWI3uGv4ozYqVu-rMZnJVqP6Jn
 junifer/markers/tests/test_markers_base.py,sha256=XYe1Z_88h2g1WX6Em4aM8VMyBuCpy5sNHbbpC0I89m4,2979
 junifer/markers/tests/test_parcel_aggregation.py,sha256=FkB0O0HjTk1CnLOn-dzNs_9_byUOISRc4jV92shN2Kc,27655
 junifer/markers/tests/test_sphere_aggregation.py,sha256=TGn5S7zKK0SJ4nHIxRZQSCpqRBmz7c8Sb8C79dLoHjE,10611
-junifer/onthefly/__init__.py,sha256=WykjD_q_zWyX40Y5iKsFXhPJLzD1NPcN-XKFKZDMXfc,184
+junifer/onthefly/__init__.py,sha256=TA6tPuw54ynDlumb9Ii-2p59hw2rGoCMe1-vQ89JzZ8,238
+junifer/onthefly/_brainprint.py,sha256=-q5j5xOkuZD_f-pjWi-b8VRqM9ZXDk1TnccuQDfLwz4,3860
 junifer/onthefly/read_transform.py,sha256=kZ-N_fKe9glaBTjhj_HXrdTtWXGI-fMoBpsawcOgsTw,4340
 junifer/onthefly/tests/test_read_transform.py,sha256=D2C3IpXQHdsJSF07v8rEwGntLGXjZOserlRhebJUAVM,4719
 junifer/pipeline/__init__.py,sha256=mmhtYuRGetBmBvKPVb9MxklcEK_R9ly-dgkzQiOp3g8,363
@@ -261,15 +262,15 @@ junifer/tests/test_main.py,sha256=GMff7jlisGM9_FsiUwWDte43j-KQJGFRYZpwRRqTkd8,37
 junifer/tests/test_stats.py,sha256=3vPMgYYpWxk8ECDFOMm3-dFBlh4XxjL83SwRBSBAHok,4155
 junifer/utils/__init__.py,sha256=F_I7WXtZMrBGGNLN09LvzBRwWKopL2k1z0UgCZvpwj0,471
 junifer/utils/fs.py,sha256=M3CKBLh4gPS6s9giyopgb1hHMXzLb6k3cung2wHVBjs,492
-junifer/utils/helpers.py,sha256=D17zdq1y92fOc-5nwnhaXfgbk8o5yHocLSct9E05YUk,1996
+junifer/utils/helpers.py,sha256=_IqnaPaOcFy1yrEyNmmg7XqQWb1wHOtxfOBnlaRYbiI,2063
 junifer/utils/logging.py,sha256=ardaiJkDfZMYvak5UIL5Etxg5Ii7inmVQSBdFLdgtb8,9781
 junifer/utils/tests/test_fs.py,sha256=WQS7cKlKEZ742CIuiOYYpueeAhY9PqlastfDVpVVtvE,923
 junifer/utils/tests/test_helpers.py,sha256=k5qqfxK8dFyuewTJyR1Qn6-nFaYNuVr0ysc18bfPjyU,929
 junifer/utils/tests/test_logging.py,sha256=duO4ou365hxwa_kwihFtKPLaL6LC5XHiyhOijrrngbA,8009
-junifer-0.0.6.dev3.dist-info/AUTHORS.rst,sha256=rmULKpchpSol4ExWFdm-qu4fkpSZPYqIESVJBZtGb6E,163
-junifer-0.0.6.dev3.dist-info/LICENSE.md,sha256=MqCnOBu8uXsEOzRZWh9EBVfVz-kE9NkXcLCrtGXo2yU,34354
-junifer-0.0.6.dev3.dist-info/METADATA,sha256=SQr23fntAuaqLWbOGxBRZLjL_GBi2Zwf5mQ84t9DaJc,8278
-junifer-0.0.6.dev3.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-junifer-0.0.6.dev3.dist-info/entry_points.txt,sha256=DxFvKq0pOqRunAK0FxwJcoDfV1-dZvsFDpD5HRqSDhw,48
-junifer-0.0.6.dev3.dist-info/top_level.txt,sha256=4bAq1R2QFQ4b3hohjys2JBvxrl0GKk5LNFzYvz9VGcA,8
-junifer-0.0.6.dev3.dist-info/RECORD,,
+junifer-0.0.6.dev11.dist-info/AUTHORS.rst,sha256=rmULKpchpSol4ExWFdm-qu4fkpSZPYqIESVJBZtGb6E,163
+junifer-0.0.6.dev11.dist-info/LICENSE.md,sha256=MqCnOBu8uXsEOzRZWh9EBVfVz-kE9NkXcLCrtGXo2yU,34354
+junifer-0.0.6.dev11.dist-info/METADATA,sha256=JERhQXlkyOo07_IQ5HaJksc0Eg9v9IWV45ZyodsOxmU,8279
+junifer-0.0.6.dev11.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+junifer-0.0.6.dev11.dist-info/entry_points.txt,sha256=DxFvKq0pOqRunAK0FxwJcoDfV1-dZvsFDpD5HRqSDhw,48
+junifer-0.0.6.dev11.dist-info/top_level.txt,sha256=4bAq1R2QFQ4b3hohjys2JBvxrl0GKk5LNFzYvz9VGcA,8
+junifer-0.0.6.dev11.dist-info/RECORD,,