;i=PddlmZddlmZmZddlmZdgZGddeZdS)) functions)QtCoreQtGui)UIGraphicsItemGradientLegendc6eZdZdZdZdZdZdZdZdZ dS) rz Draws a color gradient rectangle along with text labels denoting the value at specific points along the gradient. c||_||_tj||t jjjtj tj dddd|_ tj tj ddd|_tj tj ddd|_ddd|_tj|_|jdtj ddd|jdtj ddd|ddS)Ndr)maxmin)sizeoffsetr__init__setAcceptedMouseButtonsrQt MouseButtonNoButtonrQBrushQColorbrushQPenpentextPenlabelsQLinearGradientgradient setColorAt setZValue)selfrrs ZC:\PYTHON\MyICR_Workspace\venv\Lib\site-packages\pyqtgraph/graphicsItems/GradientLegend.pyrzGradientLegend.__init__ s  %%% $$VY%:%CDDD\%,s3s3"?"?@@ :el1Qq1122z%,q1"5"566 ** -//    EL1Q$7$7888   ELQq$9$9::: sc<||_|dS)N)rupdate)r"gs r# setGradientzGradientLegend.setGradients  r$c8||_dS)zT Set displayed gradient from a :class:`~pyqtgraph.ColorMap` object. N) getGradientr)r"colormaps r# setColorMapzGradientLegend.setColorMaps!,,.. r$c$fdtD}tj}tt|D]=}t |t|z }||||>||dvr5|tdtdidS|ddddddidS)NcBg|]}tj|z gRiS)fnintColor).0iargskargsmaxValminVals r# z3GradientLegend.setIntColorScale..&s9___A"+a??????___r$rr r) rangerrlenfloatr r( setLabelsstr) r"r7r6r4r5colorsr'r3xs ```` r#setIntColorScalezGradientLegend.setIntColorScale%s_______vW]I^I^___  ! # #s6{{## ' 'AaV$A LLF1I & & & &  5 NNCKKCKK; < < < < < NNE(OA.q%/!2DQG H H H H Hr$c<||_|dS)zWDefines labels to appear next to the color scale. Accepts a dict of {text: value} pairsN)rr&)r"ls r#r<zGradientLegend.setLabels1s  r$c tj|||||}|dS||}|||}d}d}|jD]} |tj ddddtj j j tj j jzt| } t!|| }t!|| }d} |} |} |}|}|jddkr,| |jdz}||z d| zz }||jdz }n+| |jdz}||jdz}||zd| zz}|jddkr!||jdz}||jdz }n ||jdz}||jdz}||||||g|_||j||jtj tj|| z ||dz z | z tj|| z||dz z| z}|||j !d||j "d|||j tj tj||tj||}||||j#|d| zz}|}|}|jD]} ||j| ||z zz }|$tj |||dz z ||tj j j tj j jzt| |%dS)Nr rr)&rpaint getViewBoxsavesceneTransform setTransformrectr boundingRectrQRectFr AlignmentFlag AlignLeft AlignVCenterr=rwidthheightrightlefttopbottomrrbsetPenrsetBrushrQPointFdrawRectrsetStart setFinalStoprdrawTextrestore)r"poptwidgetviewtransrI labelWidth labelHeightkrU textPaddingxRxLyTyBx3x2x1y2y1txlhlwys r#rDzGradientLegend.paint6s4T1c6222   < F ##%% yy{{   7 7Av}Q1a88&):Q:[^d^g^u_C;CEHIJEKEKLLAZ33Jk188::66KK ZZ\\ YY[[ XXZZ [[]] ;q>A  dk!n$Bj1[=0Bdil"BBdk!n$Bdil"Bj1[=0B ;q>A  dk!n$Bdil"BBdk!n$Bdil"BR2b,  4:} N2 +R A -= -K L L N2 +R A -= -K L L   4 q"%%% ""1b))) 4=!!!} N2r " " N2r " "   4  !k/ !   N NAT[^r"u--A JJv}RRT2r::FIrzs******  vvvvv^vvvvvr$