o
zi @ sr d Z ddlZddlZddlZddlZddlZddlmZ ddlZ ddl
m Z ddl m
Z
ddl mZ e e d Ze e d Zd
Zd d ZG d
d dZedkrddlmZ eddddZejddddZejddddZ ejdd ddd Z e d ejd d ddd Z e d d! Z!e!"e e!"e e!"e e!"e W d dS 1 sw Y dS dS )"z
These are utilities that allow one to embed an AudioSignal
as a playable object in a Jupyter notebook, or to play audio from
the terminal, etc.
N)NamedTemporaryFile ) templates)_close_temp_files)
format_figurez headers.htmlz widget.html.wavc C s< zdd l } W n d} Y z dd l}W | |fS td)Nr Fz8IPython must be installed in order to use this function!)ffmpyIPython ImportError)r r r L/home/ubuntu/.local/lib/python3.10/site-packages/audiotools/core/playback.py_check_imports s
r
c @ sb e Zd ZddededefddZ
dd
ededededededefddZdd ZdS ) PlayMixinNTFextdisplay return_htmlc C s
|du rt }|dsd| n|}t \}}| j}g }t|Q tdddd}|| | |j |dkrW|rWtd|dd} || |j |jdi| jdid}
|
n|} |j j | j|d } |rj|j | W d n1 stw Y |rd
|
d } | S ) a Embeds audio as a playable audio embed in a notebook, or HTML
document, etc.
Parameters
----------
ext : str, optional
Extension to use when saving the audio, by default ".wav"
display : bool, optional
This controls whether or not to display the audio when called. This
is used when the embed is the last line in a Jupyter cell, to prevent
the audio from being embedded twice, by default True
return_html : bool, optional
Whether to return the data wrapped in an HTML audio element, by default False
Returns
-------
str
Either the element for display, or the HTML string of it.
N.zw+r F)modesuffixdeletezK-write_xing 0 -codec:a libmp3lame -b:a 128k -y -hide_banner -loglevel error)inputsoutputs)dataratez )DEFAULT_EXTENSION
startswithr
sample_rater r appendwritenameFFmpegrunr Audiosrc_attr) selfr r r r r srtmpfilestmp_wav
tmp_convertedff
audio_elementr r r embed' s>
zPlayMixin.embedr 100%10pxspecshowtitle add_headers player_widthmarginplot_fnc s ddl m fdd} t \}
} d} |r/tdt|} | dt|} | j| j| t}
t |tr?t
| |}||d< |di | }| |j
}| }| td | }dd
d td D }| j|d
d}|
d| }
|
d|}
|
d|}
|
d|}
|
dt|d d}
|
dt|d d}
| j| j|
|r|r| nd}||
7 }|S dS )aE Creates a playable widget with spectrogram. Inspired (heavily) by
https://sjvasquez.github.io/blog/melnet/.
Parameters
----------
title : str, optional
Title of plot, placed in upper right of top-most axis.
ext : str, optional
Extension for embedding, by default ".mp3"
add_headers : bool, optional
Whether or not to add headers (use for first embed, False for later embeds), by default True
player_width : str, optional
Width of the player, as a string in a CSS rule, by default "100%"
margin : str, optional
Margin on all sides of player, by default "10px"
plot_fn : function, optional
Plotting function to use (by default self.specshow).
return_html : bool, optional
Whether to return the data wrapped in an HTML audio element, by default False
kwargs : dict, optional
Keyword arguments to plot_fn (by default self.specshow).
Returns
-------
HTML
HTML object.
r Nc sL t } j| ddd | d t| d}d |}|S )Ntightr ) bbox_inches
pad_inchesasciizdata:image/png;base64,{0})
ioBytesIOsavefigcloseseekbase64 b64encodereaddecodeformat)bufferdata_uritagpltr r _save_fig_to_tag s
z*PlayMixin.widget.._save_fig_to_tag PLAYER_WIDTHMARGINr/ ) g ?c s s | ] }t tjV qd S )N)randomchoicestringascii_uppercase).0_r r r s z#PlayMixin.widget..
F)r r AUDIO_SRC IMAGE_SRC
LEVELS_SRC PLAYER_IDPADDING_AMOUNTr px MAX_WIDTHr )matplotlib.pyplotpyplotr
headersreplacestrr HTMLwidget
isinstancegetattrgcfget_size_inchesdpir. r joinranger+ r# int)r$ r/ r r0 r1 r2 r3 r kwargsrG rQ r header_html widget_htmlfigpixelsrD
levels_tag player_id
audio_elemhtmlr rE r ra ` sB &
zPlayMixin.widgetc
C sv g }t |+ tddd}|| | |j t| tdddddd |jg W d
| S 1 s4w Y | S ) z
Plays an audio signal if ffplay from the ffmpeg suite of tools is installed.
Otherwise, will fail. The audio signal is written to a temporary file
and then played with ffplay.
r F)r r ffplayz-nodispz -autoexitz -hide_bannerz -loglevelerrorN)r r r r r print
subprocesscall)r$ r& r' r r r play s(
zPlayMixin.play)NTF)Nr Tr, r- r. F)__name__
__module__ __qualname__r_ boolr+ ra rx r r r r r & s2 ;
`r __main__) AudioSignalz(tests/audio/spk/f10_script4_produced.mp3 )offsetdurationWaveformwaveplotT)r3 r SpectrogramF)r r0 zWaveform + spectrogramwavespec)r3 r r0 i@ zLowpassed audioz/tmp/index.htmlw)#__doc__r= r8 rL rN rv tempfiler importlib_resources
pkg_resourcesrH r utilr r filesjoinpath read_textr] ra r r
r ry
audiotoolsr~ signal wave_html spec_html
combined_htmllow_pass lowpass_htmlopenfr r r r r s`
5
"