U td(@s@dZddlmZddlmZddlmZdZGdddeZdS) zGThis module provides implementation for a Cairo-specific matrix drawer.)islice)pi)AbstractCairoDrawer)CairoMatrixDrawercs(eZdZdZfddZddZZS)rz)Default Cairo drawer object for matrices.cstj|dddS)zConstructs the vertex drawer and associates it to the given palette. @param context: the context on which we will draw N)bbox)super__init__)selfcontext __class__T/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/igraph/drawing/cairo/matrix.pyrszCairoMatrixDrawer.__init__c& s|j}|j}|dd}|_|dd}|dkr:td|dkrJtdt|dd}t|dd}|d d } |d } |d | } |d } |dt| dkrtd| dkrd} | dkr| dk r| } | dk rddt| |j D} t | |j kr| dg|j t | | dk rfddt| |j D} t | |j krf| dg|j t | | dkrtd} | dkr|} t | tr|t} | dk rt | |std| dk r| j|jkrtd|j| dk s| dk rV|jdd} | dk r&tfdd| D| }nd}| dk rPtfdd| D| }nd}n d\}}d} t|j|}t|j|}||jd }||jd}|d!drt||t||}}||jd ||jd}}|j|j||d"}|j|j||d"}||7}||7}| dkrV||}}|}t |d t||}nd#\}}|d$|ksz|d$|kr~d%}|dkr||n |d |d%d%d%| dk r||}}| D]N}||\}}}}}}|||| |||d"||||7}q| dk r|||||t d&d'\}}| D]J}||\}}}}}}||| |||d"||||7}qN| ||}}| dkrd(d)} n|j!} |D]}!|!D]}"|"dkr||7}q| d kr|"r |d%d%d%n|dddn8| dkrTt"|"||}#|#dkrDd}#|j#||#|$|||||dkr| |d*d*d*|%n| |%||7}q|||}}q| dk r||}}|d%d%d%| j&D]}!t'd+rfd,d|!D} nfd-d|!D} | D]R}"||"d$d.\}$}%||||%d"|||$d"||"||7}q|||}}q|dkr|||d%d%d%|$||||jd ||jd|%dS)/a Draws the given Matrix in a Cairo context. @param matrix: the igraph.Matrix to plot. It accepts the following keyword arguments: - C{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). - C{palette}: the palette that can be used to map integer color indices to colors when drawing vertices - 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. rNpalettez bbox is required for Cairo plotsz#palette is required for Cairo plots grid_widthg? border_widthstyleboolean row_names col_namesvalues value_format)rrnoneNz invalid stylercSsg|] }t|qSr str.0namer r r ksz*CairoMatrixDrawer.draw..cSsg|] }t|qSr rrr r rrosFTz,values must be None, False, True or a matrixz"values must be a matrix of size %s csg|]}|dqSr!r rster rrsrcsg|]}|dqSr"r r#r%r rrs)rrZsquareg@)rr')r)r)cSsdS)Nr r r r rz(CairoMatrixDrawer.draw..g?__call__csg|] }|qSr r ritemrr rrscsg|] }|qSr r r.r0r rrs)(r r popr ValueErrorfloatgetrrZ_nrowlenextendZ_ncol isinstancelist TypeErrorshapeZ text_extentsmaxwidthheightminlefttopZset_line_widthZset_source_rgbmove_toZ show_textsave translaterotaterrestoreZ fill_preserveintZset_source_rgbaZ rectanglestrokedatahasattr)&r matrixkwdsr ZMatrixrrrrrrrrZ space_widthZmax_row_name_widthZmax_col_name_width total_widthZ total_heightZdxZdyoxZoymimaZ color_offsetZ color_ratioxyheading_hZxafillrowr/ZcidxthZtwr )r&rrdraws;                                         $    $zCairoMatrixDrawer.draw)__name__ __module__ __qualname____doc__rrY __classcell__r r r rr s rN) r] itertoolsrmathrZigraph.drawing.cairo.baser__all__rr r r rs