Source code for patato.utils.process_study

#  Copyright (c) Thomas Else 2023.
#  License: MIT

import glob
from os.path import join

from ..io.msot_data import PAData
from . import sort_key



[docs]
def get_hdf5_files(folder, filter_name="", mode="r"):
    for file in sorted(glob.glob(join(folder, "**", "*.hdf5"), recursive=True), key=sort_key):
        pa_data = PAData.from_hdf5(file, mode)
        if filter_name.lower() not in pa_data.get_scan_name().lower() and filter_name.lower() not in file.lower():
            pa_data.scan_reader.close()
            continue
        else:
            yield file, pa_data