import nibabel as nib
from pathlib import Path
import numpy as np

def process(file):
    path = Path.cwd().joinpath(file)
    data = nib.load(path)
    print(data.affine)
    print(data.header.get_data_shape())
    print(np.dot(data.affine[:3, :3], data.header.get_data_shape()))


if __name__ == '__main__':
    process('../tests/zmaps/group_23_40_27_Z.nii.gz')
    process('../tests/data/template_T2w_brain.nii.gz')
    process('../tests/data/template_T2w_brain_150um.nii.gz')
    process('../tests/data/template_T2w_brain_100um.nii.gz')
    process('../tests/data/template_T2w_brain_75um.nii.gz')