ó m[c@`sXdZddlmZmZmZmZddlZddlmZddl Z ddl m Z ddl m Z m Z ddlmZdd lmZdd lmZdd lmZdd lmZd efd„ƒYZdefd„ƒYZde fd„ƒYZeedeeedeeddedd„ Zejjde jeƒƒdS(uG Place a table below the x-axis at location loc. The table consists of a grid of cells. The grid need not be rectangular and can have holes. Cells are added by specifying their row and column. For the purposes of positioning the cell at (0, 0) is assumed to be at the top left and the cell at (max_row, max_col) is assumed to be at bottom right. You can add additional cells outside this range to have convenient ways of positioning more interesting grids. Author : John Gill Copyright : 2004 John Gill and John Hunter License : matplotlib license i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(txrangei(tartist(tArtisttallow_rasterization(t Rectangle(t docstring(tText(tBbox(tPathtCellcB`s˜eZdZdZddedddd„Zd„Zd„Zd„Z d „Z d „Z d „Z e d „ƒZd „Zd„Zd„Zd„ZRS(u; A cell is a Rectangle with some associated text. gš™™™™™¹?ukuwuc C`s“tj||d|d|d|d|ƒ|jtƒ|dkrJd}n||_td|dd|d d |d | ƒ|_|jjtƒdS( Ntwidththeightt edgecolort facecolorurighttxityittexttfontproperties(Rt__init__t set_clip_ontFalsetNonet_locR t_text( tselftxyRRRRtfillRtlocR((s/lib/python2.7/site-packages/matplotlib/table.pyR.s        cC`stj||ƒt|_dS(N(Rt set_transformtTruetstale(Rttrans((s/lib/python2.7/site-packages/matplotlib/table.pyR CscC`s$tj||ƒ|jj|ƒdS(N(Rt set_figureR(Rtfig((s/lib/python2.7/site-packages/matplotlib/table.pyR$HscC`s|jS(uReturn the cell Text intance(R(R((s/lib/python2.7/site-packages/matplotlib/table.pytget_textLscC`s|jj|ƒt|_dS(N(Rt set_fontsizeR!R"(Rtsize((s/lib/python2.7/site-packages/matplotlib/table.pyR'PscC`s |jjƒS(uReturn the cell fontsize(Rt get_fontsize(R((s/lib/python2.7/site-packages/matplotlib/table.pyR)TscC`sj|jƒ}|j|ƒ}xH|dkre||jƒkre|d8}|j|ƒ|j|ƒ}qW|S(u# Shrink font size until text fits. i(R)tget_required_widtht get_widthR'(Rtrenderertfontsizetrequired((s/lib/python2.7/site-packages/matplotlib/table.pytauto_set_font_sizeXs !  cC`sJ|jƒsdStj||ƒ|j|ƒ|jj|ƒt|_dS(N(t get_visibleRtdrawt_set_text_positionRRR"(RR,((s/lib/python2.7/site-packages/matplotlib/table.pyR1cs   c C`sä|j|ƒ}|j\}}}}|jjdƒ||d}|jdkrr|jjdƒ||d}nX|jdkr¥|jjdƒ|||j}n%|jjdƒ||d|j}|jj||fƒdS(um Set text up so it draws in the right place. Currently support 'left', 'center' and 'right' ucenterg@ulefturightgð?N(tget_window_extenttboundsRtset_verticalalignmentRtset_horizontalalignmenttPADt set_position( RR,tbboxtltbtwthRR((s/lib/python2.7/site-packages/matplotlib/table.pyR2oscC`s.|jj|ƒ}|j|jƒƒ}|jS(u' Get text bounds in axes co-ordinates. (RR3tinverse_transformedtget_data_transformR4(RR,R9tbboxa((s/lib/python2.7/site-packages/matplotlib/table.pytget_text_boundsˆscC`s.|j|ƒ\}}}}|dd|jS(u# Get width required for this cell. gð?g@(RAR7(RR,R:R;R<R=((s/lib/python2.7/site-packages/matplotlib/table.pyR*ŽscK`s|jj|ƒt|_dS(u&update the text properties with kwargsN(RtupdateR!R"(Rtkwargs((s/lib/python2.7/site-packages/matplotlib/table.pytset_text_props“sN(t__name__t __module__t__doc__R7R!RRR R$R&R'R)R/RR1R2RAR*RD(((s/lib/python2.7/site-packages/matplotlib/table.pyR 's$          t CustomCellcB`sieZdZdZidd6ed6dd6dd6Zd „Zed „ƒZejd „ƒZd „Z RS( uE A subclass of Cell where the sides may be visibly toggled. uBRTLuuopenucloseduBTu horizontaluRLuverticalcO`s/|jdƒ}tj|||Ž||_dS(Nu visible_edges(tpopR Rt visible_edges(RtargsRCRJ((s/lib/python2.7/site-packages/matplotlib/table.pyR¦scC`s|jS(N(t_visible_edges(R((s/lib/python2.7/site-packages/matplotlib/table.pyRJ«scC`s©|dkr|j|_n||jkr=|j||_n_xS|D]K}||jkrDtdj|dj|jƒdj|jƒƒƒ‚qDqDW||_t|_dS(Nu=Invalid edge param {}, must only be one of {} or string of {}u, ( Rt_edgesRLt _edge_aliasest ValueErrortformattjoinR!R"(Rtvaluetedge((s/lib/python2.7/site-packages/matplotlib/table.pyRJ¯s    cC`sµtjg}xC|jD]8}||jkr>|jtjƒq|jtjƒqWtj|dkrutj|d Retreive a custom cell from a given position iiu0Only tuples length 2 are accepted as coordinates(RtRuRi(RRwRoRp((s/lib/python2.7/site-packages/matplotlib/table.pyt __getitem__;s  cC`s|jS(N(RM(R((s/lib/python2.7/site-packages/matplotlib/table.pyRnEscC`s||_t|_dS(N(RMR!R"(RRR((s/lib/python2.7/site-packages/matplotlib/table.pyRnIs cC`s&|jd|jj|jjjdS(NgR@g333333ó?(tFONTSIZERdtdpiReR9R(R((s/lib/python2.7/site-packages/matplotlib/table.pyt_approx_text_heightNscC`s¨|dkr|jj}n|dkr6tdƒ‚n|jƒsFdS|jdƒ|j|ƒx+t|jƒD]}|j|j |ƒqpW|j dƒt |_ dS(NuNo renderer definedutable( RRdt_cachedRenderert RuntimeErrorR0t open_groupt_update_positionstsortedRiR1t close_groupRR"(RR,tkey((s/lib/python2.7/site-packages/matplotlib/table.pyR1Rs      cC`stgtj|jƒD]9\\}}}|dkr|dkr|j|ƒ^q}tj|ƒ}|j|jƒƒS(urGet a bbox, in axes co-ordinates for the cells. Only include those in the range (0,0) to (maxRow, maxCol)i(R_t iteritemsRiR3R tunionR>Rv(RR,RoRpRqtboxesR9((s/lib/python2.7/site-packages/matplotlib/table.pyt_get_grid_bboxfs "-cC`sÂt|jƒr|j||ƒS|jj}|dk r´gtj|jƒD]9\\}}}|dkrJ|dkrJ|j|ƒ^qJ}t j |ƒ}|j |j |j ƒifStifSdS(uUTest whether the mouse event occurred in the table. Returns T/F, {} iN(tcallablet _containsRdR}RR_R„RiR3R R…tcontainsRRR(Rt mouseeventR,RoRpRqR†R9((s/lib/python2.7/site-packages/matplotlib/table.pyRŠps  "-cC`sttj|jƒƒS(u)Return the Artists contained by the table(tlistR_t itervaluesRi(R((s/lib/python2.7/site-packages/matplotlib/table.pyt get_children„scC`s;gtj|jƒD]}|j|ƒ^q}tj|ƒS(u5Return the bounding box of the table in window coords(R_RRiR3R R…(RR,RqR†((s/lib/python2.7/site-packages/matplotlib/table.pyR3‰s+c C`sVi}i}x|tj|jƒD]h\\}}}|j|dƒ}t||jƒƒ||<|j|dƒ}t||jƒƒ||>>table.auto_set_column_width([-1,0,1]) col(Tuple): tuple of indexs >>>table.auto_set_column_width((-1,0,1)) col(int): index integer >>>table.auto_set_column_width(-1) >>>table.auto_set_column_width(0) >>>table.auto_set_column_width(1) N(titerRstAttributeErrorRkRVR!R"(RRpRq((s/lib/python2.7/site-packages/matplotlib/table.pytauto_set_column_width¯s cC`s‘g|jD]}|d|kr |^q }d}x3|D]+}|j|}t|j|ƒ|ƒ}q9Wx"|D]}|j|j|ƒqoWdS(u- Automagically set width for column. iiN(RiR‘R*t set_width(RRpR,RƒtcellsRRqtc((s/lib/python2.7/site-packages/matplotlib/table.pyt_auto_set_column_widthÐs,   cC`s||_t|_dS(u Automatically set font size. N(RlR!R"(RRR((s/lib/python2.7/site-packages/matplotlib/table.pyR/ßs cC`sÖt|jƒdkrdSttj|jƒƒdjƒ}g}xdtj|jƒD]P\}}|d|jkryqTn|j|ƒ}t ||ƒ}|j |ƒqTWx'tj|jƒD]}|j |ƒq»WdS(Nii( tlenRiRŒR_RR)R„RkR/tminRVR'(RR,R-R RƒRqR(((s/lib/python2.7/site-packages/matplotlib/table.pyt_auto_set_font_sizeäs"cC`sOxHtj|jƒD]4}|j|jƒ|ƒ|j|jƒ|ƒqWdS(u: Scale column widths by xscale and row heights by yscale. N(R_RRiRŸR+t set_heightR’(RtxscaletyscaleR¡((s/lib/python2.7/site-packages/matplotlib/table.pytscaleöscC`s7x'tj|jƒD]}|j|ƒqWt|_dS(uW Set the fontsize of the cell text ACCEPTS: a float in points N(R_RRiR'R!R"(RR(Rq((s/lib/python2.7/site-packages/matplotlib/table.pyR'üscC`s\xUtj|jƒD]A}|jƒ|jƒ}}|j||ƒ|j||ƒqWdS(u+Move all the artists by ox,oy (axes coords)N(R_RRitget_xtget_yR“R”(RtoxtoyR¡RR((s/lib/python2.7/site-packages/matplotlib/table.pyt_offsetsc C`sôx!|jD]}|j||ƒq W|jr=|j|ƒn|jƒ|j|ƒ}|j\}}}}|jdk rÈ|j\}} } } |j | || |ƒ||} | |} |jƒnt t |j ƒƒ\}}}}}}}}}}}}}}}}}}d|d|} d|d|} |j |||fkr_|j|} n|j |||||fkr•d|||j} n|j ||||fkrÈd|||j} n|j |||fkrð|j|} n|j |||fkrd|d|} n|j |||fkrJd|d|} n|j |||fkrp|| } n|j |||fkr•d|} n|j |||fkrºd|} n|j |||fkrà|| } n|j| | ƒdS(Ngà?iigð?(RkR¢RlR¥R›R‡R4RfRR©RR£RYRtAXESPADR®( RR,RpR9R:R;R<R=trltrbtrwtrhR¬R­tBESTtURtULtLLtLRtCLtCRtLCtUCtCtTRtTLtBLtBRtRtLtTtB((s/lib/python2.7/site-packages/matplotlib/table.pyR€sJ     K  cC`s|jS(u#return a dict of cells in the table(Ri(R((s/lib/python2.7/site-packages/matplotlib/table.pyt get_celldFsN(!RERFRGRYRzR¯RRRrRxRyR[RnR\R|RR1R‡RŠRŽtget_child_artistsR3R›RžR¢R!R/R¥R©R'R®R€RÆ(((s/lib/python2.7/site-packages/matplotlib/table.pyR]ÕsX           !      7urightuleftucenterubottomuclosedcK`sð|dkr'|dkr'tdƒ‚n|dkrft|ƒ}t|dƒ}dg|g|}nt|ƒ}t|dƒ}x8|D]0}t|ƒ|kr‰tdj|ƒƒ‚q‰q‰W|dk r1t|ƒ|krótdj|ƒƒ‚nxL|D]0}t|ƒ|krútdj|ƒƒ‚qúqúWnd|g|}|dkrbd|g|}nd}|dkr|dk r¶dg|}|d}q¶n|dkr¶d|}n|dk rït|ƒ|krïtd j|ƒƒ‚qïnd }|dkr&| dk rdg|}q?d}n| dkr?d|} n|dkr_d|g|}nt|| | |}| |_|jƒ}xqt|ƒD]c}xZt|ƒD]L}|j|||d ||d |d |||d|||d|ƒq©Wq–W|dk r]xQt|ƒD]@}|jd|d ||d |d ||d| |d| ƒqWn|dk rßxTt|ƒD]F}|j||dd |p˜dd |d ||d||d|ƒqvW|dkrß|j dƒqßn|j |ƒ|S(ut TABLE(cellText=None, cellColours=None, cellLoc='right', colWidths=None, rowLabels=None, rowColours=None, rowLoc='left', colLabels=None, colColours=None, colLoc='center', loc='bottom', bbox=None, edges='closed') Factory function to generate a Table instance. Thanks to John Gill for providing the class and table. uZAt least one argument from "cellColours" or "cellText" must be provided to create a table.iuu+Each row in 'cellText' must have {} columnsu'cellColours' must have {} rowsu.Each row in 'cellColours' must have {} columnsuwgð?u!'rowLabels' must be of length {0}iRRRRRiÿÿÿÿgV瞯Òs*"  r<ÿx  n