e`dgZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZdd lmZdd lmZmZmZGd deZGddeZy) GraphicsItemN) OrderedDict)reduce)hypot)Optional)Element) functions) GraphicsScene)Point)QtCore QtWidgetsisQObjectAlivec8eZdZdZdfd ZfdZfdZxZS)LRUz?Limit size, evicting the least recently looked-up key when fullc2||_t||i|yN)maxsizesuper__init__)selfrargskwds __class__s F/usr/lib/python3/dist-packages/pyqtgraph/graphicsItems/GraphicsItem.pyrz LRU.__init__s  $'$'cHt||}|j||Sr)r __getitem__ move_to_end)rkeyvaluers rrzLRU.__getitem__s%#C(  rc||vr|j|t| ||t||jkDrt t |}||=yyr)rr __setitem__lenrnextiter)rr r!oldestrs rr#zLRU.__setitem__sQ $;   S ! C' t9t|| #$t*%FV  $r))__name__ __module__ __qualname____doc__rrr# __classcell__)rs@rrrsE( rrcTeZdZdZedZdZdZdZdZ dZ d0d Z d Z d Z d Zd0d Zd1dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d0d Z"d!Z#d"Z$d#Z%d$Z&d0d%Z'd&Z(d'Z)d(Z*d)Z+d0d*Z,d0d+Z-d,Z.d-e/e0e1fd.e2e3e1e4e1ffd/Z5y)2raW **Bases:** :class:`object` Abstract class providing useful methods to GraphicsObject and GraphicsWidget. (This is required because we cannot have multiple inheritance with QObject subclasses.) A note about Qt's GraphicsView framework: The GraphicsView system places a lot of emphasis on the notion that the graphics within the scene should be device independent--you should be able to take the same graphics and display them on screens of different resolutions, printers, export to SVG, etc. This is nice in principle, but causes me a lot of headache in practice. It means that I have to circumvent all the device-independent expectations any time I want to operate in pixel coordinates rather than arbitrary scene coordinates. A lot of the code in GraphicsItem is devoted to this task--keeping track of view widgets and device transforms, computing the size and shape of a pixel in local item coordinates, etc. Note that in item coordinates, a pixel does not have to be square or even rectangular, so just asking how to increase a bounding rect by 2px can be a rather complex task. dcJt|dsH|jjD]/}t|tj s||j_nt|dstdt|zddg|_ d|_ d|_ d|_ d|_ d|_y)N _qtBaseClassz6Could not determine Qt base class for GraphicsItem: %sF)hasattrr __bases__ issubclassr QGraphicsItemr1 Exceptionstr_pixelVectorCache _viewWidget_viewBox_connectedView _exportOpts _cachedView)rbs rrzGraphicsItem.__init__5st^,^^-- a!8!8923DNN/ t^,TWZ[_W``a a"& " rc0|jk|j}|y|j}t|dkryt j |jjd|_|j}| t |sy|S)a Return the view widget for this item. If the scene has multiple views, only the first view is returned. The return value is cached; clear the cached value with forgetViewWidget(). If the view has been deleted by Qt, return None. Nr)r9sceneviewsr$weakrefrefr)rrArBvs r getViewWidgetzGraphicsItem.getViewWidgetFs    #JJLE}KKME5zA~&{{4::<+=+=+?+BCD      =!2rcd|_yr)r9rs rforgetViewWidgetzGraphicsItem.forgetViewWidget]s rc|j|} |j}|>|j}|yt j ||_ |jSt |dr<|jdr+t j ||_ |jS|jS#t$rYywxYw)ac Return the first ViewBox or GraphicsView which bounds this item's visible space. If this item is not contained within a ViewBox, then the GraphicsView is returned. If the item is contained inside nested ViewBoxes, then the inner-most ViewBox is returned. The result is cached; clear the cache with forgetViewBox() N implementsViewBox)r: parentItem RuntimeErrorrFrCrDr2rK)rpvbs r getViewBoxzGraphicsItem.getViewBox`s == A  A9++-Bz#(/ B }}1l+ Y0G$+KKNDM}}}}$  sB:: CCcd|_yr)r:rHs r forgetViewBoxzGraphicsItem.forgetViewBoxzs  rNc|#|j}|y|j}|jj||}|j dk(ry|S)z Return the transform that converts local item coordinates to device coordinates (usually pixels). Extends deviceTransform to automatically determine the viewportTransform. Nr)rFviewportTransformr1deviceTransform determinant)rrUviewdts rrVzGraphicsItem.deviceTransform}sa  $%%'D| $ 6 6 8     . .t5F G >> q Irc|j}|yt|dr3|jdr"|j|j dS|j S)zReturn the transform that maps from local coordinates to the item's ViewBox coordinates If there is no ViewBox, return the scene transform. Returns None if the item does not have a view.NrKrLr)rQr2rK itemTransforminnerSceneItemsceneTransformrrXs r viewTransformzGraphicsItem.viewTransforms[  < 4 &4??9+E%%d&9&9&;""$!  rc ,|j}|y|j|j|jd|j |j dddd |t jdd}n|jdk(r td|j|j |j|j |j|jf}||jdk(r&ttt|jdS|j j#|d}|"||g|_ ttt|S|}t j$t jdd|}|j|}|j'dk(ry |j)}|j+} t-j.|} t| j|j1t| j| j1f}||jd<||jd<||j |<|jdS#td|zxYw)asReturn vectors in local coordinates representing the width and height of a view pixel. If direction is specified, then return vectors parallel and orthogonal to it. Return (None, None) if pixel size is not yet defined (usually because the item has not yet been displayed) or if pixel size is below floating-point precision limit. NNNrr@z0Cannot compute pixel length for 0-length vector.zInvalid direction %s)rV setMatrixm11m12m21m22r QPointFmanhattanLengthr6xyr8tuplemapr _pixelVectorGlobalCachegetQLineFlength unitVector normalVectorfninvertQTransformp2) r directionrYr pv directionrdirLineviewDirnormView normOrthodtis r pixelVectorszGraphicsItem.pixelVectorss0 ! ! # : RVVXrvvxBFFHbffh1aK  q!,I  & & (A -NO Ovvx2668RVVXy{{}ikkmT $((+ +UD$:$:1$=>? ? ) ) - -c4 8 >&)2YD "U2' '8 --q 3Z@&&/ >> q  @))+H --/I !!"% 3778$'') *E#'')2D2G2G2I,J J$&q!$'q!,.$$S)%%a(( @2J>? ?s  JJcx|j|\}}||y|r|jS|jS)a0Return the length of one pixel in the direction indicated (in local coordinates) If ortho=True, then return the length of one pixel orthogonal to the direction indicated. Return None if pixel size is not yet defined (usually because the item has not yet been displayed). N)rr|)rrorthonormVorthoVs r pixelLengthzGraphicsItem.pixelLength s? )))4 v =FN ==? "||~rc|j}|dk(ryt|dj|djt|dj|djfS)Nrmrr@)rrrurv)rrEs r pixelSizezGraphicsItem.pixelSize.s\     adffh!)511Q4668+DEErc|j}|ytj|}|jt j ddddj SNrr@rVrrrxr r{r|rvts r pixelWidthzGraphicsItem.pixelWidth5O  ! ! # :   $vvfmmAq!Q/07799rc|j}|ytj|}|jt j ddddj Srrrs r pixelHeightzGraphicsItem.pixelHeight=rrcJ|j}|y|j|S)z Return *obj* mapped from local coordinates to device coordinates (pixels). If there is no device mapping available, return None. N)rVrxrobjrs r mapToDevicezGraphicsItem.mapToDeviceGs(  ! ! # :vvc{rc|j}|yt|tjrtj|}t j |}|j|S)z Return *obj* mapped from device coordinates (pixels) to local coordinates. If there is no device mapping available, return None. N)rV isinstancer QPointrsrrrxrs r mapFromDevicezGraphicsItem.mapFromDeviceQsT  ! ! # : c6== )..%C   $vvc{rcJ|j}|y|j|S)z Return *rect* mapped from local coordinates to device coordinates (pixels). If there is no device mapping available, return None. N)rVmapRectrrectrs rmapRectToDevicezGraphicsItem.mapRectToDevice^s)  ! ! # :zz$rct|j}|ytj|}|j|S)z Return *rect* mapped from device coordinates (pixels) to local coordinates. If there is no device mapping available, return None. N)rVrrrrs rmapRectFromDevicezGraphicsItem.mapRectFromDevicehs9  ! ! # :   $zz$rcJ|j}|y|j|Sr)r_rxrs r mapToViewzGraphicsItem.mapToViewss&    ! :vvc{rcJ|j}|y|j|Sr)r_rrs r mapRectToViewzGraphicsItem.mapRectToViewys&    ! :zz#rct|j}|ytj|}|j|Sr)r_rrrxrs r mapFromViewzGraphicsItem.mapFromViews6    ! :   $vvc{rct|j}|ytj|}|j|Sr)r_rrrrs rrhzGraphicsItem.mapRectFromViews6    ! :   $zz#rcJt|jj|Sr)r r1posrHs rrzGraphicsItem.possT&&**4011rc^|j|j|jSr)r mapFromParentrrHs rviewPoszGraphicsItem.viewPoss"~~d00<==rc8|jj|Sr)r1rMrHs rrMzGraphicsItem.parentItem  ++D11rc|5|j}|#|j|ur|j||jj||Sr)rAaddItemr1 setParentItem)rparentpscenes rrzGraphicsItem.setParentItemsK  \\^F!djjl&&@t$  ..tV<># #$$33D9 9rc L||j}|j|d}|jtjdddd}|j tj|j |j tjddzS)zReturn the rotation produced by this item's transform (this assumes there is no shear in the transform) If relativeItem is given, then the angle is determined relative to that item. rr@)rMr[rxr r{angleTop1rs)r relativeItemtrvecs rtransformAnglezGraphicsItem.transformAngles~  ??,L    -a 0ffV]]1Qq+,{{6==3668FNN1QS>S!T$'x/?"@$B[B[!\!^  GV,77= #'D   t45**2243H3HI..66t7P7PQ$$,,T-B-BC((001J1JK")++d"3D   ! ! #  % % ' '" w'1"><@sGGGcy)zkCalled when this item's view has changed (ie, the item has been added to or removed from a ViewBox)N)rrXrs rrzGraphicsItem.viewChangedrc||}|jD]H}t|tr$|j|us&|j 7|j ||Jyr)rrrrQrr)rritemchilds rrzGraphicsItem._replaceViews[ <D__& 2E%.##%0%%'!!'51  2rcy)zh Called whenever the view coordinates of the ViewBox containing this item have changed. NrrHs rrzGraphicsItem.viewRangeChanged rrcd|_y)z Called whenever the transformation matrix of the view has changed. (eg, the view range has changed or the view was resized) Invalidates the viewRect cache. N)r=rHs rrz!GraphicsItem.viewTransformChanged's  rc|j}|1t|dr$|jdr|j|yyyy)z Inform this item's container ViewBox that the bounds of this item have changed. This is used by ViewBox to react if auto-range is enabled. NrKrL)rQr2rKitemBoundsChangedr^s rinformViewBoundsChangedz$GraphicsItem.informViewBoundsChanged3sF    l ;PY@Z  " "4 (A[ ; rc|jDcgc]"}|j||j$}}ttj |Scc}w)zTReturn the union of the shapes of all descendants of this item in local coordinates.) allChildItems mapFromItemshaperoperatoradd)rcshapess r childrenShapezGraphicsItem.childrenShape<sH:>:L:L:NOQ$""1aggi0OOhllF++Ps'Ac||}g}|jD]3}|j||j|j|5|S)z>Return list of the entire item tree descending from this item.)rrdextendr)rroottreechs rrzGraphicsItem.allChildItemsAsP <D//# 0B KKO KK**2. / 0 rc.|i}|r||_yd|_y)a This method is called by exporters to inform items that they are being drawn for export with a specific set of options. Items access these via self._exportOptions. When exporting is complete, _exportOptions is set to False. NF)r<)rexportoptss r setExportModezGraphicsItem.setExportModeLs# <D #D  %D rc@t|dr|jgSgS)NgetMenu)r2r)revents rgetContextMenuszGraphicsItem.getContextMenus_s#*4#; CCrnodesreturnct)aaMethod to override to manually specify the SVG writer mechanism. Parameters ---------- nodes Dictionary keyed by the name of graphics items and the XML representation of the the item that can be written as valid SVG. Returns ------- tuple First element is the top level group for this item. The second element is a list of xml Elements corresponding to the child nodes of the item. None Return None if no XML is needed for rendering Raises ------ NotImplementedError override method to implement in subclasses of GraphicsItem See Also -------- pyqtgraph.exporters.SVGExporter._generateItemSvg The generic and default implementation )NotImplementedError)rrs r generateSvgzGraphicsItem.generateSvgbs B"!rr)F)6r)r*r+r,rryrrFrIrQrSrVr_rfrirrrrrrrrrrrrrhrrrMrrr]rrrrrrrrrrrrrdictr7rrrwlistrrrrrr(s "#h ". 4( ) 4Y)x F::      2>2=2: R8(:(xF 2  ),  %&D!"W %!" %g./ 0!"r)__all__rrC collectionsr functoolsrmathrtypingrxml.etree.ElementTreerrr rr r Qtr rrrobjectrrrrrsK  #))22 +*[ "6[ "r