o -i@sddlZddlZddlmZddlmZmZddlZddlm Z ddl m Z ddl m Z ddlmZdZd Zid d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-iZd.ed/ed0ed1ed2ee eeeff d3d4ZGd5d6d6e ZdS)7N)Path)TupleUnion)Tensor)Dataset)download_url_to_file) _extract_taraewARCTICz>http://festvox.org/cmu_arctic/packed/cmu_us_aew_arctic.tar.bz2@645cb33c0f0b2ce41384fdd8d3db2c3f5fc15c1e688baeb74d2e08cab18ab406z>http://festvox.org/cmu_arctic/packed/cmu_us_ahw_arctic.tar.bz2@024664adeb892809d646a3efd043625b46b5bfa3e6189b3500b2d0d59dfab06cz>http://festvox.org/cmu_arctic/packed/cmu_us_aup_arctic.tar.bz2@2c55bc3050caa996758869126ad10cf42e1441212111db034b3a45189c18b6fcz>http://festvox.org/cmu_arctic/packed/cmu_us_awb_arctic.tar.bz2@d74a950c9739a65f7bfc4dfa6187f2730fa03de5b8eb3f2da97a51b74df64d3cz>http://festvox.org/cmu_arctic/packed/cmu_us_axb_arctic.tar.bz2@dd65c3d2907d1ee52f86e44f578319159e60f4bf722a9142be01161d84e330ffz>http://festvox.org/cmu_arctic/packed/cmu_us_bdl_arctic.tar.bz2@26b91aaf48b2799b2956792b4632c2f926cd0542f402b5452d5adecb60942904z>http://festvox.org/cmu_arctic/packed/cmu_us_clb_arctic.tar.bz2@3f16dc3f3b97955ea22623efb33b444341013fc660677b2e170efdcc959fa7c6z>http://festvox.org/cmu_arctic/packed/cmu_us_eey_arctic.tar.bz2@8a0ee4e5acbd4b2f61a4fb947c1730ab3adcc9dc50b195981d99391d29928e8az>http://festvox.org/cmu_arctic/packed/cmu_us_fem_arctic.tar.bz2@3fcff629412b57233589cdb058f730594a62c4f3a75c20de14afe06621ef45e2z>http://festvox.org/cmu_arctic/packed/cmu_us_gka_arctic.tar.bz2@dc82e7967cbd5eddbed33074b0699128dbd4482b41711916d58103707e38c67fz>http://festvox.org/cmu_arctic/packed/cmu_us_jmk_arctic.tar.bz2@3a37c0e1dfc91e734fdbc88b562d9e2ebca621772402cdc693bbc9b09b211d73z>http://festvox.org/cmu_arctic/packed/cmu_us_ksp_arctic.tar.bz2@8029cafce8296f9bed3022c44ef1e7953332b6bf6943c14b929f468122532717z>http://festvox.org/cmu_arctic/packed/cmu_us_ljm_arctic.tar.bz2@b23993765cbf2b9e7bbc3c85b6c56eaf292ac81ee4bb887b638a24d104f921a0z>http://festvox.org/cmu_arctic/packed/cmu_us_lnh_arctic.tar.bz2@4faf34d71aa7112813252fb20c5433e2fdd9a9de55a00701ffcbf05f24a5991az>http://festvox.org/cmu_arctic/packed/cmu_us_rms_arctic.tar.bz2@c6dc11235629c58441c071a7ba8a2d067903dfefbaabc4056d87da35b72ecda4z>http://festvox.org/cmu_arctic/packed/cmu_us_rxr_arctic.tar.bz2@1fa4271c393e5998d200e56c102ff46fcfea169aaa2148ad9e9469616fbfdd9bz>http://festvox.org/cmu_arctic/packed/cmu_us_slp_arctic.tar.bz2@54345ed55e45c23d419e9a823eef427f1cc93c83a710735ec667d068c916abf1z>http://festvox.org/cmu_arctic/packed/cmu_us_slt_arctic.tar.bz2@7c173297916acf3cc7fcab2713be4c60b27312316765a90934651d367226b4ealinepath folder_audio ext_audioreturnc Csd|ddddd\}}|dd}tj||||}t|\}}||||ddfS)Nr _)stripsplitosrjoin torchaudioload) rrrr utterance_id transcript file_audiowaveform sample_rater2V/home/ubuntu/LTX-2/.venv/lib/python3.10/site-packages/torchaudio/datasets/cmuarctic.pyload_cmuarctic_item$s  r4c @szeZdZdZdZdZdZdZee dfde e e fde d e d e d d f d dZded eeee e ffddZd efddZd S) CMUARCTICa4*CMU ARCTIC* :cite:`Kominek03cmuarctic` dataset. Args: root (str or Path): Path to the directory where the dataset is found or downloaded. url (str, optional): The URL to download the dataset from or the type of the dataset to download. (default: ``"aew"``) Allowed type values are ``"aew"``, ``"ahw"``, ``"aup"``, ``"awb"``, ``"axb"``, ``"bdl"``, ``"clb"``, ``"eey"``, ``"fem"``, ``"gka"``, ``"jmk"``, ``"ksp"``, ``"ljm"``, ``"lnh"``, ``"rms"``, ``"rxr"``, ``"slp"`` or ``"slt"``. folder_in_archive (str, optional): The top-level directory of the dataset. (default: ``"ARCTIC"``) download (bool, optional): Whether to download the dataset if it is not found at root path. (default: ``False``). z txt.done.dataetcz.wavwavFrooturlfolder_in_archivedownloadr!Nc CsT|dvrd|d}d}d}tj|||}t|}tj|}tj||}tj|s4t|tj||}|dd}tj|||_|rktj|jsjtj |sft |d} t ||| dt |ntj|js{td |jd tj|j|j|j|_t|jd } t| } t| |_WddS1swYdS) N)r ahwaupawbaxbbdlclbeeyfemgkajmkkspljmlnhrmsrxrslpsltcmu_us__arcticz.tar.bz2z)http://www.festvox.org/cmu_arctic/packed/.r) hash_prefixz The path zT doesn't exist. Please check the ``root`` path or set `download=True` to download itr)r)rr*fspathbasenameisdirmkdirr(_pathisfile _CHECKSUMSgetrrexists RuntimeError _folder_text _file_text_textopencsvreaderlist_walker) selfr8r9r:r; ext_archivebase_urlrSarchivechecksumtextwalkerr2r2r3__init__Is:          "zCMUARCTIC.__init__ncCs|j|}t||j|j|jS)aiLoad 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: Transcript str: Utterance ID )rcr4rV _folder_audio _ext_audio)rdrlrr2r2r3 __getitem__s zCMUARCTIC.__getitem__cCs t|jS)N)lenrc)rdr2r2r3__len__s zCMUARCTIC.__len__)__name__ __module__ __qualname____doc__r]r\rnrmURLFOLDER_IN_ARCHIVErstrrboolrkintrrrorqr2r2r2r3r53s(  >r5)r`r)pathlibrtypingrrr+torchrtorch.utils.datartorchaudio._internalrtorchaudio.datasets.utilsrrvrwrXrxrzr4r5r2r2r2r3sd          *