o i)@sddlZddlTddlmZddlmZGdddeZGdddeZGd d d eZ Gd d d eZ Gd ddeZ GdddeZ GdddeZ GdddeZGdddeZGdddeZGdddeZGdddeZGdddeZgdZdS) N)*)Enum)Sequencecs\eZdZddedeeffdd ZddZed efd d Zed eefd d Z Z S) KoalaErrorNmessage message_stackcs0t|||_|durt|_dS||_dSN)super__init___messagelist_message_stack)selfrr __class__B/home/ubuntu/.local/lib/python3.10/site-packages/pvkoala/_koala.pyr s zKoalaError.__init__cCsJ|j}t|jdkr#|d7}tt|jD] }|d||j|f7}q|S)Nr:z [%d] %s)r lenrrange)rrirrr__str__s zKoalaError.__str__returncC|jSr )r rrrrr!zKoalaError.messagecCrr )rrrrrr%rzKoalaError.message_stack)rN) __name__ __module__ __qualname__strrr rpropertyrr __classcell__rrrrrsrc@ eZdZdS)KoalaMemoryErrorNrrrrrrrr$*r$c@r#) KoalaIOErrorNr%rrrrr'.r&r'c@r#)KoalaInvalidArgumentErrorNr%rrrrr(2r&r(c@r#)KoalaStopIterationErrorNr%rrrrr)6r&r)c@r#) KoalaKeyErrorNr%rrrrr*:r&r*c@r#)KoalaInvalidStateErrorNr%rrrrr+>r&r+c@r#)KoalaRuntimeErrorNr%rrrrr,Br&r,c@r#)KoalaActivationErrorNr%rrrrr-Fr&r-c@r#)KoalaActivationLimitErrorNr%rrrrr.Jr&r.c@r#)KoalaActivationThrottledErrorNr%rrrrr/Nr&r/c@r#)KoalaActivationRefusedErrorNr%rrrrr0Rr&r0c@seZdZdZGdddeZejeeje ej e ej e ejeejeejeejeejeejeejei ZGdddeZdededed d fd d Z d e!e"d e!e"fddZ#dddZ$dddZ%e&d e"fddZ'e&d e"fddZ(e&d e"fddZ)e&d efddZ*d e!efddZ+d S)Koalaz< Python binding for Koala noise-suppression engine. c@s<eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd S)zKoala.PicovoiceStatusesr N)rrrSUCCESS OUT_OF_MEMORYIO_ERRORINVALID_ARGUMENTSTOP_ITERATION KEY_ERROR INVALID_STATE RUNTIME_ERRORACTIVATION_ERRORACTIVATION_LIMIT_REACHEDACTIVATION_THROTTLEDACTIVATION_REFUSEDrrrrPicovoiceStatuses[srIc@r#)z Koala.CKoalaNr%rrrrCKoalawr&rJ access_key model_path library_pathrNc Cs>t|tr t|dkrtdtj|std|tj|s'td|t |}|j }t g|_ d|_ |dd|j|_ttt ttg|j_ |j|j_ |j|_tt g|j_ d|j_ |j}t t tt|jg|_ |j|_ t|j|_|||t|j}||jjur|j|d|d |j|_t|jg|j_ d|j_ |j}t|jtt g|_ |j|_ t } ||j| }||jjur|!|j|d |d | j"|_#|j$|_%t|jtt&tt&g|j%_ |j|j%_ |j'|_(t|jg|j(_ |j|j(_ |)|_*|+|_,|j-} g| _ t | _ | .d|_/dS) a Constructor. :param access_key: AccessKey obtained from Picovoice Console (https://console.picovoice.ai/) :param model_path: Absolute path to the file containing model parameters. :param library_path: Absolute path to Koala's dynamic library. rz*`access_key` should be a non-empty string.z"Could not find model file at `%s`.z/Could not find Koala's dynamic library at `%s`.Npythonutf-8zInitialization failedrrzFailed to get delay samples)0 isinstancer rr(ospathexistsr'cdll LoadLibrary pv_set_sdkc_char_pargtypesrestypeencodepv_get_error_stack_get_error_stack_funcPOINTERc_intrIpv_free_error_stack_free_error_stack_func pv_koala_initrJ_handlebyrefr=_PICOVOICE_STATUS_TO_EXCEPTION_get_error_stackpv_koala_delete _delete_funcpv_koala_delay_samplec_int32deletevalue _delay_samplepv_koala_process _process_funcc_shortpv_koala_reset _reset_funcpv_sample_rate _sample_ratepv_koala_frame_length _frame_lengthpv_koala_versiondecode_version) rrKrLrMlibrary set_sdk_func init_funcstatusdelay_sample_func delay_sample version_funcrrrr zsz              zKoala.__init__pcmcCsvt||jkrtdt||jft|j}||}|}||j||}||jjur7|j|d| dt |S)a Processes a frame of audio and returns delayed enhanced audio. :param pcm: A frame of audio samples. The number of samples per frame can be attained by calling `.frame_length`. The incoming audio needs to have a sample rate equal to `.sample_rate` and be 16-bit linearly-encoded. Koala operates on single-channel audio. Consecutive calls to `.process()` must provide consecutive frames of audio from the same source, unless `.reset()` has been called in between. :return: A frame of enhanced audio samples, stored as a sequence of 16-bit linearly-encoded integers. The output is not directly the enhanced version of the input PCM, but corresponds to samples that were given in previous calls to `.process()`. The delay in samples between the start time of the input frame and the start time of the output frame can be attained from `.delay_sample`. z@Length of input frame %d does not match required frame length %dzProcessing failedrP) r frame_lengthr(rprorcrIr=rerfr )rr frame_type enhanced_pcmr}rrrprocesss  z Koala.processcCs2||j}||jjur|j|d|ddS)z Resets Koala into a state as if it had just been newly created. Call this function in between calls to `process` that do not provide consecutive frames of audio. z Reset failedrPN)rrrcrIr=rerf)rr}rrrresets  z Koala.resetcCs||jdS)z%Releases resources acquired by Koala.N)rhrcrrrrrksz Koala.deletecCr)z)Audio sample rate accepted by `.process`.)rtrrrr sample_ratezKoala.sample_ratecCr)z9Number of audio samples per frame expected by `.process`.)rvrrrrr rzKoala.frame_lengthcCr)z Delay in samples. If the input and output of consecutive calls to `.process()` are viewed as two contiguous streams of audio data, this delay specifies the time shift between the input and output stream.)rmrrrrrszKoala.delay_samplecCr)zVersion.)ryrrrrversionrz Koala.versioncCsxtt}t}|t|t|}||jjur |j|ddt}t |j D] }| || dq(| ||S)NzUnable to get Koala error state)rrO)r^rXr_r]rdrIr=rer rrlappendrxra)rmessage_stack_refmessage_stack_depthr}rrrrrrf s   zKoala._get_error_stack)rN),rrr__doc__rrIr>r$r?r'r@r(rAr)rBr*rCr+rDr,rEr-rFr.rGr/rHr0re StructurerJr r rintrrrkr!rrrrrfrrrrr1VsJ [ r1) r1r-r.r0r/rr'r(r+r*r$r,r))rRctypesenumrtypingr Exceptionrr$r'r(r)r*r+r,r-r.r/r0objectr1__all__rrrrs&    [