from fetal.preprocess import window_intensities

from glob import glob

from os import path

from shutil import copyfile

from pathlib import Path

src_dir = '../Datasets/brain_ax'

dst_dir = '../Datasets/brain_ax_window_1_99'

Path(dst_dir).mkdir(exist_ok=True, parents=False)

ext = '.gz'

for p in glob(path.join(src_dir, '*')):

    subject_dir = Path(p).name

    Path(path.join(dst_dir, subject_dir)).mkdir(exist_ok=True)

    window_intensities(in_file=path.join(src_dir, subject_dir, 'volume.nii' + ext),

                       out_file=path.join(dst_dir, subject_dir, 'volume.nii' + ext),

                       max_percent=99)

    copyfile(src=path.join(src_dir, subject_dir, 'truth.nii' + ext),

             dst=path.join(dst_dir, subject_dir, 'truth.nii' + ext))

    print(subject_dir)