3 ƽhN@sdddlZddlZddlmZmZmZddlmZddlj Z ddl j Z ddljZGddde jZdS)N)cbook docstringrcParams)allow_rasterizationcseZdZdZddZejfddZej dddZ ej dd d Z d d Z d dZ ddZddZfddZd6fdd ZddZddZddZddZd d!Zefd"d#Zd$d%Zd&d'Zd(d)Zd7d*d+Zd,d-Zed.d/Zed0d1Z ed2d3Z!d4d5Z"Z#S)8SpineaI An 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 `~.Spine.set_position` for more information. The default position is ``('outward', 0)``. Spines are subclasses of `.Patch`, and inherit much of their behavior. Spines draw a line, a circle, or an arc depending if `~.Spine.set_patch_line`, `~.Spine.set_patch_circle`, or `~.Spine.set_patch_arc` has been called. Line-like is the default. cCsdS)Nr)selfrr7/tmp/pip-build-7iwl8md4/matplotlib/matplotlib/spines.py__str__sz Spine.__str__c stjf|||_|j|jj||_|jd|jtd|j td|j dd|_ |j d|j |jjd|_d|_d|_tjtjj|d||_d |_tj|_dS) a Parameters ---------- axes : `~matplotlib.axes.Axes` The `~.axes.Axes` instance containing the spine. spine_type : str The spine type. path : `~matplotlib.path.Path` The `.Path` instance used to draw the spine. Other Parameters ---------------- **kwargs Valid keyword arguments are: %(Patch)s nonezaxes.edgecolorzaxes.linewidthZ projectingNg@F)pathline)super__init__axesZ set_figurefigure spine_typeZ set_facecolor set_edgecolorrZ set_linewidthZ set_capstyleaxisZ set_zorder set_transform transData_bounds _smart_bounds _positionrZ_check_isinstance matplotlibr Path_path _patch_type mtransformsZIdentityTransform_patch_transform)rrrr kwargs) __class__rr rs$   zSpine.__init__z3.2cCsB||_|jdkr |jjj|n|jdkr8|jjj|d|_dS) z7Set the spine and associated axis to have smart bounds.leftrighttopbottomTN)r"r#)r$r%)rrrZyaxisset_smart_boundsZxaxisstale)rvaluerrr r&Us   zSpine.set_smart_boundscCs|jS)z*Return whether the spine has smart bounds.)r)rrrr get_smart_boundsaszSpine.get_smart_boundscCsTd|_||_|d|_|d|_||_||_tjj|||_ |j |j j d|_ dS)zSet the spine to be arc-like.arcTN)r_center_width_heightZ_theta1Z_theta2mpathrr*rrr transAxesr')rcenterradiustheta1theta2rrr set_patch_arcfs  zSpine.set_patch_arccCs8d|_||_|d|_|d|_|j|jjd|_dS)zSet the spine to be circular.circler+TN)rr,r-r.rrr0r')rr1r2rrr set_patch_circless   zSpine.set_patch_circlecCsd|_d|_dS)zSet the spine to be linear.r TN)rr')rrrr set_patch_line}szSpine.set_patch_linecCsj|jdkst|j|jd|j|jdf}|j|j}|j|j}tjj |d|dj ||_ dS)a! Notes ----- 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*r6rg?N)r*r6) rAssertionErrorZconvert_xunitsr,Zconvert_yunitsr-r.rAffine2DZscale translater)rr1widthheightrrr _recompute_transforms   zSpine._recompute_transformcs&|jdkr|j|jStjSdS)Nr*r6)r*r6)rr?rrget_patch_transform)r)r!rr r@s zSpine.get_patch_transformNc s|jtj|d}|jdkr$|S|g}|jjdg}t|jjdkrZ|j|jjdx,|D]"}|j}|j }|j }|dkrd} d} n|dkrd} d} nd} d} | |d|j j } | |d|j j } |j jr&|jd kr|j| |_|j| |_n$|jd kr&|j| |_|j| |_|jjr||jd krX|j| |_|j| |_n$|jd kr||j| |_|j| |_|j|qbWtjj|S) a Return the window extent of the spines in display space, including padding for ticks (but not their labels) See Also -------- matplotlib.axes.Axes.get_tightbbox matplotlib.axes.Axes.get_window_extent )rendererNrr9outing?Hr"r%r#r$)_adjust_locationrget_window_extentrZ majorTickslenZ minorTicksappendfrozen_sizeZ_tickdirrZdpiZ tick1lineZ get_visiblerZx0x1Zy0y1Z tick2linerZBboxunion) rrAZbbZbboxesZ tickstocheckZtickZbb0ZticklZtickdirZpadoutZpadin)r!rr rFsL              zSpine.get_window_extentcCs|jS)N)r)rrrr get_pathszSpine.get_pathcCs |jdkrd|_|j|jdS)Noutward)rOrP)r set_position)rrrr _ensure_position_is_sets zSpine._ensure_position_is_setcCs$||_|jdk r|jjd|_dS)z Register 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|jjdS)zClear the current spine.N)rrrS)rrrr rSs z Spine.clacCs|jdkrdS|jdkrh|jdkr4|jjj\}}n(|jdkrN|jjj\}}ntd|j|jrrt||g\}}|jdkr|jj j\}}|jj }n"|jdkr|jj j\}}|jj }t j |}t||g\}}||kr|}n:||k||k@}||} t| r| d}n|}t||}||kr,|}n:||k||k@}||} t| rX| d }n|}t||}n |j\}}|jd kr|jdkry|jj} Wntk rd} YnXy|jj} Wntk rd } YnX|| | }|| | }||kr ||}}tjjt j|t j||_|jdkr|jjj\} } y|jj}Wntk rh| }YnX| || |}||_||_ntd |jnf|jj}|jdkstd |jdkr||d<||d<n,|jdkr||d<||d<ntd |jdS)z4Automatically set spine bounds to the view interval.r6Nr"r#r$r%zunknown spine spine_type: %sr9rr*z#unable to set bounds for spine "%s"r+zunexpected vertices shape)r"r#)r$r%)r"r#)r$r%)r%r$)r+r+)r"r#)rr9)r9r9)r%r$)rr)r9r) rrrZviewLimZ intervalyZ intervalx ValueErrorrsortedZdataLimZ get_yticksZ get_xticksnpsortrGmaxminrZget_theta_directionAttributeErrorZget_theta_offsetr/rr*Zrad2degrZ get_roriginr.r-Zverticesshaper:)rlowhighZ viewlim_lowZ viewlim_highZ datalim_lowZ datalim_highZticksZcondZtickvals directionoffsetZrminZrmaxZroriginZscaled_diameterZv1rrr rEs                                 zSpine._adjust_locationcs|jtj|}d|_|S)NF)rErdrawr')rrAret)r!rr rb\s z Spine.drawcCsd|d kr n(t|dkrtd|dd kr2td ||_|j|j|jd k rZ|jjd |_d S)aR Set 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 place the spine inwards.) * 'axes': place the spine at the specified Axes coordinate (0 to 1). * 'data': place the spine at the specified data coordinate. Additionally, shorthand notations define a special positions: * 'center' -> ('axes', 0.5) * 'zero' -> ('data', 0.0) r1zeror+z&position should be 'center' or 2-tuplerrOrdataz:position[0] should be one of 'outward', 'axes', or 'data' NT)r1rd)rOrre)rGrVrrget_spine_transformrZ reset_ticksr')rpositionrrr rQcs    zSpine.set_positioncCs|j|jS)zReturn the spine position.)rRr)rrrr get_positionszSpine.get_positioncCs|j|j}t|tr2|dkr&d}n |dkr2d}t|dksFtd|\}}tjdd dg|d |jdkr||j j d d}n*|jdkr|j j d d}nt d|j|d kr|dkr|Sddddd|j}|t j|d}|tj||jjfSn|dkrT|jd kr,tjjdddd|d|S|jd!krtjjddddd||Sn~|dkr|jd"krr|d8}|jd#krtjtjj|d|j j|j jS|jd$krtj|j jtjjd||j jSdS)%zReturn the spine transform.r1r?rdrerr+zposition should be 2-tuplerO) position_typer"r#Zgrid)whichr$r%zunknown spine spine_type: r9)r"r#r%r$rDN)rri)rer)r"r#)r$r%rU)rUr)r9rrU)rrU)rr9)r"r#)r%r$)r#r$)r"r#)r%r$)rRr isinstancestrrGr:rZ_check_in_listrrZget_yaxis_transformZget_xaxis_transformrVrXarrayrZScaledTranslationrZdpi_scale_transr;Z from_valuesZblended_transform_factoryr<r)rrgrjamountZbase_transformZ offset_vecZ offset_dotsrrr rfs\             zSpine.get_spine_transformcCsh|jdkrtd|dkr,tj|r,|\}}|jp6d\}}|dkrH|}|dkrT|}||f|_d|_dS)a Set the spine bounds. Parameters ---------- low : float or None, optional The lower spine bound. Passing *None* leaves the limit unchanged. The bounds may also be passed as the tuple (*low*, *high*) as the first positional argument. .. ACCEPTS: (low: float, high: float) high : float or None, optional The higher spine bound. Passing *None* leaves the limit unchanged. r6z5set_bounds() method incompatible with circular spinesNT)NN)rrVrXiterable get_boundsrr')rr^r_Zold_lowZold_highrrr set_boundss  zSpine.set_boundscCs|jS)zGet the bounds of the spine.)r)rrrr rqszSpine.get_boundscKs|dkrtjd d g}nT|dkr0tjd d g}n<|dkrHtjddg}n$|dkr`tjddg}n td|||||f|}|jtd j||S)z#Create and return a linear `Spine`.r"+?r#?r%r$z"unable to make path for spine "%s"zaxes.spines.{0})rsrt)rsrt)rurt)rurt)rtrs)rtrs)rtru)rtru)r/rrVZ set_visiblerformat)clsrrr r resultrrr linear_spines zSpine.linear_spinec Ks2tjj||}||||f|} | j||||| S)z!Create and return an arc `Spine`.)r/rr*r5) rwrrr1r2r3r4r r rxrrr arc_spineszSpine.arc_spinecKs.tjj}d}||||f|}|j|||S)z%Create and return a circular `Spine`.r6)r/rZ unit_circler7)rwrr1r2r r rrxrrr circular_spines   zSpine.circular_spinecCs|j|d|_dS)aL Set the edgecolor. Parameters ---------- c : color Notes ----- This method does not modify the facecolor (which defaults to "none"), unlike the `.Patch.set_color` method defined in the parent class. Use `.Patch.set_facecolor` to set the facecolor. TN)rr')rcrrr set_color s zSpine.set_color)N)NN)$__name__ __module__ __qualname____doc__r rZdedent_interpdrrZ deprecatedr&r)r5r7r8r?r@rFrNrRrTrSrErrbrQrhrfrrrq classmethodryrzr{r} __classcell__rr)r!r r s46    8 l;   r)numpyrXrrrrZmatplotlib.artistrZmatplotlib.transformsZ transformsrZmatplotlib.patchesZpatchesZmpatchesZmatplotlib.pathr r/ZPatchrrrrr s