a ضak@sdZddlZddlZddlmZddlmZddl m Z e e Zd#ddZddZdd Zd d Zddddd d dZddddd ddZdddddZddZddZddZddddd ddZddddZdd Zd!d"ZdS)$a Adjust subplot layouts so that there are no overlapping axes or axes decorations. All axes decorations are dealt with (labels, ticks, titles, ticklabels) and some dependent artists are also dealt with (colorbar, suptitle). Layout is done via `~matplotlib.gridspec`, with one constraint per gridspec, so it is possible to have overlapping axes if the gridspecs overlap (i.e. using `~matplotlib.gridspec.GridSpecFromSubplotSpec`). Axes placed using ``figure.subplots()`` or ``figure.add_subplots()`` will participate in the layout. Axes manually placed via ``figure.add_axes()`` will not. See Tutorial: :doc:`/tutorials/intermediate/constrainedlayout_guide` N)_apic Cst|d}|ds tddStdD]x}t|||||||dt|||||dt||||t||rt |||||||dn tdt ||q(|S)a Do the constrained_layout. Called at draw time in ``figure.constrained_layout()`` Parameters ---------- fig : Figure ``Figure`` instance to do the layout in. renderer : Renderer Renderer to use. h_pad, w_pad : float Padding around the axes elements in figure-normalized units. hspace, wspace : float Fraction of the figure to dedicate to space between the axes. These are evenly spread between the gaps between the axes. A value of 0.2 for a three-column layout would have a space of 0.1 of the figure width between each column. If h/wspace < h/w_pad, then the pads are used instead. Returns ------- layoutgrid : private debugging structure NhasgridszhThere are no gridspecs with layoutgrids. Possibly did not call parent GridSpec with the "figure" keyword)h_padw_padhspacewspace)rrz{constrained_layout not applied because axes sizes collapsed to zero. Try making figure larger or axes decorations smaller.) make_layoutgridsrZ warn_externalrangemake_layout_marginsmake_margin_suptitlesmatch_submerged_marginsZupdate_variablescheck_no_collapsed_axesreposition_axes reset_margins)figrendererrrrr layoutgrids_rn/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/matplotlib/_constrained_layout.pydo_constrained_layoutAs(           rcCs|durt}d|d<t|ds4tjddd||<nB|j}t||}||}tj|ddd d |jj|jjfd ||<|j D]}t ||}q||j D]$}t|d r| }t||}q|S) z Make the layoutgrid tree. (Sub)Figures get a layoutgrid so we can have figure margins. Gridspecs that are attached to axes get a layoutgrid so axes can have margins. NFr_parentZfiglb)parentnameZpanellbT)rr parent_innernrowsncols parent_posget_subplotspec)dicthasattr mlayoutgrid LayoutGrid _subplotspec get_gridspecmake_layoutgrids_gsrowspancolspansubfigsr _localaxesas_listr )rrgsZparentlbsfigaxrrrr s0          r c Cs||vs|jdur|Sd|d<t|ds\||j}tj|dd|j|j||d||<n|j}| }||vr|t ||}||}|d|vrtj|ddd|j |j fd ||d<tj||dd|j|j||d ||<|S) zR Make the layoutgrid for a gridspec (and anything nested in the gridspec) NTr _subplot_specZgridspec)rrrrr width_ratios height_ratiostopr)rrrrr)rrrrr1r2) figurer"r#r$Z_ncolsZ_nrowsZget_width_ratiosZget_height_ratiosr0r&r'r(r))rr-rZ subplot_specZparentgsZ subspeclbrrrr'sB       r'c Cs|jD]}t||}|sdSq|jD]t}t|dr&|}||vr&||}t|jD]>}t|jD].}| ||} | j dks| j dkrhdSqhqZq&dS)z9 Check that no axes have collapsed to zero size. Fr rT) r*raxesr"r r&r rrget_inner_bboxwidthheight) rrr.okr/r-ZlgijZbbrrrrs      rrrrrc Cs|j}|}t|drB|jdur(|jn|}|jdur<|jn|}n(|jdurR|jn|}|jdurf|jn|}|d}|d}|\} } ||||ddddd} || |kr|jj dkr|| | d<|jj | kr|| | d<|| |kr|j j | kr|| | d<|j j dkr|| | d<| S) Nrrr)leftcbrightcbbottomcbtopcbleftrightr3bottomr=r>r?r@) r%r&r"rr_hspace_wspace get_geometryr)startstopr() objrrrrssr-rDrErrmarginrrrget_margin_from_paddings4          rLc Cs |jD]D}|j}t|||||||dt|dd||d} ||j| |q|jD]} t| drV| spqV| }| } | |vrdSt| ||||d} t | |\} }| d| j |j 7<| d|j| j7<| d| j|j7<| d|j| j7<| jD]}t||}t|\}}|jd }t ||\}}|dkrz|j|jjkr| d |j|7<n|dkr|j|jjkr| d |j|7<nX|dkr|j|jjkr| d |j|7<n&|j|jjkr| d |j|7<|dvrz|j|jjkrD|j |j krD| d|j |j 7<|j|jjkrz|j|jkrz| d|j|j7<|dvr|j|jjkr|j|jkr| d|j|j7<|j|jjkr|j|jkr| d|j|j7<q|| | |qVdS)z For each axes, make a margin between the *pos* layoutbox and the *axes* layoutbox be a minimum size that can accommodate the decorations on the axis. Then make room for colorbars. r<rr NrArBrCr3locationr>r=r@r?)r3rCrArB)r*r%r rLrZedit_outer_margin_minsr+r,r" get_in_layoutr r&get_pos_and_bboxx0x1y0y1 _colorbarscolorbar_get_padget_cb_parent_spans_colorbar_inforHr)r7rGr(r8)rrrrrrrr.rJZmarginsr/r-rKposbboxcbaxpadZ cbp_rspanZ cbp_cspanloccbposcbbboxrrrr sx              r rrc Cs|jj}tddg||gg}|j|j|}|j}|j}|jD]} t || |||dqF|j dur|j r|j } t |j ddr|j | dd|f||j |} ||d| jd||jdur<|j r<|j } t |jddr<|j| d|f||j|} ||d| jd||jdur|j r|j } t |jddr|j|| df||j|} ||d | jd|dS) Nrr`Z_autoposFrr3rrCrA) transFigureinvertedtransform_bbox mtransformsZBboxtransSubfigurer8r7r*r Z _suptitlerO get_positiongetattrZ set_position get_tightbboxedit_margin_minZ _supxlabelZ _supylabel) rrrrrZ inv_trans_figZpadboxZ h_pad_localZ w_pad_localr.prZrrrr nsB       r c Csf|jD]}t||qdd|D}|D]2}|}||vrR||q,||}t|jdkrt |j d|jdd|j d|jdd}t |j d|jdd|j d |jdd}|D]} | } || } | durt| jdkrt | j d| jdd| j d| jdd} | |krP| }t | j d| jdd| j d | jdd} | |kr| }q|jddD]}|j d||d q|jddD]}|j d||d qt|j dkr,t |j d |j dd|j d |j dd}t |j d |j dd|j d|j dd}|D]} | } || } | dur^t| j dkr^t t | j d | j dd| j d | j dd|g}t t | j d | j dd| j d| j dd|g}q^|j ddD]}|j d ||d q"|j ddD]}|j d ||d qHq,dS)a Make the margins that are submerged inside an Axes the same size. This allows axes that span two columns (or rows) that are offset from one another to have the same size. This gives the proper layout for something like:: fig = plt.figure(constrained_layout=True) axs = fig.subplot_mosaic("AAAB CCDD") Without this routine, the axes D will be wider than C, because the margin width between the two columns in C has no width by default, whereas the margins between the two columns of D are set by the width of the margin between A and B. However, obviously the user would like C and D to be the same size, so we need to add constraints to these "submerged" margins. This routine makes all the interior margins the same, and the spacing between the three columns in A and the two column in C are all set to the margins between the two columns of D. See test_constrained_layout::test_constrained_layout12 for an example. cSs"g|]}t|dr|r|qS)r )r"rO).0arrr sz+match_submerged_margins..rrANr=rBr>)cellr3r@rCr?) r*r Zget_axesr r&removelenr)npmaxZ margin_valsrir()rrr.ZaxsZax1Zss1Zlg1ZmaxsublZmaxsubrZax2Zss2Zlg2Zmaxsubl2Zmaxsubr2r:ZmaxsubtZmaxsubbrrrr s           r c Cstj}tj }tj}tj }|jdD]D}|}t|jj|}t|jj|}t|j j|}t|j j|}q&t ||}t ||}||fS)zA Figure out which subplotspecs this colorbar belongs to: parents) rrinfrXr minr(rGrsrHr)r ) r[ZrowstartZrowstopZcolstartZcolstoprrJr(r)rrrrWs  rWcCs||j}|jdd}||j|j}z|j|dd}WntyT|j|d}Yn0|durd|}n||j}||fS)z Get the position and the bbox for the axes. Parameters ---------- ax renderer Returns ------- pos : Bbox Position in figure coordinates. bbox : Bbox Tight bounding box in figure coordinates. T)original)rZfor_layout_only)rN)r4rf transformedrerarh TypeErrorrb)r/rrrYZ tightbboxrZrrrrP s  rPc Cs|j|j}|jD]8}||} |j| |dt|||||||dq|jD]} t | drV| snqV| } | } | \} }| |vrdS|| j| j| jd} || }| |ddddd}t| jdddD]*\}}| |jd dkrt||||d qqVdS) zF Reposition all the axes based on the new inner bounding box. )rZ)rrrrr Nrowscolsr)rArBrCr3rnrtoffset)rarer*Zget_outer_bboxZ_redo_transform_rel_figrxrr+r,r"rOr r&rFr6r(r)rc _set_position enumeraterUrXreposition_colorbar)rrrrrrrtrans_fig_to_subfigr.rZr/rJr-rrZnewbboxr~nnr[rrrr+s:       rr}cCs|jd}|d}|j}|j|j}t|\}} ||j|| d} ||j|| d} |jd} |jd} |jd}|jd}|jd}t||\}}t ||}| d vrr| || | | }| d kr|j |j }| j |j |d }|||7}|d |j|7<||d}nV|j |j }| j |j }||j |||d 7}|d |j|7<||d}n| || | | }| d kr|j|j}| j|j|d }|||7}|d |j|7<|d|}nV|j|j}| j|j}||j |||d 7}|d |j|7<|d|}||}||j|||| | dvrrd|}|||d|S)a Place the colorbar in its new place. Parameters ---------- cbax : Axes Axes for the colorbar renderer : w_pad, h_pad : float width and height padding (in fraction of figure) hspace, wspace : float width and height padding as fraction of figure size divided by number of columns or rows margin : array-like offset the colorbar needs to be pushed to in order to account for multiple colorbars rtrrzrManchorfractionaspectshrinkrNrBrAr3rC)rCr3rauto)rXr&r4rarerWZget_bbox_for_cbr6rPrVZshrunkZanchoredrQrRr7Z translatedrSrTr8rcZ set_transformrZ set_anchorZset_box_aspectZ set_aspect)rr[rr~rtr-rr cb_rspans cb_cspansZ bboxparentZpbrMrrrrr^r_ZcbpadZpbcbZlmarginZdxZbmarginZdyrrrrVsd                              rcCsh|jD]}t||q|jD]:}t|dr|r|}|}||vr||q||dS)z Reset the margins in the layoutboxes of fig. Margins are usually set as a minimum, so if the figure gets smaller the minimum needs to be zero in order for it to grow again. r N)r*rr5r"rOr r&)rrr.r/rJr-rrrrs   rcCs^|jd}|d}t|\}}||j||d}|jddvrJ|j}n|j}|jd|S)NrtrrzrM)rBrAr\)rXr&rWr6r7r8)rZcaxrtr-rrZ bboxoutersizerrrrVs   rV)NN)__doc__loggingZnumpyrrZ matplotlibrZmatplotlib.transformsZ transformsrdZmatplotlib._layoutgridZ _layoutgridr# getLogger__name___logrr r'rrLr r r rWrPrrrrVrrrrs4    ) E,, & T'a" +U