m[c@`sdZddlmZmZmZmZddlZddlmZddl Z ddl Z ddl m Z mZmZmZmZmZddlmZddlmZmZmZmZmZddlmZdd l m!Z!d d l"m#Z#d Z$d Z%dZ&dej'fdYZ(dddZ)dej*fdYZ+dddZ,dddZ-dddZ.dddZ/dddZ0defdYZ1dddZ2de!fdYZ3de3fd YZ4d!Z5ddd"Z6ddd#Z7d$efd%YZ8d&efd'YZ9dde:d(Z;d)efd*YZ<ddd+Z=d,Z>d-Z?d.Z@d/ZAdS(0ux Module containing 3D artist code and functions to convert 2D artists into 3D versions which can be added to an Axes3D. i(tabsolute_importtdivisiontprint_functiontunicode_literalsN(tzip(tartisttcbooktcolorstlinesttexttpath(t _backports(t CollectiontLineCollectiontPolyCollectiontPatchCollectiontPathCollection(t Normalize(tPatchi(tproj3dcC`s+|dd}|dkr'|d}n|S(u"Return angle between -180 and +180ihi((ta((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt norm_angles  cC`s+|dd}|dkr'|d}n|S(u Return angle between -90 and +90iiZ((R((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytnorm_text_angle&s  cC`s|dkrtjdS|dkr2tjd S|dkrKtjd S|dkrdtjd Stj|rt|dkr|StddS( Nuxiiuyuziu2'x', 'y', 'z', None or vector of length 3 expected(iii(iii(iii(iii(tnptarraytNoneRtiterabletlent ValueError(tzdir((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytget_dir_vector.s        !tText3DcB`s>eZdZddddddZdddZdZRS(uE Text object with 3D position and (in the future) direction. iuuzcK`s0tjj||||||j||dS(u *x*, *y*, *z* Position of text *text* Text string to display *zdir* Direction of text Keyword arguments are passed onto :func:`~matplotlib.text.Text`. N(tmtexttTextt__init__tset_3d_properties(tselftxtytzR Rtkwargs((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR"BscC`sI|j\}}tj|||f|_t||_t|_dS(N(t get_positionRRt _position3dRt_dir_vectTruetstale(R$R'RR%R&((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR#MscC`stj|j|j|jg|j}|dd|dd}|dd|dd}|dkr}|dkr}d}ntjtj||}|j|dd|ddf|j t |t j j ||t|_dS(Niig(Rtproj_trans_pointsR*R+tMtmathtdegreestatan2t set_positiont set_rotationRR R!tdrawtFalseR-(R$trenderertprojtdxtdytangle((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR5Ss" #(t__name__t __module__t__doc__R"R#R5(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR=s uzcC`st|_|j||dS(u"Convert a Text to a Text3D object.N(Rt __class__R#(tobjR'R((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt text_2d_to_3dcs tLine3DcB`s/eZdZdZdddZdZRS(u 3D line object. cO`s2tjj|gg|||||f|_dS(uU Keyword arguments are passed onto :func:`~matplotlib.lines.Line2D`. N(RtLine2DR"t_verts3d(R$txstystzstargsR(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR"nsiuzcC`sz|j}|j}y)t|}g|D] }|^q.}Wntk rTnXt|||||_t|_dS(N(t get_xdatat get_ydatatfloatt TypeErrort juggle_axesRDR,R-(R$RGRRERFR%((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR#us    cC`sf|j\}}}tj||||j\}}}|j||tjj||t|_ dS(N( RDRtproj_transformR/tset_dataRRCR5R6R-(R$R7txs3dtys3dtzs3dRERFRG((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR5s $(R<R=R>R"R#R5(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRBis cC`st|_|j||dS(u" Convert a 2D line to 3D. N(RBR?R#(tlineRGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt line_2d_to_3ds c C`stj|t|}|jdtdt}gt||D]'\\\}}}}|||f^q@}g|D]$\}}}t||||^qt} | S(uConvert a path to a 3D segment.tsimplifytcurves(R t broadcast_toRt iter_segmentsR6RRM( R RGRtpathsegsR%R&tcodeR'tsegtseg3d((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytpath_to_3d_segments =1cC`sPtj|t|}gt||D]\}}t|||^q(}|S(u@ Convert paths from a collection object to 3D segments. (R RWRRR](tpathsRGRR tpathztsegs((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytpaths_to_3d_segmentss1c C`stj|t|}g}g}|jdtdt}xLt||D];\\\}}}} |j||| f|j|qLWg|D]$\}}} t||| |^q} | |fS(u/Convert a path to a 3D segment with path codes.RURV(R RWRRXR6RtappendRM( R RGRR[tcodesRYR%R&RZR'R\((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytpath_to_3d_segment_with_codess(1c C`stj|t|}g}g}xOt||D]>\}}t|||\}}|j||j|q4W||fS(uP Convert paths from a collection object to 3D segments with path codes. (R RWRRRdRb( R^RGRtsegmentst codes_listR R_R`Rc((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytpaths_to_3d_segments_with_codess tLine3DCollectioncB`s>eZdZdZdZdZdZedZRS(u# A collection of 3D lines. cO`stj||||dS(uc Keyword arguments are passed onto :func:`~matplotlib.collections.LineCollection`. N(R R"(R$ReRHR(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR"scC`s||_t|_dS(u&Set the position to use for z-sorting.N(t _sort_zposR,R-(R$tval((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt set_sort_zposs cC`s&tj||_tj|gdS(u! Set 3D segments N(Rt asanyarrayt _segments3dR t set_segments(R$Re((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRnsc C`sg|jD]}tj||j^q }g|D]$\}}}tj||g^q2}tj||d}x,|D]$\}}}t|t|}qyW|S(uB Project the points according to renderer matrix. geA( RmRR.R/Rt column_stackR Rntmin( R$R7tpointstxyslistRERFRGt segments_2dtminz((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytdo_3d_projections(1cC`s*|r|j|ntj||dS(N(RuR R5(R$R7tproject((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR5s( R<R=R>R"RkRnRuR6R5(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRhs     cC`s2t|j||}t|_|j|dS(u6Convert a LineCollection to a Line3DCollection object.N(Rat get_pathsRhR?Rn(tcolRGRt segments3d((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytline_collection_2d_to_3ds tPatch3DcB`sJeZdZdZdddZdZdZdZdZRS( u 3D patch object. cO`sK|jdg}|jdd}tj||||j||dS(Nuzsuzdiruz(tpopRR"R#(R$RHR(RGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR"siuzcC`sntj|t|}gt||D]'\\}}}t||||^q(|_tj||_dS(N( R RWRRRMt _segment3dRt get_facecolort _facecolor3d(R$tvertsRGRR%R&R'((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR#s=cC`s|jS(N(t_path2d(R$((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytget_path scC`s|jS(N(t _facecolor2d(R$((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR~ sc C`s||j}t|\}}}tj||||j\}}}} tjtj||g|_ |j |_ t |S(N( R}RRtproj_transform_clipR/tmpathtPathRRoRRRRp( R$R7tsRERFRGtvxstvystvzstvis((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRus  '! cC`stj||dS(N(RR5(R$R7((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR5s( R<R=R>R"R#RR~RuR5(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR{s    t PathPatch3DcB`s/eZdZdZdddZdZRS(u 3D PathPatch object. cK`sK|jdg}|jdd}tj|||j|||dS(Nuzsuzdiruz(R|RR"R#(R$R R(RGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR""siuzcC`s/tj||jd|d||j|_dS(NRGR(R{R#tverticesRct_code3d(R$R RGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR#(sc C`s|j}t|\}}}tj||||j\}}}} tjtj||g|j |_ |j |_ t |S(N(R}RRRR/RRRRoRRRRRp( R$R7RRERFRGRRRR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRu,s  '' (R<R=R>R"R#Ru(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRs cC`sC|j}|j}|j|}t|r;|dSgSdS(u2Return a list of vertices for the path of a patch.iN(tget_patch_transformRt to_polygonsR(tpatchttransR tpolygons((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytget_patch_verts6s    cC`s,t|}t|_|j|||dS(u$Convert a Patch to a Patch3D object.N(RR{R?R#(RR'RR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytpatch_2d_to_3dAs  cC`sG|j}|j}|j|}t|_|j|||dS(u,Convert a PathPatch to a PathPatch3D object.N(RRttransform_pathRR?R#(t pathpatchR'RR RR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytpathpatch_2d_to_3dHs    tPatch3DCollectioncB`s2eZdZdZdZdZdZRS(u% A collection of 3D patches. cO`s`|jdd}|jdd}|jdt|_tj||||j||dS(u Create a collection of flat 3D patches with its normal vector pointed in *zdir* direction, and located at *zs* on the *zdir* axis. 'zs' can be a scalar or an array-like of the same length as the number of patches in the collection. Constructor arguments are the same as for :class:`~matplotlib.collections.PatchCollection`. In addition, keywords *zs=0* and *zdir='z'* are available. Also, the keyword argument "depthshade" is available to indicate whether or not to shade the patches in order to give the appearance of depth (default is *True*). This is typically desired in scatter plots. uzsiuzdiruzu depthshadeN(R|R,t _depthshadeRR"R#(R$RHR(RGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR"Ws cC`s||_t|_dS(u&Set the position to use for z-sorting.N(RiR,R-(R$Rj((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRkms cC`s|j|j}t|dkr=t|\}}n g}g}t||tj|||_|j|_ |j |_ t |_ dS(Ni(tupdate_scalarmappablet get_offsetsRRRMRt atleast_1dt _offsets3dR~Rt get_edgecolort _edgecolor3dR,R-(R$RGRtoffsetsRERF((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR#rs  !c C`s|j\}}}tj||||j\}}}}|jrTt|j|n|j} tj| |j } |j | |jrt|j |n|j } tj| |j } |j | t j|tj||g|jdkrt|StjSdS(Ni(RRRR/RtzalphaRtmcolorst to_rgba_arrayt_alphatset_facecolorsRtset_edgecolorsRt set_offsetsRRotsizeRptnan( R$R7RERFRGRRRRtfcstecs((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRus'     (R<R=R>R"RkR#Ru(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRRs    tPath3DCollectioncB`s2eZdZdZdZdZdZRS(u# A collection of 3D paths. cO`s`|jdd}|jdd}|jdt|_tj||||j||dS(u Create a collection of flat 3D paths with its normal vector pointed in *zdir* direction, and located at *zs* on the *zdir* axis. 'zs' can be a scalar or an array-like of the same length as the number of paths in the collection. Constructor arguments are the same as for :class:`~matplotlib.collections.PathCollection`. In addition, keywords *zs=0* and *zdir='z'* are available. Also, the keyword argument "depthshade" is available to indicate whether or not to shade the patches in order to give the appearance of depth (default is *True*). This is typically desired in scatter plots. uzsiuzdiruzu depthshadeN(R|R,RRR"R#(R$RHR(RGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR"s cC`s||_t|_dS(u&Set the position to use for z-sorting.N(RiR,R-(R$Rj((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRks cC`s|j|j}t|dkr=t|\}}n g}g}t||tj|||_|j|_ |j |_ t |_ dS(Ni(RRRRRMRRRR~RRRR,R-(R$RGRRRERF((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR#s  !c C`s|j\}}}tj||||j\}}}}|jrTt|j|n|j} tj| |j } |j | |jrt|j |n|j } tj| |j } |j | t j|tj||g|jdkrt|StjSdS(Ni(RRRR/RRRRRRRRRRRRRoRRpR( R$R7RERFRGRRRRRR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRus'     (R<R=R>R"RkR#Ru(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRs    cC`sSt|trt|_nt|tr6t|_n||_|j||dS(u: Convert a :class:`~matplotlib.collections.PatchCollection` into a :class:`Patch3DCollection` object (or a :class:`~matplotlib.collections.PathCollection` into a :class:`Path3DCollection` object). Keywords: *za* The location or locations to place the patches in the collection along the *zdir* axis. Defaults to 0. *zdir* The axis in which to place the patches. Default is "z". *depthshade* Whether to shade the patches to give a sense of depth. Defaults to *True*. N(t isinstanceRRR?RRRR#(RxRGRt depthshade((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytpatch_collection_2d_to_3ds    tPoly3DCollectioncB`seZdZdZiejd6ejd6ejd6ZdZ dZ e dZ dZ d Zd Zd Zd ZeZd ZeZdZdZeZdZeZdZRS(u& A collection of 3D polygons. cO`sB|jdt}tj|||||j|d|_dS(u  Create a Poly3DCollection. *verts* should contain 3D coordinates. Keyword arguments: zsort, see set_zsort for options. Note that this class does a bit of magic with the _facecolors and _edgecolors properties. uzsortN(R|R,RR"t set_zsortRt_codes3d(R$RRHR(tzsort((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR"s  uaverageuminumaxcC`su|tkrd}n|tk rG||jkr@|j|}qMtSnd}||_d|_||_t|_dS(u Set z-sorting behaviour: boolean: if True use default 'average' string: 'average', 'min' or 'max' uaverageN(R,R6t_zsort_functionsRt_zsortRit _zsortfuncR-(R$Rt zsortfunc((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRs      c C`sd}d}g}g}xD|D]<}|j||t|}|j||f|}qWt|rt|\}}} nggg}}} tjt|} tj||| | g|_||_dS(uOptimize points for projectioniN( textendRRbRRtonesRt_vect_segis( R$RytsiteitsegisRqtpRERFRGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt get_vector's    cC`s-|j|tj|gt||_dS(uSet 3D vertices.N(RRt set_vertsR6t_closed(R$Rtclosed((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR=s cC`s |j|dt||_dS(u Sets 3D vertices with path codesRN(RR6R(R$RRc((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytset_verts_and_codesDscC`sc|jd|_|jttj||_tj||_ tj ||_ t|_ dS(N( RRRiRR,Rtget_facecolorst _facecolors3dtget_edgecolorst _edgecolors3dt get_alphat_alpha3dR-(R$((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR#Ls   cC`s||_t|_dS(u&Set the position to use for z-sorting.N(RiR,R-(R$Rj((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRkWs c`s jd k r(jj_ntjj|j\}}}gj D]0\}}|||!|||!|||!f^qS}j}j } t |t |kr|j t |dd}nt | t |kr"t | dkr|} q"| j t |dd} nj rktfdtt||| Ddddt} n tdg| D]\} } } }}| ^q~}jd k rg| D]"\} } } }}j|^q}tj||ntj|jg| D]\} } } }}| ^q_t j t |kr}g| D]\} } } }}|^qV_n j _jd k rtjdgdgjgdgg}tj||j}|d dS|jdkrtj|StjSd S( u< Perform the 3D projection for this object. taxisic3`sT|]J\}\\}}}}}j|tj||g|||fVqdS(N(RRRo(t.0tidxRERFRGtfctec(R$(s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pys wstkeycS`s|dS(Ni((R%((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytzstreverseuwhoopsiiN( t_ARRt _facecolorsRRtproj_transform_vecRR/RRRtrepeatRtsortedt enumerateRR,RRRRRRt _facecolors2dt _edgecolors2dRiRRRRpR(R$R7ttxsttysttzsRRtxyzlisttcfacetcedget z_segments_2dR'RRRRRsRctzvectztrans((R$s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRu\sH !=      (/+. *  cC`s&tj||tj||_dS(N(Rt set_facecolorR~R(R$R((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRscC`s&tj||tj||_dS(N(Rt set_edgecolorRR(R$R((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRscC`s|dk r=yt|Wq=tk r9tdq=Xntjj||ytj|j|j |_ Wnt tt fk rnXytj|j |j |_Wnt tt fk rnXt|_dS(u Set the alpha tranparencies of the collection. *alpha* must be a float or *None*. ACCEPTS: float or None ualpha must be a float or NoneN(RRKRLRtArtistt set_alphaRRRRRtAttributeErrort IndexErrorRt _edgecolorsR,R-(R$talpha((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRs"  cC`s|jS(N(R(R$((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRscC`s|jS(N(R(R$((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRscC`stj||S(N(R R5(R$R7((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyR5s(R<R=R>R"RtaverageRptmaxRRRR,RRR#RkRuRRRRRRR~RRR5(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRs.          <     cC`sEt|j||\}}t|_|j|||jdS(u6Convert a PolyCollection to a Poly3DCollection object.N(RgRwRR?RR#(RxRGRt segments_3dRc((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytpoly_collection_2d_to_3ds   cC`sf|dkr|||fS|dkr2|||fS|ddkrUt||||S|||fSdS(u Reorder coordinates so that 2D xs, ys can be plotted in the plane orthogonal to zdir. zdir is normally x, y or z. However, if zdir starts with a '-' it is interpreted as a compensation for rotate_axes. uxuyiu-N(t rotate_axes(RERFRGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRMs    cC`su|dkr|||fS|dkr2|||fS|dkrK|||fS|dkrd|||fS|||fSdS(u Reorder coordinates so that the axes are rotated with zdir along the original z axis. Prepending the axis with a '-' does the inverse transform, so zdir can be x, -x, y, -y, z or -z uxu-xuyu-yN((RERFRGR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRs        cC`s=tjt|r!tj|nddddg|dfS(u=Stretch the color argument to provide the required number numii(R RWRRR(tctnum((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyt get_colorss*cC`st|t|}t|rtt|t|}d||d}gt||D]2\}}|d|d|d|d|f^q`}n|S(u6Modify the alphas of the color list according to depthigffffff?iii(RRRRpRR(RRGtnormtsatsRR((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pyRs  K(BR>t __future__RRRRtsixt six.movesRR0tnumpyRt matplotlibRRRRRR R R Rtmatplotlib.cbookR tmatplotlib.collectionsR R RRRtmatplotlib.colorsRtmatplotlib.patchesRtRRRRR!RRARCRBRTR]RaRdRgRhRzR{RRRRRRR,RRRRMRRR(((s9lib/python2.7/site-packages/mpl_toolkits/mplot3d/art3d.pytsL"   .(   &"  -$  DD