Source code for dipy.workflows.utils

"""Module for utility functions."""

from ..utils.convert import expand_range




def handle_vol_idx(vol_idx):
    """Handle user input for volume index."""
    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