qiXddlmZddlmZddlmZddgZd dZ d d dZy ) ) annotations) AsyncIterator) AudioFramecombine_audio_framessine_wave_generatorcBt|ts|S|s td|dj}|dj}d}d}|D]}}|j|k7rtd|d|j|j|k7rtd|d|j|t |j z }||jz }t|}d}|D]<}|j jd}||||t |z|t |z }>t||||S)a6 Combines one or more `rtc.AudioFrame` objects into a single `rtc.AudioFrame`. This function concatenates the audio data from multiple frames, ensuring that all frames have the same sample rate and number of channels. It efficiently merges the data by preallocating the necessary memory and copying the frame data without unnecessary reallocations. Args: buffer: A single `rtc.AudioFrame` or a list of `rtc.AudioFrame` objects to be combined. Returns: rtc.AudioFrame: A new `rtc.AudioFrame` containing the combined audio data. Raises: ValueError: If the buffer is empty. ValueError: If frames have differing sample rates. ValueError: If frames have differing numbers of channels. Example: >>> frame1 = rtc.AudioFrame( ... data=b"", sample_rate=48000, num_channels=2, samples_per_channel=1 ... ) >>> frame2 = rtc.AudioFrame( ... data=b"", sample_rate=48000, num_channels=2, samples_per_channel=1 ... ) >>> combined_frame = combine_audio_frames([frame1, frame2]) >>> combined_frame.data b'' >>> combined_frame.sample_rate 48000 >>> combined_frame.num_channels 2 >>> combined_frame.samples_per_channel 2 zbuffer is emptyrzSample rate mismatch: expected z, got z!Channel count mismatch: expected b)data sample_rate num_channelssamples_per_channel) isinstancelist ValueErrorr r lenr r bytearraycastr) bufferr r total_data_lengthtotal_samples_per_channelframer offset frame_datas C/opt/pipecat/venv/lib/python3.12/site-packages/livekit/rtc/utils.pyrr sYL fd # *++)''K!9))L ! ?    +1+fUEVEVDWX     -3L>HZHZG[\  S_,!U%>%>>! ? & 'D F"ZZ__S) 2<Vfs:./#j/!"  !5  cK ddl}|dz}t||z|z}|j||z }t |D]i}|||z|z z} ||j d|j z|z| zz} |j| dz} t| j|d|} | ky#t$r tdwxYww)a Generate sine wave audio frames. Useful for testing audio pipelines and generating test signals. Args: freq: Frequency of the sine wave in Hz. duration: Duration of the audio in seconds. sample_rate: Sample rate in Hz (default: 48000). amplitude: Amplitude of the sine wave, range [0.0, 1.0] (default: 0.3). Yields: AudioFrame: Audio frames containing sine wave data. Example: >>> import asyncio >>> async def generate_audio(): ... async for frame in sine_wave_generator(440, 1.0): ... print(f"Generated frame with {frame.samples_per_channel} samples") >>> asyncio.run(generate_audio()) rNzMnumpy is required for sine_wave_generator. Install it with: pip install numpy ir) numpy ImportErrorintarangerangesinpiint16rtobytes) freqdurationr amplitudenp blocksize total_framest_frameitsignal signal_int16rs rrrZs6  r!I;.9<=Lii "[0G <  a)mk1 1RVVAI$4q$899xx/   "          [   sCB*B"C*B??CN)rzAudioFrame | list[AudioFrame]returnr)ig333333?) r)floatr*r5r r"r+r5r4zAsyncIterator[AudioFrame]) __future__rtypingr audio_framer__all__rrrrr;s]" # "#8 9Ld 0 000 0  0r