Rc@sdZddljZddljZddljZ ddl j Z ddl m Z ddlmZmZmZmZmZddlmZddlmZddlmZddlZddlZddljZddlm Z dd l!m"Z"m#Z#d e$fd YZ%d d#d YZ&de$fdYZ'de'fdYZ(de$fdYZ)dej*fdYZ*ej+e*Z,de*fdYZ-ej+e-Z.e/dkrddl0j1Z2e2j3dd$Z4e,e4dddZ5e4j6e5ej7ddej8dZ9e5j:e9ej;e9e5j<de5j=d j>j?e@e5j=d!j>j?e@e5j=d!jAd"e2jBe2jCndS(%s Axislines includes modified implementation of the Axes class. The biggest difference is that the artists responsible to draw axis line, ticks, ticklabel and axis labels are separated out from the mpl's Axis class, which are much more than artists in the original mpl. Originally, this change was motivated to support curvilinear grid. Here are a few reasons that I came up with new axes class. * "top" and "bottom" x-axis (or "left" and "right" y-axis) can have different ticks (tick locations and labels). This is not possible with the current mpl, although some twin axes trick can help. * Curvilinear grid. * angled ticks. In the new axes class, xaxis and yaxis is set to not visible by default, and new set of artist (AxisArtist) are defined to draw axis line, ticks, ticklabels and axis label. Axes.axis attribute serves as a dictionary of these artists, i.e., ax.axis["left"] is a AxisArtist instance responsible to draw left y-axis. The default Axes.axis contains "bottom", "left", "top" and "right". AxisArtist can be considered as a container artist and has following children artists which will draw ticks, labels, etc. * line * major_ticks, major_ticklabels * minor_ticks, minor_ticklabels * offsetText * label Note that these are separate artists from Axis class of the original mpl, thus most of tick-related command in the original mpl won't work, although some effort has made to work with. For example, color and markerwidth of the ax.axis["bottom"].major_ticks will follow those of Axes.xaxis unless explicitly specified. In addition to AxisArtist, the Axes will have *gridlines* attribute, which obviously draws grid lines. The gridlines needs to be separated from the axis as some gridlines can never pass any axis. iN(tPath(tAffine2DtScaledTranslationtIdentityTransformtTransformedPathtBbox(tLineCollection(trcParams(tallow_rasterization(t AxislineStyle(t AxisArtisttGridlinesCollectiontAxisArtistHelpercBsPeZdZdefdYZdefdYZdefdYZRS(sg AxisArtistHelper should define following method with given APIs. Note that the first axes argument will be axes attribute of the caller artist. # LINE (spinal line?) def get_line(self, axes): # path : Path return path def get_line_transform(self, axes): # ... # trans : transform return trans # LABEL def get_label_pos(self, axes): # x, y : position return (x, y), trans def get_label_offset_transform(self, axes, pad_points, fontprops, renderer, bboxes, ): # va : vertical alignment # ha : horizontal alignment # a : angle return trans, va, ha, a # TICK def get_tick_transform(self, axes): return trans def get_tick_iterators(self, axes): # iter : iterable object that yields (c, angle, l) where # c, angle, l is position, tick angle, and label return iter_major, iter_minor t_BasecBs eZdZdZdZRS(s- Base class for axis helper. cCsd\|_|_dS(s gh㈵>N(gh㈵>gh㈵>(tdelta1tdelta2(tself((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt__init__|scCsdS(N((Rtaxes((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt update_lims(t__name__t __module__t__doc__RR(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR xs tFixedc BsqeZdZeddddddddZddZdZd Zd Z d Z d Z d Z RS(sI Helper class for a fixed (in the axes coordinate) axis. tleftitrightitbottomttopcCs||_|d kr(td|n|d kra|d krId}qa|d krad}qan||_ttj|j|j||_ t j ddgd d gg}d|}|j ||d d |f x axis, nth_coord = 1 -> y axis RRRRs%siigg?N(sleftsrightsbottomstop(sleftsright(sbottomstop(t_loct ValueErrortNonet nth_coordtsuperR RRt_default_passthru_ptt passthru_pttnptarrayRt_path(RtlocRt_vertst fixed_coord((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs          cCs|jS(N(R(R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt get_nth_coordscCs|jS(N(R%(RR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytget_linescCs|jS(N(t transAxes(RR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytget_line_transformscCs|jS(N(R+(RR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytget_axislabel_transformsc Cs>|j}tdd dd ddd d|\}}||fS(s label reference position in transAxes. get_label_transform() returns a transform of (transAxes+offset) Rgg?iZRg?RiR(gg?((gg?iZ(g?g?((g?g?iZ(g?g((g?gi(g?g?((g?g?i(Rtdict(RRR&tpost angle_tangent((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytget_axislabel_pos_angles   cCs#|j|jg|j}|S(N(tget_xaxis_transformtget_yaxis_transformR(RRt trans_tick((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytget_tick_transforms (ii(ii(ii(iiN( RRRR.R!RRR)R*R,R-R1R5(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs   #     tFloatingcBs#eZdZdZdZRS(cCs,||_||_ttj|jdS(N(Rt_valueR R R6R(RRtvalue((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs   cCs|jS(N(R(R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR)scCstddS(Ns6get_line method should be defined by the derived class(t RuntimeError(RR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR*s(RRRR)R*(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR6s (RRRtobjectR RR6(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR Gs/ WtAxisArtistHelperRectlinearcBs:eZdejfdYZdejfdYZRS(RcBseZddZdZRS(cCs<ttj|j|||j|jg|j|_dS(s nth_coord = along which coordinate value varies in 2d, nth_coord = 0 -> x axis, nth_coord = 1 -> y axis N(R R;RRtxaxistyaxisRtaxis(RRR&R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs c s8j}|dkr$d\n d\jj}|j}|jj|gt|D]\}}|j||^qe}jj}|j} |jj| gt| D]\}}|j||^q} j|} | |j j fd} | ||| | | fS( s tick_loc, tick_angle, tick_labelRRiZic3sxt||D]|\}}tj}||j<j|}dj|jkosdjknr||fVqqWdS(Ngg?(tziptlistR"Rttransform_pointRR(tlocstlabelstxtltctc2(t angle_normalR0Rttr2ax(s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt_fs  1(sbottomstop(iZi(iiZ( RR>tmajortlocatort formattertset_locst enumeratetminorR5R+tinverted( RRR&RKt majorLocstitvalt majorLabelsRPt minorLocst minorLabelsR4RJ((RHR0RRIs[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytget_tick_iteratorss      1  1 N(RRRRRX(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs R6cBsJeZddZdZdZdZdZdZdZRS(RcCsEttj|j||||_|j|jg|j|_dS(N( R R;R6Rt_axis_directionR<R=RR>(RRRtpassingthrough_pointtaxis_direction((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR/s  cCstjddgddgg}d|j}|j|jj}|j|j|jg}|||dd|fRKRLRMRNRORPR\R+RQ( RRR&RKRRRSRTRURPRVRWRJ((RHR0RRIs[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRXns$      1  1 ( RRRR*R,R-R1R5RX(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR6.s    (RRR RR6(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR;s;tGridHelperBasecBsGeZdZdZdZdZdZdZdZRS(cCs)t|_d|_tt|jdS(N(tTruet _force_updateRt _old_limitsR RaR(R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs  cCs|j\}}|j\}}|jsH|j||||fkr|j||||t|_||||f|_ndS(N(tget_xlimtget_ylimRcRdt_updatetFalse(RRtx1tx2ty1ty2((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs $ cCsdS(N((RRiRjRkRl((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRgscCs t|_dS(N(RbRc(R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt invalidatescCs|j S(N(Rc(R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytvalidscCsgS(s Return list of grid lines as a list of paths (list of points). *which* : "major" or "minor" *axis* : "both", "x" or "y" ((RtwhichR>((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt get_gridlinessc Csatdd|jdtddtddtd}|j||j||jj||S( s Create and return a new GridlineCollection instance. *which* : "major" or "minor" *axis* : "both", "x" or "y" t transformtcolorss grid.colort linestylessgrid.linestylet linewidthssgrid.linewidthN(R RR\Rt_set_artist_propstset_grid_helperR(Rtaxt gridlines((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt new_gridliness     ( RRRRRgRmRnRpRy(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRas     tGridHelperRectlinearcBsDeZdZdddddZdddZdddZRS(cCs tt|j||_dS(N(R RzRR(RR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRscCsn|dkr%tjd|j}ntj|||}|dkrO|}nt||d|d|}|S(Ns6'new_fixed_axis' explicitly requires the axes keyword.toffsetR[(RtwarningstwarnRR;RR (RR&RR[R{Rt_helpertaxisline((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytnew_fixed_axiss      Rc Cs|dkr%tjd|j}n||f}|j}tj||||}t||}|jj t |jj |jj |S(Ns9'new_floating_axis' explicitly requires the axes keyword.( RR|R}RR\R;R6R tlinet set_clip_onRbt set_clip_boxtbbox( RRR8R[Rtpassthrough_pointRqR~R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytnew_floating_axiss     RKtbothc Cspg}|dkrg}|jj\}}|dkrX|j|jjjjn|dkr|j|jjjjnx0|D]%}|j||g||ggqWn|d krl|jj\}} g}|jj j r |j|jj jjn|jj j r9|j|jj jjnx0|D]%} |j|| g| | ggq@Wn|S( s return list of gridline coordinates in data coordinates. *which* : "major" or "minor" *axis* : "both", "x" or "y" RRDRKRPty(RRD(Rsmajor(Rsminor(RR( RRftextendR<RKRLRPtappendReR=t _gridOnMajort _gridOnMinor( RRoR>RxRBRkRlRDRiRjR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRps(    &  &N(RRRRRRRp(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRzs   tSimpleChainedObjectscBs#eZdZdZdZRS(cCs ||_dS(N(t_objects(Rtobjects((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR<scCs/tg|jD]}t||^q }|S(N(RRtgetattr(Rtktat_a((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt __getattr__?s+cOs%x|jD]}|||q WdS(N(R(Rtkltkwargstm((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt__call__Cs(RRRRR(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR;s  tAxescBseZdefdYZdZddZdZddZdZ e e Z ddZ ddZ d Zd Zdd d d ZdZdZddZddZedZedZRS(tAxisDictcBs#eZdZdZdZRS(cCs#||_ttj|jdS(N(RR RRR(RR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRKs cCst|tr>tg|D]}tj||^q}|St|tr|jdkr|jdkr|j dkrt|j }|St dntj||SdS(NsUnsupported slice( t isinstancettupleRR.t __getitem__tslicetstartRtstoptsteptvaluesR(RRtk1tr((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyROs+-cOstjj|j||S(N(tmaxesRR>R(RtvR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR\s(RRRRR(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRJs  cOsf|jdd}t|_|r-||_nt||_tt|j|||j tdS(Nt grid_helper( tpopRRbt _axisline_ont _grid_helperRzR RRttoggle_axisline(RRtkwthelper((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR`s  cCs|dkr|j }n|rrt|_x$|jjD]}|jtq8W|jjt|jjtnPt|_x$|jjD]}|jtqW|jjt|jjtdS(N( RRRbtspinesRt set_visibleRhR<R=(Rtbts((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRqs    cCstt|jdS(N(R Rt _init_axis(R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRscCs|dkr|}n|j||_|jj}x<ddddgD](}|d|d|d||j|N(R RtgridRRRR<RRR=RbRhRxt set_whichtset_axisRtlentmartisttsetp(RRRoR>R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs%     cCsK|jr%|jj|jg}ng}|jtt|j|S(N(RRRRxRR Rt get_children(Rtchildren((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs  cCs|jjdS(N(RRm(R((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pytinvalidate_grid_helpersc Cs7|j}|j|ddddd|d|}|S(NRR[R{R(RRR(RR&R{tghR>((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs   RcCs.|j}|j||d|d|}|S(NR[R(RR(RRR8R[RR>((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs   cCs~|js&tt|j||dS|j}|jt|jj|jg|_tt|j||||_dS(N( RR RtdrawtartistsR@RRRx(Rtrenderertinframet orig_artists((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRs  )cCstt|j||}|js(|S|g}x?|jjD].}|jsYqAn|j|j|qAWtj g|D]0}|r|j dks|j dkr|^q}|S(Ni( R Rt get_tightbboxRRRt get_visibleRRtuniontwidththeight(RRtcall_axes_locatortbb0tbbRRt_bbox((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR s   FN(RRR.RRRRRRRtpropertyR>RyRRRRRRRRRhRRbR(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyRHs$            tAxesZerocBseZdZdZRS(cOstt|j||dS(N(R RR(RRR((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyR2sc Cstt|j|jj}|ddddddd|}|jj|j|jt ||j d<|dd dddd d|}|jj|j|jt ||j d t major_tickst set_tick_outRbt set_labelRtshow(((s[/opt/xtal/ccp4-6.5-linux32/lib/python2.7/site-packages/mpl_toolkits/axisartist/axislines.pyt,sH(  =c