ó ßüÚ\c@`sFddlmZmZmZddlmZdefd„ƒYZdS(i(tprint_functiontdivisiontabsolute_importi(tSchedulerPlugint GraphLayoutcB`s8eZdZd„Zddd„Zd„Zd„ZRS(s= Dynamic graph layout during computation This assigns (x, y) locations to all tasks quickly and dynamically as new tasks are added. This scales to a few thousand nodes. It is commonly used with distributed/bokeh/scheduler.py::GraphPlot, which is rendered at /graph on the diagnostic dashboard. cC`sìi|_i|_i|_||_i|_i|_d|_d|_d|_g|_ g|_ g|_ g|_ g|_ |j|ƒ|jjrèd„|jjƒDƒ}d„|jjƒDƒ}|j|jd|d|ƒndS(NicS`s8i|].\}}g|jD]}|j^q|“qS((t dependenciestkey(t.0tkttstds((sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pys $s cS`s"i|]\}}|j|“qS((tpriority(RRR ((sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pys 's RR (txtyt collisiont schedulertindext index_edgetnext_yt next_indextnext_edge_indextnewt new_edgest state_updatestvisible_updatestvisible_edge_updatest add_pluginttaskstitemst update_graph(tselfRRR ((sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pyt__init__s*                c `sXt|d‡fd†dtƒ}x0|rS|jƒ}|ˆjks$|ˆjkrZq$n||}|r"t‡fd†|DƒƒsÁ|j|ƒ|jt|d‡fd†dtƒƒq$q@t‡fd†|Dƒƒ‰t‡‡‡fd†|Dƒƒ}t ‡fd†|Dƒƒd } nd } ˆj }ˆj d 7_ | |fˆj kr¡| |} } ˆj | |f\} }|d 7}| |fˆj | | f-ttreversec3`s|]}|ˆjkVqdS(N(R (Rtdep(R(sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pys 4sc`sˆj|dƒS(Ni(R (R(R (sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pyR!7R"c3`s%|]}tˆj|jƒVqdS(N(tlenRt dependents(RR$(R(sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pys <sc3`s4|]*}ˆj|tˆj|jƒˆVqdS(N(R R%RR&(RR$(RRt total_deps(sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pys ?sc3`s|]}ˆj|VqdS(N(R (RR$(R(sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pys Bsiigš™™™™™¹?(tsortedtTruetpopR RtalltappendtextendtsumtmaxRRR RRRRRR(RRRR tkwargststackRtdepsR R told_xtold_yR$tedge((R RRR'sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pyR,sJ!    " #       c O`sJ|dkr,|jj|j||fƒn|jj|j|dfƒ|jj|}xK|jD]@}||jf}|jj|j j ||jfƒdfƒqcWx<|j D]1}|jj|j j |j|fƒdfƒq±Wy"|j |j ||j|f=Wntk rnXx'|j |j|jgD] } | |=q5WdS(Nt forgottentFalse(RR,RRRRR&RRRR*RRR R tKeyError( RRtstarttfinishtargsR0ttaskR$R5t collection((sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pyt transition[s"   & &" cC`sôg|_g|_g|_g|_g|_i|_d|_i|_d|_xœ|j D]‘}|j|j|<|jd7_|jj |ƒxY|j j |j D]D}|j|f}|j|j|<|jd7_|jj |ƒq¤Wq[WdS(s® Reset the index and refill new and new_edges From time to time GraphPlot wants to remove invisible nodes and reset all of its indices. This helps. iiN(RRRRRRRRRR R,RRRR(RRR$R5((sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pyt reset_indexss$         N(t__name__t __module__t__doc__RtNoneRR>R?(((sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pyRs  / N(t __future__RRRtpluginRR(((sClib/python2.7/site-packages/distributed/diagnostics/graph_layout.pyts