U ,:%e!@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl Z e e Ze ejjdZedddZeed d d Zd d ZddZedddZddddgZddZddZd&ddZddZd d!Zd"d#Zd$d%Z dS)'zModule to implement logics used for initializing extensions. The implementations here should be stateless. They should not depend on external state. Anything that depends on external state should happen in __init__.py Nwraps)Pathlib)rcCs(tjdkrdnd}t|d|}|S)NntZpydso.)osname_LIB_DIR)rsuffixpathrZ/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torchaudio/_extension/utils.py _get_lib_pathsr)rreturncCs0t|}|sdStj|tj|dS)a Load extension module Note: In case `torchaudio` is deployed with `pex` format, the library file is not in a standard location. In this case, we expect that `libtorchaudio` is available somewhere in the search path of dynamic loading mechanism, so that importing `_torchaudio` will have library loader find and load `libtorchaudio`. This is the reason why the function should not raising an error when the library file is not found. Returns: bool: True if the library file is found AND the library loaded without failure. False if the library file is not found (like in the case where torchaudio is deployed with pex format, thus the shared library file is in a non-standard location.). If the library file is found but there is an issue loading the library, (such as missing dependency) then this function raises the exception as-is. Raises: Exception: If the library file is found, but there is an issue loading the library file, (when underlying `ctype.DLL` throws an exception), this function will pass the exception as-is, instead of catching it and returning bool. The expected case is `OSError` thrown by `ctype.DLL` when a dynamic dependency is not found. This behavior was chosen because the expected failure case is not recoverable. If a dependency is missing, then users have to install it. FT)rexiststorchopsZ load_libraryclasses)rr rrr _load_libs   rcCsFtdddl}|jjdddl}tjj | tjjj dS)NZlibtorchaudio_soxr) rZtorchaudio.lib._torchaudio_soxrZ_torchaudio_sox set_verbosityatexitrr torchaudioZ"sox_effects_initialize_sox_effectsregisterZ sox_effects_shutdown_sox_effects)rrrrr _init_soxEs  rcCsBddddt}dddd|}|j|d }tjj|dS) Nzlibavutil.so.{ver}zlibavutil.{ver}.dylibzavutil-{ver}.dll)LinuxDarwinWindows:98)654)ver)platformsystemformatrr _torchaudioZ find_avutil) ffmpeg_verZlibname_templateZ avutil_verZ libavutilrrr_try_access_avutilQs r+)r*cCsltd|d|}d|}t|ssz._find_available_ffmpeg_ext..r)r9rrr_find_available_ffmpeg_exts rAFcCst}|stdtjd}rd|dgkr8tdn,||kr^td|ddd|D|g}t||}| | d kr| d |S) Nz[TorchAudio is not built with FFmpeg integration. Please build torchaudio with USE_FFMPEG=1.ZTORCHAUDIO_USE_FFMPEG_VERSIONr4zTTorchAudio is built in single FFmpeg mode. TORCHAUDIO_USE_FFMPEG_VERSION is ignored.zThe FFmpeg version zT (read from TORCHAUDIO_USE_FFMPEG_VERSION) is not available. Available versions are cSsg|] }|r|qSrrr=rrrr@sz _init_ffmpeg..) rAr/r environgetwarningswarn ValueErrorr<initZ get_log_levelZ set_log_level)r:r9r*extrrr _init_ffmpegs$     rJc CsNtjdddD]4}tj|rzt|Wqtk rFYqXqdS)NPATHr4;)r rCrDsplitr rZadd_dll_directoryr7)r rrr_init_dll_paths  rNcCstjj}|dk rtjjdk rt|}|ddd|d}tjjd}|dd|d}||krt d|d|d|S) NrrzjDetected that PyTorch and TorchAudio were compiled with different CUDA versions. PyTorch has CUDA version z% whereas TorchAudio has CUDA version zJ. Please install the TorchAudio version that matches your PyTorch version.) rrr)Z cuda_versionrversioncudastrrMr/)rR version_strZ ta_versionZ t_versionrrr_check_cuda_versions rVcstfdd}|S)Nc sJz tWn4tk r>}ztjd|W5d}~XYnX||S)Nzm requires sox extension which is not available. Please refer to the stacktrace above for how to resolve this.)rr7r/__name___args_kwargserrfuncrrwrappeds  z#_fail_since_no_sox..wrappedrr]r^rr\r_fail_since_no_soxsr`cstfdd}|S)Nc sNztddWn4tk rB}ztjd|W5d}~XYnX||S)NT)r:zp requires FFmpeg extension which is not available. Please refer to the stacktrace above for how to resolve this.)rJr7r/rWrXr\rrr^s z&_fail_since_no_ffmpeg..wrappedrr_rr\r_fail_since_no_ffmpegsra)F)!__doc__r0loggingr r&rE functoolsrpathlibrrr getLoggerrWr-__file__parentr rTrboolrrr+r3Z _FFMPEG_VERSr<rArJrNrVr`rarrrrs0   '