import os
import glob
import shutil
from tqdm import *
from subprocess import check_call


def create_dir(path):
    if not os.path.isdir(path):
        os.makedirs(path)
    return


src_dir_list = ["ADNI1_Screening_AD", "ADNI1_Screening_NC",
                "ADNI2_Screening_AD", "ADNI2_Screening_NC"]
dst_dir_list = ["AD", "NC", "AD", "NC"]

parent_dir = os.path.dirname(os.getcwd())
data_dir = os.path.join(parent_dir, "data")

for in_dir, out_dir in zip(src_dir_list, dst_dir_list):
    data_src_dir = os.path.join(data_dir, in_dir)
    data_dst_dir = os.path.join(data_dir, "ADNI", out_dir)
    create_dir(data_dst_dir)

    print("Move files\nfrom: {0}\nto {1}".format(data_src_dir, data_dst_dir))

    subjects = os.listdir(data_src_dir)
    for subject in tqdm(subjects):
        subj_dir = os.path.join(data_src_dir, subject)
        src_path = glob.glob(os.path.join(subj_dir, "*", "*", "*", "*"))[0]
        dst_path = os.path.join(data_dst_dir, subject + ".nii")
        shutil.copyfile(src_path, dst_path)
        check_call(["gzip", dst_path, "-f"])