:jgddlZddlZddlmZddlmZmZmZddlZddl m Z ddl m Z ddl mZmZdZdZd Zgd ZGd d e Zd ededefdZy)N)Path)OptionalTupleUnion)Dataset)download_url_to_file) _extract_tar_load_waveformz6https://speech.fit.vutbr.cz/files/quesst14Database.tgzi@@4f869e06bc066bbe9c5dde31dbd3909a0870d70291110ebbb38878dcbc2fc5e4)albanianbasqueczech nnenglishromanianslovakc eZdZdZ d deeefdedeededdf dZ d e de ee effd Z d e de e je effd Zde fd Zy)QUESST14a*QUESST14* :cite:`Mir2015QUESST2014EQ` dataset. Args: root (str or Path): Root directory where the dataset's top level directory is found subset (str): Subset of the dataset to use. Options: [``"docs"``, ``"dev"``, ``"eval"``]. language (str or None, optional): Language to get dataset for. Options: [``None``, ``albanian``, ``basque``, ``czech``, ``nnenglish``, ``romanian``, ``slovak``]. If ``None``, dataset consists of all languages. (default: ``"nnenglish"``) download (bool, optional): Whether to download the dataset if it is not found at root path. (default: ``False``) rootsubsetlanguagedownloadreturnNcJ|dvr td|#|tvrtdtttj|}tj j t}tj j||}|jddd}tj j|||_ tj j|jsNtj j|s#|s tdtt|tt!|||d k(rt#|j|d |_y|d k(rt#|j|d |_y|d k(rt#|j|d|_yy)N)docsdevevalz/`subset` must be one of ['docs', 'dev', 'eval']z"`language` must be None or one of .rz9Dataset not found. Please use `download=True` to download) hash_prefixrzlanguage_key_utterances.lstrzlanguage_key_dev.lstrzlanguage_key_eval.lst) ValueError _LANGUAGESstrosfspathpathbasenameURLjoinrsplit_pathisdirisfile RuntimeErrorr _CHECKSUMr filter_audio_pathsdata)selfrrrrr&archives q/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/torchaudio/datasets/quesst14.py__init__zQUESST14.__init__&sD 0 0NO O  HJ$>A#j/ARST Tyy77##C('',,tX.??3*1-WW\\$1 ww}}TZZ(77>>'*&'bcc$S'yI $ ' V *4::xA^_DI u_*4::xAWXDI v *4::xAXYDInc|j|}tjj||j}|t |j djfS)aGet metadata for the n-th sample from the dataset. Returns filepath instead of waveform, but otherwise returns the same fields as :py:func:`__getitem__`. Args: n (int): The index of the sample to be loaded Returns: Tuple of the following items; str: Path to audio int: Sample rate str: File name )r0r#r%relpathr* SAMPLE_RATE with_suffixname)r1r6 audio_pathr9s r3 get_metadatazQUESST14.get_metadataJsG"YYq\ ''//*djj9 Z%;%;B%?%D%DDDr5cp|j|}t|j|d|d}|f|ddzS)a:Load the n-th sample from the dataset. Args: n (int): The index of the sample to be loaded Returns: Tuple of the following items; Tensor: Waveform int: Sample rate str: File name rN)r>r r*)r1r6metadatawaveforms r3 __getitem__zQUESST14.__getitem___sA $$Q'!$**hqk8A;G{Xab\))r5c,t|jS)N)lenr0)r1s r3__len__zQUESST14.__len__ss499~r5)rF)__name__ __module__ __qualname____doc__rr"rrboolr4intrr>torchTensorrCrFr5r3rrs  #. "ZCI"Z"Z3- "Z  "Z  "ZHEcEeCcM&:E**S*U5<<c+A%B*(r5rr%rlst_namec&g}t|}t|dz |z 5}|D]V}|jj\}}|||k7r,t j dd|}|j ||z X ddd|S#1swY|SxYw)z+Extract audio paths for the given language.scoringNz^.*?\/r8)ropenstripsplitresubappend)r%rrP audio_pathsfliner=langs r3r/r/ws K :D dY) *2a 2D#zz|113 J#(8 2z:J   tj0 1  22 2 s ABB)r#rVpathlibrtypingrrrrMtorch.utils.datartorchaudio._internalrtorchaudio.datasets.utilsr r r'r:r.r!rr"r/rOr5r3rbsf )) $5B? N  [w[| r5