<reponame>djoerch/scilpy<gh_stars>0
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import nibabel as nib


def get_reference_info(reference):
    """Get basic information from a reference NIFTI file.
    :param reference: Nibabel.nifti or filepath (nii or nii.gz)
    """

    # TODO remove/replace after stateful_tractogram PR is merged
    nib_file = nib.load(reference)
    reference_shape = nib_file.get_shape()
    reference_affine = nib_file.affine

    return reference_shape, reference_affine
