+j\dZ ddlZddlZddlZddlZddlmZmZmZej Z ej dkrddl Zej jZ ej dkZGdde ZdZed kr edSdS) 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-d(S).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|_ |j dddd|_ d|_ d|_d|_d|_d|_d|_d|_ t)|d}|d=n#t*$rd}YnwxYw |d }|d =n#t*$rt,j}YnwxYw |d }|d =nB#t*$r5 |}n#t2$rt,j}YnwxYwYnwxYwt,jt,jz} ||d z}|d =n#t*$rYnwxYwg} |} | r+| | | } | +| | D]} | dtBj"d kr2tB#||||||t,j$j%g n tB#||||||tM|_'|j'(|j)|j*|r2|j'+|j',d|_-|.t,j/|j0|.t,j1d|.t,j2|j3|.t,j4|j3|.t,j5|j3|.t,j6|j7|.t,j8|j7|.t,j9|j7|.t,j:|j;|.t,j<|j=|.t,j>|j?|.t,j@|jA|.t,j@|jB|.t,jC|jD|.t,jE|jF|.t,jG|jH|.t,jI|jJ|.t,jK|jLdS)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)rrrcdSN)es Y/DATA/AppData/hermes/venv/lib/python3.11/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__ms!%"! ! !$  *}} %%a!,,,"&   !#% &  "X,''F8    FFF  **~H: * * *)HHH * &f:D6  & & & &~~''! & & &~ & &!== BwK'E7     D     HHQKKK A    A FF1IIII   + +   tVRXD%*+-;+I*J  L L L L   tVRXD%*  , , ,-.. ""4:t{;;;  <   4 4 6 6 6   9 9 ; ; ;  ", --- ")>>::: "#T%7888 ""D$6777 "$d&8999 "/4#3444 ".$"2333 ""D$4555 "-/// "%t':;;; "%t':;;; "+t{+++ "+t~... "-... "+t|,,, "+t{+++ ""DO444 "#T%566666sl-C CC C%%C>=C> D E D.-E .EE EE  E *E99 FFc||_dSTMirrors the method with the same name in vtkRenderWindowInteractor. Nr,rfrates rSetDesiredUpdateRatez&wxVTKRenderWindow.SetDesiredUpdateRates#'rc|jSrorprfs rGetDesiredUpdateRatez&wxVTKRenderWindow.GetDesiredUpdateRates &&rc||_dSrnr-rqs rSetStillUpdateRatez$wxVTKRenderWindow.SetStillUpdateRates!%rc|jSruryrvs rGetStillUpdateRatez$wxVTKRenderWindow.GetStillUpdateRates $$rcVtj|}|dS)z>Handles the wx.EVT_PAINT event for wxVTKRenderWindow. N)r0PaintDCRender)rfeventdcs rrIzwxVTKRenderWindow.OnPaint s$Z   rctjdkr2|\}}|j|||||dS)z=Handles the wx.EVT_SIZE event for wxVTKRenderWindow. rN)r0Platformr2r@rAOnSizer)rfrrBrCs rr_zwxVTKRenderWindow._OnSizes^ ;+ % %!MMOOME6   & &uf 5 5 5 E rcdSOverridable event. Nrrfrs rrzwxVTKRenderWindow.OnSize  rcdSrrrs rrazwxVTKRenderWindow.OnMoverrcZ||||dS)zMHandles the wx.EVT_ENTER_WINDOW event for wxVTKRenderWindow. N)UpdateRenderer OnEnterWindowrs rrVz wxVTKRenderWindow._OnEnterWindow$s2 E""" 5!!!!!rc|jdkr9tj|_|dSdSr)r+r0Window FindFocusSetFocusrs rrzwxVTKRenderWindow.OnEnterWindow,s? ?d " " i1133DO MMOOOOO # "rc0||dS)zMHandles the wx.EVT_LEAVE_WINDOW event for wxVTKRenderWindow. N) OnLeaveWindowrs rrXz wxVTKRenderWindow._OnLeaveWindow3s 5!!!!!rcX|jr"|jd|_dSdSr)r+rrs rrzwxVTKRenderWindow.OnLeaveWindow9s7 ? # O $ $ & & &"DOOO # #rcdSrrrs rrczwxVTKRenderWindow.OnSetFocus@rrcdSrrrs rrezwxVTKRenderWindow.OnKillFocusErrcR|j|j|rd}n0|rd}n|rd}nd}|r)|js"||_tr|| |dS)zXHandles the wx.EVT_LEFT/RIGHT/MIDDLE_DOWN events for wxVTKRenderWindow. RightLeftMiddleN) r@rsr, RightDownLeftDown MiddleDownr* _useCapture CaptureMouse OnButtonDownrfrbuttons rrLzwxVTKRenderWindow._OnButtonDownJs //0GHHH ??   FF ^^   FF      FFF  $$, $!'D  $!!### %     rc@|js|||r||dS|r||dS|r||dSdSr)r)rr OnLeftDownr OnRightDownr OnMiddleDownrs rrzwxVTKRenderWindow.OnButtonDownbsz '    & & & >>   % OOE " " " " " __   %   U # # # # #      %   e $ $ $ $ $ % %rc|jsC|r d|_dS|r d|_dSd|_dSdS)rZoomPanRotateN)r) ControlDown ShiftDownrs rrzwxVTKRenderWindow.OnLeftDownps_z &  "" &# "" &" %  & &rc&|js d|_dSdS)rrNr)rs rrzwxVTKRenderWindow.OnRightDown{s#z DJJJ  rc&|js d|_dSdS)rrNrrs rrzwxVTKRenderWindow.OnMiddleDowns#z DJJJ  rcd|j|j|rd}n0|rd}n|rd}nd}|jr-||jkr"tr|d|_| |dS)zVHandles the wx.EVT_LEFT/RIGHT/MIDDLE_UP events for wxVTKRenderWindow. rrrN) r@rsr-RightUpLeftUpMiddleUpr*r ReleaseMouse OnButtonUprs rrPzwxVTKRenderWindow._OnButtonUps //0EFFF ==?? FF \\^^ FF ^^   FFF   &&D,>">"> $!!###!%D  rcR|r||nS|r||n)|r|||jr|jr|d|_dSr) rOnLeftUpr OnRightUpr OnMiddleUpr)rrrs rrzwxVTKRenderWindow.OnButtonUps <<>> # MM% ]]__ # NN5 ! ! ! ! ^^   # OOE " " " : $   rcdSrrrs rrzwxVTKRenderWindow.OnLeftUprrcdSrrrs rrzwxVTKRenderWindow.OnRightUprrcdSrrrs rrzwxVTKRenderWindow.OnMiddleUprrc|jdkr||dS|jdkr||dS|jdkr||dSdS)rrrrN)r)rrrrs rrTzwxVTKRenderWindow.OnMotionsv :   HHUOOOOO Z8 # # KK      Z6 ! ! IIe     " !rcdSrrrs rrZzwxVTKRenderWindow.OnCharrrc0|tdkr|||tdkr||tdkr||tdkr|||dkr||dSdS)zBHandles the wx.EVT_KEY_DOWN events for wxVTKRenderWindow. rwsrlN) GetKeyCodeordReset WireframeSurface PickActorrZrs rr[zwxVTKRenderWindow.OnKeyDowns     S ) ) JJu        S ) ) NN        S ) ) LLNNN     S ) ) NN5 ! ! !      # # KK      $ #rcdSrrrs 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||j|r|jr|jr8|j| kr|j dS| r|j t| |j | |_|j dSdSdS)z2Actually renders the VTK scene on screen. N)r SetPositionr GetPosition SetFocalPoint GetFocalPointGetUpdateRegionIsEmptyrF GetHandler@rSetNextWindowInfostr WindowRemap)rflights rrzwxVTKRenderWindow.RendersT   E&E   d1==?? @ @ @    3 A A C C D D D##%%--// ,4= ,} ,$..2B2B!B!B"))+++++!! ,"44S9I9I5J5JKKK"..000 $ 0 0 "))+++++ , , , ,rc|}|}|j\}}|j}|}d|_|td|D]6}| } d\} } |dkrt||dz z } |dkr|t|z dz |dz z } | \} } }}| | kr| |kr| | kr| |kr| |_t||| z zdz | z|_ t||| z zdz | z|_ |j|_|j}|| |_n8||_||_dS)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 JJLL JJLL-5577&3355  1133 $!!!q&&  A ,,..HEBr! 1XXwqy)! eAhh&q(7195,4,@,@,B,B )VF6&f vf v(0%(-gv (Fs(J)/)0%(-gv (Fs(J)/)0%&*&;&K&K&M&M#.88::$$&&&%+%7%7%9%9"  rc|jS)z&Returns the current renderer. )rrvs rGetCurrentRendererz$wxVTKRenderWindow.GetCurrentRenderer/s $$rc|jr|}|}|j|j|z |j||jz |j||_||_|j | dSdS)z$Rotates the scene (camera). N) rrrrAzimuthr' Elevationr(OrthogonalizeViewUpResetCameraClippingRanger)rfrrrs rrzwxVTKRenderWindow.Rotate4s   A A   ' ' a 8 8 8   ) )!dk/ : : :   3 3 5 5 5DKDK  ! : : < < < KKMMMMM  rc|jr:|}|}|j}|j}|\}}}|\} } } |rl|| | | d|| \} } }| | |z |j z| |z|j z || |\} } }}|| | |||||d|| \} } }| | |z |j z| |z|j z || |\} } }}|| | |n/|\} } } || | | d|| }|d}|j||j z z}|j||j z z }| |||| |\}}}}|dkr||z }||z }||z }|| |z | z| |z | z| |z | z|| |z |z| |z |z| |z |z||_ ||_ |dSdS)z!Pans the scene (camera). r N)rrrrrrGetParallelProjection SetWorldPointWorldToDisplayGetDisplayPointSetDisplayPointr'r(DisplayToWorld GetWorldPointrrr r!r)rfrrrrcamerapPoint0pPoint1pPoint2fPoint0fPoint1fPoint2fxfyfzfwdPoint focalDepthaPoint0aPoint1rPoint0rPoint1rPoint2rPoint3s rrzwxVTKRenderWindow.PanEs  >  A A,H(F(.(:(:(<(< %WWW(.(<(<(>(> %WWW++--0 B&&wwwsCCC'')))#33552b((Adk)9)+Adk)9)+---'')))&4466 2b$$R2...&&wwwsCCC'')))#33552b((Adk)9)+Adk)9)+---'')))&4466 2b""2b,,,,-3,@,@,B,B)&&wwwsCCC'')))!1133#AY /1t{?C/1t{?C((DDD'')))4<4J4J4L4L1sNN%goG%goG%goG$$g&77%B&-&77%B&-&77%BDDD""Gg$5#@$+g$5#@$+g$5#@BBBDKDK KKMMMMM}> > rc|jr|}|}|j}|j}t jdd|j|z z}|j|z|_|r-| |z }| |n)| || ||_ ||_|dSdS)z"Zooms the scene (camera). gRQ?g?N)rrrrmathpowr(rrGetParallelScaleSetParallelScaleDollyrr'r)rfrrrrr  zoomFactor parallelScales rrzwxVTKRenderWindow.Zooms    A A,H(F$T[1_(=??J $ 1J >D ++-- 4 & 7 7 9 9* D '' 6666 Z(((11333DKDK KKMMMMM)  rNcn|jr|j|dS)zResets the camera. N)r ResetCamerarrs rrzwxVTKRenderWindow.Resets5   0  ! - - / / / rcJ|j}|}|t d|D]<}|}|=|dS)z > > > rcJ|j}|}|t d|D]<}|}|=|dS)z:Sets the current actor representation as surface. rN) rr'rrrrr(SetRepresentationToSurfacerr*s rrzwxVTKRenderWindow.Surfaces&0022++-- q## = =A&&((E      : : < < < < rc|jr^|}|}|j}|j}|j\}}||||z dz d||}|jdkrP|j dkrE|j |j |j |j d|_ |dkrc||_|j |_ |j |j |j |j|dSdS)zPicks an actor. r rN)rrrr"r@r2PickGetActorr#r& SetProperty UnRegisterr(Registerr$r) rfrrrrpickerrrr-s rrzwxVTKRenderWindow.PickActorsO    A A,H\F#199;; GW KK7Q;?C 9 9 9OO%%E!T))'4//!--d.EFFF'2243JKKK*.' $)!*.*;*G*G*I*I''001HIII!--d.BCCC KKMMMMM3  rr).r< __module__ __qualname____doc__r=rsrwrzr|rIr_rrarVrrXrrcrerLrrrrrPrrrrrTrZr[r]rrrrrrrrrrrrrrrrr r esC7C7C7J''' ''' %%% %%%        """""" ###      !!!0 % % % & & &    0"                 !!! """  ,,,.%%%N%%% "AAAF2      rr chddlm}ddlm}m}m}ddl}ddl}tj d}tj dddd }t|d}|}| ||} | d |} | | |} | | || ||dS) 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 rwxVTKRenderWindowConeExamplerWsN;:::::TTTTTTTTTT)))))))) &--C HT22 C C CE ub ) )F +--C ((--- =??Dq""$$J!!$"4"4"6"6777 I  ###LL JJLLLLLNNNNNr__main__)r9rossysr0rCrrrrr;r wx.glcanvasr>rrr rWr<rrrr\s   >B SSSSSSSSSS I ;+ $I {k) x x x x x  x x x x###J z  """""r