gGWdZddlZddlZddlZddlZddlZddl m Z ddl m Z ejeZGddZejZdZddZdS) a A layoutgrid is a nrows by ncols set of boxes, meant to be used by `._constrained_layout`, each box is analogous to a subplotspec element of a gridspec. Each box is defined by left[ncols], right[ncols], bottom[nrows] and top[nrows], and by two editable margins for each side. The main margin gets its value set by the size of ticklabels, titles, etc on each axes that is in the figure. The outer margin is the padding around the axes, and space for any colorbars. The "inner" widths and heights of these boxes are then constrained to be the same (relative the values of `width_ratios[ncols]` and `height_ratios[nrows]`). The layoutgrid is then constrained to be contained within a parent layoutgrid, its column(s) and row(s) specified when it is created. N)BboxceZdZdZ ddZdZd Zd Zd Zdd Z dZ dZ dZ ddZ dZdZdZdZddZddZddZddZddZddZddZdZdS) LayoutGridzG Analogous to a gridspec, and contained in another LayoutGrid. NrrFc   tj ||_||_||_|t z|_t|tr|jd|j|_||_ ||_ tj | |_ | tj||_ tj | |_| tj||_|jdzt|ts ||_tj|_n&||_|j|g|R|jj|_tj||ft(|_tj||ft(|_i|_i|_dD]tj||j<|j} fdt5|D|_ fdt5|D|_ fdt5|D|_ fdt5|D|_dD]` fd t5|D|j<t5|D])} | |j| d *ad D]Atj|t(|j<tj||j<B fd t5|D|_ fd t5|D|_! fdt5|D|_" fdt5|D|_#d D]` fdt5|D|j<t5|D])} | |j| d *a|$|%||_&||_'dS)N._)dtype)leftrightleftcbrightcbc2g|]}d|dS)zwidths[].0iVariablesns L/granary/edev/quantum/lib/python3.11/site-packages/matplotlib/_layoutgrid.py z'LayoutGrid.__init__..S4JJJxx2 2 2a 2 2 233JJJc2g|]}d|dS)zlefts[rrrs rrz'LayoutGrid.__init__..Ts4HHHahh"00A00011HHHrc2g|]}d|dS)zrights[rrrs rrz'LayoutGrid.__init__..Urrc2g|]}d|dS)z inner_widths[rrrs rrz'LayoutGrid.__init__..VsD444!"&X&>&>!&>&>&>??444rc 8g|]}dd|dSzmargins[z][rrrrrrtodos rrz'LayoutGrid.__init__..YL"9"9"9&'#+(b+F+F$+F+F!+F+F+F"G"G"9"9"9rstrong)bottomtopbottomcbtopcbc2g|]}d|dS)zheights[rrrs rrz'LayoutGrid.__init__..b4LLL!B!4!4!4!4!455LLLrc2g|]}d|dS)zinner_heights[rrrs rrz'LayoutGrid.__init__..csD555"#'h"'@'@A'@'@'@AA555rc2g|]}d|dS)zbottoms[rrrs rrz'LayoutGrid.__init__..er+rc2g|]}d|dS)ztops[rrrs rrz'LayoutGrid.__init__..fs4FFFAXX..!...//FFFrc 8g|]}dd|dSr!rr"s rrz'LayoutGrid.__init__..hr$r)(kiwirparent parent_pos parent_innerseq_idname isinstancernrowsncolsnp atleast_1d height_ratiosones width_ratiosSolversolver add_childemptyobjectartistschildrenmargins margin_valszerosrangewidthsleftsrights inner_widthsaddEditVariableheights inner_heightsbottomstops reset_marginsadd_constraintsh_padw_pad)selfr1r2r3r5r8r7rTrUr=r;solrrrr#s @@@r__init__zLayoutGrid.__init__$sy= $(688O fj ) ) 5!;4444DI  ]=99  !#D M,77   "D  Y_&*-- -!DK+--DKK DK F T /J / / / /+,DKxf=== %v>>>  : 5 5D&(Xe__D T " "k KJJJJU5\\JJJ HHHHH5<<HHH JJJJJU5\\JJJ 44444&+Ell444: E ED"9"9"9"9"9"9+0<<"9"9"9DL 5\\ E E##DL$6q$98DDDD E; 5 5D!#5!@!@!@DL %'Xe__D T " "LLLLLuU||LLL 55555',U||555LLLLLuU||LLL FFFFFuFFF : E ED"9"9"9"9"9"9+0<<"9"9"9DL 5\\ E E##DL$6q$98DDDD E    rcd|jdd|jd|jd}t|jD];}t|jD]"}||d|d|j|dd |j|dd |j|dd |j|dd |j |dd |j |dd|j d|dd|j d|ddz }$=|S)Nz LayoutBox: 25s xz, z, z: L(z1.3fz, Bz, Wz, Hz, innerWz, innerHz, MLr z, MRrz, ) r5r7r8rHrJvaluerPrIrNrLrOrE)rVstrrjs r__repr__zLayoutGrid.__repr__usHDIHHHDJHHHHHtz"" H HA4:&& H H!HHqHH JqM//11:HH<?0022;HH ;q>//11:HH <?0022; HH !% 1! 4 : : < <E HH !% 21 5 ; ; = =F HH!L0399;;DHH!L1!4::<<GHHHH H rc<dD]}||ddS)z Reset all the margins to zero. Must do this after changing figure size, for instance, because the relative size of the axes labels etc changes. )r rr&r'rrr(r)gN) edit_margins)rVr#s rrRzLayoutGrid.reset_marginss6 ? ) )D   dC ( ( ( ( ) )rc~|||dSN)hard_constraintsparent_constraintsgrid_constraintsrVs rrSzLayoutGrid.add_constraintss@  !!! rc t|jD]}|j||j|k|j||jd|z |jd|z |j||jd|z |jd|z kg}|D]}|j|dz t|jD]}|j||j |k|j||jd|z |jd|z |j ||jd|z |jd |z kg}|D]}|j|dz d S) zo These are the redundant constraints, plus ones that make the rest of the code easier. rrr rrequiredr'r)r&r(N) rHr8rKrJrEr? addConstraintr7rQrP)rVrhccs rrezLayoutGrid.hard_constraintss tz"" : :A+a.DJqM1;q>DL$9!$<<L+A./JqMDL$8$;;L*1-..B  : : ))!j.9999 :tz"" : :A)A,$,q/19Q<$,u"5a"88L)!,-LOdl8&>*Q-4'+b/U*)A,#%,r"f, .B  6 6A K % %a*n 5 5 5 5 6 6rc|jd|jddz |jddz }||jdz |jddz |jddz }||jdz }t d|jD]}|j||jd|z |jd|z }||j|z |jd|z |jd|z }|||j|zk}|j|dz|j|dz |j|k}|j|dz|jd|jddz |jd dz }||j dz |jd dz |jd dz }||j dz }t d|j D]}|j||jd|z |jd |z }||j |z |jd |z |jd |z }|||j |zk}|j|dz|j |dz |j|k}|j|dzdS) Nrrrr rrr%r'r)r&r() rKrErJr=rHr8r?rkrQrPr;r7)rVww0rrmhh0s rrgzLayoutGrid.grid_constraintss [^dl73A6 6 \) $Q '( A f!5a!8 8 \( #A &' "1% %q$*%% 4 4AQ$,w"7"::i(+,ATZ]"T\&%9!%<<h'*+Ab4,Q///A K % %a(l 3 3 3QU#tz!}4A K % %a(l 3 3 3 3 IaL4<.q1 1DL4I!4L L a 4<#9!#< < \* %a () #A& &q$*%% 4 4A1 U 3A 66g&q)*AT\!_$t|H'=a'@@j)!,-Ab4-a000A K % %a(l 3 3 3a!e$ ! 4A K % %a(l 3 3 3 3 4 4rc~|j|j|||||j||<dS)a Change the size of the margin for one cell. Parameters ---------- todo : string (one of 'left', 'right', 'bottom', 'top') margin to alter. size : float Size of the margin. If it is larger than the existing minimum it updates the margin size. Fraction of figure size. cell : int Cell column or row to edit. N)r? suggestValuerErFrVr#sizecells r edit_marginzLayoutGrid.edit_margins@   d!3D!94@@@'+t$$$rcf||j||kr||||dSdS)a Change the minimum size of the margin for one cell. Parameters ---------- todo : string (one of 'left', 'right', 'bottom', 'top') margin to alter. size : float Minimum size of the margin . If it is larger than the existing minimum it updates the margin size. Fraction of figure size. cell : int Cell column or row to edit. N)rFrr~s redit_margin_minzLayoutGrid.edit_margin_minsB$ $"4(. . .   T4 . . . . . / .rctt|j|D]}||||dS)a1 Change the size of all the margin of all the cells in the layout grid. Parameters ---------- todo : string (one of 'left', 'right', 'bottom', 'top') margin to alter. size : float Size to set the margins. Fraction of figure size. N)rHlenrFrrVr#rrs rrbzLayoutGrid.edit_margins/sPs4+D12233 , ,A   T4 + + + + , ,rctt|j|D]}||||dS)a Change the minimum size of all the margin of all the cells in the layout grid. Parameters ---------- todo : {'left', 'right', 'bottom', 'top'} The margin to alter. size : float Minimum size of the margin. If it is larger than the existing minimum it updates the margin size. Fraction of figure size. N)rHrrFrrs redit_all_margins_minzLayoutGrid.edit_all_margins_min?sP s4+D12233 0 0A  tQ / / / / 0 0rc|d|d|jj|d|d|jj|d|d|jjdz |d|d|jjdz |d|d|jj|d|d|jj|d|d|jjdz |d |d |jjdz d S) a< Edit all four margin minimums in one statement. Parameters ---------- margin : dict size of margins in a dict with keys 'left', 'right', 'bottom', 'top' ss : SubplotSpec defines the subplotspec these margins should be applied to r rrrrr'r)r&r(N)rcolspanstartstoprowspan)rVmarginsss redit_outer_margin_minsz!LayoutGrid.edit_outer_margin_minsRs1 VVF^RZ5EFFF Xvh'79IJJJ WfWorz7JKKK Yy(92:?Q;NOOO UF5M2:3CDDD WfWorz7GHHH Xvh'719LMMM Z );Z_q0 2 2 2 2 2rc(|j||S)z"Return the margin at this position)rF)rVr#cols r get_marginszLayoutGrid.get_marginsks%c**rctj|}tj|}tj|j|d|j|d|j|d|j|d}|S)z Return the outer bounding box of the subplot specs given by rows and cols. rows and cols can be spans. rrr) r9r:r from_extentsrJr]rPrKrQrVrurvbboxs rget_outer_bboxzLayoutGrid.get_outer_bboxos }T""}T"" JtAw  % % ' ' Lb " ( ( * * KR ! ' ' ) ) Id1g  $ $ & & ((  rc \tj|}tj|}tj|j|d|jd|dz|jd|dz|j|d|jd|dz|jd|dz|j|d|jd|dz |jd|dz |j |d|jd |dz |jd |dz }|S) z Return the inner bounding box of the subplot specs given by rows and cols. rows and cols can be spans. rr rrrr&r(rrr'r) r9r:rrrJr]rErPrKrQrs rget_inner_bboxzLayoutGrid.get_inner_bbox~s }T""}T"" ZQ & & ( ( V$T!W-33556 X&tAw/55778\$r( # ) ) + + X&tBx066889 Z(b288::;[b " ( ( * * W%d2h/55778 Y'R17799:YtAw  % % ' ' U#DG,22445 W%d1g.44667    rc tj|}tj|}tj|j|d|jd|dz|j|d|jd|dz|j|d|jd|dz |j |d|jd|dz }|S)zj Return the bounding box that includes the decorations but, *not* the colorbar... rrrrr(rr)rrs rget_bbox_for_cbzLayoutGrid.get_bbox_for_cbs3 }T""}T"" ZQ & & ( ( X&tAw/55778 \$r( # ) ) + + Z(b288::; [b " ( ( * * Y'R17799: YtAw  % % ' ' W%d1g.44667    rctj|}tj|}tj|j|d|jd|dz|j|d|j|d|jd|dz|jd|dz|j|d}|S) Return the left margin bounding box of the subplot specs given by rows and cols. rows and cols can be spans. rrrrr ) r9r:rrrJr]rErPrQrs rget_left_margin_bboxzLayoutGrid.get_left_margin_bboxs }T""}T"" ZQ & & ( ( X&tAw/55778 \$r( # ) ) + + ZQ & & ( ( X&tAw/55778 V$T!W-33556YtAw  % % ' '** rc tj|}tj|}tj|j|d|j|d|jd|dz|j|d|j|d|jd|dz|jd|dz}|S)rrrrr(r&) r9r:rrrJr]rPrErKrs rget_bottom_margin_bboxz!LayoutGrid.get_bottom_margin_bboxs }T""}T"" ZQ & & ( ( \$r( # ) ) + + \* %d2h / 5 5 7 78 [b " ( ( * * \$r( # ) ) + + X&tBx066889 \* %d2h / 5 5 7 78  rctj|}tj|}tj|j|d|jd|dz |jd|dz |j|d|j|d|jd|dz |j|d}|S)rrrrrr) r9r:rrrKr]rErPrQrs rget_right_margin_bboxz LayoutGrid.get_right_margin_bboxs }T""}T"" [b " ( ( * * W%d2h/55778 Y'R17799:\$r( # ) ) + + [b " ( ( * * Y'R17799: YtAw  % % ' '** rc tj|}tj|}tj|j|d|j|d|jd|dz |j|d|j|d|jd|dz |jd|dz }|S)rrr)rrr') r9r:rrrJr]rQrErKrs rget_top_margin_bboxzLayoutGrid.get_top_margin_bboxs }T""}T"" ZQ & & ( ( YtAw  % % ' ' W%d1g.44667 [b " ( ( * * YtAw  % % ' ' W%d1g.44667 U#DG,22445 77 rc8|jdS)zR Update the variables for the solver attached to this layoutgrid. N)r?updateVariablesrhs rupdate_variableszLayoutGrid.update_variabless ##%%%%%r) NrFrrrNNNN)r)__name__ __module__ __qualname____doc__rXr`rRrSrer@rfrgrrrbrrrrrrrrrrrrrrrrs06=>6:#OOOOb   )))   :::2JJJJ'6'6'6R%4%4%4T,,,&////*,,, 000&2222+++    0($&$$&&&&&rrc0dttzS)z5Generate a short sequential id for layoutbox objects.z%06d)next_layoutboxobjnumrrrr4r4s D)** **rc|/||}||}tjdd}||}t |jD]E}t |jD],}|||}| tj |j |j |jdddd|jd |||} | tj | j | j | jd |d |jd |||} | tj | j | j | jdd dgd|jd  |||} | tj | j | j | jdd dgd|jd  |||} | tj | j | j | jdd dgd|jd  |||} | tj | j | j | jdd dgd|jd  .G|jjD]} | t/|| |dzdS)z(Simple plotting to show where boxes are.Nzaxes.prop_cyclecolor)rurvrz0.7皙?) linewidth edgecolor facecoloralpha transformzorderrsnone)rrrrrr)?ffffff?r)rrrrrr)rrr)rrr)rrr)level)get_layout_engineexecutemplrcParamsby_keyrHr7r8r add_artistmpatches Rectanglep0widthheight transFigurerrrrrrDflat plot_children) figlgr _layoutgridscolorsrrr_bbbbichs rrrsK z,,..66s;; #  \+ , 3 3 5 5g >F -C 28__%J%Jrx$ J$ JA"""22B NN"25"(BI-2e),*,... / / / ###33C NN"3639cjA-0F-0_RIII J J J ))qq)99C NN"3639cjA-33-<__-0_RIII J J J ***::C NN"3639cjA-33-<__-0_RIII J J J +++;;C NN"3639cjA-33-<__-0_RIII J J J ((aa(88C NN"3639cjA-33-<__-0_RIII J J J JA$ JJk22 > #rq 1 1 1 122r)Nr)r itertools kiwisolverr0loggingnumpyr9 matplotlibrmatplotlib.patchespatchesrmatplotlib.transformsr getLoggerr_logrcountrr4rrrrrs$%%%%%%&&&&&&w""X&X&X&X&X&X&X&X&t#9?$$+++ /2/2/2/2/2/2r