import pathfinder
import glob
import random
import utils_lung
import utils

VALIDATION_SET_SIZE = 0.2


def read_split(path):
    d = utils.load_pkl(path)
    print d['valid']
    # print d['train']


def make_luna_validation_split():
    luna_path = pathfinder.LUNA_DATA_PATH
    file_list = sorted(glob.glob(luna_path + "/*.mhd"))
    random.seed(317070)
    all_pids = [utils_lung.extract_pid_filename(f) for f in file_list]
    validation_pids = random.sample(all_pids, int(VALIDATION_SET_SIZE * len(file_list)))
    train_pids = list(set(all_pids) - set(validation_pids))
    d = {}
    d['valid'] = validation_pids
    d['train'] = train_pids
    utils.save_pkl(d, pathfinder.LUNA_VALIDATION_SPLIT_PATH)


def make_kaggle_validation_split():
    pass


if __name__ == '__main__':
    make_luna_validation_split()
    read_split(pathfinder.LUNA_VALIDATION_SPLIT_PATH)