B ]t\"@s<dZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z yddl m Z Wn ek r|ddlm Z YnXddlmZmZddlmZddlmZmZmZdd lmZmZmZmZd d Zd d ddddddZGdddeZ GdddeZ!GdddeZ"Gddde#Z$ej%GdddeZ&dS)z+Interactive figures in the IPython notebook) b64encodeN)display JavascriptHTML)Comm)rcParamsis_interactive)Gcf)_BackendFigureCanvasBaseNavigationToolbar2)FigureCanvasWebAggCoreFigureManagerWebAggNavigationToolbar2WebAgg TimerTornadocCshg}xCs zNavigationIPy.))ZDownloadz Download plotr"r"N)__name__ __module__ __qualname__r Z toolitemsrrrrr#@sr#csreZdZeZddZddZddZddZe d d Z e dfd d Z ddZ ddZddZddZZS)FigureManagerNbAggcCsd|_t|||dS)NF)_shownr__init__)selfrrrrrr-NszFigureManagerNbAgg.__init__cCstttdS)N)rrr+get_javascript)r.rrr display_jsRszFigureManagerNbAgg.display_jscCs,|js||n |jd|_dS)NT)r,r0 _create_commr draw_idle)r.rrrshowXs   zFigureManagerNbAgg.showcCsd|_|dS)zJ A special method to re-show the figure in the notebook. FN)r,r3)r.rrrreshow`szFigureManagerNbAgg.reshowcCs t|jS)N)boolr)r.rrr connectedhszFigureManagerNbAgg.connectedNcsV|dkrt}n|}tj|d|ttjdj dd|dkrR| SdS)N)streamzweb_backend/js/nbagg_mpl.jszutf-8)encoding) ioStringIOsuperr/writepathlibZPath__file__parentZ read_textgetvalue)clsr7output) __class__rrr/ls  z!FigureManagerNbAgg.get_javascriptcCst|}|||S)N) CommSocketZadd_web_socket)r.commrrrr1ys zFigureManagerNbAgg._create_commcCs2|dxt|jD] }|qW|dS)Nclose)Z _send_eventlistron_closeclearup_closed)r.rErrrdestroy~s  zFigureManagerNbAgg.destroycCs.dd|jD|_t|jdkr*|jdS)zClear up any closed Comms.cSsh|]}|r|qSr)is_open)r%socketrrr sz4FigureManagerNbAgg.clearup_closed..rN)rrr close_event)r.rrrrIsz!FigureManagerNbAgg.clearup_closedcsfdd|jD|_dS)Ncsh|]}|jjks|qSr)rEcomm_id)r%rL)rOrrrMsz1FigureManagerNbAgg.remove_comm..)r)r.rOr)rOr remove_commszFigureManagerNbAgg.remove_comm)N)r(r)r*r#Z ToolbarClsr-r0r3r4propertyr6 classmethodr/r1rJrIrP __classcell__rr)rCrr+Ks  r+c@seZdZddZdS)FigureCanvasNbAggcOs t||S)N)r)r.argskwargsrrr new_timerszFigureCanvasNbAgg.new_timerN)r(r)r*rWrrrrrTsrTc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)rDaL Manages the Comm connection between IPython and the browser (client). Comms are 2 way, with the CommSocket being able to publish a message via the send_json method, and handle a message with on_message. On the JS side figure.send_message and figure.ws.onmessage do the sending and receiving respectively. csd__tt_ttdjytddjid_Wnt k r`t dYnXj j jd_ fdd}j|dS) Nz
matplotlibid)datazTUnable to create an IPython notebook Comm instance. Are you in the IPython notebook?Fcs$d_|dddS)NTcontentrO) _ext_closerPrI)Z close_message)rr.rr _on_closesz&CommSocket.__init__.._on_close)supports_binaryrstruuidZuuid4rrrrEAttributeError RuntimeErrorZon_msg on_messager\rH)r.rr]r)rr.rr-szCommSocket.__init__cCs|jp |jj S)N)r\rEZ_closed)r.rrrrKszCommSocket.is_opencCs0|r,y|jWntk r*YnXdS)N)rKrErFKeyError)r.rrrrHs zCommSocket.on_closecCs|jdt|idS)NrZ)rEsendjsondumps)r.r[rrr send_jsonszCommSocket.send_jsoncCs,t|d}d|}|jd|idS)Nasciizdata:image/png;base64,{0}rZ)rdecoderrEre)r.ZblobrZZdata_urirrr send_binarys zCommSocket.send_binarycCsZt|dd}|ddkr2||jn$|ddkrJ|d|_n |j|dS)Nr[rZtypeclosingr^value)rfloadsrHrrIr^Z handle_json)r.messagerrrrcs    zCommSocket.on_messageN) r(r)r*__doc__r-rKrHrhrkrcrrrrrDs  rDc@s8eZdZeZeZeddZeddZ eddZ dS) _BackendNbAggcsBt|}t|}tr*||j|dfdd|S)NrNcs tS)N)r rJ)Zevent)rrrsz?_BackendNbAgg.new_figure_manager_given_figure..)rTr+rr3rr2Z mpl_connect)rrrrr)rrnew_figure_manager_given_figures  z-_BackendNbAgg.new_figure_manager_given_figurecCs |dS)N)r3)rrrrtrigger_manager_drawsz"_BackendNbAgg.trigger_manager_drawcOsnddlm}|}|sdSt}xF|D]>}|t|drL|j|j|s(||j kr(|j |q(WdS)Nr)r _cidgcf) matplotlib._pylab_helpersr rrr3hasattrrZmpl_disconnectrvrremove)rUrVr Zmanagers interactiverrrrr3s   z_BackendNbAgg.showN) r(r)r*rTZ FigureCanvasr+Z FigureManager staticmethodrtrur3rrrrrrs  rr)'rqbase64rr9rfosr=r`ZIPython.displayrrrZipykernel.commr ImportErrorZIPython.kernel.commrXrrrwr Zmatplotlib.backend_basesr r r Z'matplotlib.backends.backend_webagg_corer rrrr r$r#r+rTobjectrDZexportrrrrrrs:   GI