o i!@sRddlZddlmZddlmZddlmZmZddl m Z m Z GdddZ dS) N) FfiHandle)audio_frame_pb2)_ensure_compatible_buffer get_address)AnyUnionc@s:eZdZdZdddeeeefdedededee e e fdfd df d d Z e dededed dfd d Ze dejd dfddZd ejfddZed e e e ffddZed efddZed efddZed efddZed efddZed efddZd efdd Zd e fd!d"Zed#e fd$d%ZdS)& AudioFramez A class that represents a frame of audio data with specific properties such as sample rate, number of channels, and samples per channel. The format of the audio data is 16-bit signed integers (int16) interleaved by channel. N)userdatadata sample_rate num_channelssamples_per_channelr returncCst|tr |d}t|}||ttj}t|}||kr$td|ttjdkr2td||_ ||_ ||_ ||_ |durGi|_ dS||_ dS)aG Initialize an AudioFrame instance. Args: data (Union[bytes, bytearray, memoryview]): The raw audio data, which must be at least `num_channels * samples_per_channel * sizeof(int16)` bytes long. sample_rate (int): The sample rate of the audio in Hz. num_channels (int): The number of audio channels (e.g., 1 for mono, 2 for stereo). samples_per_channel (int): The number of samples per channel. Raises: ValueError: If the length of `data` is smaller than the required size. BzIdata length must be >= num_channels * samples_per_channel * sizeof(int16)rz/data length must be a multiple of sizeof(int16)N) isinstance memoryviewcastrctypessizeofc_int16len ValueError_data _sample_rate _num_channels_samples_per_channel _userdata)selfr r r rr min_sizedata_lenr!Y/var/www/html/livekit_bhavya/venv/lib/python3.10/site-packages/livekit/rtc/audio_frame.py__init__s  zAudioFrame.__init__cCs*||ttj}t|}t||||S)a Create a new empty AudioFrame instance with specified sample rate, number of channels, and samples per channel. Args: sample_rate (int): The sample rate of the audio in Hz. num_channels (int): The number of audio channels (e.g., 1 for mono, 2 for stereo). samples_per_channel (int): The number of samples per channel. Returns: AudioFrame: A new AudioFrame instance with uninitialized (zeroed) data. )rrr bytearrayr )r r rsizer r!r!r"createLszAudioFrame.create owned_infocCsP|j}|j|j}tj||j}t|}t|j j  t ||j |j|jSN)infor rrr from_addressdata_ptrr$rhandleiddisposer r )r'r)r%cdatar r!r!r"_from_owned_info^s  zAudioFrame._from_owned_infocCs0t}t|j|_|j|_|j|_|j|_|Sr() proto_audioAudioFrameBufferInforrr+r r r)r audio_infor!r!r" _proto_infogs  zAudioFrame._proto_infocC|jS)zH Returns the user data associated with the audio frame. )rrr!r!r"r oszAudioFrame.userdatacCst|jddS)z Returns a memory view of the audio data as 16-bit signed integers. Returns: memoryview: A memory view of the audio data. rh)rrrr6r!r!r"r vszAudioFrame.datacCr5)zw Returns the sample rate of the audio frame. Returns: int: The sample rate in Hz. )rr6r!r!r"r zAudioFrame.sample_ratecCr5)z Returns the number of channels in the audio frame. Returns: int: The number of audio channels (e.g., 1 for mono, 2 for stereo). )rr6r!r!r"r r8zAudioFrame.num_channelscCr5)z Returns the number of samples per channel. Returns: int: The number of samples per channel. )rr6r!r!r"rr8zAudioFrame.samples_per_channelcCs |j|jS)z Returns the duration of the audio frame in seconds. Returns: float: The duration in seconds. )rr r6r!r!r"durations zAudioFrame.durationc Csddl}ddl}|9}||d}||j|d||j| |j Wdn1s5wY| WdS1sHwYdS)z Convert the audio frame data to a WAV-formatted byte stream. Returns: bytes: The audio data encoded in WAV format. rNwb) waveioBytesIOopen setnchannelsr setsampwidth setframerater writeframesrgetvalue)rr<r=wav_filewavr!r!r" to_wav_bytess    $zAudioFrame.to_wav_bytesc Cs(d|jd|jd|jd|jdd S)Nzrtc.AudioFrame(sample_rate=z, num_channels=z, samples_per_channel=z , duration=z.3f))r r rr9r6r!r!r"__repr__s zAudioFrame.__repr___c sddlm}ddldtddffdd }|j|||||| || || d| |g| || fd d d S) Nr) core_schemavaluerr csXt|tr|St|tr|d}t|tr(t|d|d|d|ddStd)Nrr r r rr r r rzInvalid type for AudioFrame)rr tupledict b64decode TypeError)rLbase64r!r"validate_audio_frames    zEAudioFrame.__get_pydantic_core_schema__..validate_audio_framerMcs"|jd|j|j|jdS)Nzutf-8rM) b64encoder decoder r r)instancerRr!r"s z9AudioFrame.__get_pydantic_core_schema__..) json_schema python_schema serialization) pydantic_corerKrSrjson_or_python_schema chain_schemamodel_fields_schema model_field str_schema int_schema no_info_plain_validator_function$plain_serializer_function_ser_schema)clsrJrKrTr!rRr"__get_pydantic_core_schema__s,      z'AudioFrame.__get_pydantic_core_schema__) __name__ __module__ __qualname____doc__rbytesr$rintrOstrrr# staticmethodr&r1OwnedAudioFrameBufferr0r2r4propertyr r r r rfloatr9rGrI classmethodrfr!r!r!r"r sJ  .     r ) r _ffi_clientr_protorr1_utilsrrtypingrrr r!r!r!r"s