m[c@`sdZddlmZmZmZmZddlZddljZ ddl j Z ddl jZddlmZddlmZmZmZmZmZddlmZddlmZddlmZddlZddlZddlj Z!d d l"m#Z#d e!j$fd YZ%d e&fdYZ'ddlm(Z(de)fdYZ*ddlm$Z$de$e*fdYZ+de j,fdYZ-de-e*fdYZ.de.e*fdYZ/defdYZ0de j(fdYZ1dS(u axis_artist.py module provides axis-related artists. They are * axis line * tick lines * tick labels * axis label * grid lines The main artist class is a AxisArtist and a GridlinesCollection. The GridlinesCollection is responsible for drawing grid lines and the AxisArtist is responsible for all other artists. The AxisArtist class has attributes that are associated with each type of artists. * line : axis line * major_ticks : major tick lines * major_ticklabels : major tick labels * minor_ticks : minor tick lines * minor_ticklabels : minor tick labels * label : axis label Typically, the AxisArtist associated with a axes will be accessed with the *axis* dictionary of the axes, i.e., the AxisArtist for the bottom axis is ax.axis["bottom"] where *ax* is an instance of axes (mpl_toolkits.axislines.Axes). Thus, ax.axis["bottom"].line is an artist associated with the axis line, and ax.axis["bottom"].major_ticks is an artist associated with the major tick lines. You can change the colors, fonts, line widths, etc. of these artists by calling suitable set method. For example, to change the color of the major ticks of the bottom axis to red, ax.axis["bottom"].major_ticks.set_color("r") However, things like the locations of ticks, and their ticklabels need to be changed from the side of the grid_helper. axis_direction -------------- AxisArtist, AxisLabel, TickLabels have *axis_direction* attribute, which adjusts the location, angle, etc.,. The *axis_direction* must be one of [left, right, bottom, top] and they follow the matplotlib convention for the rectangle axis. For example, for the *bottom* axis (the left and right is relative to the direction of the increasing coordinate), * ticklabels and axislabel are on the right * ticklabels and axislabel have text angle of 0 * ticklabels are baseline, center-aligned * axislabel is top, center-aligned The text angles are actually relative to (90 + angle of the direction to the ticklabel), which gives 0 for bottom axis. Parameter left bottom right top ticklabels location left right right left axislabel location left right right left ticklabels angle 90 0 -90 180 axislabel angle 180 0 0 180 ticklabel va center baseline center baseline axislabel va center top center bottom ticklabel ha right center right center axislabel ha right center right center Ticks are by default direct opposite side of the ticklabels. To make ticks to the same side of the ticklabels, ax.axis["bottom"].major_ticks.set_ticks_out(True) Following attributes can be customized (use set_xxx method) * Ticks : ticksize, tick_out * TickLabels : pad * AxisLabel : pad i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tPath(tAffine2DtBboxtIdentityTransformtScaledTranslationtTransformedPath(tLineCollection(trcParams(tallow_rasterizationi(t AxislineStylet BezierPathcB`s,eZdZdZdZdZRS(cO`s2tjj|gg||||_t|_dS(N(tmlinestLine2Dt__init__t_pathtFalset_invalid(tselftpathtkltkw((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRws cC`s(t|j|j|_t|_dS(N(R Rt get_transformt_transformed_pathRR(R((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytrecache|scC`s||_t|_dS(N(RtTrueR(RR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytset_paths cC`sN|jr|jn|js#dS|jd|j}|j||j|j|j|j |j |j |j |j |jr|j}|j}n|j}|j}|j||j||j|j|j|j|jdkr3|jj\}}|j|||jn|j|jddS(Nuline2du _draw_nothing( RRt_visiblet open_grouptnew_gct _set_gc_cliptset_foregroundt_colortset_antialiasedt _antialiasedt set_linewidtht _linewidtht set_alphat_alphat is_dashedt _dashcapstylet_dashjoinstylet_solidcapstylet_solidjoinstylet set_joinstylet set_capstylet set_dashest _dashOffsett_dashSeqt _lineStylest _linestyleRtget_transformed_path_and_affinet draw_pathtfrozentrestoret close_group(Rtrenderertgctcaptjointtpathtaffine((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytdraws0              (t__name__t __module__RRRRA(((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRus   tUnimplementedExceptioncB`seZRS((RBRC(((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRDs(tArtisttAttributeCopiercB`s/eZedZdZdZdZRS(cC`s)||_||_tt|jdS(N(t_klasst _ref_artisttsuperRFR(Rt ref_artisttklass((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRs  cC`s ||_dS(N(RH(Rtartist((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytset_ref_artistscC`stddS(Nuget_ref_artist must overridden(t RuntimeError(R((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytget_ref_artistscC`sad|}t|j||}|dkr]|j}|rVt||}|S|Sn|S(Nuget_uauto(tgetattrRGRO(Rt attr_namet default_valuetget_attr_method_nametcRJtattr((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytget_attribute_from_ref_artists    (RBRCRERRMRORV(((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRFs   (RtTickscB`seZdZedZdZdZdZdZdZ dZ dZ d Z d Z d Zed d gd d ggZdZRS(uO Ticks are derived from Line2D, and note that ticks themselves are markers. Thus, you should use set_mec, set_mew, etc. To change the tick size (length), you need to use set_ticksize. To change the direction of the ticks (ticks are in opposite direction of ticklabels by default), use set_tick_out(False). cK`s||_g|_|j||jdd|_|jdk rd|kr\d|dsucenterg?urightcs`s|]\}}}|VqdS(N((RRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pys subottomcs`s|]\}}}|VqdS(N((RRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pys subaselinecs`s"|]\}}}||VqdS(N((RRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pys scs`s|]\}}}|VqdS(N((RRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pys sutopcs`s|]\}}}|VqdS(N((RRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pys scs`s"|]\}}}||VqdS(N((RRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pys scs`s|]\}}}|VqdS(N((RRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pys s(ii(t!get_texts_widths_heights_descentstget_vatget_hatmax( RR;Rtwhd_listRRRRt max_ascentt max_descent((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt_get_ticklabels_offsetssP                    RucenterurightRuleftRubaselineRiZiiic C`s|js|j|_dS|j||j\}}|j|j|j}|j||xv|jD]k\\}}}}|j sqtn|j ||j ||j ||j |tj||qtW|||_dS(N(RpRRRRRqRRt_locs_angles_labelststripRtset_xtset_ytset_textRRA( RR;Rt total_widthRtxtyRtl((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRAs$        cC`s ||_dS(N(R(RRY((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytset_locs_angles_labelssc C`s|js|j|_gSg}|j||j\}}|j|j|j}|j||xs|jD]h\\}}}} |j ||j ||j ||j | t j||} |j| qzW|||_|S(N(RpRRRRRqRRRRRRRRRtappend( RR;tbboxesRRRRRRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytget_window_extentss&        c C`s|g}xo|jD]d\\}}}}|js7qn|j|\}}|j||jd|} |j| qW|S(uI return a list of width, height, descent for ticklabels. tismath(RRt is_math_texttget_text_width_height_descentt_fontpropertiesR( RR;RRRRRt clean_lineRtwhd((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRs (ucenteruright(ucenteruleft(ubaselineucenter(ubaselineucenter(RBRCRRRORRRRRRRARRR(((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRRs$     =       tGridlinesCollectioncB`s5eZdZdZdZdZdZRS(cO`sT|jdd|_|jdd|_tt|j|||jddS(uR *which* : "major" or "minor" *axis* : "both", "x" or "y" uwhichumajoruaxisubothN(R[t_whichR]RIRRtset_grid_helperR\(RRRa((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyR'scC`s ||_dS(N(R(Rtwhich((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt set_which1scC`s ||_dS(N(R](Rtaxis((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytset_axis4scC`s ||_dS(N(t _grid_helper(Rt grid_helper((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyR7scC`s|jdk r|jj|j|jj|j|j}|ro|jg|D]}tj |^qPq|jgnt t |j |dS(N( RR\t update_limRvt get_gridlinesRR]t set_segmentsRt transposeRIRRA(RR;tglR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRA:s,(RBRCRRRRRA(((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyR&s    t AxisArtistc B`sdeZdZdZedZejdZd)ddZdZ dZ dZ d Z d Z d Zd)d Zd ZdZdZdZdZdZdZdZedd*dd+dd,dd-ZdZdZd Zd!Zd"Zd#Zd$Z d%Z!d&Z"e#d'Z$d)d)d)d)d(Z%RS(.u An artist which draws axis (a line along which the n-th axes coord is constant) line, ticks, ticklabels, and axis label. g@cC`s |jjS(N(tlabelR(R((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytLABELPADOscC`s|jj|S(N(RR(Rtv((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRSsubottomcK`sQtt|j|||_||_|d kr=d }nt|_t|d|d|j|_ t |_ t |_ t |_ t |_t |_|d krd}|j|_nd}|j|_d |_||_|j|j|||j||j|j|jt|_d|_d|_d|_|j |d S( uP *axes* : axes *helper* : an AxisArtistHelper instance. iiulefturightuytickuxtickgf@gN(ii(ulefturight(!RIRRRvt_axis_artist_helperR\Rt dpi_transformRtoffset_transformRt_label_visiblet_majortick_visiblet_majorticklabel_visiblet_minortick_visiblet_minorticklabel_visibletyaxisRtxaxist_axisline_styleRt _init_linet _init_tickst_init_offsetTextt _init_labelt set_zordertZORDERRt_rotate_label_along_linet_tick_add_anglet_ticklabel_add_anglet_axislabel_add_angleR(RRvthelperR~RRt axis_name((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRWs<                      cC`s|d krtdn||_|d krM|jd|jdn|jd|jd|jj||jj|dS( uV Adjust the direction, text angle, text alignment of ticklabels, labels following the matplotlib convention for the rectangle axes. The *axis_direction* must be one of [left, right, bottom, top]. ===================== ========== ========= ========== ========== property left bottom right top ===================== ========== ========= ========== ========== ticklabels location "-" "+" "+" "-" axislabel location "-" "+" "+" "-" ticklabels angle 90 0 -90 180 ticklabel va center baseline center baseline ticklabel ha right center right center axislabel angle 180 0 0 180 axislabel va center top center bottom axislabel ha right center right center ===================== ========== ========= ========== ========== Note that the direction "+" and "-" are relative to the direction of the increasing coordinate. Also, the text angles are actually relative to (90 + angle of the direction to the ticklabel), which gives 0 for bottom axis. ulefturightutopubottomu8direction must be on of "left", "right", "top", "bottom"u-u+N(ulefturightutopubottom(uleftutop(RRtset_ticklabel_directiontset_axislabel_directiontmajor_ticklabelsRR(RR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRs      cC`s@|dkrtdn|dkr3d|_n d|_dS(u Adjust the direction of the ticklabel. ACCEPTS: [ "+" | "-" ] Note that the label_direction '+' and '-' are relative to the direction of the increasing coordinate. u+u-u!direction must be one of "+", "-"iiN(u+u-(RR(Rttick_direction((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRs   cC`s2|jdd|_|jj|jjdS(Niih(RRRtminor_ticklabels(R((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytinvert_ticklabel_directions cC`s@|dkrtdn|dkr3d|_n d|_dS(u Adjust the direction of the axislabel. ACCEPTS: [ "+" | "-" ] Note that the label_direction '+' and '-' are relative to the direction of the increasing coordinate. u+u-u!direction must be one of "+", "-"iiN(u+u-(RR(RR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRs   cC`s|jj|jS(N(Rvt transAxesR(R((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRscC`s|jS(u5 Return axis artist helper instance. (R(R((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt get_helperscK`sT|dkrtjSt|tjr4||_nt|||_|jdS(u Set the axisline style. *axisline_style* can be a string with axisline style name with optional comma-separated attributes. Alternatively, the attrs can be provided as keywords. set_arrowstyle("->,size=1.5") set_arrowstyle("->", size=1.5) Old attrs simply are forgotten. Without argument (or with arrowstyle=None), return available styles as a list of strings. N(R\R t pprint_stylest isinstancet_BaseR R (Rtaxisline_styleR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytset_axisline_styles    cC`s|jS(u4 return the current axisline style. (R (R((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytget_axisline_style!scC`s|jj|j|j}|j}|dkrot|jj|jdtddtdd||_ n||d||_ dS(uY Initialize the *line* artist that is responsible to draw the axis line. tcoloruaxes.edgecolort linewidthuaxes.linewidtht transformN( Rtget_line_transformRvRR#R\Rtget_lineR tline(RttranR!((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyR 's     cC`sa|jj|jj|j|jdk rM|jj|jj n|jj |dS(N( R)RRR(RvR#R\tset_line_mutation_scaleRtget_sizeRA(RR;((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt _draw_line7sc K`s|jj|j|j}|jdtd|}|jdtd|}|jdtd|}|jdtd|}t|d |jd ||_t|d |jd ||_ |d krtd }n td }t j d|} t d|d |jd|j |_t d|d |jd|j |_|jjd|jjd |d| |jj||jjd|jjd |d| |jj|dS(Numajor_tick_sizeu %s.major.sizeumajor_tick_padu %s.major.paduminor_tick_sizeu %s.minor.sizeuminor_tick_padu %s.minor.padRR&uxaxisuxtick.labelsizeuytick.labelsizeR|Rtfiguretfontproperties(Rtget_tick_transformRvRtgetR RWRt major_tickst minor_tickst font_managertFontPropertiesRRRRtsetR.R( RRRttranstmajor_tick_sizetmajor_tick_padtminor_tick_sizetminor_tick_padR|t fontprops((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyR >s@              c C`sg}g}|j}|j}x|D]\}}}} |d} | |7} tj| |dtjdkrx|} n |d} |j|| g|j|| | gq%W||fS(u return ticks_loc_angle, ticklabels_loc_angle_label ticks_loc_angle : list of locs and angles for ticks ticklabels_loc_angle_label : list of locs, angles and labels for tickslabels iZgf@gi(RRRRRR( Rt tick_itertticks_loc_angletticklabels_loc_angle_labelttick_add_angletticklabel_add_angleRt angle_normalt angle_tangentRt angle_labelt angle_tick((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt_get_tick_infons    $  cC`s|jd}|jjrd|jjrd|jj|jj||jj|jj|n |jjd|jjd|jj |j \}}|j |\}}|jj ||jj ||j |\}}|jj ||jj ||jj|S(Ng?i(RqR2RpRjRRRXRRtget_tick_iteratorsRvRFRnRR3R(RR;tdpi_cortmajortick_itertminortick_iterttick_loc_angletticklabel_loc_angle_label((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt _update_tickss cC`s|j|}|jj||jj||jj||jj||jjsm|jjr}|j|n|S(N(RMR2RARR3RRpt_draw_offsetText(RR;textents((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt _draw_ticksscC`s|jd}|jjrd|jjrd|jj|jj||jj|jj|n |jjd|jjd|jj |j \}}|j |\}}|jj ||jj ||jj||jj||j |\}}|jj ||jj ||jj||jj||jjsg|jjrw|j|n|jj|S(Ng?i(RqR2RpRjRRRXRRRGRvRFRnRRAR3RNR(RR;RHRIRJRKRL((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt _draw_ticks2s(RiiurightRuleftRutopRcC`s|j|\}}}}tjdd||fdddd ddd td d |d ||_|jjt|jj|jdS(Nutxytxycoordsu axes fractiontxytextit textcoordsu offset pointsR$u xtick.colortverticalalignmentthorizontalalignment(ii( t_offsetText_postmtextt AnnotationR t offsetTextRRRvt_set_artist_props(Rt directionRRRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyR s    cC`sn|jj|jjjj|jj|jj|jj |jjd}d|f|j_ dS(Ng@i( R[RRtmajort formattert get_offsettset_sizeRR,Rtxyann(RR~((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt_update_offsetText s cC`s|j|jj|dS(N(RcR[RA(RR;((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRNs cK`s|jdtd}tjd|dtd}td|}|jj|j|j}t dddd d d|d |j d |d |j |_ |j j |jj|jdd}|j j|dS(Nu labelsizeuaxes.labelsizeR|tweightuaxes.labelweightR/iu __from_axes__R$uautoRR&Rulabelpadi(R1R R4R5RRtget_axislabel_transformRvRRRRRt set_figureR.R(RRt labelsizeR<t textpropsRtlabelpad((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyR s"      c C`s;|jjsdStjdtddtd}|j|jkr|jjrd|jj s|j jr|jj r|jj }qd}nt |j j |jj }|jj||jj|j\}}|dkrdS|d}|\}}|jj||j|jjd|d|dS( NR|uaxes.labelsizeRduaxes.labelweightiiZRR(RRpR4R5R RRR2RjR3RXRRRRRRtget_axislabel_pos_angleRvR\RR6( RR;R<t axislabel_padRRRCRDRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt _update_label7s(        cC`s!|j||jj|dS(N(RlRRA(RR;((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt _draw_label_s c C`sK|jjsdStjdtddtd}|j|jkr|jjrd|jj s|j jr|jj r|jj }qd}nt |j j |jj }|jj||jj|j\}}|dkrdS|d}|\}}|jj||j|jjd|d||jj|dS( NR|uaxes.labelsizeRduaxes.labelweightiiZRR(RRpR4R5R RRR2RjR3RXRRRRRRRjRvR\RR6RA( RR;R<RkRRRCRDRR((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt _draw_label2cs*        cC`s|jj|dS(N(RR(Rts((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt set_labelscC`s-|jsdS|jj|j|jd}|jjj||g}|j||j |j j ||j |j j ||j ||j|jj||j|jj|g|D]0}|r|jdks|jdkr|^q}|r%tj|}|SdSdS(Ng?i(RpRRRvRqRRsRrRMtextendRRRRlRRRR[twidththeightRtunionR\(RR;RHRRit_bbox((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyt get_tightbboxs"   =cC`s|jsdS|jt|jj|j|jd}|jjj |||j ||j ||j ||j tdS(u*Draw the axis lines, tick lines and labelsNg?(RpRRBRRRvRqRRsRrRPR-RmR:(RR;RH((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRAs     cC`s|rttt}}}n2|dk r@ttt}}}nd\}}}|dk rd|}n|dk ry|}n|dk r|}n|dk r|jj||jj|n|dk r|jj||jj|n|dk r |jj|ndS(u@ Toggle visibility of ticks, ticklabels, and (axis) label. To turn all off, :: axis.toggle(all=False) To turn all off but ticks on :: axis.toggle(all=False, ticks=True) To turn all on but (axis) label off :: axis.toggle(all=True, label=False)) N(NNN( RR\RR2t set_visibleR3RRR(Rtallttickst ticklabelsRt_tickst _ticklabelst_label((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyttoggles&          N(iiubottomuright(iiubottomuleft(iiutopuright(iiubottomuright(&RBRCRRtpropertyRtsetterR\RRRRRRRR"R#R R-R RFRMRPRQRRXR RcRNR RlRmRnRpRvR RAR~(((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pyRGsH < 6         0  -  /       (  (  )(2Rt __future__RRRRtsixtmatplotlib.artistRLtmartisttmatplotlib.textttextRYtmatplotlib.font_managerR4tmatplotlib.pathRtmatplotlib.transformsRRRRR tmatplotlib.collectionsR t matplotlibR R twarningstnumpyRtmatplotlib.linestlinesRR!R RRt ExceptionRDREtobjectRFRWtTextRRRRR(((sBlib/python2.7/site-packages/mpl_toolkits/axisartist/axis_artist.pytUs2" (  4o!