:jhddlZddlmZddlmZmZdZeddfdeeeeeeeffdZeddfdZ y) N)Path)ListUnion)z.wavTpathscg}t|tttfr t|}n|g}|D]}t |j j tr:ttjj|}|j|itjj|s|t||||z }|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)rrr r file_pathsp file_paths u/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/torch_audiomentations/utils/file.pyfind_audio_files_in_pathsr sJ%$s+,U    q6<<> " "#7 8RWW__Q/0I   i ( WW]]1  *!1(? /  J  cjg}tj||D]\}}}t|}|D]{}tjj |} tjj | |} |j j|sb|jt| }|r|S|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_pathrr r rrootdirs filenamesfilename input_pathrs rrr(sJ!#!P dI9% ! 3H.J Z:I~~(()9:!!$y/2  3 '    r) rpathlibrtypingrrrrrrrrr-sU ! *   dT#Yc1 2B*  r