o پi7@sdZddlmZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZmZdd lmZejZGd d d eZe ddS)z Audio Reader.)absolute_importN)c_arrayc_str) _init_api) DECORDContext) DECORDError)ndarray)cpugpu) bridge_outc@seZdZdZedddfddZddZd d Zd d Zd dZ e ddZ ddZ ddZ ddZddZddZddZddZdS) AudioReadera4Individual audio reader with convenient indexing function. Parameters ---------- uri: str Path of file. ctx: decord.Context The context to decode the file, can be decord.cpu() or decord.gpu(). sample_rate: int, default is -1 Desired output sample rate of the audio, unchanged if `-1` is specified. mono: bool, default is True Desired output channel layout of the audio. Setting `True` will return the audio as mono layout. Setting `False` will return the audio channel layout intact. rTcCsd|_t|ts J|rdnd}t|dr/t|}dt|}t||j |j |d||_n t||j |j |d||_|jdurHt d|dt |j|_ |j |_ t|j|_t|j|_t|j|_||_d|_dS)Nrrreadz{} byteszError reading z...)_handle isinstancerhasattr bytearrayrformatlen_CAPI_AudioReaderGetAudioReader device_type device_id RuntimeError_CAPI_AudioReaderGetNDArray_arrayasnumpy_CAPI_AudioReaderGetDuration _duration(_CAPI_AudioReaderGetNumSamplesPerChannel_num_samples_per_channel_CAPI_AudioReaderGetNumChannels _num_channels sample_rate _num_padding)selfurictxr$monois_monobar,G/home/ubuntu/.local/lib/python3.10/site-packages/decord/audio_reader.py__init__$s*          zAudioReader.__init__cCs |jdS)zGet length of the audio. The length refer to the shape's first dimension. In this case, the length is the number of channels. Returns ------- int The number of channels in the audio track. r)shaper&r,r,r-__len__:s zAudioReader.__len__cCs|jr t|jdSdS)N)r_CAPI_AudioReaderFreer0r,r,r-__del__DszAudioReader.__del__cCs|jdusJt|tr|t||jS|dkr ||j7}||jks)|dkr2td||jt t |j dd|fS)aGet sample at `idx`. idx is the index of resampled audio, unit is sample. Parameters ---------- idx : int or slice The sample index, can be negative which means it will index backwards, or slice of sample indices. Returns ------- ndarray Samples of shape CxS, where C is the number of channels, S is the number of samples of the index or slice. NrzIndex: {} out of bound: {}) rrslice get_batchrangeindicesr! IndexErrorrr _ndarrayr)r&idxr,r,r- __getitem__Hs  zAudioReader.__getitem__cCs<|jdusJ||}t|}tt|jdd|fS)aGet entire batch of samples. Parameters ---------- indices : list of integers A list of frame indices. If negative indices detected, the indices will be indexed from backward Returns ------- ndarray Samples of shape CxS, where C is the number of channels, S is the number of samples of the slice. N)r_validate_indiceslistr r9r:rr&r7r,r,r-r5`s zAudioReader.get_batchcCs |j|jfS)zGet shape of the entire audio samples. Returns ------- (int, int) The number of channels, and the number of samples in each channel. )r#r!r0r,r,r-r/ss zAudioReader.shapecCs|jS)zGet duration of the audio. Returns ------- double Duration of the audio in secs. )rr0r,r,r-durations zAudioReader.durationcCs|jdur t|j|_|jS)zAGet number of samples needed to pad the audio to start at time 0.N)r%%_CAPI_AudioReaderGetNumPaddingSamplesrr0r,r,r-__get_num_paddings  zAudioReader.__get_num_paddingcCsBtj|jd|dffddd|_|j||j7_|S)zPad the audio samples so that it starts at time 0. Returns ------- int Number of samples padded )rrrconstant)constant_values)nppadr_AudioReader__get_num_paddingrr$r0r,r,r- add_paddings" zAudioReader.add_paddingcCst|jdS)z.Log out the basic info about the audio stream.N)_CAPI_AudioReaderGetInforr0r,r,r-get_infoszAudioReader.get_infocCst||jS)z'Convert time in seconds to sample index)mathceilr$)r& timestampr,r,r-_time_to_sampleszAudioReader._time_to_samplecsfdd|DS)z*Convert times in seconds to sample indicescsg|]}|qSr,)rN).0rMr0r,r- sz1AudioReader._times_to_samples..r,)r& timestampsr,r0r-_times_to_samplesszAudioReader._times_to_samplescCs|jdusJtj|tjd}||dk|j7<|dks.td||dk|j||jksAtd|||jk|S)zTValidate int64 integers and convert negative integers to positive by backward searchN)dtyperzInvalid negative indices: {}zOut of bound indices: {})rrEr:int64r!allr8rr?r,r,r-r=s zAudioReader._validate_indicesN)__name__ __module__ __qualname____doc__r r.r1r3r<r5propertyr/r@rGrHrJrNrRr=r,r,r,r-r s       r zdecord.audio_reader)rY __future__rctypesnumpyrErK _ffi.baserr _ffi.functionr _ffi.ndarrayrbaserr r9r r bridger c_void_pAudioReaderHandleobjectr r,r,r,r-s        %