Source code for patato.utils.process_study

#  Copyright (c) Thomas Else 2023-25.
#  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