"""

Created on November 10, 2019

functions for writing/reading data to/from disk

@modified_by: Soroosh Tayebi Arasteh <soroosh.arasteh@fau.de>

"""

import yaml

import numpy as np

import os

import warnings

import shutil

from skimage import io

import pdb

import SimpleITK as sitk

def read_config(config_path):

    """Reads config file in yaml format into a dictionary

    Parameters

    ----------

    config_path: str

        Path to the config file in yaml format

    Returns

    -------

    config dictionary

    """

    with open(config_path, 'rb') as yaml_file:

        return yaml.safe_load(yaml_file)

def write_config(params, cfg_path, sort_keys=False):

    with open(cfg_path, 'w') as f:

        yaml.dump(params, f)

def create_experiment(experiment_name, global_config_path):

    params = read_config(global_config_path)

    params['experiment_name'] = experiment_name

    create_experiment_folders(params)

    cfg_file_name = params['experiment_name'] + '_config.yaml'

    cfg_path = os.path.join(os.path.join(params['target_dir'], params['network_output_path']), cfg_file_name)

    params['cfg_path'] = cfg_path

    write_config(params, cfg_path)

    return params

def create_experiment_folders(params):

    try:

        path_keynames = ["network_output_path", "tb_logs_path", "stat_log_path", "output_data_path"]

        for key in path_keynames:

            params[key] = os.path.join(params['experiment_name'], params[key])

            os.makedirs(os.path.join(params['target_dir'], params[key]))

    except:

        raise Exception("Experiment already exist. Please try a different experiment name")

def open_experiment(experiment_name, global_config_path):

    """Open Existing Experiments

    """

    default_params = read_config(global_config_path)

    cfg_file_name = experiment_name + '_config.yaml'

    cfg_path = os.path.join(os.path.join(default_params['target_dir'], experiment_name, default_params['network_output_path']), cfg_file_name)

    params = read_config(cfg_path)

    return params

def delete_experiment(experiment_name, global_config_path):

    """Delete Existing Experiment folder

    """

    default_params = read_config(global_config_path)

    cfg_file_name = experiment_name + '_config.yaml'

    cfg_path = os.path.join(os.path.join(default_params['target_dir'], experiment_name, default_params['network_output_path']), cfg_file_name)

    params = read_config(cfg_path)

    shutil.rmtree(os.path.join(params['target_dir'], experiment_name))