o h@sdZddlZddlZddlZddlmZmZddlm Z ddl m Z ddlmZddlmZddlmZddlmZddlmZmZmZddlmZmZmZdZ d d Z!d d Z"d3ddZ#d4ddZ$Gddde j%Z&Gddde&Z'Gddde'Z(Gddde'Z)Gddde&Z*Gddde&Z+Gdd d e&Z,Gd!d"d"e&Z-Gd#d$d$e&Z.d%d&Z/Gd'd(d(e.Z0Gd)d*d*e&Z1Gd+d,d,e j%ej2Z3Gd-d.d.Z4Gd/d0d0e4Z5Gd1d2d2e4Z6dS)5ac Container classes for `.Artist`\s. `OffsetBox` The base of all container artists defined in this module. `AnchoredOffsetbox`, `AnchoredText` Anchor and align an arbitrary `.Artist` or a text relative to the parent axes or a specific anchor point. `DrawingArea` A container with fixed width and height. Children have a fixed position inside the container and may be clipped. `HPacker`, `VPacker` Containers for layouting their children vertically or horizontally. `PaddedBox` A container to add a padding around an `.Artist`. `TextArea` Contains a single `.Text` instance. N)_api _docstring)FontProperties) BboxImage)FancyBboxPatchFancyArrowPatch bbox_artist)BboxBboxBaseTransformedBboxFcs,ddddgtfdd}|S)z Decorator for the get_offset method of OffsetBox and subclasses, that allows supporting both the new signature (self, bbox, renderer) and the old signature (self, width, height, xdescent, ydescent, renderer). cStSNlocals)selfwidthheightxdescentydescentrendererrh/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/matplotlib/offsetbox.py3z$_compat_get_offset..cSr r rrbboxrrrrr4rcsbtj|g|Ri|}d|vr|dnt|d |d |d|d}|d||dS)Nrrrrrrr)rselect_matching_signaturer from_bounds)rargskwargsparamsrmethsigsrr get_offset6s z&_compat_get_offset..get_offset) functoolswraps)r"r$rr!r_compat_get_offset-s r'cOstr t|i|dSdSr )DEBUG mbbox_artist)rrrrr _bbox_artistAsr*fixedcs<tjgd|d|dkr0tdgfdd|D}|dd}|dur,|d}||fS|d krh|dur:d }t|d krM|t|t|d ndtdgfd d|D}|dd}||fS|d krt|}|durdur|td |t|}n|t|||tt|}||fSdS)a[ Pack boxes specified by their *widths*. For simplicity of the description, the terminology used here assumes a horizontal layout, but the function works equally for a vertical layout. There are three packing *mode*\s: - 'fixed': The elements are packed tight to the left with a spacing of *sep* in between. If *total* is *None* the returned total will be the right edge of the last box. A non-*None* total will be passed unchecked to the output. In particular this means that right edge of the last box may be further to the right than the returned total. - 'expand': Distribute the boxes with equal spacing so that the left edge of the first box is at 0, and the right edge of the last box is at *total*. The parameter *sep* is ignored in this mode. A total of *None* is accepted and considered equal to 1. The total is returned unchanged (except for the conversion *None* to 1). If the total is smaller than the sum of the widths, the laid out boxes will overlap. - 'equal': If *total* is given, the total space is divided in N equal ranges and each box is left-aligned within its subspace. Otherwise (*total* is *None*), *sep* must be provided and each box is left-aligned in its subspace of width ``(max(widths) + sep)``. The total width is then calculated to be ``N * (max(widths) + sep)``. Parameters ---------- widths : list of float Widths of boxes to be packed. total : float or None Intended total length. *None* if not used. sep : float or None Spacing between boxes. mode : {'fixed', 'expand', 'equal'} The packing mode. Returns ------- total : float The total width needed to accommodate the laid out boxes. offsets : array of float The left offsets of the boxes. )r+expandequal)moder+rcg|]}|qSrr.0wseprr wz'_get_packed_offsets..Nr,cr/rrr0r3rrr5r6r-z@total and sep cannot both be None when using layout mode 'equal') r check_in_listnpcumsumlensummax ValueErrorarange)widthstotalr4r.offsets_offsetsmaxhrr3r_get_packed_offsetsFs4.    rFbaselinecstjgd|ddurtdd|D|dkr7tdd|Dtdd|Df}d gt|}||fS|d vrJd f}d d |D}||fS|d vr_d f}fdd |D}||fS|dkrpd f}fdd |D}||fS)a Align boxes each specified by their ``(y0, y1)`` spans. For simplicity of the description, the terminology used here assumes a horizontal layout (i.e., vertical alignment), but the function works equally for a vertical layout. Parameters ---------- yspans List of (y0, y1) spans of boxes to be aligned. height : float or None Intended total height. If None, the maximum of the heights (``y1 - y0``) in *yspans* is used. align : {'baseline', 'left', 'top', 'right', 'bottom', 'center'} The alignment anchor of the boxes. Returns ------- (y0, y1) y range spanned by the packing. If a *height* was originally passed in, then for all alignments other than "baseline", a span of ``(0, height)`` is used without checking that it is actually large enough). descent The descent of the packing. offsets The bottom offsets of the boxes. )rGlefttoprightbottomcenter)alignNcss|] \}}||VqdSr rr1y0y1rrr sz'_get_aligned_offsets..rGcss|]\}}|VqdSr rrNrrrrQcss|]\}}|VqdSr rrNrrrrQrRr)rHrKcSsg|]\}}| qSrrrNrrrr5z(_get_aligned_offsets..)rJrIcsg|]\}}|qSrrrNrrrr5rLcs$g|]\}}||d|qS)?rrNrTrrr5s$)rr9r>minr<)yspansrrMyspanrDrrTr_get_aligned_offsetss*$ rZcseZdZdZfddZfddZejjj ddZdd Z d d Z e d d Z ddZddZddZddZddZddZdddZddZZS) OffsetBoxz The OffsetBox is a simple container artist. The child artists are meant to be drawn at a relative position to its parent. Being an artist itself, all parameters are passed on to `.Artist`. cs0tj||||dg|_d|_dS)NFrr)super__init___internal_update set_clip_on _children_offset)rrr __class__rrr^s    zOffsetBox.__init__cs(t||D]}||q dS)z Set the `.Figure` for the `.OffsetBox` and all its children. Parameters ---------- fig : `~matplotlib.figure.Figure` N)r] set_figure get_children)rfigcrcrrres   zOffsetBox.set_figurecCs0tjj|||D] }|dur||_q dSr )martistArtistaxesfsetrf)raxrhrrrrks  zOffsetBox.axescCsF||r difS|D]}||\}}|r||fSq difS)a Delegate the mouse event contains-check to the children. As a container, the `.OffsetBox` does not respond itself to mouseevents. Parameters ---------- mouseevent : `~matplotlib.backend_bases.MouseEvent` Returns ------- contains : bool Whether any values are within the radius. details : dict An artist-specific dictionary of details of the event context, such as which points are contained in the pick radius. See the individual Artist subclasses for details. See Also -------- .Artist.contains F)_different_canvasrfcontains)r mouseeventrhabrrrros   zOffsetBox.containscC||_d|_dS)a Set the offset. Parameters ---------- xy : (float, float) or callable The (x, y) coordinates of the offset in display units. These can either be given explicitly as a tuple (x, y), or by providing a function that converts the extent into the offset. This function must have the signature:: def offset(width, height, xdescent, ydescent, renderer) -> (float, float) TN)rbstalerxyrrr set_offsets zOffsetBox.set_offsetcCs.t|jr||j|j|j |j |S|jS)aO Return the offset as a tuple (x, y). The extent parameters have to be provided to handle the case where the offset is dynamically determined by a callable (see `~.OffsetBox.set_offset`). Parameters ---------- bbox : `.Bbox` renderer : `.RendererBase` subclass )callablerbrrx0rOrrrrr$#s zOffsetBox.get_offsetcCrs)zh Set the width of the box. Parameters ---------- width : float TN)rrt)rrrrr set_width6 zOffsetBox.set_widthcCrs)zj Set the height of the box. Parameters ---------- height : float TN)rrt)rrrrr set_heightAr{zOffsetBox.set_heightcCsdd|jDS)z/Return a list of the visible child `.Artist`\s.cSsg|]}|r|qSr) get_visibler1rhrrrr5NrUz2OffsetBox.get_visible_children..rarrrrget_visible_childrenLszOffsetBox.get_visible_childrencC|jS)z'Return a list of the child `.Artist`\s.rrrrrrfPzOffsetBox.get_childrencCstd)a/ Return the bbox of the offsetbox and the child offsets. The bbox should satisfy ``x0 <= x1 and y0 <= y1``. Parameters ---------- renderer : `.RendererBase` subclass Returns ------- bbox list of (xoffset, yoffset) pairs z:get_bbox_and_offsets must be overridden in derived classes)NotImplementedErrorrrrrr_get_bbox_and_child_offsetsTsz%OffsetBox._get_bbox_and_child_offsetscCs||\}}|S)z:Return the bbox of the offsetbox, ignoring parent offsets.)r)rrrrDrrrget_bboxfszOffsetBox.get_bboxNcCsb|dur |jdd}||}z |||\}}Wnty*|\}}Ynw|||SNTroot) get_figure _get_rendererrr$ TypeError translated)rrrpxpyrrrget_window_extentks   zOffsetBox.get_window_extentc Csz||\}}|||\}}t||D]\}\}}|||||f||qt||dtdddd|_dS)zm Update the location of children if necessary and draw them to the given *renderer*. FpadfillpropsN) rr$ziprrwdrawr*dictrt rrrrDrrrhoxoyrrrrvs  zOffsetBox.drawr )__name__ __module__ __qualname____doc__r^rerirjrksetterrorwr'r$rzr|rrfrrrr __classcell__rrrcrr[s$         r[cs"eZdZ  dfdd ZZS) PackerBaserNrGr+cs8t||_||_||_||_||_||_||_dS)ao Parameters ---------- pad : float, default: 0.0 The boundary padding in points. sep : float, default: 0.0 The spacing between items in points. width, height : float, optional Width and height of the container box in pixels, calculated if *None*. align : {'top', 'bottom', 'left', 'right', 'center', 'baseline'}, default: 'baseline' Alignment of boxes. mode : {'fixed', 'expand', 'equal'}, default: 'fixed' The packing mode. - 'fixed' packs the given `.Artist`\s tight with *sep* spacing. - 'expand' uses the maximal available space to distribute the artists with equal spacing in between. - 'equal': Each artist an equal fraction of the available space and is left-aligned (or top-aligned) therein. children : list of `.Artist` The artists to pack. Notes ----- *pad* and *sep* are in points and will be scaled with the renderer dpi, while *width* and *height* are in pixels. N) r]r^rrr4rr.rMra)rrr4rrrMr.childrenrcrrr^s $ zPackerBase.__init__)rrNNrGr+N)rrrr^rrrrcrrsrc@eZdZdZddZdS)VPackerz VPacker packs its children vertically, automatically adjusting their relative positions at draw time. .. code-block:: none +---------+ | Child 1 | | Child 2 | | Child 3 | +---------+ c sd}|j|}|j|}|jdur+|D]}t|tr*|jdkr*||jqfdd|D}t dd|D|j|j \\}}} t dd|D|j ||j\} } | | dd|D} | d} | | } t || ||| |gt| | fS) N?r,cg|]}|qSrrr~rrrr5rSz7VPacker._get_bbox_and_child_offsets..cSg|]}|jqSr) intervalxr1rrrrr5cSrrrTrrrrr5rcSrr)rPrrrrr5rr)points_to_pixelsrr4rr isinstancerr.rzrZrMrFrr rpaddedr) rrdpicorrr4rhbboxesryx1xoffsetsryoffsetsrrrrrs*        z#VPacker._get_bbox_and_child_offsetsNrrrrrrrrrrs rc@r)HPackera  HPacker packs its children horizontally, automatically adjusting their relative positions at draw time. .. code-block:: none +-------------------------------+ | Child 1 Child 2 Child 3 | +-------------------------------+ c sd}|j|}|j|}fdd|D}|s)tdddd|gfStdd|D|j|j \\}}}t dd|D|j ||j \} } |dj } | dd|D| 8} t| || |||gt| |fS)Nrcrrrr~rrrr5rSz7HPacker._get_bbox_and_child_offsets..rcSrr) intervalyrrrrr5rcSrr)rrrrrr5rcSrr)ryrrrrr5r)rrr4rr rrrZrrMrFrr.ryr) rrrrr4rrOrPrrrryrrrrs"      z#HPacker._get_bbox_and_child_offsetsNrrrrrrs rcsLeZdZdZddddfdd Zdd Zd d Zdd d ZddZZ S) PaddedBoxa A container to add a padding around an `.Artist`. The `.PaddedBox` contains a `.FancyBboxPatch` that is used to visualize it when rendering. .. code-block:: none +----------------------------+ | | | | | | | <--pad--> Artist | | ^ | | pad | | v | +----------------------------+ Attributes ---------- pad : float The padding in points. patch : `.FancyBboxPatch` When *draw_frame* is True, this `.FancyBboxPatch` is made visible and creates a border around the box. rFN) draw_frame patch_attrsc sPt||_|g|_tddddddd|dd |_|d ur&|j|d Sd S) a Parameters ---------- child : `~matplotlib.artist.Artist` The contained `.Artist`. pad : float, default: 0.0 The padding in points. This will be scaled with the renderer dpi. In contrast, *width* and *height* are in *pixels* and thus not scaled. draw_frame : bool Whether to draw the contained `.FancyBboxPatch`. patch_attrs : dict or None Additional parameters passed to the contained `.FancyBboxPatch`. rrrr2kr8T square,pad=0 rvrr facecolor edgecolormutation_scalesnapvisibleboxstyleN)r]r^rrarpatchupdate)rchildrrrrcrrr^s zPaddedBox.__init__cCs,|j|d}|jd||dgfS)Nrrr\)rrrarr)rrrrrrr6sz%PaddedBox._get_bbox_and_child_offsetsc Cs|||\}}|||\}}t||D]\}\}}|||||fq|||D]}||q1d|_dSNF)rr$rrrwrrrtrrrrr;s    zPaddedBox.drawcCs(|j|j|r|j|d|_dSNT)r set_boundsboundsset_mutation_scalertrrfontsizerrr update_frameIs  zPaddedBox.update_framecCs ||||j|dSr )rrrrrrrrrOszPaddedBox.draw_frame)rr ) rrrrr^rrrrrrrrcrrs rcsteZdZdZdfdd ZeddZejddZd d Zd d Z d dZ ddZ ddZ ddZ ddZZS) DrawingAreaz The DrawingArea can contain any Artist as a child. The DrawingArea has a fixed width and height. The position of children relative to the parent is fixed. The children can be clipped at the boundaries of the parent. rFcs@t||_||_||_||_||_t|_ t|_ dS)a" Parameters ---------- width, height : float Width and height of the container box. xdescent, ydescent : float Descent of the box in x- and y-direction. clip : bool Whether to clip the children to the box. N) r]r^rrrr_clip_children mtransformsAffine2Doffset_transform dpi_transform)rrrrrcliprcrrr^]s  zDrawingArea.__init__cCr)zl If the children of this DrawingArea should be clipped by DrawingArea bounding box. )rrrrr clip_childrenqszDrawingArea.clip_childrencCst||_d|_dSr)boolrrt)rvalrrrrys  cCs |j|jS)zX Return the `~matplotlib.transforms.Transform` applied to the children. )rrrrrr get_transform~s zDrawingArea.get_transformcCdSz+ set_transform is ignored. Nrrtrrr set_transformzDrawingArea.set_transformcC0||_|j|j|d|dd|_dSz Set the offset of the container. Parameters ---------- xy : (float, float) The (x, y) coordinates of the offset in display units. rr8TNrbrclear translatertrurrrrw  zDrawingArea.set_offsetcCrzReturn offset of the container.rbrrrrr$rzDrawingArea.get_offsetcCs6|d}t|j ||j ||j||j|S)Nr)rr rrrrr)rrdpi_corrrrrs zDrawingArea.get_bboxcCsZ|j||s|||jdur|j|_|jdd}|dur+||dSdS)&Add an `.Artist` to the container box.NFr)raappendis_transform_setrrrkrre)rrqrgrrr add_artists   zDrawingArea.add_artistc Cs|d}|j|j|ttddgd|jg|j |jg|j dgg| }|j D]}|j r>|j s>|js>||||q.t||dtdddd|_dS)NrrFrrr)rrrscalerTransformedPathmpathPathrrrrarclipbox _clippath set_clip_pathrr*rrt)rrrtpathrhrrrrs        zDrawingArea.draw)rrF)rrrrr^propertyrrrrrwr$rrrrrrrcrrUs   rcspeZdZdZdddfdd ZddZd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ZS)TextAreaz The TextArea is a container artist for a single Text instance. The text is placed at (0, 0) with baseline+left alignment, by default. The width and height of the TextArea instance is the width and height of its child text. NF) textpropsmultilinebaselinecsj|duri}tjdd|fi||_t|jg|_t|_t|_ |j |j|j ||_ dS)a Parameters ---------- s : str The text to be displayed. textprops : dict, default: {} Dictionary of keyword parameters to be passed to the `.Text` instance in the TextArea. multilinebaseline : bool, default: False Whether the baseline for multiline text is adjusted so that it is (approximately) center-aligned with single-line text. Nr) mtextText_textr]r^rarrr_baseline_transformr_multilinebaseline)rsrrrcrrr^s      zTextArea.__init__cCs|j|d|_dS)z&Set the text of this area as a string.TN)rset_textrtrrrrrrs  zTextArea.set_textcC |jS)z5Return the string representation of this area's text.)rget_textrrrrr zTextArea.get_textcCrs)aX Set multilinebaseline. If True, the baseline for multiline text is adjusted so that it is (approximately) center-aligned with single-line text. This is used e.g. by the legend implementation so that single-line labels are baseline-aligned, but multiline labels are "center"-aligned with them. TN)rrtrrrrset_multilinebaselines zTextArea.set_multilinebaselinecCr)z( Get multilinebaseline. )rrrrrget_multilinebaselineszTextArea.get_multilinebaselinecCrrrrrrrrrzTextArea.set_transformcCrrrrurrrrw rzTextArea.set_offsetcCrrrrrrrr$rzTextArea.get_offsetcCs|jd|jj|jr dndd\}}}|j|\}}}|j\}} |jt|dkrF|j rFd| d||} |j d|| | }n t ||| |} | |} |j } d| d| d | } t | | || S) NlpTeXF)ismathr8rVr)rHrLrJ)get_text_width_height_descentr_fontproperties get_usetex _get_layoutsizerrr<rrr>get_horizontalalignmentr r)rr_h_d_rinfoydr2hyd_newh_dharyrrrrs     zTextArea.get_bboxcCs,|j|t||dtdddd|_dSNFrrr)rrr*rrtrrrrr3s  z TextArea.draw)rrrrr^rrrr rrwr$rrrrrrcrrs  rcsXeZdZdZfddZddZddZdd Zd d Zd d Z ddZ ddZ Z S)AuxTransformBoxaZ Offset Box with the aux_transform. Its children will be transformed with the aux_transform first then will be offsetted. The absolute coordinate of the aux_transform is meaning as it will be automatically adjust so that the left-lower corner of the bounding box of children will be set to (0, 0) before the offset transform. It is similar to drawing area, except that the extent of the box is not predetermined but calculated from the window extent of its children. Furthermore, the extent of the children will be calculated in the transformed coordinate. cs(||_tt|_t|_dSr ) aux_transformr]r^rrrref_offset_transform)rrrcrrr^Hs  zAuxTransformBox.__init__cCs$|j|||d|_dS)rTN)rarrrrt)rrqrrrrPs  zAuxTransformBox.add_artistcCs|j|j|jS)zf Return the :class:`~matplotlib.transforms.Transform` applied to the children )rr rrrrrrVs zAuxTransformBox.get_transformcCrrrrrrrr_rzAuxTransformBox.set_transformcCrrrrurrrrwdrzAuxTransformBox.set_offsetcCrrrrrrrr$rrzAuxTransformBox.get_offsetcsr|j}|j|jfdd|jD}t|}|j|j |j |j |t dd|j |j S)Ncrrrr~rrrr5|rSz,AuxTransformBox.get_bbox..r)r get_matrixr rrar unionrryrO set_matrixrrr)rr_offrubrrrrvs     zAuxTransformBox.get_bboxcCs6|jD]}||qt||dtdddd|_dSr)rarr*rrt)rrrhrrrrs   zAuxTransformBox.draw) rrrrr^rrrrwr$rrrrrrcrr:s  rc seZdZdZdZdddddddd d d d Zd dddddddfdd ZddZddZddZ ddZ ddZ d&ddZ e d d!Zd&d"d#Zd$d%ZZS)'AnchoredOffsetboxa0 An offset box placed according to location *loc*. AnchoredOffsetbox has a single child. When multiple children are needed, use an extra OffsetBox to enclose them. By default, the offset box is anchored against its parent Axes. You may explicitly specify the *bbox_to_anchor*. r8r  ) z upper rightz upper leftz lower leftz lower rightrJz center leftz center rightz lower centerz upper centerrL皙?rVNT)r borderpadrpropframeonbbox_to_anchorbbox_transformc stjd i| |||||t|tr!tj|j|d}||_ ||_ ||_ |dur8t t jdd|_nt ||_t|trPd|vrP|jt jdtddddd |jd |d d |_dS)a Parameters ---------- loc : str The box location. Valid locations are 'upper left', 'upper center', 'upper right', 'center left', 'center', 'center right', 'lower left', 'lower center', 'lower right'. For backward compatibility, numeric values are accepted as well. See the parameter *loc* of `.Legend` for details. pad : float, default: 0.4 Padding around the child as fraction of the fontsize. borderpad : float, default: 0.5 Padding between the offsetbox frame and the *bbox_to_anchor*. child : `.OffsetBox` The box that will be anchored. prop : `.FontProperties` This is only used as a reference for paddings. If not given, :rc:`legend.fontsize` is used. frameon : bool Whether to draw a frame around the box. bbox_to_anchor : `.BboxBase`, 2-tuple, or 4-tuple of floats Box that is used to position the legend in conjunction with *loc*. bbox_transform : None or :class:`matplotlib.transforms.Transform` The transform for the bounding box (*bbox_to_anchor*). **kwargs All other parameters are passed on to `.OffsetBox`. Notes ----- See `.Legend` for a detailed description of the anchoring mechanism. )locNlegend.fontsizerrrrr2rTrrr)r]r^set_bbox_to_anchor set_childrstrr check_getitemcodesr6r1rrmplrcParamsr2 _from_anyrset_sizerget_size_in_pointsr) rr6rr1rr2r3r4r5rrcrrr^s*&     zAnchoredOffsetbox.__init__cCs ||_|dur |j|_d|_dS)zSet the child to be anchored.NT)_childrkrt)rrrrrr:s zAnchoredOffsetbox.set_childcCr)zReturn the child.rCrrrr get_childrzAnchoredOffsetbox.get_childcC|jgS)zReturn the list of children.rDrrrrrfszAnchoredOffsetbox.get_childrencCs.||j}|j|}|||Sr )rr2rBrrErr)rrrrrrrrs zAnchoredOffsetbox.get_bboxcCs2|jdur |jjS|j}|dur|jSt|j|S)z,Return the bbox that the box is anchored to.N)_bbox_to_anchorrkr_bbox_to_anchor_transformr )r transformrrrget_bbox_to_anchors  z$AnchoredOffsetbox.get_bbox_to_anchorc Cs|dus t|tr ||_n/zt|}Wnty'}ztd||d}~ww|dkr6|d|dddg}tj||_||_d|_ dS)aB Set the bbox that the box is anchored to. *bbox* can be a Bbox instance, a list of [left, bottom, width, height], or a list of [left, bottom] where the width and height will be assumed to be zero. The bbox will be transformed to display coordinate by the given transform. NzInvalid bbox: r rr8T) rr rGr<rr?r rrHrt)rrrIlerrrrrr9s    z$AnchoredOffsetbox.set_bbox_to_anchorcCsV|j||j}|}t|jtdd|j |j ||\}}||j ||j fS)Nr) r1rr2rBrJ_get_anchored_bboxr6r rrrryrO)rrrrr4ryrOrrrr$szAnchoredOffsetbox.get_offsetcCs&|j|j|r|j|dSdSr )rrrrrrrrr*szAnchoredOffsetbox.update_framecCs~|sdS||}||j}||||j||| ||\}}| ||f| |d|_ dSr) r}rrr2rBrrrr$rrErwrt)rrrrrrrrrr/s    zAnchoredOffsetbox.drawr )rrrrzorderr=r^r:rErfrrJr9r'r$rrrrrrcrr's: B   r'cCs(gd|}|| }|j||djS)z} Return the (x, y) position of the *bbox* anchored at the *parentbbox* with the *loc* code with the *borderpad*. ) NNENWSWSEEWrSSNC) container)ranchoredp0)r6r parentbboxr1rhrXrrrrM@s  rMcs*eZdZdZddddfdd ZZS) AnchoredTextz& AnchoredOffsetbox with Text. r0rVN)rr1r2c  sf|duri}ddh}|t|@rtdt||d|_|jj}tj|f|||j|d|dS)aq Parameters ---------- s : str Text. loc : str Location code. See `AnchoredOffsetbox`. pad : float, default: 0.4 Padding around the text as fraction of the fontsize. borderpad : float, default: 0.5 Spacing between the offsetbox frame and the *bbox_to_anchor*. prop : dict, optional Dictionary of keyword parameters to be passed to the `~matplotlib.text.Text` instance contained inside AnchoredText. **kwargs All other parameters are passed to `AnchoredOffsetbox`. Nvaverticalalignmentzr?rr2rtrrrrr^s   zAnnotationBbox.set_fontsizecCr)zReturn the fontsize in points.)r2rBrrrr get_fontsizejrzAnnotationBbox.get_fontsizec>dur |jdd|tfdd|DS)NTrcrrr!r1rrrrr5sz4AnnotationBbox.get_window_extent..rrupdate_positionsr r#rfrrrrrn  z AnnotationBbox.get_window_extentcr)NTrcrr) get_tightbboxrrrrr5{rz0AnnotationBbox.get_tightbbox..rrrrrrvrzAnnotationBbox.get_tightbboxc Cs |||j|j\}}|j|}|j\}}|j|||j|j|||j |j f|j |}|j |j||}|j ||jr|j|j|j}||} |j|| d|jvrm||jd}|j||jd|j } |j| dSdS)zHUpdate pixel positions for the annotated point, the text, and the arrow.rpatchAN)_get_xyrrrrrrwrryrrOrrrrrrrrrZrr_get_position_xyr set_positionsget set_patchA) rrox0oy0rfwfhr arrow_begin arrow_endrrrrr~s,  &     zAnnotationBbox.update_positionscCs|r ||s dS|j|jj|d|||jdur?|jjdddur9|jdd}dur9|j ||j ||j ||j || |jjd|_dS)N)gidFr)r}r open_grouprdrget_gidrrrrerrr close_grouprt)rrrgrrrrs       zAnnotationBbox.draw)NrzNr )rrrrrNrrinterpdr^rrrrrorfrerrrrrrrrrrr}s<k        $r}c@sreZdZdZdddZeddZeddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZdS) DraggableBasea Helper base class for a draggable artist (legend, offsetbox). Derived classes must override the following methods:: def save_offset(self): ''' Called when the object is picked for dragging; should save the reference position of the artist. ''' def update_offset(self, dx, dy): ''' Called during the dragging; (*dx*, *dy*) is the pixel offset from the point where the mouse drag started. ''' Optionally, you may override the following method:: def finalize_offset(self): '''Called when the mouse is released.''' In the current implementation of `.DraggableLegend` and `DraggableAnnotation`, `update_offset` places the artists in display coordinates, and `finalize_offset` recalculates their position in axes coordinate and set a relevant attribute. Fcsd||_|s |dd|_|o|jj|_|jjfddd|jfd|j fd|j ffD|_ dS)NTFc s&g|]\}}tj||qSr)r%partial disconnect_connect_picklable)r1namefunc callbacksrrr5s z*DraggableBase.__init__.. pick_eventbutton_release_eventmotion_notify_event) ref_artistpickable set_picker got_artistcanvas supports_blit _use_blitron_pick on_release on_motion_disconnectors)rruse_blitrrrr^s   zDraggableBase.__init__cCs|jjddjSr)rrrrrrrrszDraggableBase.cCsdd|jddDS)NcSsg|]}|jdqS)r)r)r1rrrrr5s z*DraggableBase...r rrrrrrs cCs|r>|jr@|j|j}|j|j}||||jr7|j |j |j |j j dd|jdS|j dSdSdSr)_check_still_parentedrxmouse_xymouse_y update_offsetrrrestore_region backgroundrrrrblit)revtdxdyrrrrs   zDraggableBase.on_motioncCs|rJ|j|jkr|jj|_|jj|_|d|_ |j rL|j rN|j d|j |jjdd}|j |j|_|j ||j dSdSdSdS)NTFr)rartistrrprrrr save_offsetrr set_animatedrrrcopy_from_bboxrrrr)rrrgrrrrs      zDraggableBase.on_pickcCsj|r/|jr1|d|_|jr3|j|j|j|jj |j |j ddSdSdSdSr) rrfinalize_offsetrrrrrrfigurerrr)reventrrrrs zDraggableBase.on_releasecCs"|jjdddur|dSdS)NFrT)rrrrrrrr sz#DraggableBase._check_still_parentedcCs|jD]}|qdS)zDisconnect the callbacks.Nr)r disconnectorrrrrs zDraggableBase.disconnectcCdSr rrrrrrzDraggableBase.save_offsetcCrr r)rrrrrrrrzDraggableBase.update_offsetcCrr rrrrrrrzDraggableBase.finalize_offsetNF)rrrrr^rrcidsrrrrrrrrrrrrrs      rcs6eZdZd fdd ZddZddZdd ZZS) DraggableOffsetBoxFcstj||d||_dSN)r)r]r^r)rrrrrcrrr^# zDraggableOffsetBox.__init__cCsD|j}|jdd}||||}|\|_|_|j|dSr)rrrr$r offsetbox_x offsetbox_yrw)rrroffsetrrrr's  zDraggableOffsetBox.save_offsetcCs$|j||j|f}|j|dSr )rrrrw)rrr loc_in_canvasrrrr.sz DraggableOffsetBox.update_offsetcCsB|j}|jdd}||}|j\}}||j||jf}|Sr)rrrrrbryrO)rrrrrrrrrrget_loc_in_canvas2s   z$DraggableOffsetBox.get_loc_in_canvasr)rrrr^rrrrrrrcrr"s rcs.eZdZdfdd ZddZddZZS) DraggableAnnotationFcstj||d||_dSr)r]r^ annotation)rrrrcrrr^<rzDraggableAnnotation.__init__cCs"|j}||j\|_|_dSr )rrrIrrr)rannrrrr@szDraggableAnnotation.save_offsetcCs.|j}||j||j|f|_dSr )rrinvertedrIrrr)rrrrrrrrDs  z!DraggableAnnotation.update_offsetr)rrrr^rrrrrrcrr;sr)r+)rG)7rr%numpyr: matplotlibr>rrmatplotlib.artistrrimatplotlib.pathpathrmatplotlib.texttextrmatplotlib.transforms transformsrmatplotlib.font_managerrmatplotlib.imagermatplotlib.patchesrrrr)r r r r(r'r*rFrZrjr[rrrrrrrr'rMr\rdrr}rrrrrrrsJ        Q3;/("XpuR5 +Gvo