# Copyright 2019 Population Health Sciences and Image Analysis, German Center for Neurodegenerative Diseases(DZNE)

#

#    Licensed under the Apache License, Version 2.0 (the "License");

#    you may not use this file except in compliance with the License.

#    You may obtain a copy of the License at

#

#        http://www.apache.org/licenses/LICENSE-2.0

#

#    Unless required by applicable law or agreed to in writing, software

#    distributed under the License is distributed on an "AS IS" BASIS,

#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

#    See the License for the specific language governing permissions and

#    limitations under the License.

import os

import fnmatch

import h5py

import numpy as np

from shutil import copy2

def locate_file(pattern, root):

    '''Locate all files matching supplied filename pattern in and below

    supplied root directory.'''

    file_path=[]

    for path, dirs, files in os.walk(os.path.abspath(root)):

        for filename in fnmatch.filter(files, pattern):

            file_path.append(os.path.join(path, filename))

    return file_path

def locate_dir(pattern, root):

    '''Locate all files matching supplied filename pattern in and below

    supplied root directory.'''

    dir_path=[]

    for path, dirs, files in os.walk(os.path.abspath(root)):

        for dirname in fnmatch.filter(dirs, pattern):

            dir_path.append(os.path.join(path, dirname))

    return dir_path

def save_dict_to_hdf5(dic, filename):

    """

    ....

    """

    with h5py.File(filename, 'w') as h5file:

        recursively_save_dict_contents_to_group(h5file, '/', dic)

def recursively_save_dict_contents_to_group(h5file, path, dic):

    """

    ....

    """

    for key, item in dic.items():

        if isinstance(item, (np.ndarray, np.int64, np.dtype(float).type, str, bytes)):

            h5file[path + key] = item

        elif isinstance(item, dict):

            recursively_save_dict_contents_to_group(h5file, path + key + '/', item)

        else:

            raise ValueError('Cannot save %s type'%type(item))

def load_dict_from_hdf5(filename):

    """

    ....

    """

    with h5py.File(filename, 'r') as h5file:

        return recursively_load_dict_contents_from_group(h5file, '/')

def recursively_load_dict_contents_from_group(h5file, path):

    """

    ....

    """

    ans = {}

    for key, item in h5file[path].items():

        if isinstance(item, h5py._hl.dataset.Dataset):

            ans[key] = item.value

        elif isinstance(item, h5py._hl.group.Group):

            ans[key] = recursively_load_dict_contents_from_group(h5file, path + key + '/')

    return ans

def move_file(source,destination):

    copy2(source,destination)

    return None