"""Converts slicer image and segmentation volumes into numpy format."""

def readImg(inputImg):

    """Reads an image file as input, returns the image as a numpy array."""

    import SimpleITK as sitk

    imgReader = sitk.ImageFileReader()

    imgReader.SetFileName(inputImg)

    image = imgReader.Execute()

    image = sitk.DICOMOrient(image, 'SPL')

    imgData = sitk.GetArrayFromImage(image)

    return imgData

def readMask(inputMask):

    """Reads an nrrd file as input, returns the image as a binary numpy array."""

    from nrrd import read

    maskReader = read(inputMask)

    maskHeader = maskReader[1:]

    maskData = maskReader[0].astype('bool')

    return maskHeader, maskData