e[ddlmZmZddlZddlmZddlmZddl m Z m Z ddl m Z dd l m Z dd lmZdd lmZd d gZGdd e ZGdd eZy))atan2degreesN) functions)Point)QtCoreQtGui) GraphicsItem)GraphicsObject)TextItem)ViewBox InfiniteLine InfLineLabelcPeZdZdZej eZej eZej eZ ej eeZ ddZ dZ dZ dZdZdZdd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZd dZdZdZ dZ!dZ"dZ#dZ$dZ%y)!ra **Bases:** :class:`GraphicsObject ` Displays a line of infinite length. This line may be dragged to indicate a position in data coordinates. =============================== =================================================== **Signals:** sigDragged(self) sigPositionChangeFinished(self) sigPositionChanged(self) sigClicked(self, ev) =============================== =================================================== Nc d|_| |_tj|| ddg|_n||_d|_|j |d|_ddg|_|j|| tdd}|j||d}|j||,|jd|jjn|j|| |_|j|_g|_d|_| | D]} |j(| ddg|_d|_d|_||in|}t1|fd|i||_yy) a8 =============== ================================================================== **Arguments:** pos Position of the line. This can be a QPointF or a single value for vertical/horizontal lines. angle Angle of line in degrees. 0 is horizontal, 90 is vertical. pen Pen to use when drawing line. Can be any arguments that are valid for :func:`mkPen `. Default pen is transparent yellow. hoverPen Pen to use when the mouse cursor hovers over the line. Only used when movable=True. movable If True, the line can be dragged to a new position by the user. bounds Optional [min, max] bounding values. Bounds are only valid if the line is vertical or horizontal. hoverPen Pen to use when drawing line when hovering over it. Can be any arguments that are valid for :func:`mkPen `. Default pen is red. label Text to be displayed in a label attached to the line, or None to show no label (default is None). May optionally include formatting strings to display the line value. labelOpts A dict of keyword arguments to use when constructing the text label. See :class:`InfLineLabel `. span Optional tuple (min, max) giving the range over the view to draw the line. For example, with a vertical line, use span=(0.5, 1) to draw only on the top half of the view. markers List of (marker, position, size) tuples, one per marker to display on the line. See the addMarker method. name Name of the item =============== ================================================================== NFr)rd)rr)colorwidthr text) _boundingRect_namer __init__maxRangemoving setMovable mouseHoveringpsetAnglersetPossetPen setHoverPenpenrspan currentPenmarkers_maxMarkerSize addMarker _endPoints_bounds _lastViewSizerlabel) selfposangler%movableboundshoverPenr. labelOptsr&r(namems F/usr/lib/python3/dist-packages/pyqtgraph/graphicsItems/InfiniteLine.pyrzInfiniteLine.__init__%s\B" % >!4LDM"DM   "Q e ;!*C C ;!C C     9DHHNN4D  E   X & ((    #" #a& !  '/YI%dDD)DDJ c4||_|j|y)z,Set whether the line is movable by the user.Nr2setAcceptHoverEventsr/r7s r8rzInfiniteLine.setMovableus  !!!$r9cP||_|j|jy)z:Set the (minimum, maximum) allowable values when dragging.N)rsetValuevalue)r/r3s r8 setBoundszInfiniteLine.setBoundszs  djjl#r9c |jddS)zDReturn the (minimum, maximum) values allowed when dragging. N)rr/s r8r3zInfiniteLine.boundss}}Qr9ctj|i||_|js"|j|_|j yy)zySet the pen for drawing the line. Allowable arguments are any that are valid for :func:`mkPen `.N)fnmkPenr%rr'update)r/argskwargss r8r#zInfiniteLine.setPens;88T,V,!!"hhDO KKM"r9ct|dk(xr;t|dtjxst|dtxrd|dvxsd|v}t j |i||_|s3|jj|jj|jr"|j|_ |jyy)aSet the pen for drawing the line while the mouse hovers over it. Allowable arguments are any that are valid for :func:`mkPen `. If the line is not movable, then hovering is also disabled. Added in version 0.9.9.r rrN)len isinstancer QPendictrErFr4setWidthr%rrr'rG)r/rHrIwidthSpecifieds r8r$zInfiniteLine.setHoverPenst9>O&tAw ;N&tAw5L'T!W:L1 '&0 $1&1  MM " "488>>#3 4   "mmDO KKM r9ctj}|dk(r'|jtjddddd|vrWtj t ddt ddt ddg}|j||jd|vrWtj t ddt ddt ddg}|j||jd|vrWtj t ddt ddt ddg}|j||jd |vrWtj t ddt ddt ddg}|j||jd |vrWtj t ddt ddt ddg}|j||jd |vrWtj t ddt ddt ddg}|j||j|jj|||ft|jDcgc] }|d d z  c}|_ |jycc}w)aAdd a marker to be displayed on the line. ============= ========================================================= **Arguments** marker String indicating the style of marker to add: ``'<|'``, ``'|>'``, ``'>|'``, ``'|<'``, ``'<|>'``, ``'>|<'``, ``'^'``, ``'v'``, ``'o'`` position Position (0.0-1.0) along the visible extent of the line to place the marker. Default is 0.5. size Size of the marker in pixels. Default is 10.0. ============= ========================================================= ogr z<|?rz|>z>|z|<^vrg@N)r QPainterPath addEllipserQRectF QPolygonFr addPolygon closeSubpathr(appendmaxr)rG)r/markerpositionsizepathr r7s r8r*zInfiniteLine.addMarkers'!!# S= OOFMM$a; < 6>sAadA OPA OOA      6>sAa uT1~ NOA OOA      6>sD!15A;dD@Q RSA OOA      6>sC%1+uT3?O PQA OOA      &=q$sAa NOA OOA      &=q$tQq# OPA OOA      T8T23!dll"C1Q4"9"CD #DsKc@g|_d|_|jy)z, Remove all markers from this line. rN)r(r)rGrCs r8 clearMarkerszInfiniteLine.clearMarkerss  r9c||_|j|j|j|jy)z Takes angle argument in degrees. 0 is horizontal; 90 is vertical. Note that the use of value() and setValue() changes if the line is not vertical or horizontal. N)r1resetTransform setRotationrG)r/r1s r8r!zInfiniteLine.setAngles2   $ r9ct|tttjfr$tj |dk(s t|}nnt|t jr!|j|jg}n3|jdk(r|dg}n|jdk(rd|g}n td|jdk(r]|jdt|d|jd|d<|jdt|d|jd|d<nk|jdk(r\|jdt|d|jd|d<|jdt|d|jd|d<|j|k7r\||_|j!t#j$|t'|j|j(j+|yy)NrZz4Must specify 2D coordinate for non-orthogonal lines.r )rLlisttuplenpndarrayndimrQPointFxyr1 Exceptionrr]minr viewTransformChangedr r"rsigPositionChangedemit)r/r0newPoss r8r"zInfiniteLine.setPoss cD%4 5bggcla>O#YF V^^ ,eegsuuw'FzzRqqS VWW :: }}Q+q 4==+;<q }}Q+q 4==+;<q ZZ1_}}Q+q 4==+;<q }}Q+q 4==+;<q 66V DF  % % '  ! !$dff 6  # # ( ( . r9c |jdSNrr rCs r8getXPoszInfiniteLine.getXPosvvayr9c |jdS)Nr ryrCs r8getYPoszInfiniteLine.getYPosr{r9c|jSNryrCs r8getPoszInfiniteLine.getPoss vv r9c|jdzdk(r|jS|jdzdk(r|jS|jS)zReturn the value of the line. Will be a single number for horizontal and vertical lines, and a list of [x,y] values for diagonal lines.rrh)r1r}rzrrCs r8r@zInfiniteLine.value sH ::c>Q <<> ! ZZ^r !<<> !;;= r9c&|j|y)zSet the position of the line. If line is horizontal or vertical, v can be a single value. Otherwise, a 2D coordinate must be specified (list, tuple and QPointF are all acceptable).N)r"r/rUs r8r?zInfiniteLine.setValues Ar9cZ|j||fk7r||f|_|jyyr)r&rG)r/mnmxs r8setSpanzInfiniteLine.setSpan%s+ 99R RDI KKM !r9c|j}|tjS|jt dd\}}|dn|j }t |jjdz |jjdz }|j|zdz|z}tj|}|j| |j||j}|j||jdzz} |j||jdzz} |j| |j!| |j#}|j%j'} |j(|k7s|j*| k7r||_| |_|j-| | f|_||_|j(S)Nr r) directionr)viewRectrrX pixelVectorsrrpr]r%rr4r) setBottomsetTopleftr&setLeftsetRight normalized getViewBoxr`r,r-prepareGeometryChanger+ _lastViewRect) r/vr_orthopxpwwbrlengthrrightvss r8_computeBoundingRectz!InfiniteLine._computeBoundingRect*s ]]_ :==? "$$uQ{$;5-QUWWY !A%t}}':':'5?3 t||  ! [[] uT1~&ffU5!_% VVE$N #Eks""$ceegsuuw/0 E  %Zddfsuuw2446!11 a1R( 2::doo33567 [[]#|| OD#t NN2  A KK1  GGD$  JJt   r9c|dk(ryy)Nr)rr)r/axisfrac orthoRanges r8 dataBoundszInfiniteLine.dataBounds}s 19r9c|jr?|jtjjj k(r|j rQd|_|j|j|jz |_ |j|_ |j|jsy|j|j|j|jz|jj!||j#r#d|_|j$j!|yyyyNTF)r2buttonrr MouseButton LeftButtonisStartrr0 mapToParent buttonDownPos cursorOffset startPositionacceptr" sigDraggedruisFinishsigPositionChangeFinishedr/evs r8mouseDragEventzInfiniteLine.mouseDragEvents <->?& ?((<.C:&'2G'2G::#q('2G'2G $'$' 4  r9c|jsy|jj}|j|jj ||j y)N)r@) isVisiblerr@setTextrupdatePosition)r/r@s r8rzInfLineLabel.valueChangedsI~~  ! T[[''e'45 r9c|jd|jj}t|j d}t|j d}|jj dzdk7rS|j}|jrt|tsytj}|j||j||jj|dj!|}tj}|j#|j|j%|j'tj(}t+|dkDrLt-|d}|jj/||d}|jj/||d}||f|_|jS)Nrrhrr )rrrrrrr1rrrLrr rVmoveTolineTo itemTransformraddRect intersectedtoSubpathPolygons QTransformrKri mapFromItem) r/lrpt1pt2viewr rpathsls r8 getEndpointszInfLineLabel.getEndpointssn ??1  %'')B 1%C A&Cyy#q((~~'z$/H'&&(  II++D1!488;'') 4,,./q);;EU1XA))//ad;C))//ad;C"CjDOr9c`d|_|j\}}|y||jz|d|jz zz}|j||jj }|B|j |j|jjdkrdndyy)Nrr r) rrrr"rr setAnchorrcenterrp)r/r r ptrs r8rzInfLineLabel.updatePositions&$$&S ;  4== 3!DMM/#: : BYY   ! > NN4<s   r9cX|jr|jtjjj k(r|j r+>+DtGYGY+Y[]_abDM    !{{}$ L68 J % %" =, 7r9)mathrrnumpyrkr%rrErrrr r r r r__all__rrrr9r8r)sG&* > *j>jZ p78p7r9