B ]t\R@snddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddlmZGddde jZdS)N) docstringrcParams)allow_rasterizationcseZdZdZddZejfddZddZdd Z d d Z d d Z ddZ ddZ fddZd9fdd ZddZddZddZddZdd Zd!d"Zefd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zed1d2Zed3d4Z ed5d6Z!d7d8Z"Z#S):Spineaan axis spine -- the line noting the data area boundaries Spines are the lines connecting the axis tick marks and noting the boundaries of the data area. They can be placed at arbitrary positions. See function:`~matplotlib.spines.Spine.set_position` for more information. The default position is ``('outward',0)``. Spines are subclasses of class:`~matplotlib.patches.Patch`, and inherit much of their behavior. Spines draw a line, a circle, or an arc depending if function:`~matplotlib.spines.Spine.set_patch_line`, function:`~matplotlib.spines.Spine.set_patch_circle`, or function:`~matplotlib.spines.Spine.set_patch_arc` has been called. Line-like is the default. cCsdS)Nr)selfrr0lib/python3.7/site-packages/matplotlib/spines.py__str__!sz Spine.__str__c stjf|||_||jj||_|d|td| td| dd|_ | d| |jjd|_d|_d|_t|tjjstd||_d |_t|_dS) z - *axes* : the Axes instance containing the spine - *spine_type* : a string specifying the spine type - *path* : the path instance used to draw the spine Valid kwargs are: %(Patch)s Znonezaxes.edgecolorzaxes.linewidthZ projectingNg@Fz4'path' must be an instance of 'matplotlib.path.Path'line)super__init__axesZ set_figurefigure spine_typeZ set_facecolor set_edgecolorrZ set_linewidthZ set_capstyleaxisZ set_zorder set_transform transData_bounds _smart_bounds _position isinstance matplotlibpathPath ValueError_path _patch_type mtransformsIdentityTransform_patch_transform)rr rrkwargs) __class__rrr $s(    zSpine.__init__cCsB||_|jdkr |jj|n|jdkr8|jj|d|_dS)z6set the spine and associated axis to have smart bounds)leftright)topbottomTN)rrr Zyaxisset_smart_boundsZxaxisstale)rvaluerrrr'Ss   zSpine.set_smart_boundscCs|jS)z&get whether the spine has smart bounds)r)rrrrget_smart_bounds^szSpine.get_smart_boundscCsTd|_||_|d|_|d|_||_||_tj|||_ | |j j d|_ dS)zset the spine to be arc-likearcTN)r_center_width_heightZ_theta1Z_theta2mpathrr+rrr transAxesr()rcenterradiustheta1theta2rrr set_patch_arcbs  zSpine.set_patch_arccCs8d|_||_|d|_|d|_||jjd|_dS)zset the spine to be circularcircler,TN)rr-r.r/rr r1r()rr2r3rrrset_patch_circleos   zSpine.set_patch_circlecCsd|_d|_dS)zset the spine to be linearr TN)rr()rrrrset_patch_lineyszSpine.set_patch_linecCsj|jdkst||jd||jdf}||j}||j}t |d|dj ||_ dS)aNOTE: This cannot be called until after this has been added to an Axes, otherwise unit conversion will fail. This makes it very important to call the accessor method and not directly access the transformation member variable. )r+r7rg?N) rAssertionErrorZconvert_xunitsr-Zconvert_yunitsr.r/rAffine2DZscale translater )rr2widthZheightrrr_recompute_transforms  zSpine._recompute_transformcs&|jdkr||jStSdS)N)r+r7)rr?r r get_patch_transform)r)r"rrr@s zSpine.get_patch_transformNcs|tj|dS)N)renderer)_adjust_locationr get_window_extent)rrA)r"rrrCszSpine.get_window_extentcCs|jS)N)r)rrrrget_pathszSpine.get_pathcCs |jdkrd|_||jdS)N)outwardg)r set_position)rrrr_ensure_position_is_sets zSpine._ensure_position_is_setcCs$||_|jdk r|jd|_dS)zregister an axis An axis should be registered with its corresponding spine from the Axes instance. This allows the spine to clear any axis properties when needed. NT)rclar()rrrrr register_axiss  zSpine.register_axiscCsd|_|jdk r|jdS)zClear the current spineN)rrrH)rrrrrHs z Spine.clacCsj||j}t|tr2|dkr&d}n |dkr2d}t|dkrFtd|\}}|dkrb|dkrbd Sd Sd S) zreturn True if directly on axes frame This is useful for determining if a spine is the edge of an old style MPL plot. If so, this function will return True. r2)r g?zero)datarr,zposition should be 2-tuplerErTFN)rGrrstrlenr)rposition position_typeamountrrr is_frame_likes  zSpine.is_frame_likecCs|jdkrdS|jdkrh|jdkr4|jjj\}}n(|jdkrN|jjj\}}ntd|j|jrrt||g\}}|jdkr|jj j\}}|j }n"|jdkr|jj j\}}|j }t |}t||g\}}||kr|}n:||k||k@}||} t| r| d}n|}t||}||kr,|}n:||k||k@}||} t| rX| d}n|}t||}n |j\}}|jdkr|jd kry|j} Wntk rd } YnXy|j} Wntk rd} YnX|| | }|| | }||kr ||}}tjt |t ||_|jd kr|jjj\} } y|j}Wntk rh| }YnX| || |}||_||_ntd |jnf|jj}|jd kstd|jdkr||d<||d<n,|jd kr||d<||d<ntd |jdS)z3automatically set spine bounds to the view intervalr7N)r#r$)r%r&zunknown spine spine_type: %srr+)r&r%r:r&z#unable to set bounds for spine "%s")r,r,zunexpected vertices shape)rr:)r:r:)rr)r:r) rrr ZviewLimZ intervalyZ intervalxrrsortedZdataLimZ get_yticksZ get_xticksnpsortrMmaxminrZget_theta_directionAttributeErrorZget_theta_offsetr0rr+Zrad2degrZ get_roriginr/r.Zverticesshaper;)rlowhighZ viewlim_lowZ viewlim_highZ datalim_lowZ datalim_highZticksZcondZtickvals directionoffsetZrminZrmaxZroriginZscaled_diameterZv1rrrrBs                                 zSpine._adjust_locationcs|t|}d|_|S)NF)rBr drawr()rrAZret)r"rrr^7s z Spine.drawc Cs||j}t|tr2|dkr&d}n |dkr2d}t|dksFtd|\}}|dksZt|dkr|d krzd tf|_nv|j d krd d ddd|j }||d d}||dd}dt |||j j f|_nt d|j d tf|_n|dkrr|j dkr(dtjd d d d|d f|_nH|j dkrRdtjdd d d d |f|_nt d|j d tf|_n|dkr|j dkr|d8}|j dkrdt|d f|_nB|j dkrdtd |f|_nt d|j d tf|_dS)z5calculate the offset transform performed by the spiner2)r g?rJ)rKrr,zposition should be 2-tuple)r rErKrEridentity)r#r$r%r&)rRr)r:r)rrR)rr:)r#r$r&r%gR@r:postz2unknown spine type "%s": no spine offset performedr )r#r$pre)r&r%rK)r$r%N)rGrrrLrMr;rr_spine_transformrZScaledTranslationrZdpi_scale_transwarningswarnr<Z from_valuesr=)rrNrOrPZ offset_vecZoffset_xZoffset_yrrr_calc_offset_transform>sv             zSpine._calc_offset_transformcCsl|dkr n(t|dkrtd|ddkr2td||_||||jdk rb|jd|_dS) apset the position of the spine Spine position is specified by a 2 tuple of (position type, amount). The position types are: * 'outward' : place the spine out from the data area by the specified number of points. (Negative values specify placing the spine inward.) * 'axes' : place the spine at the specified Axes coordinate (from 0.0-1.0). * 'data' : place the spine at the specified data coordinate. Additionally, shorthand notations define a special positions: * 'center' -> ('axes',0.5) * 'zero' -> ('data', 0.0) )r2rJr,z&position should be 'center' or 2-tupler)rEr rKz:position[0] should be one of 'outward', 'axes', or 'data' NT) rMrrrerget_spine_transformrZ reset_ticksr()rrNrrrrFs    zSpine.set_positioncCs||jS)zget the spine position)rGr)rrrr get_positionszSpine.get_positioncCs||j\}}|dkr|jj||jj|jj}|jdkrRt||jj }n*|jdkrnt|jj |}nt d|j|S|jdkr|jj dd}n(|jdkr|jj dd}nt d|j|dkr|S|dkr||S|d kr||St d |d S) zget the spine transformrK)r#r$)r%r&zunknown spine spine_type: %sZgrid)Zwhichr_r`raz unknown spine_transform type: %sN) rGrbr Z transScaleZ transLimitsr1rrZblended_transform_factoryrrZget_yaxis_transformZget_xaxis_transform)rZwhatZhowZ data_xformresultZbase_transformrrrrfs6       zSpine.get_spine_transformcCs&|jdkrtd||f|_d|_dS)zSet the bounds of the spine.r7z5set_bounds() method incompatible with circular spinesTN)rrrr()rrZr[rrr set_boundss   zSpine.set_boundscCs|jS)zGet the bounds of the spine.)r)rrrr get_boundsszSpine.get_boundscKs|dkrtddg}nT|dkr0tddg}n<|dkrHtddg}n$|dkr`tddg}n td |||||f|}|td ||S) zA (staticmethod) Returns a linear :class:`Spine`. r#)gg+?r$)g?g+?r&)g+?gr%)g+?g?z"unable to make path for spine "%s"zaxes.spines.{0})r0rrZ set_visiblerformat)clsr rr!rrhrrr linear_spines zSpine.linear_spinec Ks2tj||}||||f|} | ||||| S)z> (classmethod) Returns an arc :class:`Spine`. )r0rr+r6) rlr rr2r3r4r5r!rrhrrr arc_spineszSpine.arc_spinecKs.tj}d}||||f|}||||S)zC (staticmethod) Returns a circular :class:`Spine`. r7)r0rZ unit_circler8)rlr r2r3r!rrrhrrrcircular_spines   zSpine.circular_spinecCs||d|_dS)a Set the edgecolor. Parameters ---------- c : color or sequence of rgba tuples .. seealso:: :meth:`set_facecolor`, :meth:`set_edgecolor` For setting the edge or face color individually. TN)rr()rcrrr set_colors zSpine.set_color)N)$__name__ __module__ __qualname____doc__r rZdedent_interpdr r'r*r6r8r9r?r@rCrDrGrIrHrQrBrr^rerFrgrfrirj classmethodrmrnrorq __classcell__rr)r"rr s8/     lJ'%  r)rcZnumpyrTrrrZmatplotlib.artistrZmatplotlib.transformsZ transformsrZmatplotlib.patchesZpatchesZmpatchesZmatplotlib.pathrr0ZPatchrrrrrs