@wOGc@ sidZddklZddkZddkZddkZddklZl Z ddk l Z ddk l Z lZlZlZddklZddklZdd klZdd klZlZlZlZlZlZdd klZl Z l!Z!dd k"l#Z#dd k$l%Z%l&Z&l'Z'l(Z(l)Z)l*Z*l+Z+l,Z,dZ-de fdYZ.dS(s Place a legend on the axes at location loc. Labels are a sequence of strings and loc can be a string or an integer specifying the legend location The location codes are 'best' : 0, (only implemented for axis legends) 'upper right' : 1, 'upper left' : 2, 'lower left' : 3, 'lower right' : 4, 'right' : 5, 'center left' : 6, 'center right' : 7, 'lower center' : 8, 'upper center' : 9, 'center' : 10, Return value is a sequence of text, line instances that make up the legend i(tdivisionN(tverbosetrcParams(tArtist(t enumeratetis_string_liketiterablet silent_list(tFontProperties(tLine2D(tsegments_intersect(tPatcht RectangletRegularPolygontShadowt bbox_artistt draw_bbox(tLineCollectiontRegularPolyCollectiontPatchCollection(tText(tBboxtPointtValuetget_bbox_transformtbbox_allt unit_bboxtinverse_transform_bboxt lbwh_to_bboxc C s0|i\}}}}||}||}t|}|djotSn|djo$|i|dd|ddSn|d} x|dD]} | | f} xz||f||ff||f||ff||f||ff||f||fffD]} t| | otSqqW| } qWtS(s, Return True if and only if line cuts bbox. ii(t get_boundstlentFalsetcontainsR tTrue( tlinetbboxtminxtminytwidththeighttmaxxtmaxytntp1tp2tsegmenttedge((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pytline_cuts_bbox+s*     $     tLegendc B sIeZdZhdd<dd<dd<dd<d d <d d <d d<dd<dd<dd<ddR?R6R7tproplistt propnamestnametvaluetleftttopttextlefttHEIGHTtbottom((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyRHrsv                       $" cC s'|i|i|i|idS(N(RQRGRTt get_transform(R4ta((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyRgscC s+|id|iii|iiiS(NgR@(RNRGtdpitgetRSR#R'(R4((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyRescC s |ipdSn|id|i||ioD|io&t|idd}|i|n|ii|nt|i  ot|i  odSnx/|i D]$}|dj o|i|qqWx|i D]}|i|qW|i ddS(Ntlegendg{Gzt( t get_visiblet open_groupt_update_positionsRhR?RRftdrawRRcRaRJt close_group(R4trendererR?thtt((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyR}s$    *    c C sg}|iD]}||i|q~}g}|iD]'}|dj o||i|q;q;~}|}|i|t|} | |_t|i| } | |_ | S(s0Get a bbox for the text and lines in axes coordsN( Ratget_window_extentRcRJtextendRtsaveRRutibox( R4Rt_[1]Rt bboxesTextt_[2]Rt bboxesHandlest bboxesAllR#R((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyt_get_handle_text_bboxs->    c C s|i}g}xt||D]\}}|i\}}||i|i8}t|to||dti|i i t } t|i | } | i ||i | | id| i|i| i|i| q"t|toqtdt|i |d |fd|id|d} | i ||i | | id|i| q"t|to||dti|i i t } t|i | } |i | | id|id} |i} |id}| i|| i| | i| |i| q"t|totdt|i |d |fd|id|d} | i|i d|i!djo| i"|i!dn|i | | id|i| q"|idq"W|S( NiRAiiR&R'iRJg?g?(#ReRIRdR<R=ROR R\tonesR^tshapetfloatt update_fromRgt set_clip_boxRJtset_markersizeR:tget_markersizetappendR R tminRt get_linewidtht get_dashest get_colorst set_colort set_linewidtht set_dashesRt set_facecolort _facecolorst _edgecolorst set_edgecolor(R4RiRaRstretthandletlabeltxtytydatatleglinetptlwtdashestcolor((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyRbsX  $   #   $      #  c C s(|ipt|i}g}g}g}|ii}x|iD]}t|tpt|idt }|i dt }|i } | i ||\} } g} t | | D]} | || q~ }|i|qBWx|iD]}t|tpt|i}|i } | i|}g}|D]} ||| qA~}t}|i|t|i|qWx|iD]}t|top|i}|i } xU|D]I}| i|}g}|D]} ||| q~}|i|qWqqW|||gS(sp Returns list of vertices and extents covered by the plot. Returns a two long list. First element is a list of (x, y) vertices (in axes-coordinates) covered by all the lines and line collections, in the legend's handles. Second element is a list of bounding boxes for all the patches in the legend's handles. torig(RPtAssertionErrorRSt transAxestinverse_xy_tuptlinesROR t get_xdataRt get_ydataRut numerix_x_yRIRtpatchesR t get_vertst seq_xy_tupsRtupdateR!t collectionsRt get_lines(R4taxtverticestbboxesRtinvRtxdataRttranstxttytRtvtavertstvertsttvertsRR#thlinesR"ttlinet_[3]taline((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyt_auto_legend_dataIsH     0   '    'cC s ||_dS(s$b is a boolean. Set draw frame to bN(Rh(R4tb((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyt draw_framescC s|iS(s6return the Rectangle instance used to frame the legend(Rf(R4((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyt get_framescC s6g}|iD]!}t|to ||qq~S(s5return a list of lines.Line2D instances in the legend(RcROR (R4RR((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyRscC s?tdg}|iD]!}t|to ||qq~S(s.return a list of patch instances in the legendR (RRcROR (R4RR((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyt get_patchesscC std|iS(s1return a list of text.Text instance in the legendR(RRa(R4((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyt get_textssc C s|i}|}|}g}xb|D]Z}td|d|d|d|idddd} |i| |i| ||8}q%W|S( NRRttexttfontpropertiestverticalalignmentRqthorizontalalignmentRp(ReRRLRgR( R4RjRptupperRstposRRtlR((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyR`s       cC s |iiS(N(RfR(R4((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyRsc C sx>|iD]3}|i\}}|i||||fq Wx|iD]}t|toA|idt|idt}}|i ||||qKt|t o4|i d||i d<|i d||i d(R4RkR&R'tBESTtURtULtLLtLRtRtCLtCRtLCtUCtCRR((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyRs-  c  s3ti oti odSnfd}g}xtidid D]\}}|i\}}||\}} } } | i8} | di7} |i| | f|i|| d| fq`W|id\}} } } | i8} | di7} |i| | fxti|D]\} } | \}} t| t o5|t i i i t}| i|| dqIt| to*| i|d| | i| dqIqIWii}|ididi|i\}} } } ii|| | | d \}}tio\tidjoFii}ii}i\}}||||}}n`idjoi| | \}}n4ii| | \}}|||| }}i ||dS( Nc s.|i}ti|}|iS(N(RRRuR(RR#tbboxa(R4R(s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyt get_tboundssiiig?iig?(ii(!RRcRaRIRdR;RRROR R\RR^RRt set_ydataR Rt set_heightRtdeepcopytscaleR9RRft set_boundsRR[RRRRR(R4RRthposRttaboveRRRRRCRRttupRR#RRtxotyo((RR4s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyR|sN*      &N(t__name__t __module__t__doc__RUtzorderR5RJRHRgReR}RRbRRRRRRR`RRRRR|(((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyR0FsR             c     7 :         / (/Rt __future__RtsysRWtnumpyR\t matplotlibRRtartistRtcbookRRRRt font_managerRRR tmlabR RR R R RRRRRRRRRt transformsRRRRRRRRR/R0(((s:/Users/cmoad/workspace/matplotlib/lib/matplotlib/legend.pyss ".: