U Ñtœd&ã@s@dZddlmZddlmZddlmZdZGdd„deƒZd S) z?This module provides a dendrogram drawer for the Cairo backend.é)Úpi)Ústr_to_orientationé)ÚAbstractCairoDrawer)ÚCairoDendrogramDrawercs<eZdZdZ‡fdd„Zedd„ƒZdd„Zdd „Z‡Z S) rz,Default Cairo drawer object for dendrograms.cstƒ ||¡||_dS)aConstructs the drawer and associates it to the given palette. @param context: the context on which we will draw @param bbox: the bounding box within which we will draw. Can be anything accepted by the constructor of L{BoundingBox} (i.e., a 2-tuple, a 4-tuple or a L{BoundingBox} object). @param palette: the palette that can be used to map integer color indices to colors when drawing vertices N)ÚsuperÚ__init__Úpalette)ÚselfÚcontextÚbboxr ©Ú __class__©úX/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/cairo/dendrogram.pyrs zCairoDendrogramDrawer.__init__cCsl|jdks|j|dkr0| d¡\}}}}}}n | t|j|ƒ¡\}}}}}}|r`|||fS|||fS)zoCalculates the amount of space needed for drawing an individual vertex at the bottom of the dendrogram.NÚ)Ú_namesZ text_extentsÚstr)Údendror ÚhorizÚidxZ x_bearingÚ_ÚheightZ x_advancerrrÚ_item_box_sizes ÿ z$CairoDendrogramDrawer._item_box_sizecCsª|jdks|j|dkrdS| ||d|¡d}|rZ| |||¡| t|j|ƒ¡nL| ¡| ||¡| t d¡| d|¡| t|j|ƒ¡|  ¡dS)aCPlots a dendrogram item to the given Cairo context @param context: the Cairo context we are plotting on @param horiz: whether the dendrogram is horizontally oriented @param idx: the index of the item @param x: the X position of the item @param y: the Y position of the item NTrç@r) rrÚmove_toZ show_textrÚsaveÚ translateÚrotaterÚrestore)r rr rrÚxÚyrrrrÚ _plot_item,s   z CairoDendrogramDrawer._plot_itemc sÐddlm}ˆj‰ˆj}ˆjdkr8dd„tˆjƒDƒˆ_t| dd¡dd }|d k‰ˆ  ¡d }‡‡‡‡fd d„tˆjƒDƒ}|d krŠd nd‰|dkršd nd‰‡‡fdd„|Dƒ}|dgˆjd d}ˆ  ¡} ˆsfd\} } t | ƒD]*\} } | df|| <| t ||| dƒ7} qäˆj D]8\}}| d7} | ||d||dd| f¡q|d krø| d¡n’d\} } t | ƒD],\} } d| f|| <| t ||| dƒ7} qvˆj D]8\}}| d7} | | ||d||ddf¡qª|dkrø| d¡t dd„|Dƒƒ}t dd„|Dƒƒ}t|jƒt|jƒ}}d\}}ˆrX||8}|dkrn|}n||8}|dkrn|}ˆr‚||d7}n ||d7}|j||||||fddˆ ¡ˆ |j|j¡ˆ ddd¡ˆ d¡ˆrT|dkrðdnd}tˆjƒD]R} || d||| d} || d|| dd} ˆ ˆˆˆ| | | ¡qþnp|d krbdnd}tˆjƒD]R} || d|| dd} || d||| d} ˆ ˆˆˆ| | | ¡qpˆsHt ˆj ƒD]p\} \}}||\}}||\}}|| ˆj\}}ˆ ||¡ˆ ||¡ˆ ||¡ˆ ||¡ˆ ¡qÔn|t ˆj ƒD]p\} \}}||\}}||\}}|| ˆj\}}ˆ ||¡ˆ ||¡ˆ ||¡ˆ ||¡ˆ ¡qRˆ ¡dS)aj Draws the given Dendrogram in a Cairo context. @param dendro: the igraph.Dendrogram to plot. It accepts the following keyword arguments: - C{style}: the style of the plot. C{boolean} is useful for plotting matrices with boolean (C{True}/C{False} or 0/1) values: C{False} will be shown with a white box and C{True} with a black box. C{palette} uses the given palette to represent numbers by colors, the minimum will be assigned to palette color index 0 and the maximum will be assigned to the length of the palette. C{None} draws transparent cell backgrounds only. The default style is C{boolean} (but it may change in the future). C{None} values in the matrix are treated specially in both cases: nothing is drawn in the cell corresponding to C{None}. - C{square}: whether the cells of the matrix should be square or not. Default is C{True}. - C{grid_width}: line width of the grid shown on the matrix. If zero or negative, the grid is turned off. The grid is also turned off if the size of a cell is less than three times the given line width. Default is C{1}. Fractional widths are also allowed. - C{border_width}: line width of the border drawn around the matrix. If zero or negative, the border is turned off. Default is C{1}. - C{row_names}: the names of the rows - C{col_names}: the names of the columns. - C{values}: values to be displayed in the cells. If C{None} or C{False}, no values are displayed. If C{True}, the values come from the matrix being plotted. If it is another matrix, the values of that matrix are shown in the cells. In this case, the shape of the value matrix must match the shape of the matrix being plotted. - C{value_format}: a format string or a callable that specifies how the values should be plotted. If it is a callable, it must be a function that expects a single value and returns a string. Example: C{"%#.2f"} for floating-point numbers with always exactly two digits after the decimal point. See the Python documentation of the C{%} operator for details on the format string. If the format string is not given, it defaults to the C{str} function. If only the row names or the column names are given and the matrix is square-shaped, the same names are used for both column and row names. r)ÚLayoutNcSsg|] }t|ƒ‘qSr)r)Ú.0r rrrÚ ~sz.CairoDendrogramDrawer.draw..Ú orientationÚlrT)Zreversed_vertical)r'Úrlécsg|]}ˆ ˆˆˆ|¡‘qSr)r)r$r)r rrr rrr%ŠsZbtcs g|]\}}|ˆ|ˆf‘qSrr)r$r r!)ÚxgapÚygaprrr%’s)rr)Údimrrr(css|]}|dVqdS)rNr©r$ÚerrrÚ ³sz-CairoDendrogramDrawer.draw..css|]}|dVqdS)rNrr-rrrr/´sÚtbF)Zkeep_aspect_ratiogéÿÿÿÿ)Z igraph.layoutr#r r rÚrangeZ_nitemsrÚgetZ font_extentsZ_traverse_inorderÚ enumerateÚmaxZ_mergesÚappendZmirrorÚfloatÚwidthrZfit_intorrÚleftÚtopZset_source_rgbZset_line_widthr"rZline_toÚstroker)r rÚkwdsr#r r&Z font_heightZ item_boxesZlayoutZinorderr r!rÚelementZid1Zid2ZmaxwZmaxhr8rZdelta_xZdelta_yZsgnZx0Zy0Úx1Úy1Zx2Úy2r)r rrr r*r+rÚdrawDs¶4   ÿ ÿ *   *     þ              zCairoDendrogramDrawer.draw) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚ staticmethodrr"rAÚ __classcell__rrr rr s   rN) rEÚmathrZigraph.drawing.utilsrÚbaserÚ__all__rrrrrÚs