3 ƽh] @sdZddlmZmZmZddlZddlZddlZ ddl m Z m Z m ZmZmZddl mZmZddl mZmZmZmZddlmZdd lmZdd lmZmZmZmZm Z dd l m!Z!dd lm"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-ej.e/Z0d dZ1ddZ2ddZ3ddZ4e j5dgdgdgdgdgdgdgdgdgdgd Gd d!d!eZ6Gd"d#d#e6Z7Gd$d%d%Z8e6j9Z:ejZ;e6jej?j@e jAe6d&ejBe6jCdS)'zU The 2D line class which can draw with a variety of line styles, markers and colors. )IntegralNumberRealN)artistcbookcolors docstringrcParams)Artistallow_rasterization)_to_unmasked_float_array ls_mapper ls_mapper_rSTEP_LOOKUP_MAP) MarkerStyle)Path)Affine2DBboxBboxTransformFromBboxTransformToTransformedPath)_path) CARETLEFT CARETRIGHTCARETUP CARETDOWN CARETLEFTBASECARETRIGHTBASE CARETUPBASE CARETDOWNBASETICKLEFT TICKRIGHTTICKUPTICKDOWNcCst|trtj||}|d kr(d}d}n^|dkrHd}ttdj|}n>t|trv|\}}|dkrtjd d d d}nt d t||dk rt |}|r||;}||fS)z"Convert linestyle to dash pattern.solidNonerNdasheddashdotdottedzlines.{}_patternz3.3zPassing the dash offset as None is deprecated since %(since)s and support for it will be removed %(removal)s; pass it as zero instead.)messagezUnrecognized linestyle: %s)r%r&)r'r(r)) isinstancestrrgettupler formatrwarn_deprecated ValueErrorsum)styleoffsetdashesZdsumr66/tmp/pip-build-7iwl8md4/matplotlib/matplotlib/lines.py_get_dash_pattern!s(    r8cs>tds||fS|}|dk r2fdd|Dnd}||fS)Nzlines.scale_dashescs g|]}|dk r|ndqS)Nr6).0x)lwr6r7 Gsz!_scale_dashes..)r )r4r5r;Z scaled_offsetZ scaled_dashesr6)r;r7 _scale_dashesCs r=cCsZt|dkr8tj||d||d|dk\}|S|dd|dd}}|dd||dd|}} |d| d} |||||| | } | dk| dk@} ||d||d|dk} | | dd| ddB@} || ||| | }}||d||d|dk}|| @}| jj\}|jj\}tj||fS)z Return the indices of the segments in the polyline with coordinates (*cx*, *cy*) that are within a distance *radius* of the point (*x*, *y*). rNrr?r?)lennpnonzeroravelZ concatenate)Zcxcyr:yZradiusresZxryrdxdyZLnorm_squ candidatesZ point_hitsZpxpyZ line_hitsZpointslinesr6r6r7 segment_hitsLs  ("  rNcCs|j|j}}dd}t|tr*d|f}nt|tr._slice_or_nonergr>z9`markevery` is a tuple but its len is not 2; markevery={}zq`markevery` is a tuple with len 2 and second element is an int, but the first element is not an int; markevery={}Nz}`markevery` is a tuple with len 2 and second element is a float, but the first element is not a float or an int; markevery={}r)Zaxisz markevery=zG is a tuple with len 2, but its second element is not an int or a floatz. is iterable but not a valid numpy fancy indexz is not a recognized valuer?r?)codesverticesr+rrr.r@r1r/rslice transformrAemptyhypotTZcumsumarangeabsZnewaxisZargminuniqueiterable IndexError) markeverytpathaffineZ ax_transformrPZvertsrOstartstepZ disp_coordsdeltax0Zy0x1y1scaleZ marker_deltaZindserrr6r6r7_mark_every_pathqs^           8"*     rgZaacZdslsr;mecZmewZmfcZmfcaltms) antialiasedcolor drawstyle linestyle linewidthmarkeredgecolormarkeredgewidthmarkerfacecolormarkerfacecoloralt markersizec@s~eZdZdZddddddddZZdd d d d Zd d iZeeZeeZ e j Z e j Z e j ZdZdZdZddZdddZddZddZdd ZeeeZd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zej j!d-d.Z d/d0Z"d1d2Z#dd4d5Z$dd6d7Z%d8d9Z&d:d;Z'dd?Z*d@dAZ+dBdCZ,dDdEZ-dFdGZ.dHdIZ/dJdKZ0dLdMZ1dNdOZ2ddPdQZ3dRdSZ4dTdUZ5dVdWZ6ddYdZZ7dd[d\Z8dd]d^Z9d_d`Z:dadbZ;dcddZdidjZ?dkdlZ@eAjBdmdnZCdodpZDdqdrZEdsdtZFdudvZGdwdxZHdydzZId{d|ZJd}d~ZKddZLddZMddZNddZOddZPddZQddZRddZSddZTddZUdS)Line2Da A line - the line can have both a solid linestyle connecting all the vertices, and a marker at each vertex. Additionally, the drawing of the solid line is influenced by the drawstyle, e.g., one can create "stepped" lines in various styles. Z _draw_solidZ _draw_dashedZ_draw_dash_dotZ _draw_dotted _draw_nothing)-z--z-.:r& Z _draw_linesZ_draw_steps_midZ_draw_steps_preZ_draw_steps_post)defaultz steps-midz steps-prez steps-poststepsr>buttround projectingmiterbevelcCs|jdkrd|jdS|jdkr&dSt|jdkrld|jd|jd|jd|jd|jd |jd fSd d jtd j|j|jSdS)Nr{zLine2D()zLine2D()z#Line2D((%g,%g),(%g,%g),...,(%g,%g))rrz Line2D(%s),z ({:g},{:g})r?r?)Z_label_xr@_yjoinmapr/)selfr6r6r7__str__s  zLine2D.__str__NnonecKstj|tj|stdtj|s.td|dkr>td}|dkrNtd}|dkr^td}| dkrntd} | dkr~td} |dkrtd }|dkrtd }| dkrtd } |dkrtd }|dkrtd }|dkrtd}|dkrtd}|dkrd}d|_d|_d|_d|_ |j ||j ||j ||j |d|_d|_||_d|_d|_d|_d|_|j||j||j|d|_|j|t|| |_d|_d|_d|_|j||j | |j!|d|_"d|_#d|_$d|_%|j&| |j'| |j(| |j)||j*|||_+d|_,tj-g|_.tj-g|_/d|_0d|_1d|_2d|_3d|_4d|_5d|_6d|_7d|_8|j9||dS)a Create a `.Line2D` instance with *x* and *y* data in sequences of *xdata*, *ydata*. Additional keyword arguments are `.Line2D` properties: %(_Line2D_docstr)s See :meth:`set_linestyle` for a description of the line styles, :meth:`set_marker` for a description of the markers, and :meth:`set_drawstyle` for a description of the draw styles. zxdata must be a sequencezydata must be a sequenceNzlines.linewidthzlines.linestylez lines.markerzlines.markerfacecolorzlines.markeredgecolorz lines.colorzlines.markersizezlines.antialiasedzlines.dash_capstylezlines.dash_joinstylezlines.solid_capstylezlines.solid_joinstyler|rTF):r __init__rArZ RuntimeErrorr _dashcapstyle_dashjoinstyle_solidjoinstyle_solidcapstyleset_dash_capstyleset_dash_joinstyleset_solid_capstyleset_solid_joinstyleZ _linestyles _drawstyle _linewidth_dashSeq _dashOffset _us_dashSeq_us_dashOffset set_linewidth set_linestyle set_drawstyle_color set_colorr_marker _markevery _markersize _antialiased set_markeveryset_antialiasedset_markersize_markeredgecolor_markeredgewidth_markerfacecolor_markerfacecoloraltset_markerfacecolorset_markerfacecoloraltset_markeredgecolorset_markeredgewidthupdate pickradius ind_offsetasarray_xorig_yorig _invalidx _invalidyrr_xyr_transformed_path _subslice _x_filledset_data)rxdataydatarprormmarkerrurrrqrsrtZ fillstylerlZ dash_capstyleZsolid_capstyleZdash_joinstyleZsolid_joinstylerrnr\kwargsr6r6r7r s"                      zLine2D.__init__c CsV|j|\}}|dk r||fS|js*|jr2|jt|jdkrHdifS|j}|j\}}|j|}|j }|dddf}|dddf} |j dkrt j d|j } n|j jd|j } tjddj|jd krtj||jd | |jd | d k\} n*t|j|j|| | } |jjd r,| d } WdQRX| |j7} t| dkt| d fS)a  Test whether *mouseevent* occurred on the line. An event is deemed to have occurred "on" the line if it is less than ``self.pickradius`` (default: 5 points) away from it. Use `~.Line2D.get_pickradius` or `~.Line2D.set_pickradius` to get or set the pick radius. Parameters ---------- mouseevent : `matplotlib.backend_bases.MouseEvent` Returns ------- contains : bool Whether any values are within the radius. details : dict A dictionary ``{'ind': pointlist}``, where *pointlist* is a list of points of the line that are within the pickradius around the event position. TODO: sort returned indices by distance NrFrz,no figure set when check if mouse is on linegR@ignore)allr&r>r})ind)r&N)Z_default_containsrrrecacher@r_get_transformed_pathget_transformed_path_and_affineZtransform_pathrQfigure_logwarningrdpirAZerrstate _linestylerBr:rErNr startswithrdict) rZ mouseeventZinsideinfoZtransformed_pathpathr^xyZxtZytZpixelsrr6r6r7containss6       zLine2D.containscCs|jS)zo Return the pick radius used for containment tests. See `.contains` for more details. ) _pickradius)rr6r6r7get_pickradiusszLine2D.get_pickradiuscCs&t|t s|dkrtd||_dS)z Set the pick radius used for containment tests. See `.contains` for more details. Parameters ---------- d : float Pick radius, in points. rz pick radius should be a distanceN)r+rr1r)rdr6r6r7set_pickradiuss zLine2D.set_pickradiuscCs |jjS)z[ Return the marker fill style. See also `~.Line2D.set_fillstyle`. )r get_fillstyle)rr6r6r7rszLine2D.get_fillstylecCs|jj|d|_dS)a% Set the marker fill style. Parameters ---------- fs : {'full', 'left', 'right', 'bottom', 'top', 'none'} Possible values: - 'full': Fill the whole marker with the *markerfacecolor*. - 'left', 'right', 'bottom', 'top': Fill the marker half at the given side with the *markerfacecolor*. The other half of the marker is filled with *markerfacecoloralt*. - 'none': No filling. For examples see :ref:`marker_fill_styles`. TN)r set_fillstylestale)rfsr6r6r7rs zLine2D.set_fillstylecCs||_d|_dS)a Set the markevery property to subsample the plot when using markers. e.g., if ``every=5``, every 5-th marker will be plotted. Parameters ---------- every : None or int or (int, int) or slice or List[int] or float or (float, float) or List[bool] Which markers to plot. - every=None, every point will be plotted. - every=N, every N-th marker will be plotted starting with marker 0. - every=(start, N), every N-th marker, starting at point start, will be plotted. - every=slice(start, end, N), every N-th marker, starting at point start, up to but not including point end, will be plotted. - every=[i, j, m, n], only markers at points i, j, m, and n will be plotted. - every=[True, False, True], positions that are True will be plotted. - every=0.1, (i.e. a float) then markers will be spaced at approximately equal distances along the line; the distance along the line between markers is determined by multiplying the display-coordinate distance of the axes bounding-box diagonal by the value of every. - every=(0.5, 0.1) (i.e. a length-2 tuple of float), the same functionality as every=0.1 is exhibited but the first marker will be 0.5 multiplied by the display-coordinate-diagonal-distance along the line. For examples see :doc:`/gallery/lines_bars_and_markers/markevery_demo`. Notes ----- Setting the markevery property will only show markers at actual data points. When using float arguments to set the markevery property on irregularly spaced data, the markers will likely not appear evenly spaced because the actual data points do not coincide with the theoretical spacing between markers. When using a start offset to specify the first marker, the offset will be from the first data point which may be different from the first the visible data point if the plot is zoomed in. If zooming in on a plot when using float arguments then the actual data points that have markers will change because the distance between markers is always determined from the display-coordinates axes-bounding-box-diagonal regardless of the actual axes data limits. TN)rr)rZeveryr6r6r7rs6zLine2D.set_markeverycCs|jS)zr Return the markevery setting for marker subsampling. See also `~.Line2D.set_markevery`. )r)rr6r6r7 get_markeveryLszLine2D.get_markeverycCs4t|tr*t|t r*tjddd||_||_dS)Nz3.3zSetting the line's pick radius via set_picker is deprecated since %(since)s and will be removed %(removal)s; use set_pickradius instead.)r*)r+rboolrr0rZ_picker)rpr6r6r7 set_pickerTs  zLine2D.set_pickercCs^tddgddgg}|jj}|j||jdd|jrZ|jd|jjd}|j |}|S)NrT)rgR@g?) r get_transformrSZupdate_from_data_xy get_xydatarrrrZpadded)rrendererZbboxZtrans_data_to_xyrkr6r6r7get_window_extent_s  zLine2D.get_window_extentcCsVtjj|||dk rR|jdk r4|jjjd|j|_|jdk rR|jjjd|j|_ dS)NZunits) r axesfsetZxaxis callbacksconnectrecache_alwaysZ_xcidZyaxisZ_ycid)raxr6r6r7rjs     z Line2D.axescGs8t|dkr|\\}}n|\}}|j||j|dS)z| Set the x and y data. Parameters ---------- *args : (2, N) array or two 1D arrays rN)r@ set_xdata set_ydata)rargsr:rEr6r6r7rws    zLine2D.set_datacCs|jdddS)NT)always)r)rr6r6r7rszLine2D.recache_alwaysFc Cs|s |jr$|j|j}t|j}n|j}|s4|jrN|j|j}t|j}n|j }t j t j ||j t|_|jj\|_|_ d|_|jot|dko|j|o|jjdko|jjdko|jdko|jr2d|_t j|}|jr*|jj|_t jt|}t j|||||j||j|<n|j|_|jdk rH|jj }nd}t!|j"|jj} t#t j$| j|d|_d|_%d|_d|_dS)NFiZ rectilinearZlinearTr)_interpolation_steps)&rZconvert_xunitsrr rCrrZconvert_yunitsrrrAZ column_stackZbroadcast_arraysZastypefloatrrVrrr@ _is_sortednameZ get_xscalerZ get_clip_onisnananycopyrrWZinterprrrrrrr) rrZxconvr:ZyconvrEZnanmaskindicesZinterpolation_stepsrr6r6r7rsB          $    zLine2D.recachecCsZ|dk r@t|j|j|ddfj}ttj|j|jjd}n|j}t ||j |_ dS)z Puts a TransformedPath instance at self._transformed_path; all invalidation of the transform is then handled by the TransformedPath instance. N)r) rrrrVrrArrrrrr)rsubslicerrr6r6r7_transform_paths  zLine2D._transform_pathcCs|jdkr|j|jS)zk Return the :class:`~matplotlib.transforms.TransformedPath` instance of this line. N)rr)rr6r6r7rs zLine2D._get_transformed_pathcCs"tj||d|_d|_d|_dS)z Set the Transformation instance used by this artist. Parameters ---------- t : `matplotlib.transforms.Transform` TN)r set_transformrrr)rtr6r6r7rs zLine2D.set_transformcCs tj|S)z.Return whether x is sorted in ascending order.)rZ is_sorted)rr:r6r6r7rszLine2D._is_sortedcCsf|js dS|js|jr |jd|_|jr|jr|jj\}}|jj |d}|jj |d}t t |dd|d}|j |_|j |nd}|jrddlm}||j|}|jd|j|j|jdkr|jj\}} t|jr|j} |j| | j|jtj|j|j } | j!| dd | j"|j#| j$|j%|j&rZ|j'} |j(} n |j)} |j*} | j+| | j,| | j-|j.|j/dk r| j0|j/| j1|j2|j3|j4| || j5| j6|j7rR|j8dkrR|j} |j| | j|j| j$|j9| j"|j#tj|j:|j }tj|j;|j }tj|j;dd |j }td rtr|)rrrF)VZ get_visiblerrrrrrZ get_xboundrZ searchsortedrRmaxr_rZget_path_effectsZmatplotlib.patheffectsrZ open_groupZget_gid _lineStylesrrrr@rQZnew_gcZ _set_gc_clipZset_urlget_urlmcolorsZto_rgbarZ_alphaZset_foregroundrrrr is_dashedrrrrZ set_joinstyleZ set_capstyleZset_snapZget_snapZget_sketch_paramsZset_sketch_params set_dashesrrZ draw_pathfrozenZrestorerrrget_markeredgecolor_get_markerfacecolorr _str_equalr_str_lower_equalget_markerfacecolor get_drawstyleZ _setattr_cmZ!get_transformed_points_and_affinerrg transAxesZget_snap_thresholdr+rZpoints_to_pixelsZ get_joinstyleZ get_capstyleget_pathr get_markerreZ draw_markersZ get_alt_pathZget_alt_transformZ close_groupr)rrrbrcZi0i1rrr]r^gcZlc_rgbacaprZec_rgbaZfc_rgbaZ fcalt_rgbarelengthZ randomnessrr\Z subsampledZsnapZ marker_pathZ marker_transwZalt_marker_pathZalt_marker_transr6r6r7draws                          z Line2D.drawcCs|jS)z-Return whether antialiased rendering is used.)r)rr6r6r7get_antialiasedeszLine2D.get_antialiasedcCs|jS)zP Return the line color. See also `~.Line2D.set_color`. )r)rr6r6r7 get_coloriszLine2D.get_colorcCs|jS)zS Return the drawstyle. See also `~.Line2D.set_drawstyle`. )r)rr6r6r7r qszLine2D.get_drawstylecCs|jS)zS Return the linestyle. See also `~.Line2D.set_linestyle`. )r)rr6r6r7 get_linestyleyszLine2D.get_linestylecCs|jS)z] Return the linewidth in points. See also `~.Line2D.set_linewidth`. )r)rr6r6r7 get_linewidthszLine2D.get_linewidthcCs |jjS)zR Return the line marker. See also `~.Line2D.set_marker`. )rr)rr6r6r7rszLine2D.get_markercCsV|j}tj|drNtdrH|jjdkr.|jS|jjrH|jdkrHdS|jS|SdS) za Return the marker edge color. See also `~.Line2D.set_markeredgecolor`. rz_internal.classic_mode.rrkN)rr) rrrr rrrZ is_filledr)rrjr6r6r7rs zLine2D.get_markeredgecolorcCs|jS)zk Return the marker edge width in points. See also `~.Line2D.set_markeredgewidth`. )r)rr6r6r7get_markeredgewidthszLine2D.get_markeredgewidthcCs:|jdkrdS|r|jn|j}tj|dr2|jS|SdS)Nrr)rrrrr r)rrfcr6r6r7rs   zLine2D._get_markerfacecolorcCs |jddS)za Return the marker face color. See also `~.Line2D.set_markerfacecolor`. F)r)r)rr6r6r7r szLine2D.get_markerfacecolorcCs |jddS)zn Return the alternate marker face color. See also `~.Line2D.set_markerfacecoloralt`. T)r)r)rr6r6r7get_markerfacecoloraltszLine2D.get_markerfacecoloraltcCs|jS)z` Return the marker size in points. See also `~.Line2D.set_markersize`. )r)rr6r6r7get_markersizeszLine2D.get_markersizeTcCs|j|d|j|dfS)zb Return the xdata, ydata. If *orig* is *True*, return the original data. )orig) get_xdata get_ydata)rrr6r6r7get_dataszLine2D.get_datacCs|r |jS|jr|j|jS)z| Return the xdata. If *orig* is *True*, return the original data, else the processed data. )rrrr)rrr6r6r7r s zLine2D.get_xdatacCs|r |jS|jr|j|jS)z| Return the ydata. If *orig* is *True*, return the original data, else the processed data. )rrrr)rrr6r6r7r!s zLine2D.get_ydatacCs|js |jr|j|jS)ze Return the :class:`~matplotlib.path.Path` object associated with this line. )rrrr)rr6r6r7r s zLine2D.get_pathcCs|js |jr|j|jS)z< Return the *xy* data as a Nx2 numpy array. )rrrr)rr6r6r7rs zLine2D.get_xydatacCs|j|krd|_||_dS)zs Set whether to use antialiased rendering. Parameters ---------- b : bool TN)rr)rbr6r6r7rs zLine2D.set_antialiasedcCs||_d|_dS)zi Set the color of the line. Parameters ---------- color : color TN)rr)rrmr6r6r7r szLine2D.set_colorcCs<|dkr d}tj|j|d|j|kr2d|_d|_||_dS)a# Set the drawstyle of the plot. The drawstyle determines how the points are connected. Parameters ---------- drawstyle : {'default', 'steps', 'steps-pre', 'steps-mid', 'steps-post'}, default: 'default' For 'default', the points are connected with straight lines. The steps variants connect the points with step-like lines, i.e. horizontal lines with vertical steps. They differ in the location of the step: - 'steps-pre': The step is at the beginning of the line segment, i.e. the line will be at the y-value of point to the right. - 'steps-mid': The step is halfway between the points. - 'steps-post: The step is at the end of the line segment, i.e. the line will be at the y-value of the point to the left. - 'steps' is equal to 'steps-pre' and is maintained for backward-compatibility. For examples see :doc:`/gallery/lines_bars_and_markers/step_demo`. Nr|)rnT)r_check_in_list drawStylesrrr)rrnr6r6r7rs zLine2D.set_drawstylecCs<t|}|j|krd|_||_t|j|j|j\|_|_dS)z Set the line width in points. Parameters ---------- w : float Line width, in points. TN)rrrr=rrrr)rrr6r6r7r8s  zLine2D.set_linewidthcCsxt|trD|dkrd}tj|jt|d||jkrZinterpdrZkwdocrArr6r6r6r7s\   8 " %W,)K