LOAMRI
diff --git a/‎.bumpversion.cfg‎
Lines changed: 9 additions & 0 deletions b/‎.bumpversion.cfg‎
Lines changed: 9 additions & 0 deletions
diff --git a/‎.github/workflows/bumpversion_publish_workflow.yaml‎
Lines changed: 57 additions & 0 deletions b/‎.github/workflows/bumpversion_publish_workflow.yaml‎
Lines changed: 57 additions & 0 deletions
diff --git a/‎asltk/registration/__init__.py‎
Lines changed: 15 additions & 0 deletions b/‎asltk/registration/__init__.py‎
Lines changed: 15 additions & 0 deletions
diff --git a/‎asltk/registration/asl_normalization.py‎
Lines changed: 108 additions & 0 deletions b/‎asltk/registration/asl_normalization.py‎
Lines changed: 108 additions & 0 deletions
@@ -0,0 +1,9 @@
+[bumpversion]
+current_version = 0.6.0
+commit = True
+tag = True
+tag_name = v{new_version}
+
+[bumpversion:file:pyproject.toml]
+search = version = "{current_version}"
+replace = version = "{new_version}"
@@ -0,0 +1,57 @@
+name: Release & Publish to PyPI
+
+on:
+  workflow_dispatch:  # Manual trigger only
+    inputs:
+      bump_type:
+        description: "Version bump type (patch, minor, major)"
+        required: true
+        default: "patch"
+
+jobs:
+  release:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Validate bump_type input
+        run: |
+          if [[ ! "${{ github.event.inputs.bump_type }}" =~ ^(patch|minor|major)$ ]]; then
+            echo "Invalid bump_type: ${{ github.event.inputs.bump_type }}. Must be one of: patch, minor, major."
+            exit 1
+          fi
+
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.9'
+
+      - name: Install Poetry
+        run: pip install poetry
+
+      - name: Install bumpversion
+        run: pip install bump2version
+
+      - name: Install project dependencies
+        run: poetry install
+
+      - name: Bump version
+        run: |
+          bump2version ${{ github.event.inputs.bump_type }}
+
+      - name: Push version bump and tags to repo
+        run: |
+            git config user.name "github-actions[bot]"
+            git config user.email "github-actions[bot]@users.noreply.github.com"
+            git push
+            git push --tags
+
+      - name: Build package
+        run: poetry build
+
+      - name: Publish to PyPI
+        env:
+          POETRY_PYPI_TOKEN_PYPI: ${{ secrets.PYPI_TOKEN }}
+        run: poetry publish
@@ -21,6 +21,21 @@ def __init__(self):
         pass
 
 
+# TODO Montar classe para fazer o coregistro de ASL
+class ASLRegistration:
+
+    # Pipeline
+    # inputs: ASLData (com m0 e pcasl), BrainAtlas, resolution (1 or 2 mm)
+    # Tomar m0 e comparar orientação com o template
+    # Se necessário, corrigir orientação do template para estar coerente com o m0 (salvar a transformação e aplicar para os labels)
+    # Realizar o registro do m0 no template
+    # com a transformação do m0, deixar salvo como parametro do objeto da classe
+    # Ter metodos para aplicar transformação para o pcasl, ou mapas gerados pelo CBFMapping, MultiTE, etc.
+
+    def __init__(self):
+        pass
+
+
 def space_normalization(
     moving_image: np.ndarray,
     template_image: BrainAtlas,
 
@@ -127,6 +127,114 @@ def norm_function(vol, _):
     return new_asl, trans_m0_mtx
 
 
+def asl_template_registration(
+    asl_data: ASLData,
+    asl_data_mask: np.ndarray = None,
+    atlas_name: str = 'MNI2009',
+    verbose: bool = False,
+):
+    """
+    Register ASL data to common atlas space.
+
+    This function applies a elastic normalization to fit the subject head
+    space into the atlas template space.
+
+
+    Note:
+        This method takes in consideration the ASLData object, which contains
+        the pcasl and/or m0 image. The registration is performed using primarily
+        the `m0`image if available, otherwise it uses the `pcasl` image.
+        Therefore, choose wisely the `ref_vol` parameter, which should be a valid index
+        for the best `pcasl`volume reference to be registered to the atlas.
+
+    Args:
+        asl_data: ASLData
+            The ASLData object containing the pcasl and/or m0 image to be corrected.
+        ref_vol: (int, optional)
+            The index of the reference volume to which all other volumes will be registered.
+            Defaults to 0.
+        asl_data_mask: np.ndarray
+            A single volume image mask. This can assist the normalization method to converge
+            into the atlas space. If not provided, the full image is adopted.
+        atlas_name: str
+            The atlas type to be considered. The BrainAtlas class is applied, then choose
+            the `atlas_name` based on the ASLtk brain atlas list.
+        verbose: (bool, optional)
+            If True, prints progress messages. Defaults to False.
+
+    Raises:
+        TypeError: If the input is not an ASLData object.
+        ValueError: If ref_vol is not a valid index.
+        RuntimeError: If an error occurs during registration.
+
+    Returns:
+        tuple: ASLData object with corrected volumes and a list of transformation matrices.
+    """
+    if not isinstance(asl_data, ASLData):
+        raise TypeError('Input must be an ASLData object.')
+
+    # if not isinstance(ref_vol, int) or ref_vol < 0:
+    #     raise ValueError('ref_vol must be a non-negative integer.')
+
+    total_vols, orig_shape = collect_data_volumes(asl_data('pcasl'))
+    # if ref_vol >= len(total_vols):
+    #     raise ValueError(
+    #         'ref_vol must be a valid index based on the total ASL data volumes.'
+    #     )
+
+    if asl_data('m0') is None:
+        raise ValueError(
+            'M0 image is required for normalization. Please provide an ASLData with a valid M0 image.'
+        )
+
+    atlas = BrainAtlas(atlas_name)
+    # atlas_img = ants.image_read(atlas.get_atlas()['t1_data']).numpy()
+    atlas_img = load_image(atlas.get_atlas()['t1_data'])
+
+    def norm_function(vol, _):
+        return space_normalization(
+            moving_image=vol,
+            template_image=atlas,
+            moving_mask=asl_data_mask,
+            template_mask=None,
+            transform_type='Affine',
+            check_orientation=True,
+            orientation_verbose=verbose,
+        )
+
+    # Create a new ASLData to allocate the normalized image
+    new_asl = asl_data.copy()
+
+    tmp_vol_list = [asl_data('m0')]
+    orig_shape = asl_data('m0').shape
+
+    m0_vol_corrected, trans_m0_mtx = __apply_array_normalization(
+        tmp_vol_list, 0, orig_shape, norm_function, verbose
+    )
+    new_asl.set_image(m0_vol_corrected[0], 'm0')
+
+    # Apply the normalization transformation to all pcasl volumes
+    pcasl_vols, _ = collect_data_volumes(asl_data('pcasl'))
+    normalized_pcasl_vols = []
+    with Progress() as progress:
+        task = progress.add_task(
+            '[green]Applying normalization to pcasl volumes...',
+            total=len(pcasl_vols),
+        )
+        for vol in pcasl_vols:
+            norm_vol = apply_transformation(
+                moving_image=vol,
+                reference_image=atlas_img,
+                transforms=trans_m0_mtx,
+            )
+            normalized_pcasl_vols.append(norm_vol)
+            progress.update(task, advance=1)
+
+    new_asl.set_image(normalized_pcasl_vols, 'pcasl')
+
+    return new_asl, trans_m0_mtx
+
+
 def head_movement_correction(
     asl_data: ASLData,
     ref_vol: np.ndarray = None,