P/ju dZddlmZmZddlZej j ZdefdZ defdZ defd Z d e fd Z d eeeffd Zd eefdZd eefdZd efdZy)zModule to change the configuration of libsox, which is used by I/O functions like :py:mod:`~torchaudio.backend.sox_io_backend` and :py:mod:`~torchaudio.sox_effects`. )DictListNseedc.tj|y)zSet libsox's PRNG Args: seed (int): seed value. valid range is int32. See Also: http://sox.sourceforge.net/sox.html N)sox_extset_seed)rs ]/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torchaudio/utils/sox_utils.pyrr s T verbosityc.tj|y)aDSet libsox's verbosity Args: verbosity (int): Set verbosity level of libsox. * ``1`` failure messages * ``2`` warnings * ``3`` details of processing * ``4``-``6`` increasing levels of debug messages See Also: http://sox.sourceforge.net/sox.html N)r set_verbosity)r s r r r s )$r buffer_sizec.tj|y)zSet buffer size for sox effect chain Args: buffer_size (int): Set the size in bytes of the buffers used for processing audio. See Also: http://sox.sourceforge.net/sox.html N)rset_buffer_size)rs r rr)s K(r use_threadsc.tj|y)a>Set multithread option for sox effect chain Args: use_threads (bool): When ``True``, enables ``libsox``'s parallel effects channels processing. To use mutlithread, the underlying ``libsox`` has to be compiled with OpenMP support. See Also: http://sox.sourceforge.net/sox.html N)rset_use_threads)rs r rr5s K(r returnc<ttjS)zxList the available sox effect names Returns: Dict[str, str]: Mapping from ``effect name`` to ``usage`` )dictr list_effectsr r rrBs $$& ''r c*tjS)zoList the supported audio formats for read Returns: List[str]: List of supported audio formats )rlist_read_formatsrr r rrKs  $ $ &&r c*tjS)zpList the supported audio formats for write Returns: List[str]: List of supported audio formats )rlist_write_formatsrr r rrTs  % % ''r c*tjS)zxGet buffer size for sox effect chain Returns: int: size in bytes of buffers used for processing audio. )rget_buffer_sizerr r rr]s  " " $$r )__doc__typingrr torchaudio _extensionlazy_import_sox_extrintrr rboolrstrrrrrrr r r's    3 3 5 3 %S%" ) ) ) )(d38n('49'(DI(%%r