U & d*@sddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z GdddeZe ejd e_Gd d d eZGd d d eZddZdS)N)tqdm)ffmpeg_audiowrite)Clip)requires_duration)deprecated_version_ofextensions_dictc @sNeZdZdZdddZedddZedd d Zdd d ZedddZ dS) AudioClipaN Base class for audio clips. See ``AudioFileClip`` and ``CompositeSoundClip`` for usable classes. An AudioClip is a Clip with a ``make_frame`` attribute of the form `` t -> [ f_t ]`` for mono sound and ``t-> [ f1_t, f2_t ]`` for stereo sound (the arrays are Numpy arrays). The `f_t` are floats between -1 and 1. These bounds can be trespassed wihtout problems (the program will put the sound back into the bounds at conversion time, without much impact). Parameters ----------- make_frame A function `t-> frame at time t`. The frame does not mean much for a sound, it is just a float. What 'makes' the sound are the variations of that float in the time. nchannels Number of channels (one or two for mono or stereo). Examples --------- >>> # Plays the note A (a sine wave of frequency 440HZ) >>> import numpy as np >>> make_frame = lambda t: 2*[ np.sin(440 * 2 * np.pi * t) ] >>> clip = AudioClip(make_frame, duration=5) >>> clip.preview() NcCsht||dk r||_|dk rP||_|d}t|drJtt||_nd|_|dk rd||_ ||_ dS)Nr__iter__) r__init__fps make_frame get_framehasattrlenlist nchannelsdurationend)selfr rr Zframe0r;/tmp/pip-unpacked-wheel-0k2etfqj/moviepy/audio/AudioClip.pyr /s   zAudioClip.__init__Fc cs|dkr|j}t|}|dk r,t||}t||j}||d}tjd||ddtd} |jtt |dD]Z} | | d| | } | |kst d|t | | | | d} |j | ||||dVqrdS) zK Iterator that returns the whole sound array of the clip by chunks Nr rT)ZendpointZdtype)chunk?)nbytesquantizer buffersize) r proglogZdefault_bar_loggerintrnpZlinspaceZiter_barrrangeAssertionErrorarange to_soundarray) r chunksizeZchunk_durationr rrloggerZ totalsizeZnchunksZposposisizettrrr iter_chunks@s       zAudioClip.iter_chunksPc Cs|dkr|j}|jdkrtjntj}d||}|dkrl|j|krX||j||d|dStd|jd|}||}|rt dt d|}dd d d |} dd |d | | }|S)a Transforms the sound into an array that can be played by pygame or written in a wav file. See ``AudioClip.preview``. Parameters ------------ fps Frame rate of the sound for the conversion. 44100 for top quality. nbytes Number of bytes to encode the sound: 1 for 8bit sound, 2 for 16bit, 4 for 32bit sound. Nrr)r rrr%rgGzgGz?Zint8Zint16Zint32)r rr ) r rr ZvstackZhstackrr*r#rmaximumZminimumastype) rr)r rrrZstackerZ max_durationZ snd_arrayZinttyperrrr$Xs"    zAudioClip.to_soundarraycCsj|o |jdk}|r tddgnd}|j||dD]2}|rRt|t|jddnt|t|}q2|S)Nrr)r%r&)Zaxis)rr arrayr*r.absmax)rZstereor%r&Zmaxirrrr max_volumes 0zAudioClip.max_volumeTbarc Cs|s|jsd}n|j}|dkrptjtj|\} } zt| dddd}Wntk rntdYnXt||||||||| || d S)a, Writes an audio file from the AudioClip. Parameters ----------- filename Name of the output file fps Frames per second. If not set, it will try default to self.fps if already set, otherwise it will default to 44100 nbytes Sample width (set to 2 for 16-bit sound, 4 for 32-bit sound) codec Which audio codec should be used. If None provided, the codec is determined based on the extension of the filename. Choose 'pcm_s16le' for 16-bit wav and 'pcm_s32le' for 32-bit wav. bitrate Audio bitrate, given as a string like '50k', '500k', '3000k'. Will determine the size and quality of the output file. Note that it mainly an indicative goal, the bitrate won't necessarily be the this in the output file. ffmpeg_params Any additional parameters you would like to pass, as a list of terms, like ['-option1', 'value1', '-option2', 'value2'] write_logfile If true, produces a detailed logfile named filename + '.log' when writing the file verbose Boolean indicating whether to print infomation logger Either 'bar' or None or any Proglog logger iDNr codecrzoMoviePy couldn't find the codec associated with the filename. Provide the 'codec' parameter in write_audiofile.)r6bitrate write_logfileverbose ffmpeg_paramsr&) r ospathsplitextbasenamerKeyError ValueErrorr) rfilenamer rrr6r7r:r8r9r&nameextrrrwrite_audiofiles$. zAudioClip.write_audiofile)NNN)NNNFrN)NNFrr+)Fr+N) Nrr4NNNFTr5) __name__ __module__ __qualname____doc__r rr*r$r3rDrrrrr s(!   / r to_audiofilec@seZdZdZddZdS)AudioArrayClipa( An audio clip made from a sound array. Parameters ----------- array A Numpy array representing the sound, of size Nx1 for mono, Nx2 for stereo. fps Frames per second : speed at which the sound is supposed to be played. csRt|_|_dt||_fdd}|_ttd_ dS)Nrcst|tjrZj|t}|dk|tjk@}tt|df}j||||<|Stj|}|dks~|tjkrdjdSj|SdS)zc complicated, but must be able to handle the case where t is a list of the form sin(t) rrN) isinstancer ndarrayr r/rrr0zeros)tZ array_indsZin_arrayresultr'rrrr s z+AudioArrayClip.__init__..make_framer) rr r0r rrr rrr)rr0r r rrPrr s  zAudioArrayClip.__init__NrErFrGrHr rrrrrJsrJc@seZdZdZddZdS)CompositeAudioClipay Clip made by composing several AudioClips. An audio clip made by putting together several audio clips. Parameters ------------ clips List of audio clips, which may start playing at different times or together. If all have their ``duration`` attribute set, the duration of the composite clip is computed automatically. csrt|_ddjD}tddjD_tdd|Ds\t|_t|_fdd}|_dS)NcSsg|] }|jqSr)r.0crrr sz/CompositeAudioClip.__init__..cSsg|] }|jqSr)rrSrrrrVscSsg|] }|dkqS)Nr)rTerrrrV scshfddjD}fddtj|D}ttjrPttjf}n tj}|t|S)Ncsg|]}|qSr)Z is_playingrSrNrrrV&szCCompositeAudioClip.__init__..make_frame..cs6g|].\}}|dk r||jt|gjqS)F)rstartr r0T)rTrUpartrXrrrV(s) clipsziprKr rLrMrrsum)rNZ played_partsZsoundsZzerorPrXrr $s    z/CompositeAudioClip.__init__..make_frame) rr r\r2ranyrrr )rr\Zendsr rrPrr s    zCompositeAudioClip.__init__NrQrrrrrR srRcCshdd|D}tdg|}ddt||D}t||d}dd|D}|r^t|nd|_|S)zK The clip with the highest FPS will be the FPS of the result clip. cSsg|] }|jqSr)rrSrrrrV;sz*concatenate_audioclips..rcSsg|]\}}||qSr)Z set_start)rTrUrNrrrrV=scSsg|]}t|ddr|jqS)r N)getattrr rSrrrrVAs N)r Zcumsumr]rRZ set_durationr2r )r\Z durationsr)ZnewclipsrOZfpssrrrconcatenate_audioclips7srb)r;Znumpyr rrZ#moviepy.audio.io.ffmpeg_audiowriterrZ moviepy.CliprZmoviepy.decoratorsrZ moviepy.toolsrrrrDrIrJrRrbrrrrs    J..