o pi@sfddlZddlmZddlmZmZdZeddfdeeeeeeeffddZeddfd d Z dS) N)Path)ListUnion)z.wavTpathscCsg}t|tttfrt|}n|g}|D])}t|tr-tt j |}| |qt j |r=|t||||d7}q|S)zReturn a list of paths to all audio files with the given extension(s) contained in the list or in its directories. Also traverses subdirectories by default. )filename_endingstraverse_subdirectoriesfollow_symlinks) isinstancelisttuplesetstrlowerendswithSUPPORTED_EXTENSIONSrospathabspathappendisdirfind_audio_files)rrrr file_pathsp file_pathrT/home/ubuntu/.local/lib/python3.10/site-packages/torch_audiomentations/utils/file.pyfind_audio_files_in_paths s"    rc Cstg}tj||dD].\}}}t|}|D]}tj|} tj| |} ||r1|t | q|s7|Sq |S)zReturn a list of paths to all audio files with the given extension(s) in a directory. Also traverses subdirectories by default. ) followlinks) rwalksortedrrjoinrrrr) root_pathrrrrrootdirs filenamesfilename input_pathrrrrr(s  r) rpathlibrtypingrrrr rrrrrrs  !