B ·ôJdÊã@s¦dZddlmZmZmZddlZddlZddlZ ddl m Z m Z m Z mZmZmZddl mZmZddl 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/e 0e1¡Z2dd„Z3dd„Z4dd„Z5dd„Z6ej7e  8dgdgdgdgdgdgdgdgdgdgd œ ¡Gd!d"„d"eƒƒƒZ9Gd#d$„d$e9ƒZ:Gd%d&„d&ƒZ;e9je9j?Z?ej@ZAdS)'zK 2D lines with support for a variety of line styles, markers, colors, etc. é)ÚIntegralÚNumberÚRealNé)Ú_apiÚartistÚcbookÚcolorsÚ docstringÚrcParams)ÚArtistÚallow_rasterization)Ú_to_unmasked_float_arrayÚ ls_mapperÚ ls_mapper_rÚSTEP_LOOKUP_MAP)Ú MarkerStyle)ÚPath)ÚBboxÚBboxTransformToÚTransformedPath)Ú JoinStyleÚCapStyle)Ú_path) Ú CARETLEFTÚ CARETRIGHTÚCARETUPÚ CARETDOWNÚ CARETLEFTBASEÚCARETRIGHTBASEÚ CARETUPBASEÚ CARETDOWNBASEÚTICKLEFTÚ TICKRIGHTÚTICKUPÚTICKDOWNcCs¤t|tƒrt ||¡}|dkr(d}d}nX|dkrHd}ttd |¡ƒ}n8t|tƒrr|\}}|dkr€td|›ƒ‚ntd|›ƒ‚|dk rœt|ƒ}|rœ||;}||fS)z"Convert linestyle to dash pattern.)ZsolidÚNonerN)ZdashedZdashdotZdottedzlines.{}_patternzUnrecognized linestyle: ) Ú isinstanceÚstrrÚgetÚtupler ÚformatÚ ValueErrorÚsum)ÚstyleÚoffsetÚdashesZdsum©r1úB/var/www/html/venv/lib/python3.7/site-packages/matplotlib/lines.pyÚ_get_dash_patterns$   r3cs>tds||fS|ˆ}|dk r2‡fdd„|Dƒnd}||fS)Nzlines.scale_dashescs g|]}|dk r|ˆnd‘qS)Nr1)Ú.0Úx)Úlwr1r2ú Asz!_scale_dashes..)r )r/r0r6Z scaled_offsetZ scaled_dashesr1)r6r2Ú _scale_dashes=s r8cCsZt|ƒdkr8t ||d||d|dk¡\}|S|dd…|dd…}}|dd…||dd…|}} |d| d} |||||| | } | dk| dk@} ||d||d|dk} | | dd…| dd…B@} || ||| | }}||d||d|dk}|| @}|  ¡ ¡\}| ¡ ¡\}t ||f¡S)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*). réNéÿÿÿÿr)ÚlenÚnpÚnonzeroÚravelZ concatenate)ZcxÚcyr5ÚyZradiusÚresZxrÚyrÚdxÚdyZLnorm_sqÚuÚ candidatesZ point_hitsZpxÚpyZ line_hitsZpointsÚlinesr1r1r2Ú segment_hitsFs  ("  rIc Cs¤|j|j}}dd„}t|tƒr*d|f}nt|tƒr._slice_or_nonergr9z9`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 parentrr:)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 value)ÚcodesÚverticesr'rrr*r;r,r+rÚsliceÚ transformr<ÚemptyÚhypotÚTZcumsumÚ transAxesÚarangeÚabsZnewaxisZargminÚuniqueÚiterableÚ IndexError)Ú markeveryÚtpathÚaffineÚaxrKZvertsrJÚstartÚstepZ disp_coordsÚdeltaÚx0Zy0Úx1Úy1ÚscaleZ marker_deltaZindsÚerrr1r1r2Ú_mark_every_pathksd           8$*     rdZaaÚcZdsÚlsr6ÚmecZmewZmfcZmfcaltÚms) Ú antialiasedÚcolorÚ drawstyleÚ linestyleÚ linewidthÚmarkeredgecolorÚmarkeredgewidthÚmarkerfacecolorÚmarkerfacecoloraltÚ markersizecsÆeZdZdZddddddddœZZdd d d d œZd d iZee–Zee•Z e j Z e j Z e j ZdZe d¡ejdd„ƒƒZe d¡ejdd„ƒƒZdd„Zd“‡fdd„ Zdd„Zdd„Zdd „ZeeeƒZd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Z d+d,„Z!d-d.„Z"d/d0„Z#d”d2d3„Z$d•d4d5„Z%d6d7„Z&‡fd8d9„Z'd:d;„Z(e)dd?„Z+d@dA„Z,dBdC„Z-dDdE„Z.dFdG„Z/dHdI„Z0dJdK„Z1dLdM„Z2d–dNdO„Z3dPdQ„Z4dRdS„Z5dTdU„Z6d—dWdX„Z7d˜dYdZ„Z8d™d[d\„Z9d]d^„Z:d_d`„Z;dadb„Zdgdh„Z?didj„Z@eAjBdkdl„ƒZCdmdn„ZDdodp„ZEdqdr„ZFdsdt„ZGdudv„ZHdwdx„ZIdydz„ZJd{d|„ZKd}d~„ZL‡fdd€„ZMeAjBdd‚„ƒZNeAjBdƒd„„ƒZOd…d†„ZPd‡dˆ„ZQeAjBd‰dŠ„ƒZReAjBd‹dŒ„ƒZSddŽ„ZTdd„ZUd‘d’„ZV‡ZWS)šÚ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-postÚstepsr9z3.4cCstdd„tDƒƒS)Ncss|] }|jVqdS)N)Úvalue)r4Úcsr1r1r2ú sz"Line2D.validCap..)r*r)Úclsr1r1r2ÚvalidCapýszLine2D.validCapcCstdd„tDƒƒS)Ncss|] }|jVqdS)N)r{)r4Újsr1r1r2r}sz#Line2D.validJoin..)r*r)r~r1r1r2Ú validJoinszLine2D.validJoincCsŒ|jdkrd|j›dS|jdkr&dSt|jƒdkrld|jd|jd|jd|jd|jd|jdfSd d  td j|j|jƒ¡SdS) NrxzLine2D(ú)zLine2D()éz#Line2D((%g,%g),(%g,%g),...,(%g,%g))rr:z Line2D(%s)ú,z ({:g},{:g}))Z_labelÚ_xr;Ú_yÚjoinÚmapr+)Úselfr1r1r2Ú__str__s  zLine2D.__str__NÚnoneéc s†tƒ ¡t |¡stdƒ‚t |¡s.tdƒ‚|dkr>td}|dkrNtd}|dkr^td}|dkrntd}|dkr~td}| dkrŽtd } |dkržtd }|dkr®td }|dkr¾td }|dkrÎtd }|dkrÚd}d|_d|_d|_d|_ |  |¡|  |¡|  |¡|  |¡d|_d|_||_d|_d|_d|_d|_| |¡| |¡| |¡d|_| |¡t|| ƒ|_d|_d|_d|_| |¡|  | ¡| !|¡d|_"d|_#d|_$d|_%| &| ¡| '| ¡| (| ¡| )|¡| *|¡||_+d|_,t-|j.t/ƒr(t-|j.t0ƒs(|j.|_+t 1g¡|_2t 1g¡|_3d|_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_joinstyleryrTF)>ÚsuperÚ__init__r<rVÚ RuntimeErrorr Ú _dashcapstyleÚ_dashjoinstyleÚ_solidjoinstyleÚ_solidcapstyleÚset_dash_capstyleÚset_dash_joinstyleÚset_solid_capstyleÚset_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_markeveryÚset_antialiasedÚset_markersizeÚ_markeredgecolorÚ_markeredgewidthÚ_markerfacecolorÚ_markerfacecoloraltÚset_markerfacecolorÚset_markerfacecoloraltÚset_markeredgecolorÚset_markeredgewidthÚupdateÚ pickradiusÚ ind_offsetr'Ú_pickerrÚboolÚasarrayÚ_xorigÚ_yorigÚ _invalidxÚ _invalidyr…r†Ú_xyrÚ_transformed_pathÚ _subsliceÚ _x_filledÚset_data)r‰ÚxdataÚydatarmrlrjÚmarkerrrrornrprqZ fillstyleriZ dash_capstyleZsolid_capstyleZdash_joinstyleZsolid_joinstyler³rkrXÚkwargs)Ú __class__r1r2rŽs "                      zLine2D.__init__c CsV| |¡\}}|dk r||fS|js*|jr2| ¡t|jƒdkrHdifS| ¡}| ¡\}}| |¡}|j }|dd…df}|dd…df} |j dkr¨t   d¡|j } n|j jd|j } tjddj|jd krt ||jd | |jd | d k¡\} n*t|j|j|| | ƒ} |j d ¡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)Úall)r&Nr9rz)Úind)Z_default_containsr»rºÚrecacher;r¼Ú_get_transformed_pathÚget_transformed_path_and_affineZtransform_pathrLÚfigureÚ_logÚwarningr³Údpir<ZerrstateÚ _linestyler=r5r@rIr˜Ú startswithr´Údict) r‰Z mouseeventZinsideÚinfoZtransformed_pathÚpathrZÚxyZxtZytZpixelsrÈr1r1r2Úcontainsžs6       zLine2D.containscCs|jS)zo Return the pick radius used for containment tests. See `.contains` for more details. )Ú _pickradius)r‰r1r1r2Úget_pickradiusäszLine2D.get_pickradiuscCs$t|tƒr|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'rr,r×)r‰Údr1r1r2Úset_pickradiusìs zLine2D.set_pickradiuscCs |j ¡S)z[ Return the marker fill style. See also `~.Line2D.set_fillstyle`. )r£Ú get_fillstyle)r‰r1r1r2rÛýszLine2D.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_markerÚstale)r‰Úfsr1r1r2Ú 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)r¤rÞ)r‰Zeveryr1r1r2r§s8zLine2D.set_markeverycCs|jS)zr Return the markevery setting for marker subsampling. See also `~.Line2D.set_markevery`. )r¤)r‰r1r1r2Ú get_markeveryTszLine2D.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 _containsr³rµ)r‰Úpr1r1r2Ú set_picker\s zLine2D.set_pickercCs^tddgddggƒ}| ¡j}|j|| ¡ƒdd|jrZ|jd|jjd}|  |¡}|S)NrT)rÆgR@gà?) rÚ get_transformrNZupdate_from_data_xyÚ get_xydatar£r¥rÌrÏZpadded)r‰ÚrendererZbboxZtrans_data_to_xyrhr1r1r2Úget_window_extentks  zLine2D.get_window_extentcGs8t|ƒdkr|\\}}n|\}}| |¡| |¡dS)z| Set the x and y data. Parameters ---------- *args : (2, N) array or two 1D arrays rN)r;Ú set_xdataÚ set_ydata)r‰Úargsr5r@r1r1r2rÀvs    zLine2D.set_datacCs|jdddS)NT)Úalways)rÉ)r‰r1r1r2Úrecache_always†szLine2D.recache_alwaysFc Cs¨|s |jr$| |j¡}t|ƒ ¡}n|j}|s4|jrN| |j¡}t|ƒ ¡}n|j }t   t   ||¡¡  t¡|_|jj\|_|_ d|_|jrPt|ƒdkrP| |¡rP|jjdkrP|j ¡dkrP|jdkrP| ¡rP| ¡|jjkrPd|_t  |¡}| ¡rH|j ¡|_t  t|ƒ¡}t   |||||j|¡|j|<n|j|_|j!dk rf|j!j"}nd}t#|j$|jjŽ} t%t  &| ¡j|d|_!d|_'d|_d|_dS)NFièZ rectilinearZlinearTr)Ú_interpolation_steps)(rºZconvert_xunitsr¸rr>r…r»Zconvert_yunitsr¹r†r<Z column_stackZbroadcast_arraysZastypeÚfloatr¼rQr¾Úaxesr;Ú _is_sortedÚnameZ get_xscaler¤Z get_clip_onråÚ transDataÚisnanÚanyÚcopyr¿rSZinterprrîrr˜rr·r½) r‰rìZxconvr5Zyconvr@ZnanmaskÚindicesZinterpolation_stepsrÕr1r1r2rɉsD    "     $    zLine2D.recachecCsZ|dk r@t|j|j|dd…fjŽ}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. N)rî) rr˜r¼rQrr<r·rrîrrår½)r‰ÚsubslicerÕrr1r1r2Ú_transform_path´s  zLine2D._transform_pathcCs|jdkr| ¡|jS)zd ¡rŒt< ?| @¡d ¡sŒ|dd …|d f}| j!|dd | /¡dk rÐ| /¡\}}}|  0|d|dd|¡|j7}| A¡dkr$trrÔrZ transformsrrrZ_enumsrrrrrrrrrr r!r"r#r$r%Ú getLoggerr9rÍr3r8rIrdrCZ_define_aliasesrsrErQrr=Z lineMarkersr&r?r@r1r1r1r2Ús\    8  %[ -@H