U 5d @s dZddlZddlZddlZddlmZddlmZddlZddl m Z ddl m Z dd d d d gZ ejd edZejedddZeeeeeddeeeeedddZeeZW5QRXe dddd ZeZdd Zdd Ze ddddddddZddZ dS)z(Utility functions for dealing with filesN)Path)resource_filename)ParameterError)deprecate_positional_args find_filesexampleex list_examples example_infoZLIBROSA_DATA_DIRZlibrosazhttps://librosa.org/data/audio/)base_urlregistryZ example_dataz registry.txtz index.jsonrF)hqcCs:|tkrtd||r d}nd}tt|d|S)aQRetrieve the example recording identified by 'key'. The first time an example is requested, it will be downloaded from the remote repository over HTTPS. All subsequent requests will use a locally cached copy of the recording. For a list of examples (and their keys), see `librosa.util.list_examples`. By default, local files will be cached in the directory given by `pooch.os_cache('librosa')`. You can override this by setting an environment variable ``LIBROSA_DATA_DIR`` prior to importing librosa: >>> import os >>> os.environ['LIBROSA_DATA_DIR'] = '/path/to/store/data' >>> import librosa Parameters ---------- key : str The identifier for the track to load hq : bool If ``True``, return the high-quality version of the recording. If ``False``, return the 22KHz mono version of the recording. Returns ------- path : str The path to the requested example file Examples -------- Load "Hungarian Dance #5" by Johannes Brahms >>> y, sr = librosa.load(librosa.example('brahms')) Load "Vibe Ace" by Kevin MacLeod (the example previously packaged with librosa) in high-quality mode >>> y, sr = librosa.load(librosa.example('vibeace', hq=True)) See Also -------- librosa.util.list_examples pooch.os_cache Unknown example key: {}z.hq.oggz.oggpath) __TRACKMAPrformat __GOODBOYfetch)keyrextr6/tmp/pip-unpacked-wheel-8l90aumz/librosa/util/files.pyr*s 0cCs>tdtdttD]}td|t|dqdS)a=List the available audio recordings included with librosa. Each recording is given a unique identifier (e.g., "brahms" or "nutcracker"), listed in the first column of the output. A brief description is provided in the second column. See Also -------- util.example util.example_info zAVAILABLE EXAMPLESD--------------------------------------------------------------------z{:10} {}descN)printsortedrkeysr)rrrrr is c Csx|tkrtd|tt|dd}t|d8}td|t|dtd|D] }t|q\W5QRXdS) aDisplay licensing and metadata information for the given example recording. The first time an example is requested, it will be downloaded from the remote repository over HTTPS. All subsequent requests will use a locally cached copy of the recording. For a list of examples (and their keys), see `librosa.util.list_examples`. By default, local files will be cached in the directory given by `pooch.os_cache('librosa')`. You can override this by setting an environment variable ``LIBROSA_DATA_DIR`` prior to importing librosa. Parameters ---------- key : str The identifier for the recording (see `list_examples`) See Also -------- librosa.util.example librosa.util.list_examples pooch.os_cache rrz.txtrz {:10s} {:s}rrN)rrrrropenr)rlicensefdesclinerrrr |s T)rrecursecase_sensitivelimitoffsetcCs|dkrdddddddg}nt|tr,|g}t|}|s`td d |D}|td d |DO}t}|rt|D]}|t|d |O}qtn t||}t|}|||d}|dk r|d|}|S) a@Get a sorted list of (audio) files in a directory or directory sub-tree. Examples -------- >>> # Get all audio files in a directory sub-tree >>> files = librosa.util.find_files('~/Music') >>> # Look only within a specific directory, not the sub-tree >>> files = librosa.util.find_files('~/Music', recurse=False) >>> # Only look for mp3 files >>> files = librosa.util.find_files('~/Music', ext='mp3') >>> # Or just mp3 and ogg >>> files = librosa.util.find_files('~/Music', ext=['mp3', 'ogg']) >>> # Only get the first 10 files >>> files = librosa.util.find_files('~/Music', limit=10) >>> # Or last 10 files >>> files = librosa.util.find_files('~/Music', offset=-10) >>> # Avoid including search patterns in the path string >>> import glob >>> directory = '~/[202206] Music' >>> directory = glob.escape(directory) # Escape the special characters >>> files = librosa.util.find_files(directory) Parameters ---------- directory : str Path to look for files ext : str or list of str A file extension or list of file extensions to include in the search. Default: ``['aac', 'au', 'flac', 'm4a', 'mp3', 'ogg', 'wav']`` recurse : boolean If ``True``, then all subfolders of ``directory`` will be searched. Otherwise, only ``directory`` will be searched. case_sensitive : boolean If ``False``, files matching upper-case version of extensions will be included. limit : int > 0 or None Return at most ``limit`` files. If ``None``, all files are returned. offset : int Return files starting at ``offset`` within the list. Use negative values to offset from the end of the list. Returns ------- files : list of str The list of audio files. NZaacauZflacZm4aZmp3ZoggZwavcSsg|] }|qSr)lower.0errr szfind_files..cSsg|] }|qSr)upperr)rrrr,sr) isinstancestrsetoswalk __get_fileslistsort) directoryrr#r$r%r&filesr2rrrrs&A    cCsTtjtj|}t}|D]0}tj|dtjj|}|tt|O}q|S)z2Helper function to get files in a single directory*)r1rabspath expanduserr0joinextsepglob)Zdir_name extensionsZmyfilesZsub_extZglobstrrrrr3s r3)!__doc__r1r=jsonpathlibr pkg_resourcesrZpooch exceptionsrZ decoratorsr__all__environgetZos_cacheZ __data_pathcreaterZ load_registry__name__r/rr!loadrrr r r rr3rrrrsT     :%a