from dxpy.filesystem import Path
import os
datasets_configs = {
    'dataset_root_path': os.environ.get('PATHS_DATASET', str(Path(os.environ.get('HOME')) / 'Datas')),
    'path': '/home/hongxwing/Datas/',
    'analytical_phantom_sinogram': {
        'path': '/home/hongxwing/Datas/Phantom',
    },
    'apssr': {
        'image_type': 'sinogram',
        'target_shape': [320, 320],
        'super_resolution': {
            'nb_down_sample': 3
        },
    },
    'mice_sinograms': {
        'path': str(Path(os.environ.get('HOME'))/'Datas'/'Mice')
    }
}

config = {
    'train': {
        'summary_freq': 60,
        'ckpt_path': './save'
    },
    'datasets': datasets_configs
}


def get_config():
    from dxpy.configs import ConfigsView
    return ConfigsView(config)

def get_configs_view():
    from dxpy.configs import ConfigsView
    return ConfigsView(config)

def clear_config():
    global config
    for k in config:
        config.pop(k)
