dataLoader.py

import nibabel as nib





def niiLoader(nii_path):
    img = nib.load(str(nii_path))
    entry = {
        "data": img.get_fdata(),
        "affine": img.affine
    }
    return entry
def dataLoader(root_dir, pattern="*preproc-quasiraw_T1w.nii.gz", maxImg=float('inf')):
    results = []

    anat_dirs = root_dir.glob("sub-*/ses-*/anat")
    count = 0

    for anat_dir in anat_dirs:
        nii_files = list(anat_dir.glob(pattern))

        if not nii_files:
            continue

        for nii_path in nii_files:
            if count >= maxImg:
                return results

            parts = nii_path.name.split("_")
            sub_id = next((p for p in parts if p.startswith("sub-")), None)
            ses_id = next((p for p in parts if p.startswith("ses-")), None)

            entry = {
                "subject": sub_id,
                "session": ses_id,
                "filename": nii_path.name,
                "path": str(nii_path),
            }
            results.append(entry)
            count += 1

    return results