e=ddlZddlmZddlmZddlmZddlmZm Z ddl m Z dd l m Z dd l m Z dd lmZdd lmZd dgZGdd eZGdde Zy)N)atan2) functions)Point)QtCoreQtGui)GraphicsObject)Symbols)TextItem)UIGraphicsItem)ViewBox TargetItem TargetLabelceZdZdZej eZej eZ dfd Z fdZ dZ dZ dZ dZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZddZxZS)rzDraws a draggable target symbol (circle plus crosshair). The size of TargetItem will remain fixed on screen even as the view is zoomed. Includes an optional text label. c t |||_d|_d|_d|_|d}|j ||d}|j||d}|j||d}|j||j|_ |j|_ d|_tdd|_| tdd}|j#|t%|t&r t(||_n-t%|t.j0r||_n t3d ||_|j7|j*|j9| | y#t,$r t-dwxYw) aF Parameters ---------- pos : list, tuple, QPointF, QPoint, Optional Initial position of the symbol. Default is (0, 0) size : int Size of the symbol in pixels. Default is 10. pen : QPen, tuple, list or str Pen to use when drawing line. Can be any arguments that are valid for :func:`~pyqtgraph.mkPen`. Default pen is transparent yellow. brush : QBrush, tuple, list, or str Defines the brush that fill the symbol. Can be any arguments that is valid for :func:`~pyqtgraph.mkBrush`. Default is transparent blue. movable : bool If True, the symbol can be dragged to a new position by the user. hoverPen : QPen, tuple, list, or str Pen to use when drawing symbol when hovering over it. Can be any arguments that are valid for :func:`~pyqtgraph.mkPen`. Default pen is red. hoverBrush : QBrush, tuple, list or str Brush to use to fill the symbol when hovering over it. Can be any arguments that is valid for :func:`~pyqtgraph.mkBrush`. Default is transparent blue. symbol : QPainterPath or str QPainterPath to use for drawing the target, should be centered at ``(0, 0)`` with ``max(width, height) == 1.0``. Alternatively a string which can be any symbol accepted by :func:`~pyqtgraph.ScatterPlotItem.setSymbol` label : bool, str or callable, optional Text to be displayed in a label attached to the symbol, or None to show no label (default is None). May optionally include formatting strings to display the symbol value, or a callable that accepts x and y as inputs. If True, the label is ``x = {: >.3n}\ny = {: >.3n}`` False or None will result in no text being displayed labelOpts : dict A dict of keyword arguments to use when constructing the text label. See :class:`TargetLabel` and :class:`~pyqtgraph.TextItem` FN)rr)rrr)rrr2)rrrdrz&symbol name found in available SymbolszUnknown type provided as symbol)super__init__movablemoving_label mouseHoveringsetPen setHoverPensetBrush setHoverBrushpen currentPenbrush currentBrush_shaper_possetPos isinstancestrr _pathKeyErrorr QPainterPath TypeErrorscalesetPathsetLabel) selfpossizesymbolr hoverPenr" hoverBrushrlabel labelOpts __class__s D/usr/lib/python3/dist-packages/pyqtgraph/graphicsItems/TargetItem.pyrzTargetItem.__init__sWh    " ;C C  $H " =#E e  +J :&(( JJ !QK ;1+C C fc " I$V_  2 2 3DJ=> >  TZZ  eY' IGHH Is *EE0c t|}|j|k7r<||_t||j|j j|yy#t$rt$rtd|wxYw)aMethod to set the position to ``(x, y)`` within the plot view Parameters ---------- args : tuple or list or QtCore.QPointF or QtCore.QPoint or Point or float Two float values or a container that specifies ``(x, y)`` position where the TargetItem should be placed Raises ------ TypeError If args cannot be used to instantiate a Point z%Could not make Point from arguments: N)rr, Exceptionr%rr&sigPositionChangedemit)r0argsnewPosr8s r9r&zTargetItem.setPos{s ND\F 99 DI GN499 %  # # ( ( .     NCD8LM M Ns A!A8ctj|i||_|js"|j|_|j yy)zSet the brush that fills the symbol. Allowable arguments are any that are valid for :func:`~pyqtgraph.mkBrush`. N)fnmkBrushr"rr#updater0r>kwargss r9rzTargetItem.setBrushs<ZZ00 !! $ D  KKM"ctj|i||_|jr"|j|_|j yy)zSet the brush that fills the symbol when hovering over it. Allowable arguments are any that are valid for :func:`~pyqtgraph.mkBrush`. N)rArBr5rr#rCrDs r9rzTargetItem.setHoverBrushs<**d5f5    $D  KKM rFctj|i||_|js"|j|_|j yy)ztSet the pen for drawing the symbol. Allowable arguments are any that are valid for :func:`~pyqtgraph.mkPen`.N)rAmkPenr rr!rCrDs r9rzTargetItem.setPens;88T,V,!!"hhDO KKM"rFctj|i||_|jr"|j|_|j yy)zSet the pen for drawing the symbol when hovering over it. Allowable arguments are any that are valid for :func:`~pyqtgraph.mkPen`.N)rArIr4rr!rCrDs r9rzTargetItem.setHoverPens;$1&1   "mmDO KKM rFc>|jjSN)shape boundingRectr0s r9rNzTargetItem.boundingRectszz|((**rFc|j|j|j|j|j |j yrL)rr!rr#drawPathrM)r0p_s r9paintzTargetItem.paints6 ! 4$$% 4::< rFc>||jk7r||_d|_yrL)r)r$)r0paths r9r.zTargetItem.setPaths 4:: DJDKrFc|j5|j}| |jS||_|j|jSrL)r$ generateShaper)prepareGeometryChange)r0ss r9rMzTargetItem.shapesH ;; ""$Ayzz!DK  & & ({{rFc|j}||j|_y|jt j dd|jt j ddz }t j|}|jt j dd}tj}|j|j|jt|j|j}|j||j|j|j|j|j|jS)Nr r)deviceTransformr)r$maprQPointFrAinvertQTransformr QTransform translatexyr rotateRadiansr-)r0dtvdtidevPostrvas r9rXzTargetItem.generateShapes  ! ! # :**DK FF6>>!Q' (266&..A2F+G G!!"%q!,-     VXXZ, 1335!##%   TZZ(wwrvvdjj)**rFc6|jr5|jtjjj k7ry|j |jr<|j|j|jz |_ d|_ |jsy|j|j|j|jz|jr#d|_ |jj!|yyNTF)rbuttonrQt MouseButton LeftButtonacceptisStartr1 mapToView buttonDownPos symbolOffsetrr&isFinishsigPositionChangeFinishedr=r0evs r9mouseDragEventzTargetItem.mouseDragEvents||ryy{fii.C.C.N.NN  ::< $ T^^B+J !!#..t{{;DKt|1'/YI{{& !!#..t{{;%dCCCDKrF) N crosshairNNNNTNN)NN)__name__ __module__ __qualname____doc__rSignalobjectr<rwrr&rrrrrNrTr.rMrXrzr}rrrr1r6r/ __classcell__r8s@r9rrs 'v. - f 5   `(D/4+!   +6 6 &   DrFcReZdZdZ d fd ZdZdZdZfdZdZ dZ xZ S) raA TextItem that attaches itself to a TargetItem. This class extends TextItem with the following features : * Automatically positions adjacent to the symbol at a fixed position. * Automatically reformats text when the symbol location has changed. Parameters ---------- target : TargetItem The TargetItem to which this label will be attached to. text : str or callable, Optional Governs the text displayed, can be a fixed string or a format string that accepts the x, and y position of the target item; or be a callable method that accepts a tuple (x, y) and returns a string to be displayed. If None, an empty string is used. Default is None offset : tuple or list or QPointF or QPoint Position to set the anchor of the TargetLabel away from the center of the target in pixels, by default it is (20, 0). anchor : tuple or list or QPointF or QPoint Position to rotate the TargetLabel about, and position to set the offset value to see :class:`~pyqtgraph.TextItem` for more information. kwargs : dict kwargs contains arguments that are passed onto :class:`~pyqtgraph.TextItem` constructor, excluding text parameter c 6t|tr||_nt|ttfrt||_nct|t j t jfr.t|j|j|_n tdt|0dd|i||j|||_|j||jj j#|j$|j%y)Nz'Offset parameter is the wrong data typeanchor)r'roffsettuplelistrQPointr^rbrcr,rr setParentItemtarget setFormatr<connect valueChanged)r0rrrrrEr8s r9rzTargetLabel.__init__as fe $ DK  ..DK  ? @ FHHJ7DKEF F 11&1 6"  t &&..t/@/@A rFc|jSrL)_formatrOs r9formatzTargetLabel.formatzs ||rFcFt|smttjj |}|r|dd|j ddn6|j ||j ddn|j dd||_|jy)a Method to set how the TargetLabel should display the text. This method should be called from TargetItem.setLabel directly. Parameters ---------- text : Callable or str Details how to format the text. If Callable, then the label will display the result of ``text(x, y)`` If a fromatted string, then the output of ``text.format(x, y)`` will be displayed If a non-formatted string, then the text label will display ``text`` rr NformattableTextTF) callablerstring Formatterparse setPropertysetTextrr)r0rparseds r9rzTargetLabel.setFormat}s~&**,22489F&)A,2  !2D9 T"  !2E:   . 6  rFcJ|jj\}}|jdr>|j|jj t |t |yt|jr"|j|j ||yy)Nr)rr1propertyrrrfloatr)r0rbrcs r9rzTargetLabel.valueChangedsp{{ 1 ==* + LL,,U1XuQx@ A dll # LLa+ ,$rFcF|j}t|trs|j}t j |j j|dz|j j|dz}|j|t|-S)Nrr ) getViewBoxr'r viewPixelSizerr^rrbrcr&rr)r0viewboxr scaledOffsetr8s r9rz TargetLabel.viewTransformChangeds//# gw '#113M!>> -"22DKKMMOmTUFV4VL KK %w+--rFc@|jj|SrL) parentItemr}rxs r9r}zTargetLabel.mouseClickEvents 0044rFcV|j}|jr5|jtjj j k7ry|j|jr<|j|j|jz |_ d|_ |jsy|j|j|j|jz|jr#d|_ |j j#|yyrl)rrrmrrnrorprqrrr1rsrtrurr&rvrwr=)r0ry targetItems r9rzzTargetLabel.mouseDragEvents__& !!RYY[FII4I4I4T4T%T  ::<&0nn&6  ":'J #!%J   *11DNN26684LLM ;;= %J   0 0 5 5d ; rF))r)rg?) rrrrrrrrrr}rzrrs@r9rrFs6: 22-.5rsL *$*  'tDtDn u<(u