e_ddlddlmcmZddlmZmZmZm Z ddl Z ddl m Z ddl mZddl mZddlmZmZmZGd d ZGd d eej,Zy) )*N)cosradianssintan)Vector) functions)getConfigOption)QtCoreQtGui QtWidgetsceZdZddfd ZdZdZdZdZdZd Z d Z d Z d Z d&d Z d&dZdZdZd&dZd'dZd(dZd)dZdZdZdZdZd*dZdZdZdZdZdZdZd Z d!Z!d"Z"e#e$d#d$fd%Z%xZ&S)+ GLViewMixineuler)rotationMethodc "t||i||dvr tdtdddt j ddddddddd |d |_|jg|_tjjjtjjjtjjjtjjjtjjj tjjj"g|_i|_tj(|_|j*j,j/|j0y ) a Mixin class providing functionality for GLViewWidget ================ ============================================================== **Arguments:** rotationMethod (str): Mechanism to drive the rotation method, options are 'euler' and 'quaternion'. Defaults to 'euler'. ================ ============================================================== )r quaternionz8Rotation method should be either 'euler' or 'quaternion'r$@<-N)centerrotationdistancefov elevationazimuthviewportr)super__init__ ValueErrorr r QQuaternionoptsresetitemsr QtKey Key_RightKey_LeftKey_UpKey_Down Key_PageUp Key_PageDown noRepeatKeys keysPressedQTimerkeyTimertimeoutconnect evalKeyState)selfrargskwargs __class__s ?/usr/lib/python3/dist-packages/pyqtgraph/opengl/GLViewWidget.pyr"zGLViewMixin.__init__ s` $)&) !8 8WX XQqm**1Qq3,     #YY]]44fiimm6L6LfiimmNbNbdjdmdmdqdqdzdz}C}F}F}J}J}U}UW]W`W`WdWdWqWqr   %%d&7&78cZ|j}t|j|zSN)devicePixelRatioFintwidthr7dprs r; deviceWidthzGLViewMixin.deviceWidth0s&$$&4::<#%&&r<cZ|j}t|j|zSr>)r?r@heightrBs r; deviceHeightzGLViewMixin.deviceHeight4s&$$&4;;=3&''r<ctddd|jd<d|jd<d|jd<d|jd<d |jd <d |jd <|jtd y )z_ Initialize the widget state or reset the current state to the original state. rrrrrrrrrrNr background)r r%setBackgroundColorr r7s r;r&zGLViewMixin.reset8ss%Qqm ( $ * %!# +! ) $ *  =>r<c|jj||jr|j|j ||j yr>)r'appendisValid initialize_setViewupdater7items r;addItemzGLViewMixin.addItemEs< $ <<> OO  d r<c||jj||jd|jy)z1 Remove the item from the scene. N)r'removerPrQrRs r; removeItemzGLViewMixin.removeItemNs+ $ d r<cv|jD]}|jdg|_|jy)z2 Remove all items from the scene. N)r'rPrQrRs r;clearzGLViewMixin.clearVs3JJ D MM$    r<c(|j}|j}|js|jdkrt t }t d|z|jD]#}|jr|j%y)zN Initialize items that were not initialized during addItem(). )rrz;pyqtgraph.opengl: Requires >= OpenGL 2.0 (not ES); Found %sN) contextformat isOpenGLESversion glGetString GL_VERSION RuntimeErrorr' isInitializedrO)r7ctxfmt verStringrSs r; initializeGLzGLViewMixin.initializeGL_s|llnjjl >> s{{}v5#J/IMPYY JJ "D%%'! "r<ctj|i|j|jd<|j y)z{ Set the background color of the widget. Accepts the same arguments as :func:`~pyqtgraph.mkColor`. bgcolorN)fnmkColorgetRgbFr%rQ)r7r8kwdss r;rJzGLViewMixin.setBackgroundColoros2 "zz4848@@B ) r<cl|jd}|"dd|j|jfS|S)Nr r)r%rDrG)r7vps r; getViewportzGLViewMixin.getViewportws; YYz " :q$**,d.?.?.AB BIr<c|j|}tttt j |j tjyNdtype)projectionMatrix glMatrixMode GL_PROJECTION glLoadMatrixfnparraydatafloat32)r7regionms r; setProjectionzGLViewMixin.setProjection~s7  ! !& )]#bhhqvvxrzz:;r<c|"dd|j|jf}|j\}}}}|jd}|jd}|dz}|dz} |t dt |zz} | |z|z } | |d|z d|z zdz z} | |d|d z|z d|z zdz z} | |d|z d|z zdz z}| |d|d z|z d|z zdz z}t j}|j| | |||| |S) NrrrgMbP?g@@?@rr) rDrGror%rrr QMatrix4x4frustum)r7r|x0y0whdistrnearClipfarCliprtleftrightbottomtoptrs r;rtzGLViewMixin.projectionMatrixsO >D,,.0A0A0CDF'') B1yy$ii%<, s3-. . EAIfQils1u-12fQiq )",Q7!;<vay|A.23vay*2-#a%81<=     4Xw? r<c|j}tttt j |j tjyrq) viewMatrixru GL_MODELVIEWrwrxryrzr{)r7r}s r; setModelviewzGLViewMixin.setModelviews3 OO \"bhhqvvxrzz:;r<ctj}|jdd|jd |jddk(r|j |jdnH|j |jddz dd d |j |jd dzd d d |jd }|j|j |j |j |S) NgrrrrrZrrrr)r r translater%rotatexyz)r7rrs r;rzGLViewMixin.viewMatrixs     c3:!6 67 99% &, 6 IIdii + , IIdii ,R/Aq 9 IIdii *2-q!R 88$ fhhj[688:+ {; r<c|d|j|d|dzz |d|df}td} ttt t di|_|j|dtt}|Dcgc]}|j|jdf}}|jd |Dcgc]}|j |dc}S#tt}wxYwcc}wcc}w) zy Return a list of the items displayed in the region (x, y, w, h) relative to the widget. rrrriT)r| useItemNamesc |dS)Nr)is r;z%GLViewMixin.itemsAt..s 1r<key) rGglSelectBuffer glRenderMode GL_SELECT glInitNames glPushName _itemNamespaintGL GL_RENDERnearnamessort)r7r|bufhitsrr'rs r;itemsAtzGLViewMixin.itemsAts )T..0&)F1I2EFq SYZ[S\]V$ +  # M qM DO LLTL : *D/34!!&&!''!*%44 ~ &/45!!%55  *D45s>C"C2C7C/c|t|jnt||j||j|jd}t |t ttz|j|y)a viewport specifies the arguments to glViewport. If None, then we use self.opts['viewport'] region specifies the sub-region of self.opts['viewport'] that should be rendered. Note that we may use viewport != self.opts['viewport'] when exporting. N)r|rhr) glViewportror~rr% glClearColorglClearGL_DEPTH_BUFFER_BITGL_COLOR_BUFFER_BIT drawItemTree)r7r|r rrhs r;rzGLViewMixin.paintGLsv   ((* +  ! &) ))I&g$'::< |4r<c|+|jDcgc]}|j|}}n!|j}|j||j d|D]}|j s||ur[ t t|r.t|j||j|j<|jt#tt%t&t) |j+}t-t/j0|j3t.j4|j7||t%t&t9ycc}w#ddl m }|jtdt!|zYxYw#t#wxYw#t%t&t9wxYw)Nc"|jSr>) depthValue)as r;rz*GLViewMixin.drawItemTree..s r<rr)debugzError while drawing item %s.rrr)r' parentItem childItemsrMrvisible glPushAttribGL_ALL_ATTRIB_BITS glLoadName_idrpaintrprintExcprintstr glPopAttribrur glPushMatrix transform glMultMatrixfrxryrzr{r glPopMatrix)r7rSrrr'rrrs r;rzGLViewMixin.drawItemTreesP < $ E1alln.DQEEEOO%E LL  / 0 "A99;Dy " !34#"155)12.GGI M\*"B!"((2779BJJ"GH%%al%C .M5 " FF(NN$83t9DE M!.Ms0E.E.>AE32AF63/F$"F'' F36Gc|| td| td|||jd<|||jd<|jddk(r||v|jdj}||j| dz ||j |dz t j j||jd<|||jd<nu|||jd <|||jd <|Q|j}|jdz|jd <|j dz |jd <|jy) Nz&cannot set both rotation and elevationz$cannot set both rotation and azimuthrrrrrrrr) r#r% toEulerAnglessetZsetXr r$fromEulerAnglesrrrQ)r7posrrrreus r;setCameraPositionzGLViewMixin.setCameraPositionsa  $ !IJJ" !GHH ?"%DIIh   $,DIIj ! 99% &, 6$(;YYz*88:&GGWHRK((GGIbL)(-(9(9(I(I"(M *%#(0 *%$)2 +&"'. )$#++-)+" +&(*w| )$ r<c ,|jd}|jd}|jddk(r7t||jdjtdd|z }|St|jd}t|jd}t|j |t |zt |zz|j |t |zt|zz|j|t|zz}|S) zOReturn current position of camera based on center, dist, elevation, and azimuthrrrrrrrr) r%r rotatedVectorrrrrrr)r7rrrelevazims r;cameraPositionzGLViewMixin.cameraPositions8$yy$ 99% &, 6$))J"7"E"EfQqQUFV"XXYC 499[12D499Y/0D TCI-D 99 TCI-D 99 TCI--C  r<c :hd}|j|std||j|jd|jd|jd|jd|jdd |vr|d |jd <yy) N>rrrrrrzvalid keywords are rrrrr)rrrrrr) issupersetr#rgetr%)r7rl valid_keyss r;setCameraParamszGLViewMixin.setCameraParams-sV $$T*2:,?@ @ 488H#5@T)-+)>QZH[(,(<  > D=#E{DIIe  r<cPhd}|Dcic]}||j|c}Scc}w)N>rrrrrr)r%)r7rks r; cameraParamszGLViewMixin.cameraParams8s(V +57aTYYq\!777s#cf|jddk(rDtjj|| d}||jdz}||jd<nK|jdxx|z cc<t j |jd|zdd|jd<|j y ) zUOrbits the camera around the center position. *azim* and *elev* are given in degrees.rrrrrrgVgV@N)r%r r$rri clip_scalarrQ)r7rrqs r;orbitzGLViewMixin.orbit<s 99% &, 6!!114%A :& &A$%DIIj ! IIi D ( %'^^DIIk4JT4QSWY\%]DIIk " r<c|dk(r.|jdxxtj|||z cc<n3|dk(r|j}|jd|z }|j }|dzt dt |jdzz}||jz } tjddd} tjj| |j} tjj| | j} |jd| | z|zz| | z|zz| | z|zz|jd<n|d k(r|jd d k(r|jd j} | jtddd}| jtddd}| jtddd}|j|jd}|jdxx||| z||zz||zzzz cc<n?t |jd }t |jd}t |jd}|jd|jz j }t |dz dz}||z|jz }|t|z|z}|t||zt|t|z|zz z}|t||zt|t|z|zzz}|jdxxtj|| |z cc<n t!d|j#y)a Moves the center (look-at) position while holding the camera in place. ============== ======================================================= **Arguments:** *dx* Distance to pan in x direction *dy* Distance to pan in y direction *dz* Distance to pan in z direction *relative* String that determines the direction of dx,dy,dz. If "global", then the global coordinate system is used. If "view", then the z axis is aligned with the view direction, and x and y axes are in the plane of the view: +x points right, +y points up. If "view-upright", then x is in the global xy plane and points to the right side of the view, y is in the global xy plane and orthogonal to x, and z points in the global z direction. ============== ======================================================= Distances are scaled roughly such that a value of 1.0 moves by one pixel on screen. globalr view-uprightrrrrrviewrrrrrrz7relative argument must be global, view, or view-uprightN)r%r QVector3DrlengthrrrA crossProduct normalized conjugatedrr pixelSizerrr#rQ)r7dxdydzrelativecPoscVecrxDistxScalezVecxVecyVecqcxvyvzv scale_factorrrr fov_factorrrrs r;panzGLViewMixin.panIs. x  IIh 5??2r2#> >   '&&(D99X&-D;;=D2IC'$))E2B*C$C DDETZZ\)F??1Qq)D??//d;FFHD??//d;FFHD"&))H"5v 8J"JTTZ]]_M_"_bfiobortbt"tDIIh   yy)*l:YYz*557%%va!}6%%va!}6%%va!}6#~~tyy/BD  (#|r2#v2~27M'NN#tyy56tyy34dii./ (+d.A.A.CCKKM q\A- #j04::<?  3t9,r1 CINSYT5JR5O$OP CINSYT5JR5O$OP (#uq1"a'@@#VW W r<c|j}t|tjrStj|j d|j dz zdz}||z dzjddz}n||z j}|dztdt|jd zz}||jz S) z Return the approximate size of a screen pixel at the location pos Pos may be a Vector or an (N,3) array of locations )rr)rrr)axisrrr) r isinstancerxndarrayryreshapendimsumrrrr%rA)r7rcamrrs r;rzGLViewMixin.pixelSizes !!# c2:: &((3-''chhqj(9$(>?CWqL%%2%.3DG##%Dr Cgdii.>&? ?@@tzz|##r<cjt|dr|jn|j}||_y)Nposition)hasattrrlocalPosmousePos)r7evlposs r;mousePressEventzGLViewMixin.mousePressEvents% 'J 7r{{}R[[] r<ct|dr|jn|j}t|ds||_||jz }||_|j t j jjk(r|jt j jjzr2|j|j|jddy|j|j |jy|j t j jj k(r|jt j jjzr2|j|jd|jdy|j|j|jddyy)Nrrrr)rr)rrrrbuttonsr r( MouseButton LeftButton modifiersKeyboardModifierControlModifierrrrr MiddleButton)r7rrdiffs r;mouseMoveEventzGLViewMixin.mouseMoveEvents> 'J 7r{{}R[[]tZ( DMdmm# ::<69900;; ; !;!;!K!KK4668Q@ DFFH9dffh/ ZZ\VYY22?? ? !;!;!K!KK1dffhH4668QH @r<cyr>rr7rs r;mouseReleaseEventzGLViewMixin.mouseReleaseEvents r<cz|jj}|dk(r|jj}|jtj j jzr|jdxxd|zzcc<n|jdxxd|zzcc<|jy)Nrrg+?r) angleDeltarrrr r(rrr%rQ)r7rdeltas r; wheelEventzGLViewMixin.wheelEvents !!# A:MMO%%'E LLNVYY77GG G IIe u ,  IIj !UE\ 1 ! r<c|j|jvrO|j|jryd|j|j<|j yy)Nr)rr0accept isAutoRepeatr1r6r#s r; keyPressEventzGLViewMixin.keyPressEventsS 668t(( ( IIK )*D  RVVX &     )r<c|j|jvrN|j|jry |j|j=|j yy#t $r i|_Y$wxYwr>)rr0r*r+r1KeyErrorr6r#s r;keyReleaseEventzGLViewMixin.keyReleaseEventsq 668t(( ( IIK  &$$RVVX.     )  &#%  &sA,,A?>A?cJd}t|jdkDrn|jD]]}|tjjj k(r|j | dn|tjjjk(r|j |dn|tjjjk(r|j d| n|tjjjk(r|j d|nP|tjjjk(rn(|tjjjk(r |jjd`y|jjy)Nrr)rr)lenr1r r(r)r*rr+r,r-r.r/r3startstop)r7speedrs r;r6zGLViewMixin.evalKeyStates! t 1 $'' (&))--111JJUFJ3FIIMM222JJEJ2FIIMM000JJAUFJ3FIIMM222JJAEJ2FIIMM444FIIMM666 ##B' ( MM   r<c"|jS)zA Read the current buffer pixels out as a QImage. )grabFramebufferrKs r; readQImagezGLViewMixin.readQImages##%%r<ic tt|\}}|jd}d} d} tj||dftj } t jd} t jt j| tttd}tt||} tj| | dftj } tt dt"| | dt"t$d t't dt(dk(rt+d| j,ddzttdt"| | dt"t$| t/d} t1t2| t5t2t6| | t9tt:t2| dd||f|j<d<d|z}t?| ||z | |z D].}t?| ||z | |z D]}tA|| z||z}tA|| z||z}||z }||z }t jBt jt jDt|d|jG|||z |z ||fdd||ftt|tItd||} tjJ| tj jM| | dddd d f} | ||z | |||z f| ||z||z ||z||z f<1 d|j<d<t jt jdttd| tO|g| t jP| g| tSd| g| S#d|j<d<t jt jdttd| tO|g| t jP| g| tSd| gwwxYw) NrrrrzHOpenGL failed to create 2D texture (%dx%d); too large for this hardware.rr )r|r r.)*mapr@ makeCurrentrxemptyubyteglfboglGenFramebuffersglBindFramebufferGL_FRAMEBUFFERglEnable GL_TEXTURE_2D glGenTextures glBindTexturezeros glTexImage2DGL_PROXY_TEXTURE_2DGL_RGBAGL_UNSIGNED_BYTEglGetTexLevelParameterivGL_TEXTURE_WIDTHrashapeglGenRenderbuffersglBindRenderbufferGL_RENDERBUFFERglRenderbufferStorageGL_DEPTH_COMPONENTglFramebufferRenderbufferGL_DEPTH_ATTACHMENTr%rangeminglFramebufferTexture2DGL_COLOR_ATTACHMENT0r glGetTexImage frombufferr glDeleteTexturesglDeleteFramebuffersglDeleteRenderbuffers)r7sizer\type textureSizepaddingrrtexfb depth_bufoutputtexwidthrzp2rrx2y2w2h2s r; renderToArrayzGLViewMixin.renderToArrays#tn!   6 6XXq!Qirxx8F((+B  # #E$8$8" > ] #"C - -"H88Xhq1BD ,a(HaQXZjlp q'(;Q@PQUVV"#mptpzpz{}|}p~#~ 7Hh7Tdfj k+1-I  : !/3ExQY Z %n6I?\e f%&1aLDIIj !WBG8QwY < zx7HRK@zAQxZ73BQxZ73BABAB001E1EuGaGacpruwxyLLAaCFB(;q!RQSnLU!-5)64HD==RXX>FFxPXYZ[\`^`\`be\efDIMPRSZPZm]d\dNdfmnpqxnxfxNxIyF1W9RZ/72g:1EEFz z&%)DIIj !  # #E$8$8! < - + #'~**B40$%a)5 %)DIIj !  # #E$8$8! < - + #'~**B40$%a)5%s KM==A;O8r>)NNF)NF)NNNNN)r)'__name__ __module__ __qualname__r"rDrGr&rTrWrYrfrJror~rtrrrrrrrrrrrrrr!r$r(r,r/r6r8GL_BGRArLrn __classcell__r:s@r;rr s-4!9F'( ?" < .< 6.5"!"F#J" +8 ?B $I$    !(& *17GUYcf?r<rc$eZdZddfd ZxZS) GLViewWidgetN)devicePixelRatioct||i||jtjj j y)a Basic widget for displaying 3D data - Rotation/scale controls - Axis/grid display - Export options ================ ============================================================== **Arguments:** parent (QObject, optional): Parent QObject. Defaults to None. devicePixelRatio No longer in use. High-DPI displays should automatically detect the correct resolution. rotationMethod (str): Mechanism to drive the rotation method, options are 'euler' and 'quaternion'. Defaults to 'euler'. ================ ============================================================== N)r!r"setFocusPolicyr r( FocusPolicy ClickFocus)r7rwr8r9r:s r;r"zGLViewWidget.__init__5s5 $)&) FII11<<=r<)rorprqr"rsrts@r;rvrv4s/3>>r<rv) OpenGL.GLOpenGL.GL.framebufferobjectsGLframebufferobjectsr@mathrrrrnumpyrxrr r rir r(r r rr QOpenGLWidgetrvrr<r;rsG,,''))eeP>; 7 7>r<