Source code for dipy.workflows.utils

"""Module for utility functions."""

from dipy.utils.convert import expand_range




def handle_vol_idx(vol_idx):
    """Handle user input for volume index.

    Parameters
    ----------
    vol_idx : int, str, list, tuple
        Volume index or range of volume indices.

    Returns
    -------
    vol_idx : list
        List of volume indices.
    """
    if vol_idx is not None:
        if isinstance(vol_idx, str):
            vol_idx = expand_range(vol_idx)
        elif isinstance(vol_idx, int):
            vol_idx = [vol_idx]
        elif isinstance(vol_idx, (list, tuple)):
            vol_idx = [int(idx) for idx in vol_idx]
    return vol_idx