m[c@`sdZddlmZmZmZmZddlZddlmZm Z m Z m Z ddl Z ddl Z ddlmZddlZddljZddljZddljZddljZddljZddljZ ddl!j"Z#ddlm$Z$m%Z%ddlm&Z&ddlm'Z'm(Z(dd l!m)Z)dd l*m+Z+d d l,m-Z-d d l,m.Z.d dl,m/Z/dZ0de$fdYZ1ddZ2ddl3j4Z5e5j6j7e1dS(uR axes3d.py, original mplot3d version by John Porter Created: 23 Sep 2005 Parts fixed by Reinier Heeres Minor additions by Ben Axelrod Significant updates and revisions by Ben Root Module containing Axes3D, an object which can plot 3D objects on a 2D matplotlib figure. i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tmaptxrangetziptreduce(t defaultdict(tAxestrcParams(t _backports(t Normalizet LightSource(tBbox(t Triangulationi(tart3d(tproj3d(taxis3dcC`s+ttjddgddgg}|S(Nii(Rtnptarray(tbox((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt unit_bbox*s'tAxes3DcB`seZdZdZejZdndZdZ dZ dZ dZ dndndndndZ dZd Zd Zd Zdnd Zdndnd ZdndndZdZdZeeedZdZdZdZdZdZdZdZe ddndZ!dndndZ"dne e e dZ#dZ$dndndZ%dndne edZ&e&Z'dndne ed Z(e(Z)dndne ed!Z*e*Z+d"Z,e-j.j/je,_e,Z/e/jdnk re/jd#7_nd$Z0e-j.j1je0_e0Z1e1jdnk r7e1jd%7_nd&Z2e2Z3d'Z4d(Z5e-j.j5jdnk re-j.j5jd)e5_nd*Z6e-j.j6jdnk re-j.j6jd)e6_ne7j8d+Z9d,Z:ed-Z;d.Z<d/Z=d0Z>ed1Z?dnd2Z@d3ZAd4ZBdndnd5ZCd6ZDd7ZEd8d9d:ZFd;ZGd<ZHd=ZId>ZJd?ZKd@ZLdAZMdBZNdCZOdndndDZPdEZQdFZRdGZSdHZTdIZUe dJZVdKZWddndLZXddMZYdNZZdOZ[dPZ\dndndQZ]dndRZ^e^Z_e.j^Z`dSZaeaZbdTZcdUZddVZedWZfdXZgdYZhdZd[ZiedZd\dnd]Zjd\dnd^Zkd_ZlelZmd`ZndaZoeoZpdbZqdcd\ddZrdcd\dedne dfZsesZtdcd\dgZudndhe diZvdndjdkZwe-j.jwjew_dlZxexZydmZzRS(ou 3D axes object. u3dc `swdidjgtjD]}t|^qd6|dkrXddddg}ng_|jdd_|jdd _|jd d}|jd d}j |jd d t _ t _ t _ t _jjjd_|_|dk rKjj|d_nttj||dt||ttjjd_d_|dk rj|njdk rjjjdfd_ n d_ d_j!j"j#j$dj%j&j'ddg}|d|d\_(_)j*j+dS(Nu^ Build an :class:`Axes3D` instance in :class:`~matplotlib.figure.Figure` *fig* with *rect=[left, bottom, width, height]* in :class:`~matplotlib.figure.Figure` coordinates Optional keyword arguments: ================ ========================================= Keyword Description ================ ========================================= *azim* Azimuthal viewing angle (default -60) *elev* Elevation viewing angle (default 30) *zscale* [%(scale)s] *sharez* Other axes to share z-limits with *proj_type* 'persp' or 'ortho' (default 'persp') ================ ========================================= .. versionadded :: 1.2.1 *sharez* u | uscalegg?uazimiueleviuzscaleusharezu proj_typeuperspiudatalimtframeonuunits finalizec`sjdtS(Ntscalez(t_on_units_changedtTrue((tself(s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytwsi(ii(ii(,tjointmscaletget_scale_namestreprtNonet_cidstpopt initial_azimt initial_elevt set_proj_typeRt xy_viewLimt zz_viewLimt xy_dataLimt zz_dataLimt view_initt_readyt_sharezt_shared_z_axest _adjustabletsuperRt__init__Rt set_axis_offt set_axis_ontMt fmt_zdatat set_zscaletzaxist callbackstconnectt_zcidt mouse_initt set_top_viewtpatcht set_linewidtht transLimitstinvertedt transformt _pseudo_wt _pseudo_htfiguretadd_axes( Rtfigtrecttargstkwargstxtzscaletsharezt pseudo_bbox((Rs:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR26sN7                    cC`st|_t|_dS(N(tFalset _axis3donRtstale(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR3s cC`st|_t|_dS(N(RRPRQ(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR4s cC`s+|jjp*|jjp*|jjS(uN Return *True* if units are set on the *x*, *y*, or *z* axes (txaxist have_unitstyaxisR8(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRSscC`s|jj|S(u For artists in an axes, if the zaxis has units support, convert *z* using zaxis unit type .. versionadded :: 1.2.1 (R8t convert_units(Rtz((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytconvert_zunitsscC`stt|jd|d|d||jdksR|jdksR|jdkrVdS|dk r|jjs|jj|qn|dk r|j d|jj }||jj kr|jj ||dk r|jj|qqndS(uT Look for unit *kwargs* and update the axis instances as necessary txdatatydataRJNuzunits( R1Rt_process_unit_infoRRR"RTR8RSt update_unitsR$tunitst set_units(RRXRYtzdataRJtzunits((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRZs-   cC`sxd|j}d|j}d|j}d|j}tt|j| |ddtt|j| |dddS(Ngffffff?g?tauto(tdistR1Rtset_xlimR"tset_ylim(Rtxdwltxdwtydwltydw((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR=s      cC`stjd|jj|jj||_|j|_tjd|jj|jj||_ |j |_ tj d|j j|j j||_|j|_x*|j|j |jfD]}|jqWdS(u4Init 3D axes; overrides creation of regular X/Y axesuxuyuzN(RtXAxisR(t intervalxR*tw_xaxisRRtYAxist intervalytw_yaxisRTtZAxisR)R+tw_zaxisR8tinit3d(Rtax((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt _init_axiss   cC`s|jgtt|jS(N(R8R1Rt get_children(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRsscC`stt|j|jfS(N(R1Rt_get_axis_listR8(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRtsc C`s|p|j\}}}}}}||||||||g||||||||g||||||||g}} } tt|| | S(N(t get_w_limstlistR( Rtvalstminxtmaxxtminytmaxytminztmaxztxstystzs((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt unit_cubes $&cC`s=|dkr|j}n|j|}tj||}|S(N(R"R5RRt proj_points(RRwR5txyzsttcube((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt tunit_cubes   cC`s|j||}|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|dfg }|S( Niiiiiiii(R(RRwR5ttctedges((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt tunit_edgessc `s|jjt|_|j}|rJ||}|j|n |j|j|_|j_|j_|j _ |j _ xAt t |j dfddtD]\}}||_qWxAt t |jdfddtD]\}}||_qW|jr{|j|j|jf}x|D]}|jqCWx|D]}|jqaWntt|jdS(Ntkeyc`s |jS(N(tdo_3d_projection(tcol(trenderer(s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRstreversec`s |jS(N(R(R>(R(s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRs(R>tdrawROt_frameontget_axes_locatort apply_aspecttget_projR5tvvecteyetget_axis_positiont enumeratetsortedt collectionsRtzordertpatchesRPRRRTR8t draw_paneR1R( RRtlocatortpostiRR>taxesRq((Rs:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRs:              cC`s|j}|j||j}|dd|ddk}|dd|ddk}|dd|ddk}|||fS(Niiii(RuRR5(RRwRtxhightyhightzhigh((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR/s  cC`s*|j|jd|d|d|dS(u Callback for processing changes to axis units. Currently forces updates of data limits and view limits. tscalextscaleyRN(trelimtautoscale_view(RRRR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR7s cK`sdS(N((RtxysRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytupdate_datalim@scC`stt|jo|jS(u Get whether autoscaling is applied for all axes on plot commands .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. (R1Rtget_autoscale_ontget_autoscalez_on(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRCscC`s|jS(u Get whether autoscaling for the z-axis is applied on plot commands .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. (t _autoscaleZon(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRLscC`s'tt|j||j|dS(u  Set whether autoscaling is applied on plot commands .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. Parameters ---------- b : bool .. ACCEPTS: bool N(R1Rtset_autoscale_ontset_autoscalez_on(Rtb((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRUs cC`s ||_dS(u Set whether autoscaling for the z-axis is applied on plot commands .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. Parameters ---------- b : bool .. ACCEPTS: bool N(R(RR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRds cC`s=|dks|dkr'tdn||_t|_dS(uU Set padding of Z data limits prior to autoscaling. *m* times the data interval will be added to each end of that interval before it is used in autoscaling. accepts: float in range 0 to 1 .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. iiumargin must be in range 0 to 1N(t ValueErrort_zmarginRRQ(Rtm((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_zmarginrs  c O`s| r$| r$|j|j|jfS|jdt}|jdd}|jdd}|jdd}|sunt|dkr|d}}}ndt|dkrtjdt j |\}}n0t|d kr|\}}}n t d |dk r|j |n|dk r8|j |n|dk rT|j|n|dk }|dk }|dk } |jd |d |d |d| dS(u Convenience method to set or retrieve autoscaling margins. signatures:: margins() returns xmargin, ymargin, zmargin :: margins(margin) margins(xmargin, ymargin, zmargin) margins(x=xmargin, y=ymargin, z=zmargin) margins(..., tight=False) All forms above set the xmargin, ymargin and zmargin parameters. All keyword parameters are optional. A single argument specifies xmargin, ymargin and zmargin. The *tight* parameter is passed to :meth:`autoscale_view`, which is executed after a margin is changed; the default here is *True*, on the assumption that when margins are specified, no additional padding to match tick marks is usually desired. Setting *tight* to *None* will preserve the previous setting. Specifying any margin changes only the autoscaling; for example, if *xmargin* is not None, then *xmargin* times the X data interval will be added to each end of that interval before it is used in autoscaling. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. utightuxuyuziiiu|Passing exactly two positional arguments to Axes3D.margins is deprecated. If needed, pass them as keyword arguments insteadiu7Axes3D.margins takes at most three positional argumentsttightRRRN(t_xmargint_ymarginRR$RR"tlentwarningstwarntcbooktmplDeprecationRt set_xmargint set_ymarginRR( RRItkwRtmxtmytmzRRR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytmarginss<$        ubothc C`s|d kr!t}t}t}nx|d krCt||_}nt}|d krkt||_}nt}|d krt||_}nt}|jd|d|d|d|d S( u Convenience method for simple axis view autoscaling. See :meth:`matplotlib.axes.Axes.autoscale` for full explanation. Note that this function behaves the same, but for all three axes. Therefore, 'z' can be passed for *axis*, and 'both' applies to all three axes. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. uxubothuyuzRRRRN(uxuboth(uyuboth(uzuboth(R"Rtboolt _autoscaleXonROt _autoscaleYonRR(RtenabletaxisRRRR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt autoscales     cC`sttj|||f\}}}y8|j|j}}|dk r[|j}nWntk rrnX|jjtj||gj | |dk r|j jtj||gj | n|j dS(N( RRtasarraytflattenR"tAttributeErrorR*tupdate_from_data_xyRtTR+R(RtXtYtZthad_dataRKtyRV((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytauto_scale_xyzs$  & )cC`s|js dS|dkrd|jp^t|jdko^t|jdko^t|jdk}nt|}|_|r|jr|j j |}g|D]}|j ^q}t j j|} |jj\} } |jj} y| j| | \} } Wn5tk r:t j| | dtdd\} } nX|jdkrr| | |j} | | 8} | | 7} n|s| j| | \} } n|j| | n|r|jr|jj |}g|D]}|j ^q}t j j|} |jj\}}|jj}y|j||\}}Wn5tk rit j||dtdd\}}nX|jdkr|||j} || 8}|| 7}n|s|j||\}}n|j||n|r|jr|j j |}g|D]}|j ^q}t j j|} |j!j\}}|j"j}y|j||\}}Wn5tk rt j||dtdd\}}nX|j#dkr|||j#} || 8}|| 7}n|s|j||\}}n|j$||ndS(u Autoscale the view limits using the data limits. See :meth:`matplotlib.axes.Axes.autoscale_view` for documentation. Note that this function applies to the 3D axes, and as such adds the *scalez* to the function arguments. .. versionchanged :: 1.1.0 Function signature was changed to better match the 2D version. *tight* is now explicitly a kwarg and placed first. .. versionchanged :: 1.2.1 This is now fully functional. Nit increasingtexpanderg?(%R-R"t_tightRtimagestlinesRRRt_shared_x_axest get_siblingstdataLimt mtransformstBboxBasetunionR*RiRRtget_major_locatort nonsingularRRORt view_limitst set_xboundRt_shared_y_axesRlRTRt set_yboundRR/R+R8Rt set_zbound(RRRRRRtxsharedRqtdltbbtx0tx1txlocatortdeltatysharedty0ty1tylocatortzsharedtz0tz1tzlocator((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRsz           cC`sL|j\}}|j\}}|j\}}||||||fS(uGet 3D world limits.(t get_xlim3dt get_ylim3dt get_zlim3d(RRxRyRzR{R|R}((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRuSscO`sW|dkr*tj|r*|\}}n||krM|d8}|d7}n||fS(Ng?(R"Rtiterable(RtxmintxmaxRIRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt_determine_limsZs    c K`sAd|kr|jd}nd|kr<|jd}n|r[tdt|n|d krtj|r|\}}n|jd||f|j||j}|j||j}|j \}}|d kr|}n|d kr|}n||kr't j d||fnt j ||dt\}}|jj||\}}||f|j_|d k rt||_n|r.|jjd|x|jj|D]k}||k r|j|jjdtd ||j|jkr'|jjd k r'|jjjq'qqWnt|_||fS( un Set 3D x limits. See :meth:`matplotlib.axes.Axes.set_xlim` for full documentation. uxminuxmaxuunrecognized kwargs: %sRXuwAttempting to set identical left==right results in singular transformations; automatically expanding. left=%s, right=%sRu xlim_changedtemitR`N(R$RRvR"RRRZt_validate_converted_limitstconvert_xunitstget_xlimRRRRRORRtlimit_range_for_scaleR(RiRRR9tprocessRRRbREtcanvast draw_idleRRQ( RtlefttrightRR`Rtold_leftt old_righttother((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_xlim3dbsF            c K`sAd|kr|jd}nd|kr<|jd}n|r[tdt|n|d krtj|r|\}}n|jd||f|j||j}|j||j}|j \}}|d kr|}n|d kr|}n||kr't j d||fnt j ||dt\}}|jj||\}}||f|j_|d k rt||_n|r.|jjd|x|jj|D]k}||k r|j|jjdtd ||j|jkr'|jjd k r'|jjjq'qqWnt|_||fS( un Set 3D y limits. See :meth:`matplotlib.axes.Axes.set_ylim` for full documentation. uyminuymaxuunrecognized kwargs: %sRYuwAttempting to set identical bottom==top results in singular transformations; automatically expanding. bottom=%s, top=%sRu ylim_changedRR`N(R$RRvR"RRRZRtconvert_yunitstget_ylimRRRRRORTRR(RlRRR9RRRRcRERRRRQ( RtbottomttopRR`Rt old_bottomtold_topR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_ylim3dsF            c K`sAd|kr|jd}nd|kr<|jd}n|r[tdt|n|d krtj|r|\}}n|jd||f|j||j}|j||j}|j \}}|d kr|}n|d kr|}n||kr't j d||fnt j ||dt\}}|jj||\}}||f|j_|d k rt||_n|r.|jjd|x|jj|D]k}||k r|j|jjdtd ||j|jkr'|jjd k r'|jjjq'qqWnt|_||fS( um Set 3D z limits. See :meth:`matplotlib.axes.Axes.set_ylim` for full documentation uzminuzmaxuunrecognized kwargs: %sR^uwAttempting to set identical bottom==top results in singular transformations; automatically expanding. bottom=%s, top=%sRu zlim_changedRR`N(R$RRvR"RRRZRRWtget_zlimRRRRROR8RR)RiRRR9RR/Rtset_zlimRERRRRQ( RR R RR`RR RR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_zlim3dsF            cC`st|jjS(N(ttupleR(Ri(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRsuz .. versionchanged :: 1.1.0 This function now correctly refers to the 3D x-limits cC`st|jjS(N(RR(Rl(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRsu{ .. versionchanged :: 1.1.0 This function now correctly refers to the 3D y-limits. cC`st|jjS(uGet 3D z limits.(RR)Ri(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRscC`s$ddjtj|jjS(Nu Return the zaxis scale string %s .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. u, (RRR R8t get_scale(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt get_zscalescK`s7|jj|||jdtdt|jdS(NRR(RRt _set_scaleRROt_update_transScale(RtvalueRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_xscale"su .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. cK`s@|jj|||jdtdt|jt|_dS(NRR(RTRRRORRRQ(RRRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_yscale-s cK`s@|jj|||jdtdt|jt|_dS(u Set the scaling of the z-axis: %(scale)s ACCEPTS: [%(scale)s] Different kwargs are accepted, depending on the scale: %(scale_docs)s .. note :: Currently, Axes3D objects only supports linear scales. Other scales may or may not work, and support for these is improving with each release. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. RRN(R8RRRORRRQ(RRRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR79s cO`s|jj||S(u Set z-axis tick locations. See :meth:`matplotlib.axes.Axes.set_yticks` for more details. .. note:: Minor ticks are not supported. .. versionadded:: 1.1.0 (R8t set_ticks(RRIRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_zticksPs cC`s|jjd|S(u Return the z ticks as a list of locations See :meth:`matplotlib.axes.Axes.get_yticks` for more details. .. note:: Minor ticks are not supported. .. versionadded:: 1.1.0 tminor(R8t get_ticklocs(RR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt get_zticks\s cC`stjd|jjS(ud Get the ztick labels as a list of Text instances .. versionadded :: 1.1.0 uText zticklabel(Rt silent_listR8tget_majorticklabels(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytget_zmajorticklabelshs cC`stjd|jjS(u Get the ztick labels as a list of Text instances .. note:: Minor ticks are not supported. This function was added only for completeness. .. versionadded :: 1.1.0 uText zticklabel(RR R8tget_minorticklabels(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytget_zminorticklabelsqs cO`s|jj||S(u Set z-axis tick labels. See :meth:`matplotlib.axes.Axes.set_yticklabels` for more details. .. note:: Minor ticks are not supported by Axes3D objects. .. versionadded:: 1.1.0 (R8tset_ticklabels(RRIRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytset_zticklabels~s cC`stjd|jjd|S(u Get ztick labels as a list of Text instances. See :meth:`matplotlib.axes.Axes.get_yticklabels` for more details. .. note:: Minor ticks are not supported. .. versionadded:: 1.1.0 uText zticklabelR(RR R8tget_ticklabels(RR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytget_zticklabelss cC`s|jj|dS(u Sets up z-axis ticks and labels that treat the z data as dates. *tz* is a timezone string or :class:`tzinfo` instance. Defaults to rc value. .. note:: This function is merely provided for completeness. Axes3D objects do not officially support dates for ticks, and so this may or may not work as expected. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. N(R8t axis_date(Rttz((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt zaxis_datescC`s |jjS(u Get ztick lines as a list of Line2D instances. Note that this function is provided merely for completeness. These lines are re-calculated as the display changes. .. versionadded:: 1.1.0 (R8t get_ticklines(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytget_zticklinesscO`sdS(ua This function is currently not implemented for 3D axes. Returns *None*. N(R"(RRIRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytclabelscC`sUd|_|dkr$|j|_n ||_|dkrH|j|_n ||_dS(u| Set the elevation and azimuth of the axes. This can be used to rotate the axes programmatically. 'elev' stores the elevation angle in the z plane. 'azim' stores the azimuth angle in the x,y plane. if elev or azim are None (default), then the initial value is used which was specified in the :class:`Axes3D` constructor. i N(RaR"R&televR%tazim(RR/R0((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR,s    cC`sJ|dkrtj|_n+|dkr6tj|_ntd|dS(u Set the projection type. Parameters ---------- proj_type : str Type of projection, accepts 'persp' and 'ortho'. uperspuorthouunrecognized projection: %sN(Rtpersp_transformationt _projectiontortho_transformationR(Rt proj_type((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR's  cC`stj|jdtj|jd}}|j\}}|j\}}|j\}}tj||||||} tj dddg} | dtj |tj ||j } | dtj |tj ||j } | dtj ||j } tj | | | f}||_ | ||_|jtj|j|_t|tjdkr~tj d}ntj d}|j |j }}tj|| |}|j||}tj|| }tj||}|S( u Create the projection matrix from the current viewing position. elev stores the elevation angle in the z plane azim stores the azimuth angle in the x,y plane dist is the distance of the eye viewing point from the object point. ig?iiii(iii(iii(RtpiR/R0RRRRtworld_transformationRtcosRatsinRRtmodtabstview_transformationR2tdot(RtrelevtrazimRRtymintymaxtzmintzmaxtworldMtRtxptyptzptEtVtzfronttzbacktviewMtprojMtM0R5((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRs0 )  ++  iicC`sd|_|jj}|dk ru|jd|j}|jd|j}|jd|j}|||g|_n t j dt j |j |_t j |j |_dS(u"Initializes mouse button callbacks to enable 3D rotation of the axes. Also optionally sets the mouse buttons for 3D rotation and zooming. ============ ======================================================= Argument Description ============ ======================================================= *rotate_btn* The integer or list of integers specifying which mouse button or buttons to use for 3D rotation of the axes. Default = 1. *zoom_btn* The integer or list of integers specifying which mouse button or buttons to use to zoom the 3D axes. Default = 3. ============ ======================================================= umotion_notify_eventubutton_press_eventubutton_release_eventucAxes3D.figure.canvas is 'None', mouse rotation disabled. Set canvas then call Axes3D.mouse_init().N(R"tbutton_pressedRERt mpl_connectt_on_movet _button_presst_button_releaseR#RRRt atleast_1dttolistt _rotate_btnt _zoom_btn(Rt rotate_btntzoom_btntcanvtc1tc2tc3((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR<s   cC`stS(u Return *True* if this axes supports the zoom box button functionality. 3D axes objects do not use the zoom box button. (RO(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytcan_zoom6scC`stS(u Return *True* if this axes supports the pan/zoom button functionality. 3D axes objects do not use the pan/zoom button. (RO(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytcan_pan>scC`stt|j|jj|jdk r|jjj|j_|jjj|j_|jj\}}|j ||dt dd|jj |jjj n8|jj dy|j ddWnt k rnXt|_d|_|jtddS(u Clear axes RR`ulineariiu axes3d.gridN(R1RtclaR8R.R"tmajorRRRRORRt TypeErrorRRRtgridR (RRR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR`Fs     cC`s4x$|jD]}|jjj|q Wg|_dS(u(Disable mouse button callbacks. N(R#RERtmpl_disconnect(Rtcid((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytdisable_mouse_rotationbscC`s;|j|kr7|j|_|j|j|_|_ndS(N(tinaxestbuttonRORXRYtsxtsy(Rtevent((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRRks cC`s d|_dS(N(R"RO(RRk((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRSpscC`sNy|j|SWn6ttfk rI|jjj}||}|SXdS(u Return *z* string formatted. This function will use the :attr:`fmt_zdata` attribute if it is callable, else will fall back on the zaxis major formatter N(R6RRbR8tget_major_formattertformat_data_short(RRVtfunctval((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_zdatass  c`s;|jdkrdS|j|jkr9d|j|jfSt|jdfd\}}|\}}}|\}} } tj ||} tj || } | | } | | || | | }t j ||j\}}}|j |}|j |}|j|}d|||fS(u Given the 2D view coordinates attempt to guess a 3D coordinate. Looks for the nearest edge to the point and then assumes that the point is at the same z location as the nearest point on the edge. uu!azimuth=%d deg, elevation=%d deg Rc`s!tj|d|dfS(Nii(Rtline2d_seg_dist(tedge(txdtyd(s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRsux=%s, y=%s, z=%sN(R5R"RORVR0R/tminRRthypotRt inv_transformt format_xdatat format_ydataRp(RRsRttp0tp1RRRRRRtd0td1tdtRVRKRR~RR((RsRts:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt format_coords" $cC`s|js dS|jdkr dS|j|j}}|dkrCdS||j||j}}|j}|j}|||_|_|j|j kr|dkr|dkrdSt j |j ||d|_ t j |j ||d|_ |jt|_|jjjn|j|jkr|j\}} } } } } d|||}| ||}| | |}| | |}|j||| ||j| || ||j| || ||j|jjjndS(uMouse moving button-1 rotates by default. Can be set explicitly in mouse_init(). button-3 zooms by default. Can be set explicitly in mouse_init(). Niii(ROR5R"RXRYRiRjRCRDRVRt norm_angleR/R0RRRQRERRRWRuRRR(RRkRKRtdxtdytwthRxRyRzR{R|R}tdftdz((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRQs<    !!    cK`s1|dk r||j_n|jj|||S(uO Set zlabel. See doc for :meth:`set_ylabel` for description. N(R"R8tlabelpadtset_label_text(RtzlabeltfontdictRRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_zlabels cC`s|jj}|jS(u Get the z-label text string. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. (R8t get_labeltget_text(Rtlabel((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt get_zlabelscC`s|jS(u] Get whether the 3D axes panels are drawn. .. versionadded :: 1.1.0 (R(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt get_frame_onscC`st||_t|_dS(u Set whether the 3D axes panels are drawn. .. versionadded :: 1.1.0 Parameters ---------- b : bool .. ACCEPTS: bool N(RRRRQ(RR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_frame_ons cC`stS(u Get whether axis below is true or not. For axes3d objects, this will always be *True* .. versionadded :: 1.1.0 This function was added for completeness. (R(R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt get_axisbelows cC`st|_t|_dS(uS Set whether axis ticks and gridlines are above or below most artists. For axes3d objects, this will ignore any settings and just use *True* .. versionadded :: 1.1.0 This function was added for completeness. Parameters ---------- b : bool .. ACCEPTS: bool N(Rt _axisbelowRQ(RR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt set_axisbelows cK`s4t|rt}ntj||_t|_dS(ua Set / unset 3D grid. .. note:: Currently, this function does not behave the same as :meth:`matplotlib.axes.Axes.grid`, but it is intended to eventually support that behavior. .. versionchanged :: 1.1.0 This function was changed, but not tested. Please report any bugs. N(RRRt_string_to_boolt _draw_gridRQ(RRRJ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRcs  c K`s|jddj}|jdd}|jdd}|jddj}|dk ry|\}}||dWqttfk rtdqXn|d d krt}n]|dkrt}|d krt} qt} td n!|dkrd}n tdy~|dk r|dkrO|jj j j |n|dkrt|j j j j |n|dkr|j j j j |qn|dk r|dkr|jj j j|n|dkr|j j j j|n|dkr|j j j j|qn|dk r|dkrK|jj j j|n|dkrp|j j j j|n|dkr|j j j j|qnWntk rtdnXdS(u$ Convenience method for manipulating the ScalarFormatter used by default for linear axes in Axed3D objects. See :meth:`matplotlib.axes.Axes.ticklabel_format` for full documentation. Note that this version applies to all three axes of the Axes3D object. Therefore, the *axis* argument will also accept a value of 'z' and the value of 'both' will apply to all three axes. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. ustyleuu scilimitsu useOffsetuaxisubothiu*scilimits must be a sequence of 2 integersiusciuplainucommaucomma style remains to be addedu%s is not a valid style valueuzuyuxu0This method only works with the ScalarFormatter.N(uplainucomma(ubothuz(ubothuy(ubothuz(ubothux(ubothuy(ubothuz(ubothux(ubothuy(ubothuz(R$tlowerR"RRbRROtNotImplementedErrorRRRat formattertset_scientificRTR8tset_powerlimitst set_useOffsetR( RRJtstylet scilimitst useOffsetRRtntsbtcb((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytticklabel_format2s\                       c K`s|d k}|d k}|d k}|rC|jjj|n|rb|jjj|n|r|jjj|n|jd|d|d|d|d S( u Convenience method for controlling tick locators. See :meth:`matplotlib.axes.Axes.locator_params` for full documentation Note that this is for Axes3D objects, therefore, setting *axis* to 'both' will result in the parameters being set for all three axes. Also, *axis* can also take a value of 'z' to apply parameters to the z axis. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. uxubothuyuzRRRRN(uxuboth(uyuboth(uzuboth(RRRt set_paramsRTR8R(RRRRJt_xt_yt_z((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytlocator_paramsqs   cK`stt|j|||dkrt|}|jdd|jdd|jdd|jdd|jj|ndS( u( Convenience method for changing the appearance of ticks and tick labels. See :meth:`matplotlib.axes.Axes.tick_params` for more complete documentation. The only difference is that setting *axis* to 'both' will mean that the settings are applied to all three axes. Also, the *axis* parameter also accepts a value of 'z', which would mean to apply to only the z-axis. Also, because of how Axes3D objects are drawn very differently from regular 2D axes, some of these settings may have ambiguous meaning. For simplicity, the 'z' axis will accept settings as if it was like the 'y' axis. .. note:: While this function is currently implemented, the core part of the Axes3D object may ignore some of these settings. Future releases will fix this. Priority will be given to those who file bugs. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. uzubothutopubottomulabeltopu labelbottomN(uzuboth(R1Rt tick_paramstdictR$R"R8tset_tick_params(RRRJtzkw((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRs  cC`s,|j\}}|j||dddS(u Invert the z-axis. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. R`N(RRR"(RR R ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt invert_zaxisscC`s|j\}}||kS(u Returns True if the z-axis is inverted. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. (R(RR R ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytzaxis_invertedscC`s6|j\}}||kr(||fS||fSdS(u Returns the z-axis numerical bounds where:: lowerBound < upperBound .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. N(R(RR R ((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt get_zbounds   cC`s|dkr*tj|r*|\}}n|j\}}|dkrQ|}n|dkrf|}n|jr||kr|j||ddq|j||ddn;||kr|j||ddn|j||dddS(uE Set the lower and upper numerical bounds of the z-axis. This method will honor axes inversion regardless of parameter order. It will not change the :attr:`_autoscaleZon` attribute. .. versionadded :: 1.1.0 This function was added, but not tested. Please report any bugs. R`N(R"RRRRR(RRtuppert old_lowert old_upper((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRs      cK`s8tt|j||||}tj||||S(u Add text to the plot. kwargs will be passed on to Axes.text, except for the `zdir` keyword, which sets the direction to be used as the z direction. (R1RttextRt text_2d_to_3d(RRKRRVtstzdirRJR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRs!c O`s|j}|r[t|dtj r[|d}|d}d|krmtdqmn|jdd}|jdd}tj|t|}t t |j ||||}x'|D]} t j | d|d|qWt j||||\}}}|j|||||S( uo Plot 2D or 3D data. ========== ================================================ Argument Description ========== ================================================ *xs*, *ys* x, y coordinates of vertices *zs* z value(s), either one for all points or one for each point. *zdir* Which direction to use as z ('x', 'y' or 'z') when plotting a 2D set. ========== ================================================ Other arguments are passed on to :func:`~matplotlib.axes.Axes.plot` iiuzsu+plot() for multiple values for argument 'z'uzdiruzRR(thas_datat isinstancetsixt string_typesRbR$R t broadcast_toRR1RtplotRt line_2d_to_3dt juggle_axesR( RR~RRIRJRRRRtline((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRs    ! !c0 O`sb|j}|jdkr*tdntj|||\}}}|j\}}d|kpld|k} d|kpd|k} | r| rtdn|jdd} |jdd} |jdd } |jdd }td rD| rtt tj || d } tt tj ||d } qnM| stt tj || d } tt tj ||d } nd |kr|jd }nN|jd d}|dkr|j j }ntjtj|}d}|jdd}|jdd}|jdd}|jdd}|jdd}|jd|dk}|jdd}|r|dk r|dk r|j|||}ng}|dkr4|r4ttj |d | tj |d | }tj|df}tj|df}d}ng}xtd|d | D]}xtd|d | D]}g} x!|||fD]}!|!||t||| d f}"|!|d t||| d t|d || f}#|!t|d || |t||| d dfddd}$|!|t|d || d|fddd}%tj|"|#|$|%f}&| j|&qWtt| } tjg}'| dggtd t| D](}(| |(| |(d kr| |(^q})td|)D}*|j|)|dk rU|j|||n|j|*t|)|dkrn|rndtt|)dtdt|)d}+},}-tj|)|+tj|)|,||s('RtndimRRtbroadcast_arraystshapeR$R tinttmaxtceilR"t _get_linestget_next_colorRtmcolorstto_rgbatgett_shade_colors_lightsourcetemptyRRut concatenatetappendRvRRtsumtcrossRtPoly3DCollectiont _shade_colorstset_facecolorstset_edgecolorst set_arraytset_climtset_normtadd_collectionR(0RRRRRIRJRtrowstcolst has_stridet has_counttrstridetcstridetrcounttccounttfcolorstcolortcmaptnormtvmintvmaxt linewidthtshadet lightsourcetpolysttotptstv1tv2twhich_pttcolsettrstcstpstatztoptzlefttzbasetzrightRVtlastpRtps2tavgzsumti1ti2ti3tnormalstpolyc((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt plot_surface)s8   "("%   '!H7O  8((      cC`s|g}xo|D]g}tj|dtj|d}tj|dtj|d}|jtj||q W|S(u Generate normals for polygons by using the first three points. This normal of course might not make sense for polygons with more than three points not lying in a plane. iii(RRRR(RtpolygonsRtvertsRR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt_generate_normalss  $$c C`s<tjg|D]F}tj|rJtj|tj|dddgntj^q }tj|}t||dkr#tt ||t ||}t |||| cO`s.|j}|jdd}|dkr<|jj}ntjtj|}|j dd}|jdd}|jdd}|jdd}|j dd} |jd|dk} |jdd} t j ||\} }}d |krtj |jd } ntj |d } |d }| j }| j|}| j|}| |}tj|d tjf|d tjf|d tjffd d}tj|||}|rI|dddddfjd d }|j||dk s|dk r*|j||n|dk r|j|qn| r|ddd ddf|ddd ddf}|ddd ddf|dddddf}tj||}|j||}n|}|j||j||j| j| j| ||S(up ============= ================================================ Argument Description ============= ================================================ *X*, *Y*, *Z* Data values as 1D arrays *color* Color of the surface patches *cmap* A colormap for the surface patches. *norm* An instance of Normalize to map values to colors *vmin* Minimum value to map *vmax* Maximum value to map *shade* Whether to shade the facecolors ============= ================================================ The (optional) triangulation can be specified in one of two ways; either:: plot_trisurf(triangulation, ...) where triangulation is a :class:`~matplotlib.tri.Triangulation` object, or:: plot_trisurf(X, Y, ...) plot_trisurf(X, Y, triangles, ...) plot_trisurf(X, Y, triangles=triangles, ...) in which case a Triangulation object will be created. See :class:`~matplotlib.tri.Triangulation` for a explanation of these possibilities. The remaining arguments are:: plot_trisurf(..., Z) where *Z* is the array of values to contour, one per point in the triangulation. Other arguments are passed on to :class:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection` **Examples:** .. plot:: gallery/mplot3d/trisurf3d.py .. plot:: gallery/mplot3d/trisurf3d_2.py .. versionadded:: 1.2.0 This plotting function was added for the v1.2.0 release. ucolorucmapunormuvminuvmaxu linewidthushadeu lightsourceuZii.RiNi(RR$R"RRRRRRRRtget_from_args_and_kwargsRtget_masked_trianglesRKRRR RRtmeanRRRRRRRR(RRIRJRRRRRRRRRttriRVt trianglestxttyttztRRtavg_zRRRR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt plot_trisurfsT1        6 +  <<  icC`s|j}|j}|d|dd}x>t||D]-\}}tj|j||}tj|j||} |jd} g} g} tjt |d|} | dkrt |ddkr8d} qq8nt |dd| d}xt t tj| dD]}t tj||}t tj|d|}| j |d||d|| d|| d|gtj |d|tj |d|}tj |d|tj | d|}| j tj||q"W|j| | }|j| | }tj| d|d|}|j||j|q8Wx|D]}|jj|qpWdS(u4 Extend a contour in 3D by creating iiit facecolorst edgecolorsN(tlevelsRRRtpaths_to_3d_segmentst get_pathst get_colorRtroundRtrangeRRRRRRt set_sort_zpostadd_collection3dtremove(RtcsettstrideR1tcollsRRVR#ttopvertstbotvertsRt polyvertsRtnstepststepsizeRRRRRRtcolors2tpolycolR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt_3d_extend_contoursD    &  ,,    uzcC`sxd|}|r#|j||nQxNt|j|jD]7\}}|dk rZ|}ntj||d|q9WdS(Nu-R(RDRR1RR"Rtline_collection_2d_to_3d(RR:textend3dR;RtoffsetRVR#((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytadd_contour_set#s "  cC`sld|}x[t|j|jD]D\}}|dk rA|}ntj||d||j|q WdS(Nu-R(RR1RR"Rtpoly_collection_2d_to_3dR7(RR:RRGRVR#((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytadd_contourf_set-s  "  cO`s|jdt}|jdd}|jdd}|jdd} |j} tj||||\} } } tt|j| | | ||}|j ||||| |j |||| |S(uI Create a 3D contour plot. ========== ================================================ Argument Description ========== ================================================ *X*, *Y*, Data values as numpy.arrays *Z* *extend3d* Whether to extend contour in 3D (default: False) *stride* Stride (step size) for extending contour *zdir* The direction to use: x, y or z (default) *offset* If specified plot a projection of the contour lines on this position in plane normal to zdir ========== ================================================ The positional and other keyword arguments are passed on to :func:`~matplotlib.axes.Axes.contour` Returns a :class:`~matplotlib.axes.Axes.contour` uextend3dustrideiuzdiruzuoffsetN( R$ROR"RRt rotate_axesR1RtcontourRHR(RRRRRIRJRFR;RRGRtjXtjYtjZR:((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRL5s !$cO`sC|jdt}|jdd}|jdd}|jdd }|j}tj||\}}}|j} |j} d|kr|jd} n|d} |d }tj | | | |\} } }t| | |j |j }t t |j||||}|j||||||j| | | ||S( u Create a 3D contour plot. ========== ================================================ Argument Description ========== ================================================ *X*, *Y*, Data values as numpy.arrays *Z* *extend3d* Whether to extend contour in 3D (default: False) *stride* Stride (step size) for extending contour *zdir* The direction to use: x, y or z (default) *offset* If specified plot a projection of the contour lines on this position in plane normal to zdir ========== ================================================ Other keyword arguments are passed on to :func:`~matplotlib.axes.Axes.tricontour` Returns a :class:`~matplotlib.axes.Axes.contour` .. versionchanged:: 1.3.0 Added support for custom triangulations EXPERIMENTAL: This method currently produces incorrect output due to a longstanding bug in 3D PolyCollection rendering. uextend3dustrideiuzdiruzuoffsetuZiiN(R$ROR"RRR%RKRRRKR)R R1Rt tricontourRHR(RRIRJRFR;RRGRR(RRRRMRNROR:((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRP[s&      !!c O`s|jdd}|jdd}|j}tj||||\} } } tt|j| | | ||} |j| |||j ||||| S(u  Create a 3D contourf plot. ========== ================================================ Argument Description ========== ================================================ *X*, *Y*, Data values as numpy.arrays *Z* *zdir* The direction to use: x, y or z (default) *offset* If specified plot a projection of the filled contour on this position in plane normal to zdir ========== ================================================ The positional and keyword arguments are passed on to :func:`~matplotlib.axes.Axes.contourf` Returns a :class:`~matplotlib.axes.Axes.contourf` .. versionchanged :: 1.1.0 The *zdir* and *offset* kwargs were added. uzdiruzuoffsetN( R$R"RRRKR1RtcontourfRJR( RRRRRIRJRRGRRMRNROR:((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRQs !$cO`s|jdd}|jdd}|j}tj||\}}}|j}|j}d|kr{|jd} n|d} |d}tj||| |\} } } t| | |j |j }t t |j || ||} |j| |||j||| || S(u Create a 3D contourf plot. ========== ================================================ Argument Description ========== ================================================ *X*, *Y*, Data values as numpy.arrays *Z* *zdir* The direction to use: x, y or z (default) *offset* If specified plot a projection of the contour lines on this position in plane normal to zdir ========== ================================================ Other keyword arguments are passed on to :func:`~matplotlib.axes.Axes.tricontour` Returns a :class:`~matplotlib.axes.Axes.contour` .. versionchanged :: 1.3.0 Added support for custom triangulations EXPERIMENTAL: This method currently produces incorrect output due to a longstanding bug in 3D PolyCollection rendering. uzdiruzuoffsetuZiiN(R$R"RRR%RKRRRKR)R R1Rt tricontourfRJR(RRIRJRRGRR(RRRRMRNROR:((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRRs"      !!icC`s tj|}t|dkr0t|}nd}t|tjkrttj|d|d||j |n|t|tj krtj |d|d||j |n>t|tj krtj |d|d||j |ntt|j|dS(u) Add a 3D collection object to the plot. 2D collection types are converted to a 3D version by modifying the object and adding z coordinate information. Supported are: - PolyCollection - LineCollection - PatchCollection iRRN(RRTRRuttypetmcolltPolyCollectionRRIR7tLineCollectionREtPatchCollectiontpatch_collection_2d_to_3dR1RR(RRRRtzvalstzsortval((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR8s ic O`s\|j} tjg|||gD]'} tjtjj| tj^q"\}}}tjj|}tj|||||\}}}}}t t |j ||d|d||| } tj | } t j|t|}tj| d|d|d||jdkr9|jdkr9|jdn| sX|j|||| n| S(u1 Create a scatter plot. ============ ======================================================== Argument Description ============ ======================================================== *xs*, *ys* Positions of data points. *zs* Either an array of the same length as *xs* and *ys* or a single value to place all points in the same plane. Default is 0. *zdir* Which direction to use as z ('x', 'y' or 'z') when plotting a 2D set. *s* Size in points^2. It is a scalar or an array of the same length as *x* and *y*. *c* A color. *c* can be a single color format string, or a sequence of color specifications of length *N*, or a sequence of *N* numbers to be mapped to colors using the *cmap* and *norm* specified via kwargs (see below). Note that *c* should not be a single numeric RGB or RGBA sequence because that is indistinguishable from an array of values to be colormapped. *c* can be a 2-D array in which the rows are RGB or RGBA, however, including the case of a single row to specify the same color for all points. *depthshade* Whether or not to shade the scatter markers to give the appearance of depth. Default is *True*. ============ ======================================================== Keyword arguments are passed on to :func:`~matplotlib.axes.Axes.scatter`. Returns a :class:`~mpl_toolkits.mplot3d.art3d.Patch3DCollection` RtcRRt depthshadeg?i(RRRtraveltmatfilledRRtdelete_masked_pointsR1RtscatterRR RRRRXRRRR(RR~RRRRR[R\RIRJRttRtis_2d((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRa s ' I*cO`sO|j}tt|j||||}tj|t|}g} g} xt||D]u\} } tj | } | | j 7} | | gt| 7} tj | | |d|kra| j |dqaqaWt| dkrt t| \}}n gg}}tj||| |\}}} |j||| ||S(u Add 2D bar(s). ========== ================================================ Argument Description ========== ================================================ *left* The x coordinates of the left sides of the bars. *height* The height of the bars. *zs* Z coordinate of bars, if one value is specified they will all be placed at the same z. *zdir* Which direction to use as z ('x', 'y' or 'z') when plotting a 2D set. ========== ================================================ Keyword arguments are passed onto :func:`~matplotlib.axes.Axes.bar`. Returns a :class:`~mpl_toolkits.mplot3d.art3d.Patch3DCollection` ualphai(RR1RtbarR RRRRtget_patch_vertsRUtpatch_2d_to_3dt set_alphaRvRR(RRtheightRRRIRJRRRtverts_zsRRVtvsR~R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRdK s$ !  !uaveragec O`s|j} tjtj||||||\}}}}}}tj|} tj||}tj|}tj||}tj|}tj||}g}xt||||||D]\}}}}}}|j|||f||||f|||||f||||ff||||f|||||f||||||f|||||ff|||f||||f|||||f||||ff||||f|||||f||||||f|||||ff|||f||||f|||||f||||ff||||f|||||f||||||f|||||ffgqWg}|dkr|j j g}nt |t |krxi|D]}|j|gdqWnDt t j|}t |t |krc|dt |9}n| r|j|}|j||}n|}tj|d|d|| | }|j||j| |f||f||f| |S(u'Generate a 3D barplot. This method creates three dimensional barplot where the width, depth, height, and color of the bars can all be uniquely set. Parameters ---------- x, y, z : array-like The coordinates of the anchor point of the bars. dx, dy, dz : scalar or array-like The width, depth, and height of the bars, respectively. color : sequence of valid color specifications, optional The color of the bars can be specified globally or individually. This parameter can be: - A single color value, to color all bars the same color. - An array of colors of length N bars, to color each bar independently. - An array of colors of length 6, to color the faces of the bars similarly. - An array of colors of length 6 * N bars, to color each face independently. When coloring the faces of the boxes specifically, this is the order of the coloring: 1. -Z (bottom of box) 2. +Z (top of box) 3. -Y 4. +Y 5. -X 6. +X zsort : str, optional The z-axis sorting scheme passed onto :func:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection` shade : bool, optional (default = True) When true, this shades the dark sides of the bars (relative to the plot's source of light). Any additional keyword arguments are passed onto :func:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection` Returns ------- collection : Poly3DCollection A collection of three dimensional polygons representing the bars. itzsortt facecolorN(RRRRTRuRRtextendR"t_get_patches_for_fillRRRvRRRRRRRR(RRKRRVRRRRRkRRIRJRRxRyRzR{R|R}RtxityitzitdxitdyitdziR/R[Rt sfacecolorsR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytbar3d| sX7 34'$/'$/'$:     (ucentercK`sTtt|j|d|d||}|jj\}}|jjd||S(NRtlocgq= ףp?(R1Rt set_titlettitlet get_positiontset_y(RRRRwRJtretRKR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyRx s  cO`sudd}|j}|jdd}|jdd}|jdd}|jd t}d } t|| krtd | t|fn|| } g| D]*} t| ttfr| gn| ^q} g| D]$} t| tj j r| j ^q} tj | | } | | } | | } | r|t tj| }g| D]$} tj j| d |j^qO} ng| D]} | j^q} td | Drtjg|| |}|j||Std| Dsttd| Ddksttjd|dd}||}|dkrT||8}nB|dkrq||d8}n%|dkrtdt|ntj| d }tj| d| !jt}tjtj|ddd}|dk}||}|r$||||jd}n ||}t|dkr|tjj ||j!dd}tjg|D]}||^qr}|ddddftjj ||}t|ddf|_"|j!dd}t#|t#|}ng}tj||| |}|j||j$|dddf|dddf|dddf||S(u Plot a 3D field of arrows. call signatures:: quiver(X, Y, Z, U, V, W, **kwargs) Arguments: *X*, *Y*, *Z*: The x, y and z coordinates of the arrow locations (default is tail of arrow; see *pivot* kwarg) *U*, *V*, *W*: The x, y and z components of the arrow vectors The arguments could be array-like or scalars, so long as they they can be broadcast together. The arguments can also be masked arrays. If an element in any of argument is masked, then that corresponding quiver element will not be plotted. Keyword arguments: *length*: [1.0 | float] The length of each quiver, default to 1.0, the unit is the same with the axes *arrow_length_ratio*: [0.3 | float] The ratio of the arrow head with respect to the quiver, default to 0.3 *pivot*: [ 'tail' | 'middle' | 'tip' ] The part of the arrow that is at the grid point; the arrow rotates about this point, hence the name *pivot*. Default is 'tail' *normalize*: bool When True, all of the arrows will be the same length. This defaults to False, where the arrows will be different lengths depending on the values of u,v,w. Any additional keyword arguments are delegated to :class:`~matplotlib.collections.LineCollection` ic S`sktjj|d }|dkrB|d|}|d |}n d\}}tj|}tj|}tj|}tj||dd|||d|||g||d|||dd|| |g| ||||gg}|j} | ddddgddddgf | ddddgddddgf<|j || j |fS(uu To calculate the arrow head. uvw should be a unit vector. We normalize it here: iii(ii( RtlinalgRtmathtradiansR7R8RR R<( tuvwtangleRRKRtraR[RtRpostRneg((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt calc_arrow% s  2- Kulengthiuarrow_length_ratiog333333?upivotutailu normalizeiu-Wrong number of arguments. Expected %d got %dR cs`s!|]}t|dkVqdS(iN(R(Rtv((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pys i scs`s!|]}t|tjVqdS(N(RRtndarray(Rtk((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pys q scS`sh|]}|jqS((R(RR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pys s s itnumiumiddleg@utipuInvalid pivot argument: iRiN(ii(%RR$RORRRRtfloatRR^t MaskedArrayR RRt logical_orRt compressedRtanyRRRtalltAssertionErrortlinspacetstrt column_stacktastypetsqrtRtreshapetmultiplytoutertswapaxesRRvR(RRIRJRRtlengthtarrow_length_ratiotpivott normalizetargit input_argsRtmaskstbcastR R#tshaft_dttarrow_dttXYZtUVWRtshaftstdt head_dirstheadsR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytquiver sp.   41  1 "     "   %(/ Fc,`st|dkrd}n d}|||\}}jdkr]tdntjjdtj}t|d|dkrtj \}}}nfd|D\}}}fd} |j d d} | dkr |j j } n| | d } |j d d} | | d } |j |||tjd d d gd dd gddd gdd d ggdtj} tt} d }x|dD]}|jj|\}}}tj|}tj|}tj|}| j|j}x|D]}x|D]x}|j||d g}t|}|rz| |j||nxt|d |dD]\}}|j|||g}|j|||g}t|}t|} |r|  r| |j||q| r| r| | j||qqW|j|||dg}!|j|||g}"t|!}#|#r.| |#j|"|q.q.Wq!WqWi}$x/| jD]!\}%}&|dkr|&}'ng}'x|&D]}(|(ddd f|(dddf|(dddff})tj|(j}*||)|*ddd f<||)|*dddf<||)|*dddf<|'j|*qWtj|'d| |%d| |%|}+|j|+|+|$|% sc`stj|dkr5tj|jtj|Stj|dkrtj|d jkr~tdj|n|Stdj|dS( Niiiiu8When multidimensional, {} must match the shape of filleduInvalid {} argument(ii(ii(RRR RRRtformat(Rtname(R_(s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt_broadcast_color_arg su facecolorsu edgecolorsics`sPtj|dtj}x1t|D]#}|Vtj|ddd}q%WdS(u) Generator of cyclic permutation matices RiRiN(RRtintpR6troll(RtmatR((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pytpermutation_matrices) siNiR/R0(RRRRRRRRR"tindicesR$RnRRRRvRR<tarangeRRtitemsRRRR8(,RRIRJRtxyzRRKRRVRR/R0tsquaret voxel_facesRtpermutetpctqctrctpindstqindstrindst square_rotRtqRzti0tr1tr2R{tp2RRtpktpk2tikRtcoordt faces_indstfacest face_indstindtfacetpoly((RR_s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR sF              $    $   ?    N({t__name__t __module__t__doc__RRtGrouperR/R"R2R3R4RSRWRZR=RrRsRtRRRRRRORRRRRRRRRRRRRuRRRbRRcRRRtmaxesR RRR RRRRRt docstringtdedent_interpdR7RRR"R$R&R(R+R-R.R,R'RR<R^R_R`RfRRRSRpRRQRRRRRRRcRRRRRRRRttext3Dttext2DRtplot3DRRRRR$R.RDRHRJRLt contour3DRPRQt contourf3DRRR8Rat scatter3DRdRvRxRtquiver3DR(((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyR/s  P          2      H  U 222               /$      " 8     ? &    +    x l 0  $ 7 # 2!?1r g?cC`stjdd|}}tj||\}}tj|d|d ddtj}tj|ddd|ddd ddtjdd}||}|d}|d}|d}|||fS( u6 Return a tuple X, Y, Z with a test data set. gg@iig?g?i i(RRtmeshgridtexpR5(RRKRRRtZ1tZ2R((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt get_test_dataw s+-    (8Rt __future__RRRRRt six.movesRRRRR~RRRtnumpyRtmatplotlib.axesRRtmatplotlib.cbookRtmatplotlib.collectionsRTtmatplotlib.colorsRRtmatplotlib.docstringRtmatplotlib.scaletscaleRtmatplotlib.transformst transformsRR R R R R Rtmatplotlib.tri.triangulationRtRRRRRRtmatplotlib.projectionst projectionstprojtprojection_registrytregister(((s:lib/python2.7/site-packages/mpl_toolkits/mplot3d/axes3d.pyt sL" "    S