torchio.cli.print_info.main()

Complexity

Conditions

5

Size

Total Lines	52
Code Lines	37

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

# pylint: disable=import-outside-toplevel
@app.command()
def main(
    input_path: Path = typer.Argument(  # noqa: B008
        ...,
        exists=True,
        file_okay=True,
        dir_okay=True,
        readable=True,
    ),
    plot: bool = typer.Option(  # noqa: B008
        False,
        '--plot/--no-plot',
        '-p/-P',
        help='Plot the image using Matplotlib or Pillow.',
    ),
    show: bool = typer.Option(  # noqa: B008
        False,
        '--show/--no-show',
        '-s/-S',
        help='Show the image using specialized visualisation software.',
    ),
    label: bool = typer.Option(  # noqa: B008
        False,
        '--label/--scalar',
        '-l/-s',
        help='Use torchio.LabelMap to instantiate the image.',
    ),
    load: bool = typer.Option(  # noqa: B008
        False,
        help=(
            'Load the image from disk so that information about data type and memory '
            'can be displayed. Slower, especially for large images.'
        ),
    ),
) -> None:
    """Print information about an image and, optionally, show it.

    Example:
    $ tiohd input.nii.gz
    """
    # Imports are placed here so that the tool loads faster if not being run
    import torchio as tio

    class_ = tio.LabelMap if label else tio.ScalarImage
    image = class_(input_path)
    if load:
        image.load()
    print(image)  # noqa: T201
    if plot:
        image.plot()
    if show:
        image.show()