i!:dZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z de de fd Zd e de fd Zd e de fd Zde de fdZe fde deeefdefdZdeeefde fdZe fde dedefdZe fde dedefdZGddZdeeeefdeeeffdZy)zd Audio Processing Utilities Provides audio format conversion, resampling, and file I/O operations. N)Path)Union)SAMPLE_RATE_INSAMPLE_RATE_OUTBYTES_PER_SAMPLE BUFFER_SIZE ulaw_bytesreturnc6tj|tS)u Convert µ-law encoded audio to 16-bit PCM. Args: ulaw_bytes: µ-law encoded audio data (8kHz) Returns: 16-bit PCM audio data (8kHz) )audioopulaw2linr)r s 2/home/sas/my/fyp/ringai/ringai/audio/processing.py ulaw_to_pcmrs   J(8 99 pcm_bytesc6tj|tS)u Convert 16-bit PCM audio to µ-law encoding. Args: pcm_bytes: 16-bit PCM audio data Returns: µ-law encoded audio data )r lin2ulawr)rs r pcm_to_ulawr"s   I'7 88rpcm_8kcXtj|tdttd\}}|S)z Resample 8kHz PCM audio to 16kHz for Whisper. Args: pcm_8k: 8kHz 16-bit PCM audio data Returns: 16kHz 16-bit PCM audio data N)r ratecvrrr)rpcm_16k_s rresample_8k_to_16kr/s/  JGQ NrrcXtj|tdttd\}}|S)z Resample 16kHz PCM audio to 8kHz. Args: pcm_16k: 16kHz 16-bit PCM audio data Returns: 8kHz 16-bit PCM audio data rN)r rrrr)rrrs rresample_16k_to_8krDs/  IFA Mrpcmpath sample_ratecLt|}|jjddtjt |d5}|j d|jt|j||j|dddy#1swYyxYw)z Save PCM audio to WAV file. Args: pcm: 16-bit PCM audio data path: Output file path sample_rate: Sample rate (default: 16kHz) T)parentsexist_okwbrN) rparentmkdirwaveopenstr setnchannels setsampwidthr setframerate writeframes)rr r!wfs rsave_wavr0Ys~ :DKKdT2 3t9d #r  ()  $ s s A BB#ctjt|d5}|j|j cdddS#1swYyxYw)zu Load PCM audio from WAV file. Args: path: Input file path Returns: PCM audio data rbN)r(r)r* readframes getnframes)r r/s rload_wavr5ls? 3t9d #.r}}R]]_-...s A  Ac,t||tzz S)z Calculate duration of PCM audio in seconds. Args: pcm: PCM audio data sample_rate: Sample rate Returns: Duration in seconds )lenrrr!s rget_audio_durationr9zs s8{%55 66rcDtt||tzz dzS)z Calculate duration of PCM audio in milliseconds. Args: pcm: PCM audio data sample_rate: Sample rate Returns: Duration in milliseconds i)intr7rr8s rget_audio_duration_msr<s# CK*::;tC DDrc|eZdZdZdefdedefdZdefdZ de fd Z defd Z d Z edefd Zedefd Zy) AudioBufferz Fixed-duration audio buffer for chunk saving. Accumulates audio until buffer reaches target duration, then allows flushing for saving to disk. g? duration_secr!cb||_||_t|tz|z|_d|_y)z Initialize audio buffer. Args: duration_sec: Target buffer duration in seconds sample_rate: Sample rate of input audio rN)r?r!r;r target_size_buffer)selfr?r!s r__init__zAudioBuffer.__init__s3)&{-== LM rrc.|xj|z c_y)zAdd PCM audio to bufferNrB)rCrs raddzAudioBuffer.adds  rr cFt|j|jk\S)z'Check if buffer has reached target size)r7rBrArCs ris_fullzAudioBuffer.is_fulls4<< D$4$444rct|jd|j}|j|jd|_|S)zl Flush buffer contents. Returns: Buffered PCM data (up to target_size) N)rBrA)rCdatas rflushzAudioBuffer.flushs9||.d../||D$4$4$67  rcd|_y)zClear buffer contentsrNrFrIs rclearzAudioBuffer.clears  rc,t|jS)zCurrent buffer size in bytes)r7rBrIs rsizezAudioBuffer.sizes4<<  rcTt|j|jtzz S)z"Current buffer duration in seconds)r7rBr!rrIs rdurationzAudioBuffer.durations$4<< D$4$47G$GHHrN)__name__ __module__ __qualname____doc__rfloatr;rDbytesrGboolrJrMrOpropertyrQrSrrr>r>s.1^ U s u55 u !c!!I%IIrr> input_paths output_pathcd}|D]M} tjt|d5}||j|j z }dddO|r t ||yy#1swYxYw#t $rYwwxYw)z Merge multiple WAV files into a single file. Args: input_paths: List of input WAV file paths output_path: Output WAV file path rr2N)r(r)r*r3r4 Exceptionr0)r]r^all_pcmr r/s rmerge_audio_filesrbsG 3t9d+ :r2==99 :+&  : :   s(A1#A% A1%A. *A11 A=<A=)rWr r(ospathlibrtypingrconfigrrrr rYrrrrr*r;r0r5rXr9r<r>listrbr\rrrhs3   :E :e : 95 9U 9u*%*ET%uS$Y/c& .5d# . .7F 7E 7 7% 7:H Eu E3 EC E4I4In'4c4i(8#9'cSWiHX'r