tj\dZ ddlZddlZddlZddlZddlmZmZmZejZ ejdk(rddl ZejjZ ejdk(ZGdde ZdZed k(reyy) z A simple VTK widget for wxPython. Find wxPython info at http://wxPython.org Created by David Gobbi, December 2001 Based on vtkTkRenderWindget.py Updated to new wx namespace and some cleaning by Andrea Gavana, December 2006 N) vtkCellPicker vtkPropertyvtkRenderWindow __WXGTK__ __WXMSW__ceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d-d)Z*d*Z+d+Z,d,Z-y().wxVTKRenderWindowz A wxRenderWindow for wxPython. Use GetRenderWindow() to get the vtkRenderWindow. Create with the keyword stereo=1 in order to generate a stereo-capable window. c . d|_d|_d|_d|_d|_d|_t |_d|_t|_ |jjdddd|_ d|_ d|_d|_d|_d|_d|_d|_ t)|d}|d= |d }|d = |d }|d =t,j6t,j8z} ||d z}|d =g} |} | r$| j;| | j=} | r$| j?| D]} | jAdtBjDd k(r6tBjG||||||t,jHjJg ntBjG||||||tM|_'|jNjQ|jR|jT|r4|jNjW|jNjYd|_-|j]t,j^|j`|j]t,jbd|j]t,jd|jf|j]t,jh|jf|j]t,jj|jf|j]t,jl|jn|j]t,jp|jn|j]t,jr|jn|j]t,jt|jv|j]t,jx|jz|j]t,j||j~|j]t,j|j|j]t,j|j|j]t,j|j|j]t,j|j|j]t,j|j|j]t,j|j|j]t,j|jy#t*$rd}YkwxYw#t*$rt,j.}YwxYw#t*$r5 |j1}n#t2$rt,j4}YnwxYwYwxYw#t*$rYwxYw)zDefault class constructor. @param parent: parent window @param ID: window id @param **kw: wxPython keywords (position, size, style) plus the 'stereo' keyword N?rg-C6?stereoFpositionsizestyleGLCanvas)posrr attribList)rrrcyN)es L/DATA/.local/lib/python3.12/site-packages/vtkmodules/wx/wxVTKRenderWindow.pyz,wxVTKRenderWindow.__init__..sT)M_CurrentRenderer_CurrentCamera _CurrentZoom _CurrentLight_ViewportCenterX_ViewportCenterYr_Picker _PickedActorr_PickedPropertySetColor_PrePickedProperty_LastX_LastY_Mode _ActiveButton_wxVTKRenderWindow__OldFocus_DesiredUpdateRate_StillUpdateRateboolKeyErrorwxDefaultPositionGetSizeAttributeError DefaultSize WANTS_CHARSNO_FULL_REPAINT_ON_RESIZEappend GetParentreverseShow baseClass__name____init__glcanvasWX_GL_DOUBLEBUFFERr _RenderWindowSetSizewidthheightStereoCapableWindowOnSetStereoTypeToCrystalEyes_wxVTKRenderWindow__handleBind EVT_PAINTOnPaintEVT_ERASE_BACKGROUNDEVT_RIGHT_DOWN _OnButtonDown EVT_LEFT_DOWNEVT_MIDDLE_DOWN EVT_RIGHT_UP _OnButtonUp EVT_LEFT_UP EVT_MIDDLE_UP EVT_MOTIONOnMotionEVT_ENTER_WINDOW_OnEnterWindowEVT_LEAVE_WINDOW_OnLeaveWindowEVT_CHAROnChar OnKeyDown EVT_KEY_UPOnKeyUpEVT_SIZE_OnSizeEVT_MOVEOnMove EVT_SET_FOCUS OnSetFocusEVT_KILL_FOCUS OnKillFocus) selfparentIDargskwrrrrlps rr=zwxVTKRenderWindow.__init__msU!%"! ! !$  *} %%a!,"&   !#% &  "X,'F8  **~H: &f:D6 !=!== BwK'E7   HHQK A A FF1I    +   tVRXD%*+-;;+I+I*J  L   tVRXD%*  ,-. ""4::t{{;     4 4 6    9 9 ;  ",, - "))>: "##T%7%78 """D$6$67 "$$d&8&89 "//4#3#34 "..$"2"23 """D$4$45 "--/ "%%t':':; "%%t':':; "++t{{+ "++t~~. "--. "++t||, "++t{{+ """DOO4 "##T%5%56q F   *))H *  & &~~'! &~~ & &   sl&T8T&U+ V T#"T#&UU VU! V!U=:V<U==VV VVc||_yTMirrors the method with the same name in vtkRenderWindowInteractor. Nr,rfrates rSetDesiredUpdateRatez&wxVTKRenderWindow.SetDesiredUpdateRates #'rc|jSrorprfs rGetDesiredUpdateRatez&wxVTKRenderWindow.GetDesiredUpdateRates&&&rc||_yrnr-rqs rSetStillUpdateRatez$wxVTKRenderWindow.SetStillUpdateRates !%rc|jSruryrvs rGetStillUpdateRatez$wxVTKRenderWindow.GetStillUpdateRates$$$rcNtj|}|jy)z>Handles the wx.EVT_PAINT event for wxVTKRenderWindow. N)r0PaintDCRender)rfeventdcs rrIzwxVTKRenderWindow.OnPaint sZZ  rctjdk7r/|j\}}|jj |||j ||j y)z=Handles the wx.EVT_SIZE event for wxVTKRenderWindow. rN)r0Platformr2r@rAOnSizer)rfrrBrCs rr_zwxVTKRenderWindow._OnSizesJ ;;+ %!MMOME6    & &uf 5 E rcyOverridable event. Nrrfrs rrzwxVTKRenderWindow.OnSize rcyrrrs rrazwxVTKRenderWindow.OnMoverrcH|j||j|y)zMHandles the wx.EVT_ENTER_WINDOW event for wxVTKRenderWindow. N)UpdateRenderer OnEnterWindowrs rrVz wxVTKRenderWindow._OnEnterWindow$s  E" 5!rc|jdk(r4tjj|_|j yyr)r+r0Window FindFocusSetFocusrs rrzwxVTKRenderWindow.OnEnterWindow,s2 ??d " ii113DO MMO #rc&|j|y)zMHandles the wx.EVT_LEAVE_WINDOW event for wxVTKRenderWindow. N) OnLeaveWindowrs rrXz wxVTKRenderWindow._OnLeaveWindow3s 5!rc`|jr"|jjd|_yyr)r+rrs rrzwxVTKRenderWindow.OnLeaveWindow9s' ?? OO $ $ &"DO rcyrrrs rrczwxVTKRenderWindow.OnSetFocus@rrcyrrrs rrezwxVTKRenderWindow.OnKillFocusErrc<|jj|j|jrd}n(|j rd}n|j rd}nd}|r)|j s||_tr|j|j|y)zXHandles the wx.EVT_LEFT/RIGHT/MIDDLE_DOWN events for wxVTKRenderWindow. RightLeftMiddleN) r@rsr, RightDownLeftDown MiddleDownr* _useCapture CaptureMouse OnButtonDownrfrbuttons rrLzwxVTKRenderWindow._OnButtonDownJs //0G0GH ?? F ^^ F    FF $,,!'D !!# % rc |js|j||jr|j|y|j r|j |y|j r|j|yyr)r)rr OnLeftDownr OnRightDownr OnMiddleDownrs rrzwxVTKRenderWindow.OnButtonDownbsfzz    & >>  OOE " __    U #       e $ rc|js8|jrd|_y|jrd|_yd|_yy)rZoomPanRotateN)r) ControlDown ShiftDownrs rrzwxVTKRenderWindow.OnLeftDownps;zz  "# "" % rc,|jsd|_yy)rrNr)rs rrzwxVTKRenderWindow.OnRightDown{szzDJrc,|jsd|_yy)rrNrrs rrzwxVTKRenderWindow.OnMiddleDownszzDJrcV|jj|j|jrd}n(|j rd}n|j rd}nd}|j r,||j k(rtr|jd|_|j|y)zVHandles the wx.EVT_LEFT/RIGHT/MIDDLE_UP events for wxVTKRenderWindow. rrrN) r@rsr-RightUpLeftUpMiddleUpr*r ReleaseMouse OnButtonUprs rrPzwxVTKRenderWindow._OnButtonUps //0E0EF ==?F \\^F ^^ FF   &D,>,>">!!#!%D  rc,|jr|j|nC|jr|j|n!|j r|j ||j r|jr|jd|_yr) rOnLeftUpr OnRightUpr OnMiddleUpr)rrrs rrzwxVTKRenderWindow.OnButtonUpsf <<> MM% ]]_ NN5 ! ^^  OOE " ::$$  rcyrrrs rrzwxVTKRenderWindow.OnLeftUprrcyrrrs rrzwxVTKRenderWindow.OnRightUprrcyrrrs rrzwxVTKRenderWindow.OnMiddleUprrc|jdk(r|j|y|jdk(r|j|y|jdk(r|j|yy)rrrrN)r)rrrrs rrTzwxVTKRenderWindow.OnMotionsP ::  HHUO ZZ8 # KK  ZZ6 ! IIe "rcyrrrs rrZzwxVTKRenderWindow.OnCharrrc|jtdk(r|j||jtdk(r|j|jtdk(r|j |jtdk(r|j ||jdkr|j |yy)zBHandles the wx.EVT_KEY_DOWN events for wxVTKRenderWindow. rwsrlN) GetKeyCodeordReset WireframeSurface PickActorrZrs rr[zwxVTKRenderWindow.OnKeyDowns    S ) JJu     S ) NN     S ) LLN    S ) NN5 !     # KK  $rcyrrrs rr]zwxVTKRenderWindow.OnKeyUprrc|jS)z)Returns the current zoom factor. )rrvs r GetZoomFactorzwxVTKRenderWindow.GetZoomFactors   rc|jS)z5Returns the render window (vtkRenderWindow). )r@rvs rGetRenderWindowz!wxVTKRenderWindow.GetRenderWindows!!!rc|jS)z4Returns the current picker (vtkCellPicker). )r"rvs r GetPickerzwxVTKRenderWindow.GetPickers||rc|jr^|j}|j|jj|j |jj |j jr |jr|jr8|j|jk(r|jjy|jr||jjt|j|jj|j|_|jjyyy)z2Actually renders the VTK scene on screen. N)r SetPositionr GetPosition SetFocalPoint GetFocalPointGetUpdateRegionIsEmptyrF GetHandler@rSetNextWindowInfostr WindowRemap)rflights rrzwxVTKRenderWindow.Renders   &&E   d11==? @    3 3 A A C D##%--/4==}}$..2B!B""))+!""44S9I5JK""..0 $ 0 ""))+" 4Arcd|j}|j}|jj\}}|jj }|j }d|_|jtd|D]}|j} d\} } |dkDrt||dz z } |dkDr|t|z dz |dz z } | j\} } }}| | k\sf| |ksl| | k\sr| |ksx| |_t||| z zdz | z|_ t||| z zdz | z|_ |j j|_|j j!}|j|j|_n||_||_y)z UpdateRenderer will identify the renderer under the mouse and set up _CurrentRenderer, _CurrentCamera, and _CurrentLight. Nr)rrr g@)GetXGetYr@r2 GetRenderersGetNumberOfItemsr InitTraversalrange GetNextItemfloat GetViewportr r!GetActiveCamerar GetLightsrr'r()rfrxywindowXwindowY renderers numRenderersirenderervxvyvpxminvpyminvpxmaxvpymaxlightss rrz wxVTKRenderWindow.UpdateRenderers JJL JJL--557&&335  113 $!q&A ,,.HEBr! 1Xwqy)! eAh&q(7195,4,@,@,B )VF6&f vf v(0%(-gv (Fs(J)/)0%(-gv (Fs(J)/)0%&*&;&;&K&K&M#..88:$$&%+%7%7%9")',  rc|jS)z&Returns the current renderer. )rrvs rGetCurrentRendererz$wxVTKRenderWindow.GetCurrentRenderer/s$$$rc|jr|j}|j}|jj |j |z |jj ||jz |jj||_||_|jj|jyy)z$Rotates the scene (camera). N) rrrrAzimuthr' Elevationr(OrthogonalizeViewUpResetCameraClippingRanger)rfrrrs rrzwxVTKRenderWindow.Rotate4s   A A    ' ' a 8    ) )!dkk/ :    3 3 5DKDK  ! ! : : < KKM !rc|jr|j}|j}|j}|j}|j \}}}|j \} } } |j rH|j| | | d|j|j\} } }|j| |z |jz| |z|jz ||j|j\} } }}|j| | ||j|||d|j|j\} } }|j| |z |jz| |z|jz ||j|j\} } }}|j!| | |n|j \} } } |j| | | d|j|j}|d}|j"||jz z}|j$||jz z }|j||||j|j\}}}}|dk7r||z }||z }||z }|j| |z | z| |z | z| |z | z|j!| |z |z| |z |z| |z |z||_ ||_ |j'yy)z!Pans the scene (camera). r N)rrrrrrGetParallelProjection SetWorldPointWorldToDisplayGetDisplayPointSetDisplayPointr'r(DisplayToWorld GetWorldPointrrr r!r)rfrrrrcamerapPoint0pPoint1pPoint2fPoint0fPoint1fPoint2fxfyfzfwdPoint focalDepthaPoint0aPoint1rPoint0rPoint1rPoint2rPoint3s rrzwxVTKRenderWindow.PanEs   A A,,H((F(.(:(:(< %WWW(.(<(<(> %WWW++-&&wwwsC'')#3352b((Adkk)9)+Adkk)9)+-'')&446 2b$$R2.&&wwwsC'')#3352b((Adkk)9)+Adkk)9)+-'')&446 2b""2b,-3,@,@,B)&&wwwsC'')!113#AY //1t{{?C//1t{{?C((D'')4<4J4J4L1sN%goG%goG%goG$$g&77%B&-&77%B&-&77%BD""Gg$5#@$+g$5#@$+g$5#@BDKDK KKM} !rc|jr|j}|j}|j}|j}t j dd|j |z z}|j|z|_|jr%|j|z }|j|n!|j||j||_ ||_|jyy)z"Zooms the scene (camera). gRQ?g?N)rrrrmathpowr(rrGetParallelScaleSetParallelScaleDollyrr'r)rfrrrrr  zoomFactor parallelScales rrzwxVTKRenderWindow.Zooms   A A,,H((F$T[[1_(=?J $ 1 1J >D ++- & 7 7 9* D '' 6 Z(113DKDK KKM) !rNcp|jr|jj|jy)zResets the camera. N)r ResetCamerarrs rrzwxVTKRenderWindow.Resets(   ! ! - - / rc|jj}|j}|jt d|D]0}|j }|j j2|jy)z$ rc|jj}|j}|jt d|D]0}|j }|j j2|jy)z:Sets the current actor representation as surface. rN) rr'rrrrr(SetRepresentationToSurfacerr*s rrzwxVTKRenderWindow.Surfacesq&&002++- q#A&&(E     : : <$ rc|jrt|j}|j}|j}|j}|jj \}}|j |||z dz d||j}|jdk7r`|jdk7rQ|jj|j|jj|jd|_ |dk7rp||_|jj|_ |jj|j|jj|j|jyy)zPicks an actor. r rN)rrrr"r@r2PickGetActorr#r& SetProperty UnRegisterr(Registerr$r) rfrrrrpickerrrr-s rrzwxVTKRenderWindow.PickActors6   A A,,H\\F#1199; GW KK7Q;?C 9OO%E!!T)''4/!!--d.E.EF''2243J3JK*.' $)!*.*;*;*G*G*I'''001H1HI!!--d.B.BC KKM3 !rr).r< __module__ __qualname____doc__r=rsrwrzr|rIr_rrarVrrXrrcrerLrrrrrPrrrrrTrZr[r]rrrrrrrrrrrrrrrrr r esC7J' ' % %    "" #  !0 % &  0"     ! "  ,.%N% "AF2  rr cddlm}ddlm}m}m}ddl}ddl}tjd}tjdddd }t|d}|}|jj||} | jd |} | j| j!|} | j#| |j%| |j'|j)y) z)Like it says, just a simple example. r) vtkConeSource)vtkActorvtkPolyDataMapper vtkRendererNFr )r@)r)vtkmodules.vtkFiltersSourcesr;vtkmodules.vtkRenderingCorer<r=r>vtkmodules.vtkRenderingOpenGL2vtkmodules.vtkInteractionStyler0AppFramer r AddRenderer SetResolutionSetInputConnection GetOutputPort SetMapperAddActorr:MainLoop) r;r<r=r> vtkmodulesappframewidgetrencone coneMapper coneActors rwxVTKRenderWindowConeExamplerWs;TT)) &&-C HHT22 CE ub )F -C ((- ?Dq"$J!!$"4"4"67 I  #LL JJLLLNr__main__)r9rossysr0rCrrrrr;r wx.glcanvasr>rrr rWr<rrrr\s >B SS II ;;+ $$I {{k) x  x x#J z "r