dj d Z ddlZddlZddlmZ ddlZddlmZ ddl m
Z
ej G d d Z G d d e
Zy)
zMediaPipe audio data. N)Optional) audio_data_c) doc_controlsc 4 e Zd ZU dZdZeed< dZee ed< y)AudioDataFormatzAudio format metadata.
Attributes:
num_channels: the number of channels of the audio data.
sample_rate: the audio sample rate.
num_channelsN sample_rate)
__name__
__module__ __qualname____doc__r int__annotations__r
r float z/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/mediapipe/tasks/python/components/containers/audio_data.pyr r s! ,!%+x%r r c : e Zd ZdZ e fdededdfdZd Z ddej d ed
eddfd Z
e ddej d e e
dd fd
Zedefd Zedefd Zedej fd Zej( dej, fd Zy) AudioDataz!MediaPipe Tasks' audio container.
buffer_length audio_formatreturnNc || _ t j || j j gt j | _ y)zInitializes the `AudioData` object.
Args:
buffer_length: the length of the audio buffer.
audio_format: the audio format metadata.
)dtypeN)
_audio_formatnpzerosr float32_buffer)selfr r s r __init__zAudioData.__init__) s6 &D88]D,>,>,K,K L"$**.DLr c : | j j d y)z2Clears the internal buffer and fill it with zeros.r N)r fillr! s r clearzAudioData.clear8 s LLar srcoffsetsizec
< t |j dk( r<| j j dk7 rlt d| j j d |j d | j j k7 r#t d| j j d |dk r t | }||z t | kD r t d| d| d t | t | t | j
k\ rL||z t | j
z
}t | j
}||||z j
| _ y |}t j | j
| d
| _ ||||z j
| j
| d d d f< y ) a Loads the audio data from a NumPy array.
Args:
src: A NumPy source array contains the input audio.
offset: An optional offset for loading a slice of the `src` array to the
buffer.
size: An optional size parameter denoting the number of samples to load
from the `src` array.
Raises:
ValueError: If the input array has an incorrect shape or if
`offset` + `size` exceeds the length of the `src` array.
r z s8* &
' ' 3x3t||$$ D=3t||#44jT\\"hJ$9:??Adl eWWT\\E6:dl!$VFTM!:!?!?!AdllE67A: r r
c |j t j } | |j d t t |j dk( rdn|j d | }|j
| |S )a Creates an `AudioData` object from a NumPy array.
Args:
src: A NumPy source array contains the input audio.
sample_rate: the optional audio sample rate.
Returns:
An `AudioData` object that contains a copy of the NumPy source array as
the data.
r r )r r
)r r )astyper r r. r r- r5 )clsr' r
objs r create_from_arrayzAudioData.create_from_arrayj sg **RZZ
C
iil$ !#))n 1syy|#% &C
Jr c | j S )z#Gets the audio format of the audio.)r r% s r r zAudioData.audio_format s r c 4 | j j d S )z#Gets the sample count of the audio.r )r r. r% s r r zAudioData.buffer_length s << a r c | j S )zGets the internal buffer.)r r% s r bufferzAudioData.buffer s <<r c 4 t j | j j | j j | j
j j t
j t j | j
j S )z2Converts the object to a ctypes audio data object.)r r
audio_dataaudio_data_size) r
AudioDataCr r r
r ctypesdata_asPOINTERc_floatr) r% s r to_ctypeszAudioData.to_ctypes sg " "&&33%%11<<&&..v~~fnn/MN )) r )r )N)r r r
r r r r" r& r ndarrayr5 classmethodr r r: propertyr r r> r do_not_generate_docsr rB rG r r r r r & s# )
'6&7
.
. $
.
. %&"$,B ::,B !,B ,B *.,B\ 8< ZZ%-e_@K , O !S ! ! bjj $$ 00 %r r )r rC dataclassestypingr numpyr ,mediapipe.tasks.python.components.containersr 1mediapipe.tasks.python.core.optional_dependenciesr dataclassr objectr r r r rT sI
E J
& & &r rr