ó ÎYc @`s‹dZddlmZmZddlZddlZddlm Z m Z m Z ddl m Z ddlmZmZddlmZmZmZddlmZmZdd lmZdd lmZmZmZdd lmZdd l m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ddd„ƒYZ,defd„ƒYZ-ee-ƒddde/de/e/d„ƒZ0dS(sùThis is the Bokeh charts interface. It gives you a high level API to build complex plot is a simple way. This is the Chord class which lets you build your Chord charts just passing the arguments to the Chart class and calling the proper functions. i(tabsolute_importtdivisionN(tcostsintpi(t Dimension(tcreate_and_buildtBuilder(t MarkerAttrt ColorAttrtCatAttr(tcolor_in_equal_spacethelp(tRange1d(tArctBeziertText(t GlyphRenderer(tColumnDataSource( tInstancetBooltStringtArraytFloattAnytSeqtEithertInttAreacB`seZdZd„ZRS(sÀ It represents an arc area. It will create a list of available points through the arc representing that area and then those points will be used as start and end for the beziers lines. cC`s€||_||_||_tj|||ƒ}g|D]}t|ƒt|ƒg^q7|_|jt|jƒks|t ‚dS(N( tn_connt start_pointt end_pointtnptlinspaceRRt free_pointstlentAssertionError(tselfRRRtfree_points_anglestangle((s>lib/python2.7/site-packages/bkcharts/builders/chord_builder.pyt__init__,s    .(t__name__t __module__t__doc__R((((s>lib/python2.7/site-packages/bkcharts/builders/chord_builder.pyR'st ChordBuildercB`sãeZdZieƒd6eƒd6eƒd6ZdgZedƒZ e e ƒZ e e ƒZ e e ƒZeƒZeƒZeƒZeƒZeeƒƒZeeeeƒeƒƒƒƒZd„Zd„Zd„Zd„Z RS( s‚ This is the Chord builder and it is in charge of plotting Chord graphs in an easy and intuitive way. Essentially, we provide a way to ingest the data, make the proper calculations and push the references into a source object. We additionally make calculations for the ranges. And finally add the needed glyphs (markers) taking the references from the source. tcolortmarkertstacktvaluescC`s?|jsdnd}t| |ƒ|_t| |ƒ|_dS(Ngš™™™™™ñ?gÍÌÌÌÌÌü?(tlabelR tx_rangety_range(R%trng((s>lib/python2.7/site-packages/bkcharts/builders/chord_builder.pyt set_rangesWscC`s@|jsö|jj|j}|jj|j}|j|ƒjƒ}|jd}tj t j ||fƒd|d|ƒ}|j s–t |ƒ|_ n|jdkrüxB|jjjƒD].\}}|||j||jcd7lib/python2.7/site-packages/bkcharts/builders/chord_builder.pytsetup\s0  * &$#' cC`s`|jjddƒ|jjddƒ|jjƒ}||jƒdt}tj|jddƒ}||d)|jƒ}gt|jdƒD]}t ||jdƒ^q‘}t j i|d d6|dd6|d6ƒ}t |ƒ|_ |jrtd }|d |dd }t j i|d 6tj|ƒ|d 6tj|ƒ|d 6t|jƒd6ƒ} t | ƒ|_ng} xHt|jdƒD]3} | jt|| |d | |d| ƒƒqŽWg} x¸t|jƒD]§\} }| | }|| }|| }xzt|ƒD]l\}}| |}xStt|ƒƒD]?}|jjƒ}|jjƒ}| j||||gƒq;WqWqÛWt j | dtƒ}ddddddg|_|jjdƒd|d<|jjdƒd|d<|jjdƒd|d<|jjdƒd|d<|j jdƒ|j jdƒjƒd|_ t |ƒ|_!dS(Ntaxisiiiiÿÿÿÿt start_anglet end_anglet line_colorgš™™™™™ñ?g@tanglesttext_xttext_yttexttdtypetstart_xtstart_ytend_xtend_ytcolorstweighttfloat64tcx0tcy0tcx1tcy1i¸ ("RGtsumtdiagonalRR RAR>tcumsumtrangeR R?R@Rt arcs_dataR1RRRBt text_dataR<Rt enumerateRIR"tpopRKR6R_tastypeR`RaRbRdtconnection_data(R%tweights_of_areastareas_in_radianstpointstareaRcRnt text_radiusRZttext_positionst all_areastitall_connectionstjtregion1RNR-Rdtktregion2RORStp1tp2tconnections_df((s>lib/python2.7/site-packages/bkcharts/builders/chord_builder.pyt process_data€sT5  7      1    '/cc`stddddddddd d d d d d d d d dddƒ }td|jd|ƒVtddddddddddddddƒ}td|jd|ƒV|jridd6dd6d d!6d"d#6}tdd$dd%d&d&d'd(|}td|jd|ƒVnd)S(*sˆUse the marker glyphs to display the arcs and beziers. Takes reference points from data loaded at the ColumnDataSource. tx0R_ty0R`tx1Raty1RbRfRgRhRit line_alphaRdRYRct data_sourcetglyphtxitytradiusit line_widthi RWRXs#000000t text_colort8ptttext_font_sizetleftt text_aligntmiddlet text_baselineR[R\R]R'RZN(RRRsRRnR1RRo(R%tbezierstarcst text_propstlabels((s>lib/python2.7/site-packages/bkcharts/builders/chord_builder.pytyield_renderersÂs>        (!R)R*R+R RR tdefault_attributest dimensionsRR0RRRnRoRsRR:R;RRCRR8RR1RRRRRGR5RUR„R›(((s>lib/python2.7/site-packages/bkcharts/builders/chord_builder.pyR,9s&             !  $ Bc K`s…||d<||d<||d<||d<||d<||d<||d` to render a chord graph from a variety of value forms. This chart displays the inter-relationships between data in a matrix. The data can be generated by the chart interface. Given a :class:`DataFrame `, select two columns to be used as arcs with `source` and `target` attributes, passing by the name of those columns. The :class:`Chord ` chart will then deduce the relationship between the arcs. The value of the connections can be inferred automatically by counting `source` and `target`. If you prefer you can assign a fixed value for all the connections with `value` simply passing by a number. A third option is to pass a reference to a third column in the :class:`DataFrame ` with the values for the connections. If you want to plot the relationships in a squared matrix, simply pass the matrix and set `square_matrix` attribute to `True`. Reference: `Chord diagram on Wikipedia `_ Args: data (:ref:`userguide_charts_data_types`): the data source for the chart. source (list(str) or str, optional): Data source to use as origin of the connection to a destination. target (list(str) or str, optional): Data source to use as destination of a connection. value (list(num) or num, optional): The value the connection should have. square_matrix (bool, optional): If square matrix, avoid any calculations during the setup. label (list(str), optional): The labels to be put in the areas. Returns: :class:`Chart`: includes glyph renderers that generate the chord Examples: .. bokeh-plot:: :source-position: above import pandas as pd from bkcharts import Chord from bokeh.io import show, output_file from bokeh.sampledata.les_mis import data nodes = data['nodes'] links = data['links'] nodes_df = pd.DataFrame(nodes) links_df = pd.DataFrame(links) source_data = links_df.merge(nodes_df, how='left', left_on='source', right_index=True) source_data = source_data.merge(nodes_df, how='left', left_on='target', right_index=True) source_data = source_data[source_data["value"] > 5] # Select those with 5 or more connections chord_from_df = Chord(source_data, source="name_x", target="name_y", value="value") output_file('chord_from_df.html') show(chord_from_df) R:R;RCR8R1txgridtygridiN(RR,tFalseR“tvisibletbelowRDtoutline_line_color( tdataRNRORCR8R1RžRŸtkwtchart((s>lib/python2.7/site-packages/bkcharts/builders/chord_builder.pytChordðs:        ((1R+t __future__RRtnumpyR tpandasR?tmathRRRtbkcharts.propertiesRtbkcharts.builderRRtbkcharts.attributesRR R tbkcharts.utilsR R t bokeh.modelsR tbokeh.models.glyphsRRRtbokeh.models.renderersRtbokeh.models.sourcesRtbokeh.core.propertiesRRRRRRRRRRR,RDR R§(((s>lib/python2.7/site-packages/bkcharts/builders/chord_builder.pyts"   @·