o i@sddlZddlZddlZddlmZmZmZddlZe dkr,ej ej e dZn(e dkr>ej ej e dZne dkrPej ej e dZned ej esaed eeeZejgej_ejgej_ejeejeejgej_ejej_ejej_ejej_eZd Zed eZej Z!d ejfddZ"dejfddZ#dejdej$d eej$ejffddZ%deejeejfdej$d eej$ej$ffddZ&dS)N)ListTupleUnionDarwinzlibrnnoise.dylibWindowsz rnnoise.dllLinuxz librnnoise.sozUnsupported operating systemzRnNoise library not found: iireturncCs tdS)z~ Create a new RNNoise denoising state. Returns: ctypes.c_void_p: Pointer to the RNNoise denoising state. N)librnnoise_creater r E/home/ubuntu/.local/lib/python3.10/site-packages/pyrnnoise/rnnoise.pycreate1s r statecCst|dS)z Destroy an RNNoise denoising state. Args: state (ctypes.c_void_p): Pointer to the RNNoise denoising state to destroy. N)r rnnoise_destroy)rr r r destroy;srframecCs|jtjtjfvr|dkr|dkr|dt}|jtks#J|tj}t|}|t kr;t |dt |f}|j t t j}t|||}|td||fS)aA Process a single frame of audio data with RNNoise. Args: state (ctypes.c_void_p): Pointer to the RNNoise denoising state. frame (np.ndarray): Input audio frame, should be a 1D numpy array. Returns: Tuple[np.ndarray, ctypes.c_float]: Processed audio frame and speech probability. gg?irN)dtypenpfloat32float64minmaxastypeDTYPElen FRAME_SIZEpadctypesdata_asPOINTERc_floatr rnnoise_process_frame)rr frame_sizeptr speech_probr r r process_mono_frameEs*  r%statescCsl|jdkr t||S|jdksJt||jdksJddt||D}t|\}}t|t|fS)a Process a frame of audio data with RNNoise, supporting both mono and stereo. Args: states (Union[ctypes.c_void_p, List[ctypes.c_void_p]]): Pointer to the RNNoise denoising state or a list of states for each channel. frame (np.ndarray): Input audio frame, can be 1D (mono) or 2D (stereo). Returns: Tuple[np.ndarray, np.ndarray]: Processed audio frame and speech probabilities. rcSsg|] \}}t||qSr )r%).0r mono_framer r r nsz!process_frame..)ndimr%rshapeziprvstack)r&r processedframes speech_probsr r r process_frame\s  r3)'rosplatformtypingrrrnumpyrsystempathjoindirname__file__ LIBRNNOISEOSErrorexistsCDLLr c_void_pr argtypesrrr r!restypec_intrnnoise_get_frame_sizer SAMPLE_RATE FRAME_SIZE_MSint16rr rndarrayr%r3r r r r sL              &