a Sick@sdZddlZddlmZmZmZddlZddlZddl Z ddl m Z m Z mZmZddlmZmZddl mZmZmZmZddlmZdd lmZdd lmZmZmZdd l m!Z!m"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/e0e1Z2ddZ3ddZ4ddZ5ddZ6ej7e 8dgdgdgdgdgdgdgdgdgdgd Gd!d"d"eZ9Gd#d$d$e9Z:Gd%d&d&Z;e9je9j?Z?ej@ZAdS)'zK 2D lines with support for a variety of line styles, markers, colors, etc. N)IntegralNumberReal)_apicbookcolors _docstring)Artistallow_rasterization)_to_unmasked_float_array ls_mapper ls_mapper_rSTEP_LOOKUP_MAP) MarkerStyle)Path)BboxBboxTransformToTransformedPath) JoinStyleCapStyle)_path) CARETLEFT CARETRIGHTCARETUP CARETDOWN CARETLEFTBASECARETRIGHTBASE CARETUPBASE CARETDOWNBASETICKLEFT TICKRIGHTTICKUPTICKDOWNcCst|trt||}|dvr(d}d}nZ|dvrJd}ttjd|}n8t|trt|\}}|durtd|ntd||durt |}|r||;}||fS)z"Convert linestyle to dash pattern.)solidNonerN)dasheddashdotdottedzlines.{}_patternzUnrecognized linestyle: ) isinstancestrr gettuplemplrcParamsformat ValueErrorsum)styleoffsetdashesdsumr6L/var/www/html/django/DPS/env/lib/python3.9/site-packages/matplotlib/lines.py_get_dash_pattern!s$   r8cs@tjds||fS|}|dur4fdd|Dnd}||fS)Nzlines.scale_dashescs g|]}|dur|ndqSNr6).0xlwr6r7 Cz!_scale_dashes..)r-r.)r3r4r=Z scaled_offsetZ scaled_dashesr6r<r7 _scale_dashes?s r@cCsZt|dkr8t||d||d|dk\}|S|dd|dd}}|dd||dd|}} |d| d} |||||| | } | dk| dk@} ||d||d|dk} | | dd| ddB@} || ||| | }}||d||d|dk}|| @}| \}|\}t||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*). rNr)lennpnonzeroravel concatenate)cxcyr;yradiusresxryrdxdyZLnorm_squ candidatesZ point_hitspxpyZ line_hitspointslinesr6r6r7 segment_hitsHs  ("  rWc Cs|j|j}}dd}t|tr*d|f}nt|tr._slice_or_nonergrAz9`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={}z]markevery is specified relative to the axes size, but the line does not have a Axes as parentr)axisrBz 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 value)codesverticesr)rrr,rCr0r/rslicerDisfiniteall transformemptyhypotTcumsum transAxesarangeabsnewaxisargminuniqueiterable IndexError) markeverytpathaffineaxr[vertsrYstartstepfinZfvertsZ disp_coordsdeltax0y0x1y1scaleZ marker_deltaindserrr6r6r7_mark_every_pathms|           8$*      r}aacdslsr=mecmewmfcZmfcaltms) antialiasedcolor drawstyle linestyle linewidthmarkeredgecolormarkeredgewidthmarkerfacecolormarkerfacecoloralt markersizecseZdZdZddddddddZZdd d d d Zd d iZieeZgeeZ e j Z e j Z e j ZdZddZejddddfdd ZddZddZeddddd ZeeeZd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zdd-d.Z d/d0Z!d1d2Z"dd4d5Z#dd6d7Z$d8d9Z%fd:d;Z&dd?Z)d@dAZ*dBdCZ+dDdEZ,dFdGZ-dHdIZ.dJdKZ/dLdMZ0dNdOZ1dPdQZ2ddRdSZ3dTdUZ4dVdWZ5dXdYZ6dd[d\Z7dd]d^Z8dd_d`Z9dadbZ:dcddZ;dedfZdkdlZ?dmdnZ@dodpZAeBjCdqdrZDdsdtZEdudvZFdwdxZGdydzZHd{d|ZId}d~ZJddZKddZLddZMfddZNeBjCddZOeBjCddZPddZQddZReBjCddZSeBjCddZTddZUddZVddZWZXS)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)----.:r% Z _draw_linesZ_draw_steps_midZ_draw_steps_preZ_draw_steps_post)defaultz steps-midz steps-prez steps-poststepsrAcCs|jdkrd|jdS|jdur&dSt|jdkrld|jd|jd|jd|jd|jd|jdfSd d td j|j|jSdS) NrzLine2D()zLine2D()z#Line2D((%g,%g),(%g,%g),...,(%g,%g))rrBz Line2D(%s),z ({:g},{:g}))_label_xrC_yjoinmapr/selfr6r6r7__str__s  zLine2D.__str__z3.6r)nameNnonec stt|stdt|s.td|dur@tjd}|durRtjd}|durdtjd}|durvtjd}|durtjd}|durtjd }|durtjd }|durtjd }|durtjd }|durtjd }|durd}d|_d|_d|_ d|_ | || || |||d|_d|_||_d|_d|_||||||d|_|||durd}t|tst|| |_n||_d|_||d|_d|_d|_ |!||"||#|d|_$d|_%d|_&d|_'|(| |)| |*| |+| |,|||_-d|_.t|j/t0rbt|j/t1sb|j/|_-t2g|_3t2g|_4d|_5d|_6d|_7d|_8d|_9d|_:d|_;d|_||dS)a Create a `.Line2D` instance with *x* and *y* data in sequences of *xdata*, *ydata*. Additional keyword arguments are `.Line2D` properties: %(Line2D:kwdoc)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.markerz lines.colorzlines.markersizezlines.antialiasedzlines.dash_capstylezlines.dash_joinstylezlines.solid_capstylezlines.solid_joinstyler)rNrrTF)?super__init__rDrk RuntimeErrorr-r. _dashcapstyle_dashjoinstyle_solidjoinstyle_solidcapstyleset_dash_capstyleset_dash_joinstyleset_solid_capstyleset_solid_joinstyle _linestyles _drawstyle _linewidth_unscaled_dash_pattern _dash_pattern set_linewidth set_linestyle set_drawstyle_color set_colorr)r_marker _gapcolor set_gapcolor _markevery _markersize _antialiased set_markeveryset_antialiasedset_markersize_markeredgecolor_markeredgewidth_markerfacecolor_markerfacecoloraltset_markerfacecolorset_markerfacecoloraltset_markeredgecolorset_markeredgewidth_internal_update _pickradius ind_offset_pickerrboolasarray_xorig_yorig _invalidx _invalidyrr_xyr_transformed_path _subslice _x_filledset_data)rxdataydatarrrgapcolormarkerrrrrr fillstylerZ dash_capstylesolid_capstyleZdash_joinstylesolid_joinstyle pickradiusrrmkwargs __class__r6r7rs$                                   zLine2D.__init__c Csl||\}}|dur||fS|js*|jr2|t|jdkrHdifS|}|\}}||}|j }|dddf}|dddf} |j durt d|j } n|j jd|j } tjddt|jd vrt||jd | |jd | d k\} n*t|j|j|| | } |jd r,| d } Wdn1sB0Y| |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 lineR@ignore)r_)r%NrAr)ind)_default_containsrrrecacherCr_get_transformed_pathget_transformed_path_and_affinetransform_pathr\figure_logwarningrdpirDerrstate _linestylerEr;rJrWr startswithrdict) r mouseeventinsideinfoZtransformed_pathpathroxyxtytpixelsrr6r6r7containss:      ( zLine2D.containscCs|jS)zo Return the pick radius used for containment tests. See `.contains` for more details. )rrr6r6r7get_pickradiusszLine2D.get_pickradiusdrcCs$t|tr|dkrtd||_dS)z Set the pick radius used for containment tests. See `.contains` for more details. Parameters ---------- pickradius : float Pick radius, in points. rz pick radius should be a distanceN)r)rr0r)rrr6r6r7set_pickradiuss zLine2D.set_pickradiuscCs |jS)z[ Return the marker fill style. See also `~.Line2D.set_fillstyle`. )r get_fillstylerr6r6r7rszLine2D.get_fillstylecCs |t|j|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) set_markerrr get_markerstale)rfsr6r6r7 set_fillstyleszLine2D.set_fillstylecCs||_d|_dS)af 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 index *start*, will be plotted. - ``every=slice(start, end, N)``: every N-th marker, starting at index *start*, up to but not including index *end*, will be plotted. - ``every=[i, j, m, ...]``: only markers at the given indices will be plotted. - ``every=[True, False, True, ...]``: only positions that are True will be plotted. The list must have the same length as the data points. - ``every=0.1``, (i.e. a float): markers will be spaced at approximately equal visual 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): similar to ``every=0.1`` but the first marker will be offset along the line by 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 *markevery* will still only draw markers at actual data points. While the float argument form aims for uniform visual spacing, it has to coerce from the ideal spacing to the nearest available data point. Depending on the number and distribution of data points, the result may still not look evenly spaced. 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)rZeveryr6r6r7rs8zLine2D.set_markeverycCs|jS)zr Return the markevery setting for marker subsampling. See also `~.Line2D.set_markevery`. )rrr6r6r7 get_markeveryUszLine2D.get_markeverycCs$t|r||_n ||||_dS)z Set the event picker details for the line. Parameters ---------- p : float or callable[[Artist, Event], tuple[bool, dict]] If a float, it is used as the pick radius in points. N)callableZ _containsrr)rpr6r6r7 set_picker]s  zLine2D.set_pickercCs&tddgddgg}|||S)z"Get the bounding box of this line.r)rupdate_from_data_xy get_xydata)rbboxr6r6r7get_bboxlszLine2D.get_bboxcCs^tddgddgg}|j}|j||dd|jrZ|jd|jjd}| |}|S)NrT)rrg?) r get_transformr`rrrrrrpadded)rrendererrZtrans_data_to_xyrr6r6r7get_window_extentrs  zLine2D.get_window_extentcGs8t|dkr|\\}}n|\}}||||dS)z| Set the x and y data. Parameters ---------- *args : (2, N) array or two 1D arrays rN)rC set_xdata set_ydata)rargsr;rJr6r6r7r}s    zLine2D.set_datacCs|jdddS)NT)always)rrr6r6r7recache_alwaysszLine2D.recache_alwaysFc Cs|s |jr$||j}t|}n|j}|s4|jrN||j}t|}n|j }t t || t|_|jj\|_|_ d|_|jrPt|dkrP||rP|jjdkrP|jdkrP|jdurP|rP||jjkrPd|_t |}|rH|j|_t t|}t |||||j||j|<n|j|_|j!durf|j!j"}nd}t#|j$|jj} t%t &| j|d|_!d|_'d|_d|_dS)NFiZ rectilinearlinearTr_interpolation_steps)(rconvert_xunitsrr rFrrconvert_yunitsrrrD column_stackbroadcast_arraysastypefloatrrcraxesrC _is_sortedrZ get_xscaler get_clip_onr transDataisnananycopyrrfinterprrrrrrr) rrZxconvr;ZyconvrJZnanmaskindicesZinterpolation_stepsrr6r6r7rsR    "         zLine2D.recachecCsZ|dur@t|j|j|ddfj}tt|j|jjd}n|j}t || |_ dS)z Put a TransformedPath instance at self._transformed_path; all invalidation of the transform is then handled by the TransformedPath instance. Nr) rrrrcrrDrrrrrr)rsubslicerrr6r6r7_transform_paths zLine2D._transform_pathcCs|jdur||jS)zd rts  zLine2D.set_dash_joinstylecCs"t|}|j|krd|_||_dS)z How to join segments if the line is solid (not `~Line2D.is_dashed`). The default joinstyle is :rc:`lines.solid_joinstyle`. Parameters ---------- s : `.JoinStyle` or %(JoinStyle)s TN)rrrrr6r6r7rNs  zLine2D.set_solid_joinstylecCs|jjS)zl Return the `.JoinStyle` for dashed lines. See also `~.Line2D.set_dash_joinstyle`. )rrrr6r6r7get_dash_joinstyle^szLine2D.get_dash_joinstylecCs|jjS)zl Return the `.JoinStyle` for solid lines. See also `~.Line2D.set_solid_joinstyle`. )rrrr6r6r7get_solid_joinstylefszLine2D.get_solid_joinstylecCs"t|}|j|krd|_||_dS)z How to draw the end caps if the line is `~Line2D.is_dashed`. The default capstyle is :rc:`lines.dash_capstyle`. Parameters ---------- s : `.CapStyle` or %(CapStyle)s TN)rrrrrcsr6r6r7rns  zLine2D.set_dash_capstylecCs"t|}|j|krd|_||_dS)z How to draw the end caps if the line is solid (not `~Line2D.is_dashed`) The default capstyle is :rc:`lines.solid_capstyle`. Parameters ---------- s : `.CapStyle` or %(CapStyle)s TN)rrrrr6r6r7r~s  zLine2D.set_solid_capstylecCs|jjS)zj Return the `.CapStyle` for dashed lines. See also `~.Line2D.set_dash_capstyle`. )rrrr6r6r7get_dash_capstyleszLine2D.get_dash_capstylecCs|jjS)zj Return the `.CapStyle` for solid lines. See also `~.Line2D.set_solid_capstyle`. )rrrr6r6r7get_solid_capstyleszLine2D.get_solid_capstylecCs |jdvS)z Return whether line has a dashed linestyle. A custom linestyle is assumed to be dashed, we do not inspect the ``onoffseq`` directly. See also `~.Line2D.set_linestyle`. )rrrrlrr6r6r7rEs zLine2D.is_dashed)NNNNNNNNNrNNNNNNrNN)N)F)N)F)T)T)T)Y__name__ __module__ __qualname____doc__ lineStylesr@Z _drawStyles_lZ _drawStyles_srZ drawStyleKeysrmarkersZfilled_markers fillstyles fillStyleszorderrrmake_keyword_onlyrrrrename_parameterrpropertyrrr rr r rrrrrr.rr/r%r rhrirjrXrkrmrnrrSrqrTrWrtrurzrwrxr]rrrrrrrr interpdrrrrrrrrrrPrrrrrrrrrrE __classcell__r6r6rr7rs    F   ;  +        #-           rcs4eZdZdZfddZddZfddZZS)_AxLinezo A helper class that implements `~.Axes.axline`, by recomputing the artist transform at draw time. c s\tjddgddgfi||dur.|dus>|durF|durFtd||_||_||_dS)Nrrz.Exactly one of 'xy2' and 'slope' must be given)rr TypeError_slope_xy1_xy2)rxy1xy2sloperrr6r7rsz_AxLine.__init__cCsx|j}|j|j|j}|jdur||j|jg\\}}\}}||}||}t||rt||rt d||fd||fdtj } q||} n||j\}}|j } |j|j \\} } \} } t | dr| |f}| |f}nvt| r|| f}|| f}nXt| || || f| || || f|| || | f|| || | fg\}}}}tt||g|j|jS)Nz3Cannot draw a line through two identical points (x=z, y=rr)r$ _transformr'Z transScalerr`rrDallcloser0infrviewLimiscloseisinfsortedrrZ transLimitsre)rrpZpoints_transformrxryx2y2rOrPrZvxloZvyloZvxhiZvyhirrstoprr6r6r7rsL        z_AxLine.get_transformcsd|_t|dSr9)rrrh)rrrr6r7rhsz _AxLine.draw)rrrrrrrhrr6r6rr7rs )rc@s4eZdZdZddZeddZddZdd Zd S) VertexSelectora Manage the callbacks to maintain a list of selected vertices for `.Line2D`. Derived classes should override the `process_selected` method to do something with the picks. Here is an example which highlights the selected verts with red circles:: import numpy as np import matplotlib.pyplot as plt import matplotlib.lines as lines class HighlightSelected(lines.VertexSelector): def __init__(self, line, fmt='ro', **kwargs): super().__init__(line) self.markers, = self.axes.plot([], [], fmt, **kwargs) def process_selected(self, ind, xs, ys): self.markers.set_data(xs, ys) self.canvas.draw() fig, ax = plt.subplots() x, y = np.random.rand(2, 30) line, = ax.plot(x, y, 'bs-', picker=5) selector = HighlightSelected(line) plt.show() cCsT|jdurtd|dur&td|j|_||_|jjd|j|_t |_ dS)z Parameters ---------- line : `.Line2D` The line must already have been added to an `~.axes.Axes` and must have its picker property set. Nz'You must first add the line to the Axesz2You must first set the picker property of the line pick_event) r$r get_pickerlinecanvas callbacks_connect_picklableonpickcidsetr)rrr6r6r7rs  zVertexSelector.__init__cCs |jjjSr9)r$rrrr6r6r7r?zVertexSelector.cCsdS)a Default "do nothing" implementation of the `process_selected` method. Parameters ---------- ind : list of int The indices of the selected vertices. xs, ys : array-like The coordinates of the selected vertices. Nr6)rrxsysr6r6r7process_selecteds zVertexSelector.process_selectedcCsV|j|jurdS|jt|jN_t|j}|j\}}||||||dS)zs`   8  %_ Z@I