o hT@szdZddlZddlZddlZddlZddlZddl m Z ddl m Z eeZGdddZeZddZd dd ZdS) 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)Bboxc@seZdZdZ    d4ddZd d Zd d Zd dZddZdddZ ddZ ddZ ddZ d5ddZ ddZddZd d!Zd"d#Zdd$d%Zdd&d'Zdd(d)Zdd*d+Zdd,d-Zdd.d/Zdd0d1Zd2d3ZdS)6 LayoutGridzG Analogous to a gridspec, and contained in another LayoutGrid. NrrFc sltj||_||_|t|_t|tr|jd|j|_||_||_ t | |_ | dur4t ||_ t | |_| durDt ||_|jdt|tsTt|_n |j|g|R|j|_t j||ftd|_t j||ftd|_i|_i|_dD] t ||j<q}|j} fddt|D|_fddt|D|_dD]$fddt|D|j<t|D] } | |j| d qqd D]t j|td|j<t ||j<q·fd dt|D|_fd dt|D|_d D]%fd dt|D|j<t|D]} | |j| d qq|| |||_!||_"dS)N._)dtype)leftrightleftcbrightcbc g|] }d|dqS)zlefts[].0iVariablesnrj/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/matplotlib/_layoutgrid.py M z'LayoutGrid.__init__..cr)zrights[rrrrrrrNrc &g|]}dd|dqSzmargins[z][rrrrrtodorrrPstrong)bottomtopbottomcbtopcbcr)zbottoms[rrrrrrrYrcr)ztops[rrrrrrrZrc rrrrrrrr\r)#kiwir parent_pos parent_innerseq_idname isinstancernrowsncolsnp atleast_1d height_ratiosones width_ratiosSolversolver add_childemptyobjectartistschildrenmargins margin_valszerosrangeleftsrightsaddEditVariablebottomstops reset_marginsadd_constraintsh_padw_pad) selfparentr%r&r(r+r*rCrDr0r.solrrrr__init__$sf             zLayoutGrid.__init__cCsd|jdd|jd|jd}t|jD]d}t|jD]\}||d|d|j|dd |j|dd |j|dd |j|dd |j d |dd|j d|dd|j d|dd|j d|dd7}qq|S)Nz LayoutBox: 25s xz, z, z: Lz1.3fz, Bz, Rz, Tz, MLr z, MRr z, MBr z, MTr!z, ) r(r*r+r;r<valuer?r=r@r8)rEstrrjrrr__repr__is,   zLayoutGrid.__repr__cCsdD]}||dqdS)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 r r r!r r r"r#gN) edit_margins)rErrrrrAxszLayoutGrid.reset_marginscCs||||dSN)hard_constraintsparent_constraintsgrid_constraints)rErFrrrrBs  zLayoutGrid.add_constraintscCst|jD]?}|j||j|k|j||jd||jd||j||jd||jd|kg}|D] }|j|dBq9qt|jD]?}|j||j |k|j||jd||jd||j ||jd||jd |kg}|D] }|j|dBq~qJd S) zo These are the redundant constraints, plus ones that make the rest of the code easier. r r r r requiredr!r#r r"N) r;r+r=r<r8r2 addConstraintr*r@r?)rErhccrrrrRs6    zLayoutGrid.hard_constraintsrcCs"||jtt|t|<dSrQ)r7r,ix_r-)rEchildrrNrrrr3s"zLayoutGrid.add_childc Cst|ts0|jd|dk|jd|d|dk|jd|d|dk|jd|dkg}n|j\}}t|}t|}|j|d}|j|d}|j|d}|j|d}|j r||j d|d7}||j d|d7}||j d|d8}||j d |d8}||j d |d8}||j d |d8}||j d |d7}||j d |d7}|jd|k|jd|k|jd|k|jd|kg}|D] } |j | dBqdS)Nrrr r r r r!r#r r"rU) r)rr<r=r@r?r%r,r-r&r8r2rV) rErFrWrowscolsr r r!r rXrrrrSs<        zLayoutGrid.parent_constraintscCs|jd|jdd|jdd}||jd|jdd|jdd}||jd}td|jD]O}|j||jd||jd|}||j||jd||jd|}|||j|k}|j|dB|j|d|j|k}|j|dBq5|jd|jdd|jd d}||j d|jd d|jd d}||j d}td|j D]O}|j||jd||jd |}||j ||jd ||jd |}|||j |k}|j|dB|j |d|j|k}|j|dBqdS) Nrr r r r rrr!r#r r") r=r8r<r0r;r+r2rVr@r?r.r*)rEww0rrXhh0rrrrTsF    &   zLayoutGrid.grid_constraintscCs*|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)r2 suggestValuer8r9rErsizecellrrr edit_marginszLayoutGrid.edit_margincCs(||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)r9rhrerrredit_margin_min szLayoutGrid.edit_margin_mincC*tt|j|D] }||||q 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)r;lenr9rhrErrfrrrrrP"s zLayoutGrid.edit_marginscCrj)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)r;rkr9rirlrrredit_all_margins_min2szLayoutGrid.edit_all_margins_mincCs|d|d|jj|d|d|jj|d|d|jjd|d|d|jjd|d|d|jj|d|d|jj|d|d|jjd|d |d |jjdd 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 r r rr r!r#r r"N)ricolspanstartstoprowspan)rEmarginssrrredit_outer_margin_minsEs  z!LayoutGrid.edit_outer_margin_minscCs|j||S)z"Return the margin at this position)r9)rErcolrrr get_margins^szLayoutGrid.get_marginscCs`t|}t|}t|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. rr[) r,r-r from_extentsr<rLr?r=r@rEr^r_bboxrrrget_outer_bboxbs  zLayoutGrid.get_outer_bboxc Cst|}t|}t|j|d|jd|d|jd|d|j|d|jd|d|jd|d|j|d|jd|d|jd|d|j |d|jd |d|jd |d}|S) z Return the inner bounding box of the subplot specs given by rows and cols. rows and cols can be spans. rr r r[r r"r r r!r# r,r-rrwr<rLr8r?r=r@rxrrrget_inner_bboxqs2  zLayoutGrid.get_inner_bboxc Cst|}t|}t|j|d|jd|d|j|d|jd|d|j|d|jd|d|j |d|jd|d}|S)zj Return the bounding box that includes the decorations but, *not* the colorbar... rr r[r"r r#r{rxrrrget_bbox_for_cbs"   zLayoutGrid.get_bbox_for_cbcCst|}t|}t|j|d|jd|d|j|d|j|d|jd|d|jd|d|j|d}|S) Return the left margin bounding box of the subplot specs given by rows and cols. rows and cols can be spans. rr r[r ) r,r-rrwr<rLr8r?r@rxrrrget_left_margin_bboxs  zLayoutGrid.get_left_margin_bboxc Cst|}t|}t|j|d|j|d|jd|d|j|d|j|d|jd|d|jd|d}|S)r~rr[r"r ) r,r-rrwr<rLr?r8r=rxrrrget_bottom_margin_bboxs   z!LayoutGrid.get_bottom_margin_bboxcCst|}t|}t|j|d|jd|d|jd|d|j|d|j|d|jd|d|j|d}|S)r~r[r r r) r,r-rrwr=rLr8r?r@rxrrrget_right_margin_bboxs  z LayoutGrid.get_right_margin_bboxc Cst|}t|}t|j|d|j|d|jd|d|j|d|j|d|jd|d|jd|d}|S)r~rr#r[r!) r,r-rrwr<rLr@r8r=rxrrrget_top_margin_bboxs  zLayoutGrid.get_top_margin_bboxcCs|jdS)zR Update the variables for the solver attached to this layoutgrid. N)r2updateVariables)rErrrupdate_variablesszLayoutGrid.update_variables) NrFrrrNNNN)r)__name__ __module__ __qualname____doc__rHrOrArBrRr3rSrTrhrirPrmrtrvrzr|r}rrrrrrrrrrs8 E  (*         rcCs dttS)z5Generate a short sequential id for layoutbox objects.z%06d)next_layoutboxobjnumrrrrr's r'c Cs|dur||}||}tjdd}||}t|jD]}t|jD]}|j||d}| t j |j |j |jdddd|jdd |j||d} | t j | j | j | jd |d |jd d |j||d} | t j | j | j | jdd dgd|jd d |j||d} | t j | j | j | jdd dgd|jd d |j||d} | t j | j | j | jdd dgd|jd d |j||d} | t j | j | j | jdd dgd|jd d q(q!|jjD]} | durt|| |ddqdS)z(Simple plotting to show where boxes are.Nzaxes.prop_cyclecolor)r^r_rz0.7皙?) linewidth edgecolor facecoloralpha transformzorderr\none)rrrrrr)?ffffff?r)rrrrrr)rrr)rrr)rrr)level)get_layout_engineexecutemplrcParamsby_keyr;r*r+rz add_artistmpatches Rectanglep0widthheight transFigurer|rrrrr7flat plot_children) figlgr _layoutgridscolorsrurrNbbbbichrrrrsx %r)Nr)r itertools kiwisolverr$loggingnumpyr, matplotlibrmatplotlib.patchespatchesrmatplotlib.transformsr getLoggerr_logrcountrr'rrrrrs    P