o iR\@s"dZddlZddlZddlZddlZeeZ ddZ GdddZ d7dd Z d7d d Z d7d d Zd7ddZd8ddZd8ddZd9ddZd9ddZd:ddZd;ddZd;ddZd;d d!Zd;d"d#Zd;d$d%Zd;d&d'Zd;d(d)ZeZd*d+Zd,d-Zd.d/Z d0d1Z!d2d3Z"dd?ZdQd@dAZ dQdBdCZ!dQdDdEZ"dFdGZ#dHdIZ$dJdKZ% dRdLdMZ&dNdOZ'dS)S LayoutBoxzD Basic rectangle representation using kiwi solver variables NFrrrc Cstj} ||_||_|jd} |durt|_d|_n |j|_d|_||||_||_ | |_ | | d|_ | | d|_ | | d|_ | | d|_| | d|_| | d|_| | d |_| | d |_| | d |_| | d |_| | d |_| | d|_| | d|_| | d|_| | d|_| | d|_| | d|_| | d|_| | d|_| | d|_|\}}|\}}||_||_ |!g|_"d|_#|j r|$| |_%| |_&dS)N_rtopbottomleftrightwidthheighth_centerv_center min_width min_height pref_width pref_height left_margin right_margin bottom_margin top_marginleft_margin_minright_margin_minbottom_margin_mintop_margin_min)'kiwiVariableparentnameSolversolverconstrained_layout_called add_childartistpossubplotrrrrrrrrrrrrr r!r"r#r$r%r&r' tightheight tightwidthadd_constraintschildren subplotspecconstrain_marginsh_padw_pad)selfr*r+r4r3r0 lower_left upper_rightr1r2r9r:r)snrrrrr r r __init__3sX    zLayoutBox.__init__cCs|j}||js||jd||jd|j|j|jjk}|j|dB|j|jk}|j|dB||j sK||j d||j d|j |jj |j k}|j|dB|j |j k}|j|dB||j s||j d||j d|j |j|jjk}|j|dB|j |j k}|j|dB||js||jd||jd|j|jj|jk}|j|dB|j|jk}|j|dBdS)a Only do this for pos. This sets a variable distance margin between the position of the axes and the outer edge of the axes. Margins are variable because they change with the figure size. Margin minimums are set to make room for axes decorations. However, the margins can be larger if we are mathicng the position size to other axes. strongg-C6?requiredN)r-hasEditVariabler$addEditVariable suggestValuer rr* addConstraintr%r!rr&r"rr'r#r)r;solcr r r r8qs:        zLayoutBox.constrain_marginscCs|j|g7_dSN)r6r;childr r r r/zLayoutBox.add_childcCs2z |j|WdStytdYdSw)Nz3Tried to remove child that doesn't belong to parent)r6remove ValueErrorrrrIr r r remove_childs  zLayoutBox.remove_childcCsN|j}|j|jfD]}||d||dq ||jr%|dSdS)NgeA)r-rrrCrDhard_constraintsr*parent_constrain)r;rFir r r r5s  zLayoutBox.add_constraintscCsP|j}|j|jk|j|jk|j|jk|j|jkg}|D] }|j|dBqdS)NrA)r*rrrrr-rE)r;r*hcrGr r r rQs    zLayoutBox.parent_constraincCs~|j|j|jk|j|j|jk|j|j|jdk|j|j|jdk|j|jk|j|j kg}|D] }|j |dBq2dS)Ng?rA) rrrrrrrrrrr-rE)r;rSrGr r r rPs  zLayoutBox.hard_constraintscCs|j}|jr d}nd}|j|k}|D] }||dBq|jr"d}nd}|j|k}|D] }||dBq+|j|k|j|kg}|D] }||dBqAdS)NrOg4@rAiI)r-r4rrEr3rrr)r;rFsuggestrGrRr r r soft_constraintss&  zLayoutBox.soft_constraintscCs||_|dS)z/Replace the parent of this with the new parent.N)r*rQ)r;r*r r r set_parents zLayoutBox.set_parentr@cCsB|j|k|j|k|j|k|j|kg}|D] }|j||BqdSrHrrrrr-rE)r;rrrrstrengthrSrGr r r constrain_geometryszLayoutBox.constrain_geometrycCsJ|j|jk|j|jk|j|jk|j|jkg}|D] }|j||BqdS)zJ Make the layoutbox have same position as other layoutbox NrW)r;otherrXrSrGr r r constrain_sames    zLayoutBox.constrain_samecC&|j|jj|k}|j||BdSrH)rr*r-rEr;marginrXrGr r r constrain_left_marginzLayoutBox.constrain_left_margincC|j|j|dSrH)r-rDr$r;r^r r r edit_left_margin_minrKzLayoutBox.edit_left_margin_mincC&|j|jj|k}|j||BdSrH)rr*r-rEr]r r r constrain_right_marginr`z LayoutBox.constrain_right_margincCrarH)r-rDr%rbr r r edit_right_margin_minrKzLayoutBox.edit_right_margin_mincCr\rH)rr*r-rEr]r r r constrain_bottom_marginr`z!LayoutBox.constrain_bottom_margincCrarH)r-rDr&rbr r r edit_bottom_margin_min rKz LayoutBox.edit_bottom_margin_mincCrdrH)rr*r-rEr]r r r constrain_top_marginr`zLayoutBox.constrain_top_margincCrarH)r-rDr'rbr r r edit_top_margin_minrKzLayoutBox.edit_top_margin_mincCs$|j|j|j|jfSrH)rvaluerrrr;r r r get_rectszLayoutBox.get_rectcCs|jdS)zp Update *all* the variables that are part of the solver this LayoutBox is created with. N)r-updateVariablesrlr r r update_variablesszLayoutBox.update_variablescCs<|j}|jfD] }||s|||q||j|dS)z Set the height of the layout box. This is done as an editable variable so that the value can change due to resizing. N)r-rrBrCrD)r;rrXrFrRr r r edit_height s    zLayoutBox.edit_heightcC|j|k}|j||BdS)zr Constrain the height of the layout box. height is either a float or a layoutbox.height. Nrr-rEr;rrXrGr r r constrain_height- zLayoutBox.constrain_heightcC|j|k}|j||BdSrHrrrsr r r constrain_height_min5 zLayoutBox.constrain_height_mincCs<|j}|jfD] }||s|||q||j|dSrH)r-rrBrCrD)r;rrXrFrRr r r edit_width9s    zLayoutBox.edit_widthcCrq)zq Constrain the width of the layout box. *width* is either a float or a layoutbox.width. Nrr-rEr;rrXrGr r r constrain_width@ruzLayoutBox.constrain_widthcCrvrHrzr{r r r constrain_width_minHrxzLayoutBox.constrain_width_mincC|j|k}|j||BdSrH)rr-rE)r;rrXrGr r r constrain_leftLrxzLayoutBox.constrain_leftcCr~rH)rr-rE)r;rrXrGr r r constrain_bottomPrxzLayoutBox.constrain_bottomcCr~rH)rr-rE)r;rrXrGr r r constrain_rightTrxzLayoutBox.constrain_rightcCr~rH)rr-rE)r;rrXrGr r r constrain_topXrxzLayoutBox.constrain_topcC |jdd}|dddkS)zV Helper to check if this layoutbox is the layoutbox of a subplotspec. .Nssr+splitr;r+r r r _is_subplotspec_layoutbox\z#LayoutBox._is_subplotspec_layoutboxcCr)zS Helper to check if this layoutbox is the layoutbox of a gridspec. rrNgridspecrrr r r _is_gridspec_layoutboxcrz LayoutBox._is_gridspec_layoutboxcCs.|jr|g}ng}|jD]}||7}q |S)z Find children of this layout box that are subplots. We want to line poss up, and this is an easy way to find them all. )r2r6find_child_subplots)r;subplotsrJr r r rjs  zLayoutBox.find_child_subplotsc(st||||d}|}|\}}|j} dd} d} d| } | } d}d}| |||d}||}|jdurQ||t|jfdd|jD}n|g|}dg|g|d}tt||gj | |||d}||}|j dur||t|j fd d|j D}n|g|}dg|g|d}tt||gj fd dt |D}fd dt |D}fd dt |D}fd dt |D}t |j |\}}t |j|\}}t||||} t||||}!t||||}"t||||}#|#|"}$|!| }%|j} |j| j| j|"k|j| j| j| k|j| j|$k|j| j|%kg}&|&D] }'|j|'dBqB|S)a Make a layout box from a subplotspec. The layout box is constrained to be a fraction of the width/height of the parent, and be a fraction of the parent width/height from the left/bottom of the parent. Therefore the parent can move around and the layout for the subplot spec should move with it. The parent is *usually* the gridspec that made the subplotspec.?? )r*r+r0r1rO?rNcg|]}|qSr r .0r) netHeighttrr r z5LayoutBox.layout_from_subplotspec..rcrr r r)netWidthrr r rrcsg|] }d|qSrr rrowNumcellHsrr r rcs g|] }d|dqSrrr rrr r r csg|] }d|qSrr rcolNumcellWsrr r rrcs g|] }d|dqSrr rrr r rrrA)r get_gridspec get_geometryr*_row_height_ratiossumnpcumsum column_stackflat_col_width_ratiosrangedivmodnum1num2minmaxrrrrr-rE)(r;subspecr+r0r1lbgsnrowsncolsr*rrtotWidth totHeighthspacewspacecellHsepH cellHeights sepHeightscellWsepW cellWidths sepWidthsfigTops figBottomsfigLefts figRightsrowNum1colNum1rowNum2colNum2 figBottomfigTopfigLeftfigRightrrcsrGr )rrrrrrrr layout_from_subplotspecwsf        z!LayoutBox.layout_from_subplotspecc CsHd|jdd|jdd|jdd|jdd|jdd S) Nz LayoutBox: 25sz , (left: z1.3fz) (bot: z ) (right: z) (top: ))r+rrkrrrrlr r r __repr__s     zLayoutBox.__repr__) NrFFNrrFFNNr@)rNF)(__name__ __module__ __qualname____doc__r?r8r/rNr5rQrPrUrVrYr[r_rcrerfrgrhrirjrmrorprtrwryr|r}rrrrrrrrrr r r r r.sV >0                  Srr@cCsHtdt|D]}||dj|||jk}||j||BqdS)zb Stack LayoutBox instances from left to right. *padding* is in figure-relative units. rNrlenrrr-rEboxespaddingrXrRrGr r r hstacksrcCsHtdt|D]}||dj|||jk}||j||BqdS)z-Stack LayoutBox instances from left to right.rNrrr r r hpackrcCHtdt|D]}||dj|||jk}||j||BqdS-Stack LayoutBox instances from top to bottom.rNrrrrr-rErr r r vstackrrcCrrrrr r r vpackrrmediumcCn|dur tt|}tdt|D]"}||dj||j||d||k}||j||BqdSrNr)ronesrrrr-rE)r height_ratiosrXrRrGr r r match_heights rcCrr)rrrrrr-rE)r width_ratiosrXrRrGr r r match_widthsrrcCt||dt||ddS)Nr)r)rr)rrrr r r vstackeq  rcCr)Nr)r)rr)rrrr r r hstackeqrrcCsFg}|ddD]}t|d|t||k}|dj||BqdS)Nrr)getattrr-rE)rattrrXconsboxr r r aligns rrcCv|d}|}t|D]}|j}q |ddD]"}|}t|D]}|j}q|j|j|j|jk}|j|dBqdSNrrr@)rr*rr-rErlevelsbox0top0nrtopbrGr r r match_top_margins  rcCrr)rr*rr-rErr r r match_bottom_margins*rr cCrr)rr*rr-rErr r r match_left_margins7rr cCrr)rr*rr-rErr r r match_right_marginsDrr cCt||dt||ddSN)r)r r rrr r r match_width_marginsQrrcCr r )rr rr r r match_height_marginsVrrcCr r )rrrr r r match_margins[rrcCs dttS)z5Generate a short sequential id for layoutbox objects.z%06d)next_layoutboxobjnumr r r r seq_idcs rcCs t||jD]}t|qdS)z$Print the children of the layoutbox.N)printr6print_childrenrrJr r r rhs  rcCs:|dur|jdur|jt|dSt|jdSdS)zT Make all elements in this tree None, signalling not to do any more layout. N)r*r-reset nonechildrennonetreerr r r ros    rcCs$|jD]}t|qd|j_d}dSrH)r6rr0 _layoutboxrr r r r|s  rcCs<|jdurtdtdt|tddSt|jdS)zPrint the tree of layoutboxes.NzLayoutBox Tree z==============  )r*rr print_treerr r r rs  rTc Cspddl}ddlm}t||jjr-|gd}|gd|d| |j n|}ddl m }|jdd}|rEtd||jD]m} |rPt| ||j| j| jf| j| jd d ||d |dkr| jd d } |ddkr|j| j| j| d|||dn|j| j| j| ddd|||dt|| |d|dqHdS)z(Simple plotting to show where boxes are.rN)rOrOrr)rrrgffffff?g333333?zaxes.prop_cyclecolorzLevel:noneg?)fcalphaecrrr )sizerrr)havar%rr)levelprintit) matplotlibmatplotlib.pyplotpyplot isinstancefigureFigureadd_axes set_facecolor set_alphadrawrrmatplotlib.patchespatchesrcParamsby_keyrr6 add_patch Rectanglerrkrrrr+rtextrr plot_children) r rr(r)r*pltaxr5colorsrJr+r r r r;sJ         r;)rr@)Nr)rNr)r)rT)$r itertools kiwisolverr(loggingnumpyr getLoggerrrrrrrrrrrrrrrr r r rrrcountrrrrrrr;r r r r sD  (