B ]t\ @sdZddlZddlZddlmZddlmZ ddl m Z ddl mZddlmZmZmZddlmZddlmZmZddlmZddlmZddlmZdd lmZdd l mZd Z d d ZdGddZ!dHddZ"Gddde j#Z$Gddde$Z%Gddde%Z&Gddde%Z'Gddde$Z(Gddde$Z)Gd d!d!e$Z*Gd"d#d#e$Z+Gd$d%d%e$Z,Gd&d'd'e,Z-Gd(d)d)e$Z.Gd*d+d+e j#eZ/Gd,d-d-e0Z1Gd.d/d/e1Z2Gd0d1d1e1Z3e4d2krddl5m6Z7e78d3Z9e9:e7;d4Ze?d5@d6d6d7ZAe.eAfd8ddd9e>ZBeZEe/eEd;dd?d@e=dAdBdCde=dDdEdF ZFeEsz'_get_packed_offsets..Nexpandcsg|] }|qSrr)rr)rrrrTsZequalzUnknown mode : %s)zipnpZcumsumlensummaxarange ValueError) wd_listZtotalrmodeZw_listZd_listZoffsets_offsetsZmaxhr)rr_get_packed_offsets4s0      r%baselinecsdkrtdd|D|dkr^tdd|D}tdd|D}||dd|D}nj|d krzd }d d|D}nN|d krd }fd d|D}n.|dkrd }fdd|D}ntd|f||fS)a) Given a list of (height, descent) of each boxes, align the boxes with *align* and calculate the y-offsets of each boxes. total width and the offset positions of each items according to *mode*. xdescent is analogous to the usual descent, but along the x-direction. xdescent values are currently ignored. *hd_list* : list of (width, xdescent) of boxes to be aligned. *sep* : spacing between boxes *height* : Intended total length. None if not used. *align* : align mode. 'baseline', 'top', 'bottom', or 'center'. Ncss|]\}}|VqdS)Nr)rhdrrr tsz'_get_aligned_offsets..r&css|]\}}||VqdS)Nr)rr'r(rrrr)wscss|]\}}|VqdS)Nr)rr'r(rrrr)xscSsg|] \}}dqS)gr)rr'r(rrrrzsz(_get_aligned_offsets..)lefttopgcSsg|] \}}|qSrr)rr'r(rrrr}s)rightZbottomcsg|]\}}||qSrr)rr'r()heightrrrscentercs g|]\}}|d|qS)g?r)rr'r()r-rrrszUnknown Align mode : %s)rr!)hd_listr-alignheight_descentZdescentr$r)r-r_get_aligned_offsetses$r2cseZdZdZfddZddZejjj ddZdd Z d d Z d d Z ddZ ddZddZddZddZddZddZddZZS) OffsetBoxz The OffsetBox is a simple container artist. The child artist are meant to be drawn at a relative position to its parent. cs(tj|||dg|_d|_dS)NF)rr)super__init__Z set_clip_on _children_offset)selfrr) __class__rrr5s zOffsetBox.__init__cCs.tj||x|D]}||qWdS)z^ Set the figure accepts a class:`~matplotlib.figure.Figure` instance N)martistArtist set_figure get_children)r8figcrrrr<szOffsetBox.set_figurecCs4tjj||x|D]}|dk r||_qWdS)N)r:r;axesfsetr=)r8axr?rrrr@szOffsetBox.axescCs4x*|D]}||\}}|r ||fSq WdifS)NF)r=contains)r8 mouseeventr?abrrrrCs  zOffsetBox.containscCs||_d|_dS)a> Set the offset. Parameters ---------- xy : (float, float) or callable The (x,y) coordinates of the offset in display units. A callable must have the signature:: def offset(width, height, xdescent, ydescent, renderer) -> (float, float) TN)r7stale)r8xyrrr set_offsets zOffsetBox.set_offsetcCs"t|jr||||||S|jS)zC Get the offset accepts extent of the box )callabler7)r8widthr-xdescentydescentrendererrrr get_offsetszOffsetBox.get_offsetcCs||_d|_dS)z6 Set the width accepts float TN)rKrG)r8rKrrr set_widthszOffsetBox.set_widthcCs||_d|_dS)z7 Set the height accepts float TN)r-rG)r8r-rrr set_heightszOffsetBox.set_heightcCsdd|jDS)z? Return a list of visible artists it contains. cSsg|]}|r|qSr) get_visible)rr?rrrrsz2OffsetBox.get_visible_children..)r6)r8rrrget_visible_childrenszOffsetBox.get_visible_childrencCs|jS)z7 Return a list of artists it contains. )r6)r8rrrr=szOffsetBox.get_childrencCs tddS)N) Exception)r8rNrrrget_extent_offsetsszOffsetBox.get_extent_offsetscCs ||\}}}}}||||fS)z@ Return with, height, xdescent, ydescent of box )rV)r8rNrr'xdydr$rrr get_extentszOffsetBox.get_extentc CsD||\}}}}}||||||\}}tj||||||S)z8 get the bounding box in display space. )rVrO mtransformsr from_bounds) r8rNrr'rWrXr$pxpyrrrget_window_extentszOffsetBox.get_window_extentc Cs||\}}}}}||||||\}}x>t||D],\} \} } | || || f| |q:Wt||dtdddd|_dS)zm Update the location of children if necessary and draw them to the given *renderer*. Fg)pad)fillpropsN) rVrOrrSrIdrawr dictrG) r8rNrKr-rLrMr$r\r]r?oxoyrrrrbszOffsetBox.draw)__name__ __module__ __qualname____doc__r5r<r:r;r@setterrCrIrOrPrQrSr=rVrYr^rb __classcell__rr)r9rr3s     r3cseZdZdfdd ZZS) PackerBaseNcs8t||_||_||_||_||_||_||_dS)a Parameters ---------- pad : float, optional Boundary pad. sep : float, optional Spacing between items. width : float, optional height : float, optional Width and height of the container box, calculated if `None`. align : str, optional Alignment of boxes. Can be one of ``top``, ``bottom``, ``left``, ``right``, ``center`` and ``baseline`` mode : str, optional Packing mode. Notes ----- *pad* and *sep* need to given in points and will be scale with the renderer dpi, while *width* and *height* need to be in pixels. N) r4r5r-rKrr_r#r0r6)r8r_rrKr-r0r#children)r9rrr5s zPackerBase.__init__)NNNNNNN)rfrgrhr5rkrr)r9rrlsrlcs*eZdZdZd fdd ZddZZS) VPackerz The VPacker has its children packed vertically. It automatically adjust the relative positions of children in the drawing time. Nr&rc st|||||||dS)ap Parameters ---------- pad : float, optional Boundary pad. sep : float, optional Spacing between items. width : float, optional height : float, optional width and height of the container box, calculated if `None`. align : str, optional Alignment of boxes. mode : str, optional Packing mode. Notes ----- *pad* and *sep* need to given in points and will be scale with the renderer dpi, while *width* and *height* need to be in pixels. N)r4r5)r8r_rrKr-r0r#rm)r9rrr5BszVPacker.__init__cs$d}|j|}|j|}|jdk rZx0|D]$}t|tr2|jdkr2||jq2Wfdd|D}dd|D}dd|D}t ||j|j \}} } dd|D} t | |j ||j\} } | d d|D}| |d }| |}||}|d || d || |||t t| |fS) zN update offset of childrens and return the extents of the box g?Nrcsg|]}|qSr)rY)rr?)rNrrrqsz.VPacker.get_extent_offsets..cSs$g|]\}}}}|||||fqSrr)rrr'rWrXrrrrsscSsg|]\}}}}||fqSrr)rrr'rWrXrrrruscSsg|]\}}}}||fqSrr)rrr'rWrXrrrrzscSsg|]\}}}}|qSrr)rrr'rWrXrrrr~sr)points_to_pixelsr_rrKrS isinstancerlr#rPr2r0r%r-listr)r8rNdpicorr_rr?whd_listr"rKrLxoffsets pack_listr-Z yoffsets_yoffsetsrMr)rNrrVcs0        zVPacker.get_extent_offsets)NNNNr&rN)rfrgrhrir5rVrkrr)r9rrn=s rncs*eZdZdZd fdd ZddZZS) HPackerz The HPacker has its children packed horizontally. It automatically adjusts the relative positions of children at draw time. Nr&rc st|||||||dS)aW Parameters ---------- pad : float, optional Boundary pad. sep : float, optional Spacing between items. width : float, optional height : float, optional Width and height of the container box, calculated if `None`. align : str Alignment of boxes. mode : str Packing mode. Notes ----- *pad* and *sep* need to given in points and will be scale with the renderer dpi, while *width* and *height* need to be in pixels. N)r4r5)r8r_rrKr-r0r#rm)r9rrr5szHPacker.__init__cs2d}|j|}|j|}fdd|D}|sNd|d|||gfS|jdkrtdd|D}tdd|D}||}n|jd|}d d|D} t| |j|j\}}} d d|D} t| |j ||j \} } | d d|D}|d d}||}| d||d|||||t t || fS) zM update offset of children and return the extents of the box g?csg|]}|qSr)rY)rr?)rNrrrsz.HPacker.get_extent_offsets..roNcss|]\}}}}||VqdS)Nr)rrr'rWrXrrrr)sz-HPacker.get_extent_offsets..css|]\}}}}|VqdS)Nr)rrr'rWrXrrrr)scSsg|]\}}}}||fqSrr)rrr'rWrXrrrrscSsg|]\}}}}||fqSrr)rrr'rWrXrrrrscSsg|]\}}}}|qSrr)rrr'rWrXrrrrsr) rpr_rrSr-rr2r0r%rKr#rrr)r8rNrsr_rrtr1rMr-r/rwrvrKZ xoffsets_rurLr)rNrrVs2         zHPacker.get_extent_offsets)NNNNr&rN)rfrgrhrir5rVrkrr)r9rrxs rxcs@eZdZd fdd ZddZddZdd d Zd d ZZS) PaddedBoxNFc s^t||_|g|_tdddddddd|_|jjdd d |d k rT|j|||_d S) z *pad* : boundary pad .. note:: *pad* need to given in points and will be scale with the renderer dpi, while *width* and *height* need to be in pixels. )ggg?rkrT)rHrKr- facecolor edgecolormutation_scalesnapsquarer)r_N) r4r5r_r6rpatch set_boxstyleupdate _drawFrame)r8childr_ draw_frameZ patch_attrs)r9rrr5s   zPaddedBox.__init__cCsT|d}|j|}|jd|\}}}}|d||d|||||dgfS)zN update offset of childrens and return the extents of the box g?rro)rr)rpr_r6rY)r8rNrsr_rr'rWrXrrrrVs    zPaddedBox.get_extent_offsetsc Cs||\}}}}}||||||\}}x4t||D]"\} \} } | || || fq:W||x|D]} | |qtWd|_dS)zm Update the location of children if necessary and draw them to the given *renderer*. FN)rVrOrrSrIrrbrG) r8rNrKr-rLrMr$r\r]r?rdrerrrrbs zPaddedBox.drawcCs4|j|j|j|j|j|r*|j|d|_dS)NT)r set_boundsx0y0rKr-set_mutation_scalerG)r8bboxfontsizerrr update_frames   zPaddedBox.update_framecCs*||}|||jr&|j|dS)N)r^rrrrb)r8rNrrrrr!s  zPaddedBox.draw_frame)NFN)N) rfrgrhr5rVrbrrrkrr)r9rrys  rycs|eZdZdZdfdd ZeddZejddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZZS) 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. FcsXt||_||_||_||_||_t|_ |j |j ddt|_ dS)z *width*, *height* : width and height of the container box. *xdescent*, *ydescent* : descent of the box in x- and y-direction. *clip* : Whether to clip the children rN) r4r5rKr-rLrM_clip_childrenrZAffine2Doffset_transformclear translate dpi_transform)r8rKr-rLrMZclip)r9rrr52s   zDrawingArea.__init__cCs|jS)zl If the children of this DrawingArea should be clipped by DrawingArea bounding box. )r)r8rrr clip_childrenHszDrawingArea.clip_childrencCst||_d|_dS)NT)boolrrG)r8valrrrrPs cCs |j|jS)zf Return the :class:`~matplotlib.transforms.Transform` applied to the children )rr)r8rrr get_transformUszDrawingArea.get_transformcCsdS)z+ set_transform is ignored. Nr)r8trrr set_transform\szDrawingArea.set_transformcCs0||_|j|j|d|dd|_dS)z Set the offset of the container. Parameters ---------- xy : (float, float) The (x,y) coordinates of the offset in display units. rrTN)r7rrrrG)r8rHrrrrIbs  zDrawingArea.set_offsetcCs|jS)z1 return offset of the container. )r7)r8rrrrOqszDrawingArea.get_offsetcCs8||\}}}}|\}}tj||||||S)z8 get the bounding box in display space. )rYrOrZrr[)r8rNrr'rWrXrdrerrrr^ws zDrawingArea.get_window_extentcCs.|d}|j||j||j||j|fS)z@ Return with, height, xdescent, ydescent of box g?)rprKr-rLrM)r8rNdpi_corrrrrYs zDrawingArea.get_extentcCsP|j||s"|||jdk r4|j|_|j}|dk rL||dS)z?Add any :class:`~matplotlib.artist.Artist` to the container boxN)r6appendZis_transform_setrrr@figurer<)r8rEr>rrr add_artists  zDrawingArea.add_artistc Cs|d}|j|j||ttddgd|jg|j |jg|j dgg| }x4|j D]*}|j r|j s|js||||q`Wt||dtdddd|_dS)z# Draw the children g?rFg)r_)r`raN)rprrZscalerZZTransformedPathmpathZPathr-rKrr6rZclipboxZ _clippathZ set_clip_pathrbr rcrG)r8rNrZtpathr?rrrrbs       zDrawingArea.draw)rrF)rfrgrhrir5propertyrrjrrrIrOr^rYrrbrkrr)r9rr*s    rc@szeZdZdZdddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZdS)TextAreaz The TextArea is contains a single Text instance. The text is placed at (0,0) with baseline+left alignment. The width and height of the TextArea instance is the width and height of the its child text. NTcCs|dkr i}d|krd|d<tjdd|f||_t||jg|_t|_|j |j ddt|_ |j |j|j ||_ ||_dS)a Parameters ---------- s : str a string to be displayed. textprops : `~matplotlib.font_manager.FontProperties`, optional multilinebaseline : bool, optional If `True`, baseline for multiline text is adjusted so that it is (approximatedly) center-aligned with singleline text. minimumdescent : bool, optional If `True`, the box has a minimum descent of "p". Nvar&r)mtextZText_textr3r5r6rZrrrr_baseline_transformr_multilinebaseline_minimumdescent)r8s textpropsZmultilinebaselineminimumdescentrrrr5s       zTextArea.__init__cCs|j|d|_dS)z&Set the text of this area as a string.TN)rset_textrG)r8rrrrrs zTextArea.set_textcCs |jS)z5Returns the string representation of this area's text)rget_text)r8rrrrszTextArea.get_textcCs||_d|_dS)z Set multilinebaseline . If True, baseline for multiline text is adjusted so that it is (approximatedly) center-aligned with singleline text. TN)rrG)r8rrrrset_multilinebaselineszTextArea.set_multilinebaselinecCs|jS)z) get multilinebaseline . )r)r8rrrget_multilinebaselineszTextArea.get_multilinebaselinecCs||_d|_dS)z Set minimumdescent . If True, extent of the single line text is adjusted so that it has minimum descent of "p" TN)rrG)r8rrrrset_minimumdescentszTextArea.set_minimumdescentcCs|jS)z% get minimumdescent. )r)r8rrrget_minimumdescentszTextArea.get_minimumdescentcCsdS)z+ set_transform is ignored. Nr)r8rrrrr szTextArea.set_transformcCs0||_|j|j|d|dd|_dS)z Set the offset of the container. Parameters ---------- xy : (float, float) The (x,y) coordinates of the offset in display units. rrTN)r7rrrrG)r8rHrrrrIs  zTextArea.set_offsetcCs|jS)z1 return offset of the container. )r7)r8rrrrOszTextArea.get_offsetcCs8||\}}}}|\}}tj||||||S)z8 get the bounding box in display space. )rYrOrZrr[)r8rNrr'rWrXrdrerrrr^%s zTextArea.get_window_extentcCs|j|jj\}}|jd|jjdd\}}}|j|\}}} |j|j} } |dd} |jt |dkr|j rd| d||} |j d| | | } n,t ||| | }| rt | |} || } | | d| fS) NZlpF)ismathrrrg?g)rZ is_math_textZget_text_width_height_descentZ_fontpropertiesZ _get_layoutrKr-rrrrrrr)r8rNZ clean_liner_Zh_Zd_rinfor(rr'lineZd_newZh_drrrrY-s    zTextArea.get_extentcCs,|j|t||dtdddd|_dS)z# Draw the children Fg)r_)r`raN)rrbr rcrG)r8rNrrrrbMs z TextArea.draw)NNT)rfrgrhrir5rrrrrrrrIrOr^rYrbrrrrrs  (   rc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)AuxTransformBoxaX Offset Box with the aux_transform. Its children will be transformed with the aux_transform first then will be offseted. 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. cCsJ||_t|t|_|j|jddt|_|jdS)Nr) aux_transformr3r5rZrrrrref_offset_transform)r8rrrrr5fs    zAuxTransformBox.__init__cCs$|j|||d|_dS)z?Add any :class:`~matplotlib.artist.Artist` to the container boxTN)r6rrrrG)r8rErrrrts zAuxTransformBox.add_artistcCs|j|j|jS)zf Return the :class:`~matplotlib.transforms.Transform` applied to the children )rrr)r8rrrrzs zAuxTransformBox.get_transformcCsdS)z+ set_transform is ignored. Nr)r8rrrrrszAuxTransformBox.set_transformcCs0||_|j|j|d|dd|_dS)z Set the offset of the container. Parameters ---------- xy : (float, float) The (x,y) coordinates of the offset in display units. rrTN)r7rrrrG)r8rHrrrrIs  zAuxTransformBox.set_offsetcCs|jS)z1 return offset of the container. )r7)r8rrrrOszAuxTransformBox.get_offsetcCs8||\}}}}|\}}tj||||||S)z8 get the bounding box in display space. )rYrOrZrr[)r8rNrr'rWrXrdrerrrr^s z!AuxTransformBox.get_window_extentcs||j}|j|jfdd|jD}tj|}|j|j |j |jj |}|j ||j |jddfS)Ncsg|]}|qSr)r^)rr?)rNrrrsz.AuxTransformBox.get_extent..g)rZ to_valuesrrr6rZrunionrrrZmatrix_from_valuesZ set_matrixrKr-)r8rNZ_offZbboxesZubZmtxr)rNrrYs      zAuxTransformBox.get_extentcCs:x|jD]}||qWt||dtdddd|_dS)z# Draw the children Fg)r_)r`raN)r6rbr rcrG)r8rNr?rrrrbs zAuxTransformBox.drawN) rfrgrhrir5rrrrIrOr^rYrbrrrrrXs  rc seZdZdZdZdddddddd d d d Zd)fdd ZddZddZddZ ddZ ddZ d*ddZ dd Z d+d!d"Zd,d#d$Zd%d&Zd'd(ZZS)-AnchoredOffsetboxa7 An offset box placed according to the legend location loc. AnchoredOffsetbox has a single child. When multiple children is needed, use other OffsetBox class to enclose them. By default, the offset box is anchored against its parent axes. You may explicitly specify the bbox_to_anchor. rro ) z upper rightz upper leftz lower leftz lower rightr,z center leftz center rightz lower centerz upper centerr.皙??NTc stjf| |||||t|trjy|j|}Wn,tk rhtd|d |jfYnX||_ ||_ ||_ |dkrt tdd|_n6t|trt f||_d|kr|jtdn||_tdddd d |jd d |_|jjd dd||_dS)a loc is a string or an integer specifying the legend location. The valid location codes are:: 'upper right' : 1, 'upper left' : 2, 'lower left' : 3, 'lower right' : 4, 'right' : 5, (same as 'center right', for back-compatibility) 'center left' : 6, 'center right' : 7, 'lower center' : 8, 'upper center' : 9, 'center' : 10, pad : pad around the child for drawing a frame. given in fraction of fontsize. borderpad : pad between offsetbox frame and the bbox_to_anchor, child : OffsetBox instance that will be anchored. prop : font property. This is only used as a reference for paddings. frameon : draw a frame box if True. bbox_to_anchor : bbox to anchor. Use self.axes.bbox if None. bbox_transform : with which the bbox_to_anchor will be transformed. z4Unrecognized location "%s". Valid locations are %s z Nzlegend.fontsize)sizer)ggg?rrzT)rHrKr-r{r|r}r~rr)r_)r4r5set_bbox_to_anchor set_childrqstrcodesKeyErrorr!joinloc borderpadr_rrproprcZset_sizerget_size_in_pointsrrr) r8rr_rrrframeonbbox_to_anchorZbbox_transformr)r9rrr5s4%      zAnchoredOffsetbox.__init__cCs ||_|dk r|j|_d|_dS)zset the child to be anchoredNT)_childr@rG)r8rrrrr%szAnchoredOffsetbox.set_childcCs|jS)zreturn the child)r)r8rrr get_child,szAnchoredOffsetbox.get_childcCs|jgS)zreturn the list of children)r)r8rrrr=0szAnchoredOffsetbox.get_childrencCsT||\}}}}||j}|j|}|d||d|||||fS)zq return the extent of the artist. The extent of the child added with the pad is returned ro)rrYrprrr_)r8rNrr'rWrXrr_rrrrY4s zAnchoredOffsetbox.get_extentcCs6|jdkr|jjS|j}|dkr&|jSt|j|SdS)zB return the bbox that the legend will be anchored N)_bbox_to_anchorr@r_bbox_to_anchor_transformr)r8 transformrrrget_bbox_to_anchor?s z$AnchoredOffsetbox.get_bbox_to_anchorcCs|dkst|tr||_nZy t|}Wn$tk rJtdt|YnX|dkrh|d|dddg}tj||_||_ d|_ dS)aF set the bbox that the child will be anchored. *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 argument for bbox : %srorrT) rqrrr TypeErrorr!rrr[rrG)r8rrlrrrrMs   z$AnchoredOffsetbox.set_bbox_to_anchorcCsJ||||\}}}}||||||\}}t||||||S)z8 get the bounding box in display space. )_update_offset_funcrYrOrr[)r8rNrr'rWrXrdrerrrr^fs z#AnchoredOffsetbox.get_window_extentcCs4|dkr||j}||fdd}||dS)zs Update the offset func which depends on the dpi of the renderer (because of the padding). Nc SsHtdd||}|j|}|} ||j|| |\} } | || |fS)Nr)rr[rr_get_anchored_bboxr) rr'rWrXrNrr8rrrrrrrrr7xs  z6AnchoredOffsetbox._update_offset_func.._offset)rprrrI)r8rNrr7rrrros   z%AnchoredOffsetbox._update_offset_funccCs.|j|j|j|j|j|r*|j|dS)N)rrrrrKr-r)r8rrrrrrs zAnchoredOffsetbox.update_framec Cs|s dS||j}||||jrP||}||||j || |\}}}}| |||||\}} | || f| |d|_dS)zdraw the artistNF)rRrprrrrr^rrrbrYrOrrIrG) r8rNrrrKr-rLrMr\r]rrrrbs    zAnchoredOffsetbox.drawcCs|tddksttd\ }}}}} } } } } }}|d|d|d| d| d| d| d| d |d |d i }||}|| }|j||d }|j|jfS) z{ return the position of the bbox anchored at the parentbbox with the loc code, with the borderpad. r ZNEZNWZSWZSEEWSNC) container)rangeAssertionErrorZpaddedZanchoredrr)r8rrZ parentbboxrZBESTZURZULZLLZLRRZCLZCRZLCZUCrZ anchor_coefsr?rZ anchored_boxrrrrs  z$AnchoredOffsetbox._get_anchored_bbox)rrNNTNN)N)N)N)rfrgrhrizorderrr5rrr=rYrrr^rrrbrrkrr)r9rrs6D    rcs"eZdZdZdfdd ZZS) AnchoredTextz& AnchoredOffsetbox with Text. 皙??Nc  sn|dkr i}ddddh}|t|@r.tdt||dd|_|jj}tj|f|||j|d |dS) a Parameters ---------- s : string Text. loc : str Location code. pad : float, optional Pad between the text and the frame as fraction of the font size. borderpad : float, optional Pad between the frame and the axes (or *bbox_to_anchor*). prop : dictionary, optional, default: None Dictionary of keyword parameters to be passed to the `~matplotlib.text.Text` instance contained inside AnchoredText. Notes ----- Other keyword parameters of `AnchoredOffsetbox` are also allowed. NZhaZhorizontalalignmentrZverticalalignmentzSMixing horizontalalignment or verticalalignment with AnchoredText is not supported.F)rr)r_rrr) setwarningswarnrZtxtrZget_fontpropertiesr4r5) r8rrr_rrrZ badkwargsfp)r9rrr5s    zAnchoredText.__init__)rrN)rfrgrhrir5rkrr)r9rrsrc @s^eZdZdddZdd Zd d Zd d ZddZddZddZ ddZ ddZ ddZ dS) OffsetImagerN@FTc KsVt|| |_tf|j||||||| d| |_|jg|_||||dS)N)rcmapnorm interpolationorigin filternorm filterradresample) r3r5_dpi_corr r^imager6set_zoomset_data) r8arrzoomrrrrrrrrrrrrr5s    zOffsetImage.__init__cCs$t||_|j|jd|_dS)NT)rZasarray_datarrrG)r8rrrrr s zOffsetImage.set_datacCs|jS)N)r)r8rrrget_dataszOffsetImage.get_datacCs||_d|_dS)NT)_zoomrG)r8rrrrrszOffsetImage.set_zoomcCs|jS)N)r)r8rrrget_zoomszOffsetImage.get_zoomcCs|jS)z1 return offset of the container. )r7)r8rrrrO.szOffsetImage.get_offsetcCs|jgS)N)r)r8rrrr=4szOffsetImage.get_childrencCs8||\}}}}|\}}tj||||||S)z8 get the bounding box in display space. )rYrOrZrr[)r8rNrr'rWrXrdrerrrr^7s zOffsetImage.get_window_extentc Cs^|jr|d}nd}|}|}|jdd\}}||||||}}||ddfS)Ng?ror)rrprrshape) r8rNrrdataZnyZnxrr'rrrrY?s zOffsetImage.get_extentcCs|j|d|_dS)z# Draw the children FN)rrbrG)r8rNrrrrbLs zOffsetImage.draw) rNNNNrrFT) rfrgrhr5rrrrrOr=r^rYrbrrrrrs$  rc @seZdZdZdZddZejd"d d Ze d d Z e j dd Z e ddZ e j ddZ ddZ ddZddZd#ddZd$ddZddZddZd d!ZdS)%AnnotationBboxzD Annotation-like class, but with offsetbox instead of Text. rcCsd|jd|jdfS)NzAnnotationBbox(%g,%g)rr)rH)r8rrr__str__[szAnnotationBbox.__str__NrT皙?g?g?c  Kstjj|f| tj||||d||_| |_|| |dkrH||_n||_|dkr^||_n||_| dk r|j dd|_ t d|j|_ n d|_ d|_ | |_ tdddd d |jd d |_|jjd |d| r|jjf| ||_dS)ao *offsetbox* : OffsetBox instance *xycoords* : same as Annotation but can be a tuple of two strings which are interpreted as x and y coordinates. *boxcoords* : similar to textcoords as Annotation but can be a tuple of two strings which are interpreted as x and y coordinates. *box_alignment* : a tuple of two floats for a vertical and horizontal alignment of the offset box w.r.t. the *boxcoords*. The lower-left corner is (0.0) and upper-right corner is (1.1). other parameters are identical to that of Annotation. )xycoordsannotation_clipNrelpos)g?g?rrrr)ggg?rrzT)rHrKr-r{r|r}r~r)r_)r r )r:r;r5r offsetbox arrowprops set_fontsizexybox boxcoordspop _arrow_relposr arrow_patch_box_alignmentrrrrrrr)r8r rHrrrrr_rZ box_alignment bboxpropsr rrrrrr5^s<   zAnnotationBbox.__init__cCs|jS)N)r)r8rrrxyannszAnnotationBbox.xyanncCs||_d|_dS)NT)rrG)r8rrrrrscCs|jS)N)r)r8rrr anncoordsszAnnotationBbox.anncoordscCs||_d|_dS)NT)rrG)r8ZcoordsrrrrscCs|j|\}}||fS)N)r rC)r8eventrZtinforrrrCszAnnotationBbox.containscCs"|j|jg}|jr||j|S)N)r rrr)r8rmrrrr=s  zAnnotationBbox.get_childrencCs4|jdk r|j||j|tj||dS)N)rr<r r:r;)r8r>rrrr<s   zAnnotationBbox.set_figurecCs&|dkrtd}t|d|_d|_dS)z( set fontsize in points Nzlegend.fontsize)rT)rrrrG)r8rrrrr s zAnnotationBbox.set_fontsizecCs |jS)z+ return fontsize in points )rr)r8rrrr get_fontsizeszAnnotationBbox.get_fontsizecCsF||}|||||}|j||jrB|j|dS)zQ Update the pixel positions of the annotated point and the text. N)_get_position_xy_update_position_xyboxrprrrr)r8rNxy_pixelr}rrrupdate_positionss    zAnnotationBbox.update_positionscCs|j\}}t|jtrT|j\}}|||||\}}|||||\} } || } } n|||||j\} } |j|\} }}}|j\}}|j| || || |||f|j |}|j |j |j |j|j|\}}||}}|jr||}}|j}|j}|j |j|d} |j |j|d} |j| | f||f|j}|d|}||}|j||d|j }|j|dS)z` Update the pixel positions of the annotation text and the arrow patch. rrr}patchAN)rrqrtupleZ_get_xyr rYrrIr^rrrrrKr-r copyrrZ set_positionsrrrrprZ set_patchA)r8rNrxyZxcoordZycoordZx1Zy1Zx2Zy2Zox0Zoy0rr'rWrXZ_fwZ_fhrZox1Zoy1rrr(rZfsr}rrrrrs8     (         z%AnnotationBbox._update_position_xyboxcCs|dk r||_|sdS||}|||s4dS|||jdk rt|jjdkrh|jdk rh|j|j_|j||jr|j ||j |d|_ dS)zN Draw the :class:`Annotation` object to the given *renderer*. NF) Z _rendererrRrZ _check_xyrrrrbrrr rG)r8rNrrrrrb*s         zAnnotationBbox.draw) NrNTrNrNNN)N)N)rfrgrhrirrr Zdedent_interpdr5rrrjrrCr=r<r rrrrbrrrrrUs2 @     r;AttributeError)r8ZcidZc1rrrr?s  zDraggableBase.disconnectcCs |j|S)N)r#rC)r8r9r1rrrr*szDraggableBase.artist_pickercCsdS)Nr)r8rrrr<szDraggableBase.save_offsetcCsdS)Nr)r8r2r3rrrr0szDraggableBase.update_offsetcCsdS)Nr)r8rrrr=szDraggableBase.finalize_offsetN)F)rfrgrhrir5r4r8r(r)r-r?r*r<r0r=rrrrr"Gs    r"c@s.eZdZd ddZddZddZdd Zd S) DraggableOffsetBoxFcCstj|||d||_dS)N)r,)r"r5r )r8r#r r,rrrr5szDraggableOffsetBox.__init__cCsN|j}|jj}||\}}}}||||||}|\|_|_|j|dS)N)r rr6rYrO offsetbox_x offsetbox_yrI)r8r rNrr'rWrXoffsetrrrr<s  zDraggableOffsetBox.save_offsetcCs$|j||j|f}|j|dS)N)rBrCr rI)r8r2r3 loc_in_canvasrrrr0sz DraggableOffsetBox.update_offsetc Cs>|j}|jj}||\}}}}|j\}}||||f} | S)N)r rr6rYr7) r8r rNrr'rWrXrdrerErrrget_loc_in_canvass  z$DraggableOffsetBox.get_loc_in_canvasN)F)rfrgrhr5r<r0rFrrrrrAs rAc@s&eZdZd ddZddZddZdS) DraggableAnnotationFcCstj|||d||_dS)N)r,)r"r5 annotation)r8rHr,rrrr5szDraggableAnnotation.__init__cCs"|j}||j\|_|_dS)N)rHrrrrdre)r8annrrrr<szDraggableAnnotation.save_offsetcCs.|j}||j||j|f|_dS)N)rHrinvertedrrdrer)r8r2r3rIrrrr0s z!DraggableAnnotation.update_offsetN)F)rfrgrhr5r<r0rrrrrGs rG__main__rygp@ro)rrr)drO)g?g?)rPrz offset pointsTg?roundr!)ZboxstyleZfcz->)Z arrowstyle)rrrrr_rrr )r)r&)IrirZnumpyrZmatplotlib.transformsZ transformsrZZmatplotlib.artistr9r:Zmatplotlib.texttextrZmatplotlib.pathpathrrrrZmatplotlib.font_managerrZmatplotlib.patchesrrZ matplotlibrr Zmatplotlib.imager r rr r r%r2r;r3rlrnrxryrrrrrrrobjectr"rArGrfZmatplotlib.pyplotZpyplotZpltrr>ZclfZsubplotrBrcrr ZreshaperEZmyimagerrIZmyimage2rIrbZshowrrrrs           1 %,LNS+ny.is{