Rc@sddlmZddlZddlZddlZddlZejZddlm Z ddl j Z ddlm Z ddlmZddlmZddlmZddlmZmZmZdd lmZmZmZmZmZmZmZmZm Z m!Z!ddl"j#Z$d efd YZ%d efd YZ&defdYZ'defdYZ(defdYZ)de fdYZ*e%e*_%e&e*_&e'e*_'e(e*_(e)e*_)dS(i(tprint_functionN(tAxes(tcbook(t docstring(tCircle(tPath(t FormattertLocatortFormatStrFormatter( tAffine2Dt Affine2DBasetBboxtBboxTransformTotIdentityTransformt TransformtTransformWrappertScaledTranslationtblended_transform_factorytBboxTransformToMaxOnlytPolarTransformcBsweZdZdZdZeZdedZ dZ e j je _dZ e j je _dZ e j je _RS(s The base polar transform. This handles projection *theta* and *r* into Cartesian coordinate space *x* and *y*, but does not perform the ultimate affine transformation into the correct position. icCs#tj|||_||_dS(N(Rt__init__t_axist _use_rmin(tselftaxistuse_rmin((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR!s  c Csetj|jtj}|jdk ri|jrB|jjj}nd}|jj }|jj }nd}d}d}|ddddf}|ddddf}|ddddf}|ddddf} ||9}||7}||}|dk} tj | tj |tj ||(tj | tj |tj|| (|S(Niii(tnptemptytshapetfloat_RtNoneRtviewLimtymintget_theta_offsettget_theta_directiontwheretnantcostsin( Rttrtxytrmint theta_offsetttheta_directiontttrtxtytmask((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyttransform_non_affine&s*     &&cCsv|j}t|dkrH|d|dkrHt|j||jS|j|j}t|j|j|jS(Niii(ii(ii(tverticestlenRt transformtcodest interpolatedt_interpolation_steps(RtpathR2tipath((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyttransform_path_non_affineDs  &cCstj|j|jS(N(t PolarAxestInvertedPolarTransformRR(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytinvertedLsN(t__name__t __module__t__doc__t input_dimst output_dimstFalset is_separableRtTrueRR1RR:R=(((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRs   t PolarAffinecBs8eZdZdZdZejje_dZRS(s The affine part of the polar projection. Scales the output so that maximum radius rests on the edge of the axes circle. cCs<tj|||_||_|j||d|_dS(s *limits* is the view limit of the data. The only part of its bounds that is used is ymax (for the radius maximum). The theta range is always fixed to (0, 2pi). N(R Rt_scale_transformt_limitst set_childrenRt_mtx(Rtscale_transformtlimits((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRVs    cCs{|jrt|jj|j}|j|j}tjd|jdd}|j |_ d|_ d|_n|j S(Ng?i( t_invalidRHt transformedRGtymaxR R tscalet translatet get_matrixRJRt _inverted(Rt limits_scaledtyscaletaffine((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRRbs      cCsiS(N((R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyt __getstate__os(R>R?R@RRRR RW(((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRFQs  R<cBs_eZdZdZdZeZdedZ dZ e j je _dZ e j je _RS(sy The inverse of the polar transform, mapping Cartesian coordinate space *x* and *y* back to *theta* and *r*. icCs#tj|||_||_dS(N(RRRR(RRR((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR|s  c Cs#|jdk rQ|jr*|jjj}nd}|jj}|jj}nd}d}d}|ddddf}|ddddf}tj||||}tj ||}tj |dkdtj ||}||8}||9}||7}tj ||fdS(Niii( RRRRR R!R"RtsqrttarccosR#tpit concatenate( RR(R)R*R+R.R/R-ttheta((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR1s$ &   cCstj|j|jS(N(R;RRR(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR=sN( R>R?R@RARBRCRDRRERR1RR=(((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR<ss  tThetaFormattercBseZdZddZRS(s Used to format the *theta* tick labels. Converts the native unit of radians into degrees and adds a degree symbol. cCs?tdr(td r(d|tjdSd|tjdSdS(Ns text.usetexstext.latex.unicodes $%0.0f^\circ$gf@u%0.0f°(trcParamsRRZ(RR.tpos((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyt__call__sN(R>R?R@RR`(((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR]st RadialLocatorcBsMeZdZdZdZdZdZdZdZdZ RS(s Used to locate radius ticks. Ensures that all ticks are strictly positive. For all other tasks, it delegates to the base :class:`~matplotlib.ticker.Locator` (which may be different depending on the scale of the *r*-axis. cCs ||_dS(N(tbase(RRb((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRscCs/|j}g|D]}|dkr|^qS(Ni(Rb(RtticksR.((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR`s cCs |jjS(N(Rbt autoscale(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRdscCs|jj|S(N(Rbtpan(Rtnumsteps((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRescCs|jj|S(N(Rbtzoom(Rt direction((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRgscCs |jjS(N(Rbtrefresh(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRiscCs%|jj||\}}d|fS(Ni(Rbt view_limits(Rtvmintvmax((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRjs( R>R?R@RR`RdReRgRiRj(((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRas      R;cBseZdZdZdZejje_dZdZdZddZ dZ d Z dd Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZejd'd'd'dZ!ejd'd'd'dZ"dZ#dZ$d Z%d!Z&d"Z'd#Z(d$Z)d%Z*d&Z+RS((s A polar graph projection, where the input dimensions are *theta*, *r*. Theta starts pointing east and goes anti-clockwise. tpolarcOs|jdd|_|jdd|_|jdd|_|jd kr^tjdntj||||j ddd d d |j d S(s Create a new Polar Axes for a polar plot. The following optional kwargs are supported: - *resolution*: The number of points of interpolation between each pair of data points. Set to 1 to disable interpolation. t resolutioniR*iR+sThe resolution kwarg to Polar plots is now ignored. If you need to interpolate data points, consider running cbook.simple_linear_interpolation on the data before passing to matplotlib.tequalt adjustabletboxtanchortCN(Ni( tpopRnt_default_theta_offsett_default_theta_directionRtwarningstwarnRRt set_aspecttcla(Rtargstkwargs((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRs  cCstj||jjd|jj|jt|j_t j ddd}|j ||j j |j|j j|jtd|jjd|j jd|j jdt|j|j|j|jdS(Ng?ggv@gF@spolaraxes.gridtnonetlabel1On(RRzttitletset_ytxaxistset_major_formatterR]REtisDefault_majfmtRtarangetset_thetagridstyaxistset_major_locatorRatget_major_locatortgridR^tset_ticks_positiontset_tick_paramstset_theta_offsetRutset_theta_directionRv(Rtangles((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRzs   "cCs2tj||_tj||_|jdS(sAmove this out of __init__ because non-separable axes don't use itN(tmaxistXAxisRtYAxisRt_update_transScale(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyt _init_axis scCs|t|j|_tt|_|j||_|j|dt|_ |j |j|j |_ |j|j|j |j|_ |j |j ttj|j|_tjdd|_|j|j|_tjddd|_|j|j|_tjtjdd|j |_tddt|_|jtjddd|j|_dS(NRgg?g?g@g6@gv@(R tbboxt transAxesRR t transScaleRttransProjectionRCttransPureProjectionRFRttransProjectionAffinet transDataR tunitt_xaxis_transformR RQt_theta_label1_positiont_xaxis_text1_transformt_theta_label2_positiont_xaxis_text2_transformRPRRZt_yaxis_transformRt_r_label_positiont_yaxis_text_transform(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyt_set_lim_and_transformss*     RcCs|dkst|jS(Nttick1ttick2R(RRsgrid(tAssertionErrorR(Rtwhich((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytget_xaxis_transformJscCs|jddfS(Ntcenter(R(Rtpad((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytget_xaxis_text1_transformNscCs|jddfS(NR(R(RR((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytget_xaxis_text2_transformQscCs|dkst|jS(NRRR(stick1stick2sgrid(RR(RR((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytget_yaxis_transformTscCs{|jjd}|dkr/|jddfS|dkrK|jddfS|dkrg|jddfS|jddfSdS( NigV@tbottomtleftgf@trightgp@ttop(Rt to_valuesR(RRtangle((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytget_yaxis_text1_transformXs   cCs{|jjd}|dkr/|jddfS|dkrK|jddfS|dkrg|jddfS|jddfSdS( NigV@RRgf@Rgp@R(RRR(RRR((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytget_yaxis_text2_transformcs   cCs tddS(Ng?(g?g?(R(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyt_gen_axes_patchnscCsitjj|ddd6S(Ng?Rm(g?g?(tmspinestSpinetcircular_spine(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyt_gen_axes_spinesqscCs||j_dS(N(Rty1(Rtrmax((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytset_rmaxuscCs |jjS(N(RRO(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytget_rmaxxscCs||j_dS(N(Rty0(RR)((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytset_rmin{scCs |jjS(N(RR (R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytget_rmin~scCs ||_dS(sB Set the offset for the location of 0 in radians. N(t _theta_offset(Rtoffset((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRscCs|jS(sB Get the offset for the location of 0 in radians. (R(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR!scCs|itjdd6tjdd6tjd6tjdd6tjdd 6tjd d 6d d 6tjdd6}|j||S(s Sets the location of theta's zero. (Calls set_theta_offset with the correct value in radians under the hood.) May be one of "N", "NW", "W", "SW", "S", "SE", "E", or "NE". g?tNg?tNWtWg?tSWg?tSg?tSEitEg?tNE(RRZR(Rtloctmapping((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pytset_theta_zero_locations cCsX|dkrd|_n<|d kr0d|_n$|d krH||_n tddS( s Set the direction in which theta increases. clockwise, -1: Theta increases in the clockwise direction counterclockwise, anticlockwise, 1: Theta increases in the counterclockwise direction t clockwiseitcounterclockwiset anticlockwiseis6direction must be 1, -1, clockwise or counterclockwiseN(R(RR(ii(t _directiont ValueError(RRh((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyRs      cCs|jS(s Get the direction in which theta increases. -1: Theta increases in the clockwise direction 1: Theta increases in the counterclockwise direction (R(R((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR"s cOsTd|kr"|jd|dR?R@tnameRRRzRRRRRRRRRRRRRRRR!RRR"RRRRRtdedent_interpdRRRRRRRRRRRR(((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyR;sR   7                  )+     (+t __future__RRRwtnumpyRt matplotlibR^tmatplotlib.axesRtmatplotlib.axisRRRRtmatplotlib.patchesRtmatplotlib.pathRtmatplotlib.tickerRRRtmatplotlib.transformsR R R R R RRRRRtmatplotlib.spinestspinesRRRFR<R]RaR;(((sV/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/matplotlib/projections/polar.pyts4     F;".!