eddlZddlmZddlZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z dd l m Z mZmZdd lmZd d lmZe j&j(ZdgZGddeZy)N)Callable)colormap)debug) functions)functions_qimage)getConfigOption)Point)QtCoreQtGui QtWidgets)getCupy)GraphicsObject ImageItemceZdZdZej Zej eZd1dZ dZ dZ dZ dZ dZd Zd2d Zd Zd Zd Zd2dZedZd2dZd2dZdZdZdZd3dZd4dZdZdZdZ dZ!d5dZ"dZ#dZ$dZ%dZ& d6d Z'd!Z(d"Z)d#Z*d$Z+d%Z,d&Z-d'Z.d(Z/d)Z0d*Z1d+Z2d,Z3d1d-Z4d7d.Z5d/Z6d0Z7y)8rzG **Bases:** :class:`GraphicsObject ` Nc tj|d|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_t%d|_dx|_|_|j-|j&d|_d|_d|_||j4|fi|y|j6di|y)a See :func:`~pyqtgraph.ImageItem.setOpts` for further keyword arguments and and :func:`~pyqtgraph.ImageItem.setImage` for information on supported formats. Parameters ---------- image: np.ndarray, optional Image data NFrrTimageAxisOrder)r__init__menuimageqimage paintModelevelslutautoDownsample _colorMap_lastDownsample_processingBuffer_displayBuffer_renderRequired _unrenderable_xp_defferedLevels _imageHasNansr axisOrder_dataTransform_inverseDataTransform_update_data_transforms drawKernelborder removablesetImagesetOpts)selfrkargss C/usr/lib/python3/dist-packages/pyqtgraph/graphicsItems/ImageItem.pyrzImageItem.__init__s %    #%!%"#"#!()9:;??d8 $$dnn6    DMM% )5 ) DLL !5 !c2||_|jy)as Change the composition mode of the item. This is useful when overlaying multiple items. Parameters ---------- mode : ``QtGui.QPainter.CompositionMode`` Composition of the item, often used when overlaying items. Common options include: ``QPainter.CompositionMode.CompositionMode_SourceOver`` (Default) Image replaces the background if it is opaque. Otherwise, it uses the alpha channel to blend the image with the background. ``QPainter.CompositionMode.CompositionMode_Overlay`` Image color is mixed with the background color to reflect the lightness or darkness of the background ``QPainter.CompositionMode.CompositionMode_Plus`` Both the alpha and color of the image and background pixels are added together. ``QPainter.CompositionMode.CompositionMode_Plus`` The output is the image color multiplied by the background. See ``QPainter::CompositionMode`` in the Qt Documentation for more options and details N)rupdate)r1modes r3setCompositionModezImageItem.setCompositionModeGs8 r4cXtj||_|jy)z Defines the border drawn around the image. Accepts all arguments supported by :func:`~pyqtgraph.mkPen`. N)fnmkPenr-r6r1bs r3 setBorderzImageItem.setBorderfs hhqk  r4ct|jy|jdk(rdnd}|jj|S)N col-majorrrrr(shaper1axiss r3widthzImageItem.widthn6 :: NNk1qqzz%%r4ct|jy|jdk(rdnd}|jj|S)Nr@rrrArCs r3heightzImageItem.heighttrFr4c|jy|jjdk(r|jjdSdS)Nrr)rndimrBr1s r3channelszImageItem.channelszs8 :: &*jjoo&:tzz"AAr4c |jtjddddStjddt|j t|j S)N)rr QRectFfloatrErHrLs r3 boundingRectzImageItem.boundingRectsL :: ==RR0 0}}RU4::<%8% :NOOr4c|j||_||_y||jj|}||_|r|j yy)a Sets image scaling levels. See :func:`makeARGB ` for more details on how levels are applied. Parameters ---------- levels: array_like - ``[blackLevel, whiteLevel]`` sets black and white levels for monochrome data and can be used with a lookup table. - ``[[minR, maxR], [minG, maxG], [minB, maxB]]`` sets individual scaling for RGB values. Not compatible with lookup tables. update: bool, optional Controls if image immediately updates to reflect the new levels. N)r%rr&asarray updateImage)r1rr6s r3 setLevelszImageItem.setLevelssU 88  DK#)D   XX%%f-F      r4c|jS)z Returns the list representing the current level settings. See :func:`~setLevels`. When ``autoLevels`` is active, the format is ``[blackLevel, whiteLevel]``. )rrLs r3 getLevelszImageItem.getLevelss {{r4c t|tjr||_n6t|trtj ||_n t d|j|jjdy)a Sets a color map for false color display of a monochrome image. Parameters ---------- colorMap : :class:`~pyqtgraph.ColorMap` or `str` A string argument will be passed to :func:`colormap.get() ` z.'colorMap' argument must be ColorMap or string)nPtsN) isinstancerColorMaprstrget TypeErrorsetLookupTablegetLookupTable)r1colorMaps r3 setColorMapzImageItem.setColorMapsa h 1 1 2%DN # &%\\(3DNLM M T^^:::DFr4c|jS)z^ Returns the assigned :class:`pyqtgraph.ColorMap`, or `None` if not available )rrLs r3 getColorMapzImageItem.getColorMaps~~r4c||jurC|j|j||j}||_|r|jyyy)a6 Sets lookup table ``lut`` to use for false color display of a monochrome image. See :func:`makeARGB ` for more information on how this is used. Optionally, `lut` can be a callable that accepts the current image as an argument and returns the lookup table to use. Ordinarily, this table is supplied by a :class:`~pyqtgraph.HistogramLUTItem`, :class:`~pyqtgraph.GradientEditorItem` or :class:`~pyqtgraph.ColorBarItem`. Setting ``update = False`` avoids an immediate image update. N)rr%_ensure_proper_substraterU)r1rr6s r3razImageItem.setLookupTablesP dhh xx#33CBDH  " r4cZ|&t|tst||jr|St}||k(r)t||js|j |}|S|t k(r?|(t||jr|j }|St j|}|SN)r\rndarrayrrTnumpyr_)data substratecupys r3rhz"ImageItem._ensure_proper_substrates <:dH5D)J[J[9\Ky  Zdll%C<<%D  % JtT\\$Bxxz }}T* r4c@||_d|_|jy)a Controls automatic downsampling for this ImageItem. If `active` is `True`, the image is automatically downsampled to match the screen resolution. This improves performance for large images and reduces aliasing. If `autoDownsample` is not specified, then ImageItem will choose whether to downsample the image based on its size. `False` disables automatic downsampling. TN)rr#r6)r1actives r3setAutoDownsamplezImageItem.setAutoDownsamples%# r4c Pd|vr6|d}|dvr td||_|j|jd|vr|j|dd|vr|j |d|d|vr|j |d|d|vr|j |dd |vr|j|d d |vr|j|d d |vr|d |_ d |_ d |vr|j|d d|vr|j|d|r|jy y )a Sets display and processing options for this ImageItem. :func:`~pyqtgraph.ImageItem.__init__` and :func:`~pyqtgraph.ImageItem.setImage` support all keyword arguments listed here. Parameters ---------- autoDownsample: bool See :func:`~pyqtgraph.ImageItem.setAutoDownsample`. axisOrder: str | `'col-major'`: The shape of the array represents (width, height) of the image. This is the default. | `'row-major'`: The shape of the array represents (height, width). border: bool Sets a pen to draw to draw an image border. See :func:`~pyqtgraph.ImageItem.setBorder`. compositionMode: See :func:`~pyqtgraph.ImageItem.setCompositionMode` colorMap: :class:`~pyqtgraph.ColorMap` or `str` Sets a color map. A string will be passed to :func:`colormap.get() ` lut: array_like Sets a color lookup table to use when displaying the image. See :func:`~pyqtgraph.ImageItem.setLookupTable`. levels: array_like Shape of (min, max). Sets minimum and maximum values to use when rescaling the image data. By default, these will be set to the estimated minimum and maximum values in the image. If the image array has dtype uint8, no rescaling is necessary. See :func:`~pyqtgraph.ImageItem.setLevels`. opacity: float Overall opacity for an RGB image. Between 0.0-1.0. rect: :class:`QRectF`, :class:`QRect` or array_like Displays the current image within the specified rectangle in plot coordinates. If ``array_like``, should be of the of ``floats (`x`,`y`,`w`,`h`)`` . See :func:`~pyqtgraph.ImageItem.setRect`. update : bool, optional Controls if image immediately updates to reflect the new options. r() row-majorr@z3axisOrder must be either 'row-major' or 'col-major'rcr)r6ropacitycompositionModer-r.Nrrect) ValueErrorr(r+rdrarV setOpacityr8r>r.rrrsetRectr6)r1r6r2vals r3r0zImageItem.setOptssHH %  $C44 !VWW DN  ( ( 8     U:. / E>   e V  < u  NN5?6N :   OOE), -  %  # #E*;$< = u  NN5? + % ";/DNDI u $  " "5)9#: ; U? LLv '  KKM r4c:t|dk(r|jyt|dtjtj fr|d}n&t |ddr|d}tj|}tj}|j|j|j|j|j|jz |j|jz |j|y)a setRect(rect) or setRect(x,y,w,h) Sets translation and scaling of this ImageItem to display the current image within the rectangle given as ``rect`` (:class:`QtCore.QRect` or :class:`QtCore.QRectF`), or described by parameters `x, y, w, h`, defining starting position, width and height. This method cannot be used before an image is assigned. See the :ref:`examples ` for how to manually set transformations. rN__len__)lenresetTransformr\r rPQRecthasattrr QTransform translatelefttopscalerErH setTransform)r1argsrwtrs r3rzzImageItem.setRect,s t9>    !  d1g v||< =7DtAwy)Aw==4)D     TYY[$((*-  ,dkkmdkkm.KL "r4crd|_|j|j|jy)z, Clears the assigned image. N)rprepareGeometryChangeinformViewBoundsChangedr6rLs r3clearzImageItem.clearEs,  ""$ $$& r4cJtj|dddztj|_|jt k(rD|jj|dddz|jj|_d|_y|j|_d|_y)Nr)dtype)rlemptyubyter"r%rr!r)r1rBs r3_buildQImageBufferzImageItem._buildQImageBufferNs~#kk%)d*:%++N 88wy %)XX^^E"1I4DDHHNN^%[D " &*%8%8D " r4c xtj}d}||jy|j}t }|r|j |nt |_d}||jk7}|rd|_|xs3|jduxs#|j|jjk7} |j}||_d|_ |jjddkDs|jjddkDr d|vrd|d<| r |j|j|| d|vrd}nd}|rm|jd d } |j| \} } | | k(s6|jj| s|jj| rd} d } | | g|d<||j dd di||d|_|j%||r|j&j)|j*%|j*} d|_|j-| yy)a Updates the image displayed by this ImageItem. For more information on how the image is processed before displaying, see :func:`~pyqtgraph.makeARGB`. For backward compatibility, image data is assumed to be in column-major order (column, row) by default. However, most data is stored in row-major order (row, column). It can either be transposed before assignment:: imageitem.setImage(imagedata.T) or the interpretation of the data can be changed locally through the ``axisOrder`` keyword or by changing the `imageAxisOrder` :ref:`global configuration option ` All keywords supported by :func:`~pyqtgraph.ImageItem.setOpts` are also allowed here. Parameters ---------- image: np.ndarray, optional Image data given as NumPy array with an integer or floating point dtype of any bit depth. A 2-dimensional array describes single-valued (monochromatic) data. A 3-dimensional array is used to give individual color components. The third dimension must be of length 3 (RGB) or 4 (RGBA). rect: QRectF or QRect or array_like, optional If given, sets translation and scaling to display the image within the specified rectangle. If ``array_like`` should be the form of floats ``[x, y, w, h]`` See :func:`~pyqtgraph.ImageItem.setRect` autoLevels: bool, optional If `True`, ImageItem will automatically select levels based on the maximum and minimum values encountered in the data. For performance reasons, this search subsamples the images and may miss individual bright or or dark points in the data set. If `False`, the search will be omitted. The default is `False` if a ``levels`` keyword argument is given, and `True` otherwise. levelSamples: int, default 65536 When determining minimum and maximum values, ImageItem only inspects a subset of pixels no larger than this number. Setting this larger than the total number of pixels considers all values. FNTrirrr levelSamplesi) targetSizer6r)rProfilerrr%rget_array_modulerlr!rBviewr'rrpop quickMinMaxisnanr0r#r6sigImageChangedemitr&rV)r1r autoLevelsr2profile gotNewDataold_xpcpprocessingSubstrateChanged shapeChanged level_samplesmnmxrs r3r/zImageItem.setImageVsN.." =zz!XXFB57r**51UDHJ)/488); &))-&6o$**:LoPUP[P[_c_i_i_o_oPoLJJLEDJ!%D zz"W, 0@0@0Cg0M#50.2E*+**,,,.  5 " ! !IIne>"-1C!"gE(O  +E+U+ #    % % '    +))F#'D NNF $ ,r4cdtj|_tj|_|jdk(ro|jj dd|jj d|jj dd|jj dyy)zF Sets up the transforms needed to map between input array and display rtriN)r rr)r*r(rrotate)r1r(s r3r+z!ImageItem._update_data_transformss#..0%*%5%5%7" >>[ (    % %a ,    & &s +  & & , ,Q 3  & & - -c 2 )r4c|jS)a Returns the transform that maps from this image's input array to its local coordinate system. This transform corrects for the transposition that occurs when image data is interpreted in row-major order. :meta private: )r)rLs r3 dataTransformzImageItem.dataTransforms"""r4c|jS)zReturn the transform that maps from this image's local coordinate system to its input array. See dataTransform() for more information. :meta private: )r*rLs r3inverseDataTransformzImageItem.inverseDataTransforms)))r4c8|jj|Srj)r*mapr1objs r3 mapToDatazImageItem.mapToDatas))--c22r4c8|jj|Srj)r)rrs r3 mapFromDatazImageItem.mapFromDatas""&&s++r4c |j}|dkrd} |jdd\}}||z|krn||kDr |dddddf}n |dddddf}:|jj||jj |fS)a Estimates the min/max values of the image data by subsampling. Subsampling is performed at regular strides chosen to evaluate a number of samples equal to or less than `targetSize`. Returns (`min`, `max`). rN)rrBr%nanminnanmax)r1rrmhws r3rzImageItem.quickMinMaxszz >J::bq>DAq1u "E1uCaCG}B!G} xxt$dhhood&;;;r4cPddi}|j||j|i|S)NrF)r6r/)r1rr2defaultss r3rUzImageItem.updateImages4 %  t}}d/h//r4cd|_|j|jjdk(ry|jjdk(s|jjddk(r|j |j |j|_t|j tr7|j |j |jd|j}n|j }nd}|jr|j\}}|y|jdk(rddgnddg}tj|j||d}tj|||d}||f|_|jdk(r y|j}|jdk(r|jd dk(r|d }|jd k(r|j!dd}|j"}|j$I|j&j(d k(xr)|jj+|j-|_d}|@|j&|jj.k7rt1j2d t4dn$|j$st7j8|||}|$d|_d|_||_d|_ d|_y|j:)|j:jdd|jddk7r|jC|jtjD||||j:|jtGk(r&|j:jI|j<tjJ|j<tLjNjPjR|_d|_ d|_y)NTrrrrZrt)rDrJr).rr@fzUsing non-uint8 LUTs is an undocumented accidental feature and may be removed at some point in the future. Please open an issue if you instead believe this to be worthy of protected inclusion in pyqtgraph.) stacklevel)rrF)rroutput)out)*r$rsizerKrBrhrr%r\rr_computeDownsampleFactorsr(r: downsampler swapaxesrr'rkindrminuint8warningswarnDeprecationWarningrtry_make_qimager!r"rr#rmakeARGBrr_ndarray_to_qimager QImageFormat Format_ARGB32)r1rxdsydsaxesrrrs r3renderzImageItem.renders! :: A!5  ::??a 4::#3#3A#6!#;44TXXtxxHDH$((H-33DHHTZZ4MtxxXhhC   557HC{!^^{:Aq6ADMM$**cQ@EMM%47;E$':D zzQJJE ::?u{{2!3&ME >>[ (NN1a(E    %   C',uyy{+    ?syyDHHNN: MMY#q  2 ##%55eFPSTF  %)D ""&D  DK#(D !&D    ! ! )T-C-C-I-I"1-MQVQ\Q\]_^_Q`-`  # #EKK 0 Es6$:P:PQ 88wy  " " & &4+>+> & ?**4+>+> @S@S@a@ab $"r4ctj}|jy|jr%|j |j ry|d|j #|j|j |d|jdk(r|jjddn|jjddddd}|jtjddg||j|d|j;|j|j|j!|j#yy)Nz render QImagez set comp moder@rrrz p.drawImage)rrrr#rr$rr8r(rB drawImager rPrr-setPendrawRectrR)r1prrrBs r3paintzImageItem.paintWs.." ::      KKM!! O $ >> %  0 O $(,+(E   !$4::K[K[\^]^K_`dbd`dKe FMM!A-u-t{{;  ;; " HHT[[ ! JJt((* + #r4cv|jr|j|jj|g|y)z Saves this image to file. Note that this saves the visible image (after scale/color changes), not the original data. N)r#rrsave)r1fileNamers r3rzImageItem.saveks/    KKM )D)r4c >|j|jjdk(ry|dk(rtdt|jj |jj d|z tdt|jj |jj d|z f}|jj|r||f}|jdd|ddd|df}t|trT|dk(rN|jj|j}|jj|j} | |k(r| dz } |jj|s|jj| ry|jjdvr[t|jj | |z dz }g}|dkDrG|jj!|| d |zz|t }n|jj#|| d }t%|dk(r|jj'|| f}||d <t)} |rg} t+|j d D]} |d| f} | |jj-| } |jj.| fi|}| r;| j1| j3|ddd | j3|df| j1|ddd |df| S||jj-|}|jj.|fi|} | r+| j3| ddd | j3| dfS| ddd | dfS)a Returns `x` and `y` arrays containing the histogram values for the current image. For an explanation of the return format, see :func:`numpy.histogram()`. The `step` argument causes pixels to be skipped when computing the histogram to save time. If `step` is 'auto', then a step is chosen such that the analyzed data has dimensions approximating `targetImageSize` for each axis. The `bins` argument and any extra keyword arguments are passed to :func:`numpy.histogram()`. If `bins` is `auto`, a bin number is automatically chosen based on the image characteristics: * Integer images will have approximately `targetHistogramSize` bins, with each bin having an integer width. * All other types will have `targetHistogramSize` bins. If `perChannel` is `True`, then a histogram is computed for each channel, and the output is a list of the results. NrNNautoruig@@rOg)\(?rbinsr.)rrmaxintr%ceilrBisscalarr\r^ritemrrrrarangelinspacer~rTrrangeisfinite histogramappendasnumpy)r1rstep perChanneltargetImageSizetargetHistogramSizekwdsstepDatarrrhististepChanrs r3 getHistogramzImageItem.getHistogramts/, :: A!5 6>3txx}}TZZ-=-=a-@?-RSTU3txx}}TZZ-=-=a-@?-RSTUWD 88  T "$>"-. 3#CF+#DHH$5$5h$?@&DHH&&x848KKAaD"I!6 1Q48H IJKK1crAaD 12 3K 1 1( ;>!)DI II  yg> ?]]9[.#I499UF    $ $T%7%7 8 II   '%DII yyr4c|js|j|jtjj j rg|jtjj j |jtjj jy|jsA|jr4|jtjj jyyyrj) isExitr, acceptDragsr rrr acceptClicksr#r.rs r3 hoverEventzImageItem.hoverEventsyy{t:r~~fiiNcNcNnNn?o OOFII11<< = OOFII11== > OOFII11== >"0r4cyrjrrs r3 tabletEventzImageItem.tabletEvents r4c|jdk(r3t|jt|jg}n2t|jt|jg}|j}|j }d|j dg}d|j dg}|d|dz |d|dz |j dzg}|d|dz |d|dz |j dzg}dD]} td||  } td|jj d|| z } || xx| | zz cc<|| xx| | zz cc<td||  } td|jj d|| z } || xx| | zz cc<|| xx| | zz cc<t|d|dt|d|df}t|d|dt|d|df}|j}|}t|jtr!|j||j||||y||}|jdk(r@|.||}|j|d|z z||zz|j|<nO||j|<n?|jdk(r|j|xx|z cc<ntd|jz|jy)Nr@rr)rrsetaddzUnknown draw mode '%s')r(rrrr,drawKernelCenterrBrrslicedrawMaskr\drawModer ExceptionrU)r1rr dkkcsxsytxtyrdx1dx2dy1dy2tsssmasksrcs r3rzImageItem.drawAt%s >>[ (suuw<SUUW.Csuuw<SUUW.C __  " " _ _!fr!unc!fr!unbhhqk9 :!fr!unc!fr!unbhhqk9 : Aq"Q%=.Ca))!,RU23C qESW E qESW Eq"Q%=.Ca))!,RU23C qESW E qESW E BqE"Q% %1be"4 5BqE"Q% %1be"4 5}} dmmX . MM"djj$B ;b'C}}%#8D%)ZZ^qv%>t%KDJJrN%(DJJrN%' 2#% 84== HII    r4c<||_||_||_||_yrj)r,rArDrC)r1kernelrRcenterr7s r3 setDrawKernelzImageItem.setDrawKernelQs  &  r4ctj|_|jjj |j |jj dy)Nr)r QTimer removeTimertimeoutr3emitRemoveRequestedstartrLs r3r4zImageItem.removeClickedWsD!==?   (()A)AB q!r4c|jjj|j|jj |yrj)rZr[ disconnectr\sigRemoveRequestedrrLs r3r\zImageItem.emitRemoveRequested]s6   ++D,D,DE $$T*r4rj)Tr)r@)g.A)rrFr)NN)rrr?)8__name__ __module__ __qualname____doc__r Signalrobjectr`rr8r>rErHrMrRrVrXrdrfra staticmethodrhrrr0rzrrr/r+rrrrrrUrrrrrrrr rrr!r%r$r'r;r=rrWr4r\rr4r3rrs5$fmmoO&v.("T>& & B P 2G" #$   @D2`%D3 # *3,<( 0U#n,(*X[),I-VJ4B  :&& ? *X " +r4)rcollections.abcrrlrrrr:rr r rr r r util.cupy_helperrrQCoreApplicationr__all__rrr4r3rnsR$ ))&*  # # - - -J +J +r4